nolo-cli 0.1.12 → 0.1.14

package/database/actions/write.ts

@@ -0,0 +1,126 @@
+// 文件路径: database/actions/write.ts
+
+import type { AppThunkApi } from "app/store";
+import { getRuntimeServerContext } from "database/runtimeServerContext";
+import { DataType } from "create/types";
+import { normalizeTimeFields, logger } from "./common";
+import {
+  resolveReplicationServers,
+  scheduleWriteReplication,
+} from "./replication";
+import { toast } from "app/utils/toast";
+
+// 辅助函数：保存到本地 DB
+const saveToClientDb = async (
+  clientDb: any,
+  dbKey: string,
+  data: any
+): Promise<void> => {
+  if (!clientDb) {
+    logger.error({ dbKey }, "Client database is undefined in saveToClientDb");
+    throw new Error("Client database instance is required");
+  }
+  try {
+    await clientDb.put(dbKey, data);
+    logger.debug({ dbKey }, "Data saved successfully to local database.");
+  } catch (err: any) {
+    logger.error({ err, dbKey }, "Failed to save data to local database");
+    throw new Error(`Local database put failed for ${dbKey}: ${err.message}`);
+  }
+};
+
+/**
+ * Write Action: 写入新数据项。
+ * 1. 验证数据类型。
+ * 2. 规范化数据（添加时间戳、dbKey、userId）。
+ * 3. 保存数据到本地数据库。
+ * 4. 若在线，异步将完整数据写入所有服务器。
+ */
+export const writeAction = async (
+  writeConfig: { data: any; customKey: string; userId?: string },
+  thunkApi: AppThunkApi
+): Promise<any> => {
+  const { db: clientDb } = thunkApi.extra;
+  if (!clientDb) {
+    throw new Error("Client database instance is required in writeAction");
+  }
+
+  const state = thunkApi.getState();
+  const { currentServer, syncServers, currentUserId } =
+    getRuntimeServerContext(state);
+
+  const { data, customKey } = writeConfig;
+  const userId = writeConfig.userId || currentUserId;
+
+  // 1. 基础参数校验
+  if (!data || !customKey) {
+    const errorMsg =
+      "Invalid arguments for writeAction: data and customKey are required.";
+    logger.error(errorMsg, { writeConfig });
+    toast.error(errorMsg);
+    throw new Error(errorMsg);
+  }
+
+  // 2. 类型校验（保持原有行为：非法类型只告警，不阻塞）
+  const VALID_TYPES = [
+    DataType.MSG,
+    DataType.CYBOT,
+    DataType.DOC,
+    DataType.DIALOG,
+    DataType.NOTIFICATION,
+    DataType.TOKEN,
+    DataType.TRANSACTION,
+    DataType.SPACE,
+    DataType.SETTING,
+    DataType.TABLE,
+    DataType.TABLE_ROW,
+    DataType.EMAIL,
+  ];
+  if (!data.type || !VALID_TYPES.includes(data.type)) {
+    logger.warn(
+      `Invalid data type "${data.type}" for writeAction with key ${customKey}. Proceeding anyway.`
+    );
+  }
+
+  try {
+    // 3. 规范化数据（时间字段 / dbKey / userId）
+    const willSaveData = normalizeTimeFields({
+      ...data,
+      dbKey: customKey,
+      userId,
+    });
+
+    // 4. 本地保存
+    await saveToClientDb(clientDb, customKey, willSaveData);
+
+    // 5. 计算远程服务器列表（currentServer + syncServers，带去重 + 离线判断）
+    const servers = resolveReplicationServers(currentServer, syncServers);
+
+    const serverWriteConfig = {
+      data: willSaveData,
+      customKey,
+      userId,
+    };
+
+    // 6. 后台异步同步到远程（若在线且有可用服务器）
+    if (servers.length > 0) {
+      logger.debug(
+        `[writeAction] Initiating background sync for key: ${customKey} to ${servers.length} servers.`
+      );
+      scheduleWriteReplication(servers, serverWriteConfig, state);
+    } else {
+      logger.warn(
+        "[writeAction] No available servers, data only saved locally.",
+        { customKey }
+      );
+    }
+
+    return willSaveData;
+  } catch (error: any) {
+    const errorMessage = `Write action failed for ${customKey}: ${error?.message || "Unknown error"
+      }`;
+    logger.error("[writeAction] Error:", error);
+    toast.error(`Failed to save data for ${customKey}.`);
+    throw new Error(errorMessage);
+  }
+};

