Apply-AssemblyVersionAndDefaults.ps1

param (
    [Parameter(Mandatory = $true)] [string] $buildNumber,
    [Parameter(Mandatory = $true)] [string] $solutionDirectory,
    [Parameter(Mandatory = $true)] [string] $UKHOAssemblyCompany,
    [Parameter(Mandatory = $true)] [string] $UKHOAssemblyCopyright,
    [Parameter(Mandatory = $true)] [string] $UKHOAssemblyVersionPrefix,
    [Parameter(Mandatory = $true)] [string] $UKHOAssemblyProduct,
    [Parameter(Mandatory = $true)] [string] $SourceRevisionId
)

Write-Host "Build number: " $buildNumber

#UKHO.file-share-dotnet-client_merge_20210428.5
$buildNumberRegex = "(.+)_202([0-9]{3,5})\.([0-9]{1,2})"
$validBuildNumber = $buildNumber -match $buildNumberRegex

if ($validBuildNumber -eq $false) {
    $errorMessage = "Build number passed in must be in the following format: (BuildDefinitionName)_.(date:yyyyMMdd)(rev:.r)"
    Write-Error $errorMessage
    throw $errorMessage
}

# Magic var $Matches comes from the above regex match statement: $buildNumber -match $buildNumberRegex
$buildRevisionNumber = $Matches.2 + "." + $Matches.3
$versionToApply = $UKHOAssemblyVersionPrefix + $buildRevisionNumber
Write-Host "##vso[task.setvariable variable=NuGetVersion;isOutput=true]$($versionToApply)"

$assemblyValues = @{
    "Company"           = $UKHOAssemblyCompany;
    "Copyright"         = $UKHOAssemblyCopyright;
    "Description"       = $UKHOAssemblyProduct;
    "Product"           = $UKHOAssemblyProduct;
    "AssemblyVersion"   = $versionToApply;
    "FileVersion"       = $versionToApply;
    "Version"           = $versionToApply;
    "SourceRevisionId"  = $SourceRevisionId;
}

function UpdateOrAddAttribute($xmlContent, $assemblyKey, $newValue, $namespace) {
    $propertyGroup = $xmlContent.Project.PropertyGroup
    if ($propertyGroup -is [array]) {
        $propertyGroup = $propertyGroup[0]
    }

    $propertyGroupNode = $propertyGroup.$assemblyKey

    if ($null -ne $propertyGroupNode) {
        Write-Host "Assembly key $assemblyKey has been located in source file - updating with value: " $newValue
        $propertyGroup.$assemblyKey = $newValue
        return $xmlContent
    }

    Write-Host "Assembly key $assemblyKey could not be located in source file - appending value " $newValue

    $newChild = $xmlContent.CreateElement($assemblyKey, $namespace)
    $newChild.InnerText = $newValue
    $propertyGroup.AppendChild($newChild)

    return $propertyGroupNode
}

(Get-ChildItem -Path $solutionDirectory -File -Filter "*.csproj" -Recurse) | ForEach-Object {
    $file = $_

    Write-Host "Updating assembly file at path: $file"
    [xml]$xmlContent = (Get-Content $file.FullName)

    $assemblyValues.Keys | ForEach-Object {
        $key = $_

        UpdateOrAddAttribute $xmlContent $key $assemblyValues[$key] $xmlContent.DocumentElement.NamespaceURI
    }

    $xmlContent.Save($file.FullName)
}