DocumentDB

1.0

DocumentDB.psm1

    function Get-DocDBKey([System.String]$Verb = '',[System.String]$ResourceId = '',
            [System.String]$ResourceType = '',[System.String]$Date = '',[System.String]$masterKey = '') {
        Add-Type -AssemblyName System.Web
        $keyBytes = [System.Convert]::FromBase64String($masterKey) 
        $text = @($Verb.ToLowerInvariant() + "`n" + $ResourceType.ToLowerInvariant() + "`n" + $ResourceId + "`n" + $Date.ToLowerInvariant() + "`n" + "" + "`n")
        $body =[Text.Encoding]::UTF8.GetBytes($text)
        $hmacsha = new-object -TypeName System.Security.Cryptography.HMACSHA256 -ArgumentList (,$keyBytes) 
        $hash = $hmacsha.ComputeHash($body)
        $signature = [System.Convert]::ToBase64String($hash)
        
        return [System.Web.HttpUtility]::UrlEncode($('type=master&ver=1.0&sig=' + $signature))
    }

    function Get-UTCDate() {
        $date = $(Get-Date).ToUniversalTime()
        return $date.ToString("r", [System.Globalization.CultureInfo]::InvariantCulture)
    }

    function Get-DocDBDatabases([string]$accountName, [string]$key) {
        $BaseUri = "https://" + $accountName + ".documents.azure.com"
        $uri = $BaseUri + "/dbs"
        $headers = New-DocDBHeader -resType dbs -key $key
        $response = Invoke-RestMethod -Uri $uri -Method Get -Headers $headers
        return $response.Databases
    }

    function Get-DocDBCollections([string]$DBName, [string]$accountName, [string]$key){
        $BaseUri = "https://" + $accountName + ".documents.azure.com"
        $uri = $BaseUri + "/" + "dbs/" + $DBName + "/colls"
        $headers = New-DocDBHeader -resType colls -resourceId $("dbs/" + $DBName) -key $key
        $response = Invoke-RestMethod -Uri $uri -Method Get -Headers $headers
        return $response.DocumentCollections
   }

    function New-DocDBHeader([string]$action = "get",[string]$resType, [string]$resourceId, [String]$key) {
        $apiDate = Get-UTCDate
        $auth = Get-DocDBKey -Verb $action -ResourceType $resType -ResourceId $resourceId -Date $apiDate -masterKey $Key
        $headers = New-Object "System.Collections.Generic.Dictionary[[String],[String]]"
        $headers.Add("x-ms-date", $apiDate) 
        $headers.Add("Authorization", $auth)
        $headers.Add("x-ms-version", '2015-12-16')
        return $headers
    }

    #Post json query
    function New-DocDBQuery([switch]$NoClean, [string]$JSONQuery, [string]$DBName, [string]$collection, [string]$accountName, [string]$key){
        $BaseUri = "https://" + $accountName + ".documents.azure.com"
        $collName = "dbs/"+$DBName+"/colls/" + $collection
        $DBName = "dbs/" + $databaseName
        $headers = New-DocDBHeader -action Post -resType docs -resourceId $collName -key $key
        $headers.Add("x-ms-documentdb-is-upsert", "true")
        $uri = $BaseUri + "/" + $collName + "/docs"

        Write-host ("Calling " + $uri)
        try
            {
            $JSONQuery | ConvertFrom-Json|out-null
            }
        catch
            {
            Throw "Problem with JSON input"
            break
            }
        try
            {
            $response = Invoke-RestMethod $uri -Method Post -Body $JSONQuery -ContentType 'application/query+json' -Headers $headers
            }
        catch
            {
            Throw $_.Exception.Message
            break
            }
        if($NoClean)
            {
            return $response
            }
        else
            {
            return Get-CleanDocDBResponse -DocDBOutput $response
            }
    }

    #Post Document
    function Set-DocDBDocument{
        [CmdletBinding(DefaultParameterSetName='JSON')]
        param (
            [Parameter(Mandatory = $true, ParameterSetName = 'JSON')]
            [string]$JSONdocument,
            [string]$DBName, 
            [string]$collection,
            [string]$accountName,
            [string]$key,
            [Parameter(Mandatory = $true, ParameterSetName = 'PSO')]
            [PSCustomObject]$PSdocument
        )
        if($PSdocument)
            {
            $document = $PSdocument|ConvertTo-Json
            }
        elseif($JSONDocument)
            {
       try
            {
            $JSONDocument | ConvertFrom-Json|out-null
            }
        catch
            {
            Throw "Problem with JSON input"
            break
            }
            $document = $JSONDocument
            }
        $BaseUri = "https://" + $accountName + ".documents.azure.com"
        $collName = "dbs/"+$DBName+"/colls/" + $collection
        $DBName = "dbs/" + $databaseName
        $headers = New-DocDBHeader -action Post -resType docs -resourceId $collName -key $key
        $headers.Add("x-ms-documentdb-is-upsert", "true")
        $uri = $BaseUri + "/" + $collName + "/docs"

        Write-host ("Calling " + $uri)

        $response = Invoke-RestMethod $uri -Method Post -Body $document -ContentType 'application/json' -Headers $headers
        return $response
    }
    
    function Get-CleanDocDBResponse($DocDBOutput){
    $CleanResponse = $DocDBOutput.Documents|Select-Object * -ExcludeProperty _*
    return $CleanResponse
    }