package/database/client/db.native.ts

@@ -0,0 +1,73 @@
+// database/client/db.native.ts
+// React Native 版数据库 - 使用 @nolo/leveldb (monorepo 内部包)
+
+import Level from "@nolo/leveldb";
+
+// 扩展全局对象类型
+declare global {
+    var noloDbInstance: Level | null;
+}
+
+/**
+ * 获取或创建数据库实例
+ * 使用全局变量存储实例，防止 Fast Refresh 导致模块重载时丢失引用而重复打开数据库
+ */
+export function getDb(): Level {
+    if (!global.noloDbInstance) {
+        console.log('[DB Native] Creating new LevelDB instance');
+        try {
+            global.noloDbInstance = new Level("nolo", { valueEncoding: "json" });
+        } catch (e: any) {
+            console.error('[DB Native] Failed to open database:', e?.message ?? e);
+            throw new Error(
+                `[DB Native] 数据库启动失败，请尝试重启应用。原因: ${e?.message ?? '未知错误'}`
+            );
+        }
+    }
+    return global.noloDbInstance;
+}
+
+/**
+ * 创建数据库实例（用于 store 初始化）
+ */
+export function createDb(): Level {
+    return getDb();
+}
+
+/**
+ * 关闭当前数据库实例
+ */
+export async function closeDb(): Promise<void> {
+    if (global.noloDbInstance) {
+        try {
+            await global.noloDbInstance.close();
+            console.log('[DB Native] Database closed');
+        } catch (e) {
+            console.warn('[DB Native] Error closing database:', e);
+        }
+        global.noloDbInstance = null;
+    }
+}
+
+/**
+ * 关闭所有数据库（用于热重载清理）
+ */
+export async function closeAllDatabases(): Promise<void> {
+    try {
+        global.noloDbInstance = null;
+        await Level.closeAll();
+        console.log('[DB Native] All databases closed');
+    } catch (e) {
+        console.warn('[DB Native] Failed to close all databases:', e);
+    }
+}
+
+/**
+ * 检查是否为 React Native 环境
+ */
+export const isNative = true;
+
+// 导出数据库实例（兼容旧代码）
+export const browserDb = getDb();
+
+export default Level;

package/database/client/db.ts

@@ -0,0 +1,51 @@
+// database/client/db.ts
+// Web 版数据库 - 使用 level 包 (基于 IndexedDB)
+
+import { Level } from "level";
+
+// 数据库单例
+let dbInstance: Level<string, any> | null = null;
+
+/**
+ * 获取或创建数据库实例
+ */
+export function getDb(): Level<string, any> {
+    if (!dbInstance) {
+        dbInstance = new Level("nolo", { valueEncoding: "json" });
+    }
+    return dbInstance;
+}
+
+/**
+ * 创建数据库实例（用于 store 初始化）
+ */
+export function createDb(): Level<string, any> {
+    return getDb();
+}
+
+/**
+ * 关闭数据库
+ */
+export async function closeDb(): Promise<void> {
+    if (dbInstance) {
+        await dbInstance.close();
+        dbInstance = null;
+    }
+}
+
+/**
+ * 关闭所有数据库（Web 版只有一个实例）
+ */
+export async function closeAllDatabases(): Promise<void> {
+    await closeDb();
+}
+
+/**
+ * 检查是否为 React Native 环境
+ */
+export const isNative = false;
+
+// 导出数据库实例（兼容旧代码）
+export const browserDb = getDb();
+
+export default Level;

