js-bao 0.3.0 → 0.3.1

package/dist/cloudflare.js

@@ -59,14 +59,138 @@ var init_DocumentQueryTranslator = __esm({
 
 // src/models/metaSync.ts
 import * as Y from "yjs";
+function inferFieldType(value) {
+  if (value instanceof Y.Map) return "stringset";
+  switch (typeof value) {
+    case "string":
+      return "string";
+    case "number":
+      return "number";
+    case "boolean":
+      return "boolean";
+    default:
+      return null;
+  }
+}
 function registerFunctionDefault(fn, name) {
   KNOWN_FUNCTION_DEFAULTS.set(fn, name);
 }
-var KNOWN_FUNCTION_DEFAULTS;
+function encodeDefault(value) {
+  if (value === void 0 || value === null) return void 0;
+  if (typeof value === "function") {
+    const name = KNOWN_FUNCTION_DEFAULTS.get(value);
+    return name ? `$${name}` : "$unknown_function";
+  }
+  return value;
+}
+function clearMetaSyncCache(yDoc) {
+  if (yDoc) {
+    _syncedCache.delete(yDoc);
+  }
+}
+function syncModelMeta(yDoc, modelName, schema) {
+  let synced = _syncedCache.get(yDoc);
+  if (synced?.has(modelName)) return;
+  if (!synced) {
+    synced = /* @__PURE__ */ new Set();
+    _syncedCache.set(yDoc, synced);
+  }
+  const meta = yDoc.getMap(`_meta_${modelName}`);
+  for (const [fieldName, fieldOpts] of schema.fields.entries()) {
+    syncFieldMeta(meta, fieldName, fieldOpts);
+  }
+  const compoundConstraints = schema.resolvedUniqueConstraints.filter(
+    (c) => c.fields.length > 1
+  );
+  if (compoundConstraints.length > 0) {
+    let constraints = meta.get("_constraints");
+    if (!constraints) {
+      constraints = new Y.Map();
+      meta.set("_constraints", constraints);
+    }
+    for (const constraint of compoundConstraints) {
+      syncConstraintMeta(constraints, constraint);
+    }
+  }
+  const relationships = schema.options.relationships;
+  if (relationships && Object.keys(relationships).length > 0) {
+    let rels = meta.get("_relationships");
+    if (!rels) {
+      rels = new Y.Map();
+      meta.set("_relationships", rels);
+    }
+    for (const [relName, relConfig] of Object.entries(relationships)) {
+      syncRelationshipMeta(rels, relName, relConfig);
+    }
+  }
+  synced.add(modelName);
+}
+function syncFieldMeta(metaMap, fieldName, fieldOpts) {
+  let fieldMeta = metaMap.get(fieldName);
+  if (!fieldMeta) {
+    fieldMeta = new Y.Map();
+    metaMap.set(fieldName, fieldMeta);
+  }
+  setIfChanged(fieldMeta, "type", fieldOpts.type);
+  if (fieldOpts.indexed) setIfChanged(fieldMeta, "indexed", true);
+  if (fieldOpts.unique) setIfChanged(fieldMeta, "unique", true);
+  if (fieldOpts.required) setIfChanged(fieldMeta, "required", true);
+  if (fieldOpts.autoAssign) setIfChanged(fieldMeta, "autoAssign", true);
+  if (fieldOpts.maxLength !== void 0) setIfChanged(fieldMeta, "maxLength", fieldOpts.maxLength);
+  if (fieldOpts.maxCount !== void 0) setIfChanged(fieldMeta, "maxCount", fieldOpts.maxCount);
+  const encoded = encodeDefault(fieldOpts.default);
+  if (encoded !== void 0) setIfChanged(fieldMeta, "default", encoded);
+}
+function syncConstraintMeta(constraintsMap, constraint) {
+  let cMeta = constraintsMap.get(constraint.name);
+  if (!cMeta) {
+    cMeta = new Y.Map();
+    constraintsMap.set(constraint.name, cMeta);
+  }
+  setIfChanged(cMeta, "type", "unique");
+  const fieldsJson = JSON.stringify(constraint.fields);
+  setIfChanged(cMeta, "fields", fieldsJson);
+}
+function syncRelationshipMeta(relsMap, relName, relConfig) {
+  let relMeta = relsMap.get(relName);
+  if (!relMeta) {
+    relMeta = new Y.Map();
+    relsMap.set(relName, relMeta);
+  }
+  for (const [key, value] of Object.entries(relConfig)) {
+    if (value !== void 0) {
+      setIfChanged(relMeta, key, value);
+    }
+  }
+}
+function syncInferredMeta(yDoc, modelName, recordData) {
+  const meta = yDoc.getMap(`_meta_${modelName}`);
+  for (const [fieldName, value] of Object.entries(recordData)) {
+    if (fieldName.startsWith("_")) continue;
+    let fieldMeta = meta.get(fieldName);
+    if (!fieldMeta) {
+      fieldMeta = new Y.Map();
+      meta.set(fieldName, fieldMeta);
+    }
+    if (!fieldMeta.has("type")) {
+      const inferredType = inferFieldType(value);
+      if (inferredType) {
+        fieldMeta.set("type", inferredType);
+      }
+    }
+  }
+}
+function setIfChanged(map, key, value) {
+  if (map.get(key) !== value) {
+    map.set(key, value);
+  }
+}
+var KNOWN_FUNCTION_DEFAULTS, _syncedCache;
 var init_metaSync = __esm({
   "src/models/metaSync.ts"() {
     "use strict";
     KNOWN_FUNCTION_DEFAULTS = /* @__PURE__ */ new WeakMap();
+    _syncedCache = /* @__PURE__ */ new WeakMap();
   }
 });
 
@@ -158,6 +282,23 @@ var init_BaseModel = __esm({
   }
 });
 
+// src/models/relationshipManager.ts
+var init_relationshipManager = __esm({
+  "src/models/relationshipManager.ts"() {
+    "use strict";
+    init_BaseModel();
+  }
+});
+
+// src/models/ModelRegistry.ts
+var init_ModelRegistry = __esm({
+  "src/models/ModelRegistry.ts"() {
+    "use strict";
+    init_BaseModel();
+    init_relationshipManager();
+  }
+});
+
 // src/initialize-do.ts
 init_BaseModel();
 
@@ -302,7 +443,8 @@ var DOClientEngine = class extends DatabaseEngine {
       data,
       stringSets,
       ifNotExists: options?.ifNotExists,
-      condition: options?.condition
+      condition: options?.condition,
+      upsertOn: options?.upsertOn
     };
     const response = await this.doFetch("/save", request);
     return response.id;
@@ -669,22 +811,24 @@ async function initJsBaoDO(options) {
       if (!docId) {
         throw new Error("Not connected to a document. Call connectDocument() first.");
       }
-      if (!data.id) {
-        throw new Error("Record must have an 'id' field");
-      }
       let stringSets;
       let ifNotExists;
       let condition;
+      let upsertOn;
       if (options2 && typeof options2 === "object") {
-        if (options2.ifNotExists !== void 0 || options2.stringSets !== void 0 || options2.condition !== void 0) {
+        if (options2.ifNotExists !== void 0 || options2.stringSets !== void 0 || options2.condition !== void 0 || options2.upsertOn !== void 0) {
           stringSets = options2.stringSets;
           ifNotExists = options2.ifNotExists;
           condition = options2.condition;
+          upsertOn = options2.upsertOn;
         } else {
           stringSets = options2;
         }
       }
-      return engine.saveModel(resolveModelName(model), data.id, data, stringSets, { ifNotExists, condition });
+      if (!data.id && !upsertOn) {
+        throw new Error("Record must have an 'id' field (or use upsertOn)");
+      }
+      return engine.saveModel(resolveModelName(model), data.id, data, stringSets, { ifNotExists, condition, upsertOn });
     },
     patch: async (model, id, data, options2) => {
       const docId = engine.getCurrentDocument();
@@ -843,22 +987,24 @@ function createModelAccessor(engine, modelName) {
       return result.data.length > 0 ? result.data[0] : null;
     },
     save: (data, options) => {
-      if (!data.id) {
-        throw new Error("Record must have an 'id' field");
-      }
       let stringSets;
       let ifNotExists;
       let condition;
+      let upsertOn;
       if (options && typeof options === "object") {
-        if (options.ifNotExists !== void 0 || options.stringSets !== void 0 || options.condition !== void 0) {
+        if (options.ifNotExists !== void 0 || options.stringSets !== void 0 || options.condition !== void 0 || options.upsertOn !== void 0) {
           stringSets = options.stringSets;
           ifNotExists = options.ifNotExists;
           condition = options.condition;
+          upsertOn = options.upsertOn;
         } else {
           stringSets = options;
         }
       }
-      return engine.saveModel(modelName, data.id, data, stringSets, { ifNotExists, condition });
+      if (!data.id && !upsertOn) {
+        throw new Error("Record must have an 'id' field (or use upsertOn)");
+      }
+      return engine.saveModel(modelName, data.id, data, stringSets, { ifNotExists, condition, upsertOn });
     },
     patch: (id, data, options) => {
       let stringSets;
@@ -945,22 +1091,24 @@ function connectDoDb(options) {
       return result.data.length > 0 ? result.data[0] : null;
     },
     save: (model, data, options2) => {
-      if (!data.id) {
-        throw new Error("Record must have an 'id' field");
-      }
       let stringSets;
       let ifNotExists;
       let condition;
+      let upsertOn;
       if (options2 && typeof options2 === "object") {
-        if (options2.ifNotExists !== void 0 || options2.stringSets !== void 0 || options2.condition !== void 0) {
+        if (options2.ifNotExists !== void 0 || options2.stringSets !== void 0 || options2.condition !== void 0 || options2.upsertOn !== void 0) {
           stringSets = options2.stringSets;
           ifNotExists = options2.ifNotExists;
           condition = options2.condition;
+          upsertOn = options2.upsertOn;
         } else {
           stringSets = options2;
         }
       }
-      return engine.saveModel(resolveModelName(model), data.id, data, stringSets, { ifNotExists, condition });
+      if (!data.id && !upsertOn) {
+        throw new Error("Record must have an 'id' field (or use upsertOn)");
+      }
+      return engine.saveModel(resolveModelName(model), data.id, data, stringSets, { ifNotExists, condition, upsertOn });
     },
     patch: (model, id2, data, options2) => {
       let stringSets;
@@ -1026,10 +1174,417 @@ function connectDoDb(options) {
   Logger.info(`[connectDoDb] Connected to document: ${id} at ${endpoint}`);
   return db;
 }
+
+// src/utils/yDocSchema.ts
+import * as Y3 from "yjs";
+function discoverSchema(yDoc) {
+  const models = {};
+  const metaNames = /* @__PURE__ */ new Set();
+  for (const key of yDoc.share.keys()) {
+    if (!key.startsWith("_meta_")) continue;
+    const map = materializeMap(yDoc, key);
+    if (!map) continue;
+    const modelName = key.slice("_meta_".length);
+    metaNames.add(modelName);
+    models[modelName] = readModelMeta(map);
+  }
+  for (const key of yDoc.share.keys()) {
+    if (key.startsWith("_")) continue;
+    if (metaNames.has(key)) continue;
+    const map = materializeMap(yDoc, key);
+    if (!map || map.size === 0) continue;
+    const inferred = inferModelFromData(map);
+    if (inferred) models[key] = inferred;
+  }
+  return { models };
+}
+function discoverModelNames(yDoc) {
+  const names = [];
+  for (const key of yDoc.share.keys()) {
+    if (key.startsWith("_")) continue;
+    const map = materializeMap(yDoc, key);
+    if (map) names.push(key);
+  }
+  return names.sort();
+}
+function materializeMap(yDoc, key) {
+  try {
+    const map = yDoc.getMap(key);
+    return map instanceof Y3.Map ? map : null;
+  } catch {
+    return null;
+  }
+}
+function readModelMeta(metaMap) {
+  const fields = {};
+  let constraints;
+  let relationships;
+  for (const [key, value] of metaMap.entries()) {
+    if (key === "_constraints" && value instanceof Y3.Map) {
+      constraints = readConstraints(value);
+    } else if (key === "_relationships" && value instanceof Y3.Map) {
+      relationships = readRelationships(value);
+    } else if (value instanceof Y3.Map) {
+      fields[key] = readFieldMeta(value);
+    }
+  }
+  const model = { fields };
+  if (constraints && Object.keys(constraints).length > 0) {
+    model.constraints = constraints;
+  }
+  if (relationships && Object.keys(relationships).length > 0) {
+    model.relationships = relationships;
+  }
+  return model;
+}
+function readFieldMeta(fieldMap) {
+  const field = { type: fieldMap.get("type") ?? "unknown" };
+  if (fieldMap.get("indexed") === true) field.indexed = true;
+  if (fieldMap.get("unique") === true) field.unique = true;
+  if (fieldMap.get("required") === true) field.required = true;
+  if (fieldMap.get("autoAssign") === true) field.autoAssign = true;
+  const def = fieldMap.get("default");
+  if (def !== void 0) field.default = def;
+  const maxLength = fieldMap.get("maxLength");
+  if (maxLength !== void 0) field.maxLength = maxLength;
+  const maxCount = fieldMap.get("maxCount");
+  if (maxCount !== void 0) field.maxCount = maxCount;
+  return field;
+}
+function inferModelFromData(dataMap) {
+  const fields = {};
+  let sampled = 0;
+  for (const [_recordId, recordValue] of dataMap.entries()) {
+    if (!(recordValue instanceof Y3.Map)) continue;
+    if (++sampled > 5) break;
+    for (const [fieldName, value] of recordValue.entries()) {
+      if (fieldName.startsWith("_")) continue;
+      if (fields[fieldName]) continue;
+      const type = inferTypeFromValue(value);
+      if (type) fields[fieldName] = { type };
+    }
+  }
+  if (Object.keys(fields).length === 0) return null;
+  return { fields };
+}
+function inferTypeFromValue(value) {
+  if (value instanceof Y3.Map) return "stringset";
+  switch (typeof value) {
+    case "string":
+      return "string";
+    case "number":
+      return "number";
+    case "boolean":
+      return "boolean";
+    default:
+      return null;
+  }
+}
+function readConstraints(constraintsMap) {
+  const out = {};
+  for (const [name, value] of constraintsMap.entries()) {
+    if (!(value instanceof Y3.Map)) continue;
+    let fields = [];
+    const rawFields = value.get("fields");
+    if (typeof rawFields === "string") {
+      try {
+        fields = JSON.parse(rawFields);
+      } catch {
+      }
+    }
+    out[name] = {
+      type: value.get("type") ?? "unknown",
+      fields
+    };
+  }
+  return out;
+}
+function readRelationships(relsMap) {
+  const out = {};
+  for (const [name, value] of relsMap.entries()) {
+    if (!(value instanceof Y3.Map)) continue;
+    const rel = {};
+    for (const [k, v] of value.entries()) {
+      rel[k] = v;
+    }
+    out[name] = rel;
+  }
+  return out;
+}
+var CAMEL_TO_SNAKE = {
+  autoAssign: "auto_assign",
+  maxLength: "max_length",
+  maxCount: "max_count",
+  relatedIdField: "related_id_field",
+  joinModel: "join_model",
+  joinModelLocalField: "join_model_local_field",
+  joinModelRelatedField: "join_model_related_field",
+  joinModelOrderByField: "join_model_order_by_field",
+  joinModelOrderDirection: "join_model_order_direction",
+  orderByField: "order_by_field",
+  orderDirection: "order_direction"
+};
+function toSnake(key) {
+  return CAMEL_TO_SNAKE[key] ?? key;
+}
+function tomlValue(v) {
+  if (typeof v === "string") {
+    return `"${v.replace(/\\/g, "\\\\").replace(/"/g, '\\"').replace(/\n/g, "\\n").replace(/\r/g, "\\r").replace(/\t/g, "\\t")}"`;
+  }
+  return String(v);
+}
+function schemaToToml(schema) {
+  const lines = [];
+  for (const [modelName, model] of Object.entries(schema.models)) {
+    if (lines.length > 0) lines.push("", "");
+    lines.push(`[models.${modelName}]`);
+    for (const [fieldName, field] of Object.entries(model.fields)) {
+      lines.push("");
+      lines.push(`[models.${modelName}.fields.${fieldName}]`);
+      lines.push(`type = ${tomlValue(field.type)}`);
+      if (field.autoAssign) lines.push("auto_assign = true");
+      if (field.indexed) lines.push("indexed = true");
+      if (field.unique) lines.push("unique = true");
+      if (field.required) lines.push("required = true");
+      if (field.maxLength !== void 0) lines.push(`max_length = ${field.maxLength}`);
+      if (field.maxCount !== void 0) lines.push(`max_count = ${field.maxCount}`);
+      if (field.default !== void 0) lines.push(`default = ${tomlValue(field.default)}`);
+    }
+    if (model.relationships) {
+      for (const [relName, rel] of Object.entries(model.relationships)) {
+        lines.push("");
+        lines.push(`[models.${modelName}.relationships.${relName}]`);
+        for (const [k, v] of Object.entries(rel)) {
+          if (v === void 0) continue;
+          lines.push(`${toSnake(k)} = ${tomlValue(v)}`);
+        }
+      }
+    }
+    if (model.constraints) {
+      for (const [cName, c] of Object.entries(model.constraints)) {
+        lines.push("");
+        lines.push(`[[models.${modelName}.unique_constraints]]`);
+        lines.push(`name = ${tomlValue(cName)}`);
+        lines.push(`fields = [${c.fields.map((f) => tomlValue(f)).join(", ")}]`);
+      }
+    }
+  }
+  lines.push("");
+  return lines.join("\n");
+}
+
+// src/models/tomlLoader.ts
+import { parse as parseToml } from "smol-toml";
+
+// src/models/schema.ts
+init_BaseModel();
+init_ModelRegistry();
+function defineModelSchema(input) {
+  const { name, fields } = input;
+  const options = {
+    name,
+    uniqueConstraints: input.options?.uniqueConstraints ? [...input.options.uniqueConstraints] : void 0,
+    relationships: input.options?.relationships
+  };
+  const schema = {
+    name,
+    fields,
+    options,
+    runtimeShape: void 0,
+    buildRuntimeShape(modelClass) {
+      if (schema.runtimeShape && schema.runtimeShape.class === modelClass) {
+        return schema.runtimeShape;
+      }
+      const fieldsMap = /* @__PURE__ */ new Map();
+      for (const [fieldName, fieldOptions] of Object.entries(fields)) {
+        fieldsMap.set(fieldName, { ...fieldOptions });
+      }
+      const resolvedUniqueConstraints = resolveUniqueConstraints(
+        name,
+        fieldsMap,
+        options.uniqueConstraints
+      );
+      schema.runtimeShape = {
+        class: modelClass,
+        options,
+        fields: fieldsMap,
+        resolvedUniqueConstraints
+      };
+      return schema.runtimeShape;
+    }
+  };
+  return schema;
+}
+function resolveUniqueConstraints(modelName, fields, customConstraints) {
+  const resolved = [];
+  for (const [fieldName, options] of fields.entries()) {
+    if (options.unique) {
+      resolved.push({
+        name: `${modelName}_${fieldName}_unique`,
+        fields: [fieldName]
+      });
+    }
+  }
+  if (customConstraints) {
+    for (const constraint of customConstraints) {
+      const missingField = constraint.fields.find(
+        (field) => !fields.has(field)
+      );
+      if (missingField) {
+        console.warn(
+          `[defineModelSchema] Unique constraint "${constraint.name}" for model "${modelName}" references unknown field "${missingField}". Skipping.`
+        );
+        continue;
+      }
+      if (resolved.some((item) => item.name === constraint.name)) {
+        console.warn(
+          `[defineModelSchema] Duplicate unique constraint name "${constraint.name}" in model "${modelName}".`
+        );
+      }
+      resolved.push({
+        name: constraint.name,
+        fields: [...constraint.fields]
+      });
+    }
+  }
+  return resolved;
+}
+
+// src/models/tomlLoader.ts
+var VALID_FIELD_TYPES = /* @__PURE__ */ new Set([
+  "string",
+  "number",
+  "boolean",
+  "date",
+  "id",
+  "stringset"
+]);
+function parseFieldOptions(raw) {
+  if (!raw.type || !VALID_FIELD_TYPES.has(raw.type)) {
+    throw new Error(
+      `Invalid field type "${raw.type}". Must be one of: ${[...VALID_FIELD_TYPES].join(", ")}`
+    );
+  }
+  const opts = { type: raw.type };
+  if (raw.indexed === true) opts.indexed = true;
+  if (raw.unique === true) opts.unique = true;
+  if (raw.required === true) opts.required = true;
+  if (raw.auto_assign === true) opts.autoAssign = true;
+  if (raw.max_length !== void 0) opts.maxLength = raw.max_length;
+  if (raw.max_count !== void 0) opts.maxCount = raw.max_count;
+  if (raw.default !== void 0) opts.default = raw.default;
+  return opts;
+}
+function requireField(raw, field, context) {
+  if (!raw[field]) {
+    throw new Error(`Relationship ${context}: missing required field "${field}"`);
+  }
+}
+function parseRelationship(raw) {
+  const type = raw.type;
+  if (type === "refersTo") {
+    requireField(raw, "model", "refersTo");
+    requireField(raw, "related_id_field", "refersTo");
+    return {
+      type: "refersTo",
+      model: raw.model,
+      relatedIdField: raw.related_id_field
+    };
+  }
+  if (type === "hasMany") {
+    requireField(raw, "model", "hasMany");
+    requireField(raw, "related_id_field", "hasMany");
+    const rel = {
+      type: "hasMany",
+      model: raw.model,
+      relatedIdField: raw.related_id_field
+    };
+    if (raw.order_by_field) rel.orderByField = raw.order_by_field;
+    if (raw.order_direction) rel.orderDirection = raw.order_direction;
+    return rel;
+  }
+  if (type === "hasManyThrough") {
+    requireField(raw, "model", "hasManyThrough");
+    requireField(raw, "join_model", "hasManyThrough");
+    requireField(raw, "join_model_local_field", "hasManyThrough");
+    requireField(raw, "join_model_related_field", "hasManyThrough");
+    const rel = {
+      type: "hasManyThrough",
+      model: raw.model,
+      joinModel: raw.join_model,
+      joinModelLocalField: raw.join_model_local_field,
+      joinModelRelatedField: raw.join_model_related_field
+    };
+    if (raw.join_model_order_by_field)
+      rel.joinModelOrderByField = raw.join_model_order_by_field;
+    if (raw.join_model_order_direction)
+      rel.joinModelOrderDirection = raw.join_model_order_direction;
+    return rel;
+  }
+  throw new Error(`Unknown relationship type: ${type}`);
+}
+function loadSchemaFromTomlString(tomlString) {
+  const parsed = parseToml(tomlString);
+  const models = parsed.models;
+  if (!models || typeof models !== "object") {
+    throw new Error("TOML schema must have a [models] section");
+  }
+  const schemas = [];
+  for (const [modelName, modelDef] of Object.entries(models)) {
+    const fields = {};
+    if (modelDef.fields) {
+      for (const [fieldName, fieldDef] of Object.entries(modelDef.fields)) {
+        fields[fieldName] = parseFieldOptions(fieldDef);
+      }
+    }
+    let relationships;
+    if (modelDef.relationships) {
+      relationships = {};
+      for (const [relName, relDef] of Object.entries(
+        modelDef.relationships
+      )) {
+        relationships[relName] = parseRelationship(relDef);
+      }
+    }
+    let uniqueConstraints;
+    if (modelDef.unique_constraints) {
+      uniqueConstraints = [];
+      for (const raw of modelDef.unique_constraints) {
+        uniqueConstraints.push({
+          name: raw.name,
+          fields: [...raw.fields]
+        });
+      }
+    }
+    schemas.push(
+      defineModelSchema({
+        name: modelName,
+        fields,
+        options: {
+          uniqueConstraints,
+          relationships
+        }
+      })
+    );
+  }
+  return schemas;
+}
+
+// src/cloudflare.ts
+init_metaSync();
 export {
   DOClientEngine,
+  clearMetaSyncCache,
   connectDoDb,
+  discoverModelNames,
+  discoverSchema,
   getActiveDOEngine,
+  inferFieldType,
   initJsBaoDO,
-  resetJsBaoDO
+  loadSchemaFromTomlString,
+  registerFunctionDefault,
+  resetJsBaoDO,
+  schemaToToml,
+  syncInferredMeta,
+  syncModelMeta
 };