@nocobase/database 0.5.0-alpha.38 → 0.7.0-alpha.10

package/esm/update-associations.js

@@ -0,0 +1,362 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { Model } from './model';
+import { UpdateGuard } from './update-guard';
+function isUndefinedOrNull(value) {
+    return typeof value === 'undefined' || value === null;
+}
+function isStringOrNumber(value) {
+    return typeof value === 'string' || typeof value === 'number';
+}
+function getKeysByPrefix(keys, prefix) {
+    return keys.filter((key) => key.startsWith(`${prefix}.`)).map((key) => key.substring(prefix.length + 1));
+}
+export function modelAssociations(instance) {
+    return instance.constructor.associations;
+}
+export function belongsToManyAssociations(instance) {
+    const associations = modelAssociations(instance);
+    return Object.entries(associations)
+        .filter((entry) => {
+        const [key, association] = entry;
+        return association.associationType == 'BelongsToMany';
+    })
+        .map((association) => {
+        return association[1];
+    });
+}
+export function modelAssociationByKey(instance, key) {
+    return modelAssociations(instance)[key];
+}
+export function updateModelByValues(instance, values, options) {
+    return __awaiter(this, void 0, void 0, function* () {
+        if (!(options === null || options === void 0 ? void 0 : options.sanitized)) {
+            const guard = new UpdateGuard();
+            //@ts-ignore
+            guard.setModel(instance.constructor);
+            guard.setBlackList(options.blacklist);
+            guard.setWhiteList(options.whitelist);
+            guard.setAssociationKeysToBeUpdate(options.updateAssociationValues);
+            values = guard.sanitize(values);
+        }
+        yield instance.update(values, options);
+        yield updateAssociations(instance, values, options);
+    });
+}
+export function updateThroughTableValue(instance, throughName, throughValues, source, transaction = null) {
+    return __awaiter(this, void 0, void 0, function* () {
+        // update through table values
+        for (const belongsToMany of belongsToManyAssociations(instance)) {
+            // @ts-ignore
+            const throughModel = belongsToMany.through.model;
+            const throughModelName = throughModel.name;
+            if (throughModelName === throughModelName) {
+                const where = {
+                    [belongsToMany.foreignKey]: instance.get(belongsToMany.sourceKey),
+                    [belongsToMany.otherKey]: source.get(belongsToMany.targetKey),
+                };
+                return yield throughModel.update(throughValues, {
+                    where,
+                    transaction,
+                });
+            }
+        }
+    });
+}
+/**
+ * update association of instance by values
+ * @param instance
+ * @param values
+ * @param options
+ */
+export function updateAssociations(instance, values, options = {}) {
+    return __awaiter(this, void 0, void 0, function* () {
+        // if no values set, return
+        if (!values) {
+            return;
+        }
+        let newTransaction = false;
+        let transaction = options.transaction;
+        if (!transaction) {
+            newTransaction = true;
+            transaction = yield instance.sequelize.transaction();
+        }
+        const keys = Object.keys(values);
+        for (const key of Object.keys(modelAssociations(instance))) {
+            if (keys.includes(key)) {
+                yield updateAssociation(instance, key, values[key], Object.assign(Object.assign({}, options), { transaction }));
+            }
+        }
+        // update through table values
+        for (const belongsToMany of belongsToManyAssociations(instance)) {
+            // @ts-ignore
+            const throughModel = belongsToMany.through.model;
+            const throughModelName = throughModel.name;
+            if (values[throughModelName] && options.sourceModel) {
+                const where = {
+                    [belongsToMany.foreignKey]: instance.get(belongsToMany.sourceKey),
+                    [belongsToMany.otherKey]: options.sourceModel.get(belongsToMany.targetKey),
+                };
+                yield throughModel.update(values[throughModel.name], {
+                    where,
+                    context: options.context,
+                    transaction,
+                });
+            }
+        }
+        if (newTransaction) {
+            yield transaction.commit();
+        }
+    });
+}
+function isReverseAssociationPair(a, b) {
+    const typeSet = new Set();
+    typeSet.add(a.associationType);
+    typeSet.add(b.associationType);
+    if (typeSet.size == 1 && typeSet.has('BelongsToMany')) {
+        return (a.through.tableName === b.through.tableName &&
+            a.target.name === b.source.name &&
+            b.target.name === a.source.name &&
+            a.foreignKey === b.otherKey &&
+            a.sourceKey === b.targetKey &&
+            a.otherKey === b.foreignKey &&
+            a.targetKey === b.sourceKey);
+    }
+    if ((typeSet.has('HasOne') && typeSet.has('BelongsTo')) || (typeSet.has('HasMany') && typeSet.has('BelongsTo'))) {
+        const sourceAssoc = a.associationType == 'BelongsTo' ? b : a;
+        const targetAssoc = sourceAssoc == a ? b : a;
+        return (sourceAssoc.source.name === targetAssoc.target.name &&
+            sourceAssoc.foreignKey === targetAssoc.foreignKey &&
+            sourceAssoc.sourceKey === targetAssoc.targetKey);
+    }
+    return false;
+}
+/**
+ * update model association by key
+ * @param instance
+ * @param key
+ * @param value
+ * @param options
+ */
+export function updateAssociation(instance, key, value, options = {}) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const association = modelAssociationByKey(instance, key);
+        if (!association) {
+            return false;
+        }
+        if (options.associationContext && isReverseAssociationPair(association, options.associationContext)) {
+            return false;
+        }
+        switch (association.associationType) {
+            case 'HasOne':
+            case 'BelongsTo':
+                return updateSingleAssociation(instance, key, value, options);
+            case 'HasMany':
+            case 'BelongsToMany':
+                return updateMultipleAssociation(instance, key, value, options);
+        }
+    });
+}
+/**
+ * update belongsTo and HasOne
+ * @param model
+ * @param key
+ * @param value
+ * @param options
+ */
+export function updateSingleAssociation(model, key, value, options = {}) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const association = modelAssociationByKey(model, key);
+        if (!association) {
+            return false;
+        }
+        if (!['undefined', 'string', 'number', 'object'].includes(typeof value)) {
+            return false;
+        }
+        const { context, updateAssociationValues = [], transaction = yield model.sequelize.transaction() } = options;
+        const keys = getKeysByPrefix(updateAssociationValues, key);
+        try {
+            // set method of association
+            const setAccessor = association.accessors.set;
+            const removeAssociation = () => __awaiter(this, void 0, void 0, function* () {
+                yield model[setAccessor](null, { transaction });
+                model.setDataValue(key, null);
+                if (!options.transaction) {
+                    yield transaction.commit();
+                }
+                return true;
+            });
+            if (isUndefinedOrNull(value)) {
+                return yield removeAssociation();
+            }
+            if (isStringOrNumber(value)) {
+                yield model[setAccessor](value, { context, transaction });
+                if (!options.transaction) {
+                    yield transaction.commit();
+                }
+                return true;
+            }
+            if (value instanceof Model) {
+                yield model[setAccessor](value, { context, transaction });
+                model.setDataValue(key, value);
+                if (!options.transaction) {
+                    yield transaction.commit();
+                }
+                return true;
+            }
+            const createAccessor = association.accessors.create;
+            let dataKey;
+            let M;
+            if (association.associationType === 'BelongsTo') {
+                M = association.target;
+                // @ts-ignore
+                dataKey = association.targetKey;
+            }
+            else {
+                M = association.source;
+                dataKey = M.primaryKeyAttribute;
+            }
+            if (isStringOrNumber(value[dataKey])) {
+                let instance = yield M.findOne({
+                    where: {
+                        [dataKey]: value[dataKey],
+                    },
+                    transaction,
+                });
+                if (instance) {
+                    yield model[setAccessor](instance, { context, transaction });
+                    if (updateAssociationValues.includes(key)) {
+                        yield instance.update(value, Object.assign(Object.assign({}, options), { transaction }));
+                    }
+                    yield updateAssociations(instance, value, Object.assign(Object.assign({}, options), { transaction, associationContext: association, updateAssociationValues: keys }));
+                    model.setDataValue(key, instance);
+                    if (!options.transaction) {
+                        yield transaction.commit();
+                    }
+                    return true;
+                }
+            }
+            const instance = yield model[createAccessor](value, { context, transaction });
+            yield updateAssociations(instance, value, Object.assign(Object.assign({}, options), { transaction, associationContext: association, updateAssociationValues: keys }));
+            model.setDataValue(key, instance);
+            // @ts-ignore
+            if (association.targetKey) {
+                model.setDataValue(association.foreignKey, instance[dataKey]);
+            }
+            if (!options.transaction) {
+                yield transaction.commit();
+            }
+        }
+        catch (error) {
+            if (!options.transaction) {
+                yield transaction.rollback();
+            }
+            throw error;
+        }
+    });
+}
+/**
+ * update multiple association of model by value
+ * @param model
+ * @param key
+ * @param value
+ * @param options
+ */
+export function updateMultipleAssociation(model, key, value, options = {}) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const association = modelAssociationByKey(model, key);
+        if (!association) {
+            return false;
+        }
+        if (!['undefined', 'string', 'number', 'object'].includes(typeof value)) {
+            return false;
+        }
+        const { context, updateAssociationValues = [], transaction = yield model.sequelize.transaction() } = options;
+        const keys = getKeysByPrefix(updateAssociationValues, key);
+        try {
+            const setAccessor = association.accessors.set;
+            const createAccessor = association.accessors.create;
+            if (isUndefinedOrNull(value)) {
+                yield model[setAccessor](null, { transaction, context });
+                model.setDataValue(key, null);
+                return;
+            }
+            if (isStringOrNumber(value)) {
+                yield model[setAccessor](value, { transaction, context });
+                return;
+            }
+            if (!Array.isArray(value)) {
+                value = [value];
+            }
+            const list1 = []; // to be setted
+            const list2 = []; // to be added
+            for (const item of value) {
+                if (isUndefinedOrNull(item)) {
+                    continue;
+                }
+                if (isStringOrNumber(item)) {
+                    list1.push(item);
+                }
+                else if (item instanceof Model) {
+                    list1.push(item);
+                }
+                else if (item.sequelize) {
+                    list1.push(item);
+                }
+                else if (typeof item === 'object') {
+                    list2.push(item);
+                }
+            }
+            // associate targets in lists1
+            yield model[setAccessor](list1, { transaction, context });
+            const list3 = [];
+            for (const item of list2) {
+                const pk = association.target.primaryKeyAttribute;
+                const through = association.through ? association.through.model.name : null;
+                const accessorOptions = {
+                    context,
+                    transaction,
+                };
+                const throughValue = item[through];
+                if (throughValue) {
+                    accessorOptions['through'] = throughValue;
+                }
+                if (isUndefinedOrNull(item[pk])) {
+                    // create new record
+                    const instance = yield model[createAccessor](item, accessorOptions);
+                    yield updateAssociations(instance, item, Object.assign(Object.assign({}, options), { transaction, associationContext: association, updateAssociationValues: keys }));
+                    list3.push(instance);
+                }
+                else {
+                    // set & update record
+                    const instance = yield association.target.findByPk(item[pk], {
+                        transaction,
+                    });
+                    const addAccessor = association.accessors.add;
+                    yield model[addAccessor](item[pk], accessorOptions);
+                    if (updateAssociationValues.includes(key)) {
+                        yield instance.update(item, Object.assign(Object.assign({}, options), { transaction }));
+                    }
+                    yield updateAssociations(instance, item, Object.assign(Object.assign({}, options), { transaction, associationContext: association, updateAssociationValues: keys }));
+                    list3.push(instance);
+                }
+            }
+            model.setDataValue(key, list1.concat(list3));
+            if (!options.transaction) {
+                yield transaction.commit();
+            }
+        }
+        catch (error) {
+            yield transaction.rollback();
+            throw error;
+        }
+    });
+}
+//# sourceMappingURL=update-associations.js.map

