Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
PowerShell Script for Deploying Cosmos DB Server-Side Scripts
function GenerateCosmosDBAuthToken
{
param(
[Parameter(Mandatory=$true)]
[string]
$Verb,
[Parameter(Mandatory=$true)]
[string]
$ResourceType,
[Parameter(Mandatory=$true)]
[string]
$ResourceId,
[Parameter(Mandatory=$true)]
[string]
$Date,
[Parameter(Mandatory=$true)]
[string]
$Key,
[Parameter(Mandatory=$false)]
[string]
$KeyType = 'master',
[Parameter(Mandatory=$false)]
[string]
$TokenVersion = '1.0'
)
$payload = "$($verb.ToLowerInvariant())`n$($resourceType.ToLowerInvariant())`n$resourceId`n$($date.ToLowerInvariant())`n`n"
$hmacSha256 = New-Object System.Security.Cryptography.HMACSHA256
$hmacSha256.Key = [System.Convert]::FromBase64String($key)
$encoding = [System.Text.Encoding]::UTF8
$payloadHash = $hmacSha256.ComputeHash($encoding.GetBytes($payload))
$signature = [System.Convert]::ToBase64String($payloadHash)
return [System.Web.HttpUtility]::UrlEncode("type=$keyType&ver=$tokenVersion&sig=$signature")
}
function SubmitCosmosDbApiRequest
{
param(
[Parameter(Mandatory=$true)]
[string]
$Verb,
[Parameter(Mandatory=$true)]
[string]
$ResourceId,
[Parameter(Mandatory=$true)]
[string]
$ResourceType,
[Parameter(Mandatory=$true)]
[string]
$Url,
[Parameter(Mandatory=$true)]
[string]
$BodyJson,
[Parameter(Mandatory=$true)]
[string]
$Key
)
$date = (Get-Date).ToUniversalTime().ToString('r')
$authToken = GenerateCosmosDBAuthToken -Verb $Verb -ResourceType $ResourceType -ResourceId $ResourceId -Date $date -Key $Key
# Add the headers
$headers = New-Object "System.Collections.Generic.Dictionary[[String],[String]]"
$headers.Add("Authorization", $authToken)
$headers.Add("x-ms-version", '2015-08-06')
$headers.Add("x-ms-date", $date)
# Send the request and handle the result
try
{
Invoke-RestMethod $Url `
-Headers $headers `
-Method $Verb `
-ContentType 'application/json' `
-Body $BodyJson
}
catch
{
$statusCode = $_.Exception.Response.StatusCode.value__
if ($statusCode -eq 409)
{
return 'AlreadyExists'
}
else
{
throw
}
}
}
function CreateCosmosDBObject
{
param(
[Parameter(Mandatory=$true)]
[string]
$AccountName,
[Parameter(Mandatory=$true)]
[string]
$AccountKey,
[Parameter(Mandatory=$true)]
[string]
$DatabaseName,
[Parameter(Mandatory=$true)]
[string]
$CollectionName,
[Parameter(Mandatory=$true)]
[string]
[ValidateSet('Trigger', 'UDF', 'StoredProcedure')]
$ObjectType,
[Parameter(Mandatory=$true)]
[string]
$ObjectName,
[Parameter(Mandatory=$true)]
[string]
$DefinitionJson
)
if ($ObjectType -eq 'Trigger')
{
$objectResourceType = 'triggers'
}
elseif ($ObjectType -eq 'UDF')
{
$objectResourceType = 'udfs'
}
elseif ($ObjectType -eq 'StoredProcedure')
{
$objectResourceType = 'sprocs'
}
# Try creating the object
Write-Host "Installing $ObjectType '$ObjectName' to Cosmos DB collection '$CollectionName' in database '$DatabaseName'..."
$createResult = SubmitCosmosDbApiRequest `
-Verb 'POST' `
-ResourceId "dbs/$DatabaseName/colls/$CollectionName" `
-ResourceType $objectResourceType `
-Url "https://$AccountName.documents.azure.com/dbs/$DatabaseName/colls/$CollectionName/$objectResourceType" `
-Key $AccountKey `
-BodyJson $DefinitionJson
# If that failed because the object already exists, update the object
if ($createResult -eq 'AlreadyExists')
{
Write-Host "$ObjectType already exists. Updating..."
SubmitCosmosDbApiRequest `
-Verb 'PUT' `
-ResourceId "dbs/$DatabaseName/colls/$CollectionName/$objectResourceType/$ObjectName" `
-ResourceType $objectResourceType `
-Url "https://$AccountName.documents.azure.com/dbs/$DatabaseName/colls/$CollectionName/$objectResourceType/$ObjectName" `
-Key $AccountKey `
-BodyJson $DefinitionJson | Out-Null
}
}
function DeployUserDefinedFunction
{
param(
[Parameter(Mandatory=$true)]
[string]
$AccountName,
[Parameter(Mandatory=$true)]
[string]
$AccountKey,
[Parameter(Mandatory=$true)]
[string]
$DatabaseName,
[Parameter(Mandatory=$true)]
[string]
$CollectionName,
[Parameter(Mandatory=$true)]
[string]
$UserDefinedFunctionName,
[Parameter(Mandatory=$true)]
[string]
$SourceFilePath
)
# Assemble the UDF definition to send to Cosmos DB
Write-Host 'Preparing UDF...'
$sourceFileContents = Get-Content $SourceFilePath | Out-String
$definition = @{
body = $sourceFileContents
id = $UserDefinedFunctionName
}
$definitionJson = $definition | ConvertTo-Json
CreateCosmosDBObject `
-AccountName $AccountName `
-AccountKey $AccountKey `
-DatabaseName $DatabaseName `
-CollectionName $CollectionName `
-ObjectType UDF `
-ObjectName $UserDefinedFunctionName `
-Definition $definitionJson
}
function DeployStoredProcedure
{
param(
[Parameter(Mandatory=$true)]
[string]
$AccountName,
[Parameter(Mandatory=$true)]
[string]
$AccountKey,
[Parameter(Mandatory=$true)]
[string]
$DatabaseName,
[Parameter(Mandatory=$true)]
[string]
$CollectionName,
[Parameter(Mandatory=$true)]
[string]
$StoredProcedureName,
[Parameter(Mandatory=$true)]
[string]
$SourceFilePath
)
# Assemble the stored procedure definition to send to Cosmos DB
Write-Host 'Preparing stored procedure...'
$sourceFileContents = Get-Content $SourceFilePath | Out-String
$definition = @{
body = $sourceFileContents
id = $StoredProcedureName
}
$definitionJson = $definition | ConvertTo-Json
CreateCosmosDBObject `
-AccountName $AccountName `
-AccountKey $AccountKey `
-DatabaseName $DatabaseName `
-CollectionName $CollectionName `
-ObjectType StoredProcedure `
-ObjectName $StoredProcedureName `
-Definition $definitionJson
}
function DeployTrigger
{
param(
[Parameter(Mandatory=$true)]
[string]
$AccountName,
[Parameter(Mandatory=$true)]
[string]
$AccountKey,
[Parameter(Mandatory=$true)]
[string]
$DatabaseName,
[Parameter(Mandatory=$true)]
[string]
$CollectionName,
[Parameter(Mandatory=$true)]
[string]
$TriggerName,
[Parameter(Mandatory=$true)]
[string]
[ValidateSet('Pre', 'Post')]
$TriggerType,
[Parameter(Mandatory=$true)]
[string]
[ValidateSet('All', 'Create', 'Delete', 'Replace')]
$TriggerOperation,
[Parameter(Mandatory=$true)]
[string]
$SourceFilePath
)
# Assemble the trigger definition to send to Cosmos DB
Write-Host 'Preparing trigger...'
$sourceFileContents = Get-Content $SourceFilePath | Out-String
$definition = @{
body = $sourceFileContents
id = $TriggerName
triggerOperation = $TriggerOperation
triggerType = $TriggerType
}
$definitionJson = $definition | ConvertTo-Json
CreateCosmosDBObject `
-AccountName $AccountName `
-AccountKey $AccountKey `
-DatabaseName $DatabaseName `
-CollectionName $CollectionName `
-ObjectType Trigger `
-ObjectName $TriggerName `
-Definition $definitionJson
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.