feat(db): support Nitro useDatabase()

playground/server/api/db-test.ts → playground/server/api/tests/db.ts

@@ -1,6 +1,20 @@
 import { sqliteTable, integer, text } from 'drizzle-orm/sqlite-core'
 
 export default defineEventHandler(async () => {
+  const db0 = useDatabase()
+
+  const _tables = await db0.sql`
+    SELECT
+      name,
+      type
+    FROM
+      sqlite_schema
+    WHERE
+      type = 'table' AND
+      name NOT LIKE 'sqlite_%' and name NOT LIKE '_litestream_%' and name NOT LIKE '__drizzle%'
+    ;`
+
+  console.log(_tables.rows)
   const db = useDrizzle()
 
   const tables = await db.all(sql`

playground/server/api/tests/kv.ts

@@ -0,0 +1,12 @@
+export default eventHandler(async () => {
+  const kv = hubKV()
+
+  await kv.set('vue', { year: 2014 })
+  await kv.set('vue:nuxt', { year: 2016 })
+  await kv.set('vue:quasar', { version: 2015 })
+  await kv.set('react', { version: 2013 })
+  await kv.set('react:next', { version: 2016 })
+  await kv.set('react:gatsby', { version: 2015 })
+
+  return kv.keys()
+})

src/module/index.ts

@@ -96,6 +96,17 @@ export default defineNuxtModule<ModuleOptions>({
       }
     })
 
+    // Bind `useDatabase()` to `hubDatabase()`
+    nuxt.options.nitro.experimental = defu(nuxt.options.nitro.experimental, {
+      database: true
+    })
+    nuxt.options.nitro.database = defu(nuxt.options.nitro.database, {
+      default: {
+        connector: 'cloudflare-d1',
+        options: { bindingName: 'DB' }
+      }
+    })
+
     // nuxt prepare or production mode, stop here
     if (nuxt.options._prepare || !nuxt.options.dev) {
       return

src/module/runtime/bindings.dev.ts

@@ -46,10 +46,13 @@ async function getBindingsProxy() {
     persist: { path: wranglerConfig.persistDir },
   })
 
+  // @ts-ignore
+  globalThis.__cf_env__ = proxy.env
+
   Object.keys(proxy.env).forEach((key) => {
     // @ts-ignore
     if (!globalThis[key]) {
-    // @ts-ignore
+      // @ts-ignore
       globalThis[key] = proxy.env[key]
     }
   })