Simple module to manage pull request creation within neovim

pull_request.lua

local M = {}

---@return 'main' | 'develop'
local function decide_base_branch()
  if vim.fn.system 'git branch --contains develop' == 'error: malformed object name develop' then
    return 'main'
  end

  return 'develop'
end

---@return string
local function return_project_root()
  return vim.fn.substitute(vim.fn.system 'git rev-parse --show-toplevel', '\n', '', '')
end

function M.open_pr_template()
  local template_path = return_project_root() .. '/.github/pull_request_template.md'

  vim.cmd('tabe ' .. vim.fn.tempname())

  if vim.fn.filereadable(template_path) then
    vim.fn.setline(1, vim.fn.readfile(template_path))
    vim.bo.filetype = 'markdown'
  end
end

function M.create_pr()
  local filetype = vim.fn.expand '%'
  local base_branch = decide_base_branch()
  local reviewers = vim.fn.getenv 'REVIEWERS'

  local gh_pr_cmd = string.format('gh pr -a @me create -B %s -r %s -F "%s"', base_branch, reviewers, filetype)

  vim.cmd('!tmux neww ' .. gh_pr_cmd)
end

function M.open_pr()
  local cmd = 'gh pr view -w'
  vim.cmd('!' .. cmd)
end

return M