package/database/client/fetchUserData.ts

@@ -0,0 +1,61 @@
+import { getUserDataPrefixes } from "../queryPrefixes";
+
+interface FetchUserDataOptions {
+    includeDeleted?: boolean;
+}
+
+const attachQueriedKey = (key: string, value: any) => {
+    if (!value || typeof value !== "object") return value;
+    if (
+        typeof value.dbKey === "string" && value.dbKey.trim().length > 0
+    ) {
+        return value;
+    }
+    return {
+        ...value,
+        dbKey: key,
+    };
+};
+
+// 支持单类型或多类型查询
+// db should be passed from caller (e.g. thunk extra)
+export async function fetchUserData(
+    db: any,
+    types: string | string[],
+    userId: string,
+    options: FetchUserDataOptions = {}
+) {
+    const results: Record<string, any[]> = {};
+    const typeArray = Array.isArray(types) ? types : [types];
+    const includeDeleted = options.includeDeleted === true;
+
+    try {
+        for (const type of typeArray) {
+            const prefixes = getUserDataPrefixes(type, userId);
+            results[type] = [];
+            for (const prefix of prefixes) {
+                let iterator = db.iterator({
+                    gte: prefix,
+                    lte: `${prefix}\uffff`,
+                });
+
+                if (iterator && typeof iterator.then === "function") {
+                    iterator = await iterator;
+                }
+
+                // @ts-ignore - iterator compatibility
+                for await (const [key, value] of iterator) {
+                    const hydrated = attachQueriedKey(String(key), value);
+                    if (!hydrated) continue;
+                    if (!includeDeleted && hydrated.deletedAt) continue;
+                    results[type].push(hydrated);
+                }
+            }
+        }
+
+        return Array.isArray(types) ? results : results[types] ?? [];
+    } catch (error) {
+        console.error("Query error:", error);
+        throw error;
+    }
+}

package/database/client/handleError.ts

@@ -0,0 +1,19 @@
+import i18n from "app/i18n";
+export const handleError = (error, handleUnauthorized?) => {
+  let message;
+  switch (error.message) {
+    case "400":
+      message = i18n.t("errors.validationError");
+      break;
+    case "401":
+      message = i18n.t("errors.unauthorized");
+      handleUnauthorized && handleUnauthorized();
+      break;
+    case "500":
+    default:
+      message = i18n.t("errors.serverError");
+      break;
+  }
+
+  return message;
+};

package/database/client/queryRequest.ts

@@ -0,0 +1,21 @@
+import { API_ENDPOINTS } from "database/config";
+
+export const noloQueryRequest = async (queryConfig: any) => {
+  const { server } = queryConfig;
+  const { queryUserId, options } = queryConfig;
+
+  const queryParams = new URLSearchParams({
+    limit: options.limit?.toString() ?? "",
+  });
+  const url = `${API_ENDPOINTS.DATABASE}/query/${queryUserId}?${queryParams}`;
+  const fullUrl = server + url;
+  let headers = {
+    "Content-Type": "application/json",
+  };
+  const body = JSON.stringify(options.condition);
+  return fetch(fullUrl, {
+    method: "POST",
+    headers,
+    body,
+  });
+};

package/database/config.ts

@@ -0,0 +1,21 @@
+// packages/database/config.ts
+
+// 这个文件现在是前后端通用的，不包含任何后端模块如 'fs' 或 'path'。
+export const API_VERSION = "/api/v1";
+export const SERVERS = {
+  MAIN: "https://nolo.chat",
+  US: "https://us.nolo.chat",
+} as const;
+export const NOLO_CLUSTER_SERVERS = Object.values(SERVERS);
+
+export const API_ENDPOINTS = {
+  DATABASE: `${API_VERSION}/db`,
+  SHARE: `${API_VERSION}/share`,
+  USERS: `${API_VERSION}/users`,
+  WEATHER: `${API_VERSION}/weather`,
+  HI: `${API_VERSION}/hi`,
+  CHAT: `${API_VERSION}/chat`,
+  EXECUTE_SQL: `${API_VERSION}/sqlite/execute_sql`,
+  // --- 新增端点 ---
+  TRANSACTIONS: `${API_VERSION}/transactions`,
+};

