@librechat/data-schemas 0.0.42 → 0.0.44

package/dist/methods/share.es.js

@@ -0,0 +1,472 @@
+import { nanoid } from 'nanoid';
+import { Constants } from 'librechat-data-provider';
+import logger from '../config/winston.es.js';
+
+class ShareServiceError extends Error {
+    constructor(message, code) {
+        super(message);
+        this.name = 'ShareServiceError';
+        this.code = code;
+    }
+}
+function memoizedAnonymizeId(prefix) {
+    const memo = new Map();
+    return (id) => {
+        if (!memo.has(id)) {
+            memo.set(id, `${prefix}_${nanoid()}`);
+        }
+        return memo.get(id);
+    };
+}
+const anonymizeConvoId = memoizedAnonymizeId('convo');
+const anonymizeAssistantId = memoizedAnonymizeId('a');
+const anonymizeMessageId = (id) => id === Constants.NO_PARENT ? id : memoizedAnonymizeId('msg')(id);
+function anonymizeConvo(conversation) {
+    if (!conversation) {
+        return null;
+    }
+    const newConvo = { ...conversation };
+    if (newConvo.assistant_id) {
+        newConvo.assistant_id = anonymizeAssistantId(newConvo.assistant_id);
+    }
+    return newConvo;
+}
+function anonymizeMessages(messages, newConvoId) {
+    if (!Array.isArray(messages)) {
+        return [];
+    }
+    const idMap = new Map();
+    return messages.map((message) => {
+        var _a, _b;
+        const newMessageId = anonymizeMessageId(message.messageId);
+        idMap.set(message.messageId, newMessageId);
+        const anonymizedAttachments = (_a = message.attachments) === null || _a === void 0 ? void 0 : _a.map((attachment) => {
+            return {
+                ...attachment,
+                messageId: newMessageId,
+                conversationId: newConvoId,
+            };
+        });
+        return {
+            ...message,
+            messageId: newMessageId,
+            parentMessageId: idMap.get(message.parentMessageId || '') ||
+                anonymizeMessageId(message.parentMessageId || ''),
+            conversationId: newConvoId,
+            model: ((_b = message.model) === null || _b === void 0 ? void 0 : _b.startsWith('asst_'))
+                ? anonymizeAssistantId(message.model)
+                : message.model,
+            attachments: anonymizedAttachments,
+        };
+    });
+}
+/**
+ * Filter messages up to and including the target message (branch-specific)
+ * Similar to getMessagesUpToTargetLevel from fork utilities
+ */
+function getMessagesUpToTarget(messages, targetMessageId) {
+    var _a, _b;
+    if (!messages || messages.length === 0) {
+        return [];
+    }
+    // If only one message and it's the target, return it
+    if (messages.length === 1 && ((_a = messages[0]) === null || _a === void 0 ? void 0 : _a.messageId) === targetMessageId) {
+        return messages;
+    }
+    // Create a map of parentMessageId to children messages
+    const parentToChildrenMap = new Map();
+    for (const message of messages) {
+        const parentId = message.parentMessageId || Constants.NO_PARENT;
+        if (!parentToChildrenMap.has(parentId)) {
+            parentToChildrenMap.set(parentId, []);
+        }
+        (_b = parentToChildrenMap.get(parentId)) === null || _b === void 0 ? void 0 : _b.push(message);
+    }
+    // Find the target message
+    const targetMessage = messages.find((msg) => msg.messageId === targetMessageId);
+    if (!targetMessage) {
+        // If target not found, return all messages for backwards compatibility
+        return messages;
+    }
+    const visited = new Set();
+    const rootMessages = parentToChildrenMap.get(Constants.NO_PARENT) || [];
+    let currentLevel = rootMessages.length > 0 ? [...rootMessages] : [targetMessage];
+    const results = new Set(currentLevel);
+    // Check if the target message is at the root level
+    if (currentLevel.some((msg) => msg.messageId === targetMessageId) &&
+        targetMessage.parentMessageId === Constants.NO_PARENT) {
+        return Array.from(results);
+    }
+    // Iterate level by level until the target is found
+    let targetFound = false;
+    while (!targetFound && currentLevel.length > 0) {
+        const nextLevel = [];
+        for (const node of currentLevel) {
+            if (visited.has(node.messageId)) {
+                continue;
+            }
+            visited.add(node.messageId);
+            const children = parentToChildrenMap.get(node.messageId) || [];
+            for (const child of children) {
+                if (visited.has(child.messageId)) {
+                    continue;
+                }
+                nextLevel.push(child);
+                results.add(child);
+                if (child.messageId === targetMessageId) {
+                    targetFound = true;
+                }
+            }
+        }
+        currentLevel = nextLevel;
+    }
+    return Array.from(results);
+}
+/** Factory function that takes mongoose instance and returns the methods */
+function createShareMethods(mongoose) {
+    /**
+     * Get shared messages for a public share link
+     */
+    async function getSharedMessages(shareId) {
+        try {
+            const SharedLink = mongoose.models.SharedLink;
+            const share = (await SharedLink.findOne({ shareId, isPublic: true })
+                .populate({
+                path: 'messages',
+                select: '-_id -__v -user',
+            })
+                .select('-_id -__v -user')
+                .lean());
+            if (!(share === null || share === void 0 ? void 0 : share.conversationId) || !share.isPublic) {
+                return null;
+            }
+            /** Filtered messages based on targetMessageId if present (branch-specific sharing) */
+            let messagesToShare = share.messages;
+            if (share.targetMessageId) {
+                messagesToShare = getMessagesUpToTarget(share.messages, share.targetMessageId);
+            }
+            const newConvoId = anonymizeConvoId(share.conversationId);
+            const result = {
+                shareId: share.shareId || shareId,
+                title: share.title,
+                isPublic: share.isPublic,
+                createdAt: share.createdAt,
+                updatedAt: share.updatedAt,
+                conversationId: newConvoId,
+                messages: anonymizeMessages(messagesToShare, newConvoId),
+            };
+            return result;
+        }
+        catch (error) {
+            logger.error('[getSharedMessages] Error getting share link', {
+                error: error instanceof Error ? error.message : 'Unknown error',
+                shareId,
+            });
+            throw new ShareServiceError('Error getting share link', 'SHARE_FETCH_ERROR');
+        }
+    }
+    /**
+     * Get shared links for a specific user with pagination and search
+     */
+    async function getSharedLinks(user, pageParam, pageSize = 10, isPublic = true, sortBy = 'createdAt', sortDirection = 'desc', search) {
+        var _a;
+        try {
+            const SharedLink = mongoose.models.SharedLink;
+            const Conversation = mongoose.models.Conversation;
+            const query = { user, isPublic };
+            if (pageParam) {
+                if (sortDirection === 'desc') {
+                    query[sortBy] = { $lt: pageParam };
+                }
+                else {
+                    query[sortBy] = { $gt: pageParam };
+                }
+            }
+            if (search && search.trim()) {
+                try {
+                    const searchResults = await Conversation.meiliSearch(search, {
+                        filter: `user = "${user}"`,
+                    });
+                    if (!((_a = searchResults === null || searchResults === void 0 ? void 0 : searchResults.hits) === null || _a === void 0 ? void 0 : _a.length)) {
+                        return {
+                            links: [],
+                            nextCursor: undefined,
+                            hasNextPage: false,
+                        };
+                    }
+                    const conversationIds = searchResults.hits.map((hit) => hit.conversationId);
+                    query['conversationId'] = { $in: conversationIds };
+                }
+                catch (searchError) {
+                    logger.error('[getSharedLinks] Meilisearch error', {
+                        error: searchError instanceof Error ? searchError.message : 'Unknown error',
+                        user,
+                    });
+                    return {
+                        links: [],
+                        nextCursor: undefined,
+                        hasNextPage: false,
+                    };
+                }
+            }
+            const sort = {};
+            sort[sortBy] = sortDirection === 'desc' ? -1 : 1;
+            const sharedLinks = await SharedLink.find(query)
+                .sort(sort)
+                .limit(pageSize + 1)
+                .select('-__v -user')
+                .lean();
+            const hasNextPage = sharedLinks.length > pageSize;
+            const links = sharedLinks.slice(0, pageSize);
+            const nextCursor = hasNextPage
+                ? links[links.length - 1][sortBy]
+                : undefined;
+            return {
+                links: links.map((link) => ({
+                    shareId: link.shareId || '',
+                    title: (link === null || link === void 0 ? void 0 : link.title) || 'Untitled',
+                    isPublic: link.isPublic,
+                    createdAt: link.createdAt || new Date(),
+                    conversationId: link.conversationId,
+                })),
+                nextCursor,
+                hasNextPage,
+            };
+        }
+        catch (error) {
+            logger.error('[getSharedLinks] Error getting shares', {
+                error: error instanceof Error ? error.message : 'Unknown error',
+                user,
+            });
+            throw new ShareServiceError('Error getting shares', 'SHARES_FETCH_ERROR');
+        }
+    }
+    /**
+     * Delete all shared links for a user
+     */
+    async function deleteAllSharedLinks(user) {
+        try {
+            const SharedLink = mongoose.models.SharedLink;
+            const result = await SharedLink.deleteMany({ user });
+            return {
+                message: 'All shared links deleted successfully',
+                deletedCount: result.deletedCount,
+            };
+        }
+        catch (error) {
+            logger.error('[deleteAllSharedLinks] Error deleting shared links', {
+                error: error instanceof Error ? error.message : 'Unknown error',
+                user,
+            });
+            throw new ShareServiceError('Error deleting shared links', 'BULK_DELETE_ERROR');
+        }
+    }
+    /**
+     * Delete shared links by conversation ID
+     */
+    async function deleteConvoSharedLink(user, conversationId) {
+        if (!user || !conversationId) {
+            throw new ShareServiceError('Missing required parameters', 'INVALID_PARAMS');
+        }
+        try {
+            const SharedLink = mongoose.models.SharedLink;
+            const result = await SharedLink.deleteMany({ user, conversationId });
+            return {
+                message: 'Shared links deleted successfully',
+                deletedCount: result.deletedCount,
+            };
+        }
+        catch (error) {
+            logger.error('[deleteConvoSharedLink] Error deleting shared links', {
+                error: error instanceof Error ? error.message : 'Unknown error',
+                user,
+                conversationId,
+            });
+            throw new ShareServiceError('Error deleting shared links', 'SHARE_DELETE_ERROR');
+        }
+    }
+    /**
+     * Create a new shared link for a conversation
+     */
+    async function createSharedLink(user, conversationId, targetMessageId) {
+        if (!user || !conversationId) {
+            throw new ShareServiceError('Missing required parameters', 'INVALID_PARAMS');
+        }
+        try {
+            const Message = mongoose.models.Message;
+            const SharedLink = mongoose.models.SharedLink;
+            const Conversation = mongoose.models.Conversation;
+            const [existingShare, conversationMessages] = await Promise.all([
+                SharedLink.findOne({
+                    conversationId,
+                    user,
+                    isPublic: true,
+                    ...(targetMessageId && { targetMessageId }),
+                })
+                    .select('-_id -__v -user')
+                    .lean(),
+                Message.find({ conversationId, user }).sort({ createdAt: 1 }).lean(),
+            ]);
+            if (existingShare && existingShare.isPublic) {
+                logger.error('[createSharedLink] Share already exists', {
+                    user,
+                    conversationId,
+                    targetMessageId,
+                });
+                throw new ShareServiceError('Share already exists', 'SHARE_EXISTS');
+            }
+            else if (existingShare) {
+                await SharedLink.deleteOne({
+                    conversationId,
+                    user,
+                    ...(targetMessageId && { targetMessageId }),
+                });
+            }
+            const conversation = (await Conversation.findOne({ conversationId, user }).lean());
+            // Check if user owns the conversation
+            if (!conversation) {
+                throw new ShareServiceError('Conversation not found or access denied', 'CONVERSATION_NOT_FOUND');
+            }
+            // Check if there are any messages to share
+            if (!conversationMessages || conversationMessages.length === 0) {
+                throw new ShareServiceError('No messages to share', 'NO_MESSAGES');
+            }
+            const title = conversation.title || 'Untitled';
+            const shareId = nanoid();
+            await SharedLink.create({
+                shareId,
+                conversationId,
+                messages: conversationMessages,
+                title,
+                user,
+                ...(targetMessageId && { targetMessageId }),
+            });
+            return { shareId, conversationId };
+        }
+        catch (error) {
+            if (error instanceof ShareServiceError) {
+                throw error;
+            }
+            logger.error('[createSharedLink] Error creating shared link', {
+                error: error instanceof Error ? error.message : 'Unknown error',
+                user,
+                conversationId,
+                targetMessageId,
+            });
+            throw new ShareServiceError('Error creating shared link', 'SHARE_CREATE_ERROR');
+        }
+    }
+    /**
+     * Get a shared link for a conversation
+     */
+    async function getSharedLink(user, conversationId) {
+        if (!user || !conversationId) {
+            throw new ShareServiceError('Missing required parameters', 'INVALID_PARAMS');
+        }
+        try {
+            const SharedLink = mongoose.models.SharedLink;
+            const share = (await SharedLink.findOne({ conversationId, user, isPublic: true })
+                .select('shareId -_id')
+                .lean());
+            if (!share) {
+                return { shareId: null, success: false };
+            }
+            return { shareId: share.shareId || null, success: true };
+        }
+        catch (error) {
+            logger.error('[getSharedLink] Error getting shared link', {
+                error: error instanceof Error ? error.message : 'Unknown error',
+                user,
+                conversationId,
+            });
+            throw new ShareServiceError('Error getting shared link', 'SHARE_FETCH_ERROR');
+        }
+    }
+    /**
+     * Update a shared link with new messages
+     */
+    async function updateSharedLink(user, shareId) {
+        if (!user || !shareId) {
+            throw new ShareServiceError('Missing required parameters', 'INVALID_PARAMS');
+        }
+        try {
+            const SharedLink = mongoose.models.SharedLink;
+            const Message = mongoose.models.Message;
+            const share = (await SharedLink.findOne({ shareId, user })
+                .select('-_id -__v -user')
+                .lean());
+            if (!share) {
+                throw new ShareServiceError('Share not found', 'SHARE_NOT_FOUND');
+            }
+            const updatedMessages = await Message.find({ conversationId: share.conversationId, user })
+                .sort({ createdAt: 1 })
+                .lean();
+            const newShareId = nanoid();
+            const update = {
+                messages: updatedMessages,
+                user,
+                shareId: newShareId,
+            };
+            const updatedShare = (await SharedLink.findOneAndUpdate({ shareId, user }, update, {
+                new: true,
+                upsert: false,
+                runValidators: true,
+            }).lean());
+            if (!updatedShare) {
+                throw new ShareServiceError('Share update failed', 'SHARE_UPDATE_ERROR');
+            }
+            anonymizeConvo(updatedShare);
+            return { shareId: newShareId, conversationId: updatedShare.conversationId };
+        }
+        catch (error) {
+            logger.error('[updateSharedLink] Error updating shared link', {
+                error: error instanceof Error ? error.message : 'Unknown error',
+                user,
+                shareId,
+            });
+            throw new ShareServiceError(error instanceof ShareServiceError ? error.message : 'Error updating shared link', error instanceof ShareServiceError ? error.code : 'SHARE_UPDATE_ERROR');
+        }
+    }
+    /**
+     * Delete a shared link
+     */
+    async function deleteSharedLink(user, shareId) {
+        if (!user || !shareId) {
+            throw new ShareServiceError('Missing required parameters', 'INVALID_PARAMS');
+        }
+        try {
+            const SharedLink = mongoose.models.SharedLink;
+            const result = await SharedLink.findOneAndDelete({ shareId, user }).lean();
+            if (!result) {
+                return null;
+            }
+            return {
+                success: true,
+                shareId,
+                message: 'Share deleted successfully',
+            };
+        }
+        catch (error) {
+            logger.error('[deleteSharedLink] Error deleting shared link', {
+                error: error instanceof Error ? error.message : 'Unknown error',
+                user,
+                shareId,
+            });
+            throw new ShareServiceError('Error deleting shared link', 'SHARE_DELETE_ERROR');
+        }
+    }
+    // Return all methods
+    return {
+        getSharedLink,
+        getSharedLinks,
+        createSharedLink,
+        updateSharedLink,
+        deleteSharedLink,
+        getSharedMessages,
+        deleteAllSharedLinks,
+        deleteConvoSharedLink,
+    };
+}
+
+export { createShareMethods };
+//# sourceMappingURL=share.es.js.map

