Public/Set-LabSnippet.ps1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
function Set-LabSnippet
{
    [CmdletBinding()]
    param
    (
        [Parameter(Mandatory, ValueFromPipeline, ValueFromPipelineByPropertyName)]
        [string]
        $Name,

        [string[]]
        $DependsOn,

        [ValidateSet('Sample','Snippet', 'CustomRole')]
        [string]
        $Type,

        [string[]]
        $Tag,

        [scriptblock]
        $ScriptBlock,

        [switch]
        $NoExport
    )

    process
    {
        $schnippet = Get-LabSnippet -Name $Name
        if (-not $schnippet)
        {
            Write-PSFMessage -Level Warning -Message "Snippet $Name not found"
            break
        }

        if (-not $Tag)
        {
            $Tag = $schnippet.Tag
        }

        foreach ($dependency in $DependsOn)
        {
            if ($Tag -contains "DependsOn_$($dependency)") { Continue }

            $Tag += "DependsOn_$($dependency)"
        }

        if (-not $Description)
        {
            $Description = $schnippet.Description
        }

        if (-not $ScriptBlock)
        {
            $ScriptBlock = $schnippet.ScriptBlock
        }

        Set-PSFScriptblock -Name $Name -Description $Description -Tag $Tag -Scriptblock $ScriptBlock -Global
        
        if ($NoExport) { return }

        Export-LabSnippet -Name $Name -DependsOn $DependsOn
    }
}