orca/src/main/runtime/runtime-github-repository-query-commands.ts

import type { ListWorkItemsResult } from '../../shared/github/work-item-types'
import type { Repo } from '../../shared/repo-types'
import type { LocalProjectGhExecOptions } from '../project-runtime-git-options'
import {
  countWorkItems,
  getWorkItem,
  getWorkItemByOwnerRepo,
  listAssignableUsers,
  listIssues,
  listLabels,
  listWorkItems,
  type MainWorkItem
} from '../github/client'
import {
  listGhAccountBindingInventory,
  validateGhAccountBinding
} from '../github/gh-account-binding-inventory'
import { getRateLimit } from '../github/rate-limit'
import { githubHostFromIdentityKey } from '../../shared/github/repository-identity-key'
import { getWorkItemDetails } from '../github/work-item-details'

type LocalGitArgs = [] | [LocalProjectGhExecOptions]

type RuntimeGitHubRepositoryQueryCommandsDeps = {
  resolveRepo: (selector: string) => Promise<Repo>
  getLocalGitArgs: (repo: Repo) => LocalGitArgs
}

export class RuntimeGitHubRepositoryQueryCommands {
  constructor(private readonly deps: RuntimeGitHubRepositoryQueryCommandsDeps) {}

  async listRepoWorkItems(
    repoSelector: string,
    limit?: number,
    query?: string,
    page?: number,
    noCache?: boolean
  ): Promise<ListWorkItemsResult<MainWorkItem>> {
    const repo = await this.deps.resolveRepo(repoSelector)
    return listWorkItems(
      repo.path,
      limit,
      query,
      page,
      repo.issueSourcePreference,
      repo.connectionId ?? null,
      noCache,
      ...this.deps.getLocalGitArgs(repo)
    )
  }

  async listRepoIssues(
    repoSelector: string,
    limit?: number
  ): Promise<Awaited<ReturnType<typeof listIssues>>['items']> {
    const repo = await this.deps.resolveRepo(repoSelector)
    const result = await listIssues(
      repo.path,
      limit,
      repo.issueSourcePreference,
      repo.connectionId ?? null,
      ...this.deps.getLocalGitArgs(repo)
    )
    return result.items
  }

  async getRepoWorkItem(
    repoSelector: string,
    number: number,
    type?: 'issue' | 'pr'
  ): Promise<Awaited<ReturnType<typeof getWorkItem>>> {
    const repo = await this.deps.resolveRepo(repoSelector)
    return getWorkItem(
      repo.path,
      number,
      type,
      repo.connectionId ?? null,
      this.deps.getLocalGitArgs(repo)[0] ?? {},
      repo.issueSourcePreference
    )
  }

  async getRepoWorkItemByOwnerRepo(
    repoSelector: string,
    ownerRepo: { owner: string; repo: string; host?: string },
    number: number,
    type: 'issue' | 'pr'
  ): Promise<Awaited<ReturnType<typeof getWorkItemByOwnerRepo>>> {
    const repo = await this.deps.resolveRepo(repoSelector)
    return getWorkItemByOwnerRepo(
      repo.path,
      ownerRepo,
      number,
      type,
      repo.connectionId ?? null,
      ...this.deps.getLocalGitArgs(repo)
    )
  }

  async getRepoWorkItemDetails(
    repoSelector: string,
    number: number,
    type?: 'issue' | 'pr'
  ): Promise<Awaited<ReturnType<typeof getWorkItemDetails>>> {
    const repo = await this.deps.resolveRepo(repoSelector)
    return getWorkItemDetails(
      repo.path,
      number,
      type,
      repo.connectionId ?? null,
      this.deps.getLocalGitArgs(repo)[0] ?? {},
      repo.issueSourcePreference
    )
  }

  async countRepoWorkItems(repoSelector: string, query?: string): Promise<number> {
    const repo = await this.deps.resolveRepo(repoSelector)
    return countWorkItems(
      repo.path,
      query,
      repo.issueSourcePreference,
      repo.connectionId ?? null,
      ...this.deps.getLocalGitArgs(repo)
    )
  }

  async listRepoLabels(repoSelector: string): Promise<Awaited<ReturnType<typeof listLabels>>> {
    const repo = await this.deps.resolveRepo(repoSelector)
    return listLabels(
      repo.path,
      repo.issueSourcePreference,
      repo.connectionId ?? null,
      ...this.deps.getLocalGitArgs(repo)
    )
  }

  async listRepoAssignableUsers(
    repoSelector: string
  ): Promise<Awaited<ReturnType<typeof listAssignableUsers>>> {
    const repo = await this.deps.resolveRepo(repoSelector)
    return listAssignableUsers(
      repo.path,
      repo.issueSourcePreference,
      repo.connectionId ?? null,
      ...this.deps.getLocalGitArgs(repo)
    )
  }

  getGitHubRateLimit(options?: {
    force?: boolean
  }): Promise<Awaited<ReturnType<typeof getRateLimit>>> {
    return getRateLimit(options)
  }

  async listGitHubBindableAccounts(
    repoSelector: string,
    options?: { refreshCapability?: boolean }
  ): Promise<Awaited<ReturnType<typeof listGhAccountBindingInventory>>> {
    const repo = await this.deps.resolveRepo(repoSelector)
    const requiredHost = githubHostFromIdentityKey(repo.gitRemoteIdentity?.canonicalKey)
    return listGhAccountBindingInventory(this.deps.getLocalGitArgs(repo)[0] ?? {}, {
      refreshCapability: options?.refreshCapability === true,
      ...(requiredHost ? { requiredHost } : {})
    })
  }

  async validateGitHubAccountBinding(
    repoSelector: string,
    binding: { host: string; user: string }
  ): Promise<Awaited<ReturnType<typeof validateGhAccountBinding>>> {
    const repo = await this.deps.resolveRepo(repoSelector)
    const requiredHost = githubHostFromIdentityKey(repo.gitRemoteIdentity?.canonicalKey)
    return validateGhAccountBinding(
      binding,
      this.deps.getLocalGitArgs(repo)[0] ?? {},
      requiredHost ? { requiredHost } : {}
    )
  }
}