package/dist/methods/share.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"share.es.js","sources":["../../src/methods/share.ts"],"sourcesContent":[null],"names":[],"mappings":";;;;AAOA,MAAM,iBAAkB,SAAQ,KAAK,CAAA;IAEnC,WAAA,CAAY,OAAe,EAAE,IAAY,EAAA;QACvC,KAAK,CAAC,OAAO,CAAC;AACd,QAAA,IAAI,CAAC,IAAI,GAAG,mBAAmB;AAC/B,QAAA,IAAI,CAAC,IAAI,GAAG,IAAI;IAClB;AACD;AAED,SAAS,mBAAmB,CAAC,MAAc,EAAA;AACzC,IAAA,MAAM,IAAI,GAAG,IAAI,GAAG,EAAkB;IACtC,OAAO,CAAC,EAAU,KAAI;QACpB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;AACjB,YAAA,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAA,EAAG,MAAM,CAAA,CAAA,EAAI,MAAM,EAAE,CAAA,CAAE,CAAC;QACvC;AACA,QAAA,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,CAAW;AAC/B,IAAA,CAAC;AACH;AAEA,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,OAAO,CAAC;AACrD,MAAM,oBAAoB,GAAG,mBAAmB,CAAC,GAAG,CAAC;AACrD,MAAM,kBAAkB,GAAG,CAAC,EAAU,KACpC,EAAE,KAAK,SAAS,CAAC,SAAS,GAAG,EAAE,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;AAElE,SAAS,cAAc,CAAC,YAA+D,EAAA;IACrF,IAAI,CAAC,YAAY,EAAE;AACjB,QAAA,OAAO,IAAI;IACb;AAEA,IAAA,MAAM,QAAQ,GAAG,EAAE,GAAG,YAAY,EAAE;AACpC,IAAA,IAAI,QAAQ,CAAC,YAAY,EAAE;QACzB,QAAQ,CAAC,YAAY,GAAG,oBAAoB,CAAC,QAAQ,CAAC,YAAY,CAAC;IACrE;AACA,IAAA,OAAO,QAAQ;AACjB;AAEA,SAAS,iBAAiB,CAAC,QAAsB,EAAE,UAAkB,EAAA;IACnE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;AAC5B,QAAA,OAAO,EAAE;IACX;AAEA,IAAA,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB;AACvC,IAAA,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,KAAI;;QAC9B,MAAM,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC;QAC1D,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,YAAY,CAAC;AAQ1C,QAAA,MAAM,qBAAqB,GAAG,CAAA,EAAA,GAAC,OAAO,CAAC,WAAmC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,GAAG,CAC7E,CAAC,UAAU,KAAI;YACb,OAAO;AACL,gBAAA,GAAG,UAAU;AACb,gBAAA,SAAS,EAAE,YAAY;AACvB,gBAAA,cAAc,EAAE,UAAU;aAC3B;AACH,QAAA,CAAC,CACF;QAED,OAAO;AACL,YAAA,GAAG,OAAO;AACV,YAAA,SAAS,EAAE,YAAY;YACvB,eAAe,EACb,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC;AACxC,gBAAA,kBAAkB,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC;AACnD,YAAA,cAAc,EAAE,UAAU;YAC1B,KAAK,EAAE,CAAA,CAAA,EAAA,GAAA,OAAO,CAAC,KAAK,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,UAAU,CAAC,OAAO,CAAC;AACvC,kBAAE,oBAAoB,CAAC,OAAO,CAAC,KAAK;kBAClC,OAAO,CAAC,KAAK;AACjB,YAAA,WAAW,EAAE,qBAAqB;SACrB;AACjB,IAAA,CAAC,CAAC;AACJ;AAEA;;;AAGG;AACH,SAAS,qBAAqB,CAAC,QAAsB,EAAE,eAAuB,EAAA;;IAC5E,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;AACtC,QAAA,OAAO,EAAE;IACX;;AAGA,IAAA,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,CAAA,CAAA,EAAA,GAAA,QAAQ,CAAC,CAAC,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,SAAS,MAAK,eAAe,EAAE;AACvE,QAAA,OAAO,QAAQ;IACjB;;AAGA,IAAA,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAwB;AAC3D,IAAA,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;QAC9B,MAAM,QAAQ,GAAG,OAAO,CAAC,eAAe,IAAI,SAAS,CAAC,SAAS;QAC/D,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE;AACtC,YAAA,mBAAmB,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC;QACvC;QACA,CAAA,EAAA,GAAA,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,IAAI,CAAC,OAAO,CAAC;IAClD;;AAGA,IAAA,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,SAAS,KAAK,eAAe,CAAC;IAC/E,IAAI,CAAC,aAAa,EAAE;;AAElB,QAAA,OAAO,QAAQ;IACjB;AAEA,IAAA,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU;AACjC,IAAA,MAAM,YAAY,GAAG,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE;IACvE,IAAI,YAAY,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC;AAChF,IAAA,MAAM,OAAO,GAAG,IAAI,GAAG,CAAa,YAAY,CAAC;;AAGjD,IAAA,IACE,YAAY,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,SAAS,KAAK,eAAe,CAAC;AAC7D,QAAA,aAAa,CAAC,eAAe,KAAK,SAAS,CAAC,SAAS,EACrD;AACA,QAAA,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;IAC5B;;IAGA,IAAI,WAAW,GAAG,KAAK;IACvB,OAAO,CAAC,WAAW,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;QAC9C,MAAM,SAAS,GAAiB,EAAE;AAClC,QAAA,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE;YAC/B,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE;gBAC/B;YACF;AACA,YAAA,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;AAC3B,YAAA,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE;AAC9D,YAAA,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE;gBAC5B,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE;oBAChC;gBACF;AACA,gBAAA,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC;AACrB,gBAAA,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;AAClB,gBAAA,IAAI,KAAK,CAAC,SAAS,KAAK,eAAe,EAAE;oBACvC,WAAW,GAAG,IAAI;gBACpB;YACF;QACF;QACA,YAAY,GAAG,SAAS;IAC1B;AAEA,IAAA,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;AAC5B;AAEA;AACM,SAAU,kBAAkB,CAAC,QAAmC,EAAA;AACpE;;AAEG;IACH,eAAe,iBAAiB,CAAC,OAAe,EAAA;AAC9C,QAAA,IAAI;AACF,YAAA,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAkC;AACrE,YAAA,MAAM,KAAK,IAAI,MAAM,UAAU,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE;AAChE,iBAAA,QAAQ,CAAC;AACR,gBAAA,IAAI,EAAE,UAAU;AAChB,gBAAA,MAAM,EAAE,iBAAiB;aAC1B;iBACA,MAAM,CAAC,iBAAiB;iBACxB,IAAI,EAAE,CAAwD;AAEjE,YAAA,IAAI,EAAC,KAAK,KAAA,IAAA,IAAL,KAAK,uBAAL,KAAK,CAAE,cAAc,CAAA,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;AAC7C,gBAAA,OAAO,IAAI;YACb;;AAGA,YAAA,IAAI,eAAe,GAAiB,KAAK,CAAC,QAAQ;AAClD,YAAA,IAAI,KAAK,CAAC,eAAe,EAAE;gBACzB,eAAe,GAAG,qBAAqB,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,eAAe,CAAC;YAChF;YAEA,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,CAAC,cAAc,CAAC;AACzD,YAAA,MAAM,MAAM,GAA2B;AACrC,gBAAA,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,OAAO;gBACjC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,SAAS,EAAE,KAAK,CAAC,SAAS;AAC1B,gBAAA,cAAc,EAAE,UAAU;AAC1B,gBAAA,QAAQ,EAAE,iBAAiB,CAAC,eAAe,EAAE,UAAU,CAAC;aACzD;AAED,YAAA,OAAO,MAAM;QACf;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,CAAC,KAAK,CAAC,8CAA8C,EAAE;AAC3D,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,eAAe;gBAC/D,OAAO;AACR,aAAA,CAAC;AACF,YAAA,MAAM,IAAI,iBAAiB,CAAC,0BAA0B,EAAE,mBAAmB,CAAC;QAC9E;IACF;AAEA;;AAEG;AACH,IAAA,eAAe,cAAc,CAC3B,IAAY,EACZ,SAAgB,EAChB,QAAA,GAAmB,EAAE,EACrB,QAAA,GAAoB,IAAI,EACxB,MAAA,GAAiB,WAAW,EAC5B,aAAA,GAAwB,MAAM,EAC9B,MAAe,EAAA;;AAEf,QAAA,IAAI;AACF,YAAA,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAkC;AACrE,YAAA,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,YAAsC;AAC3E,YAAA,MAAM,KAAK,GAA+B,EAAE,IAAI,EAAE,QAAQ,EAAE;YAE5D,IAAI,SAAS,EAAE;AACb,gBAAA,IAAI,aAAa,KAAK,MAAM,EAAE;oBAC5B,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,EAAE,SAAS,EAAE;gBACpC;qBAAO;oBACL,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,EAAE,SAAS,EAAE;gBACpC;YACF;AAEA,YAAA,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE;AAC3B,gBAAA,IAAI;oBACF,MAAM,aAAa,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,MAAM,EAAE;wBAC3D,MAAM,EAAE,CAAA,QAAA,EAAW,IAAI,CAAA,CAAA,CAAG;AAC3B,qBAAA,CAAC;AAEF,oBAAA,IAAI,EAAC,CAAA,EAAA,GAAA,aAAa,aAAb,aAAa,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAb,aAAa,CAAE,IAAI,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAA,EAAA,CAAE,MAAM,CAAA,EAAE;wBAChC,OAAO;AACL,4BAAA,KAAK,EAAE,EAAE;AACT,4BAAA,UAAU,EAAE,SAAS;AACrB,4BAAA,WAAW,EAAE,KAAK;yBACnB;oBACH;AAEA,oBAAA,MAAM,eAAe,GAAG,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,cAAc,CAAC;oBAC3E,KAAK,CAAC,gBAAgB,CAAC,GAAG,EAAE,GAAG,EAAE,eAAe,EAAE;gBACpD;gBAAE,OAAO,WAAW,EAAE;AACpB,oBAAA,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;AACjD,wBAAA,KAAK,EAAE,WAAW,YAAY,KAAK,GAAG,WAAW,CAAC,OAAO,GAAG,eAAe;wBAC3E,IAAI;AACL,qBAAA,CAAC;oBACF,OAAO;AACL,wBAAA,KAAK,EAAE,EAAE;AACT,wBAAA,UAAU,EAAE,SAAS;AACrB,wBAAA,WAAW,EAAE,KAAK;qBACnB;gBACH;YACF;YAEA,MAAM,IAAI,GAA2B,EAAE;AACvC,YAAA,IAAI,CAAC,MAAM,CAAC,GAAG,aAAa,KAAK,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC;YAEhD,MAAM,WAAW,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,KAAK;iBAC5C,IAAI,CAAC,IAAI;AACT,iBAAA,KAAK,CAAC,QAAQ,GAAG,CAAC;iBAClB,MAAM,CAAC,YAAY;AACnB,iBAAA,IAAI,EAAE;AAET,YAAA,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,GAAG,QAAQ;YACjD,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC;YAE5C,MAAM,UAAU,GAAG;kBACd,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAA6B;kBACtD,SAAS;YAEb,OAAO;gBACL,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,MAAM;AAC1B,oBAAA,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;oBAC3B,KAAK,EAAE,CAAA,IAAI,KAAA,IAAA,IAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,KAAI,UAAU;oBAChC,QAAQ,EAAE,IAAI,CAAC,QAAQ;AACvB,oBAAA,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI,IAAI,EAAE;oBACvC,cAAc,EAAE,IAAI,CAAC,cAAc;AACpC,iBAAA,CAAC,CAAC;gBACH,UAAU;gBACV,WAAW;aACZ;QACH;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,CAAC,KAAK,CAAC,uCAAuC,EAAE;AACpD,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,eAAe;gBAC/D,IAAI;AACL,aAAA,CAAC;AACF,YAAA,MAAM,IAAI,iBAAiB,CAAC,sBAAsB,EAAE,oBAAoB,CAAC;QAC3E;IACF;AAEA;;AAEG;IACH,eAAe,oBAAoB,CAAC,IAAY,EAAA;AAC9C,QAAA,IAAI;AACF,YAAA,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAkC;YACrE,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC;YACpD,OAAO;AACL,gBAAA,OAAO,EAAE,uCAAuC;gBAChD,YAAY,EAAE,MAAM,CAAC,YAAY;aAClC;QACH;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,CAAC,KAAK,CAAC,oDAAoD,EAAE;AACjE,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,eAAe;gBAC/D,IAAI;AACL,aAAA,CAAC;AACF,YAAA,MAAM,IAAI,iBAAiB,CAAC,6BAA6B,EAAE,mBAAmB,CAAC;QACjF;IACF;AAEA;;AAEG;AACH,IAAA,eAAe,qBAAqB,CAClC,IAAY,EACZ,cAAsB,EAAA;AAEtB,QAAA,IAAI,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE;AAC5B,YAAA,MAAM,IAAI,iBAAiB,CAAC,6BAA6B,EAAE,gBAAgB,CAAC;QAC9E;AAEA,QAAA,IAAI;AACF,YAAA,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAkC;AACrE,YAAA,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;YACpE,OAAO;AACL,gBAAA,OAAO,EAAE,mCAAmC;gBAC5C,YAAY,EAAE,MAAM,CAAC,YAAY;aAClC;QACH;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,CAAC,KAAK,CAAC,qDAAqD,EAAE;AAClE,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,eAAe;gBAC/D,IAAI;gBACJ,cAAc;AACf,aAAA,CAAC;AACF,YAAA,MAAM,IAAI,iBAAiB,CAAC,6BAA6B,EAAE,oBAAoB,CAAC;QAClF;IACF;AAEA;;AAEG;AACH,IAAA,eAAe,gBAAgB,CAC7B,IAAY,EACZ,cAAsB,EACtB,eAAwB,EAAA;AAExB,QAAA,IAAI,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE;AAC5B,YAAA,MAAM,IAAI,iBAAiB,CAAC,6BAA6B,EAAE,gBAAgB,CAAC;QAC9E;AACA,QAAA,IAAI;AACF,YAAA,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAiC;AACjE,YAAA,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAkC;AACrE,YAAA,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,YAAsC;YAE3E,MAAM,CAAC,aAAa,EAAE,oBAAoB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC9D,UAAU,CAAC,OAAO,CAAC;oBACjB,cAAc;oBACd,IAAI;AACJ,oBAAA,QAAQ,EAAE,IAAI;AACd,oBAAA,IAAI,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;iBAC5C;qBACE,MAAM,CAAC,iBAAiB;AACxB,qBAAA,IAAI,EAAmC;gBAC1C,OAAO,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE;AACrE,aAAA,CAAC;AAEF,YAAA,IAAI,aAAa,IAAI,aAAa,CAAC,QAAQ,EAAE;AAC3C,gBAAA,MAAM,CAAC,KAAK,CAAC,yCAAyC,EAAE;oBACtD,IAAI;oBACJ,cAAc;oBACd,eAAe;AAChB,iBAAA,CAAC;AACF,gBAAA,MAAM,IAAI,iBAAiB,CAAC,sBAAsB,EAAE,cAAc,CAAC;YACrE;iBAAO,IAAI,aAAa,EAAE;gBACxB,MAAM,UAAU,CAAC,SAAS,CAAC;oBACzB,cAAc;oBACd,IAAI;AACJ,oBAAA,IAAI,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;AAC5C,iBAAA,CAAC;YACJ;AAEA,YAAA,MAAM,YAAY,IAAI,MAAM,YAAY,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAEzE;;YAGR,IAAI,CAAC,YAAY,EAAE;AACjB,gBAAA,MAAM,IAAI,iBAAiB,CACzB,yCAAyC,EACzC,wBAAwB,CACzB;YACH;;YAGA,IAAI,CAAC,oBAAoB,IAAI,oBAAoB,CAAC,MAAM,KAAK,CAAC,EAAE;AAC9D,gBAAA,MAAM,IAAI,iBAAiB,CAAC,sBAAsB,EAAE,aAAa,CAAC;YACpE;AAEA,YAAA,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,IAAI,UAAU;AAE9C,YAAA,MAAM,OAAO,GAAG,MAAM,EAAE;YACxB,MAAM,UAAU,CAAC,MAAM,CAAC;gBACtB,OAAO;gBACP,cAAc;AACd,gBAAA,QAAQ,EAAE,oBAAoB;gBAC9B,KAAK;gBACL,IAAI;AACJ,gBAAA,IAAI,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;AAC5C,aAAA,CAAC;AAEF,YAAA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE;QACpC;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,KAAK,YAAY,iBAAiB,EAAE;AACtC,gBAAA,MAAM,KAAK;YACb;AACA,YAAA,MAAM,CAAC,KAAK,CAAC,+CAA+C,EAAE;AAC5D,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,eAAe;gBAC/D,IAAI;gBACJ,cAAc;gBACd,eAAe;AAChB,aAAA,CAAC;AACF,YAAA,MAAM,IAAI,iBAAiB,CAAC,4BAA4B,EAAE,oBAAoB,CAAC;QACjF;IACF;AAEA;;AAEG;AACH,IAAA,eAAe,aAAa,CAC1B,IAAY,EACZ,cAAsB,EAAA;AAEtB,QAAA,IAAI,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE;AAC5B,YAAA,MAAM,IAAI,iBAAiB,CAAC,6BAA6B,EAAE,gBAAgB,CAAC;QAC9E;AAEA,QAAA,IAAI;AACF,YAAA,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAkC;AACrE,YAAA,MAAM,KAAK,IAAI,MAAM,UAAU,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE;iBAC7E,MAAM,CAAC,cAAc;iBACrB,IAAI,EAAE,CAAgC;YAEzC,IAAI,CAAC,KAAK,EAAE;gBACV,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE;YAC1C;AAEA,YAAA,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE;QAC1D;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE;AACxD,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,eAAe;gBAC/D,IAAI;gBACJ,cAAc;AACf,aAAA,CAAC;AACF,YAAA,MAAM,IAAI,iBAAiB,CAAC,2BAA2B,EAAE,mBAAmB,CAAC;QAC/E;IACF;AAEA;;AAEG;AACH,IAAA,eAAe,gBAAgB,CAAC,IAAY,EAAE,OAAe,EAAA;AAC3D,QAAA,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE;AACrB,YAAA,MAAM,IAAI,iBAAiB,CAAC,6BAA6B,EAAE,gBAAgB,CAAC;QAC9E;AAEA,QAAA,IAAI;AACF,YAAA,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAkC;AACrE,YAAA,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAiC;AACjE,YAAA,MAAM,KAAK,IAAI,MAAM,UAAU,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE;iBACtD,MAAM,CAAC,iBAAiB;iBACxB,IAAI,EAAE,CAAyB;YAElC,IAAI,CAAC,KAAK,EAAE;AACV,gBAAA,MAAM,IAAI,iBAAiB,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;YACnE;AAEA,YAAA,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,IAAI,EAAE;AACtF,iBAAA,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;AACrB,iBAAA,IAAI,EAAE;AAET,YAAA,MAAM,UAAU,GAAG,MAAM,EAAE;AAC3B,YAAA,MAAM,MAAM,GAAG;AACb,gBAAA,QAAQ,EAAE,eAAe;gBACzB,IAAI;AACJ,gBAAA,OAAO,EAAE,UAAU;aACpB;AAED,YAAA,MAAM,YAAY,IAAI,MAAM,UAAU,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE;AACjF,gBAAA,GAAG,EAAE,IAAI;AACT,gBAAA,MAAM,EAAE,KAAK;AACb,gBAAA,aAAa,EAAE,IAAI;AACpB,aAAA,CAAC,CAAC,IAAI,EAAE,CAAyB;YAElC,IAAI,CAAC,YAAY,EAAE;AACjB,gBAAA,MAAM,IAAI,iBAAiB,CAAC,qBAAqB,EAAE,oBAAoB,CAAC;YAC1E;YAEA,cAAc,CAAC,YAAY,CAAC;YAE5B,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,YAAY,CAAC,cAAc,EAAE;QAC7E;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,CAAC,KAAK,CAAC,+CAA+C,EAAE;AAC5D,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,eAAe;gBAC/D,IAAI;gBACJ,OAAO;AACR,aAAA,CAAC;AACF,YAAA,MAAM,IAAI,iBAAiB,CACzB,KAAK,YAAY,iBAAiB,GAAG,KAAK,CAAC,OAAO,GAAG,4BAA4B,EACjF,KAAK,YAAY,iBAAiB,GAAG,KAAK,CAAC,IAAI,GAAG,oBAAoB,CACvE;QACH;IACF;AAEA;;AAEG;AACH,IAAA,eAAe,gBAAgB,CAC7B,IAAY,EACZ,OAAe,EAAA;AAEf,QAAA,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE;AACrB,YAAA,MAAM,IAAI,iBAAiB,CAAC,6BAA6B,EAAE,gBAAgB,CAAC;QAC9E;AAEA,QAAA,IAAI;AACF,YAAA,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAkC;AACrE,YAAA,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;YAE1E,IAAI,CAAC,MAAM,EAAE;AACX,gBAAA,OAAO,IAAI;YACb;YAEA,OAAO;AACL,gBAAA,OAAO,EAAE,IAAI;gBACb,OAAO;AACP,gBAAA,OAAO,EAAE,4BAA4B;aACtC;QACH;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,CAAC,KAAK,CAAC,+CAA+C,EAAE;AAC5D,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,eAAe;gBAC/D,IAAI;gBACJ,OAAO;AACR,aAAA,CAAC;AACF,YAAA,MAAM,IAAI,iBAAiB,CAAC,4BAA4B,EAAE,oBAAoB,CAAC;QACjF;IACF;;IAGA,OAAO;QACL,aAAa;QACb,cAAc;QACd,gBAAgB;QAChB,gBAAgB;QAChB,gBAAgB;QAChB,iBAAiB;QACjB,oBAAoB;QACpB,qBAAqB;KACtB;AACH;;;;"}

