Sync eng/common directory with azure-sdk-tools repository for Tools PR 989 (#647)

This commit is contained in:
Azure SDK Bot 2020-09-23 18:12:39 -07:00 committed by GitHub
parent fdcb150621
commit 7d9fd0de57
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 109 additions and 114 deletions

View File

@ -15,6 +15,7 @@ parameters:
ScriptDirectory: eng/common/scripts
GHReviewersVariable: ''
GHTeamReviewersVariable: ''
GHAssignessVariable: ''
# Multiple labels seperated by comma, e.g. "bug, APIView"
PRLabels: ''
@ -75,19 +76,6 @@ steps:
-PRTitle "${{ parameters.PRTitle }}"
-PRBody "${{ coalesce(parameters.PRBody, parameters.CommitMsg, parameters.PRTitle) }}"
-PRLabels "${{ parameters.PRLabels}}"
- task: PowerShell@2
displayName: Tag a Reviewer on PR
condition: and(succeeded(), eq(variables['HasChanges'], 'true'))
continueOnError: true
inputs:
pwsh: true
workingDirectory: ${{ parameters.WorkingDirectory }}
filePath: ${{ parameters.ScriptDirectory }}/add-pullrequest-reviewers.ps1
arguments: >
-RepoOwner "${{ parameters.RepoOwner }}"
-RepoName "$(RepoNameWithoutOwner)"
-AuthToken "$(azuresdk-github-pat)"
-GitHubUsers "$(${{ parameters.GHReviewersVariable }})"
-GitHubTeams "$(${{ parameters.GHTeamReviewersVariable }})"
-PRNumber "$(Submitted.PullRequest.Number)"
-UserReviewers "$(${{ parameters.GHReviewersVariable }})"
-TeamReviewers "$(${{ parameters.GHTeamReviewersVariable }})"
-Assignees "$(${{ parameters.GHAssignessVariable }})"

View File

@ -49,40 +49,64 @@ param(
[string]$PRBody = $PRTitle,
[Parameter(Mandatory = $false)]
[string]$PRLabels
[string]$PRLabels,
[Parameter(Mandatory = $false)]
[string]$UserReviewers,
[Parameter(Mandatory = $false)]
[string]$TeamReviewers,
[Parameter(Mandatory = $false)]
[string]$Assignees
)
$headers = @{
Authorization = "bearer $AuthToken"
}
$baseURI = "https://api.github.com/repos"
function SplitMembers ($membersString)
{
return @($membersString.Split(",") | % { $_.Trim() } | ? { return $_ })
}
function InvokeGitHubAPI($apiURI, $method, $body) {
$resp = Invoke-RestMethod -Method $method -Headers $headers -Body ($body | ConvertTo-Json) -Uri $apiURI -MaximumRetryCount 3
Write-Host -f green "These members have been added to: https://github.com/$RepoOwner/$RepoName/pull/$prNumber"
($body | Format-List | Write-Output)
$resp | Write-Verbose
}
function AddReviewers ($prNumber, $users, $teams) {
$uri = "$baseURI/$RepoOwner/$RepoName/pulls/$prNumber/requested_reviewers"
$userAdditions = SplitMembers -membersString $users
$teamAdditions = SplitMembers -membersString $teams
$postResp = @{}
if ($userAdditions) {
$postResp["reviewers"] = @($userAdditions)
}
if ($teamAdditions) {
$postResp["team_reviewers"] = @($teamAdditions)
}
return InvokeGitHubAPI -apiURI $uri -method 'Post' -body $postResp
}
function AddLabelsAndOrAssignees ($prNumber, $labels, $assignees) {
$uri = "$baseURI/$RepoOwner/$RepoName/issues/$prNumber"
$labelAdditions = SplitMembers -membersString $labels
$assigneeAdditions = SplitMembers -membersString $assignees
$postResp = @{}
if ($assigneeAdditions) {
$postResp["assignees"] = @($assigneeAdditions)
}
if ($labelAdditions) {
$postResp["labels"] = @($labelAdditions)
}
return InvokeGitHubAPI -apiURI $uri -method 'Post' -body $postResp
}
$query = "state=open&head=${PROwner}:${PRBranch}&base=${BaseBranch}"
function AddLabels([int] $prNumber, [string] $prLabelString)
{
# Adding labels to the pr.
if (-not $prLabelString) {
Write-Verbose "There are no labels added to the PR."
return
}
# Parse the labels from string to array
$prLabelArray = @($prLabelString.Split(",") | % { $_.Trim() } | ? { return $_ })
$prLabelUri = "https://api.github.com/repos/$RepoOwner/$RepoName/issues/$prNumber"
$labelRequestData = @{
labels = $prLabelArray
}
try {
$resp = Invoke-RestMethod -Method PATCH -Headers $headers $prLabelUri -Body ($labelRequestData | ConvertTo-Json)
}
catch {
Write-Error "Invoke-RestMethod $prLabelUri failed with exception:`n$_"
}
$resp | Write-Verbose
Write-Host -f green "Label(s) [$prLabelArray] added to pull request: https://github.com/$RepoOwner/$RepoName/pull/$prNumber"
}
try {
$resp = Invoke-RestMethod -Headers $headers "https://api.github.com/repos/$RepoOwner/$RepoName/pulls?$query"
}
@ -93,11 +117,18 @@ catch {
$resp | Write-Verbose
if ($resp.Count -gt 0) {
try {
Write-Host -f green "Pull request already exists $($resp[0].html_url)"
# setting variable to reference the pull request by number
Write-Host "##vso[task.setvariable variable=Submitted.PullRequest.Number]$($resp[0].number)"
AddLabels $resp[0].number $PRLabels
AddReviewers -prNumber $resp[0].number -users $UserReviewers -teams $TeamReviewers
AddLabelsAndOrAssignees -prNumber $resp[0].number -labels $PRLabels -assignees $Assignees
}
catch {
Write-Error "Call to GitHub API failed with exception:`n$_"
exit 1
}
}
else {
$data = @{
@ -112,17 +143,17 @@ else {
$resp = Invoke-RestMethod -Method POST -Headers $headers `
"https://api.github.com/repos/$RepoOwner/$RepoName/pulls" `
-Body ($data | ConvertTo-Json)
$resp | Write-Verbose
Write-Host -f green "Pull request created https://github.com/$RepoOwner/$RepoName/pull/$($resp.number)"
# setting variable to reference the pull request by number
Write-Host "##vso[task.setvariable variable=Submitted.PullRequest.Number]$($resp.number)"
AddReviewers -prNumber $resp.number -users $UserReviewers -teams $TeamReviewers
AddLabelsAndOrAssignees -prNumber $resp.number -labels $PRLabels -assignees $Assignees
}
catch {
Write-Error "Invoke-RestMethod [https://api.github.com/repos/$RepoOwner/$RepoName/pulls] failed with exception:`n$_"
Write-Error "Call to GitHub API failed with exception:`n$_"
exit 1
}
$resp | Write-Verbose
Write-Host -f green "Pull request created https://github.com/$RepoOwner/$RepoName/pull/$($resp.number)"
# setting variable to reference the pull request by number
Write-Host "##vso[task.setvariable variable=Submitted.PullRequest.Number]$($resp.number)"
AddLabels $resp.number $PRLabels
}
}

View File

@ -1,61 +0,0 @@
param(
[Parameter(Mandatory = $true)]
$RepoOwner,
[Parameter(Mandatory = $true)]
$RepoName,
[Parameter(Mandatory = $false)]
$GitHubUsers = "",
[Parameter(Mandatory = $false)]
$GitHubTeams = "",
[Parameter(Mandatory = $true)]
$PRNumber,
[Parameter(Mandatory = $true)]
$AuthToken
)
function AddMembers($memberName, $additionSet) {
$headers = @{
Authorization = "bearer $AuthToken"
}
$uri = "https://api.github.com/repos/$RepoOwner/$RepoName/pulls/$PRNumber/requested_reviewers"
$errorOccurred = $false
foreach ($id in $additionSet) {
try {
$postResp = @{}
$postResp[$memberName] = @($id)
$postResp = $postResp | ConvertTo-Json
Write-Host $postResp
$resp = Invoke-RestMethod -Method Post -Headers $headers -Body $postResp -Uri $uri -MaximumRetryCount 3
$resp | Write-Verbose
}
catch {
Write-Host "Error attempting to add $user `n$_"
$errorOccurred = $true
}
}
return $errorOccurred
}
# at least one of these needs to be populated
if (-not $GitHubUsers -and -not $GitHubTeams) {
Write-Host "No user provided for addition, exiting."
exit 0
}
$userAdditions = @($GitHubUsers.Split(",") | % { $_.Trim() } | ? { return $_ })
$teamAdditions = @($GitHubTeams.Split(",") | % { $_.Trim() } | ? { return $_ })
$errorsOccurredAddingUsers = AddMembers -memberName "reviewers" -additionSet $userAdditions
$errorsOccurredAddingTeams = AddMembers -memberName "team_reviewers" -additionSet $teamAdditions
if ($errorsOccurredAddingUsers -or $errorsOccurredAddingTeams) {
exit 1
}

View File

@ -0,0 +1,37 @@
param (
[Parameter(Mandatory = $true)]
[string]$RepoOwner,
[Parameter(Mandatory = $true)]
[string]$RepoName,
[Parameter(Mandatory = $true)]
$PullRequestNumber,
[Parameter(Mandatory = $true)]
$VsoPRCreatorVariable,
[Parameter(Mandatory = $false)]
$AuthToken
)
$headers = @{ }
if ($AuthToken) {
$headers = @{
Authorization = "bearer $AuthToken"
}
}
try
{
$prApiUrl = "https://api.github.com/repos/$RepoOwner/$RepoName/pulls/${PullRequestNumber}"
$response = Invoke-RestMethod -Headers $headers $prApiUrl
Write-Host "##vso[task.setvariable variable=$VsoPRCreatorVariable;]$($response.user.login)"
}
catch
{
Write-Error "Invoke-RestMethod ${prApiUrl} failed with exception:`n$_"
exit 1
}