TFSPowershell.Utils.psm1

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
$rootPath = $PSScriptRoot

# .ExternalHelp .\MAML\TFSPowershell.Utils.Help.xml
Function Get-RestApiCredentials
{
    [CmdletBinding()]
    Param ()
    $credFile = "$rootPath\restcreds`_$env:COMPUTERNAME.xml"
    
    try
    {
        $credentials = Import-Clixml $credFile
    }
    catch 
    {
        $exception = $_.Exception.Message
        "Credentials file: $credFile. Exception: $exception" | Out-File "c:\temp\log.txt"
        Write-Verbose "The credential file $credFile was not found. Use Set-RestApiCredentials to save the credentials."
    
        $credentials = $null
    }
    
    return $credentials
}

# .ExternalHelp .\MAML\TFSPowershell.Utils.Help.xml
Function Set-RestApiCredentials
{
    [CmdletBinding()]
    Param()
    $credentials = Get-Credential
    $username = $credentials.UserName
    $user = $username.Replace("\","-")

    $fileName = "restcreds`_$env:COMPUTERNAME.xml"
    $credentials | Export-Clixml "$rootPath\$fileName"
}

# .ExternalHelp .\MAML\TFSPowershell.Utils.Help.xml
Function Clear-RestAPICredentials
{
    $credFile = "$rootPath\restcreds`_$env:COMPUTERNAME.xml"
    if (Test-Path $credFile)
    {
        Remove-Item $credFile -Force
    }
    else 
    {
        Write-Verbose "No credentials found. No operation done."    
    }
}

# .ExternalHelp .\MAML\TFSPowershell.Utils.Help.xml
Function Set-TFSPowershellTempDir
{
    [CmdletBinding()]
    Param(
        [Parameter(mandatory=$true)]
        [string] $TFSPowershellTempDirLocation
    )
    Write-Verbose "Creating environment variable 'TFSPowershellTempDirLocation'..."
    [Environment]::SetEnvironmentVariable("TFSPowershellTempDirLocation", $TFSPowershellTempDirLocation, "Machine")
}

# .ExternalHelp .\MAML\TFSPowershell.Utils.Help.xml
Function Invoke-RestAPICall
{
    [CmdletBinding()]
    Param(
        [Parameter(mandatory=$true)]
        [string] $Uri,
        [Parameter(mandatory=$true)]
        [string] $Method,
        $Headers,
        $Body
    )

    $command = "Invoke-RestMethod -Uri ""$Uri"" -Method ""$Method"" -ContentType application/json "
    $credentials = Get-RestApiCredentials

    #Selecting authentication method
    if (!$credentials)
    {
        Write-Verbose "No rest api credentials found. The default credentials will be considered to the call."
        $command += "-UseDefaultCredentials "
    }
    else 
    {
        if ([regex]::match($Uri, "https:\/\/.*.visualstudio\.com\/.*", [System.Text.RegularExpressions.RegexOptions]::IgnoreCase).Success)
        {
            $base64authinfo = [Convert]::ToBase64String([Text.Encoding]::ASCII.GetBytes(("{0}:{1}" -f $credentials.UserName, $credentials.GetNetworkCredential().Password)))
            $Headers = @{Authorization=("Basic {0}" -f $base64authinfo)}

            $command += '-Headers $Headers '
        }
        else 
        {
            $command += '-Credential $credentials '
        }
    }
    if ($Method -ne "GET" -and $Method -ne "DELETE")
    {
        $command += '-Body $Body '
    }

    Write-Verbose $command 
    return Invoke-Expression $command
}