213 lines · 7.4 KB · TypeScript
RawDownload
import { Option, Schema } from "effect";import { NonNegativeInt, PositiveInt, TrimmedNonEmptyString } from "./baseSchemas";import { DEFAULT_GIT_TEXT_GENERATION_MODEL } from "./model"; const TrimmedNonEmptyStringSchema = TrimmedNonEmptyString; // Domain Types export const GitStackedAction = Schema.Literals(["commit", "commit_push", "commit_push_pr"]);export type GitStackedAction = typeof GitStackedAction.Type;const GitCommitStepStatus = Schema.Literals(["created", "skipped_no_changes"]);const GitPushStepStatus = Schema.Literals([  "pushed",  "skipped_not_requested",  "skipped_up_to_date",]);const GitBranchStepStatus = Schema.Literals(["created", "skipped_not_requested"]);const GitPrStepStatus = Schema.Literals(["created", "opened_existing", "skipped_not_requested"]);const GitStatusPrState = Schema.Literals(["open", "closed", "merged"]);const GitPullRequestReference = TrimmedNonEmptyStringSchema;const GitPullRequestState = Schema.Literals(["open", "closed", "merged"]);const GitPreparePullRequestThreadMode = Schema.Literals(["local", "worktree"]); export const GitBranch = Schema.Struct({  name: TrimmedNonEmptyStringSchema,  isRemote: Schema.optional(Schema.Boolean),  remoteName: Schema.optional(TrimmedNonEmptyStringSchema),  current: Schema.Boolean,  isDefault: Schema.Boolean,  worktreePath: TrimmedNonEmptyStringSchema.pipe(Schema.NullOr),});export type GitBranch = typeof GitBranch.Type; const GitWorktree = Schema.Struct({  path: TrimmedNonEmptyStringSchema,  branch: TrimmedNonEmptyStringSchema,});const GitResolvedPullRequest = Schema.Struct({  number: PositiveInt,  title: TrimmedNonEmptyStringSchema,  url: Schema.String,  baseBranch: TrimmedNonEmptyStringSchema,  headBranch: TrimmedNonEmptyStringSchema,  state: GitPullRequestState,});export type GitResolvedPullRequest = typeof GitResolvedPullRequest.Type; // RPC Inputs export const GitStatusInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,});export type GitStatusInput = typeof GitStatusInput.Type; export const GitPullInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,});export type GitPullInput = typeof GitPullInput.Type; export const GitRunStackedActionInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,  action: GitStackedAction,  commitMessage: Schema.optional(TrimmedNonEmptyStringSchema.check(Schema.isMaxLength(10_000))),  featureBranch: Schema.optional(Schema.Boolean),  filePaths: Schema.optional(    Schema.Array(TrimmedNonEmptyStringSchema).check(Schema.isMinLength(1)),  ),  textGenerationModel: Schema.optional(TrimmedNonEmptyStringSchema).pipe(    Schema.withConstructorDefault(() => Option.some(DEFAULT_GIT_TEXT_GENERATION_MODEL)),  ),});export type GitRunStackedActionInput = typeof GitRunStackedActionInput.Type; export const GitListBranchesInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,});export type GitListBranchesInput = typeof GitListBranchesInput.Type; export const GitCreateWorktreeInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,  branch: TrimmedNonEmptyStringSchema,  newBranch: Schema.optional(TrimmedNonEmptyStringSchema),  path: Schema.NullOr(TrimmedNonEmptyStringSchema),});export type GitCreateWorktreeInput = typeof GitCreateWorktreeInput.Type; export const GitPullRequestRefInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,  reference: GitPullRequestReference,});export type GitPullRequestRefInput = typeof GitPullRequestRefInput.Type; export const GitPreparePullRequestThreadInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,  reference: GitPullRequestReference,  mode: GitPreparePullRequestThreadMode,});export type GitPreparePullRequestThreadInput = typeof GitPreparePullRequestThreadInput.Type; export const GitRemoveWorktreeInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,  path: TrimmedNonEmptyStringSchema,  force: Schema.optional(Schema.Boolean),});export type GitRemoveWorktreeInput = typeof GitRemoveWorktreeInput.Type; export const GitCreateBranchInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,  branch: TrimmedNonEmptyStringSchema,});export type GitCreateBranchInput = typeof GitCreateBranchInput.Type; export const GitCheckoutInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,  branch: TrimmedNonEmptyStringSchema,});export type GitCheckoutInput = typeof GitCheckoutInput.Type; export const GitInitInput = Schema.Struct({  cwd: TrimmedNonEmptyStringSchema,});export type GitInitInput = typeof GitInitInput.Type; // RPC Results const GitStatusPr = Schema.Struct({  number: PositiveInt,  title: TrimmedNonEmptyStringSchema,  url: Schema.String,  baseBranch: TrimmedNonEmptyStringSchema,  headBranch: TrimmedNonEmptyStringSchema,  state: GitStatusPrState,}); export const GitStatusResult = Schema.Struct({  branch: TrimmedNonEmptyStringSchema.pipe(Schema.NullOr),  hasWorkingTreeChanges: Schema.Boolean,  workingTree: Schema.Struct({    files: Schema.Array(      Schema.Struct({        path: TrimmedNonEmptyStringSchema,        insertions: NonNegativeInt,        deletions: NonNegativeInt,      }),    ),    insertions: NonNegativeInt,    deletions: NonNegativeInt,  }),  hasUpstream: Schema.Boolean,  aheadCount: NonNegativeInt,  behindCount: NonNegativeInt,  pr: Schema.NullOr(GitStatusPr),});export type GitStatusResult = typeof GitStatusResult.Type; export const GitListBranchesResult = Schema.Struct({  branches: Schema.Array(GitBranch),  isRepo: Schema.Boolean,  hasOriginRemote: Schema.Boolean,});export type GitListBranchesResult = typeof GitListBranchesResult.Type; export const GitCreateWorktreeResult = Schema.Struct({  worktree: GitWorktree,});export type GitCreateWorktreeResult = typeof GitCreateWorktreeResult.Type; export const GitResolvePullRequestResult = Schema.Struct({  pullRequest: GitResolvedPullRequest,});export type GitResolvePullRequestResult = typeof GitResolvePullRequestResult.Type; export const GitPreparePullRequestThreadResult = Schema.Struct({  pullRequest: GitResolvedPullRequest,  branch: TrimmedNonEmptyStringSchema,  worktreePath: TrimmedNonEmptyStringSchema.pipe(Schema.NullOr),});export type GitPreparePullRequestThreadResult = typeof GitPreparePullRequestThreadResult.Type; export const GitRunStackedActionResult = Schema.Struct({  action: GitStackedAction,  branch: Schema.Struct({    status: GitBranchStepStatus,    name: Schema.optional(TrimmedNonEmptyStringSchema),  }),  commit: Schema.Struct({    status: GitCommitStepStatus,    commitSha: Schema.optional(TrimmedNonEmptyStringSchema),    subject: Schema.optional(TrimmedNonEmptyStringSchema),  }),  push: Schema.Struct({    status: GitPushStepStatus,    branch: Schema.optional(TrimmedNonEmptyStringSchema),    upstreamBranch: Schema.optional(TrimmedNonEmptyStringSchema),    setUpstream: Schema.optional(Schema.Boolean),  }),  pr: Schema.Struct({    status: GitPrStepStatus,    url: Schema.optional(Schema.String),    number: Schema.optional(PositiveInt),    baseBranch: Schema.optional(TrimmedNonEmptyStringSchema),    headBranch: Schema.optional(TrimmedNonEmptyStringSchema),    title: Schema.optional(TrimmedNonEmptyStringSchema),  }),});export type GitRunStackedActionResult = typeof GitRunStackedActionResult.Type; export const GitPullResult = Schema.Struct({  status: Schema.Literals(["pulled", "skipped_up_to_date"]),  branch: TrimmedNonEmptyStringSchema,  upstreamBranch: TrimmedNonEmptyStringSchema.pipe(Schema.NullOr),});export type GitPullResult = typeof GitPullResult.Type;