GitHubIssueAppendEnhancer.tsx

import { IssueOpenedIcon } from "@primer/octicons-react"
import OverType, { type OverTypeInstance } from "overtype"
import type React from "react"
import { LinkOutOfPopup } from "@/components/LinkOutOfPopup"
import type {
  CommentEnhancer,
  CommentSpot,
  StrippedLocation,
} from "@/lib/enhancer"
import { logger } from "@/lib/logger"
import { fixupOvertype, modifyDOM } from "../overtype-misc"
import {
  commonGitHubOptions,
  isInProjectCommentBox,
  isProjectUrl,
  parseProjectIssueParam,
  prepareGitHubHighlighter,
} from "./github-common"

const GH_ISSUE_APPEND = "GH_ISSUE_APPEND" as const

export interface GitHubIssueAppendSpot extends CommentSpot {
  type: typeof GH_ISSUE_APPEND
  title: string
  domain: string
  slug: string // owner/repo
  number: number // issue number, undefined for new issues
}

export class GitHubIssueAppendEnhancer
  implements CommentEnhancer<GitHubIssueAppendSpot>
{
  forSpotTypes(): string[] {
    return [GH_ISSUE_APPEND]
  }

  tryToEnhance(
    textarea: HTMLTextAreaElement,
    location: StrippedLocation
  ): GitHubIssueAppendSpot | null {
    if (textarea.id === "feedback") {
      return null
    }
    if (location.host !== "github.com") {
      return null
    }

    // Don't enhance textareas that are within the issue/PR body editing container
    const bodyContainer = textarea.closest(".react-issue-body")
    if (bodyContainer) {
      return null
    }

    // Check for project URLs with issue parameter first
    if (isProjectUrl(location.pathname)) {
      const params = new URLSearchParams(location.search)
      // Only match textareas within Shared-module__CommentBox (those are for adding new comments)
      if (isInProjectCommentBox(textarea)) {
        const issueInfo = parseProjectIssueParam(params)
        if (issueInfo) {
          const unique_key = `github.com:${issueInfo.slug}:${issueInfo.number}`
          // For project views, the title is in the side panel dialog
          const title =
            document
              .querySelector('[data-testid="issue-title"]')
              ?.textContent?.trim() || ""
          return {
            domain: location.host,
            number: issueInfo.number,
            slug: issueInfo.slug,
            title,
            type: GH_ISSUE_APPEND,
            unique_key,
          }
        }
      }
      return null
    }

    // Parse GitHub URL structure: /owner/repo/issues/123 or /owner/repo/pull/456
    logger.debug(`${this.constructor.name} examing url`, location.pathname)

    const match = location.pathname.match(
      /^\/([^/]+)\/([^/]+)(?:\/issues\/(\d+))/
    )
    logger.debug(`${this.constructor.name} found match`, location.pathname)
    if (!match) return null

    const [, owner, repo, numberStr] = match
    const slug = `${owner}/${repo}`
    const number = parseInt(numberStr!, 10)
    const unique_key = `github.com:${slug}:${number}`
    const title = document
      .querySelector("main h1")!
      .textContent!.replace(/\s*#\d+$/, "")
      .trim()
    return {
      domain: location.host,
      number,
      slug,
      title,
      type: GH_ISSUE_APPEND,
      unique_key,
    }
  }

  enhance(
    textArea: HTMLTextAreaElement,
    _spot: GitHubIssueAppendSpot
  ): OverTypeInstance {
    prepareGitHubHighlighter()
    const overtypeContainer = modifyDOM(textArea)
    return fixupOvertype(
      new OverType(overtypeContainer, {
        ...commonGitHubOptions,
        minHeight: "100px",
        placeholder: "Use Markdown to format your comment",
      })
    )
  }

  tableUpperDecoration(spot: GitHubIssueAppendSpot): React.ReactNode {
    return (
      <>
        <span className="flex h-4 w-4 flex-shrink-0 items-center justify-center">
          <IssueOpenedIcon size={16} />
        </span>
        <span>
          #{spot.number}{" "}
          <LinkOutOfPopup href={`https://${spot.domain}/${spot.slug}`}>
            {spot.slug}
          </LinkOutOfPopup>
        </span>
      </>
    )
  }

  tableTitle(spot: GitHubIssueAppendSpot): string {
    return spot.title
  }
}