package/database/dbActionThunks.ts

@@ -0,0 +1 @@
+export { readAndWait, patch, upsert, write } from "./dbSlice";

package/database/dbSlice.ts

@@ -0,0 +1,149 @@
+// 文件路径: database/dbSlice.ts
+
+import {
+  asyncThunkCreator,
+  buildCreateSlice,
+  createEntityAdapter,
+  type PayloadAction,
+} from "@reduxjs/toolkit";
+import type { RootState } from "app/store";
+
+// Import actions
+import { removeAction } from "./actions/remove";
+import { readAction } from "./actions/read";
+import { readAndWaitAction } from "./actions/readAndWait";
+import { writeAction } from "./actions/write";
+import { patchAction } from "./actions/patch";
+import { upsertAction } from "./actions/upsert";
+import { uploadFileAction } from "./actions/upload";
+import { readFileContentAction } from "./actions/fileContent";
+import { shareResourceAction } from "share/action";
+
+// Use dbKey as the entity's unique identifier
+export const dbAdapter = createEntityAdapter<any>({
+  selectId: (entity: any) => entity.dbKey,
+});
+
+// Selectors
+export const {
+  selectById,
+  selectEntities,
+  selectAll,
+  selectIds,
+  selectTotal,
+} = dbAdapter.getSelectors((state: RootState) => state.db);
+
+// Initial state
+const initialState = dbAdapter.getInitialState({});
+
+// Create slice with async thunks
+const createSliceWithThunks = buildCreateSlice({
+  creators: { asyncThunk: asyncThunkCreator },
+});
+
+// Slice definition
+const dbSlice = createSliceWithThunks({
+  name: "db",
+  initialState,
+  reducers: (create) => ({
+    // Async Thunks
+    read: create.asyncThunk(readAction, {
+      fulfilled: (state, action) => {
+        if (action.payload && Object.keys(action.payload).length > 0) {
+          dbAdapter.upsertOne(state, action.payload);
+        }
+      },
+    }),
+    readAndWait: create.asyncThunk(readAndWaitAction, {
+      fulfilled: (state, action) => {
+        if (action.payload && Object.keys(action.payload).length > 0) {
+          dbAdapter.upsertOne(state, action.payload);
+        }
+      },
+    }),
+    remove: create.asyncThunk(removeAction, {
+      fulfilled: (state, action) => {
+        const { dbKey } = action.payload;
+        if (dbKey) dbAdapter.removeOne(state, dbKey);
+      },
+    }),
+    write: create.asyncThunk(writeAction, {
+      fulfilled: (state, action) => {
+        if (
+          action.payload &&
+          action.payload.dbKey &&
+          Object.keys(action.payload).length > 0
+        ) {
+          dbAdapter.upsertOne(state, action.payload);
+        }
+      },
+    }),
+    patch: create.asyncThunk(patchAction, {
+      fulfilled: (state, action) => {
+        const { payload } = action;
+        if (payload && payload.dbKey && Object.keys(payload).length > 0) {
+          dbAdapter.upsertOne(state, payload);
+        }
+      },
+    }),
+    upsert: create.asyncThunk(upsertAction, {
+      fulfilled: (state, action) => {
+        if (
+          action.payload &&
+          action.payload.dbKey &&
+          Object.keys(action.payload).length > 0
+        ) {
+          dbAdapter.upsertOne(state, action.payload);
+        }
+      },
+    }),
+    // 文件上传（avatar / Slate / Space 等统一走这里）
+    upload: create.asyncThunk(uploadFileAction, {
+      fulfilled: (state, action) => {
+        const payload = action.payload;
+        if (payload && payload.dbKey && Object.keys(payload).length > 0) {
+          dbAdapter.upsertOne(state, payload);
+        }
+      },
+    }),
+    // 读取文件内容（优先本地 IndexedDB，无状态副作用）
+    readFileContent: create.asyncThunk(readFileContentAction, {
+      // fulfilled 时不修改 db state；由调用方通过 unwrap() 拿返回值使用
+    }),
+    share: create.asyncThunk(shareResourceAction, {
+      fulfilled: (state, action) => {
+        if (action.payload && action.payload.key) {
+          // We might want to store the shared object in local DB state too?
+          // writeAction already does it. This is just for return value.
+          // Actually, writeAction is dispatched inside shareResourceAction.
+          // Does writeAction update state? Yes.
+          // So here we might not need to do anything extra to state, 
+          // just define the thunk.
+        }
+      }
+    }),
+    // SSR 预取：服务端直接注入实体到 db slice，供首屏 hydrate 使用
+    upsertSSREntity: create.reducer((state, action: PayloadAction<any>) => {
+      if (action.payload && action.payload.dbKey) {
+        dbAdapter.upsertOne(state, action.payload);
+      }
+    }),
+  }),
+});
+
+// Export actions
+export const {
+  remove,
+  read,
+  readAndWait,
+  write,
+  patch,
+  upsert,
+  upload,
+  readFileContent,
+  share,
+  upsertSSREntity,
+} = dbSlice.actions;
+
+// Export the reducer
+export default dbSlice.reducer;

