@casual-simulation/aux-records 3.2.13 → 3.2.14-alpha.7890390188

package/MemoryStore.js

@@ -18,10 +18,9 @@ var __rest = (this && this.__rest) || function (s, e) {
         }
     return t;
 };
-import { sortBy } from 'lodash';
+import { cloneDeep, orderBy, sortBy } from 'lodash';
 import { v4 as uuid } from 'uuid';
-import { getExpireTime, } from './PolicyStore';
-import { DEFAULT_ANY_RESOURCE_POLICY_DOCUMENT, DEFAULT_PUBLIC_READ_POLICY_DOCUMENT, DEFAULT_PUBLIC_WRITE_POLICY_DOCUMENT, PUBLIC_READ_MARKER, PUBLIC_WRITE_MARKER, } from '@casual-simulation/aux-common';
+import { getExpireTime, getSubjectUserId, } from './PolicyStore';
 import { DateTime } from 'luxon';
 export class MemoryStore {
     get users() {
@@ -109,6 +108,10 @@ export class MemoryStore {
         this._moderationConfiguration = null;
         this._recordNotifications = [];
         this._comIdRequests = [];
+        this._resourcePermissionAssignments = [];
+        this._markerPermissionAssignments = [];
+        // TODO: Support global permissions
+        // private _globalPermissionAssignments: GlobalPermissionAssignment[] = [];
         this.maxAllowedInstSize = Infinity;
         this._subscriptionConfiguration = config.subscriptions;
         this._privoConfiguration = (_a = config.privo) !== null && _a !== void 0 ? _a : null;
@@ -117,6 +120,50 @@ export class MemoryStore {
         this.roles = {};
         this.roleAssignments = {};
     }
+    /**
+     * Constructs a deep clone of this memory store.
+     * Effectively copies all the data in the store into a new one.
+     */
+    clone() {
+        const newStore = new MemoryStore({
+            subscriptions: cloneDeep(this._subscriptionConfiguration),
+            privo: cloneDeep(this._privoConfiguration),
+            moderation: cloneDeep(this._moderationConfiguration),
+        });
+        newStore._users = cloneDeep(this._users);
+        newStore._loginRequests = cloneDeep(this._loginRequests);
+        newStore._oidLoginRequests = cloneDeep(this._oidLoginRequests);
+        newStore._sessions = cloneDeep(this._sessions);
+        newStore._subscriptions = cloneDeep(this._subscriptions);
+        newStore._periods = cloneDeep(this._periods);
+        newStore._invoices = cloneDeep(this._invoices);
+        newStore._records = cloneDeep(this._records);
+        newStore._recordKeys = cloneDeep(this._recordKeys);
+        newStore._studios = cloneDeep(this._studios);
+        newStore._studioAssignments = cloneDeep(this._studioAssignments);
+        newStore._aiChatMetrics = cloneDeep(this._aiChatMetrics);
+        newStore._aiImageMetrics = cloneDeep(this._aiImageMetrics);
+        newStore._aiSkyboxMetrics = cloneDeep(this._aiSkyboxMetrics);
+        newStore._dataBuckets = cloneDeep(this._dataBuckets);
+        newStore._eventBuckets = cloneDeep(this._eventBuckets);
+        newStore._files = cloneDeep(this._files);
+        newStore._fileUploadUrl = cloneDeep(this._fileUploadUrl);
+        newStore._emailRules = cloneDeep(this._emailRules);
+        newStore._smsRules = cloneDeep(this._smsRules);
+        newStore._userInstReports = cloneDeep(this._userInstReports);
+        newStore._instRecords = cloneDeep(this._instRecords);
+        newStore._subscriptionConfiguration = cloneDeep(this._subscriptionConfiguration);
+        newStore._privoConfiguration = cloneDeep(this._privoConfiguration);
+        newStore._moderationConfiguration = cloneDeep(this._moderationConfiguration);
+        newStore._recordNotifications = cloneDeep(this._recordNotifications);
+        newStore._comIdRequests = cloneDeep(this._comIdRequests);
+        newStore._resourcePermissionAssignments = cloneDeep(this._resourcePermissionAssignments);
+        newStore._markerPermissionAssignments = cloneDeep(this._markerPermissionAssignments);
+        newStore.maxAllowedInstSize = this.maxAllowedInstSize;
+        newStore.roles = cloneDeep(this.roles);
+        newStore.roleAssignments = cloneDeep(this.roleAssignments);
+        return newStore;
+    }
     saveComIdRequest(request) {
         return __awaiter(this, void 0, void 0, function* () {
             this._comIdRequests.push(request);
@@ -398,6 +445,252 @@ export class MemoryStore {
             });
         });
     }
+    getUserPrivacyFeatures(userId) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield this._getUserPrivacyFeatures(userId);
+        });
+    }
+    getRecordOwnerPrivacyFeatures(recordName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const record = yield this.getRecordByName(recordName);
+            if (!record || !record.ownerId) {
+                return null;
+            }
+            return yield this._getUserPrivacyFeatures(record.ownerId);
+        });
+    }
+    getPermissionForSubjectAndResource(subjectType, subjectId, recordName, resourceKind, resourceId, action, currentTimeMs) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const existingRoles = subjectType === 'user'
+                ? yield this.listRolesForUser(recordName, subjectId)
+                : subjectType === 'inst'
+                    ? yield this.listRolesForInst(recordName, subjectId)
+                    : [];
+            const roles = existingRoles.map((r) => r.role);
+            const assignment = this._resourcePermissionAssignments.find((p) => p.recordName === recordName &&
+                ((p.subjectType === 'role' &&
+                    roles.indexOf(p.subjectId) >= 0) ||
+                    (p.subjectType === subjectType &&
+                        p.subjectId === subjectId)) &&
+                p.resourceKind === resourceKind &&
+                p.resourceId === resourceId &&
+                (p.action === null || p.action === action) &&
+                (!p.expireTimeMs || p.expireTimeMs > currentTimeMs));
+            return {
+                success: true,
+                permissionAssignment: assignment,
+            };
+        });
+    }
+    getPermissionForSubjectAndMarkers(subjectType, subjectId, recordName, resourceKind, markers, action, currentTimeMs) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const existingRoles = subjectType === 'user'
+                ? yield this.listRolesForUser(recordName, subjectId)
+                : subjectType === 'inst'
+                    ? yield this.listRolesForInst(recordName, subjectId)
+                    : [];
+            const roles = existingRoles.map((r) => r.role);
+            const assignment = this._markerPermissionAssignments.find((p) => p.recordName === recordName &&
+                markers.indexOf(p.marker) >= 0 &&
+                ((p.subjectType === 'role' &&
+                    roles.indexOf(p.subjectId) >= 0) ||
+                    (p.subjectType === subjectType &&
+                        p.subjectId === subjectId)) &&
+                p.resourceKind === resourceKind &&
+                (p.action === null || p.action === action) &&
+                (!p.expireTimeMs || p.expireTimeMs > currentTimeMs));
+            return {
+                success: true,
+                permissionAssignment: assignment,
+            };
+        });
+    }
+    // TODO: Support global permissions
+    // async assignGlobalPermissionToSubject(
+    //     subjectType: SubjectType,
+    //     subjectId: string,
+    //     resourceKind: ResourceKinds,
+    //     action: ActionKinds,
+    //     options: PermissionOptions,
+    //     expireTimeMs: number
+    // ): Promise<AssignGlobalPermissionToSubjectResult> {
+    //     const userId = getSubjectUserId(subjectType, subjectId);
+    //     const assignment: GlobalPermissionAssignment = {
+    //         id: uuid(),
+    //         userId,
+    //         subjectType,
+    //         subjectId,
+    //         resourceKind,
+    //         action,
+    //         options,
+    //         expireTimeMs
+    //     };
+    //     this._globalPermissionAssignments.push(assignment);
+    //     return {
+    //         success: true,
+    //         permissionAssignment: assignment,
+    //     };
+    // }
+    assignPermissionToSubjectAndResource(recordName, subjectType, subjectId, resourceKind, resourceId, action, options, expireTimeMs) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const assignmentIndex = this._resourcePermissionAssignments.findIndex((a) => a.recordName === recordName &&
+                a.subjectType === subjectType &&
+                a.subjectId === subjectId &&
+                a.resourceKind === resourceKind &&
+                a.resourceId === resourceId &&
+                a.action === action);
+            if (assignmentIndex >= 0) {
+                const assignment = this._resourcePermissionAssignments[assignmentIndex];
+                this._resourcePermissionAssignments[assignmentIndex] = Object.assign(Object.assign({}, assignment), { options,
+                    expireTimeMs });
+                return {
+                    success: true,
+                    permissionAssignment: assignment,
+                };
+            }
+            const userId = getSubjectUserId(subjectType, subjectId);
+            const assignment = {
+                id: uuid(),
+                recordName,
+                userId,
+                subjectType,
+                subjectId,
+                resourceKind,
+                resourceId,
+                action,
+                options,
+                expireTimeMs,
+            };
+            this._resourcePermissionAssignments.push(assignment);
+            return {
+                success: true,
+                permissionAssignment: assignment,
+            };
+        });
+    }
+    assignPermissionToSubjectAndMarker(recordName, subjectType, subjectId, resourceKind, marker, action, options, expireTimeMs) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const assignmentIndex = this._markerPermissionAssignments.findIndex((a) => a.recordName === recordName &&
+                a.subjectType === subjectType &&
+                a.subjectId === subjectId &&
+                a.marker === marker &&
+                a.resourceKind === resourceKind &&
+                a.action === action);
+            if (assignmentIndex >= 0) {
+                const assignment = this._markerPermissionAssignments[assignmentIndex];
+                this._markerPermissionAssignments[assignmentIndex] = Object.assign(Object.assign({}, assignment), { options,
+                    expireTimeMs });
+                return {
+                    success: true,
+                    permissionAssignment: assignment,
+                };
+            }
+            const userId = getSubjectUserId(subjectType, subjectId);
+            const assignment = {
+                id: uuid(),
+                recordName,
+                userId,
+                subjectType,
+                subjectId,
+                resourceKind,
+                marker,
+                action,
+                options,
+                expireTimeMs,
+            };
+            this._markerPermissionAssignments.push(assignment);
+            return {
+                success: true,
+                permissionAssignment: assignment,
+            };
+        });
+    }
+    getMarkerPermissionAssignmentById(id) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            return ((_a = this._markerPermissionAssignments.find((a) => a.id === id)) !== null && _a !== void 0 ? _a : null);
+        });
+    }
+    getResourcePermissionAssignmentById(id) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            return ((_a = this._resourcePermissionAssignments.find((a) => a.id === id)) !== null && _a !== void 0 ? _a : null);
+        });
+    }
+    deleteResourcePermissionAssignment(assigment) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this._resourcePermissionAssignments =
+                this._resourcePermissionAssignments.filter((p) => p.id !== assigment.id);
+            return {
+                success: true,
+            };
+        });
+    }
+    deleteResourcePermissionAssignmentById(id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this._resourcePermissionAssignments =
+                this._resourcePermissionAssignments.filter((p) => p.id !== id);
+            return {
+                success: true,
+            };
+        });
+    }
+    deleteMarkerPermissionAssignment(assigment) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this._markerPermissionAssignments =
+                this._markerPermissionAssignments.filter((p) => p.id !== assigment.id);
+            return {
+                success: true,
+            };
+        });
+    }
+    deleteMarkerPermissionAssignmentById(id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this._markerPermissionAssignments =
+                this._markerPermissionAssignments.filter((p) => p.id !== id);
+            return {
+                success: true,
+            };
+        });
+    }
+    listPermissionsInRecord(recordName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const resourceAssignments = this._resourcePermissionAssignments.filter((p) => p.recordName === recordName);
+            const markerAssignments = this._markerPermissionAssignments.filter((p) => p.recordName === recordName);
+            return {
+                success: true,
+                resourceAssignments,
+                markerAssignments,
+            };
+        });
+    }
+    listPermissionsForResource(recordName, resourceKind, resourceId) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this._resourcePermissionAssignments.filter((p) => p.recordName === recordName &&
+                p.resourceKind === resourceKind &&
+                p.resourceId === resourceId);
+        });
+    }
+    listPermissionsForMarker(recordName, marker) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this._markerPermissionAssignments.filter((p) => p.recordName === recordName && p.marker === marker);
+        });
+    }
+    listPermissionsForSubject(recordName, subjectType, subjectId) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const resourceAssignments = this._resourcePermissionAssignments.filter((p) => p.recordName === recordName &&
+                p.subjectType === subjectType &&
+                p.subjectId === subjectId);
+            const markerAssignments = this._markerPermissionAssignments.filter((p) => p.recordName === recordName &&
+                p.subjectType === subjectType &&
+                p.subjectId === subjectId);
+            return {
+                success: true,
+                resourceAssignments,
+                markerAssignments,
+            };
+        });
+    }
     countRecords(filter) {
         return __awaiter(this, void 0, void 0, function* () {
             let count = 0;
@@ -866,6 +1159,46 @@ export class MemoryStore {
                 success: true,
                 items,
                 totalCount: count,
+                marker: null,
+            };
+        });
+    }
+    listDataByMarker(request) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            const marker = request.marker;
+            let record = this._getDataRecord(request.recordName);
+            let items = [];
+            const address = request.startingAddress;
+            const sortAscending = ((_a = request.sort) !== null && _a !== void 0 ? _a : 'ascending') === 'ascending';
+            let count = 0;
+            for (let [key, item] of record.entries()) {
+                if (item.markers.includes(marker)) {
+                    count += 1;
+                    if (!address ||
+                        (sortAscending && key > address) ||
+                        (!sortAscending && key < address)) {
+                        items.push({
+                            address: key,
+                            data: item.data,
+                            markers: item.markers,
+                        });
+                    }
+                }
+            }
+            if (request.sort) {
+                if (request.sort === 'ascending') {
+                    items = sortBy(items, (i) => i.address);
+                }
+                else if (request.sort === 'descending') {
+                    items = orderBy(items, (i) => i.address, 'desc');
+                }
+            }
+            return {
+                success: true,
+                items,
+                totalCount: count,
+                marker: marker,
             };
         });
     }
