All files / src/entities/files schema.ts

100% Statements 8/8
100% Branches 0/0
100% Functions 0/0
100% Lines 8/8

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 7019x 19x                   19x     19x               19x                               19x                       19x               19x                      
import { z } from "zod";
import { flexibleBoolean, requiredId } from "../utils";
 
// ============================================================================
// manage_files - CQRS Command Tool (discriminated union schema)
// Actions: single, batch, upload
// Uses z.discriminatedUnion() for type-safe action handling.
// Schema pipeline flattens to flat JSON Schema for AI clients that don't support oneOf.
// ============================================================================
 
// --- Shared fields ---
const projectIdField = requiredId.describe("Project ID or URL-encoded path");
 
// --- Helper schema for batch file actions ---
const BatchFileActionSchema = z.object({
  file_path: z.string().describe("Path to the file"),
  content: z.string().describe("File content"),
  encoding: z.enum(["text", "base64"]).optional().describe("Content encoding"),
  execute_filemode: flexibleBoolean.optional().describe("Set executable permission"),
});
 
// --- Action: single ---
const SingleActionSchema = z.object({
  action: z.literal("single").describe("Create or update a single file"),
  project_id: projectIdField,
  file_path: z.string().describe("Path to the file"),
  content: z.string().describe("File content (text or base64 encoded)"),
  commit_message: z.string().describe("Commit message"),
  branch: z.string().describe("Target branch name"),
  start_branch: z.string().optional().describe("Base branch to start from"),
  encoding: z.enum(["text", "base64"]).optional().describe("Content encoding (default: text)"),
  author_email: z.string().optional().describe("Commit author email"),
  author_name: z.string().optional().describe("Commit author name"),
  last_commit_id: z.string().optional().describe("Last known commit ID for conflict detection"),
  execute_filemode: flexibleBoolean.optional().describe("Set executable permission"),
});
 
// --- Action: batch ---
const BatchActionSchema = z.object({
  action: z.literal("batch").describe("Commit multiple files atomically"),
  project_id: projectIdField,
  branch: z.string().describe("Target branch name"),
  commit_message: z.string().describe("Commit message"),
  files: z.array(BatchFileActionSchema).min(1).describe("Files to commit (at least one required)"),
  start_branch: z.string().optional().describe("Base branch to start from"),
  author_email: z.string().optional().describe("Commit author email"),
  author_name: z.string().optional().describe("Commit author name"),
});
 
// --- Action: upload ---
const UploadActionSchema = z.object({
  action: z.literal("upload").describe("Upload a file as markdown attachment"),
  project_id: projectIdField,
  file: z.string().describe("Base64 encoded file content"),
  filename: z.string().describe("Name of the file"),
});
 
// --- Discriminated union combining all actions ---
export const ManageFilesSchema = z.discriminatedUnion("action", [
  SingleActionSchema,
  BatchActionSchema,
  UploadActionSchema,
]);
 
// ============================================================================
// Type exports
// ============================================================================
 
export type ManageFilesInput = z.infer<typeof ManageFilesSchema>;