classes/RestMethod/FormRestMethod.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
using module ..\AtlassianContext.psm1
using module .\RestMethodQueryParams.psm1
using module .\RestMethod.psm1

class FormRestMethod : RestMethod {

    #####################
    # HIDDEN PROPERTIES #
    #####################
    
    #####################
    # PUBLIC PROPERTIES #
    #####################

    # Form values for a multipart request
    [hashtable]
    $Form

    ################
    # CONSTRUCTORS #
    ################

    #form only
    FormRestMethod(
        [string]$FunctionPath,
        [string]$HttpMethod,
        [hashtable]$Form
    ) : base($FunctionPath,$HttpMethod) {
        $this.FormInit($Form)
    }

    #form + query
    FormRestMethod(
        [string]$FunctionPath,
        [string]$HttpMethod,
        [RestMethodQueryParams]$Query,
        [hashtable]$Form
    ) : base($FunctionPath,$HttpMethod,$Query) {
        $this.FormInit($Form)
    }

    ##################
    # HIDDEN METHODS #
    ##################

    hidden
    [void]
    FormInit([hashtable]$Form){
        $this.Form = $Form
        if(!$this.Headers.ContainsKey("X-Atlassian-Token")) {
            $this.Headers.Add("X-Atlassian-Token","no-check")
        }
    }

    hidden
    [string]
    BodyLines([string]$Boundary) {
        $fileName = $this.Form.file.Name
        $fileBytes = [System.IO.File]::ReadAllBytes($this.Form.file);
        $fileEnc = [System.Text.Encoding]::GetEncoding('ISO-8859-1').GetString($fileBytes);
        $LF = "`r`n";

        return ( 
            "--$Boundary",
            "Content-Disposition: form-data; name=`"file`"; filename=`"$fileName`"",
            "Content-Type: application/octet-stream$LF",
            $fileEnc,
            "--$Boundary--$LF" 
        ) -join $LF
    }

    ##################
    # PUBLIC METHODS #
    ##################

    [object]
    Invoke(
        [AtlassianContext]$AtlassianContext
    ){
        $AtlassianContext = [RestMethod]::FillContext($AtlassianContext)
        $boundary = [System.Guid]::NewGuid().ToString()
        $bodyLines = $this.BodyLines($boundary)
        $invokeSplat = @{
            Uri = $this.Uri($AtlassianContext)
            Method = $this.HttpMethod
            Headers = $this.HeadersToSend($AtlassianContext) 
            MaximumRetryCount = $AtlassianContext.Retries
            RetryIntervalSec = $AtlassianContext.RetryDelay
            ContentType = "multipart/form-data; boundary=`"$boundary`""
            Body = $bodyLines
        }
        return [RestMethod]::RootInvoke($invokeSplat)
    }
}