@@ -1126,85 +1459,87 @@ export class MemoryStore {
         }
         return record;
     }
-    listUserPolicies(recordName, startingMarker) {
-        var _a;
-        return __awaiter(this, void 0, void 0, function* () {
-            const recordPolicies = (_a = this.policies[recordName]) !== null && _a !== void 0 ? _a : {};
-            const keys = sortBy(Object.keys(recordPolicies));
-            let results = [];
-            let start = !startingMarker;
-            for (let key of keys) {
-                if (start) {
-                    results.push({
-                        marker: key,
-                        document: recordPolicies[key].document,
-                        markers: recordPolicies[key].markers,
-                    });
-                }
-                else if (key === startingMarker || key > startingMarker) {
-                    start = true;
-                }
-            }
-            return {
-                success: true,
-                policies: results,
-                totalCount: results.length,
-            };
-        });
-    }
-    getUserPolicy(recordName, marker) {
-        var _a;
-        return __awaiter(this, void 0, void 0, function* () {
-            const policy = (_a = this.policies[recordName]) === null || _a === void 0 ? void 0 : _a[marker];
-            if (!policy) {
-                return {
-                    success: false,
-                    errorCode: 'policy_not_found',
-                    errorMessage: 'The policy was not found.',
-                };
-            }
-            return {
-                success: true,
-                document: policy.document,
-                markers: policy.markers,
-            };
-        });
-    }
-    updateUserPolicy(recordName, marker, policy) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (!this.policies[recordName]) {
-                this.policies[recordName] = {};
-            }
-            this.policies[recordName][marker] = {
-                document: policy.document,
-                markers: policy.markers,
-            };
-            return {
-                success: true,
-            };
-        });
-    }
-    listPoliciesForMarkerAndUser(recordName, userId, marker) {
-        var _a;
-        return __awaiter(this, void 0, void 0, function* () {
-            const policies = [DEFAULT_ANY_RESOURCE_POLICY_DOCUMENT];
-            if (marker === PUBLIC_READ_MARKER) {
-                policies.push(DEFAULT_PUBLIC_READ_POLICY_DOCUMENT);
-            }
-            else if (marker === PUBLIC_WRITE_MARKER) {
-                policies.push(DEFAULT_PUBLIC_WRITE_POLICY_DOCUMENT);
-            }
-            const policy = (_a = this.policies[recordName]) === null || _a === void 0 ? void 0 : _a[marker];
-            if (policy) {
-                policies.push(policy.document);
-            }
-            return {
-                policies,
-                recordOwnerPrivacyFeatures: yield this._getRecordOwnerPrivacyFeatures(recordName),
-                userPrivacyFeatures: yield this._getUserPrivacyFeatures(userId),
-            };
-        });
-    }
+    // async listUserPolicies(
+    //     recordName: string,
+    //     startingMarker: string
+    // ): Promise<ListUserPoliciesStoreResult> {
+    //     const recordPolicies = this.policies[recordName] ?? {};
+    //     const keys = sortBy(Object.keys(recordPolicies));
+    //     let results: ListedUserPolicy[] = [];
+    //     let start = !startingMarker;
+    //     for (let key of keys) {
+    //         if (start) {
+    //             results.push({
+    //                 marker: key,
+    //                 document: recordPolicies[key].document,
+    //                 markers: recordPolicies[key].markers,
+    //             });
+    //         } else if (key === startingMarker || key > startingMarker) {
+    //             start = true;
+    //         }
+    //     }
+    //     return {
+    //         success: true,
+    //         policies: results,
+    //         totalCount: results.length,
+    //     };
+    // }
+    // async getUserPolicy(
+    //     recordName: string,
+    //     marker: string
+    // ): Promise<GetUserPolicyResult> {
+    //     const policy = this.policies[recordName]?.[marker];
+    //     if (!policy) {
+    //         return {
+    //             success: false,
+    //             errorCode: 'policy_not_found',
+    //             errorMessage: 'The policy was not found.',
+    //         };
+    //     }
+    //     return {
+    //         success: true,
+    //         document: policy.document,
+    //         markers: policy.markers,
+    //     };
+    // }
+    // async updateUserPolicy(
+    //     recordName: string,
+    //     marker: string,
+    //     policy: UserPolicyRecord
+    // ): Promise<UpdateUserPolicyResult> {
+    //     if (!this.policies[recordName]) {
+    //         this.policies[recordName] = {};
+    //     }
+    //     this.policies[recordName][marker] = {
+    //         document: policy.document,
+    //         markers: policy.markers,
+    //     };
+    //     return {
+    //         success: true,
+    //     };
+    // }
+    // async listPoliciesForMarkerAndUser(
+    //     recordName: string,
+    //     userId: string,
+    //     marker: string
+    // ): Promise<ListMarkerPoliciesResult> {
+    //     const policies = [DEFAULT_ANY_RESOURCE_POLICY_DOCUMENT];
+    //     if (marker === PUBLIC_READ_MARKER) {
+    //         policies.push(DEFAULT_PUBLIC_READ_POLICY_DOCUMENT);
+    //     } else if (marker === PUBLIC_WRITE_MARKER) {
+    //         policies.push(DEFAULT_PUBLIC_WRITE_POLICY_DOCUMENT);
+    //     }
+    //     const policy = this.policies[recordName]?.[marker];
+    //     if (policy) {
+    //         policies.push(policy.document);
+    //     }
+    //     return {
+    //         policies,
+    //         recordOwnerPrivacyFeatures:
+    //             await this._getRecordOwnerPrivacyFeatures(recordName),
+    //         userPrivacyFeatures: await this._getUserPrivacyFeatures(userId),
+    //     };
+    // }
     _getRecordOwnerPrivacyFeatures(recordName) {
         return __awaiter(this, void 0, void 0, function* () {
             let record = yield this.getRecordByName(recordName);
@@ -1356,34 +1691,46 @@ export class MemoryStore {
             };
         });
     }
