@wabot-dev/framework 0.1.0-beta.3 → 0.1.0-beta.30

package/dist/src/{controller/channel → feature/chat-controller}/ChatResolver.js

@@ -1,8 +1,9 @@
 import { __decorate, __metadata } from 'tslib';
-import { ChatRepository } from '../../core/chat/repository/IChatRepository.js';
-import { Chat } from '../../core/chat/Chat.js';
-import '../../core/user/IUserRepository.js';
-import { injectable } from '../../injection/index.js';
+import { Chat } from '../chat-bot/Chat.js';
+import { ChatRepository } from '../chat-bot/ChatRepository.js';
+import { injectable } from '../../core/injection/index.js';
+import 'uuid';
+import '../chat-bot/metadata/ChatBotMetadataStore.js';
 
 let ChatResolver = class ChatResolver {
     chatRepository;

package/dist/src/{controller → feature/chat-controller}/metadata/ControllerMetadataStore.js

@@ -1,5 +1,5 @@
 import { __decorate } from 'tslib';
-import { singleton } from '../../injection/index.js';
+import { singleton } from '../../../core/injection/index.js';
 
 let ControllerMetadataStore = class ControllerMetadataStore {
     channels = new Map();

package/dist/src/{controller → feature/chat-controller}/metadata/controller/@chatController.js

@@ -1,4 +1,4 @@
-import { container, injectable } from '../../../injection/index.js';
+import { container, injectable } from '../../../../core/injection/index.js';
 import { ControllerMetadataStore } from '../ControllerMetadataStore.js';
 
 function chatController(config) {

package/dist/src/feature/chat-controller/runChatControllers.js

@@ -0,0 +1,69 @@
+import { container } from '../../core/injection/index.js';
+import { Chat } from '../chat-bot/Chat.js';
+import { ChatBot } from '../chat-bot/ChatBot.js';
+import { ChatMemory } from '../chat-bot/ChatMemory.js';
+import { ChatRepository } from '../chat-bot/ChatRepository.js';
+import 'uuid';
+import { ChatBotMetadataStore } from '../chat-bot/metadata/ChatBotMetadataStore.js';
+import 'reflect-metadata';
+import '../mindset/metadata/MindsetMetadataStore.js';
+import { Mindset } from '../mindset/IMindset.js';
+import '../mindset/MindsetOperator.js';
+import { ChatResolver } from './ChatResolver.js';
+import { ControllerMetadataStore } from './metadata/ControllerMetadataStore.js';
+import { Container } from '../../core/injection/Container.js';
+
+async function prepareChatContainer(container, messageContext, mindsetCtor) {
+    const chatContainer = container.createChildContainer();
+    chatContainer.register(Container, { useValue: chatContainer });
+    chatContainer.registerInstance(Chat, messageContext.chat);
+    const chatRepository = container.resolve(ChatRepository);
+    const chatMemory = await chatRepository.findMemory(messageContext.chat.id);
+    if (!chatMemory) {
+        throw new Error('Not found Chat Memory for Chat with Id=' + messageContext.chat.id);
+    }
+    chatContainer.registerInstance(ChatMemory, chatMemory);
+    const chatBotMetadataStore = container.resolve(ChatBotMetadataStore);
+    const chatBots = chatBotMetadataStore.getChatBotsMetadata();
+    for (const chatBotMetadata of chatBots) {
+        chatContainer.beforeResolution(chatBotMetadata.constructor, (a, b) => {
+            const subContainer = chatContainer.createChildContainer();
+            subContainer.register(Mindset, { useClass: chatBotMetadata.mindsetConstructor });
+            const chatBot = subContainer.resolve(ChatBot);
+            chatContainer.register(chatBotMetadata.injectionToken, { useValue: chatBot });
+        });
+    }
+    return chatContainer;
+}
+function runChatControllers(controllers) {
+    const metadataStore = container.resolve(ControllerMetadataStore);
+    const chatResolver = container.resolve(ChatResolver);
+    for (const controllerCtor of controllers) {
+        const chatControllerMetadata = metadataStore.getChatControllerMetadata(controllerCtor);
+        if (!chatControllerMetadata) {
+            continue;
+        }
+        for (const channelMetadata of chatControllerMetadata.channels) {
+            const channelContainer = container.createChildContainer();
+            if (channelMetadata.channelConfig) {
+                channelContainer.register(channelMetadata.channelConfig.constructor, {
+                    useValue: channelMetadata.channelConfig,
+                });
+            }
+            const channel = channelContainer.resolve(channelMetadata.channelConstructor);
+            channel.listen(async (channelMessage) => {
+                const chat = await chatResolver.resolve(channelMessage.chatConnection);
+                const chatContainer = await prepareChatContainer(channelContainer, {
+                    chat,
+                    message: channelMessage.message,
+                    reply: channelMessage.reply,
+                });
+                const chatController = chatContainer.resolve(channelMetadata.controllerConstructor);
+                chatController[channelMetadata.functionName](channelMessage);
+            });
+            channel.connect();
+        }
+    }
+}
+
+export { runChatControllers };

package/dist/src/{channels → feature}/express/ExpressProvider.js

@@ -3,7 +3,7 @@ import { HttpServerProvider } from '../http/HttpServerProvider.js';
 import express from 'express';
 import { singleton } from 'tsyringe';
 import bodyParser from 'body-parser';
-import { Logger } from '../../logger/Logger.js';
+import { Logger } from '../../core/logger/Logger.js';
 
 let ExpressProvider = class ExpressProvider {
     httpServerProvider;

package/dist/src/{channels → feature}/http/HttpServerProvider.js

@@ -1,5 +1,5 @@
 import { __decorate } from 'tslib';
-import { Logger } from '../../logger/Logger.js';
+import { Logger } from '../../core/logger/Logger.js';
 import { Server } from 'http';
 import { singleton } from 'tsyringe';
 

package/dist/src/{mindset → feature/mindset}/MindsetOperator.js

@@ -1,8 +1,8 @@
 import { __decorate, __metadata } from 'tslib';
-import { injectable } from '../injection/index.js';
+import { injectable } from '../../core/injection/index.js';
 import { Mindset } from './IMindset.js';
 import { MindsetMetadataStore } from './metadata/MindsetMetadataStore.js';
-import { Container } from '../injection/Container.js';
+import { Container } from '../../core/injection/Container.js';
 
 let MindsetOperator = class MindsetOperator {
     mindset;
@@ -22,64 +22,66 @@ let MindsetOperator = class MindsetOperator {
     limits() {
         return this.mindset.limits();
     }
-    async callFunction(name, params) {
-        const fnMetadata = this.metadata.modules
-            .map((module) => module.functions)
-            .flat()
-            .find((fn) => fn.name === name);
-        if (!fnMetadata) {
-            throw new Error(`Function ${name} not found`);
-        }
-        const paramsObj = JSON.parse(params);
-        const module = this.container.resolve(fnMetadata.moduleConstructor);
-        try {
-            const response = await module[name](paramsObj);
-            if (!response) {
-                return 'success';
-            }
-            return response.toString();
-        }
-        catch (error) {
-            return `Error: ${error}`;
-        }
+    async systemPrompt() {
+        let [identity, skills, limits] = await Promise.all([
+            this.identity(),
+            this.skills(),
+            this.limits(),
+        ]);
+        const language = identity.language.replaceAll('#', ' ');
+        const name = identity.name.replaceAll('#', ' ');
+        const age = identity.age ? identity.age.toString().replaceAll('#', ' ') : null;
+        const personality = identity.personality ? identity.personality.replaceAll('#', ' ') : null;
+        skills = skills.replaceAll('#', ' ');
+        limits = limits.replaceAll('#', ' ');
+        const systemPrompt = `
+         # System Instructions
+         you should act as a assistant.
+         your main language is ${language}.
+         your name is ${name}.
+         ${age ? 'you are ' + age + ' years old.' : ''}
+         
+          ${personality ? '## Personality (in your main language) \n' + personality : ''}
+  
+          ## Skills (in your main language)
+          ${skills}
+  
+          ## System limitations (in your main language)
+          ${limits}
+  
+          ## Chat memory
+          Next you will receive a chat history,
+          you should use this information to answer the user.
+      `;
+        return systemPrompt;
     }
-    async allFunctionsDescriptors() {
+    tools() {
         return this.metadata.modules
-            .map((module) => module.functions.map((fn) => this.functionDescriptor(fn)))
+            .map((module) => module.functions.map((fn) => this.tool(fn, module)))
             .flat();
     }
-    functionDescriptor(fn) {
+    tool(fn, module) {
         const description = fn.config.description.replaceAll('#', ' ');
         return {
-            type: 'function',
+            language: module.config.language ?? 'english',
             name: fn.name,
             description,
-            parameters: {
-                type: 'object',
-                properties: fn.params.reduce((prev, param) => ({
-                    ...prev,
-                    [param.name]: this.toolParam(param),
-                }), {}),
-                required: fn.params.filter((param) => !param.config.optional).map((param) => param.name)
-            },
+            parameters: fn.params.map((param) => this.toolParameter(param)),
         };
     }
-    toolParam(param) {
-        const addons = {
-            description: `
+    toolParameter(param) {
+        let description = `
       ### description (in your main language)
       ${param.config.description.replaceAll('#', ' ')}
-      `,
-        };
+      `;
         const type = (() => {
             if (param.type === Number)
                 return 'number';
             if (param.type === String)
                 return 'string';
             if (param.type === Date) {
-                addons.description = `${addons.description}
+                description = `${description}
           ### format: ISO 8681 - YYYY-MM-DDTHH:mm:ssZ
-          ${addons.description}
         `;
                 return 'string';
             }
@@ -87,9 +89,31 @@ let MindsetOperator = class MindsetOperator {
         })();
         return {
             type,
-            ...addons,
+            name: param.name,
+            description,
         };
     }
+    async callFunction(name, params) {
+        const fnMetadata = this.metadata.modules
+            .map((module) => module.functions)
+            .flat()
+            .find((fn) => fn.name === name);
+        if (!fnMetadata) {
+            throw new Error(`Function ${name} not found`);
+        }
+        const paramsObj = JSON.parse(params);
+        const module = this.container.resolve(fnMetadata.moduleConstructor);
+        try {
+            const response = await module[name](paramsObj);
+            if (!response) {
+                return 'success';
+            }
+            return response.toString();
+        }
+        catch (error) {
+            return `Error: ${error}`;
+        }
+    }
 };
 MindsetOperator = __decorate([
     injectable(),

package/dist/src/{mindset → feature/mindset}/metadata/MindsetMetadataStore.js

@@ -1,5 +1,5 @@
 import { __decorate } from 'tslib';
-import { singleton } from '../../injection/index.js';
+import { singleton } from '../../../core/injection/index.js';
 import { MINDSET_FUNCTION_DECORATION_FUNCTION } from './functions/decoratorNames.js';
 import { MINDSET_DECORATION_MINDSET } from './mindsets/decoratorNames.js';
 import { MINDSET_MODULE_DECORATION_MODULE } from './modules/decoratorNames.js';

package/dist/src/{mindset → feature/mindset}/metadata/functions/@mindsetFunction.js

@@ -1,5 +1,5 @@
 import 'reflect-metadata';
-import { container } from '../../../injection/index.js';
+import { container } from '../../../../core/injection/index.js';
 import { MindsetMetadataStore } from '../MindsetMetadataStore.js';
 import { MINDSET_FUNCTION_DECORATION_FUNCTION } from './decoratorNames.js';
 

package/dist/src/{mindset → feature/mindset}/metadata/mindsets/@mindset.js

@@ -1,5 +1,5 @@
 import 'reflect-metadata';
-import { container, injectable } from '../../../injection/index.js';
+import { container, injectable } from '../../../../core/injection/index.js';
 import { MindsetMetadataStore } from '../MindsetMetadataStore.js';
 import { MINDSET_DECORATION_MINDSET } from './decoratorNames.js';
 

package/dist/src/{mindset → feature/mindset}/metadata/modules/@mindsetModule.js

@@ -1,5 +1,4 @@
-import 'reflect-metadata';
-import { container, injectable } from '../../../injection/index.js';
+import { container, injectable } from '../../../../core/injection/index.js';
 import { MindsetMetadataStore } from '../MindsetMetadataStore.js';
 import { MINDSET_MODULE_DECORATION_MODULE } from './decoratorNames.js';
 

package/dist/src/{mindset → feature/mindset}/metadata/params/@param.js

@@ -1,6 +1,6 @@
 import 'reflect-metadata';
 import { MindsetMetadataStore } from '../MindsetMetadataStore.js';
-import { container } from '../../../injection/index.js';
+import { container } from '../../../../core/injection/index.js';
 import { PARAM_DECORATION_PARAM } from './decoratorNames.js';
 
 function param(config) {

package/dist/src/feature/rest-controller/auth/Auth.js

@@ -0,0 +1,24 @@
+import { __decorate } from 'tslib';
+import { CustomError } from '../../../core/error/CustomError.js';
+import { scoped, Lifecycle } from '../../../core/injection/index.js';
+
+let Auth = class Auth {
+    authInfo = null;
+    require() {
+        if (!this.authInfo) {
+            throw new CustomError({ message: 'Unauthorized', httpCode: 401 });
+        }
+        return this.authInfo;
+    }
+    assign(authInfo) {
+        if (this.authInfo) {
+            throw new CustomError({ message: 'Authorization info already assigned', httpCode: 401 });
+        }
+        this.authInfo = authInfo;
+    }
+};
+Auth = __decorate([
+    scoped(Lifecycle.ContainerScoped)
+], Auth);
+
+export { Auth };

package/dist/src/feature/rest-controller/metadata/@get.js

@@ -0,0 +1,19 @@
+import { container } from '../../../core/injection/index.js';
+import { RestControllerMetadataStore } from './RestControllerMetadataStore.js';
+
+function get(config) {
+    return function (target, propertyKey) {
+        const functionName = propertyKey.toString();
+        const paramsTypes = Reflect.getMetadata('design:paramtypes', target, functionName);
+        const store = container.resolve(RestControllerMetadataStore);
+        store.saveEndPointMetadata({
+            controllerConstructor: target.constructor,
+            functionName,
+            method: 'get',
+            path: typeof config === 'string' ? config : config?.path,
+            paramsTypes,
+        });
+    };
+}
+
+export { get };

package/dist/src/feature/rest-controller/metadata/@middleware.js

@@ -0,0 +1,16 @@
+import { container } from '../../../core/injection/index.js';
+import { RestControllerMetadataStore } from './RestControllerMetadataStore.js';
+
+function middleware(middlewareConstructor) {
+    return function (target, propertyKey) {
+        const functionName = propertyKey.toString();
+        const store = container.resolve(RestControllerMetadataStore);
+        store.saveMiddlewareMetadata({
+            controllerConstructor: target.constructor,
+            functionName,
+            middlewareConstructor: middlewareConstructor,
+        });
+    };
+}
+
+export { middleware };

package/dist/src/feature/rest-controller/metadata/@post.js

@@ -0,0 +1,19 @@
+import { container } from '../../../core/injection/index.js';
+import { RestControllerMetadataStore } from './RestControllerMetadataStore.js';
+
+function post(config) {
+    return function (target, propertyKey) {
+        const functionName = propertyKey.toString();
+        const paramsTypes = Reflect.getMetadata('design:paramtypes', target, functionName);
+        const store = container.resolve(RestControllerMetadataStore);
+        store.saveEndPointMetadata({
+            controllerConstructor: target.constructor,
+            functionName,
+            method: 'post',
+            path: typeof config === 'string' ? config : config?.path,
+            paramsTypes,
+        });
+    };
+}
+
+export { post };

package/dist/src/feature/rest-controller/metadata/@restController.js

@@ -0,0 +1,15 @@
+import { container, injectable } from '../../../core/injection/index.js';
+import { RestControllerMetadataStore } from './RestControllerMetadataStore.js';
+
+function restController(config) {
+    return function (target) {
+        const metaDataStore = container.resolve(RestControllerMetadataStore);
+        metaDataStore.saveControllerMetadata({
+            controllerConstructor: target,
+            path: typeof config === 'string' ? config : config.path,
+        });
+        injectable()(target);
+    };
+}
+
+export { restController };

package/dist/src/feature/rest-controller/metadata/RestControllerMetadataStore.js

@@ -0,0 +1,50 @@
+import { __decorate } from 'tslib';
+import { singleton } from '../../../core/injection/index.js';
+
+let RestControllerMetadataStore = class RestControllerMetadataStore {
+    endPoints = new Map();
+    middlewares = new Map();
+    restControllers = new Map();
+    saveControllerMetadata(controllerMetadata) {
+        this.restControllers.set(controllerMetadata.controllerConstructor, controllerMetadata);
+    }
+    saveEndPointMetadata(endPointMetadata) {
+        let controllerEndPoints = this.endPoints.get(endPointMetadata.controllerConstructor);
+        if (!controllerEndPoints) {
+            this.endPoints.set(endPointMetadata.controllerConstructor, (controllerEndPoints = new Map()));
+        }
+        controllerEndPoints.set(endPointMetadata.functionName, endPointMetadata);
+    }
+    saveMiddlewareMetadata(middlewareMetadata) {
+        let controllerMiddlewares = this.middlewares.get(middlewareMetadata.controllerConstructor);
+        if (!controllerMiddlewares) {
+            this.middlewares.set(middlewareMetadata.controllerConstructor, (controllerMiddlewares = new Map()));
+        }
+        let methodMiddlewares = controllerMiddlewares.get(middlewareMetadata.functionName);
+        if (!methodMiddlewares) {
+            controllerMiddlewares.set(middlewareMetadata.functionName, (methodMiddlewares = []));
+        }
+        methodMiddlewares.unshift(middlewareMetadata);
+    }
+    getControllerEndPointsInfo(controllerConstructor) {
+        const controller = this.restControllers.get(controllerConstructor);
+        if (!controller) {
+            throw new Error(`${controllerConstructor.name} should be decorated with @restController`);
+        }
+        const endPoints = this.endPoints.get(controllerConstructor);
+        if (!endPoints?.size) {
+            // TODO: Warning
+            return [];
+        }
+        return [...endPoints.values()].map((endPoint) => ({
+            ...endPoint,
+            middlewares: this.middlewares.get(endPoint.controllerConstructor)?.get(endPoint.functionName) ?? [],
+            controller: this.restControllers.get(endPoint.controllerConstructor),
+        }));
+    }
+};
+RestControllerMetadataStore = __decorate([
+    singleton()
+], RestControllerMetadataStore);
+
+export { RestControllerMetadataStore };

package/dist/src/feature/rest-controller/runRestControllers.js

@@ -0,0 +1,75 @@
+import path from 'path';
+import { container } from '../../core/injection/index.js';
+import { RestControllerMetadataStore } from './metadata/RestControllerMetadataStore.js';
+import { Logger } from '../../core/logger/Logger.js';
+import { ExpressProvider } from '../express/ExpressProvider.js';
+import '../../core/validation/metadata/ValidationMetadataStore.js';
+import { validate } from '../../core/validation/validate.js';
+import { CustomError } from '../../core/error/CustomError.js';
+
+function buildRequest(req) {
+    return Object.assign({}, req.body, req.query, req.params);
+}
+function runRestControllers(controllers) {
+    const logger = new Logger('wabot:rest');
+    const metadataStore = container.resolve(RestControllerMetadataStore);
+    const expressProvider = container.resolve(ExpressProvider);
+    const expressApp = expressProvider.getExpress();
+    controllers.forEach((controller) => {
+        const endPoints = metadataStore.getControllerEndPointsInfo(controller);
+        endPoints.forEach((endPoint) => {
+            const method = endPoint.method;
+            const route = path.join(endPoint.controller.path, endPoint.path ?? '');
+            logger.info(`config ${endPoint.method.toUpperCase()} ${route}`);
+            expressApp[method](route, async (req, res) => {
+                const requestContainer = container.createChildContainer();
+                try {
+                    const middlewares = endPoint.middlewares.map((x) => requestContainer.resolve(x.middlewareConstructor));
+                    for (const middleware of middlewares) {
+                        await middleware.handle(req, res, requestContainer);
+                    }
+                    const controllerInstance = requestContainer.resolve(endPoint.controllerConstructor);
+                    const endPointArgs = [];
+                    let defaultArgFound = false;
+                    for (let paramIndex = 0; paramIndex < endPoint.paramsTypes.length; paramIndex++) {
+                        const paramType = endPoint.paramsTypes[paramIndex];
+                        if (defaultArgFound) {
+                            throw new Error(`Cant determine de parameter ${paramIndex} value`);
+                        }
+                        defaultArgFound = true;
+                        if (typeof paramType === 'function') {
+                            const { value, error } = validate(buildRequest(req), paramType);
+                            if (error) {
+                                throw new CustomError({ httpCode: 400, message: error.description, info: error });
+                            }
+                            endPointArgs.push(value);
+                        }
+                    }
+                    const response = await controllerInstance[endPoint.functionName].apply(controllerInstance, endPointArgs);
+                    res.status(200).json(response);
+                }
+                catch (err) {
+                    if (err instanceof Error) {
+                        const keys = Object.keys(err).filter((key) => !['message', 'stack'].includes(key));
+                        const { httpCode, ...info } = keys.reduce((acc, key) => {
+                            acc[key] = err[key];
+                            return acc;
+                        }, {});
+                        res
+                            .status(httpCode ?? 500)
+                            .json({ error: { message: err.message, stack: err.stack, ...info } });
+                    }
+                    else {
+                        res.status(500).json({ error: { message: 'Unknown error' } });
+                    }
+                }
+                finally {
+                    requestContainer.dispose();
+                }
+            });
+        });
+    });
+    expressProvider.listen();
+}
+
+export { runRestControllers };

package/dist/src/{channels → feature}/socket/SocketServerProvider.js

@@ -1,5 +1,5 @@
 import { __decorate, __metadata } from 'tslib';
-import { Logger } from '../../logger/Logger.js';
+import { Logger } from '../../core/logger/Logger.js';
 import { HttpServerProvider } from '../http/HttpServerProvider.js';
 import { Server } from 'socket.io';
 import { singleton } from 'tsyringe';