// src/schemas/flux/compose.ts
import { z } from "zod";
import {
responseFormatSchema,
paginationSchema,
hostSchema,
projectSchema,
preprocessWithDiscriminator,
jsFilterSchema
} from "../common.js";
import { DEFAULT_LOG_LINES, MAX_LOG_LINES } from "../../constants.js";
/**
* Compose subaction schemas for Flux tool (9 subactions)
*/
export const composeListSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:list"),
action: z.literal("compose"),
subaction: z.literal("list"),
host: hostSchema.optional(),
name_filter: z.string().optional().describe("Partial match on project name"),
...paginationSchema.shape,
response_format: responseFormatSchema
})
.describe("List all Docker Compose projects")
);
export const composeStatusSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:status"),
action: z.literal("compose"),
subaction: z.literal("status"),
host: hostSchema.optional(),
project: projectSchema,
service_filter: z.string().optional().describe("Filter to specific service(s)"),
...paginationSchema.shape,
response_format: responseFormatSchema
})
.describe("Get Docker Compose project status")
);
export const composeUpSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:up"),
action: z.literal("compose"),
subaction: z.literal("up"),
host: hostSchema.optional(),
project: projectSchema,
detach: z.boolean().default(true).describe("Run in background"),
response_format: responseFormatSchema
})
.describe("Start a Docker Compose project")
);
export const composeDownSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:down"),
action: z.literal("compose"),
subaction: z.literal("down"),
host: hostSchema.optional(),
project: projectSchema,
remove_volumes: z.boolean().default(false).describe("Delete volumes (destructive!)"),
force: z.boolean().default(false).describe("Required when remove_volumes=true to confirm destructive action"),
response_format: responseFormatSchema
})
.refine(
(data) => !data.remove_volumes || data.force,
{
message: "force=true is required when remove_volumes=true to prevent accidental data loss",
path: ["force"]
}
)
.describe("Stop a Docker Compose project")
);
export const composeRestartSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:restart"),
action: z.literal("compose"),
subaction: z.literal("restart"),
host: hostSchema.optional(),
project: projectSchema,
response_format: responseFormatSchema
})
.describe("Restart a Docker Compose project")
);
export const composeLogsSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:logs"),
action: z.literal("compose"),
subaction: z.literal("logs"),
host: hostSchema.optional(),
project: projectSchema,
service: z.string().optional().describe("Target specific service"),
lines: z.number().int().min(1).max(MAX_LOG_LINES).default(DEFAULT_LOG_LINES),
since: z.string().optional(),
until: z.string().optional(),
grep: jsFilterSchema.optional(),
response_format: responseFormatSchema
})
.describe("Get Docker Compose project logs")
);
export const composeBuildSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:build"),
action: z.literal("compose"),
subaction: z.literal("build"),
host: hostSchema.optional(),
project: projectSchema,
service: z.string().optional().describe("Target specific service"),
no_cache: z.boolean().default(false).describe("Rebuild from scratch"),
response_format: responseFormatSchema
})
.describe("Build Docker Compose project images")
);
export const composePullSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:pull"),
action: z.literal("compose"),
subaction: z.literal("pull"),
host: hostSchema.optional(),
project: projectSchema,
service: z.string().optional(),
response_format: responseFormatSchema
})
.describe("Pull Docker Compose project images")
);
export const composeRecreateSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:recreate"),
action: z.literal("compose"),
subaction: z.literal("recreate"),
host: hostSchema.optional(),
project: projectSchema,
service: z.string().optional(),
response_format: responseFormatSchema
})
.describe("Recreate Docker Compose project containers")
);
export const composeRefreshSchema = z.preprocess(
preprocessWithDiscriminator,
z
.object({
action_subaction: z.literal("compose:refresh"),
action: z.literal("compose"),
subaction: z.literal("refresh"),
host: hostSchema,
response_format: responseFormatSchema
})
.describe("Refresh compose project cache by scanning filesystem")
);
// Type exports for each compose subaction
export type ComposeListInput = z.infer<typeof composeListSchema>;
export type ComposeStatusInput = z.infer<typeof composeStatusSchema>;
export type ComposeUpInput = z.infer<typeof composeUpSchema>;
export type ComposeDownInput = z.infer<typeof composeDownSchema>;
export type ComposeRestartInput = z.infer<typeof composeRestartSchema>;
export type ComposeLogsInput = z.infer<typeof composeLogsSchema>;
export type ComposeBuildInput = z.infer<typeof composeBuildSchema>;
export type ComposePullInput = z.infer<typeof composePullSchema>;
export type ComposeRecreateInput = z.infer<typeof composeRecreateSchema>;
export type ComposeRefreshInput = z.infer<typeof composeRefreshSchema>;
/**
* Union type of all compose action inputs
* Used for type-safe handling of compose subactions
*/
export type ComposeActionInput =
| ComposeListInput
| ComposeStatusInput
| ComposeUpInput
| ComposeDownInput
| ComposeRestartInput
| ComposeLogsInput
| ComposeBuildInput
| ComposePullInput
| ComposeRecreateInput
| ComposeRefreshInput;
