ConvertTo-SSMDocument.ps1

ConvertTo-SSMDocument.ps1

function ConvertTo-SSMDocument
{
<#
.Synopsis
    Convert a script into SSM Command document
.DESCRIPTION
    This cmdlet convert a script into SSM Command document.
.EXAMPLE
PS:\ > ConvertTo-SSMDocument -Expression 'Get-Service'
{
  "schemaVersion": "2.2",
  "description": "SSM Command Document",
  "parameters": {
    "Message": {
      "type": "String",
      "description": "SSM parameter description",
      "default": "hello hello"
    }
  },
  "mainSteps": [
    {
      "action": "aws:runPowerShellScript",
      "name": "example",
      "inputs": {
        "runCommand": [
            "Get-Service"
        ]
      }
    }
  ]
}
.EXAMPLE
$scriptpath = 'C:\temp\demo.ps1'
ConvertTo-SSMDocument -Path $scriptpath
{
    "schemaVersion": "2.2",
    "description": "SSM Command Document",
    "parameters": {
      "Message": {
        "type": "String",
        "description": "SSM parameter description",
        "default": "hello hello"
      }
    },
    "mainSteps": [
      {
        "action": "aws:runPowerShellScript",
        "name": "example",
        "inputs": {
          "runCommand": [
              "if (-Not (Get-Module -name 'ActiveDirectory' -ErrorAction stop -ListAvailable)){",
  "    Add-WindowsFeature -Name \"RSAT-AD-PowerShell\" -IncludeAllSubFeature -verbose",
  "}"
          ]
        }
      }
    ]
}
.NOTES
    Author: Lawrence Hwang
#>
    [cmdletbinding(DefaultParameterSetName = 'Expression')]
    param(
        # Convert an existing script to SSM document.
        [parameter(Mandatory = $true,
            ParameterSetName = 'File')]
        [string]$Path,

        # Convert a one liner to SSM document.
        [parameter(Mandatory = $true,
            ParameterSetName = 'Expression')]
        [string]$Expression,

        #SSM document description
        [parameter(Mandatory = $false,
            ParameterSetName = 'File')]
        [parameter(Mandatory = $false,
            ParameterSetName = 'Expression')]
        [string]$SSMDocDescription = 'SSM Command Document',

        # SSM Document type. Currently only support Command type.
        [parameter(Mandatory = $false,
            ParameterSetName = 'File')]
        [parameter(Mandatory = $false,
            ParameterSetName = 'Expression')]
        [string]$DocumentType = 'Command'

    )

    switch ($PSCmdlet.ParameterSetName)
    {
        'File' {$RawContent = Get-Content $Path}
        'Expression' {$RawContent = $Expression}
    }


    $RawContent = $RawContent.replace('\', '\\').replace('"', '\"')
    $powershellscript = [string]::Join("`",`n`"", $RawContent)
    $powershellscript = '"' + $powershellscript + '"'
    $ssmdoc = @"
{
  "schemaVersion": "2.2",
  "description": "$SSMDocDescription",
  "parameters": {
    "Message": {
      "type": "String",
      "description": "SSM parameter description",
      "default": "hello hello"
    }
  },
  "mainSteps": [
    {
      "action": "aws:runPowerShellScript",
      "name": "example",
      "inputs": {
        "runCommand": [
            $powershellscript
        ]
      }
    }
  ]
}
"@

    $ssmdoc
}