Implementing Drizzle to add views and like for post

server/api/posts/[slug].post.ts

@@ -0,0 +1,17 @@
+import { useValidatedParams, z } from 'h3-zod'
+
+export default defineEventHandler(async (event) => {
+  const { slug } = await useValidatedParams(event, {
+    slug: z.string()
+  })
+  return useDB().insert(tables.posts).values({
+    slug
+  }).onConflictDoUpdate({
+    target: tables.posts.slug,
+    set: {
+      slug,
+      views: sql`${tables.posts.views}
+      + 1`
+    }
+  }).returning().get()
+})

server/api/posts/index.get.ts

@@ -0,0 +1,3 @@
+export default defineEventHandler(() => {
+  return useDB().query.posts.findMany()
+})

server/api/posts/like/[slug].put.ts

@@ -0,0 +1,13 @@
+import { useValidatedParams, z } from 'h3-zod'
+
+export default defineEventHandler(async (event) => {
+  const { slug } = await useValidatedParams(event, {
+    slug: z.string()
+  })
+  return useDB().update(tables.posts)
+    .set({
+      likes: sql`${tables.posts.likes}
+      + 1`
+    })
+    .where(eq(tables.posts.slug, slug))
+})

server/database/schema.ts

@@ -0,0 +1,9 @@
+import { integer, sqliteTable, text } from 'drizzle-orm/sqlite-core'
+import { sql } from 'drizzle-orm'
+
+export const posts = sqliteTable('posts', {
+  slug: text('slug').primaryKey(),
+  likes: integer('likes').default(0),
+  views: integer('views').default(0),
+  createdAt: text('created_at').default(sql`(CURRENT_DATE)`)
+})

server/plugins/migrations.ts

@@ -0,0 +1,16 @@
+import { consola } from 'consola'
+import { migrate } from 'drizzle-orm/d1/migrator'
+
+export default defineNitroPlugin(async () => {
+  if (!import.meta.dev) return
+
+  onHubReady(async () => {
+    await migrate(useDB(), { migrationsFolder: 'server/database/migrations' })
+      .then(() => {
+        consola.success('Database migrations done')
+      })
+      .catch((err) => {
+        consola.error('Database migrations failed', err)
+      })
+  })
+})

server/utils/db.ts

@@ -0,0 +1,10 @@
+import { drizzle } from 'drizzle-orm/d1'
+import * as schema from '../database/schema'
+
+export { sql, eq, and, or, asc, desc, sum } from 'drizzle-orm'
+
+export const tables = schema
+
+export function useDB() {
+  return drizzle(hubDatabase(), { schema })
+}