UNPKG

@mathrunet/masamune

Version:

Manages packages for the server portion (NodeJS) of the Masamune framework.

125 lines 4.71 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.FirestoreModelVectorValueConverter = exports.ModelVectorValueConverter = void 0; const model_field_value_converter_1 = require("../model_field_value_converter"); const utils_1 = require("../../utils"); const model_field_value_1 = require("../model_field_value"); const firebase_admin_1 = require("firebase-admin"); const firestore_1 = require("@google-cloud/firestore"); /** * ModelVectorValue ModelFieldValueConverter. * * ModelVectorValue用のModelFieldValueConverter。 */ class ModelVectorValueConverter extends model_field_value_converter_1.ModelFieldValueConverter { /** * ModelVectorValue ModelFieldValueConverter. * * ModelVectorValue用のModelFieldValueConverter。 */ constructor() { super(); } type = "ModelVectorValue"; convertFrom(key, value, original) { if (value !== null && typeof value === "object" && "@type" in value && value["@type"] === this.type) { const vector = value["@vector"] ?? []; return { [key]: new model_field_value_1.ModelVectorValue(vector, "server"), }; } return null; } convertTo(key, value, original) { if (value instanceof model_field_value_1.ModelVectorValue) { return { [key]: { "@type": this.type, "@vector": value["@vector"], "@source": value["@source"], }, }; } return null; } } exports.ModelVectorValueConverter = ModelVectorValueConverter; /** * FirestoreConverter for [ModelVectorValue]. * * [ModelVectorValue]用のFirestoreConverter。 */ class FirestoreModelVectorValueConverter extends model_field_value_converter_1.FirestoreModelFieldValueConverter { /** * FirestoreConverter for [ModelVectorValue]. * * [ModelVectorValue]用のFirestoreConverter。 */ constructor() { super(); } type = "ModelVectorValue"; convertFrom(key, value, original, firestoreInstance) { if (value instanceof firestore_1.VectorValue) { const targetKey = `#${key}`; const targetMap = original[targetKey] ?? {}; const type = targetMap["@type"] ?? ""; if (type == this.type) { const vectorValue = value.toArray(); const vector = []; for (const item of vectorValue) { vector.push(Number(item)); } return { [key]: { "@type": this.type, "@vector": vector, }, [targetKey]: null, }; } } return null; } convertTo(key, value, original, firestoreInstance) { if (value != null && typeof value === "object" && "@type" in value) { const type = value["@type"] ?? ""; if (type === this.type) { const fromUser = (value["@source"] ?? "") === "user"; const val = value["@vector"] ?? []; const targetKey = `#${key}`; const result = { [targetKey]: { "@type": this.type, "@target": key, }, }; if (fromUser) { result[key] = firebase_admin_1.firestore.FieldValue.vector(val); } return result; } } else if (Array.isArray(value)) { const list = value.filter((e) => e != null && typeof e === "object" && "@type" in e); if (list.length > 0 && list.every((e) => e["@type"] === this.type)) { throw new Error("ModelVectorValue cannot be included in a listing or map. It must be placed in the top field."); } } else if ((0, utils_1.isDynamicMap)(value)) { const map = {}; for (const k in value) { const v = value[k]; if (v != null && typeof v === "object" && "@type" in v) { map[k] = v; } } if (Object.keys(map).length > 0 && Object.values(map).every((e) => e["@type"] === this.type)) { throw new Error("ModelVectorValue cannot be included in a listing or map. It must be placed in the top field."); } } return null; } } exports.FirestoreModelVectorValueConverter = FirestoreModelVectorValueConverter; //# sourceMappingURL=model_vector_value_converter.js.map