55 lines
1.8 KiB
TypeScript
55 lines
1.8 KiB
TypeScript
|
import { defineAction } from "@agent-native/core/action";
|
||
|
|
import { and, eq } from "drizzle-orm";
|
||
|
|
import { z } from "zod";
|
||
|
|
|
||
|
|
import { db, schema } from "../server/db/index.js";
|
||
|
|
import { quoteFavoriteCounter, withActionSpan } from "../server/otel.js";
|
||
|
|
|
||
|
|
export default defineAction({
|
||
|
|
description:
|
||
|
|
"Favorite or unfavorite a quote for a given visitor (toggles based on current state). Returns the new favorited state and total favorite count for the quote.",
|
||
|
|
schema: z.object({
|
||
|
|
quoteId: z.coerce.number().int().describe("The quote's id"),
|
||
|
|
visitorId: z.string().min(1).describe("Anonymous visitor id (from browser storage)"),
|
||
|
|
}),
|
||
|
|
run: async (args) => {
|
||
|
|
return withActionSpan(
|
||
|
|
"toggle-favorite",
|
||
|
|
{ quoteId: args.quoteId },
|
||
|
|
async () => {
|
||
|
|
const existing = await db()
|
||
|
|
.select()
|
||
|
|
.from(schema.favorites)
|
||
|
|
.where(
|
||
|
|
and(
|
||
|
|
eq(schema.favorites.quote_id, args.quoteId),
|
||
|
|
eq(schema.favorites.visitor_id, args.visitorId),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
.limit(1);
|
||
|
|
|
||
|
|
let favorited: boolean;
|
||
|
|
if (existing.length > 0) {
|
||
|
|
await db().delete(schema.favorites).where(eq(schema.favorites.id, existing[0].id));
|
||
|
|
favorited = false;
|
||
|
|
quoteFavoriteCounter.add(1, { action: "remove" });
|
||
|
|
} else {
|
||
|
|
await db().insert(schema.favorites).values({
|
||
|
|
quote_id: args.quoteId,
|
||
|
|
visitor_id: args.visitorId,
|
||
|
|
});
|
||
|
|
favorited = true;
|
||
|
|
quoteFavoriteCounter.add(1, { action: "add" });
|
||
|
|
}
|
||
|
|
|
||
|
|
const allFavorites = await db()
|
||
|
|
.select()
|
||
|
|
.from(schema.favorites)
|
||
|
|
.where(eq(schema.favorites.quote_id, args.quoteId));
|
||
|
|
|
||
|
|
return { quoteId: args.quoteId, favorited, favoriteCount: allFavorites.length };
|
||
|
|
},
|
||
|
|
);
|
||
|
|
},
|
||
|
|
});
|