package/esm/update-associations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-associations.js","sourceRoot":"","sources":["../src/update-associations.ts"],"names":[],"mappings":";;;;;;;;;AAUA,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAEhC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,SAAS,iBAAiB,CAAC,KAAU;IACnC,OAAO,OAAO,KAAK,KAAK,WAAW,IAAI,KAAK,KAAK,IAAI,CAAC;AACxD,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAU;IAClC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAC;AAChE,CAAC;AAED,SAAS,eAAe,CAAC,IAAc,EAAE,MAAc;IACrD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;AAC3G,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,QAAe;IAC/C,OAAsB,QAAQ,CAAC,WAAY,CAAC,YAAY,CAAC;AAC3D,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAAe;IACvD,MAAM,YAAY,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IACjD,OAAO,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC;SAChC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QAChB,MAAM,CAAC,GAAG,EAAE,WAAW,CAAC,GAAG,KAAK,CAAC;QACjC,OAAO,WAAW,CAAC,eAAe,IAAI,eAAe,CAAC;IACxD,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE;QACnB,OAAsB,WAAW,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,QAAe,EAAE,GAAW;IAChE,OAAO,iBAAiB,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAgB,CAAC;AACzD,CAAC;AAyBD,MAAM,UAAgB,mBAAmB,CAAC,QAAe,EAAE,MAAmB,EAAE,OAAuB;;QACrG,IAAI,CAAC,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,CAAA,EAAE;YACvB,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;YAChC,YAAY;YACZ,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YACrC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACtC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACtC,KAAK,CAAC,4BAA4B,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;YACpE,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;SACjC;QAED,MAAM,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACvC,MAAM,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;CAAA;AAED,MAAM,UAAgB,uBAAuB,CAC3C,QAAe,EACf,WAAmB,EACnB,aAAkB,EAClB,MAAa,EACb,WAAW,GAAG,IAAI;;QAElB,8BAA8B;QAC9B,KAAK,MAAM,aAAa,IAAI,yBAAyB,CAAC,QAAQ,CAAC,EAAE;YAC/D,aAAa;YACb,MAAM,YAAY,GAAG,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC;YACjD,MAAM,gBAAgB,GAAG,YAAY,CAAC,IAAI,CAAC;YAE3C,IAAI,gBAAgB,KAAK,gBAAgB,EAAE;gBACzC,MAAM,KAAK,GAAG;oBACZ,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC;oBACjE,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC;iBAC9D,CAAC;gBAEF,OAAO,MAAM,YAAY,CAAC,MAAM,CAAC,aAAa,EAAE;oBAC9C,KAAK;oBACL,WAAW;iBACZ,CAAC,CAAC;aACJ;SACF;IACH,CAAC;CAAA;AAED;;;;;GAKG;AACH,MAAM,UAAgB,kBAAkB,CAAC,QAAe,EAAE,MAAW,EAAE,UAAoC,EAAE;;QAC3G,2BAA2B;QAC3B,IAAI,CAAC,MAAM,EAAE;YACX,OAAO;SACR;QAED,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,IAAI,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QAEtC,IAAI,CAAC,WAAW,EAAE;YAChB,cAAc,GAAG,IAAI,CAAC;YACtB,WAAW,GAAG,MAAM,QAAQ,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC;SACtD;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEjC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,EAAE;YAC1D,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;gBACtB,MAAM,iBAAiB,CAAC,QAAQ,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,kCAC7C,OAAO,KACV,WAAW,IACX,CAAC;aACJ;SACF;QAED,8BAA8B;QAC9B,KAAK,MAAM,aAAa,IAAI,yBAAyB,CAAC,QAAQ,CAAC,EAAE;YAC/D,aAAa;YACb,MAAM,YAAY,GAAG,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC;YACjD,MAAM,gBAAgB,GAAG,YAAY,CAAC,IAAI,CAAC;YAE3C,IAAI,MAAM,CAAC,gBAAgB,CAAC,IAAI,OAAO,CAAC,WAAW,EAAE;gBACnD,MAAM,KAAK,GAAG;oBACZ,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC;oBACjE,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC;iBAC3E,CAAC;gBAEF,MAAM,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE;oBACnD,KAAK;oBACL,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,WAAW;iBACZ,CAAC,CAAC;aACJ;SACF;QAED,IAAI,cAAc,EAAE;YAClB,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;SAC5B;IACH,CAAC;CAAA;AAED,SAAS,wBAAwB,CAAC,CAAM,EAAE,CAAM;IAC9C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;IAC1B,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;IAC/B,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;IAE/B,IAAI,OAAO,CAAC,IAAI,IAAI,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE;QACrD,OAAO,CACL,CAAC,CAAC,OAAO,CAAC,SAAS,KAAK,CAAC,CAAC,OAAO,CAAC,SAAS;YAC3C,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI;YAC/B,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI;YAC/B,CAAC,CAAC,UAAU,KAAK,CAAC,CAAC,QAAQ;YAC3B,CAAC,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS;YAC3B,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,UAAU;YAC3B,CAAC,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS,CAC5B,CAAC;KACH;IAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,EAAE;QAC/G,MAAM,WAAW,GAAG,CAAC,CAAC,eAAe,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7D,MAAM,WAAW,GAAG,WAAW,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAE7C,OAAO,CACL,WAAW,CAAC,MAAM,CAAC,IAAI,KAAK,WAAW,CAAC,MAAM,CAAC,IAAI;YACnD,WAAW,CAAC,UAAU,KAAK,WAAW,CAAC,UAAU;YACjD,WAAW,CAAC,SAAS,KAAK,WAAW,CAAC,SAAS,CAChD,CAAC;KACH;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAgB,iBAAiB,CACrC,QAAe,EACf,GAAW,EACX,KAAU,EACV,UAAoC,EAAE;;QAEtC,MAAM,WAAW,GAAG,qBAAqB,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QAEzD,IAAI,CAAC,WAAW,EAAE;YAChB,OAAO,KAAK,CAAC;SACd;QAED,IAAI,OAAO,CAAC,kBAAkB,IAAI,wBAAwB,CAAC,WAAW,EAAE,OAAO,CAAC,kBAAkB,CAAC,EAAE;YACnG,OAAO,KAAK,CAAC;SACd;QAED,QAAQ,WAAW,CAAC,eAAe,EAAE;YACnC,KAAK,QAAQ,CAAC;YACd,KAAK,WAAW;gBACd,OAAO,uBAAuB,CAAC,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;YAChE,KAAK,SAAS,CAAC;YACf,KAAK,eAAe;gBAClB,OAAO,yBAAyB,CAAC,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;SACnE;IACH,CAAC;CAAA;AAED;;;;;;GAMG;AACH,MAAM,UAAgB,uBAAuB,CAC3C,KAAY,EACZ,GAAW,EACX,KAAU,EACV,UAAoC,EAAE;;QAEtC,MAAM,WAAW,GAAuB,qBAAqB,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAE1E,IAAI,CAAC,WAAW,EAAE;YAChB,OAAO,KAAK,CAAC;SACd;QAED,IAAI,CAAC,CAAC,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,OAAO,KAAK,CAAC,EAAE;YACvE,OAAO,KAAK,CAAC;SACd;QAED,MAAM,EAAE,OAAO,EAAE,uBAAuB,GAAG,EAAE,EAAE,WAAW,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,GAAG,OAAO,CAAC;QAC7G,MAAM,IAAI,GAAG,eAAe,CAAC,uBAAuB,EAAE,GAAG,CAAC,CAAC;QAE3D,IAAI;YACF,4BAA4B;YAC5B,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC;YAE9C,MAAM,iBAAiB,GAAG,GAAS,EAAE;gBACnC,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;gBAChD,KAAK,CAAC,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;gBAC9B,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;oBACxB,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;iBAC5B;gBACD,OAAO,IAAI,CAAC;YACd,CAAC,CAAA,CAAC;YAEF,IAAI,iBAAiB,CAAC,KAAK,CAAC,EAAE;gBAC5B,OAAO,MAAM,iBAAiB,EAAE,CAAC;aAClC;YAED,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE;gBAC3B,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;gBAC1D,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;oBACxB,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;iBAC5B;gBACD,OAAO,IAAI,CAAC;aACb;YAED,IAAI,KAAK,YAAY,KAAK,EAAE;gBAC1B,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;gBAC1D,KAAK,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAE/B,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;oBACxB,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;iBAC5B;gBACD,OAAO,IAAI,CAAC;aACb;YAED,MAAM,cAAc,GAAG,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC;YACpD,IAAI,OAAe,CAAC;YACpB,IAAI,CAAmB,CAAC;YACxB,IAAI,WAAW,CAAC,eAAe,KAAK,WAAW,EAAE;gBAC/C,CAAC,GAAG,WAAW,CAAC,MAA0B,CAAC;gBAC3C,aAAa;gBACb,OAAO,GAAG,WAAW,CAAC,SAAS,CAAC;aACjC;iBAAM;gBACL,CAAC,GAAG,WAAW,CAAC,MAA0B,CAAC;gBAC3C,OAAO,GAAG,CAAC,CAAC,mBAAmB,CAAC;aACjC;YAED,IAAI,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE;gBACpC,IAAI,QAAQ,GAAQ,MAAM,CAAC,CAAC,OAAO,CAAC;oBAClC,KAAK,EAAE;wBACL,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC;qBAC1B;oBACD,WAAW;iBACZ,CAAC,CAAC;gBAEH,IAAI,QAAQ,EAAE;oBACZ,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;oBAE7D,IAAI,uBAAuB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;wBACzC,MAAM,QAAQ,CAAC,MAAM,CAAC,KAAK,kCAAO,OAAO,KAAE,WAAW,IAAG,CAAC;qBAC3D;oBAED,MAAM,kBAAkB,CAAC,QAAQ,EAAE,KAAK,kCACnC,OAAO,KACV,WAAW,EACX,kBAAkB,EAAE,WAAW,EAC/B,uBAAuB,EAAE,IAAI,IAC7B,CAAC;oBACH,KAAK,CAAC,YAAY,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;oBAClC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;wBACxB,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;qBAC5B;oBACD,OAAO,IAAI,CAAC;iBACb;aACF;YAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YAC9E,MAAM,kBAAkB,CAAC,QAAQ,EAAE,KAAK,kCACnC,OAAO,KACV,WAAW,EACX,kBAAkB,EAAE,WAAW,EAC/B,uBAAuB,EAAE,IAAI,IAC7B,CAAC;YACH,KAAK,CAAC,YAAY,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAClC,aAAa;YACb,IAAI,WAAW,CAAC,SAAS,EAAE;gBACzB,KAAK,CAAC,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;aAC/D;YACD,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;gBACxB,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;aAC5B;SACF;QAAC,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;gBACxB,MAAM,WAAW,CAAC,QAAQ,EAAE,CAAC;aAC9B;YACD,MAAM,KAAK,CAAC;SACb;IACH,CAAC;CAAA;AAED;;;;;;GAMG;AACH,MAAM,UAAgB,yBAAyB,CAC7C,KAAY,EACZ,GAAW,EACX,KAAU,EACV,UAAoC,EAAE;;QAEtC,MAAM,WAAW,GAA4B,qBAAqB,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAE/E,IAAI,CAAC,WAAW,EAAE;YAChB,OAAO,KAAK,CAAC;SACd;QAED,IAAI,CAAC,CAAC,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,OAAO,KAAK,CAAC,EAAE;YACvE,OAAO,KAAK,CAAC;SACd;QAED,MAAM,EAAE,OAAO,EAAE,uBAAuB,GAAG,EAAE,EAAE,WAAW,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,GAAG,OAAO,CAAC;QAC7G,MAAM,IAAI,GAAG,eAAe,CAAC,uBAAuB,EAAE,GAAG,CAAC,CAAC;QAE3D,IAAI;YACF,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC;YAE9C,MAAM,cAAc,GAAG,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC;YACpD,IAAI,iBAAiB,CAAC,KAAK,CAAC,EAAE;gBAC5B,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;gBACzD,KAAK,CAAC,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;gBAC9B,OAAO;aACR;YAED,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE;gBAC3B,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC1D,OAAO;aACR;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;gBACzB,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC;aACjB;YAED,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,eAAe;YACjC,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,cAAc;YAChC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;gBACxB,IAAI,iBAAiB,CAAC,IAAI,CAAC,EAAE;oBAC3B,SAAS;iBACV;gBACD,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE;oBAC1B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAClB;qBAAM,IAAI,IAAI,YAAY,KAAK,EAAE;oBAChC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAClB;qBAAM,IAAI,IAAI,CAAC,SAAS,EAAE;oBACzB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAClB;qBAAM,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;oBACnC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAClB;aACF;YAED,8BAA8B;YAC9B,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,KAAK,GAAG,EAAE,CAAC;YACjB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;gBACxB,MAAM,EAAE,GAAG,WAAW,CAAC,MAAM,CAAC,mBAAmB,CAAC;gBAElD,MAAM,OAAO,GAAS,WAAY,CAAC,OAAO,CAAC,CAAC,CAAO,WAAY,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;gBAE1F,MAAM,eAAe,GAAG;oBACtB,OAAO;oBACP,WAAW;iBACZ,CAAC;gBAEF,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;gBAEnC,IAAI,YAAY,EAAE;oBAChB,eAAe,CAAC,SAAS,CAAC,GAAG,YAAY,CAAC;iBAC3C;gBAED,IAAI,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE;oBAC/B,oBAAoB;oBACpB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;oBACpE,MAAM,kBAAkB,CAAC,QAAQ,EAAE,IAAI,kCAClC,OAAO,KACV,WAAW,EACX,kBAAkB,EAAE,WAAW,EAC/B,uBAAuB,EAAE,IAAI,IAC7B,CAAC;oBACH,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;iBACtB;qBAAM;oBACL,sBAAsB;oBACtB,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAM,IAAI,CAAC,EAAE,CAAC,EAAE;wBAChE,WAAW;qBACZ,CAAC,CAAC;oBACH,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC;oBAE9C,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,eAAe,CAAC,CAAC;oBACpD,IAAI,uBAAuB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;wBACzC,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,kCAAO,OAAO,KAAE,WAAW,IAAG,CAAC;qBAC1D;oBACD,MAAM,kBAAkB,CAAC,QAAQ,EAAE,IAAI,kCAClC,OAAO,KACV,WAAW,EACX,kBAAkB,EAAE,WAAW,EAC/B,uBAAuB,EAAE,IAAI,IAC7B,CAAC;oBACH,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;iBACtB;aACF;YAED,KAAK,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC7C,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;gBACxB,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;aAC5B;SACF;QAAC,OAAO,KAAK,EAAE;YACd,MAAM,WAAW,CAAC,QAAQ,EAAE,CAAC;YAC7B,MAAM,KAAK,CAAC;SACb;IACH,CAAC;CAAA","sourcesContent":["import {\n  Association,\n  BelongsTo,\n  BelongsToMany,\n  HasMany,\n  HasOne,\n  Hookable,\n  ModelCtor,\n  Transactionable,\n} from 'sequelize';\nimport { Model } from './model';\nimport { TransactionAble } from './repository';\nimport { UpdateGuard } from './update-guard';\n\nfunction isUndefinedOrNull(value: any) {\n  return typeof value === 'undefined' || value === null;\n}\n\nfunction isStringOrNumber(value: any) {\n  return typeof value === 'string' || typeof value === 'number';\n}\n\nfunction getKeysByPrefix(keys: string[], prefix: string) {\n  return keys.filter((key) => key.startsWith(`${prefix}.`)).map((key) => key.substring(prefix.length + 1));\n}\n\nexport function modelAssociations(instance: Model) {\n  return (<typeof Model>instance.constructor).associations;\n}\n\nexport function belongsToManyAssociations(instance: Model): Array<BelongsToMany> {\n  const associations = modelAssociations(instance);\n  return Object.entries(associations)\n    .filter((entry) => {\n      const [key, association] = entry;\n      return association.associationType == 'BelongsToMany';\n    })\n    .map((association) => {\n      return <BelongsToMany>association[1];\n    });\n}\n\nexport function modelAssociationByKey(instance: Model, key: string): Association {\n  return modelAssociations(instance)[key] as Association;\n}\n\ntype UpdateValue = { [key: string]: any };\n\ninterface UpdateOptions extends TransactionAble {\n  filter?: any;\n  filterByTk?: number | string;\n  // 字段白名单\n  whitelist?: string[];\n  // 字段黑名单\n  blacklist?: string[];\n  // 关系数据默认会新建并建立关联处理，如果是已存在的数据只关联，但不更新关系数据\n  // 如果需要更新关联数据，可以通过 updateAssociationValues 指定\n  updateAssociationValues?: string[];\n  sanitized?: boolean;\n  sourceModel?: Model;\n}\n\ninterface UpdateAssociationOptions extends Transactionable, Hookable {\n  updateAssociationValues?: string[];\n  sourceModel?: Model;\n  context?: any;\n  associationContext?: any;\n}\n\nexport async function updateModelByValues(instance: Model, values: UpdateValue, options?: UpdateOptions) {\n  if (!options?.sanitized) {\n    const guard = new UpdateGuard();\n    //@ts-ignore\n    guard.setModel(instance.constructor);\n    guard.setBlackList(options.blacklist);\n    guard.setWhiteList(options.whitelist);\n    guard.setAssociationKeysToBeUpdate(options.updateAssociationValues);\n    values = guard.sanitize(values);\n  }\n\n  await instance.update(values, options);\n  await updateAssociations(instance, values, options);\n}\n\nexport async function updateThroughTableValue(\n  instance: Model,\n  throughName: string,\n  throughValues: any,\n  source: Model,\n  transaction = null,\n) {\n  // update through table values\n  for (const belongsToMany of belongsToManyAssociations(instance)) {\n    // @ts-ignore\n    const throughModel = belongsToMany.through.model;\n    const throughModelName = throughModel.name;\n\n    if (throughModelName === throughModelName) {\n      const where = {\n        [belongsToMany.foreignKey]: instance.get(belongsToMany.sourceKey),\n        [belongsToMany.otherKey]: source.get(belongsToMany.targetKey),\n      };\n\n      return await throughModel.update(throughValues, {\n        where,\n        transaction,\n      });\n    }\n  }\n}\n\n/**\n * update association of instance by values\n * @param instance\n * @param values\n * @param options\n */\nexport async function updateAssociations(instance: Model, values: any, options: UpdateAssociationOptions = {}) {\n  // if no values set, return\n  if (!values) {\n    return;\n  }\n\n  let newTransaction = false;\n  let transaction = options.transaction;\n\n  if (!transaction) {\n    newTransaction = true;\n    transaction = await instance.sequelize.transaction();\n  }\n\n  const keys = Object.keys(values);\n\n  for (const key of Object.keys(modelAssociations(instance))) {\n    if (keys.includes(key)) {\n      await updateAssociation(instance, key, values[key], {\n        ...options,\n        transaction,\n      });\n    }\n  }\n\n  // update through table values\n  for (const belongsToMany of belongsToManyAssociations(instance)) {\n    // @ts-ignore\n    const throughModel = belongsToMany.through.model;\n    const throughModelName = throughModel.name;\n\n    if (values[throughModelName] && options.sourceModel) {\n      const where = {\n        [belongsToMany.foreignKey]: instance.get(belongsToMany.sourceKey),\n        [belongsToMany.otherKey]: options.sourceModel.get(belongsToMany.targetKey),\n      };\n\n      await throughModel.update(values[throughModel.name], {\n        where,\n        context: options.context,\n        transaction,\n      });\n    }\n  }\n\n  if (newTransaction) {\n    await transaction.commit();\n  }\n}\n\nfunction isReverseAssociationPair(a: any, b: any) {\n  const typeSet = new Set();\n  typeSet.add(a.associationType);\n  typeSet.add(b.associationType);\n\n  if (typeSet.size == 1 && typeSet.has('BelongsToMany')) {\n    return (\n      a.through.tableName === b.through.tableName &&\n      a.target.name === b.source.name &&\n      b.target.name === a.source.name &&\n      a.foreignKey === b.otherKey &&\n      a.sourceKey === b.targetKey &&\n      a.otherKey === b.foreignKey &&\n      a.targetKey === b.sourceKey\n    );\n  }\n\n  if ((typeSet.has('HasOne') && typeSet.has('BelongsTo')) || (typeSet.has('HasMany') && typeSet.has('BelongsTo'))) {\n    const sourceAssoc = a.associationType == 'BelongsTo' ? b : a;\n    const targetAssoc = sourceAssoc == a ? b : a;\n\n    return (\n      sourceAssoc.source.name === targetAssoc.target.name &&\n      sourceAssoc.foreignKey === targetAssoc.foreignKey &&\n      sourceAssoc.sourceKey === targetAssoc.targetKey\n    );\n  }\n\n  return false;\n}\n\n/**\n * update model association by key\n * @param instance\n * @param key\n * @param value\n * @param options\n */\nexport async function updateAssociation(\n  instance: Model,\n  key: string,\n  value: any,\n  options: UpdateAssociationOptions = {},\n) {\n  const association = modelAssociationByKey(instance, key);\n\n  if (!association) {\n    return false;\n  }\n\n  if (options.associationContext && isReverseAssociationPair(association, options.associationContext)) {\n    return false;\n  }\n\n  switch (association.associationType) {\n    case 'HasOne':\n    case 'BelongsTo':\n      return updateSingleAssociation(instance, key, value, options);\n    case 'HasMany':\n    case 'BelongsToMany':\n      return updateMultipleAssociation(instance, key, value, options);\n  }\n}\n\n/**\n * update belongsTo and HasOne\n * @param model\n * @param key\n * @param value\n * @param options\n */\nexport async function updateSingleAssociation(\n  model: Model,\n  key: string,\n  value: any,\n  options: UpdateAssociationOptions = {},\n) {\n  const association = <HasOne | BelongsTo>modelAssociationByKey(model, key);\n\n  if (!association) {\n    return false;\n  }\n\n  if (!['undefined', 'string', 'number', 'object'].includes(typeof value)) {\n    return false;\n  }\n\n  const { context, updateAssociationValues = [], transaction = await model.sequelize.transaction() } = options;\n  const keys = getKeysByPrefix(updateAssociationValues, key);\n\n  try {\n    // set method of association\n    const setAccessor = association.accessors.set;\n\n    const removeAssociation = async () => {\n      await model[setAccessor](null, { transaction });\n      model.setDataValue(key, null);\n      if (!options.transaction) {\n        await transaction.commit();\n      }\n      return true;\n    };\n\n    if (isUndefinedOrNull(value)) {\n      return await removeAssociation();\n    }\n\n    if (isStringOrNumber(value)) {\n      await model[setAccessor](value, { context, transaction });\n      if (!options.transaction) {\n        await transaction.commit();\n      }\n      return true;\n    }\n\n    if (value instanceof Model) {\n      await model[setAccessor](value, { context, transaction });\n      model.setDataValue(key, value);\n\n      if (!options.transaction) {\n        await transaction.commit();\n      }\n      return true;\n    }\n\n    const createAccessor = association.accessors.create;\n    let dataKey: string;\n    let M: ModelCtor<Model>;\n    if (association.associationType === 'BelongsTo') {\n      M = association.target as ModelCtor<Model>;\n      // @ts-ignore\n      dataKey = association.targetKey;\n    } else {\n      M = association.source as ModelCtor<Model>;\n      dataKey = M.primaryKeyAttribute;\n    }\n\n    if (isStringOrNumber(value[dataKey])) {\n      let instance: any = await M.findOne({\n        where: {\n          [dataKey]: value[dataKey],\n        },\n        transaction,\n      });\n\n      if (instance) {\n        await model[setAccessor](instance, { context, transaction });\n\n        if (updateAssociationValues.includes(key)) {\n          await instance.update(value, { ...options, transaction });\n        }\n\n        await updateAssociations(instance, value, {\n          ...options,\n          transaction,\n          associationContext: association,\n          updateAssociationValues: keys,\n        });\n        model.setDataValue(key, instance);\n        if (!options.transaction) {\n          await transaction.commit();\n        }\n        return true;\n      }\n    }\n\n    const instance = await model[createAccessor](value, { context, transaction });\n    await updateAssociations(instance, value, {\n      ...options,\n      transaction,\n      associationContext: association,\n      updateAssociationValues: keys,\n    });\n    model.setDataValue(key, instance);\n    // @ts-ignore\n    if (association.targetKey) {\n      model.setDataValue(association.foreignKey, instance[dataKey]);\n    }\n    if (!options.transaction) {\n      await transaction.commit();\n    }\n  } catch (error) {\n    if (!options.transaction) {\n      await transaction.rollback();\n    }\n    throw error;\n  }\n}\n\n/**\n * update multiple association of model by value\n * @param model\n * @param key\n * @param value\n * @param options\n */\nexport async function updateMultipleAssociation(\n  model: Model,\n  key: string,\n  value: any,\n  options: UpdateAssociationOptions = {},\n) {\n  const association = <BelongsToMany | HasMany>modelAssociationByKey(model, key);\n\n  if (!association) {\n    return false;\n  }\n\n  if (!['undefined', 'string', 'number', 'object'].includes(typeof value)) {\n    return false;\n  }\n\n  const { context, updateAssociationValues = [], transaction = await model.sequelize.transaction() } = options;\n  const keys = getKeysByPrefix(updateAssociationValues, key);\n\n  try {\n    const setAccessor = association.accessors.set;\n\n    const createAccessor = association.accessors.create;\n    if (isUndefinedOrNull(value)) {\n      await model[setAccessor](null, { transaction, context });\n      model.setDataValue(key, null);\n      return;\n    }\n\n    if (isStringOrNumber(value)) {\n      await model[setAccessor](value, { transaction, context });\n      return;\n    }\n\n    if (!Array.isArray(value)) {\n      value = [value];\n    }\n\n    const list1 = []; // to be setted\n    const list2 = []; // to be added\n    for (const item of value) {\n      if (isUndefinedOrNull(item)) {\n        continue;\n      }\n      if (isStringOrNumber(item)) {\n        list1.push(item);\n      } else if (item instanceof Model) {\n        list1.push(item);\n      } else if (item.sequelize) {\n        list1.push(item);\n      } else if (typeof item === 'object') {\n        list2.push(item);\n      }\n    }\n\n    // associate targets in lists1\n    await model[setAccessor](list1, { transaction, context });\n\n    const list3 = [];\n    for (const item of list2) {\n      const pk = association.target.primaryKeyAttribute;\n\n      const through = (<any>association).through ? (<any>association).through.model.name : null;\n\n      const accessorOptions = {\n        context,\n        transaction,\n      };\n\n      const throughValue = item[through];\n\n      if (throughValue) {\n        accessorOptions['through'] = throughValue;\n      }\n\n      if (isUndefinedOrNull(item[pk])) {\n        // create new record\n        const instance = await model[createAccessor](item, accessorOptions);\n        await updateAssociations(instance, item, {\n          ...options,\n          transaction,\n          associationContext: association,\n          updateAssociationValues: keys,\n        });\n        list3.push(instance);\n      } else {\n        // set & update record\n        const instance = await association.target.findByPk<any>(item[pk], {\n          transaction,\n        });\n        const addAccessor = association.accessors.add;\n\n        await model[addAccessor](item[pk], accessorOptions);\n        if (updateAssociationValues.includes(key)) {\n          await instance.update(item, { ...options, transaction });\n        }\n        await updateAssociations(instance, item, {\n          ...options,\n          transaction,\n          associationContext: association,\n          updateAssociationValues: keys,\n        });\n        list3.push(instance);\n      }\n    }\n\n    model.setDataValue(key, list1.concat(list3));\n    if (!options.transaction) {\n      await transaction.commit();\n    }\n  } catch (error) {\n    await transaction.rollback();\n    throw error;\n  }\n}\n"]}

package/esm/update-guard.d.ts

@@ -0,0 +1,26 @@
+import { ModelCtor } from 'sequelize';
+import { AssociationKeysToBeUpdate, BlackList, WhiteList } from './repository';
+declare type UpdateValueItem = string | number | UpdateValues;
+declare type UpdateValues = {
+    [key: string]: UpdateValueItem | Array<UpdateValueItem>;
+};
+declare type UpdateAction = 'create' | 'update';
+export declare class UpdateGuard {
+    model: ModelCtor<any>;
+    action: UpdateAction;
+    private associationKeysToBeUpdate;
+    private blackList;
+    private whiteList;
+    setAction(action: UpdateAction): void;
+    setModel(model: ModelCtor<any>): void;
+    setAssociationKeysToBeUpdate(associationKeysToBeUpdate: AssociationKeysToBeUpdate): void;
+    setWhiteList(whiteList: WhiteList): void;
+    setBlackList(blackList: BlackList): void;
+    /**
+     * Sanitize values by whitelist blacklist
+     * @param values
+     */
+    sanitize(values: UpdateValues): {};
+    static fromOptions(model: any, options: any): UpdateGuard;
+}
+export {};

package/esm/update-guard.js

@@ -0,0 +1,122 @@
+import lodash from 'lodash';
+import { Model } from './model';
+export class UpdateGuard {
+    setAction(action) {
+        this.action = action;
+    }
+    setModel(model) {
+        this.model = model;
+    }
+    setAssociationKeysToBeUpdate(associationKeysToBeUpdate) {
+        if (this.action == 'create') {
+            this.associationKeysToBeUpdate = Object.keys(this.model.associations);
+        }
+        else {
+            this.associationKeysToBeUpdate = associationKeysToBeUpdate;
+        }
+    }
+    setWhiteList(whiteList) {
+        this.whiteList = whiteList;
+    }
+    setBlackList(blackList) {
+        this.blackList = blackList;
+    }
+    /**
+     * Sanitize values by whitelist blacklist
+     * @param values
+     */
+    sanitize(values) {
+        values = lodash.clone(values);
+        if (!this.model) {
+            throw new Error('please set model first');
+        }
+        const associations = this.model.associations;
+        const associationsValues = lodash.pick(values, Object.keys(associations));
+        // build params of association update guard
+        const listOfAssociation = (list, association) => {
+            if (list) {
+                list = list
+                    .filter((whiteListKey) => whiteListKey.startsWith(`${association}.`))
+                    .map((whiteListKey) => whiteListKey.replace(`${association}.`, ''));
+                if (list.length == 0) {
+                    return undefined;
+                }
+                return list;
+            }
+            return undefined;
+        };
+        // sanitize association values
+        Object.keys(associationsValues).forEach((association) => {
+            let associationValues = associationsValues[association];
+            const filterAssociationToBeUpdate = (value) => {
+                const associationKeysToBeUpdate = this.associationKeysToBeUpdate || [];
+                if (associationKeysToBeUpdate.includes(association)) {
+                    return value;
+                }
+                const associationObj = associations[association];
+                const associationKeyName = associationObj.associationType == 'BelongsTo' || associationObj.associationType == 'HasOne'
+                    ? associationObj.targetKey
+                    : associationObj.target.primaryKeyAttribute;
+                if (value[associationKeyName]) {
+                    return lodash.pick(value, [associationKeyName, ...Object.keys(associationObj.target.associations)]);
+                }
+                return value;
+            };
+            const sanitizeValue = (value) => {
+                const associationUpdateGuard = new UpdateGuard();
+                associationUpdateGuard.setModel(associations[association].target);
+                ['whiteList', 'blackList', 'associationKeysToBeUpdate'].forEach((optionKey) => {
+                    associationUpdateGuard[`set${lodash.upperFirst(optionKey)}`](listOfAssociation(this[optionKey], association));
+                });
+                return associationUpdateGuard.sanitize(filterAssociationToBeUpdate(value));
+            };
+            if (Array.isArray(associationValues)) {
+                associationValues = associationValues.map((value) => {
+                    if (typeof value == 'string' || typeof value == 'number') {
+                        return value;
+                    }
+                    else {
+                        return sanitizeValue(value);
+                    }
+                });
+            }
+            else if (typeof associationValues === 'object' && associationValues !== null) {
+                associationValues = sanitizeValue(associationValues);
+            }
+            // set association values to sanitized value
+            values[association] = associationValues;
+        });
+        if (values instanceof Model) {
+            return values;
+        }
+        let valuesKeys = Object.keys(values || {});
+        // handle whitelist
+        if (this.whiteList) {
+            valuesKeys = valuesKeys.filter((valueKey) => {
+                return (this.whiteList.findIndex((whiteKey) => {
+                    const keyPaths = whiteKey.split('.');
+                    return keyPaths[0] === valueKey;
+                }) !== -1);
+            });
+        }
+        // handle blacklist
+        if (this.blackList) {
+            valuesKeys = valuesKeys.filter((valueKey) => !this.blackList.includes(valueKey));
+        }
+        const result = valuesKeys.reduce((obj, key) => {
+            lodash.set(obj, key, values[key]);
+            return obj;
+        }, {});
+        return result;
+    }
+    static fromOptions(model, options) {
+        const guard = new UpdateGuard();
+        guard.setModel(model);
+        guard.setWhiteList(options.whitelist);
+        guard.setBlackList(options.blacklist);
+        guard.setAction(lodash.get(options, 'action', 'update'));
+        guard.setAssociationKeysToBeUpdate(options.updateAssociationValues);
+        return guard;
+    }
+}
+//# sourceMappingURL=update-guard.js.map

package/esm/update-guard.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-guard.js","sourceRoot":"","sources":["../src/update-guard.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAUhC,MAAM,OAAO,WAAW;IAOtB,SAAS,CAAC,MAAoB;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,QAAQ,CAAC,KAAqB;QAC5B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,4BAA4B,CAAC,yBAAoD;QAC/E,IAAI,IAAI,CAAC,MAAM,IAAI,QAAQ,EAAE;YAC3B,IAAI,CAAC,yBAAyB,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;SACvE;aAAM;YACL,IAAI,CAAC,yBAAyB,GAAG,yBAAyB,CAAC;SAC5D;IACH,CAAC;IAED,YAAY,CAAC,SAAoB;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,YAAY,CAAC,SAAoB;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED;;;OAGG;IACH,QAAQ,CAAC,MAAoB;QAC3B,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE9B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;YACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;SAC3C;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;QAC7C,MAAM,kBAAkB,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QAE1E,2CAA2C;QAC3C,MAAM,iBAAiB,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE;YAC9C,IAAI,IAAI,EAAE;gBACR,IAAI,GAAG,IAAI;qBACR,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,WAAW,GAAG,CAAC,CAAC;qBACpE,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,WAAW,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;gBAEtE,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE;oBACpB,OAAO,SAAS,CAAC;iBAClB;gBAED,OAAO,IAAI,CAAC;aACb;YAED,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC;QAEF,8BAA8B;QAC9B,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;YACtD,IAAI,iBAAiB,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;YAExD,MAAM,2BAA2B,GAAG,CAAC,KAAK,EAAE,EAAE;gBAC5C,MAAM,yBAAyB,GAAG,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC;gBAEvE,IAAI,yBAAyB,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE;oBACnD,OAAO,KAAK,CAAC;iBACd;gBAED,MAAM,cAAc,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;gBAEjD,MAAM,kBAAkB,GACtB,cAAc,CAAC,eAAe,IAAI,WAAW,IAAI,cAAc,CAAC,eAAe,IAAI,QAAQ;oBACzF,CAAC,CAAO,cAAe,CAAC,SAAS;oBACjC,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,mBAAmB,CAAC;gBAEhD,IAAI,KAAK,CAAC,kBAAkB,CAAC,EAAE;oBAC7B,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,kBAAkB,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;iBACrG;gBAED,OAAO,KAAK,CAAC;YACf,CAAC,CAAC;YAEF,MAAM,aAAa,GAAG,CAAC,KAAK,EAAE,EAAE;gBAC9B,MAAM,sBAAsB,GAAG,IAAI,WAAW,EAAE,CAAC;gBACjD,sBAAsB,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC;gBAElE,CAAC,WAAW,EAAE,WAAW,EAAE,2BAA2B,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC5E,sBAAsB,CAAC,MAAM,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC;gBAChH,CAAC,CAAC,CAAC;gBAEH,OAAO,sBAAsB,CAAC,QAAQ,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC,CAAC;YAC7E,CAAC,CAAC;YAEF,IAAI,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE;gBACpC,iBAAiB,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oBAClD,IAAI,OAAO,KAAK,IAAI,QAAQ,IAAI,OAAO,KAAK,IAAI,QAAQ,EAAE;wBACxD,OAAO,KAAK,CAAC;qBACd;yBAAM;wBACL,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC;qBAC7B;gBACH,CAAC,CAAC,CAAC;aACJ;iBAAM,IAAI,OAAO,iBAAiB,KAAK,QAAQ,IAAI,iBAAiB,KAAK,IAAI,EAAE;gBAC9E,iBAAiB,GAAG,aAAa,CAAC,iBAAiB,CAAC,CAAC;aACtD;YAED,4CAA4C;YAC5C,MAAM,CAAC,WAAW,CAAC,GAAG,iBAAiB,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,IAAI,MAAM,YAAY,KAAK,EAAE;YAC3B,OAAO,MAAM,CAAC;SACf;QAED,IAAI,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAE3C,mBAAmB;QACnB,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAC1C,OAAO,CACL,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,EAAE;oBACpC,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBACrC,OAAO,QAAQ,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC;gBAClC,CAAC,CAAC,KAAK,CAAC,CAAC,CACV,CAAC;YACJ,CAAC,CAAC,CAAC;SACJ;QAED,mBAAmB;QACnB,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;SAClF;QAED,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClC,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO;QAC/B,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;QAChC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACtB,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QACzD,KAAK,CAAC,4BAA4B,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QACpE,OAAO,KAAK,CAAC;IACf,CAAC;CACF","sourcesContent":["import lodash from 'lodash';\nimport { ModelCtor } from 'sequelize';\nimport { Model } from './model';\nimport { AssociationKeysToBeUpdate, BlackList, WhiteList } from './repository';\n\ntype UpdateValueItem = string | number | UpdateValues;\n\ntype UpdateValues = {\n  [key: string]: UpdateValueItem | Array<UpdateValueItem>;\n};\n\ntype UpdateAction = 'create' | 'update';\nexport class UpdateGuard {\n  model: ModelCtor<any>;\n  action: UpdateAction;\n  private associationKeysToBeUpdate: AssociationKeysToBeUpdate;\n  private blackList: BlackList;\n  private whiteList: WhiteList;\n\n  setAction(action: UpdateAction) {\n    this.action = action;\n  }\n\n  setModel(model: ModelCtor<any>) {\n    this.model = model;\n  }\n\n  setAssociationKeysToBeUpdate(associationKeysToBeUpdate: AssociationKeysToBeUpdate) {\n    if (this.action == 'create') {\n      this.associationKeysToBeUpdate = Object.keys(this.model.associations);\n    } else {\n      this.associationKeysToBeUpdate = associationKeysToBeUpdate;\n    }\n  }\n\n  setWhiteList(whiteList: WhiteList) {\n    this.whiteList = whiteList;\n  }\n\n  setBlackList(blackList: BlackList) {\n    this.blackList = blackList;\n  }\n\n  /**\n   * Sanitize values by whitelist blacklist\n   * @param values\n   */\n  sanitize(values: UpdateValues) {\n    values = lodash.clone(values);\n\n    if (!this.model) {\n      throw new Error('please set model first');\n    }\n\n    const associations = this.model.associations;\n    const associationsValues = lodash.pick(values, Object.keys(associations));\n\n    // build params of association update guard\n    const listOfAssociation = (list, association) => {\n      if (list) {\n        list = list\n          .filter((whiteListKey) => whiteListKey.startsWith(`${association}.`))\n          .map((whiteListKey) => whiteListKey.replace(`${association}.`, ''));\n\n        if (list.length == 0) {\n          return undefined;\n        }\n\n        return list;\n      }\n\n      return undefined;\n    };\n\n    // sanitize association values\n    Object.keys(associationsValues).forEach((association) => {\n      let associationValues = associationsValues[association];\n\n      const filterAssociationToBeUpdate = (value) => {\n        const associationKeysToBeUpdate = this.associationKeysToBeUpdate || [];\n\n        if (associationKeysToBeUpdate.includes(association)) {\n          return value;\n        }\n\n        const associationObj = associations[association];\n\n        const associationKeyName =\n          associationObj.associationType == 'BelongsTo' || associationObj.associationType == 'HasOne'\n            ? (<any>associationObj).targetKey\n            : associationObj.target.primaryKeyAttribute;\n\n        if (value[associationKeyName]) {\n          return lodash.pick(value, [associationKeyName, ...Object.keys(associationObj.target.associations)]);\n        }\n\n        return value;\n      };\n\n      const sanitizeValue = (value) => {\n        const associationUpdateGuard = new UpdateGuard();\n        associationUpdateGuard.setModel(associations[association].target);\n\n        ['whiteList', 'blackList', 'associationKeysToBeUpdate'].forEach((optionKey) => {\n          associationUpdateGuard[`set${lodash.upperFirst(optionKey)}`](listOfAssociation(this[optionKey], association));\n        });\n\n        return associationUpdateGuard.sanitize(filterAssociationToBeUpdate(value));\n      };\n\n      if (Array.isArray(associationValues)) {\n        associationValues = associationValues.map((value) => {\n          if (typeof value == 'string' || typeof value == 'number') {\n            return value;\n          } else {\n            return sanitizeValue(value);\n          }\n        });\n      } else if (typeof associationValues === 'object' && associationValues !== null) {\n        associationValues = sanitizeValue(associationValues);\n      }\n\n      // set association values to sanitized value\n      values[association] = associationValues;\n    });\n\n    if (values instanceof Model) {\n      return values;\n    }\n\n    let valuesKeys = Object.keys(values || {});\n\n    // handle whitelist\n    if (this.whiteList) {\n      valuesKeys = valuesKeys.filter((valueKey) => {\n        return (\n          this.whiteList.findIndex((whiteKey) => {\n            const keyPaths = whiteKey.split('.');\n            return keyPaths[0] === valueKey;\n          }) !== -1\n        );\n      });\n    }\n\n    // handle blacklist\n    if (this.blackList) {\n      valuesKeys = valuesKeys.filter((valueKey) => !this.blackList.includes(valueKey));\n    }\n\n    const result = valuesKeys.reduce((obj, key) => {\n      lodash.set(obj, key, values[key]);\n      return obj;\n    }, {});\n\n    return result;\n  }\n\n  static fromOptions(model, options) {\n    const guard = new UpdateGuard();\n    guard.setModel(model);\n    guard.setWhiteList(options.whitelist);\n    guard.setBlackList(options.blacklist);\n    guard.setAction(lodash.get(options, 'action', 'update'));\n    guard.setAssociationKeysToBeUpdate(options.updateAssociationValues);\n    return guard;\n  }\n}\n"]}

package/lib/collection-importer.d.ts

@@ -0,0 +1,7 @@
+export declare type ImportFileExtension = 'js' | 'ts' | 'json';
+export declare class ImporterReader {
+    directory: string;
+    extensions: Set<string>;
+    constructor(directory: string, extensions?: ImportFileExtension[]);
+    read(): Promise<any[]>;
+}