use Next.js

packages/next/server/db.ts

@@ -0,0 +1,110 @@
+/* eslint-disable import/no-mutable-exports */
+
+import idxJson from '~/build/idx.json'
+import Loki, { Collection } from 'lokijs'
+import lunr from 'lunr'
+import * as z from 'zod'
+
+import { buildPath } from './dir'
+
+const _zMatter = {
+  priority: z.number().optional(),
+  title: z.string(),
+  image: z.string().optional(),
+  tag: z.array(z.string()).optional(),
+  category: z.array(z.string()).optional()
+}
+
+export const zMatter = z.object({
+  ..._zMatter,
+  date: z.date().optional(),
+  createdAt: z.date().optional(),
+  updatedAt: z.date().optional()
+})
+
+export type IMatter = z.infer<typeof zMatter>
+
+const zMatterSerialized = z.object({
+  ..._zMatter,
+  date: z.number().optional(),
+  createdAt: z.number().optional(),
+  updatedAt: z.number().optional()
+})
+
+export type IMatterSerialized = z.infer<typeof zMatterSerialized>
+
+export const zPost = z
+  .object({
+    path: z.string(),
+    content: z.string(),
+    excerptHtml: z.string(),
+    contentHtml: z.string()
+  })
+  .merge(zMatterSerialized)
+
+export type IPost = z.infer<typeof zPost> & {
+  path: string
+  title: string
+  content: string
+  excerptHtml: string
+  contentHtml: string
+}
+
+export type IPostTeaser = {
+  path: string
+  date?: number
+  title: string
+  image?: string
+  content: string
+  excerptHtml: string
+}
+
+export interface IPostSearch {
+  path: string
+  title: string
+  tag?: string[]
+  category?: string[]
+  /**
+   * Cleaned and jieba-separated markdown
+   */
+  content: string
+}
+
+export let db: Loki
+
+export let dbPost: Collection<IPost>
+
+export let idx: lunr.Index
+
+export async function initDatabase () {
+  if (db) {
+    return
+  }
+
+  idx = lunr.Index.load(idxJson)
+
+  return new Promise<void>((resolve) => {
+    db = new Loki(buildPath('data.loki'), {
+      autoload: true,
+      autoloadCallback: () => {
+        dbPost = db.getCollection('post')
+
+        if (!dbPost) {
+          dbPost = db.addCollection('post', {
+            unique: ['path'],
+            indices: [
+              'date',
+              'createdAt',
+              'updatedAt',
+              'priority',
+              'tag',
+              'category'
+            ]
+          })
+        }
+
+        resolve()
+      }
+    })
+  })
+}