proto.io 0.0.171 → 0.0.173

package/dist/index.mjs

@@ -3,8 +3,8 @@ import { Server } from '@o2ter/server-js';
 import { Q as QueryValidator, r as resolveColumn, a as resolveDataType, g as generateId } from './internals/random-BCpwYpyw.mjs';
 import { P as PVK } from './internals/private-BUpLAMZi.mjs';
 import { prototypes, asyncStream, asyncIterableToArray, isBinaryData, base64ToBuffer } from '@o2ter/utils-js';
-import { T as TQuery, M as MASTER_USER_HEADER_NAME, a as MASTER_PASS_HEADER_NAME, A as AUTH_COOKIE_KEY, b as TUser, P as ProtoType, s as serialize, d as deserialize, U as UPLOAD_TOKEN_HEADER_NAME } from './internals/index-D0hHgn2P.mjs';
-export { c as ProtoClient, e as classExtends, j as isFile, f as isObject, i as isQuery, h as isRole, g as isUser } from './internals/index-D0hHgn2P.mjs';
+import { T as TQuery, M as MASTER_USER_HEADER_NAME, a as MASTER_PASS_HEADER_NAME, A as AUTH_COOKIE_KEY, b as TUser, P as ProtoType, s as serialize, d as deserialize, U as UPLOAD_TOKEN_HEADER_NAME } from './internals/index-BZNPlw1L.mjs';
+export { c as ProtoClient, e as classExtends, j as isFile, f as isObject, i as isQuery, h as isRole, g as isUser } from './internals/index-BZNPlw1L.mjs';
 import { i as isPointer, a as isRelation, d as decodeUpdateOp, T as TObject, b as isShape, c as defaultObjectKeyTypes, e as isPrimitive } from './internals/index-BYbMU-Ao.mjs';
 import jwt from 'jsonwebtoken';
 import { Blob } from 'node:buffer';
@@ -183,6 +183,13 @@ const dispatcher = (proto, options, disableSecurity) => {
                 throw Error('No permission');
             return proto.storage.atomic((storage) => storage.updateOne(_validator.decodeQuery(normalize(query), 'update'), normalize(_validator.validateFields(query.className, update, 'update', QueryValidator.patterns.path))));
         },
+        async updateMany(query, update) {
+            QueryValidator.recursiveCheck(query, update);
+            const _validator = await validator();
+            if (!_validator.validateCLPs(query.className, 'update'))
+                throw Error('No permission');
+            return proto.storage.atomic((storage) => storage.updateMany(_validator.decodeQuery(normalize(query), 'update'), normalize(_validator.validateFields(query.className, update, 'update', QueryValidator.patterns.path))));
+        },
         async upsertOne(query, update, setOnInsert) {
             QueryValidator.recursiveCheck(query, update, setOnInsert);
             const _validator = await validator();
@@ -202,6 +209,25 @@ const dispatcher = (proto, options, disableSecurity) => {
                 }
             }
         },
+        async upsertMany(query, update, setOnInsert) {
+            QueryValidator.recursiveCheck(query, update, setOnInsert);
+            const _validator = await validator();
+            if (!_validator.validateCLPs(query.className, 'create', 'update'))
+                throw Error('No permission');
+            const _query = _validator.decodeQuery(normalize(query), 'update');
+            const _update = normalize(_validator.validateFields(query.className, update, 'update', QueryValidator.patterns.path));
+            const _setOnInsert = normalize(_validator.validateFields(query.className, setOnInsert, 'create', QueryValidator.patterns.name));
+            while (true) {
+                try {
+                    return await proto.storage.atomic((storage) => storage.upsertMany(_query, _update, _setOnInsert), { lockTable: query.className, retry: true });
+                }
+                catch (e) {
+                    if (proto.storage.isDuplicateIdError(e))
+                        continue;
+                    throw e;
+                }
+            }
+        },
         async deleteOne(query) {
             QueryValidator.recursiveCheck(query);
             const _validator = await validator();
@@ -440,6 +466,33 @@ class _ProtoQuery extends TQuery {
         }
         return result;
     }
