schemas.ts

import { CodeHostType } from '@sourcebot/db';
import z from 'zod';
import { fileTreeItemSchema, fileTreeNodeSchema } from './types';

export const getTreeRequestSchema = z.object({
    repoName: z.string(),
    revisionName: z.string(),
    paths: z.array(z.string()),
});

export const getTreeResponseSchema = z.object({
    tree: fileTreeNodeSchema,
});

export const getFilesRequestSchema = z.object({
    repoName: z.string(),
    revisionName: z.string(),
});

export const getFilesResponseSchema = z.array(fileTreeItemSchema);

export const fileSourceRequestSchema = z.object({
    path: z.string(),
    repo: z.string(),
    ref: z.string().optional(),
});

export const fileSourceResponseSchema = z.object({
    source: z.string(),
    language: z.string(),
    path: z.string(),
    repo: z.string(),
    repoCodeHostType: z.nativeEnum(CodeHostType),
    repoDisplayName: z.string().optional(),
    repoExternalWebUrl: z.string().optional(),
    webUrl: z.string(),
    externalWebUrl: z.string().optional(),
});

export const getDiffRequestSchema = z.object({
    repo: z.string().describe('The fully-qualified repository name.'),
    base: z.string().describe('The base git ref (branch, tag, or commit SHA) to diff from.'),
    head: z.string().describe('The head git ref (branch, tag, or commit SHA) to diff to.'),
    path: z.string().optional().describe('Restrict the diff to changes touching this file path. Omit to diff all changes between the two refs.'),
});

const hunkRangeSchema = z.object({
    start: z.number().int().describe('The 1-based line number where the range starts.'),
    lines: z.number().int().describe('The number of lines the range spans.'),
});

const diffHunkSchema = z.object({
    oldRange: hunkRangeSchema.describe('The line range in the old file.'),
    newRange: hunkRangeSchema.describe('The line range in the new file.'),
    heading: z.string().optional().describe('Optional context heading extracted from the @@ line, typically the enclosing function or class name.'),
    body: z.string().describe('The diff content, with each line prefixed by a space (context), + (addition), or - (deletion).'),
});

const fileDiffSchema = z.object({
    oldPath: z.string().nullable().describe('The file path before the change. `null` for added files.'),
    newPath: z.string().nullable().describe('The file path after the change. `null` for deleted files.'),
    hunks: z.array(diffHunkSchema).describe('The list of changed regions within the file.'),
});

export const getDiffResponseSchema = z.object({
    files: z.array(fileDiffSchema).describe('The list of changed files.'),
});

export const listCommitsQueryParamsSchema = z.object({
    repo: z.string().describe('The fully-qualified repository name.'),
    query: z.string().optional().describe('Filter commits by message content (case-insensitive). Interpreted as a POSIX BRE regex — escape metacharacters for literal matching.'),
    since: z.string().optional().describe('Return commits after this date. Accepts ISO 8601 or relative formats (e.g. `30 days ago`).'),
    until: z.string().optional().describe('Return commits before this date. Accepts ISO 8601 or relative formats.'),
    author: z.string().optional().describe('Filter commits by author name or email (case-insensitive). Interpreted as a POSIX BRE regex — escape metacharacters for literal matching.'),
    ref: z.string().optional().describe('The git ref (branch, tag, or commit SHA) to list commits from. Defaults to `HEAD`.'),
    path: z.string().optional().describe('Restrict commits to those that touch this file path.'),
    page: z.coerce.number().int().positive().default(1),
    perPage: z.coerce.number().int().positive().max(100).default(50),
});

export const commitSchema = z.object({
    hash: z.string().describe('The full commit SHA.'),
    date: z.string().describe('The commit date in ISO 8601 format.'),
    message: z.string().describe('The commit subject line.'),
    refs: z.string().describe('Refs pointing to this commit (e.g. branch or tag names).'),
    body: z.string().describe('The commit body (everything after the subject line).'),
    authorName: z.string(),
    authorEmail: z.string(),
});

export const getCommitQueryParamsSchema = z.object({
    repo: z.string().describe('The fully-qualified repository name.'),
    ref: z.string().describe('The git ref (commit SHA, branch, or tag).'),
});

export const commitDetailSchema = commitSchema.extend({
    parents: z.array(z.string()).describe('The parent commit SHAs.'),
});

export const listCommitAuthorsQueryParamsSchema = z.object({
    repo: z.string().describe('The fully-qualified repository name.'),
    ref: z.string().optional().describe('The git ref (branch, tag, or commit SHA) to list authors from. Defaults to `HEAD`.'),
    path: z.string().optional().describe('Restrict authors to those who touched this file path.'),
    page: z.coerce.number().int().positive().default(1),
    perPage: z.coerce.number().int().positive().max(100).default(50),
});

export const commitAuthorSchema = z.object({
    name: z.string(),
    email: z.string(),
    commitCount: z.number().int().nonnegative(),
});

export const fileBlameRequestSchema = z.object({
    path: z.string().describe('The file path to blame, relative to the repository root.'),
    repo: z.string().describe('The fully-qualified repository name.'),
    ref: z.string().optional().describe('The git ref (branch, tag, or commit SHA) to blame at. Defaults to the repository\'s default branch.'),
});

export const blameRangeSchema = z.object({
    hash: z.string().describe('The hash of the commit that last modified the lines in this range.'),
    path: z.string().describe('The file path as it existed at the attributing commit. May differ from the current path due to renames.'),
    startLine: z.number().int().positive().describe('The 1-based line number where the range begins (inclusive).'),
    lineCount: z.number().int().positive().describe('The number of contiguous lines in this range.'),
});

export const blamePreviousSchema = z.object({
    hash: z.string().describe('The hash of the commit that previously affected these lines (i.e., the next step backwards in the blame walk).'),
    path: z.string().describe('The file path as it existed at the previous commit. May differ from the current path due to renames.'),
});

export const blameCommitSchema = z.object({
    hash: z.string().describe('The full commit SHA.'),
    date: z.string().describe('The commit date in ISO 8601 format.'),
    message: z.string().describe('The commit subject line.'),
    authorName: z.string(),
    authorEmail: z.string(),
    previous: blamePreviousSchema.optional().describe('Pointer to the previous commit that affected these lines, with the file path as it existed there. Absent when the commit introduced the lines (no earlier history to walk to).'),
});

export const fileBlameResponseSchema = z.object({
    ranges: z.array(blameRangeSchema).describe('Contiguous, non-overlapping line ranges ordered by startLine. Each range is attributed to a single commit.'),
    commits: z.record(z.string(), blameCommitSchema).describe('Commit metadata keyed by hash, deduplicated across ranges.'),
});