63 lines
1.4 KiB
TypeScript
63 lines
1.4 KiB
TypeScript
import { call, os } from "@orpc/server";
|
|
import { z } from "zod";
|
|
import { randomUUID } from "crypto";
|
|
import { createKV } from "@/server/lib/create-kv";
|
|
|
|
const TreatmentSchema = z.object({
|
|
id: z.string(),
|
|
name: z.string(),
|
|
description: z.string(),
|
|
duration: z.number(), // duration in minutes
|
|
price: z.number(), // price in cents
|
|
category: z.string(),
|
|
});
|
|
|
|
type Treatment = z.output<typeof TreatmentSchema>;
|
|
|
|
const kv = createKV<Treatment>("treatments");
|
|
|
|
const create = os
|
|
.input(TreatmentSchema.omit({ id: true }))
|
|
.handler(async ({ input }) => {
|
|
const id = randomUUID();
|
|
const treatment = { id, ...input };
|
|
await kv.setItem(id, treatment);
|
|
return treatment;
|
|
});
|
|
|
|
const update = os
|
|
.input(TreatmentSchema)
|
|
.handler(async ({ input }) => {
|
|
await kv.setItem(input.id, input);
|
|
return input;
|
|
});
|
|
|
|
const remove = os.input(z.string()).handler(async ({ input }) => {
|
|
await kv.removeItem(input);
|
|
});
|
|
|
|
const list = os.handler(async () => {
|
|
return kv.getAllItems();
|
|
});
|
|
|
|
const get = os.input(z.string()).handler(async ({ input }) => {
|
|
return kv.getItem(input);
|
|
});
|
|
|
|
const live = {
|
|
list: os.handler(async function* ({ signal }) {
|
|
yield call(list, {}, { signal });
|
|
for await (const _ of kv.subscribe()) {
|
|
yield call(list, {}, { signal });
|
|
}
|
|
}),
|
|
};
|
|
|
|
export const router = {
|
|
create,
|
|
update,
|
|
remove,
|
|
list,
|
|
get,
|
|
live,
|
|
}; |