package/dist/methods/spendTokens.cjs

@@ -0,0 +1,109 @@
+'use strict';
+
+var winston = require('../config/winston.cjs');
+
+function createSpendTokensMethods(_mongoose, transactionMethods) {
+    /**
+     * Creates up to two transactions to record the spending of tokens.
+     */
+    async function spendTokens(txData, tokenUsage) {
+        var _a;
+        const { promptTokens, completionTokens } = tokenUsage;
+        winston.debug(`[spendTokens] conversationId: ${txData.conversationId}${(txData === null || txData === void 0 ? void 0 : txData.context) ? ` | Context: ${txData === null || txData === void 0 ? void 0 : txData.context}` : ''} | Token usage: `, { promptTokens, completionTokens });
+        let prompt, completion;
+        const normalizedPromptTokens = Math.max(promptTokens !== null && promptTokens !== void 0 ? promptTokens : 0, 0);
+        try {
+            if (promptTokens !== undefined) {
+                prompt = await transactionMethods.createTransaction({
+                    ...txData,
+                    tokenType: 'prompt',
+                    rawAmount: promptTokens === 0 ? 0 : -normalizedPromptTokens,
+                    inputTokenCount: normalizedPromptTokens,
+                });
+            }
+            if (completionTokens !== undefined) {
+                completion = await transactionMethods.createTransaction({
+                    ...txData,
+                    tokenType: 'completion',
+                    rawAmount: completionTokens === 0 ? 0 : -Math.max(completionTokens, 0),
+                    inputTokenCount: normalizedPromptTokens,
+                });
+            }
+            if (prompt || completion) {
+                winston.debug('[spendTokens] Transaction data record against balance:', {
+                    user: txData.user,
+                    prompt: prompt === null || prompt === void 0 ? void 0 : prompt.prompt,
+                    promptRate: prompt === null || prompt === void 0 ? void 0 : prompt.rate,
+                    completion: completion === null || completion === void 0 ? void 0 : completion.completion,
+                    completionRate: completion === null || completion === void 0 ? void 0 : completion.rate,
+                    balance: (_a = completion === null || completion === void 0 ? void 0 : completion.balance) !== null && _a !== void 0 ? _a : prompt === null || prompt === void 0 ? void 0 : prompt.balance,
+                });
+            }
+            else {
+                winston.debug('[spendTokens] No transactions incurred against balance');
+            }
+        }
+        catch (err) {
+            winston.error('[spendTokens]', err);
+        }
+    }
+    /**
+     * Creates transactions to record the spending of structured tokens.
+     */
+    async function spendStructuredTokens(txData, tokenUsage) {
+        var _a, _b, _c, _d, _e, _f, _g;
+        const { promptTokens, completionTokens } = tokenUsage;
+        winston.debug(`[spendStructuredTokens] conversationId: ${txData.conversationId}${(txData === null || txData === void 0 ? void 0 : txData.context) ? ` | Context: ${txData === null || txData === void 0 ? void 0 : txData.context}` : ''} | Token usage: `, { promptTokens, completionTokens });
+        let prompt, completion;
+        try {
+            if (promptTokens) {
+                const input = Math.max((_a = promptTokens.input) !== null && _a !== void 0 ? _a : 0, 0);
+                const write = Math.max((_b = promptTokens.write) !== null && _b !== void 0 ? _b : 0, 0);
+                const read = Math.max((_c = promptTokens.read) !== null && _c !== void 0 ? _c : 0, 0);
+                const totalInputTokens = input + write + read;
+                prompt = await transactionMethods.createStructuredTransaction({
+                    ...txData,
+                    tokenType: 'prompt',
+                    inputTokens: -input,
+                    writeTokens: -write,
+                    readTokens: -read,
+                    inputTokenCount: totalInputTokens,
+                });
+            }
+            if (completionTokens) {
+                const totalInputTokens = promptTokens
+                    ? Math.max((_d = promptTokens.input) !== null && _d !== void 0 ? _d : 0, 0) +
+                        Math.max((_e = promptTokens.write) !== null && _e !== void 0 ? _e : 0, 0) +
+                        Math.max((_f = promptTokens.read) !== null && _f !== void 0 ? _f : 0, 0)
+                    : undefined;
+                completion = await transactionMethods.createTransaction({
+                    ...txData,
+                    tokenType: 'completion',
+                    rawAmount: -Math.max(completionTokens, 0),
+                    inputTokenCount: totalInputTokens,
+                });
+            }
+            if (prompt || completion) {
+                winston.debug('[spendStructuredTokens] Transaction data record against balance:', {
+                    user: txData.user,
+                    prompt: prompt === null || prompt === void 0 ? void 0 : prompt.prompt,
+                    promptRate: prompt === null || prompt === void 0 ? void 0 : prompt.rate,
+                    completion: completion === null || completion === void 0 ? void 0 : completion.completion,
+                    completionRate: completion === null || completion === void 0 ? void 0 : completion.rate,
+                    balance: (_g = completion === null || completion === void 0 ? void 0 : completion.balance) !== null && _g !== void 0 ? _g : prompt === null || prompt === void 0 ? void 0 : prompt.balance,
+                });
+            }
+            else {
+                winston.debug('[spendStructuredTokens] No transactions incurred against balance');
+            }
+        }
+        catch (err) {
+            winston.error('[spendStructuredTokens]', err);
+        }
+        return { prompt, completion };
+    }
+    return { spendTokens, spendStructuredTokens };
+}
+
+exports.createSpendTokensMethods = createSpendTokensMethods;
+//# sourceMappingURL=spendTokens.cjs.map

