feat: validate -s and -u with regex

Author: ryoppippi

commands/daily.ts

@@ -3,6 +3,7 @@ import Table from "cli-table3";
 import { define } from "gunshi";
 import pc from "picocolors";
 import { type LoadOptions, loadUsageData } from "../data-loader.ts";
+import { parseDateArg } from "../date-validation.ts";
 import { log, logger } from "../logger.ts";
 import { formatCurrency, formatNumber } from "../utils.ts";
 
@@ -11,14 +12,16 @@ export const dailyCommand = define({
 	description: "Show usage report grouped by date",
 	args: {
 		since: {
-			type: "string",
+			type: "custom",
 			short: "s",
 			description: "Filter from date (YYYYMMDD format)",
+			parse: parseDateArg,
 		},
 		until: {
-			type: "string",
+			type: "custom",
 			short: "u",
 			description: "Filter until date (YYYYMMDD format)",
+			parse: parseDateArg,
 		},
 		path: {
 			type: "string",

commands/session.ts

@@ -3,6 +3,7 @@ import Table from "cli-table3";
 import { define } from "gunshi";
 import pc from "picocolors";
 import { type LoadOptions, loadSessionData } from "../data-loader.ts";
+import { parseDateArg } from "../date-validation.ts";
 import { log, logger } from "../logger.ts";
 import { formatCurrency, formatNumber } from "../utils.ts";
 
@@ -11,14 +12,16 @@ export const sessionCommand = define({
 	description: "Show usage report grouped by conversation session",
 	args: {
 		since: {
-			type: "string",
+			type: "custom",
 			short: "s",
 			description: "Filter from date (YYYYMMDD format)",
+			parse: parseDateArg,
 		},
 		until: {
-			type: "string",
+			type: "custom",
 			short: "u",
 			description: "Filter until date (YYYYMMDD format)",
+			parse: parseDateArg,
 		},
 		path: {
 			type: "string",

date-validation.ts

@@ -0,0 +1,10 @@
+import * as v from "valibot";
+import { dateSchema } from "./types";
+
+export const parseDateArg = (value: string): string => {
+	const result = v.safeParse(dateSchema, value);
+	if (!result.success) {
+		throw new TypeError(result.issues[0].message);
+	}
+	return result.output;
+};

types.ts

@@ -34,3 +34,8 @@ export type ModelSpec = v.InferOutput<typeof ModelSpecSchema>;
 export const LiteLLMModelPricesSchema = v.record(v.string(), ModelSpecSchema);
 
 export type LiteLLMModelPrices = v.InferOutput<typeof LiteLLMModelPricesSchema>;
+
+export const dateSchema = v.pipe(
+	v.string(),
+	v.regex(/^\d{8}$/, "Date must be in YYYYMMDD format"),
+);