package/database/email.ts

@@ -0,0 +1,42 @@
+import type { DataType } from "create/types";
+
+export type EmailOwnerType = "user" | "agent";
+export type EmailMailbox = "inbox" | "sent" | "archive" | "trash" | "drafts";
+export type EmailStatus =
+  | "received"
+  | "draft"
+  | "queued"
+  | "sent"
+  | "failed";
+
+export type EmailParticipant = {
+  email: string;
+  name?: string;
+};
+
+export type EmailRecord = {
+  dbKey: string;
+  type: DataType.EMAIL;
+  ownerType: EmailOwnerType;
+  ownerId: string;
+  tenantId: string;
+  spaceId?: string | null;
+  mailbox: EmailMailbox;
+  status: EmailStatus;
+  from: EmailParticipant;
+  to: EmailParticipant[];
+  cc?: EmailParticipant[];
+  bcc?: EmailParticipant[];
+  replyTo?: EmailParticipant[];
+  subject: string;
+  text?: string;
+  html?: string;
+  messageId?: string;
+  threadId?: string;
+  inReplyTo?: string;
+  references?: string[];
+  tags?: string[];
+  meta?: Record<string, unknown>;
+  createdAt: string | number;
+  updatedAt: string | number;
+};

package/database/fileRing.ts

