jax-hono 1.0.20 → 1.0.21

package/plugins/mongoose/model.ts

@@ -1,322 +1,322 @@
-import { basename, parse } from "node:path";
-import mongoose, {
-  Schema,
-  model as mongooseModel,
-  type InferSchemaType,
-  type Model,
-  type SchemaOptions,
-} from "mongoose";
-import config from "../../config";
-import { rankSetter, rankedFields } from "./schema";
-import dayjs from "dayjs";
-
-export { rankSetter };
-
-type PageOptions = {
-  page?: number | string;
-  pageSize?: number | string;
-  sort?: Record<string, unknown>;
-};
-
-export type ModelStatics = {
-  findPage(
-    filter?: Record<string, unknown>,
-    options?: PageOptions,
-  ): Promise<{
-    total: number;
-    list: unknown[];
-    maxPage: number;
-    page: number;
-    pageSize: number;
-  }>;
-  tree(options?: {
-    filter?: Record<string, unknown>;
-    sort?: Record<string, unknown>;
-  }): Promise<unknown[]>;
-  flatTree(options?: {
-    filter?: Record<string, unknown>;
-    sort?: Record<string, unknown>;
-  }): Promise<unknown[]>;
-};
-
-export function createLooseModel(filePathOrName: string, collection?: string) {
-  const name = getModelName(filePathOrName);
-  const schema = new Schema({}, { strict: false });
-
-  applyDefaultSchemaOptions(schema);
-  installCommonFields(schema);
-  installModelStatics(schema);
-
-  return (
-    mongoose.models[name] ??
-    mongooseModel(name, schema, collection ?? getCollectionName(name))
-  );
-}
-
-export function createSchemaModel<TSchema extends Schema>(
-  filePathOrName: string,
-  schema: TSchema,
-  collection?: string,
-): Model<InferSchemaType<TSchema>> & ModelStatics {
-  const name = getModelName(filePathOrName);
-
-  applyDefaultSchemaOptions(schema);
-  installCommonFields(schema);
-  installRankHooksIfNeeded(schema);
-  installModelStatics(schema);
-
-  return (mongoose.models[name] ??
-    mongooseModel<InferSchemaType<TSchema>>(
-      name,
-      schema as any,
-      collection ?? getCollectionName(name),
-    )) as Model<InferSchemaType<TSchema>> & ModelStatics;
-}
-
-export const createModel = createSchemaModel;
-
-const rankHooksInstalled = Symbol.for("jax-hono.mongoose.rankHooksInstalled");
-
-export function getModelName(filePathOrName: string) {
-  const baseName =
-    filePathOrName.includes("/") || filePathOrName.includes("\\")
-      ? parse(basename(filePathOrName)).name.replace(/\.model$/, "")
-      : filePathOrName;
-
-  return toPascalCase(baseName);
-}
-
-export function getCollectionName(modelName: string) {
-  const prefix =
-    (config as any).mongoose?.prefix ?? (config as any).mongo?.prefix ?? "";
-  const name = toSnakeCase(modelName);
-
-  return prefix ? `${prefix}_${name}` : name;
-}
-
-export function schemaOptions(): SchemaOptions {
-  return {
-    timestamps: true,
-    versionKey: false,
-    toJSON: {
-      virtuals: true,
-      getters: true,
-      transform(_doc: unknown, ret: any) {
-        // if (ret._id && !ret.id) {
-        //   ret.id = ret._id.toString();
-        // }
-
-        if (ret.createdAt) {
-          ret.createdAt = dayjs(ret.createdAt).format("YYYY-MM-DD HH:mm:ss");
-        }
-
-        if (ret.updatedAt) {
-          ret.updatedAt = dayjs(ret.updatedAt).format("YYYY-MM-DD HH:mm:ss");
-        }
-
-        delete ret.__v;
-        delete ret._id;
-
-        return ret;
-      },
-    },
-    toObject: {
-      virtuals: true,
-      getters: true,
-    },
-  };
-}
-
-export function applyDefaultSchemaOptions(schema: Schema) {
-  const defaults = schemaOptions();
-
-  if (schema.options.timestamps === undefined) {
-    schema.set("timestamps", defaults.timestamps);
-  }
-
-  if (
-    schema.options.versionKey === undefined ||
-    schema.options.versionKey === "__v"
-  ) {
-    schema.set("versionKey", defaults.versionKey);
-  }
-
-  const toJSON = {
-    ...defaults.toJSON,
-    ...(schema.options.toJSON ?? {}),
-  } as any;
-
-  schema.set("toJSON", toJSON);
-  schema.set("toObject", {
-    ...defaults.toObject,
-    ...(schema.options.toObject ?? {}),
-  } as any);
-}
-
-export function installRankHooks(schema: Schema) {
-  const fields: Record<string, unknown> = {};
-
-  if (!schema.path("rank")) {
-    fields.rank = rankedFields.rank;
-  }
-
-  if (!schema.path("isTop")) {
-    fields.isTop = rankedFields.isTop;
-  }
-
-  if (Object.keys(fields).length) {
-    schema.add(fields as any);
-  }
-
-  if ((schema as any)[rankHooksInstalled]) {
-    return;
-  }
-
-  (schema as any)[rankHooksInstalled] = true;
-
-  // 修改
-  schema.pre(["updateOne", "findOneAndUpdate", "updateMany"], function () {
-    console.log('updateOne');
-    normalizeRank(this.getUpdate());
-  });
-
-  // 新增
-  schema.pre("save", function () {
-    console.log('save');
-    normalizeRank(this);
-  });
-}
-
-function installRankHooksIfNeeded(schema: Schema) {
-  if (schema.path("rank") && schema.path("isTop")) {
-    installRankHooks(schema);
-  }
-}
-
-export function installAutoNo(schema: Schema, field: string, start = 0) {
-  schema.pre("save", async function () {
-    if (!this.isNew || this.get(field)) {
-      return;
-    }
-
-    const model = this.constructor as any;
-    const doc = await model.findOne({}).sort({ [field]: -1 });
-
-    this.set(field, (doc?.[field] ?? start) + 1);
-  });
-}
-
-export function installModelStatics(schema: Schema) {
-  schema.static(
-    "findPage",
-    async function findPage(filter = {}, options: PageOptions = {}) {
-      const page = Number(options.page || 1);
-      const pageSize = Number(options.pageSize || 20);
-      const sort = (options.sort ?? { _id: -1 }) as any;
-      const total = await this.countDocuments(filter);
-      let query = this.find(filter).sort(sort);
-
-      if (pageSize !== -1) {
-        query = query.skip((page - 1) * pageSize).limit(pageSize);
-      }
-
-      const list = await query;
-
-      return {
-        total,
-        list,
-        maxPage: pageSize === -1 ? 1 : Math.ceil(total / pageSize),
-        page,
-        pageSize,
-      };
-    },
-  );
-
-  schema.static("tree", async function tree(options: any = {}) {
-    const docs = await this.find(options.filter ?? {})
-      .sort(options.sort ?? {})
-      .lean({ virtuals: true });
-    const byId = new Map<string, any>();
-    const tree: any[] = [];
-
-    for (const doc of docs) {
-      const id = String(doc._id ?? doc.id);
-      byId.set(id, { ...doc, id, children: [] });
-    }
-
-    for (const item of byId.values()) {
-      const parentId = item.parentId ? String(item.parentId) : "";
-      const parent = parentId ? byId.get(parentId) : undefined;
-
-      if (parent) {
-        parent.children.push(item);
-      } else {
-        tree.push(item);
-      }
-    }
-
-    return tree;
-  });
-
-  schema.static(
-    "flatTree",
-    async function flatTree(this: any, options: any = {}) {
-      const tree = await this.tree(options);
-      const list: any[] = [];
-
-      walkTree(tree, 1, list);
-
-      return list;
-    },
-  );
-}
-
-export function installCommonFields(schema: Schema) {
-  const fields: Record<string, unknown> = {};
-
-  if (!schema.path("isDelete")) {
-    fields.isDelete = { type: Boolean, default: false };
-  }
-
-  if (!schema.path("isOpen")) {
-    fields.isOpen = { type: Boolean, default: false };
-  }
-
-  if (Object.keys(fields).length > 0) {
-    schema.add(fields as any);
-  }
-}
-
-export function toSnakeCase(value: string) {
-  return value
-    .replace(/([a-z0-9])([A-Z])/g, "$1_$2")
-    .replace(/[-\s]+/g, "_")
-    .toLowerCase();
-}
-
-function toPascalCase(value: string) {
-  const camel = value.replace(/[-_\s]+([a-zA-Z0-9])/g, (_match, char: string) =>
-    char.toUpperCase(),
-  );
-
-  return camel.charAt(0).toUpperCase() + camel.slice(1);
-}
-
-function walkTree(items: any[], depth: number, list: any[]) {
-  for (const item of items) {
-    const { children = [], ...data } = item;
-
-    list.push({ ...data, children, index: depth });
-    walkTree(children, depth + 1, list);
-  }
-}
-
-function normalizeRank(data: any) {
-  if (data.rank != undefined) {
-    if (data.rank) {
-      data.isTop = true
-    } else {
-      data.isTop = false
-    }
-  }
-}
+import { basename, parse } from "node:path";
+import mongoose, {
+  Schema,
+  model as mongooseModel,
+  type InferSchemaType,
+  type Model,
+  type SchemaOptions,
+} from "mongoose";
+import config from "../../config";
+import { rankSetter, rankedFields } from "./schema";
+import dayjs from "dayjs";
+
+export { rankSetter };
+
+type PageOptions = {
+  page?: number | string;
+  pageSize?: number | string;
+  sort?: Record<string, unknown>;
+};
+
+export type ModelStatics = {
+  findPage(
+    filter?: Record<string, unknown>,
+    options?: PageOptions,
+  ): Promise<{
+    total: number;
+    list: unknown[];
+    maxPage: number;
+    page: number;
+    pageSize: number;
+  }>;
+  tree(options?: {
+    filter?: Record<string, unknown>;
+    sort?: Record<string, unknown>;
+  }): Promise<unknown[]>;
+  flatTree(options?: {
+    filter?: Record<string, unknown>;
+    sort?: Record<string, unknown>;
+  }): Promise<unknown[]>;
+};
+
+export function createLooseModel(filePathOrName: string, collection?: string) {
+  const name = getModelName(filePathOrName);
+  const schema = new Schema({}, { strict: false });
+
+  applyDefaultSchemaOptions(schema);
+  installCommonFields(schema);
+  installModelStatics(schema);
+
+  return (
+    mongoose.models[name] ??
+    mongooseModel(name, schema, collection ?? getCollectionName(name))
+  );
+}
+
+export function createSchemaModel<TSchema extends Schema>(
+  filePathOrName: string,
+  schema: TSchema,
+  collection?: string,
+): Model<InferSchemaType<TSchema>> & ModelStatics {
+  const name = getModelName(filePathOrName);
+
+  applyDefaultSchemaOptions(schema);
+  installCommonFields(schema);
+  installRankHooksIfNeeded(schema);
+  installModelStatics(schema);
+
+  return (mongoose.models[name] ??
+    mongooseModel<InferSchemaType<TSchema>>(
+      name,
+      schema as any,
+      collection ?? getCollectionName(name),
+    )) as Model<InferSchemaType<TSchema>> & ModelStatics;
+}
+
+export const createModel = createSchemaModel;
+
+const rankHooksInstalled = Symbol.for("jax-hono.mongoose.rankHooksInstalled");
+
+export function getModelName(filePathOrName: string) {
+  const baseName =
+    filePathOrName.includes("/") || filePathOrName.includes("\\")
+      ? parse(basename(filePathOrName)).name.replace(/\.model$/, "")
+      : filePathOrName;
+
+  return toPascalCase(baseName);
+}
+
+export function getCollectionName(modelName: string) {
+  const prefix =
+    (config as any).mongoose?.prefix ?? (config as any).mongo?.prefix ?? "";
+  const name = toSnakeCase(modelName);
+
+  return prefix ? `${prefix}_${name}` : name;
+}
+
+export function schemaOptions(): SchemaOptions {
+  return {
+    timestamps: true,
+    versionKey: false,
+    toJSON: {
+      virtuals: true,
+      getters: true,
+      transform(_doc: unknown, ret: any) {
+        // if (ret._id && !ret.id) {
+        //   ret.id = ret._id.toString();
+        // }
+
+        if (ret.createdAt) {
+          ret.createdAt = dayjs(ret.createdAt).format("YYYY-MM-DD HH:mm:ss");
+        }
+
+        if (ret.updatedAt) {
+          ret.updatedAt = dayjs(ret.updatedAt).format("YYYY-MM-DD HH:mm:ss");
+        }
+
+        delete ret.__v;
+        delete ret._id;
+
+        return ret;
+      },
+    },
+    toObject: {
+      virtuals: true,
+      getters: true,
+    },
+  };
+}
+
+export function applyDefaultSchemaOptions(schema: Schema) {
+  const defaults = schemaOptions();
+
+  if (schema.options.timestamps === undefined) {
+    schema.set("timestamps", defaults.timestamps);
+  }
+
+  if (
+    schema.options.versionKey === undefined ||
+    schema.options.versionKey === "__v"
+  ) {
+    schema.set("versionKey", defaults.versionKey);
+  }
+
+  const toJSON = {
+    ...defaults.toJSON,
+    ...(schema.options.toJSON ?? {}),
+  } as any;
+
+  schema.set("toJSON", toJSON);
+  schema.set("toObject", {
+    ...defaults.toObject,
+    ...(schema.options.toObject ?? {}),
+  } as any);
+}
+
+export function installRankHooks(schema: Schema) {
+  const fields: Record<string, unknown> = {};
+
+  if (!schema.path("rank")) {
+    fields.rank = rankedFields.rank;
+  }
+
+  if (!schema.path("isTop")) {
+    fields.isTop = rankedFields.isTop;
+  }
+
+  if (Object.keys(fields).length) {
+    schema.add(fields as any);
+  }
+
+  if ((schema as any)[rankHooksInstalled]) {
+    return;
+  }
+
+  (schema as any)[rankHooksInstalled] = true;
+
+  // 修改
+  schema.pre(["updateOne", "findOneAndUpdate", "updateMany"], function () {
+    console.log('updateOne');
+    normalizeRank(this.getUpdate());
+  });
+
+  // 新增
+  schema.pre("save", function () {
+    console.log('save');
+    normalizeRank(this);
+  });
+}
+
+function installRankHooksIfNeeded(schema: Schema) {
+  if (schema.path("rank") && schema.path("isTop")) {
+    installRankHooks(schema);
+  }
+}
+
+export function installAutoNo(schema: Schema, field: string, start = 0) {
+  schema.pre("save", async function () {
+    if (!this.isNew || this.get(field)) {
+      return;
+    }
+
+    const model = this.constructor as any;
+    const doc = await model.findOne({}).sort({ [field]: -1 });
+
+    this.set(field, (doc?.[field] ?? start) + 1);
+  });
+}
+
+export function installModelStatics(schema: Schema) {
+  schema.static(
+    "findPage",
+    async function findPage(filter = {}, options: PageOptions = {}) {
+      const page = Number(options.page || 1);
+      const pageSize = Number(options.pageSize || 20);
+      const sort = (options.sort ?? { _id: -1 }) as any;
+      const total = await this.countDocuments(filter);
+      let query = this.find(filter).sort(sort);
+
+      if (pageSize !== -1) {
+        query = query.skip((page - 1) * pageSize).limit(pageSize);
+      }
+
+      const list = await query;
+
+      return {
+        total,
+        list,
+        maxPage: pageSize === -1 ? 1 : Math.ceil(total / pageSize),
+        page,
+        pageSize,
+      };
+    },
+  );
+
+  schema.static("tree", async function tree(options: any = {}) {
+    const docs = await this.find(options.filter ?? {})
+      .sort(options.sort ?? {})
+      .lean({ virtuals: true });
+    const byId = new Map<string, any>();
+    const tree: any[] = [];
+
+    for (const doc of docs) {
+      const id = String(doc._id ?? doc.id);
+      byId.set(id, { ...doc, id, children: [] });
+    }
+
+    for (const item of byId.values()) {
+      const parentId = item.parentId ? String(item.parentId) : "";
+      const parent = parentId ? byId.get(parentId) : undefined;
+
+      if (parent) {
+        parent.children.push(item);
+      } else {
+        tree.push(item);
+      }
+    }
+
+    return tree;
+  });
+
+  schema.static(
+    "flatTree",
+    async function flatTree(this: any, options: any = {}) {
+      const tree = await this.tree(options);
+      const list: any[] = [];
+
+      walkTree(tree, 1, list);
+
+      return list;
+    },
+  );
+}
+
+export function installCommonFields(schema: Schema) {
+  const fields: Record<string, unknown> = {};
+
+  if (!schema.path("isDelete")) {
+    fields.isDelete = { type: Boolean, default: false };
+  }
+
+  if (!schema.path("isOpen")) {
+    fields.isOpen = { type: Boolean, default: false };
+  }
+
+  if (Object.keys(fields).length > 0) {
+    schema.add(fields as any);
+  }
+}
+
+export function toSnakeCase(value: string) {
+  return value
+    .replace(/([a-z0-9])([A-Z])/g, "$1_$2")
+    .replace(/[-\s]+/g, "_")
+    .toLowerCase();
+}
+
+function toPascalCase(value: string) {
+  const camel = value.replace(/[-_\s]+([a-zA-Z0-9])/g, (_match, char: string) =>
+    char.toUpperCase(),
+  );
+
+  return camel.charAt(0).toUpperCase() + camel.slice(1);
+}
+
+function walkTree(items: any[], depth: number, list: any[]) {
+  for (const item of items) {
+    const { children = [], ...data } = item;
+
+    list.push({ ...data, children, index: depth });
+    walkTree(children, depth + 1, list);
+  }
+}
+
+function normalizeRank(data: any) {
+  if (data.rank != undefined) {
+    if (data.rank) {
+      data.isTop = true
+    } else {
+      data.isTop = false
+    }
+  }
+}