+    async updateMany(update, options) {
+        const context = options?.context ?? {};
+        const silent = _.castArray(options?.silent ?? []);
+        const beforeSave = _.includes(silent, 'beforeSave') ? null : this._proto[PVK].triggers?.beforeSave?.[this.className];
+        const afterSave = _.includes(silent, 'afterSave') ? null : this._proto[PVK].triggers?.afterSave?.[this.className];
+        if (_.isFunction(beforeSave) || _.isFunction(afterSave)) {
+            const objects = this._objectMethods(await asyncIterableToArray(this._dispatcher(options).find(this._queryOptions)));
+            if (_.isEmpty(objects))
+                return 0;
+            if (_.isFunction(beforeSave)) {
+                await Promise.all(_.map(objects, object => beforeSave(proxy(Object.setPrototypeOf({ object, context }, options?.session ?? this._proto)))));
+            }
+            if (!_.isFunction(afterSave)) {
+                return this._dispatcher(options).updateMany({
+                    ...this._queryOptions,
+                    filter: { _id: { $in: _.map(objects, x => x.objectId) } },
+                }, update);
+            }
+            const updated = await Promise.all(_.map(objects, x => this._dispatcher(options).updateOne({
+                ...this._queryOptions,
+                filter: { _id: { $eq: x.objectId } },
+            }, update)));
+            await Promise.all(_.map(updated, object => afterSave(proxy(Object.setPrototypeOf({ object, context }, options?.session ?? this._proto)))));
+            return updated.length;
+        }
+        return this._dispatcher(options).updateMany(this._queryOptions, update);
+    }
     async upsertOne(update, setOnInsert, options) {
         const context = options?.context ?? {};
         const silent = _.castArray(options?.silent ?? []);
@@ -478,6 +531,47 @@ class _ProtoQuery extends TQuery {
         }
         return result;
     }
