adding changes to update a PR (#60)

* adding changes to update a pr
including a second envar to set a state, along with refactor
of main body to be multiple functions, and adding docstrings
* fixing missing variable
* Try removing maintainer can modify
* Update pull-request.py

Signed-off-by: vsoch <vsoch@users.noreply.github.com>

.github/stale.yml

@@ -0,0 +1,17 @@
+# Number of days of inactivity before an issue becomes stale
+daysUntilStale: 60
+# Number of days of inactivity before a stale issue is closed
+daysUntilClose: 7
+# Issues with these labels will never be considered stale
+exemptLabels:
+  - pinned
+  - security
+# Label to use when marking an issue as stale
+staleLabel: wontfix
+# Comment to post when marking an issue as stale. Set to `false` to disable
+markComment: >
+  This issue has been automatically marked as stale because it has not had
+  recent activity. It will be closed if no further activity occurs. Thank you
+  for your contributions.
+# Comment to post when closing a stale issue. Set to `false` to disable
+closeComment: false

CHANGELOG.md

@@ -0,0 +1,19 @@
+# CHANGELOG
+
+This is a manually generated log to track changes to the repository for each release. 
+Each section should include general headers such as **Implemented enhancements** 
+and **Merged pull requests**. All closed issued and bug fixes should be 
+represented by the pull requests that fixed them. Critical items to know are:
+
+ - renamed commands
+ - deprecated / removed commands
+ - changed defaults
+ - backward incompatible changes
+
+
+Versions correspond with GitHub releases that can be referenced with @ using actions.
+
+## [master](https://github.com/vsoch/pull-request-action/tree/master) (master)
+  - output and environment variables for PR number and return codes (1.0.5)
+  - added support for reviewer (individual and team) assignments (1.0.4)
+  - added support for maintainer can modify and assignees (1.0.3)

Dockerfile

@@ -7,8 +7,9 @@ LABEL "com.github.actions.description"="Create a pull request when a branch is c
 LABEL "com.github.actions.icon"="activity"
 LABEL "com.github.actions.color"="yellow"
 
-RUN apk --no-cache add curl wget git bash jq
-COPY pull-request.sh /pull-request.sh
+RUN apk --no-cache add python3 py3-pip git bash && \
+    pip3 install requests
+COPY pull-request.py /pull-request.py
 
-RUN chmod u+x /pull-request.sh
-ENTRYPOINT ["/pull-request.sh"]
+RUN chmod u+x /pull-request.py
+ENTRYPOINT ["python3", "/pull-request.py"]

README.md

@@ -22,14 +22,14 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: pull-request-action
-        uses: vsoch/pull-request-action@1.0.2
+        uses: vsoch/pull-request-action@1.0.6
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           BRANCH_PREFIX: "update/"
           PULL_REQUEST_BRANCH: "master"
 ```
 
-## Environment Variables
+## Environment Variable Inputs
 
 Unlike standard actions, this action just uses variables from the environment.
 
@@ -39,14 +39,56 @@ Unlike standard actions, this action just uses variables from the environment.
 | PULL_REQUEST_BRANCH | open pull request against this branch | false | master |
 | PULL_REQUEST_FROM_BRANCH | if a branch isn't found in your GitHub payload, use this branch | false | |
 | PULL_REQUEST_BODY | the body for the pull request | false | |
-| PULL REQUEST_TITLE | the title for the pull request | false | |
-| PULL REQUEST_DRAFT | should this be a draft PR? | false | false |
+| PULL_REQUEST_TITLE | the title for the pull request | false | |
+| PULL_REQUEST_DRAFT | should this be a draft PR? | false | unset |
+| MAINTAINER_CANT_MODIFY | Do not allow the maintainer to modify the PR | false | unset |
+| PULL_REQUEST_ASSIGNEES | A list (string with spaces) of users to assign | false | unset |
+| PULL_REQUEST_REVIEWERS | A list (string with spaces) of users to assign review | false | unset |
+| PULL_REQUEST_TEAM_REVIEWERS | A list (string with spaces) of teams to assign review | false | unset |
+| PASS_ON_ERROR | Instead of failing on an error response, pass | unset |
+| PASS_IF_EXISTS | Instead of failing if the pull request already exists, pass | unset |
+| PULL_REQUEST_UPDATE | If the pull request already exists, update it | unset |
+| PULL_REQUEST_STATE | If `PULL_REQUEST_UPDATE` is true, update to this state (open, closed) | open |
 
-All booleans should be lowercase.
+For `PULL_REQUEST_DRAFT`, `PASS_ON_ERROR`, `PASS_IF_EXISTS`, and `MAINTAINER_CANT_MODIFY`, these are
+treated as environment booleans. If they are defined in the environment, they trigger the
+"true" condition. E.g.,:
+
+ - Define `MAINTAINER_CANT_MODIFY` if you don't want the maintainer to be able to modify the pull request.
+ - Define `PULL_REQUEST_DRAFT` if you want the PR to be a draft.
+ - Define `PASS_ON_ERROR` if you want the PR to not exit given any non 200/201 response.
+ - Define `PASS_IF_EXISTS` if you want the PR to not exit given the pull request is already open.
+ - Define `PULL_REQUEST_UPDATE` if you want the pull request to be updated if it already exits.
+
+For `PULL_REQUEST_ASSIGNEES`, `PULL_REQUEST_REVIEWERS`, and `PULL_REQUEST_TEAM_REVIEWERS` 
+you can provide a string of one or more GitHub usernames (or team names) to
+assign to the issue. Note that only users with push access can add assigness to 
+an issue or PR, they are ignored otherwise.
 
 The `GITHUB_TOKEN` secret is required to interact and authenticate with the GitHub API to open
 the pull request. The example is [deployed here](https://github.com/vsoch/pull-request-action-example) with an example opened (and merged) [pull request here](https://github.com/vsoch/pull-request-action-example/pull/1) if needed.
 
+## Outputs
+
+The action sets a few useful output and environment variables. An output can
+be referenced later as `${{ steps.<stepname>.outputs.<output-name> }}`.
+An environment variable of course can be referenced as you usually would.
+
+| Name | Description | Environment | 
+|------|-------------|-------------|
+| pull_request_number |If the pull request is opened, this is the number for it. | PULL_REQUEST_NUMBER |
+| pull_request_url |If the pull request is opened, the html url for it. | PULL_REQUEST_URL |
+| pull_request_return_code | Return code for the pull request | PULL_REQUEST_RETURN_CODE |
+| assignees_return_code | Return code for the assignees request | ASSIGNEES_RETURN_CODE |
+| reviewers_return_code | Return code for the reviewers request | REVIEWERS_RETURN_CODE |
+
+See the [examples/outputs-example.yml](examples/outputs-example.yml) for how this works. 
+In this example, we can reference `${{ steps.pull_request.outputs.pull_request_url }}`
+in either another environment variable declaration, or within a run statement to access
+our variable `pull_request_url` that was generated in a step with id `pull_request`.
+The screenshot below shows the example in action to interact with outputs in several ways.
+
+![img/outputs.png](img/outputs.png)
 
 ## Examples
 
@@ -54,7 +96,7 @@ Example workflows are provided in [examples](examples), and please contribute an
 examples that you might have to help other users! We will walk through a basic
 example here for a niche case. Let's say that we are opening a pull request on the release event. This would mean
 that the payload's branch variable would be null. We would need to define `PULL_REQUEST_FROM`. How would
-we do that? We can [set environment variables](https://help.github.com/en/actions/reference/development-tools-for-github-actions#set-an-environment-variable-set-env) for next steps. Here is an example:
+we do that? We can [set environment variables](https://github.com/actions/toolkit/blob/main/docs/commands.md#environment-files) for next steps. Here is an example:
 
 ```yaml
 name: Pull Request on Branch Push
@@ -70,9 +112,9 @@ jobs:
         run: |
             # do custom parsing of your code / date to derive a branch from
             PR_BRANCH_FROM=release-v$(cat VERSION)
-            ::set-env name=PULL_REQUEST_FROM_BRANCH::${PR_BRANCH_FROM}
+            export "PULL_REQUEST_FROM_BRANCH=${PR_BRANCH_FROM}" >> $GITHUB_ENV
       - name: pull-request-action
-        uses: vsoch/pull-request-action@1.0.2
+        uses: vsoch/pull-request-action@1.0.6
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           PULL_REQUEST_BRANCH: "master"

action.yml

@@ -7,3 +7,14 @@ runs:
 branding:
   icon: 'activity'  
   color: 'yellow'
+outputs:
+  pull_request_number:
+    description: 'If the pull request is opened, this is the number for it.'
+  pull_request_url:
+    description: 'If the pull request is opened, the html url for it.'
+  pull_request_return_code:
+    description: 'The pull request return code.'
+  assignees_return_code:
+    description: 'The add assignees post return code.'
+  reviewers_return_code:
+    description: 'The add reviewers post return code.'

examples/assignees-example.yml

@@ -0,0 +1,17 @@
+name: Pull Request on Branch Push
+on:
+  push:
+    branches-ignore:
+      - devel
+jobs:
+  auto-pull-request:
+    name: PullRequestAction
+    runs-on: ubuntu-latest
+    steps:
+      - name: pull-request-action
+        uses: vsoch/pull-request-action@1.0.3
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          BRANCH_PREFIX: "update/"
+          PULL_REQUEST_BRANCH: "master"
+          PULL_REQUEST_ASSIGNEES: vsoch

examples/branch-from-environment.yml

@@ -0,0 +1,49 @@
+name: derive-branch-from-environment
+
+on:
+  schedule:
+    # Weekly
+    - cron: 0 0 * * 0
+
+jobs:
+  DoSomeUpdate:
+    runs-on: ubuntu-latest
+    steps:
+    - name: Checkout Repository
+      uses: actions/checkout@v1
+    - name: Install or Do Something to Change repository
+      run: |
+         echo "This is a new file." >> newfile.txt
+
+    - name: Checkout New Branch
+      env:
+        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        BRANCH_AGAINST: "master"
+      run: |
+        printf "GitHub Actor: ${GITHUB_ACTOR}\n"
+        export BRANCH_FROM="update/newfile-$(date '+%Y-%m-%d')"
+        git remote set-url origin "https://x-access-token:${GITHUB_TOKEN}@github.com/${GITHUB_REPOSITORY}.git"
+        git branch
+        git checkout -b "${BRANCH_FROM}" || git checkout "${BRANCH_FROM}"
+        git branch
+
+        git config --global user.name "github-actions"
+        git config --global user.email "github-actions@users.noreply.github.com"
+
+        git add newfile.txt
+
+        if git diff-index --quiet HEAD --; then
+           printf "No changes\n"
+        else
+           printf "Changes\n"
+           git commit -m "Automated deployment to update software database $(date '+%Y-%m-%d')"
+           git push origin "${BRANCH_FROM}"
+        fi
+        # Here is where we are setting the environment variable!
+        echo "PULL_REQUEST_FROM_BRANCH=${BRANCH_FROM}" >> $GITHUB_ENV
+
+    - name: Open Pull Request
+      uses: vsoch/pull-request-action@1.0.6
+      env:
+        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        PULL_REQUEST_BRANCH: "master"

examples/custom-body-example.yml

@@ -0,0 +1,55 @@
+name: Hotfix Branch Pull Request
+on:
+  push:
+    branches-ignore:
+      - master
+      - production
+
+# See https://github.com/vsoch/pull-request-action/issues/47#issuecomment-707109132
+
+jobs:
+  auto-pull-request:
+    name: PullRequestAction
+    runs-on: ubuntu-latest
+    steps:
+      - name: Generate branch name
+        uses: actions/github-script@v3
+        id: set-branch-name
+        with:
+          script: |
+            const capitalize = (name) => name.charAt(0).toUpperCase() + name.slice(1);
+            const emoji = context.payload.ref.startsWith("refs/heads/feature")
+              ? "✨ "
+              : context.payload.ref.startsWith("refs/heads/hotfix")
+              ? "🚑 "
+              : "";
+            return `${emoji}${capitalize(
+              context.payload.ref
+                .replace("refs/heads/", "")
+                .replace(/-/g, " ")
+                .replace("feature ", "")
+                .replace("hotfix ", "")
+            )}`;
+          result-encoding: string
+      - name: Set branch name
+        run: echo "PULL_REQUEST_TITLE=${{steps.set-branch-name.outputs.result}}" >> $GITHUB_ENV
+      - name: Generate PR body
+        uses: actions/github-script@v3
+        id: set-pr-body
+        with:
+          script: |
+            return `I'm opening this pull request for this branch, pushed by @${
+              context.payload.head_commit.author.username
+            } with ${context.payload.commits.length} commit${
+              context.payload.commits.length === 1 ? "" : "s"
+            }.`;
+          result-encoding: string
+      - name: Set PR body
+        run: echo "PULL_REQUEST_BODY=${{steps.set-pr-body.outputs.result}}" >> $GITHUB_ENV
+      - name: pull-request-action
+        uses: vsoch/pull-request-action@1.0.10
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          BRANCH_PREFIX: "hotfix-"
+          PULL_REQUEST_BRANCH: "production"
+          PULL_REQUEST_REVIEWERS: "AnandChowdhary"

examples/outputs-example.yml

@@ -0,0 +1,28 @@
+name: Pull Request on Branch Push
+on:
+  push:
+    branches-ignore:
+      - devel
+jobs:
+  auto-pull-request:
+    name: PullRequestAction
+    runs-on: ubuntu-latest
+    steps:
+      - name: pull-request-action
+        id: pull_request
+        uses: vsoch/pull-request-action@1.0.6
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          BRANCH_PREFIX: "update/"
+          PULL_REQUEST_BRANCH: "master"
+          PULL_REQUEST_REVIEWERS: vsoch
+      - name: Test outputs
+        env:
+          pull_request_number_output: ${{ steps.pull_request.outputs.pull_request_number }}
+          pull_request_url_output: ${{ steps.pull_request.outputs.pull_request_url }}
+        run: |
+          echo "Pull request number from output: ${pull_request_number_output}"
+          echo "Pull request url from output: ${pull_request_url_output}"
+          echo "Pull request number from environment: ${PULL_REQUEST_NUMBER}"
+          echo "Pull request url from environment: ${PULL_REQUEST_URL}"
+          echo "Another way to specify from output ${{ steps.pull_request.outputs.pull_request_number }}"

examples/reviewers-example.yml

@@ -0,0 +1,17 @@
+name: Pull Request on Branch Push
+on:
+  push:
+    branches-ignore:
+      - devel
+jobs:
+  auto-pull-request:
+    name: PullRequestAction
+    runs-on: ubuntu-latest
+    steps:
+      - name: pull-request-action
+        uses: vsoch/pull-request-action@1.0.4
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          BRANCH_PREFIX: "update/"
+          PULL_REQUEST_BRANCH: "master"
+          PULL_REQUEST_REVIEWERS: vsoch

pull-request.py

@@ -0,0 +1,455 @@
+#!/usr/bin/env python3
+
+import sys
+import os
+import json
+import requests
+
+################################################################################
+# Helper Functions
+################################################################################
+
+
+def get_envar(name):
+    """
+    Given a name, return the corresponding environment variable. Exit if not
+    defined, as using this function indicates the envar is required.
+
+    Parameters:
+    name (str): the name of the environment variable
+    """
+    value = os.environ.get(name)
+    if not value:
+        sys.exit("%s is required for vsoch/pull-request-action" % name)
+    return value
+
+
+def check_events_json():
+    """the github events json is required in order to indicate that we are
+    in an action environment.
+    """
+    events = get_envar("GITHUB_EVENT_PATH")
+    if not os.path.exists(events):
+        sys.exit("Cannot find Github events file at ${GITHUB_EVENT_PATH}")
+    print("Found ${GITHUB_EVENT_PATH} at %s" % events)
+    return events
+
+
+def abort_if_fail(response, reason):
+    """If PASS_ON_ERROR, don't exit. Otherwise exit with an error and print
+    the reason.
+
+    Parameters:
+    response (requests.Response) : an unparsed response from requests
+    reason                 (str) : a message to print to the user for fail.
+    """
+    message = "%s: %s: %s\n %s" % (
+        reason,
+        response.status_code,
+        response.reason,
+        response.json(),
+    )
+
+    if os.environ.get("PASS_ON_ERROR"):
+        print("Error, but PASS_ON_ERROR is set, continuing: %s" % message)
+    else:
+        sys.exit(message)
+
+
+def parse_into_list(values):
+    """A list of reviewers or assignees to parse from a string to a list
+
+    Parameters:
+    values (str) : a list of space separated, quoted values to parse to a list
+    """
+    if values:
+        values = values.replace('"', "").replace("'", "")
+    if not values:
+        return []
+    return [x.strip() for x in values.split(" ")]
+
+
+def set_env(name, value):
+    """helper function to echo a key/value pair to the environement file
+
+    Parameters:
+    name (str)  : the name of the environment variable
+    value (str) : the value to write to file
+    """
+    environment_file_path = os.environ.get("GITHUB_ENV")
+
+    with open(environment_file_path, "a") as environment_file:
+        environment_file.write("%s=%s" % (name, value))
+
+
+def open_pull_request(title, body, target, source, is_draft=False, can_modify=True):
+    """Open pull request opens a pull request with a given body and content,
+    and sets output variables. An unparsed response is returned.
+
+    Parameters:
+    title       (str) : the title to set for the new pull request
+    body        (str) : the body to set for the new pull request
+    target      (str) : the target branch
+    source      (str) : the source branch
+    is_draft   (bool) : indicate the pull request is a draft
+    can_modify (bool) : indicate the maintainer can modify
+    """
+    print("No pull request from %s to %s is open, continuing!" % (source, target))
+
+    # Post the pull request
+    data = {
+        "title": title,
+        "body": body,
+        "base": target,
+        "head": source,
+        "draft": is_draft,
+        "maintainer_can_modify": can_modify,
+    }
+    print("Data for opening pull request: %s" % data)
+    response = requests.post(PULLS_URL, json=data, headers=HEADERS)
+    if response.status_code != 201:
+        abort_if_fail(response, "Unable to create pull request")
+
+    return response
+
+
+def update_pull_request(entry, title, body, target, state=None):
+    """Given an existing pull request, update it.
+
+    Parameters:
+    entry      (dict) : the pull request metadata
+    title       (str) : the title to set for the new pull request
+    body        (str) : the body to set for the new pull request
+    target      (str) : the target branch
+    state      (bool) : the state of the PR (open, closed)
+    """
+    print("PULL_REQUEST_UPDATE is set, updating existing pull request.")
+
+    data = {
+        "title": title,
+        "body": body,
+        "base": target,
+        "state": state or "open",
+    }
+    # PATCH /repos/{owner}/{repo}/pulls/{pull_number}
+    url = "%s/%s" % (PULLS_URL, entry.get("number"))
+    print("Data for updating pull request: %s" % data)
+    response = requests.patch(url, json=data, headers=HEADERS)
+    if response.status_code != 200:
+        abort_if_fail(response, "Unable to create pull request")
+
+    return response
+
+
+def set_pull_request_groups(response):
+    """Given a response for an open or updated PR, set metadata
+
+    Parameters:
+    response (requests.Response) : a requests response, unparsed
+    """
+    # Expected return codes are 0 for success
+    pull_request_return_code = (
+        0 if response.status_code == 201 else response.status_code
+    )
+    response = response.json()
+    print("::group::github response")
+    print(response)
+    print("::endgroup::github response")
+    number = response.get("number")
+    html_url = response.get("html_url")
+    print("Number opened for PR is %s" % number)
+    set_env("PULL_REQUEST_NUMBER", number)
+    print("::set-output name=pull_request_number::%s" % number)
+    set_env("PULL_REQUEST_RETURN_CODE", pull_request_return_code)
+    print("::set-output name=pull_request_return_code::%s" % pull_request_return_code)
+    set_env("PULL_REQUEST_URL", html_url)
+    print("::set-output name=pull_request_url::%s" % html_url)
+
+
+def list_pull_requests(target, source):
+    """Given a target and source, return a list of pull requests that match
+    (or simply exit given some kind of error code)
+
+    Parameters:
+    target (str) : the target branch
+    source (str) : the source branch
+    """
+    # Check if the branch already has a pull request open
+    params = {"base": target, "head": source, "state": "open"}
+    print("Params for checking if pull request exists: %s" % params)
+    response = requests.get(PULLS_URL, params=params)
+
+    # Case 1: 404 might warrant needing a token
+    if response.status_code == 404:
+        response = requests.get(PULLS_URL, params=params, headers=HEADERS)
+    if response.status_code != 200:
+        abort_if_fail(response, "Unable to retrieve information about pull requests")
+
+    return response.json()
+
+
+def add_assignees(entry):
+    """Given a pull request metadata (from create or update) add assignees
+
+    Parameters:
+    entry (dict) : the pull request metadata
+    """
+    # Remove leading and trailing quotes
+    assignees = parse_into_list(assignees)
+    number = entry.get("number")
+
+    print(
+        "Attempting to assign %s to pull request with number %s" % (assignees, number)
+    )
+
+    # POST /repos/:owner/:repo/issues/:issue_number/assignees
+    data = {"assignees": assignees}
+    ASSIGNEES_URL = "%s/%s/assignees" % (ISSUE_URL, number)
+    response = requests.post(ASSIGNEES_URL, json=data, headers=HEADERS)
+    if response.status_code != 201:
+        abort_if_fail(response, "Unable to create assignees")
+
+    assignees_return_code = 0 if response.status_code == 201 else response.status_code
+    print("::group::github assignees response")
+    print(response.json())
+    print("::endgroup::github assignees response")
+    set_env("ASSIGNEES_RETURN_CODE", assignees_return_code)
+    print("::set-output name=assignees_return_code::%s" % assignees_return_code)
+
+
+def find_pull_request(listing, source):
+    """Given a listing and a source, find a pull request based on the source
+    (the branch name).
+
+    Parameters:
+    listing (list) : the list of PR objects (dict) to parse over
+    source   (str) : the source (head) branch to look for
+    """
+    if listing:
+        for entry in listing:
+            if entry.get("head", {}).get("ref", "") == source:
+                print("Pull request from %s is already open!" % source)
+                return entry
+
+
+def add_reviewers(entry, reviewers, team_reviewers):
+    """Given regular or team reviewers, add them to a PR.
+
+    Parameters:
+    entry (dict) : the pull request metadata
+    """
+    print("Found reviewers: %s and team reviewers: %s" % (reviewers, team_reviewers))
+    team_reviewers = parse_into_list(team_reviewers)
+    reviewers = parse_into_list(reviewers)
+    print("Parsed reviewers: %s and team reviewers: %s" % (reviewers, team_reviewers))
+
+    # POST /repos/:owner/:repo/pulls/:pull_number/requested_reviewers
+    REVIEWERS_URL = "%s/%s/requested_reviewers" % (PULLS_URL, entry.get("number"))
+
+    data = {"reviewers": reviewers, "team_reviewers": team_reviewers}
+    response = requests.post(REVIEWERS_URL, json=data, headers=HEADERS)
+    if response.status_code != 201:
+        abort_if_fail(response, "Unable to assign reviewers")
+    reviewers_return_code = 0 if response.status_code == 201 else response.status_code
+
+    print("::group::github reviewers response")
+    print(response.json())
+    print("::endgroup::github reviewers response")
+    set_env("REVIEWERS_RETURN_CODE", reviewers_return_code)
+    print("::set-output name=reviewers_return_code::%s" % reviewers_return_code)
+    print("Add reviewers return code: %s" % reviewers_return_code)
+
+
+################################################################################
+# Global Variables (we can't use GITHUB_ prefix)
+################################################################################
+
+API_VERSION = "v3"
+BASE = "https://api.github.com"
+
+HEADERS = {
+    "Authorization": "token %s" % get_envar("GITHUB_TOKEN"),
+    "Accept": "application/vnd.github.%s+json;application/vnd.github.antiope-preview+json;application/vnd.github.shadow-cat-preview+json"
+    % API_VERSION,
+}
+
+# URLs
+REPO_URL = "%s/repos/%s" % (BASE, get_envar("GITHUB_REPOSITORY"))
+ISSUE_URL = "%s/issues" % REPO_URL
+PULLS_URL = "%s/pulls" % REPO_URL
+
+
+def create_pull_request(
+    source,
+    target,
+    body,
+    title,
+    assignees,
+    reviewers,
+    team_reviewers,
+    is_draft=False,
+    can_modify=True,
+    state="open",
+):
+    """Create pull request is the base function that determines if the PR exists,
+    and then updates or creates it depending on user preferences.
+    """
+    listing = list_pull_requests(target, source)
+
+    # Determine if the pull request is already open
+    entry = find_pull_request(listing, source)
+    response = None
+
+    # Case 1: we found the PR, the user wants to pass
+    if entry and os.environ.get("PASS_IF_EXISTS"):
+        print("PASS_IF_EXISTS is set, exiting with success status.")
+        sys.exit(0)
+
+    # Does the user want to update the existing PR?
+    if entry and os.environ.get("PULL_REQUEST_UPDATE"):
+        response = update_pull_request(entry, title, body, target, state)
+        set_pull_request_groups(response)
+
+    # If it's not open, we open a new pull request
+    elif not entry:
+        response = open_pull_request(title, body, target, source, is_draft, can_modify)
+        set_pull_request_groups(response)
+
+    # If we have a response, parse into json (no longer need retvals)
+    response = response.json() if response else None
+
+    # If we have opened or updated, we can add assignees
+    if response and assignees:
+        add_assignees(response)
+    if response and (reviewers or team_reviewers):
+        add_reviewers(response, reviewers, team_reviewers)
+
+
+def main():
+    """main primarily parses environment variables to prepare for creation"""
+
+    # path to file that contains the POST response of the event
+    # Example: https://github.com/actions/bin/tree/master/debug
+    # Value: /github/workflow/event.json
+    check_events_json()
+
+    branch_prefix = os.environ.get("BRANCH_PREFIX", "")
+    print("Branch prefix is %s" % branch_prefix)
+    if not branch_prefix:
+        print("No branch prefix is set, all branches will be used.")
+
+    # Default to master to support older, will eventually change to main
+    pull_request_branch = os.environ.get("PULL_REQUEST_BRANCH", "master")
+    print("Pull requests will go to %s" % pull_request_branch)
+
+    # Pull request draft
+    pull_request_draft = os.environ.get("PULL_REQUEST_DRAFT")
+    if not pull_request_draft:
+        print("No explicit preference for draft PR: created PRs will be normal PRs.")
+        pull_request_draft = False
+    else:
+        print("PULL_REQUEST_DRAFT set to a value: created PRs will be draft PRs.")
+        pull_request_draft = True
+
+    # If an update is true, we can change the state
+    pull_request_state = os.environ.get("PULL_REQUEST_STATE", "open")
+    if pull_request_state not in ["open", "closed"]:
+        sys.exit("State is required to be one of 'open' or 'closed'")
+
+    # Maintainer can modify, defaults to CAN, unless user sets MAINTAINER_CANT_MODIFY
+    maintainer_can_modify = os.environ.get("MAINTAINER_CANT_MODIFY")
+    if not maintainer_can_modify:
+        print("No preference for maintainer being able to modify: default is true.")
+        maintainer_can_modify = True
+    else:
+        print(
+            "MAINTAINER_CANT_MODIFY set to a value: maintainer will not be able to modify."
+        )
+        maintainer_can_modify = False
+
+    # Assignees
+    assignees = os.environ.get("PULL_REQUEST_ASSIGNEES")
+    if not assignees:
+        print("PULL_REQUEST_ASSIGNEES is not set, no assignees.")
+    else:
+        print("PULL_REQUEST_ASSIGNEES is set, %s" % assignees)
+
+    # Reviewers (individual and team)
+
+    reviewers = os.environ.get("PULL_REQUEST_REVIEWERS")
+    team_reviewers = os.environ.get("PULL_REQUEST_TEAM_REVIEWERS")
+    if not reviewers:
+        print("PULL_REQUEST_REVIEWERS is not set, no reviewers.")
+    else:
+        print("PULL_REQUEST_REVIEWERS is set, %s" % reviewers)
+
+    if not team_reviewers:
+        print("PULL_REQUEST_TEAM_REVIEWERS is not set, no team reviewers.")
+    else:
+        print("PULL_REQUEST_TEAM_REVIEWERS is set, %s" % team_reviewers)
+
+    # The user is allowed to explicitly set the name of the branch
+    from_branch = os.environ.get("PULL_REQUEST_FROM_BRANCH")
+    if not from_branch:
+        print("PULL_REQUEST_FROM_BRANCH is not set, checking branch in payload.")
+        with open(check_events_json(), "r") as fd:
+            from_branch = json.loads(fd.read()).get("ref")
+        from_branch = from_branch.replace("refs/heads/", "").strip("/")
+    else:
+        print("PULL_REQUEST_FROM_BRANCH is set.")
+
+    # At this point, we must have a branch
+    if from_branch:
+        print("Found branch %s to open PR from" % from_branch)
+    else:
+        sys.exit(
+            "You are required to define PULL_REQUEST_FROM_BRANCH in the environment."
+        )
+
+    # If it's to the target branch, ignore it
+    if from_branch == pull_request_branch:
+        print("Target and current branch are identical (%s), skipping." % from_branch)
+        sys.exit(0)
+
+    # If the prefix for the branch matches
+    if not branch_prefix or from_branch.startswith(branch_prefix):
+
+        # Pull request body (optional)
+        pull_request_body = os.environ.get(
+            "PULL_REQUEST_BODY",
+            "This is an automated pull request to update from branch %s" % from_branch,
+        )
+
+        print("::group::pull request body")
+        print(pull_request_body)
+        print("::endgroup::pull request body")
+
+        # Pull request title (optional)
+        pull_request_title = os.environ.get(
+            "PULL_REQUEST_TITLE", "Update from %s" % from_branch
+        )
+        print("::group::pull request title")
+        print(pull_request_title)
+        print("::endgroup::pull request title")
+
+        # Create the pull request
+        create_pull_request(
+            target=pull_request_branch,
+            source=from_branch,
+            body=pull_request_body,
+            title=pull_request_title,
+            is_draft=pull_request_draft,
+            can_modify=maintainer_can_modify,
+            assignees=assignees,
+            reviewers=reviewers,
+            team_reviewers=team_reviewers,
+            state=pull_request_state,
+        )
+
+
+if __name__ == "__main__":
+    print("==========================================================================")
+    print("START: Running Pull Request on Branch Update Action!")
+    main()
+    print("==========================================================================")
+    print("END: Finished")

pull-request.sh

@@ -1,163 +0,0 @@
-#!/bin/bash
-
-# Suggested by Github actions to be strict
-set -e
-set -o pipefail
-
-################################################################################
-# Global Variables (we can't use GITHUB_ prefix)
-################################################################################
-
-API_VERSION=v3
-BASE=https://api.github.com
-AUTH_HEADER="Authorization: token ${GITHUB_TOKEN}"
-HEADER="Accept: application/vnd.github.${API_VERSION}+json"
-HEADER="${HEADER}; application/vnd.github.antiope-preview+json; application/vnd.github.shadow-cat-preview+json"
-
-# URLs
-REPO_URL="${BASE}/repos/${GITHUB_REPOSITORY}"
-PULLS_URL=$REPO_URL/pulls
-
-################################################################################
-# Helper Functions
-################################################################################
-
-
-check_credentials() {
-
-    if [[ -z "${GITHUB_TOKEN}" ]]; then
-        printf "You must include the GITHUB_TOKEN as an environment variable.\n"
-        exit 1
-    fi
-
-}
-
-check_events_json() {
-
-    if [[ ! -f "${GITHUB_EVENT_PATH}" ]]; then
-        printf "Cannot find Github events file at ${GITHUB_EVENT_PATH}\n";
-        exit 1;
-    fi
-    printf "Found ${GITHUB_EVENT_PATH}\n";
-
-}
-
-create_pull_request() {
-
-    # JSON strings
-    SOURCE="$(echo -n "${1}" | jq --raw-input --slurp ".")"  # from this branch
-    TARGET="$(echo -n "${2}" | jq --raw-input --slurp ".")"  # pull request TO this target
-    BODY="$(echo -n "${3}" | jq --raw-input --slurp ".")"    # this is the content of the message
-    TITLE="$(echo -n "${4}" | jq --raw-input --slurp ".")"   # pull request title
-
-    # JSON boolean
-    if [[ "${5}" ==  "true" ]]; then                         # if PRs are draft
-        DRAFT="true";
-    else
-        DRAFT="false";
-    fi
-
-    # Check if the branch already has a pull request open
-
-    DATA="{\"base\":${TARGET}, \"head\":${SOURCE}, \"body\":${BODY}}"
-    RESPONSE=$(curl -sSL -H "${AUTH_HEADER}" -H "${HEADER}" --user "${GITHUB_ACTOR}" -X GET --data "${DATA}" ${PULLS_URL})
-    PR=$(echo "${RESPONSE}" | jq --raw-output '.[] | .head.ref')
-    printf "Response ref: ${PR}\n"
-
-    # Option 1: The pull request is already open
-    if [[ "${PR}" == "${SOURCE}" ]]; then
-        printf "Pull request from ${SOURCE} to ${TARGET} is already open!\n"
-
-    # Option 2: Open a new pull request
-    else
-        # Post the pull request
-        DATA="{\"title\":${TITLE}, \"body\":${BODY}, \"base\":${TARGET}, \"head\":${SOURCE}, \"draft\":${DRAFT}}"
-        printf "curl --user ${GITHUB_ACTOR} -X POST --data ${DATA} ${PULLS_URL}\n"
-        curl -sSL -H "${AUTH_HEADER}" -H "${HEADER}" --user "${GITHUB_ACTOR}" -X POST --data "${DATA}" ${PULLS_URL}
-        echo $?
-    fi
-}
-
-
-main () {
-
-    # path to file that contains the POST response of the event
-    # Example: https://github.com/actions/bin/tree/master/debug
-    # Value: /github/workflow/event.json
-    check_events_json;
-
-    # User specified branch to PR to, and check
-    if [ -z "${BRANCH_PREFIX}" ]; then
-        printf "No branch prefix is set, all branches will be used.\n"
-        BRANCH_PREFIX=""
-        printf "Branch prefix is $BRANCH_PREFIX\n"
-    fi
-
-    if [ -z "${PULL_REQUEST_BRANCH}" ]; then
-        PULL_REQUEST_BRANCH=master
-    fi
-    printf "Pull requests will go to ${PULL_REQUEST_BRANCH}\n"
-
-    if [ -z "${PULL_REQUEST_DRAFT}" ]; then
-        printf "No explicit preference for draft PR: created PRs will be normal PRs.\n"
-        PULL_REQUEST_DRAFT="false"
-    else
-        printf "Environment variable PULL_REQUEST_DRAFT set to a value: created PRs will be draft PRs.\n"
-        PULL_REQUEST_DRAFT="true"
-    fi
-
-    # The user is allowed to explicitly set the name of the branch
-    if [ -z "${PULL_REQUEST_FROM_BRANCH}" ]; then
-        printf "PULL_REQUEST_FROM_BRANCH is not set, checking branch in payload.\n"
-        BRANCH=$(jq --raw-output .ref "${GITHUB_EVENT_PATH}");
-        BRANCH=$(echo "${BRANCH/refs\/heads\//}")
-    else
-        printf "PULL_REQUEST_FROM_BRANCH is set.\n"
-        BRANCH="${PULL_REQUEST_FROM_BRANCH}"        
-    fi
-
-    # At this point, we must have a branch
-    if [[ "$BRANCH" != "null" ]]; then
-        printf "Found branch $BRANCH to open PR from\n"
-    else
-        printf "No branch in payload, you are required to define PULL_REQUEST_FROM_BRANCH in the environment.\n"
-        exit 1
-    fi
-
-    # If it's to the target branch, ignore it
-    if [[ "${BRANCH}" == "${PULL_REQUEST_BRANCH}" ]]; then
-        printf "Target and current branch are identical (${BRANCH}), skipping.\n"
-    else
-
-        # If the prefix for the branch matches
-        if  [[ $BRANCH == ${BRANCH_PREFIX}* ]]; then
-
-            # Ensure we have a GitHub token
-            check_credentials
-
-            # Pull request body (optional)
-            if [ -z "${PULL_REQUEST_BODY}" ]; then
-                echo "No pull request body is set, will use default."
-                PULL_REQUEST_BODY="This is an automated pull request to update the container collection ${BRANCH}"
-            fi
-            printf "Pull request body is ${PULL_REQUEST_BODY}\n"
-
-            # Pull request title (optional)
-            if [ -z "${PULL_REQUEST_TITLE}" ]; then
-                printf "No pull request title is set, will use default.\n"
-                PULL_REQUEST_TITLE="Update container ${BRANCH}"
-            fi
-            printf "Pull request title is ${PULL_REQUEST_TITLE}\n"
-
-            create_pull_request "${BRANCH}" "${PULL_REQUEST_BRANCH}" "${PULL_REQUEST_BODY}" "${PULL_REQUEST_TITLE}" "${PULL_REQUEST_DRAFT}"
-
-        fi
-
-    fi
-}
-
-echo "==========================================================================
-START: Running Pull Request on Branch Update Action!";
-main;
-echo "==========================================================================
-END: Finished";