Skip to content

Commit

Permalink
Sync eng/common directory with azure-sdk-tools for PR 1088 (#14482)
Browse files Browse the repository at this point in the history
* Add pipeline configuration for cleaning up upstream branches

* Fix powershell naming

* Switch to use github API

* Use github API to delete branches

Co-authored-by: Chidozie Ononiwu <chononiw@microsoft.com>
  • Loading branch information
azure-sdk and chidozieononiwu authored Oct 14, 2020
1 parent d2cfea4 commit 23e3963
Show file tree
Hide file tree
Showing 2 changed files with 106 additions and 0 deletions.
44 changes: 44 additions & 0 deletions eng/common/scripts/Delete-RemoteBranches.ps1
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
param(
$RepoOwner,
$RepoName,
$BranchPrefix,
$AuthToken
)

. "${PSScriptRoot}\common.ps1"

LogDebug "Operating on Repo [ $RepoName ]"
try{
$branches = (List-References -RepoOwner $RepoOwner -RepoName $RepoName -Ref "heads/$BranchPrefix").ref
}
catch {
LogError "List-References failed with exception:`n$_"
exit 1
}

foreach ($branch in $branches)
{
try {
$branchName = $branch.Replace("refs/heads/","")
$head = "${RepoOwner}/${RepoName}:${branchName}"
LogDebug "Operating on branch [ $branchName ]"
$pullRequests = List-PullRequests -RepoOwner $RepoOwner -RepoName $RepoName -head $head
}
catch
{
LogError "List-PullRequests failed with exception:`n$_"
exit 1
}

if ($pullRequests.Count -eq 0)
{
LogDebug "Branch [ $branchName ] in repo [ $RepoName ] has no associated Pull Request. Deleting Branch"
try{
Delete-References -RepoOwner $RepoOwner -RepoName $RepoName -Ref ($branch.Remove(0,5)) -AuthToken $AuthToken
}
catch {
LogError "Delete-References failed with exception:`n$_"
exit 1
}
}
}
62 changes: 62 additions & 0 deletions eng/common/scripts/Invoke-GitHubAPI.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,24 @@ function Invoke-GitHubAPIPatch {
return $resp
}

function Invoke-GitHubAPIDelete {
param (
[Parameter(Mandatory = $true)]
$apiURI,
[Parameter(Mandatory = $true)]
$token
)

$resp = Invoke-RestMethod `
-Method DELETE `
-Uri $apiURI `
-Headers (Get-GitHubHeaders -token $token) `
-MaximumRetryCount 3

return $resp
}


function Invoke-GitHubAPIGet {
param (
[Parameter(Mandatory = $true)]
Expand Down Expand Up @@ -105,6 +123,27 @@ function List-PullRequests {
return Invoke-GitHubAPIGet -apiURI $uri
}

#
<#
.PARAMETER Ref
Ref to search for
Pass 'heads/<branchame> ,tags/<tag name>, or nothing
#>
function List-References {
param (
[Parameter(Mandatory = $true)]
$RepoOwner,
[Parameter(Mandatory = $true)]
$RepoName,
$Ref
)

$uri = "$GithubAPIBaseURI/$RepoOwner/$RepoName/git/matching-refs/"
if ($Ref) { $uri += "$Ref" }

return Invoke-GitHubAPIGet -apiURI $uri
}

function Add-IssueComment {
param (
[Parameter(Mandatory = $true)]
Expand Down Expand Up @@ -229,4 +268,27 @@ function Update-Issue {
}

return Invoke-GitHubAPIPatch -apiURI $uri -body $parameters -token $AuthToken
}

function Delete-References {
param (
[Parameter(Mandatory = $true)]
$RepoOwner,
[Parameter(Mandatory = $true)]
$RepoName,
[ValidateNotNullOrEmpty()]
[Parameter(Mandatory = $true)]
$Ref,
[Parameter(Mandatory = $true)]
$AuthToken
)

if ($Ref.Trim().Length -eq 0)
{
throw "You must supply a valid 'Ref' Parameter to 'Delete-Reference'."
}

$uri = "$GithubAPIBaseURI/$RepoOwner/$RepoName/git/refs/$Ref"

return Invoke-GitHubAPIDelete -apiURI $uri -token $AuthToken
}

0 comments on commit 23e3963

Please sign in to comment.