package/dist/methods/spendTokens.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"spendTokens.cjs","sources":["../../src/methods/spendTokens.ts"],"sourcesContent":[null],"names":["logger"],"mappings":";;;;AAeM,SAAU,wBAAwB,CACtC,SAAoC,EACpC,kBAGC,EAAA;AAED;;AAEG;AACH,IAAA,eAAe,WAAW,CACxB,MAAmB,EACnB,UAAgE,EAAA;;AAEhE,QAAA,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,UAAU;AACrD,QAAAA,OAAM,CAAC,KAAK,CACV,iCAAiC,MAAM,CAAC,cAAc,CAAA,EACpD,CAAA,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,MAAA,GAAA,MAAA,GAAN,MAAM,CAAE,OAAO,IAAG,CAAA,YAAA,EAAe,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,MAAA,GAAA,MAAA,GAAN,MAAM,CAAE,OAAO,CAAA,CAAE,GAAG,EACvD,CAAA,gBAAA,CAAkB,EAClB,EAAE,YAAY,EAAE,gBAAgB,EAAE,CACnC;QACD,IAAI,MAAqC,EAAE,UAAyC;AACpF,QAAA,MAAM,sBAAsB,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,KAAA,IAAA,IAAZ,YAAY,KAAA,MAAA,GAAZ,YAAY,GAAI,CAAC,EAAE,CAAC,CAAC;AAC7D,QAAA,IAAI;AACF,YAAA,IAAI,YAAY,KAAK,SAAS,EAAE;AAC9B,gBAAA,MAAM,GAAG,MAAM,kBAAkB,CAAC,iBAAiB,CAAC;AAClD,oBAAA,GAAG,MAAM;AACT,oBAAA,SAAS,EAAE,QAAQ;AACnB,oBAAA,SAAS,EAAE,YAAY,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,sBAAsB;AAC3D,oBAAA,eAAe,EAAE,sBAAsB;AACxC,iBAAA,CAAC;YACJ;AAEA,YAAA,IAAI,gBAAgB,KAAK,SAAS,EAAE;AAClC,gBAAA,UAAU,GAAG,MAAM,kBAAkB,CAAC,iBAAiB,CAAC;AACtD,oBAAA,GAAG,MAAM;AACT,oBAAA,SAAS,EAAE,YAAY;AACvB,oBAAA,SAAS,EAAE,gBAAgB,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC;AACtE,oBAAA,eAAe,EAAE,sBAAsB;AACxC,iBAAA,CAAC;YACJ;AAEA,YAAA,IAAI,MAAM,IAAI,UAAU,EAAE;AACxB,gBAAAA,OAAM,CAAC,KAAK,CAAC,wDAAwD,EAAE;oBACrE,IAAI,EAAE,MAAM,CAAC,IAAI;AACjB,oBAAA,MAAM,EAAE,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAN,MAAM,CAAE,MAAM;AACtB,oBAAA,UAAU,EAAE,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAN,MAAM,CAAE,IAAI;AACxB,oBAAA,UAAU,EAAE,UAAU,KAAA,IAAA,IAAV,UAAU,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAV,UAAU,CAAE,UAAU;AAClC,oBAAA,cAAc,EAAE,UAAU,KAAA,IAAA,IAAV,UAAU,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAV,UAAU,CAAE,IAAI;AAChC,oBAAA,OAAO,EAAE,CAAA,EAAA,GAAA,UAAU,KAAA,IAAA,IAAV,UAAU,uBAAV,UAAU,CAAE,OAAO,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,EAAA,GAAI,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAN,MAAM,CAAE,OAAO;AAChD,iBAAA,CAAC;YACJ;iBAAO;AACL,gBAAAA,OAAM,CAAC,KAAK,CAAC,wDAAwD,CAAC;YACxE;QACF;QAAE,OAAO,GAAG,EAAE;AACZ,YAAAA,OAAM,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,CAAC;QACpC;IACF;AAEA;;AAEG;AACH,IAAA,eAAe,qBAAqB,CAClC,MAAmB,EACnB,UAGC,EAAA;;AAED,QAAA,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,UAAU;AACrD,QAAAA,OAAM,CAAC,KAAK,CACV,2CAA2C,MAAM,CAAC,cAAc,CAAA,EAC9D,CAAA,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,MAAA,GAAA,MAAA,GAAN,MAAM,CAAE,OAAO,IAAG,CAAA,YAAA,EAAe,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,MAAA,GAAA,MAAA,GAAN,MAAM,CAAE,OAAO,CAAA,CAAE,GAAG,EACvD,CAAA,gBAAA,CAAkB,EAClB,EAAE,YAAY,EAAE,gBAAgB,EAAE,CACnC;QACD,IAAI,MAAqC,EAAE,UAAyC;AACpF,QAAA,IAAI;YACF,IAAI,YAAY,EAAE;AAChB,gBAAA,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAA,EAAA,GAAA,YAAY,CAAC,KAAK,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,EAAA,GAAI,CAAC,EAAE,CAAC,CAAC;AAClD,gBAAA,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAA,EAAA,GAAA,YAAY,CAAC,KAAK,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,EAAA,GAAI,CAAC,EAAE,CAAC,CAAC;AAClD,gBAAA,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAA,EAAA,GAAA,YAAY,CAAC,IAAI,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,EAAA,GAAI,CAAC,EAAE,CAAC,CAAC;AAChD,gBAAA,MAAM,gBAAgB,GAAG,KAAK,GAAG,KAAK,GAAG,IAAI;AAC7C,gBAAA,MAAM,GAAG,MAAM,kBAAkB,CAAC,2BAA2B,CAAC;AAC5D,oBAAA,GAAG,MAAM;AACT,oBAAA,SAAS,EAAE,QAAQ;oBACnB,WAAW,EAAE,CAAC,KAAK;oBACnB,WAAW,EAAE,CAAC,KAAK;oBACnB,UAAU,EAAE,CAAC,IAAI;AACjB,oBAAA,eAAe,EAAE,gBAAgB;AAClC,iBAAA,CAAC;YACJ;YAEA,IAAI,gBAAgB,EAAE;gBACpB,MAAM,gBAAgB,GAAG;AACvB,sBAAE,IAAI,CAAC,GAAG,CAAC,CAAA,EAAA,GAAA,YAAY,CAAC,KAAK,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,EAAA,GAAI,CAAC,EAAE,CAAC,CAAC;wBACpC,IAAI,CAAC,GAAG,CAAC,CAAA,EAAA,GAAA,YAAY,CAAC,KAAK,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,EAAA,GAAI,CAAC,EAAE,CAAC,CAAC;wBACpC,IAAI,CAAC,GAAG,CAAC,CAAA,EAAA,GAAA,YAAY,CAAC,IAAI,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,EAAA,GAAI,CAAC,EAAE,CAAC;sBAClC,SAAS;AACb,gBAAA,UAAU,GAAG,MAAM,kBAAkB,CAAC,iBAAiB,CAAC;AACtD,oBAAA,GAAG,MAAM;AACT,oBAAA,SAAS,EAAE,YAAY;oBACvB,SAAS,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC;AACzC,oBAAA,eAAe,EAAE,gBAAgB;AAClC,iBAAA,CAAC;YACJ;AAEA,YAAA,IAAI,MAAM,IAAI,UAAU,EAAE;AACxB,gBAAAA,OAAM,CAAC,KAAK,CAAC,kEAAkE,EAAE;oBAC/E,IAAI,EAAE,MAAM,CAAC,IAAI;AACjB,oBAAA,MAAM,EAAE,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAN,MAAM,CAAE,MAAM;AACtB,oBAAA,UAAU,EAAE,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAN,MAAM,CAAE,IAAI;AACxB,oBAAA,UAAU,EAAE,UAAU,KAAA,IAAA,IAAV,UAAU,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAV,UAAU,CAAE,UAAU;AAClC,oBAAA,cAAc,EAAE,UAAU,KAAA,IAAA,IAAV,UAAU,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAV,UAAU,CAAE,IAAI;AAChC,oBAAA,OAAO,EAAE,CAAA,EAAA,GAAA,UAAU,KAAA,IAAA,IAAV,UAAU,uBAAV,UAAU,CAAE,OAAO,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,EAAA,GAAI,MAAM,KAAA,IAAA,IAAN,MAAM,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAN,MAAM,CAAE,OAAO;AAChD,iBAAA,CAAC;YACJ;iBAAO;AACL,gBAAAA,OAAM,CAAC,KAAK,CAAC,kEAAkE,CAAC;YAClF;QACF;QAAE,OAAO,GAAG,EAAE;AACZ,YAAAA,OAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,GAAG,CAAC;QAC9C;AAEA,QAAA,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE;IAC/B;AAEA,IAAA,OAAO,EAAE,WAAW,EAAE,qBAAqB,EAAE;AAC/C;;;;"}