enarxbot-assigned

#!/usr/bin/python3
# SPDX-License-Identifier: Apache-2.0

from githubgql import githubgql

import constants
import json
import sys
import os

if os.environ["GITHUB_EVENT_NAME"] not in ["pull_request_target", "issues"]:
    sys.exit(0)

owner, repo = os.environ["GITHUB_REPOSITORY"].split("/")

with open(os.environ["GITHUB_EVENT_PATH"]) as f:
    event = json.load(f)

if event["action"] != "assigned":
    sys.exit(0)

id = event.get('pull_request', None)
if id is None:
    id_type = "issue"
    id = event['issue']['node_id']
else:
    id_type = "pr"
    id = id['node_id']


# Find out what projects the issue/PR is in, as well as their assignees.
try:
    result = githubgql.graphql(
        """
        query($id:ID!, $cardCursor:String, $assigneeCursor:String) {
            node(id:$id) {
                ... on PullRequest {
                    number
                    assignees(first:100, after:$assigneeCursor) {
                        pageInfo { endCursor hasNextPage }
                        nodes {
                            login
                        }
                    }
                    projectCards(first:100, archivedStates:NOT_ARCHIVED, after:$cardCursor) {
                        pageInfo { endCursor hasNextPage }
                        nodes {
                            id
                            column {
                                id
                                name
                                project {
                                    id
                                    name
                                }
                            }
                        }
                    }
                }
                ... on Issue {
                    number
                    assignees(first:100, after:$assigneeCursor) {
                        pageInfo { endCursor hasNextPage }
                        nodes {
                            login
                        }
                    }
                    projectCards(first:100, archivedStates:NOT_ARCHIVED, after:$cardCursor) {
                        pageInfo { endCursor hasNextPage }
                        nodes {
                            id
                            column {
                                id
                                name
                                project {
                                    id
                                    name
                                }
                            }
                        }
                    }
                }
            }
        }
        """,
        id=id,
        page={
            "cardCursor": ["node", "projectCards"],
            "assigneeCursor": ["node", "assignees"],
        }
    )
except githubgql.TokenError as e:
    print(e.error)
    sys.exit(0)

# Construct lists of assignees, projects, and columns.
assignees = result["node"]["assignees"]["nodes"]
cards = result["node"]["projectCards"]["nodes"]
columns = {card["column"]["project"]["id"]: card["column"]["name"] for card in cards}

input = None

if len(assignees) > 0 and constants.PROJECTS["Sprint"] not in columns.keys():
    # Issues must additionally be present in the "Assigned" column of the
    # Planning board to be moved.
    if id_type == "issue" and columns.get(constants.PROJECTS["Planning"]) == "Assigned":
        # Print status.
        print(f"Moving issue {owner}/{repo}#{result['node']['number']} to Sprint board")
        input = {
            "projectColumnId": constants.COLUMNS["Sprint"]["Assigned"],
            "contentId": id
        }

    if id_type == "pr":
        # Print status.
        print(f"Moving PR {owner}/{repo}#{result['node']['number']} to Sprint board")
        input = {
            "projectColumnId": constants.COLUMNS["Sprint"]["Reviewing"],
            "contentId": id
        }

# If this issue/PR needs to be moved, move it.
if input is not None:
    try:
        githubgql.graphql(
            """
            mutation($input:AddProjectCardInput!) {
                addProjectCard(input:$input) {
                    clientMutationId
                }
            }
            """,
            input=input
        )
    except githubgql.GraphQLError as e:
        if e.errors[0]["message"] != "Project already has the associated issue":
            raise
        print(f"Project already has this card. Skipping addition.")