-    updateUserRoles(recordName, userId, update) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (!this.roleAssignments[recordName]) {
-                this.roleAssignments[recordName] = {};
-            }
-            const assignments = update.roles
-                .filter((r) => getExpireTime(r.expireTimeMs) > Date.now())
-                .map((r) => (Object.assign(Object.assign({}, r), { expireTimeMs: r.expireTimeMs === Infinity ? null : r.expireTimeMs })));
-            this.roleAssignments[recordName][userId] = assignments;
-            return {
-                success: true,
-            };
-        });
-    }
-    updateInstRoles(recordName, inst, update) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (!this.roleAssignments[recordName]) {
-                this.roleAssignments[recordName] = {};
-            }
-            const assignments = update.roles
-                .filter((r) => getExpireTime(r.expireTimeMs) > Date.now())
-                .map((r) => (Object.assign(Object.assign({}, r), { expireTimeMs: r.expireTimeMs === Infinity ? null : r.expireTimeMs })));
-            this.roleAssignments[recordName][inst] = assignments;
-            return {
-                success: true,
-            };
-        });
-    }
+    // async updateUserRoles(
+    //     recordName: string,
+    //     userId: string,
+    //     update: UpdateRolesUpdate
+    // ): Promise<UpdateUserRolesResult> {
+    //     if (!this.roleAssignments[recordName]) {
+    //         this.roleAssignments[recordName] = {};
+    //     }
+    //     const assignments = update.roles
+    //         .filter((r) => getExpireTime(r.expireTimeMs) > Date.now())
+    //         .map((r) => ({
+    //             ...r,
+    //             expireTimeMs:
+    //                 r.expireTimeMs === Infinity ? null : r.expireTimeMs,
+    //         }));
+    //     this.roleAssignments[recordName][userId] = assignments;
+    //     return {
+    //         success: true,
+    //     };
+    // }
+    // async updateInstRoles(
+    //     recordName: string,
+    //     inst: string,
+    //     update: UpdateRolesUpdate
+    // ): Promise<UpdateUserRolesResult> {
+    //     if (!this.roleAssignments[recordName]) {
+    //         this.roleAssignments[recordName] = {};
+    //     }
+    //     const assignments = update.roles
+    //         .filter((r) => getExpireTime(r.expireTimeMs) > Date.now())
+    //         .map((r) => ({
+    //             ...r,
+    //             expireTimeMs:
+    //                 r.expireTimeMs === Infinity ? null : r.expireTimeMs,
+    //         }));
+    //     this.roleAssignments[recordName][inst] = assignments;
+    //     return {
+    //         success: true,
+    //     };
+    // }
     _getRolesForEntity(recordName, id) {
         var _a, _b, _c, _d;
         const roles = (_b = (_a = this.roles[recordName]) === null || _a === void 0 ? void 0 : _a[id]) !== null && _b !== void 0 ? _b : new Set();