import mof powershell sdk update

Author: avkaur

src/ResourceManager/Automation/Commands.Automation/Cmdlet/ImportAzureAutomationDscNodeConfiguration.cs

@@ -0,0 +1,71 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using System.Management.Automation;
+using System.Security.Permissions;
+using Microsoft.Azure.Commands.Automation.Common;
+using Microsoft.Azure.Commands.Automation.Model;
+using Microsoft.WindowsAzure.Commands.Utilities.Common;
+
+namespace Microsoft.Azure.Commands.Automation.Cmdlet
+{
+    /// <summary>
+    /// Imports dsc node configuration script
+    /// </summary>
+    [Cmdlet(VerbsData.Import, "ImportAzureAutomationDscNodeConfiguration")]
+    [OutputType(typeof(DscConfiguration))]
+    public class ImportAzureAutomationDscNodeConfiguration : AzureAutomationBaseCmdlet
+    {       
+        /// <summary>
+        /// Gets or sets the source path.
+        /// </summary>
+        [Parameter(Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "Path to the node configuration script .mof to import.")]
+        [Alias("Path")]
+        [ValidateNotNullOrEmpty]
+        public string SourcePath { get; set; }
+
+        /// <summary>
+        /// Gets or sets the node configuration name.
+        /// </summary>
+        [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The dsc node configuration name for which .mof is imported.")]
+        [Alias("Name")]
+        public string Name { get; set; }
+
+        /// <summary>
+        /// Gets or sets the configuration name for the node configuration.
+        /// </summary>
+        [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The dsc configuration name for .mof imported")]
+        [Alias("ConfigurationName")]
+        public string ConfigurationName { get; set; }
+        
+        /// <summary>
+        /// Execute this cmdlet.
+        /// </summary>
+        [PermissionSet(SecurityAction.Demand, Name = "FullTrust")]
+        public override void ExecuteCmdlet()
+        {
+            var nodeConfiguration = this.AutomationClient.CreateNodeConfiguration(
+                    this.ResourceGroupName,
+                    this.AutomationAccountName,
+                    this.SourcePath,
+                    this.Name,
+                    this.ConfigurationName);
+
+            this.WriteObject(nodeConfiguration);
+        }
+    }
+}

src/ResourceManager/Automation/Commands.Automation/Commands.ResourceManagement.Automation.csproj

@@ -117,6 +117,7 @@
   </ItemGroup>
   <ItemGroup>
     <Compile Include="Cmdlet\AzureAutomationBaseCmdlet.cs" />
+    <Compile Include="Cmdlet\ImportAzureAutomationDscNodeConfiguration.cs" />
     <Compile Include="Cmdlet\ExportAzureAutomationDscConfiguration.cs" />
     <Compile Include="Cmdlet\ExportAzureAutomationDscNodeReportContent.cs" />
     <Compile Include="Cmdlet\GetAzureAutomationCertificate.cs" />

src/ResourceManager/Automation/Commands.Automation/Common/AutomationClientDSC.cs

@@ -1118,6 +1118,63 @@ public Model.NodeConfiguration GetNodeConfiguration(string resourceGroupName, st
             }
         }
 
+        public Model.NodeConfiguration CreateNodeConfiguration(
+            string resourceGroupName,
+            string automationAccountName,
+            string sourcePath,
+            string nodeConfiguraionName,
+            string configurationName)
+        {
+            using (var request = new RequestSettings(this.automationManagementClient))
+            {
+                Requires.Argument("ResourceGroupName", resourceGroupName).NotNull();
+                Requires.Argument("AutomationAccountName", automationAccountName).NotNull();
+                Requires.Argument("SourcePath", sourcePath).NotNull();
+
+                string fileContent = null;
+                string nodeConfigurationName = String.Empty;
+
+                try
+                {
+                    if (File.Exists(Path.GetFullPath(sourcePath)))
+                    {
+                        fileContent = System.IO.File.ReadAllText(sourcePath);
+                    }
+                }
+                catch (Exception)
+                {
+                    // exception in accessing the file path
+                    throw new FileNotFoundException(
+                                        string.Format(
+                                            CultureInfo.CurrentCulture,
+                                            Resources.ConfigurationSourcePathInvalid));
+                }
+
+                var nodeConfigurationCreateParameters = new DscNodeConfigurationCreateOrUpdateParameters()
+                {
+                    Name = nodeConfiguraionName,
+                    Source = new Microsoft.Azure.Management.Automation.Models.ContentSource()
+                    {
+                        // only embeddedContent supported for now
+                        ContentType = Model.ContentSourceType.embeddedContent.ToString(),
+                        Value = fileContent
+                    },
+                    Configuration = new DscConfigurationAssociationProperty() 
+                    {
+                        Name = configurationName
+                    }
+                };
+
+                var configuration =
+                    this.automationManagementClient.NodeConfigurations.CreateOrUpdate(
+                        resourceGroupName,
+                        automationAccountName,
+                        nodeConfigurationCreateParameters).NodeConfiguration;
+
+                return new Model.NodeConfiguration(resourceGroupName, automationAccountName, configuration);
+            }
+        }
+
         #endregion
 
         #region dsc reports

src/ResourceManager/Automation/Commands.Automation/Common/IAutomationClient.cs

@@ -59,6 +59,8 @@ public interface IAutomationClient
         IEnumerable<NodeConfiguration> ListNodeConfigurationsByConfigurationName(string resourceGroupName, string automationAccountName, string configurationName, string rollupStatus);
 
         IEnumerable<NodeConfiguration> ListNodeConfigurations(string resourceGroupName, string automationAccountName, string rollupStatus);
+
+        NodeConfiguration CreateNodeConfiguration(string resourceGroupName, string automationAccountName, string sourcePath, IDictionary tags);
         #endregion
 
         #region Configurations