@@ -0,0 +1,51 @@
+// 文件路径: database/fileRing.ts
+
+/**
+ * 简单的 FNV-1a 32bit 字符串哈希
+ * - 稳定、实现简单，适合作为 hash ring 的排序依据
+ */
+const fnv1a32 = (str: string): number => {
+    let hash = 0x811c9dc5;
+    for (let i = 0; i < str.length; i++) {
+        hash ^= str.charCodeAt(i);
+        hash = (hash * 0x01000193) >>> 0; // 32bit
+    }
+    return hash >>> 0;
+};
+
+/**
+ * 通用服务器选择函数：
+ *
+ * 给定一组服务器和一个“key”（可以是 tenantId / userId / orgId / fileId …），
+ * 选择 replicaCount 个服务器作为该 key 的“归属服务器”。
+ *
+ * 规则：
+ * - 对每个 server 计算 hash(server + "::" + key)
+ * - 按 hash 升序排序
+ * - 取前 replicaCount 个
+ *
+ * 说明：
+ * - 这不是严格意义上的一致性哈希环，但分布效果 + 实现复杂度比较均衡；
+ * - 之后要支持更多服务器，只需要在配置中把新 server 加入列表即可，
+ *   这里不需要改代码。
+ */
+export const chooseServersByKey = (
+    allServers: string[],
+    key: string,
+    replicaCount: number
+): string[] => {
+    if (!allServers.length || replicaCount <= 0) return [];
+
+    const uniqueServers = Array.from(new Set(allServers)).filter((s) => !!s);
+    if (uniqueServers.length === 0) return [];
+
+    const scored = uniqueServers.map((server) => ({
+        server,
+        score: fnv1a32(`${server}::${key}`),
+    }));
+
+    scored.sort((a, b) => a.score - b.score);
+
+    const limit = Math.min(replicaCount, scored.length);
+    return scored.slice(0, limit).map((item) => item.server);
+};

package/database/fileSharding.ts

@@ -0,0 +1,70 @@
+// 文件路径: database/fileSharding.ts
+
+import { chooseServersForFile } from "./fileRing";
+
+/**
+ * 当前“切片”策略参数：
+ *
+ * - REPLICA_COUNT = 3:
+ *   在 ring 中尽量挑 3 台不同的服务器存副本。
+ *
+ * - REQUIRED_SHARDS = 2:
+ *   语义上表示：只要有 2 份副本可用就能工作（当前是完整副本，所以任意一份都能用）。
+ *
+ * 将来如果要上真正的纠删码（2-of-3），可以保留这两个名字，
+ * 只是把 encode/decode 的实现替换掉即可。
+ */
+export const REPLICA_COUNT = 3;
+export const REQUIRED_SHARDS = 2;
+
+/**
+ * 基于 hash ring 选择本次写入要用的服务器列表。
+ *
+ * 要求：
+ * - 优先根据 fileId 在 allServers 上做 ring 选择 REPLICA_COUNT 个；
+ * - 无论 ring 如何选择，currentServer 必须包含在结果中（保证当前前端指向的服务器上总有一份）。
+ *
+ * 说明：
+ * - 现在每个 server 存的是完整文件，将来可以改成“每个 server 存一个 shard blob”，
+ *   但对 upload 调用方来说只是「发给这些 servers」这一点不变。
+ */
+export const planReplicaServersForFile = (
+    allServers: string[],
+    currentServer: string | null | undefined,
+    fileId: string
+): string[] => {
+    const uniqueServers = Array.from(new Set(allServers)).filter(Boolean);
+    if (!uniqueServers.length) return [];
+
+    const fromRing = chooseServersForFile(uniqueServers, fileId, REPLICA_COUNT);
+    const set = new Set(fromRing);
+
+    if (currentServer && uniqueServers.includes(currentServer)) {
+        set.add(currentServer);
+    }
+
+    return Array.from(set);
+};
+
+/**
+ * 预留：将来真正做 2-of-3 纠删码时在这里实现。
+ *
+ * encode:
+ *   接收完整 File/Blob，返回若干 shard（索引 + blob），
+ *   上层再根据 shardIndex + planReplicaServersForFile 决定各 shard 分布。
+ *
+ * decode:
+ *   接收至少 REQUIRED_SHARDS 份 shard，重建完整 Blob。
+ */
+
+// export async function encodeToShards(
+//   file: File | Blob
+// ): Promise<Array<{ shardIndex: number; blob: Blob }>> {
+//   // TODO: 将来在这里实现真正的 2-of-3 切片编码
+// }
+
+// export async function decodeFromShards(
+//   shards: Array<{ shardIndex: number; blob: Blob }>
+// ): Promise<Blob> {
+//   // TODO: 将来在这里实现解码逻辑
+// }