chore: optimize code logic

app/components/ExportAction.tsx

@@ -32,7 +32,6 @@ export function ExportAction() {
       })
     } else {
       setExportState({ ...exportState, range: "all", status: "running", logs: [] })
-      // 异步的
       sendJsonMessage<WSMessgae<ExportFnProps>>({
         type: "export",
         data: {
@@ -82,7 +81,7 @@ export function ExportAction() {
 
     else
       return "idle"
-  }, [exportState])
+  }, [exportState, databaseStatus])
   const allStatus = useMemo(() => {
     if (!databaseStatus.maimemo_base || (exportState.status === "running" && exportState.range === "selected"))
       return "disabled"

src/db.ts

@@ -1,8 +1,8 @@
 import path from "node:path"
-import fs, { ensureDir } from "fs-extra"
+import fs from "fs-extra"
 import Database from "libsql"
-import { databaseDir, exportedDir } from "./dir"
-import { type DatabaseStatus, type Target, targets } from "@/types"
+import { databaseDir } from "./dir"
+import type { DatabaseStatus } from "@/types"
 
 const dbOpt = {
   readonly: true,
@@ -45,33 +45,8 @@ function r(f: string) {
 }
 
 export function checkDatabases() {
-  if (databases.maimemo_base?.db === undefined)
-    databases.maimemo_base = c(r("maimemo_base.db"), `SELECT * FROM  "BK_VOC_TB" ORDER BY "id" LIMIT 2`)
-  if (databases.maimemo_cloud?.db === undefined)
-    databases.maimemo_cloud = c(r("maimemo_cloud.db"), `SELECT * FROM  "notepad" ORDER BY "id" LIMIT 2`)
-  if (databases.ecdict?.db === undefined)
-    databases.ecdict = c(r("ecdict_ultimate.db"), `SELECT * FROM  "stardict" ORDER BY "id" LIMIT 2`)
+  databases.maimemo_base = c(r("maimemo_base.db"), `SELECT * FROM  "BK_VOC_TB" ORDER BY "id" LIMIT 2`)
+  databases.maimemo_cloud = c(r("maimemo_cloud.db"), `SELECT * FROM  "notepad" ORDER BY "id" LIMIT 2`)
+  databases.ecdict = c(r("ecdict_ultimate.db"), `SELECT * FROM  "stardict" ORDER BY "id" LIMIT 2`)
   return databases
 }
-
-export async function ensureExportedFolder() {
-  await ensureDir(exportedDir)
-  return {
-    path: exportedDir,
-    status: true,
-  }
-}
-
-export async function ensureTargetFolders(folder: string) {
-  const folders: Record<Target, string> = {
-    word: "",
-    list: "",
-    translation: "",
-  }
-  for (const target of targets) {
-    folders[target] = path.join(exportedDir, folder, target)
-    await ensureDir(folders[target])
-  }
-
-  return folders
-}

src/dir.ts

@@ -1,7 +1,31 @@
 import { fileURLToPath } from "node:url"
 import { join } from "node:path"
+import { ensureDir } from "fs-extra"
+import { type Target, targets } from "@/types"
 
 // 在 esm 中 import.meta.url 应该是模块文件原始的位置，比较方便用于定位。
 export const projectDir = fileURLToPath(new URL("..", import.meta.url))
 export const databaseDir = join(projectDir, "database")
 export const exportedDir = join(projectDir, "exported")
+
+export async function ensureExportedFolder() {
+  await ensureDir(exportedDir)
+  return {
+    path: exportedDir,
+    status: true,
+  }
+}
+
+export async function ensureTargetFolders(folder: string) {
+  const folders: Record<Target, string> = {
+    word: "",
+    list: "",
+    translation: "",
+  }
+  for (const target of targets) {
+    folders[target] = join(exportedDir, folder, target)
+    await ensureDir(folders[target])
+  }
+
+  return folders
+}

src/export.ts

@@ -1,12 +1,12 @@
 import { join } from "node:path"
 import fs from "fs-extra"
-import { getLibWords, getLibs, translateAll } from "./get"
-import { checkDatabases, databases, ensureTargetFolders } from "./db"
+import { getLibWords, getLibs, translateAll } from "./query"
+import { databases } from "./db"
 import { transform } from "./transform"
+import { ensureTargetFolders } from "./dir"
 import type { ExportFnProps, ExportLog, Target, TrafficLights, Word } from "@/types"
 
 export async function exportLib({ selected, range, type, options, fnEvery }: ExportFnProps & { fnEvery: (log: ExportLog) => Promise<boolean> }) {
-  checkDatabases()
   const targetFolders = await ensureTargetFolders(options.folderName)
 
   let libs = selected

src/get.ts → src/query.ts

@@ -1,12 +1,11 @@
 import { sql } from "./sql"
 import { translateByDict } from "./translate"
 import { transform } from "./transform"
-import { checkDatabases, databases } from "./db"
+import { databases } from "./db"
 import type { LibType, Library, Target, Word } from "@/types"
 
 let memorizedWords: string[] = []
 export function getLibs(type: LibType = "base"): Library[] {
-  checkDatabases()
   if (!databases.maimemo_base?.db)
     return []
   if (type === "base") {
@@ -41,7 +40,6 @@ export function previewLibWords({
   exculedMemorized = false,
   target = "word",
 }: Props) {
-  checkDatabases()
   const words = getLibWords({ id, type, exculedMemorized })
   if (target === "translation") {
     if (!databases.maimemo_base?.db)
@@ -62,7 +60,6 @@ export function getLibWords({
   exculedMemorized = false,
 }: Omit<Props, "target">): Word[] {
   try {
-    checkDatabases()
     if (!id || !databases.maimemo_base?.db)
       return []
     let words: Word[] = []

src/translate.ts

@@ -1,7 +1,6 @@
-import { checkDatabases, databases } from "./db"
+import { databases } from "./db"
 
 export function translateByDict(word: string): string {
-  checkDatabases()
   if (databases.ecdict?.db === undefined)
     throw new Error("ecdict database not found")
   // 缩写

trpcServer.ts

@@ -2,8 +2,9 @@ import { defineEventHandler, toWebRequest } from "vinxi/http"
 import { initTRPC } from "@trpc/server"
 import { fetchRequestHandler } from "@trpc/server/adapters/fetch"
 import { z } from "zod"
-import { checkDatabases, ensureExportedFolder } from "./src/db"
-import { getLibs, previewLibWords } from "./src/get"
+import { ensureExportedFolder } from "src/dir"
+import { checkDatabases } from "./src/db"
+import { getLibs, previewLibWords } from "./src/query"
 
 const t = initTRPC.create()
 

tsconfig.json

@@ -22,5 +22,6 @@
     "allowSyntheticDefaultImports": true,
     "skipLibCheck": false
   },
-  "include": ["./app/**/*.*", "./src/**/*.ts", "./shared/**/*.ts", "./scripts/*.ts", "./*.ts", "eslint.config.js"]
+  "include": ["./**/*.ts", "./**/*.tsx"],
+  "exxclude": ["node_modules/**.*"]
 }