-
-
Save realdubb/637cfcafb3adc27d3bfc22d632ccbefe to your computer and use it in GitHub Desktop.
Create a (draft) pull request using GitHub CLI
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/bin/bash | |
| # Create a (draft) pull request using GitHub CLI. | |
| # It assigns the PR to the current user, fills in the title from the first commit, | |
| # and uses the PR template file for the description. | |
| set -euo pipefail | |
| # Colors for output | |
| RED='\033[0;31m' | |
| GREEN='\033[0;32m' | |
| YELLOW='\033[0;33m' | |
| NO_COLOR='\033[0m' | |
| # Helper functions | |
| print_message() { echo -e "\n${1}${2}${NO_COLOR}\n"; } | |
| error_message() { print_message "${RED}" "Error: $1"; } | |
| warning_message() { print_message "${YELLOW}" "Warning: $1"; } | |
| success_message() { print_message "${GREEN}" "$1"; } | |
| PR_TEMPLATE=$(curl -s https://raw.githubusercontent.com/gumroad/.github/main/pull_request_template.md) | |
| gh_pr_create() { | |
| gh pr create -a @me --fill-first --body "$PR_TEMPLATE" "$@" | |
| } | |
| create_pr() { | |
| git push || { | |
| error_message "Failed to push changes." | |
| exit 1 | |
| } | |
| if gh_pr_create "--draft"; then | |
| success_message "Draft pull request created successfully!" | |
| elif gh_pr_create; then | |
| success_message "Pull request created!" | |
| else | |
| error_message "Failed to create PR." | |
| exit 1 | |
| fi | |
| } | |
| build_pr_description() { | |
| local pr_title=$(gh pr view --json title -q .title) | |
| local current_description=$(gh pr view --json body -q .body) | |
| # Get the diff for files in select folders and root, excluding all other folders | |
| local pr_diff=$(gh pr diff | awk '/^diff --git/ {in_folder=($0 ~ " b/(app|config|db|scripts)/| b/[^/]+$")} in_folder {print}') | |
| local system_content="You are an expert software engineer. that writes high-quality, concise pull request descriptions based on code diffs, | |
| titles, and existing descriptions." | |
| local user_content="Carefully review the provided context: title, PR template, and diff. | |
| Update the placeholder content from the What section, and the placeholder content from the Why section. | |
| Under the Checklist section, update the checklist items that are relevant to the changes (don't update self-review, | |
| that will be done manually). | |
| The content generated MUST use the imperative tense. | |
| Title: | |
| <title> | |
| $pr_title | |
| </title> | |
| PR Template: | |
| <pr_template> | |
| $PR_TEMPLATE | |
| </pr_template> | |
| Current Description: | |
| <current_description> | |
| $current_description | |
| </current_description> | |
| Diff (excluding test files): | |
| <diff> | |
| $pr_diff | |
| </diff>" | |
| local payload=$(jq -n \ | |
| --arg system_content "$system_content" \ | |
| --arg user_content "$user_content" \ | |
| '{ | |
| model: "gpt-3.5-turbo", | |
| messages: [ | |
| { role: "system", content: $system_content }, | |
| { role: "user", content: $user_content } | |
| ] | |
| }') | |
| local response=$(curl -s -X POST https://api.openai.com/v1/chat/completions \ | |
| -H "Content-Type: application/json" \ | |
| -H "Authorization: Bearer ${PERSONAL_OPENAI_API_KEY:-}" \ | |
| -d "$payload") | |
| local content=$(echo "$response" | jq -r '.choices[0].message.content') | |
| if [ -z "$content" ] || [ "$content" == "null" ]; then | |
| echo "$response" | jq '.' >&2 | |
| return 1 | |
| fi | |
| printf '%s' "$content" | |
| } | |
| # Use OpenAI API to generate an updated description | |
| function update_pr_description() { | |
| if [ -z "${PERSONAL_OPENAI_API_KEY:-}" ]; then | |
| warning_message "Set PERSONAL_OPENAI_API_KEY in .env.development.local for AI-generated PR descriptions." | |
| return 0 | |
| fi | |
| local updated_description | |
| if ! updated_description=$(build_pr_description 2>&1); then | |
| error_message "Failed to build PR description." | |
| error_message "$updated_description" | |
| return 0 | |
| fi | |
| echo "Updating pull request description..." | |
| echo "$updated_description" | gh pr edit --body-file - || { | |
| error_message "Failed to update PR description." | |
| return 0 | |
| } | |
| success_message "Pull request description updated successfully!" | |
| } | |
| function open_pr_in_browser() { | |
| gh pr view --web | |
| } | |
| # Check if GitHub CLI is installed | |
| if ! command -v gh &>/dev/null; then | |
| error_message "GitHub CLI (gh) is not installed. Please visit https://cli.github.com/" | |
| exit 1 | |
| fi | |
| # Source the .env.development.local file if it exists | |
| if [ -f .env.development.local ]; then | |
| source .env.development.local | |
| fi | |
| create_pr | |
| update_pr_description | |
| open_pr_in_browser |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment