{ pkgs, lib }:
# Single `hive-forge <verb>` CLI wrapping common Forgejo API operations.
# Reads credentials from the environment:
#
#   HIVE_FORGE_URL      -- base URL, e.g. http://localhost:3000
#   HIVE_FORGE_REPO     -- default repo, e.g. hyperhive/hyperhive
#   HYPERHIVE_STATE_DIR -- state dir; forge-token lives here
#
# Usage: hive-forge <verb> [args...]
# Verbs: view, issue, pr, comment, assign, close, labels, pr-reviews,
#        branches, tree-sha
pkgs.writeShellApplication {
  name = "hive-forge";
  runtimeInputs = [
    pkgs.curl
    pkgs.jq
  ];
  text = ''
    : "''${HIVE_FORGE_URL:=http://localhost:3000}"
    : "''${HIVE_FORGE_REPO:=hyperhive/hyperhive}"
    _state_dir="''${HYPERHIVE_STATE_DIR:-}"
    _token_file="''${_state_dir:+$_state_dir/}forge-token"
    if [ ! -f "$_token_file" ]; then
      echo "hive-forge: no forge-token at $_token_file" >&2
      exit 1
    fi
    _token=$(cat "$_token_file")
    FORGE_API="$HIVE_FORGE_URL/api/v1"

    forge_get() {
      ${pkgs.curl}/bin/curl -sf \
        -H "Authorization: token $_token" \
        -H "Accept: application/json" \
        "$1"
    }
    forge_post() {
      ${pkgs.curl}/bin/curl -sf -X POST \
        -H "Authorization: token $_token" \
        -H "Content-Type: application/json" \
        -H "Accept: application/json" \
        -d "$2" "$1"
    }
    forge_patch() {
      ${pkgs.curl}/bin/curl -sf -X PATCH \
        -H "Authorization: token $_token" \
        -H "Content-Type: application/json" \
        -H "Accept: application/json" \
        -d "$2" "$1"
    }
    forge_delete() {
      local _url="$1"
      local _body="''${2:-}"
      if [ -n "$_body" ]; then
        ${pkgs.curl}/bin/curl -sf -X DELETE \
          -H "Authorization: token $_token" \
          -H "Content-Type: application/json" \
          -H "Accept: application/json" \
          -d "$_body" "$_url"
      else
        ${pkgs.curl}/bin/curl -sf -X DELETE \
          -H "Authorization: token $_token" \
          -H "Accept: application/json" \
          "$_url"
      fi
    }

    cmd_view() {
      # view <number> [repo]
      # Dump title + body + all comments for an issue or PR.
      if [ $# -lt 1 ]; then echo "usage: hive-forge view <number> [repo]" >&2; exit 1; fi
      local _n="$1" _repo="''${2:-$HIVE_FORGE_REPO}"
      local _issue _title _body _state _user _is_pr _kind
      _issue=$(forge_get "$FORGE_API/repos/$_repo/issues/$_n")
      _title=$(printf '%s' "$_issue" | jq -r '.title')
      _body=$(printf '%s' "$_issue" | jq -r '.body // ""')
      _state=$(printf '%s' "$_issue" | jq -r '.state')
      _user=$(printf '%s' "$_issue" | jq -r '.user.login')
      _is_pr=$(printf '%s' "$_issue" | jq -r '.pull_request != null')
      _kind="issue"
      if [ "$_is_pr" = "true" ]; then _kind="PR"; fi
      printf '# %s #%s [%s] by %s\n' "$_kind" "$_n" "$_state" "$_user"
      printf '%s\n' "$_title"
      if [ -n "$_body" ]; then printf '\n%s\n' "$_body"; fi
      local _comments _count
      _comments=$(forge_get "$FORGE_API/repos/$_repo/issues/$_n/comments?limit=50")
      _count=$(printf '%s' "$_comments" | jq 'length')
      if [ "$_count" -gt 0 ]; then
        printf '\n---\n## Comments (%s)\n\n' "$_count"
        printf '%s' "$_comments" | jq -r '.[] | "**\(.user.login)**: \(.body)\n"'
      fi
    }

    cmd_issue() {
      # issue <number> [repo]
      # Print key fields of an issue as JSON.
      if [ $# -lt 1 ]; then echo "usage: hive-forge issue <number> [repo]" >&2; exit 1; fi
      local _n="$1" _repo="''${2:-$HIVE_FORGE_REPO}"
      forge_get "$FORGE_API/repos/$_repo/issues/$_n" \
        | jq '{number,title,state,user:.user.login,assignees:[.assignees[]?.login],labels:[.labels[]?.name],body}'
    }

    cmd_pr() {
      # pr <number> [repo]
      # Print key fields of a PR as JSON.
      if [ $# -lt 1 ]; then echo "usage: hive-forge pr <number> [repo]" >&2; exit 1; fi
      local _n="$1" _repo="''${2:-$HIVE_FORGE_REPO}"
      forge_get "$FORGE_API/repos/$_repo/pulls/$_n" \
        | jq '{number,title,state,merged,user:.user.login,head_sha:.head.sha,head_branch:.head.label,base_branch:.base.label}'
    }

    cmd_comment() {
      # comment <number> [body | -f <file> | - for stdin]
      # Post a comment on an issue or PR.
      if [ $# -lt 1 ]; then echo "usage: hive-forge comment <number> [body | -f <file> | -]" >&2; exit 1; fi
      local _n="$1"; shift
      local _body
      if [ $# -eq 0 ] || [ "''${1:-}" = "-" ]; then
        _body=$(cat)
      elif [ "''${1:-}" = "-f" ]; then
        _body=$(cat "$2")
      else
        _body="$*"
      fi
      local _payload
      _payload=$(jq -n --arg body "$_body" '{body:$body}')
      forge_post "$FORGE_API/repos/$HIVE_FORGE_REPO/issues/$_n/comments" "$_payload" \
        | jq '{id,url:.html_url}'
    }

    cmd_assign() {
      # assign <number> <user> [--remove]
      # Assign or unassign a user on an issue or PR.
      if [ $# -lt 2 ]; then echo "usage: hive-forge assign <number> <user> [--remove]" >&2; exit 1; fi
      local _n="$1" _user="$2" _remove="''${3:-}"
      local _payload
      _payload=$(jq -n --arg u "$_user" '{assignees:[$u]}')
      if [ "$_remove" = "--remove" ]; then
        forge_delete "$FORGE_API/repos/$HIVE_FORGE_REPO/issues/$_n/assignees" "$_payload" \
          | jq '{number,assignees:[.assignees[]?.login]}'
      else
        forge_post "$FORGE_API/repos/$HIVE_FORGE_REPO/issues/$_n/assignees" "$_payload" \
          | jq '{number,assignees:[.assignees[]?.login]}'
      fi
    }

    cmd_close() {
      # close <number>
      # Close an issue or PR.
      if [ $# -lt 1 ]; then echo "usage: hive-forge close <number>" >&2; exit 1; fi
      forge_patch "$FORGE_API/repos/$HIVE_FORGE_REPO/issues/$1" '{"state":"closed"}' \
        | jq '{number,state}'
    }

    cmd_labels() {
      # labels <number> [list|add|remove] [labels...]
      # Manage labels on an issue or PR. Default action is list.
      if [ $# -lt 1 ]; then echo "usage: hive-forge labels <number> [list|add|remove] [labels...]" >&2; exit 1; fi
      local _n="$1" _action="''${2:-list}"
      shift; shift || true
      local _all _ids _id _label
      case "$_action" in
        list)
          forge_get "$FORGE_API/repos/$HIVE_FORGE_REPO/issues/$_n/labels" | jq '[.[].name]'
          ;;
        add)
          _all=$(forge_get "$FORGE_API/repos/$HIVE_FORGE_REPO/labels?limit=100")
          _ids=$(printf '%s' "$_all" | jq -r --args '[.[] | select(.name == ($ARGS.positional[])) | .id]' -- "$@")
          forge_post "$FORGE_API/repos/$HIVE_FORGE_REPO/issues/$_n/labels" "{\"labels\":$_ids}" \
            | jq '[.[].name]'
          ;;
        remove)
          _all=$(forge_get "$FORGE_API/repos/$HIVE_FORGE_REPO/labels?limit=100")
          for _label in "$@"; do
            _id=$(printf '%s' "$_all" | jq -r --arg n "$_label" '.[] | select(.name==$n) | .id')
            if [ -n "$_id" ]; then
              forge_delete "$FORGE_API/repos/$HIVE_FORGE_REPO/issues/$_n/labels/$_id"
            fi
          done
          forge_get "$FORGE_API/repos/$HIVE_FORGE_REPO/issues/$_n/labels" | jq '[.[].name]'
          ;;
        *)
          echo "unknown action: $_action (use list, add, remove)" >&2; exit 1
          ;;
      esac
    }

    cmd_pr_reviews() {
      # pr-reviews <number> [repo]
      # List reviews on a PR with id/state/user/body.
      if [ $# -lt 1 ]; then echo "usage: hive-forge pr-reviews <number> [repo]" >&2; exit 1; fi
      local _n="$1" _repo="''${2:-$HIVE_FORGE_REPO}"
      forge_get "$FORGE_API/repos/$_repo/pulls/$_n/reviews" \
        | jq '[.[] | {id,state,user:.user.login,body,comments_count}]'
    }

    cmd_branches() {
      # branches [pattern] [repo]
      # List branches, optionally filtered by a grep pattern.
      local _pattern="''${1:-}" _repo="''${2:-$HIVE_FORGE_REPO}"
      local _result
      _result=$(forge_get "$FORGE_API/repos/$_repo/branches?limit=100" | jq -r '.[].name')
      if [ -n "$_pattern" ]; then
        printf '%s\n' "$_result" | grep "$_pattern" || true
      else
        printf '%s\n' "$_result"
      fi
    }

    cmd_tree_sha() {
      # tree-sha <ref> [repo]
      # Print the tree SHA of the commit at the given branch or commit SHA.
      if [ $# -lt 1 ]; then echo "usage: hive-forge tree-sha <ref> [repo]" >&2; exit 1; fi
      local _ref="$1" _repo="''${2:-$HIVE_FORGE_REPO}"
      local _commit_sha
      _commit_sha=$(forge_get "$FORGE_API/repos/$_repo/branches/$_ref" 2>/dev/null \
        | jq -r '.commit.id // empty') || true
      if [ -z "$_commit_sha" ]; then
        _commit_sha="$_ref"
      fi
      forge_get "$FORGE_API/repos/$_repo/git/commits/$_commit_sha" \
        | jq -r '.tree.sha // .sha'
    }

    cmd_attach_issue() {
      # attach-issue <number> <file> [repo]
      # Upload a file as an attachment to an issue. Prints the download URL.
      if [ $# -lt 2 ]; then echo "usage: hive-forge attach-issue <number> <file> [repo]" >&2; exit 1; fi
      local _n="$1" _file="$2" _repo="''${3:-$HIVE_FORGE_REPO}"
      if [ ! -f "$_file" ]; then echo "hive-forge attach-issue: file not found: $_file" >&2; exit 1; fi
      ${pkgs.curl}/bin/curl -sf -X POST \
        -H "Authorization: token $_token" \
        -F "attachment=@$_file" \
        "$FORGE_API/repos/$_repo/issues/$_n/assets" \
        | jq -r '.browser_download_url'
    }

    cmd_attach_comment() {
      # attach-comment <comment-id> <file> [repo]
      # Upload a file as an attachment to an issue comment. Prints the download URL.
      if [ $# -lt 2 ]; then echo "usage: hive-forge attach-comment <comment-id> <file> [repo]" >&2; exit 1; fi
      local _id="$1" _file="$2" _repo="''${3:-$HIVE_FORGE_REPO}"
      if [ ! -f "$_file" ]; then echo "hive-forge attach-comment: file not found: $_file" >&2; exit 1; fi
      ${pkgs.curl}/bin/curl -sf -X POST \
        -H "Authorization: token $_token" \
        -F "attachment=@$_file" \
        "$FORGE_API/repos/$_repo/issues/comments/$_id/assets" \
        | jq -r '.browser_download_url'
    }

    VERB="''${1:-}"
    if [ -z "$VERB" ]; then
      echo "usage: hive-forge <verb> [args...]" >&2
      echo "verbs: view, issue, pr, comment, assign, close, labels, pr-reviews, branches, tree-sha, attach-issue, attach-comment" >&2
      exit 1
    fi
    shift

    case "$VERB" in
      view)           cmd_view "$@" ;;
      issue)          cmd_issue "$@" ;;
      pr)             cmd_pr "$@" ;;
      comment)        cmd_comment "$@" ;;
      assign)         cmd_assign "$@" ;;
      close)          cmd_close "$@" ;;
      labels)         cmd_labels "$@" ;;
      pr-reviews)     cmd_pr_reviews "$@" ;;
      branches)       cmd_branches "$@" ;;
      tree-sha)       cmd_tree_sha "$@" ;;
      attach-issue)   cmd_attach_issue "$@" ;;
      attach-comment) cmd_attach_comment "$@" ;;
      *)
        echo "hive-forge: unknown verb '$VERB'" >&2
        echo "verbs: view, issue, pr, comment, assign, close, labels, pr-reviews, branches, tree-sha, attach-issue, attach-comment" >&2
        exit 1
        ;;
    esac
  '';
}