+    async upsertMany(update, setOnInsert, options) {
+        const context = options?.context ?? {};
+        const silent = _.castArray(options?.silent ?? []);
+        const beforeSave = _.includes(silent, 'beforeSave') ? null : this._proto[PVK].triggers?.beforeSave?.[this.className];
+        const afterSave = _.includes(silent, 'afterSave') ? null : this._proto[PVK].triggers?.afterSave?.[this.className];
+        if (_.isFunction(beforeSave) || _.isFunction(afterSave)) {
+            const objects = this._objectMethods(await asyncIterableToArray(this._dispatcher(options).find(this._queryOptions)));
+            if (!_.isEmpty(objects) && _.isFunction(beforeSave)) {
+                await Promise.all(_.map(objects, object => beforeSave(proxy(Object.setPrototypeOf({ object, context }, options?.session ?? this._proto)))));
+            }
+            if (_.isEmpty(objects)) {
+                const result = await this._dispatcher(options).insert({
+                    className: this.className,
+                    includes: this[PVK].options.includes,
+                    matches: this[PVK].options.matches,
+                }, setOnInsert);
+                if (!result)
+                    throw Error('Unable to insert document');
+                if (_.isFunction(afterSave)) {
+                    await afterSave(proxy(Object.setPrototypeOf({ object: result, context }, options?.session ?? this._proto)));
+                }
+                return { updated: 0, inserted: 1 };
+            }
+            if (!_.isFunction(afterSave)) {
+                return {
+                    inserted: 0,
+                    updated: await this._dispatcher(options).updateMany({
+                        ...this._queryOptions,
+                        filter: { _id: { $in: _.map(objects, x => x.objectId) } },
+                    }, update),
+                };
+            }
+            const updated = await Promise.all(_.map(objects, x => this._dispatcher(options).updateOne({
+                ...this._queryOptions,
+                filter: { _id: { $eq: x.objectId } },
+            }, update)));
+            await Promise.all(_.map(updated, object => afterSave(proxy(Object.setPrototypeOf({ object, context }, options?.session ?? this._proto)))));
+            return { updated: updated.length, inserted: 0 };
+        }
+        return this._dispatcher(options).upsertMany(this._queryOptions, update, setOnInsert);
+    }
     async deleteOne(options) {
         const context = options?.context ?? {};
         const silent = _.castArray(options?.silent ?? []);
@@ -848,7 +942,7 @@ class ProtoInternal {
         validateSchemaName(options.schema);
         const schema = mergeSchema(defaultSchema, options.fileStorage.schema, options.schema);
         validateSchema(schema);
-        if (!_.every(options.roleInheritKeys, k => {
+        if (!_.every(options.roleResolver?.inheritKeys, k => {
             const type = resolveDataType(schema, 'Role', k);
             return type && isRelation(type) && _.includes(['User', 'Role'], type.target);
         })) {
@@ -1354,7 +1448,7 @@ class ProtoService extends ProtoType {
     constructor(options) {
         super();
         this[PVK] = new ProtoInternal({
-            roleInheritKeys: [],
+            roleResolver: {},
             objectIdSize: 10,
             maxFetchLimit: 1000,
             maxUploadSize: 20 * 1024 * 1024,
@@ -1441,30 +1535,36 @@ class ProtoService extends ProtoType {
         return _.assign(payload, _.isFunction(attrs) ? attrs(payload) : attrs);
     }
     async userRoles(user) {
-        const roleInheritKeys = this[PVK].options.roleInheritKeys;
-        const schema = this.schema;
-        const userKeys = _.filter(roleInheritKeys, k => {
-            const type = resolveDataType(schema, 'Role', k);
-            return !!type && isRelation(type) && type.target === 'User';
-        });
-        const roleKeys = _.filter(roleInheritKeys, k => {
-            const type = resolveDataType(schema, 'Role', k);
-            return !!type && isRelation(type) && type.target === 'Role';
-        });
-        let queue = await this.Query('Role')
-            .or(_.map(_.uniq(['users', ...userKeys]), k => q => q.isIntersect(k, [user])))
-            .includes('name')
-            .find({ master: true });
-        let roles = queue;
-        while (!_.isEmpty(queue)) {
-            queue = await this.Query('Role')
-                .or(_.map(_.uniq(['roles', ...roleKeys]), k => q => q.isIntersect(k, queue)))
-                .notContainsIn('_id', _.compact(_.map(roles, x => x.objectId)))
+        const self = this;
+        const { inheritKeys, resolver } = self[PVK].options.roleResolver;
+        const defaultResolver = async () => {
+            const schema = self.schema;
+            const userKeys = _.filter(inheritKeys, k => {
+                const type = resolveDataType(schema, 'Role', k);
+                return !!type && isRelation(type) && type.target === 'User';
+            });
+            const roleKeys = _.filter(inheritKeys, k => {
+                const type = resolveDataType(schema, 'Role', k);
+                return !!type && isRelation(type) && type.target === 'Role';
+            });
+            let queue = await self.Query('Role')
+                .or(_.map(_.uniq(['users', ...userKeys]), k => q => q.isIntersect(k, [user])))
                 .includes('name')
                 .find({ master: true });
-            roles = _.uniqBy([...roles, ...queue], x => x.objectId);
-        }
-        return roles;
+            let roles = queue;
+            while (!_.isEmpty(queue)) {
+                queue = await self.Query('Role')
+                    .or(_.map(_.uniq(['roles', ...roleKeys]), k => q => q.isIntersect(k, queue)))
+                    .notContainsIn('_id', _.compact(_.map(roles, x => x.objectId)))
+                    .includes('name')
+                    .find({ master: true });
+                roles = _.uniqBy([...roles, ...queue], x => x.objectId);
+            }
+            return roles;
+        };
+        if (resolver)
+            return resolver(user, defaultResolver);
+        return defaultResolver();
     }
     async becomeUser(req, user, options) {
         if (!user.objectId)
@@ -1769,7 +1869,9 @@ var classesRoute = (router, proto) => {
             case 'insert': return query.insert(attributes, opts);
             case 'insertMany': return query.insertMany(attributes, opts);
             case 'updateOne': return query.updateOne(update, opts);
+            case 'updateMany': return query.updateMany(update, opts);
             case 'upsertOne': return query.upsertOne(update, setOnInsert, opts);
+            case 'upsertMany': return query.upsertMany(update, setOnInsert, opts);
             case 'deleteOne': return query.deleteOne(opts);
             case 'deleteMany': return query.deleteMany(opts);
             default: throw Error('Invalid operation');
@@ -1873,6 +1975,16 @@ var classesRoute = (router, proto) => {
         const payload = proto.connect(req);
         await response(res, async () => payload.refs(payload.Object(name, id), { master: payload.isMaster }));
     });
+    router.patch('/classes/:name', Server.text({ type: '*/*' }), async (req, res) => {
+        res.setHeader('Cache-Control', ['no-cache', 'no-store']);
+        const { name } = req.params;
+        const classes = proto.classes();
+        if (!_.includes(classes, name))
+            return void res.sendStatus(404);
+        const payload = proto.connect(req);
+        const update = _.mapValues(deserialize(req.body), v => ({ $set: v }));
+        await response(res, () => createQuery(payload, req, false).updateMany(update, { master: payload.isMaster }));
+    });
     router.patch('/classes/:name/:id', Server.text({ type: '*/*' }), async (req, res) => {
         res.setHeader('Cache-Control', ['no-cache', 'no-store']);
         const { name, id } = req.params;