115 lines
3.6 KiB
TypeScript
115 lines
3.6 KiB
TypeScript
import { experimental_trpcMiddleware, TRPCError } from "@trpc/server";
|
|
import { and, desc, eq, inArray, notExists } from "drizzle-orm";
|
|
import { z } from "zod";
|
|
|
|
import { db, SqliteError } from "@lifetracker/db";
|
|
import { categories, days, hours, users } from "@lifetracker/db/schema";
|
|
import {
|
|
zDaySchema, ZDay,
|
|
ZHour
|
|
} from "@lifetracker/shared/types/days";
|
|
import type { Context } from "../index";
|
|
import { authedProcedure, router } from "../index";
|
|
import { dateFromInput, hoursListInUTC } from "@lifetracker/shared/utils/days";
|
|
import { closestIndexTo, format } from "date-fns";
|
|
import { TZDate } from "@date-fns/tz";
|
|
import { hoursAppRouter, hourColors, hourJoinsQuery, getHours, createHour } from "./hours";
|
|
import spacetime from "spacetime";
|
|
import { getHourFromTime, getTimeFromHour } from "@lifetracker/shared/utils/hours";
|
|
|
|
export async function createDay(date: string, ctx: Context) {
|
|
return await ctx.db.transaction(async (trx) => {
|
|
try {
|
|
// Create the Day object
|
|
const dayRes = await trx
|
|
.insert(days)
|
|
.values({
|
|
userId: ctx.user!.id,
|
|
date: date,
|
|
})
|
|
.returning({
|
|
id: days.id,
|
|
date: days.date,
|
|
mood: days.mood,
|
|
comment: days.comment,
|
|
});
|
|
|
|
return dayRes[0];
|
|
} catch (e) {
|
|
if (e instanceof SqliteError) {
|
|
if (e.code == "SQLITE_CONSTRAINT_UNIQUE") {
|
|
throw new TRPCError({
|
|
code: "BAD_REQUEST",
|
|
message: "This day already exists",
|
|
});
|
|
}
|
|
}
|
|
throw new TRPCError({
|
|
code: "INTERNAL_SERVER_ERROR",
|
|
message: "Something went wrong",
|
|
});
|
|
}
|
|
});
|
|
}
|
|
|
|
export async function getDay(ctx: Context, input: { dateQuery: string, timezone: string }) {
|
|
const dayDate = dateFromInput(input);
|
|
const dayRes = await ctx.db.select({
|
|
id: days.id,
|
|
date: days.date,
|
|
mood: days.mood,
|
|
comment: days.comment,
|
|
})
|
|
.from(days)
|
|
.where(eq(days.date, dayDate));
|
|
|
|
const day = dayRes.length == 0
|
|
? (await createDay(dayDate, ctx))
|
|
: (dayRes[0]);
|
|
|
|
const dayHours: ZHour[] = await getHours(ctx, input);
|
|
|
|
return {
|
|
hours: dayHours,
|
|
...day
|
|
}
|
|
}
|
|
|
|
export const daysAppRouter = router({
|
|
get: authedProcedure
|
|
.input(z.object({
|
|
dateQuery: z.string(),
|
|
timezone: z.string(),
|
|
}))
|
|
.output(zDaySchema)
|
|
.query(async ({ input, ctx }) => {
|
|
// Get a Day
|
|
return await getDay(ctx, input);
|
|
|
|
}),
|
|
update: authedProcedure
|
|
.input(
|
|
z.object({
|
|
mood: z.string().optional().or(z.number()),
|
|
comment: z.string().optional(),
|
|
dateQuery: z.string(),
|
|
timezone: z.string().optional(),
|
|
}),
|
|
)
|
|
.mutation(async ({ input, ctx }) => {
|
|
const { dateQuery, timezone, ...updatedProps } = input;
|
|
|
|
// Convert mood to number, if it exists
|
|
if (updatedProps.mood) {
|
|
updatedProps.mood = parseInt(updatedProps.mood);
|
|
}
|
|
const res = await ctx.db
|
|
.update(days)
|
|
.set(updatedProps)
|
|
.where(eq(days.date, dateFromInput({ dateQuery: dateQuery, timezone: timezone ?? ctx.user.timezone })
|
|
)).returning();
|
|
|
|
return res[0];
|
|
}),
|
|
});
|