homeio

1import { z } from "zod";
2
3export type ScheduledTaskType = "shell" | "script" | "restart-app" | "backup" | "pull-images";
4
5export type ScheduledTaskConfig =
6  | { type: "shell"; command: string }
7  | { type: "script"; script: string }
8  | { type: "restart-app"; appId: string; appName: string }
9  | { type: "backup" }
10  | { type: "pull-images" };
11
12export type ScheduledTaskRecord = {
13  id: string;
14  label: string;
15  taskType: ScheduledTaskType;
16  taskConfig: ScheduledTaskConfig;
17  cronExpression: string;
18  enabled: boolean;
19  lastRunAt: string | null;
20  lastRunStatus: "success" | "error" | null;
21  lastRunOutput: string | null;
22  nextRunAt: string | null;
23  createdAt: string;
24};
25
26export type CreateScheduledTaskInput = {
27  label: string;
28  taskType: ScheduledTaskType;
29  taskConfig: ScheduledTaskConfig;
30  cronExpression: string;
31  enabled?: boolean;
32};
33
34export type UpdateScheduledTaskInput = {
35  label?: string;
36  taskType?: ScheduledTaskType;
37  taskConfig?: ScheduledTaskConfig;
38  cronExpression?: string;
39  enabled?: boolean;
40};
41
42export type ScheduledTasksListResponse = {
43  tasks: ScheduledTaskRecord[];
44};
45
46export const CRON_PRESETS = [
47  { label: "Every hour", value: "0 * * * *" },
48  { label: "Daily at 3 AM", value: "0 3 * * *" },
49  { label: "Weekly (Sun 3 AM)", value: "0 3 * * 0" },
50  { label: "Monthly (1st, 3 AM)", value: "0 3 1 * *" },
51  { label: "Custom…", value: "custom" },
52] as const;
53
54export const SHELL_COMMAND_ALLOWLIST = [
55  "df -h",
56  "docker system prune -f",
57  "docker image prune -af",
58  "docker volume prune -f",
59  "apt-get update",
60  "apt-get upgrade -y",
61  "apt-get autoremove -y",
62  "journalctl --vacuum-size=500M",
63  "sync && echo 3 > /proc/sys/vm/drop_caches",
64] as const;
65
66// 5-field cron: basic validation (digits, *, commas, hyphens, slashes)
67const cronField = /^(\*|[0-9,\-*/]+)$/;
68export const CRON_EXPRESSION_REGEX = /^(\*|[0-9,\-*/]+)\s+(\*|[0-9,\-*/]+)\s+(\*|[0-9,\-*/]+)\s+(\*|[0-9,\-*/]+)\s+(\*|[0-9,\-*/]+)$/;
69
70const taskConfigSchema = z.discriminatedUnion("type", [
71  z.object({ type: z.literal("shell"), command: z.enum(SHELL_COMMAND_ALLOWLIST) }),
72  z.object({ type: z.literal("script"), script: z.string().min(1).max(32_768) }),
73  z.object({ type: z.literal("restart-app"), appId: z.string().min(1), appName: z.string().min(1) }),
74  z.object({ type: z.literal("backup") }),
75  z.object({ type: z.literal("pull-images") }),
76]);
77
78export const createScheduledTaskSchema = z.object({
79  label: z.string().min(1).max(200),
80  taskType: z.enum(["shell", "script", "restart-app", "backup", "pull-images"]),
81  taskConfig: taskConfigSchema,
82  cronExpression: z.string().regex(CRON_EXPRESSION_REGEX, "Invalid cron expression"),
83  enabled: z.boolean().optional(),
84});
85
86export const updateScheduledTaskSchema = z.object({
87  label: z.string().min(1).max(200).optional(),
88  taskType: z.enum(["shell", "script", "restart-app", "backup", "pull-images"]).optional(),
89  taskConfig: taskConfigSchema.optional(),
90  cronExpression: z.string().regex(CRON_EXPRESSION_REGEX, "Invalid cron expression").optional(),
91  enabled: z.boolean().optional(),
92});
93