@vida-global/apps-tools 1.0.1

package/lib/userContext.js

@@ -0,0 +1,218 @@
+const { logger } = require('@vida-global/core');
+const   _        = require("lodash");
+
+
+class VaderUserContext {
+    constructor(
+        manager
+    ) {
+        this.manager = manager;
+        this.context = {};
+    }
+
+    APPID_VERSION_KEY (appId, appVersion) {
+        return `${appId}:${appVersion}`
+    }
+
+    USER_CONFIG_KEY () {
+        return `vader:user:${this.get('user').id}:config`
+    }
+
+    USER_APPS_KEY(user) {
+        return `vader:user:${this.get('user').id}:installed`
+    }
+
+    async load(ctx) {
+        ctx = ctx || {}
+        for (const [key, value] of Object.entries(ctx)) {
+            this.set(key, value)
+        }
+        if (ctx.user && !this.manager.attachedToAppServer) {
+            const userConfig = await this.manager.redisClient.hgetall(this.USER_CONFIG_KEY())
+            if (userConfig) {
+                for (const [key, value] of Object.entries(userConfig)) {
+                    try {
+                        this.set(key, JSON.parse(value))
+                    } catch (e) {
+                        this.set(key, value)
+                    }
+                }
+            }
+            // Load installed apps
+            const userAppsKey = this.USER_APPS_KEY()
+            const installedApps = await this.manager.redisClient.hgetall(userAppsKey) || {}
+            const installedAppIds = Object.keys(installedApps)
+            this.set('installedApps', installedAppIds)
+
+            // Load appVersionIds
+            const appVersionIds = await this.getAppVersionIds()
+            this.set('appVersionIds', appVersionIds)
+
+            // Load enabled functions for each installed app
+            const enabledFunctions = {}
+            for (const appVersionId of installedAppIds) {
+                const [appId, appVersion] = appVersionId.split(':')
+                enabledFunctions[appVersionId] = this.getEnabledFunctions(appId, appVersion)
+            }
+            this.set('enabledFunctions', enabledFunctions)
+        }
+    }
+
+    get(name) {
+        // && name !== 'customVaderUrl' && name !== 'customVaderToken'
+        if (!this.context.hasOwnProperty(name)) {
+            // logger.warn(`VADER: User Context Missing Key: ${name}`);
+            return null;
+        }
+        return this.context[name];
+    }
+
+    set(name, value) {
+        this.context[name] = value;
+    }
+
+    toDict() {
+        return {...this.context}
+    }
+
+    toJSON() {
+        return JSON.stringify(this.context)
+    }
+
+    async getProviderConfig (providerType, providerName, appId = null, appVersion = null) {
+        let providerKey = `${providerType}:${providerName}`
+        if (appId && appVersion) {
+           providerKey = `${appId}:${appVersion}:${providerKey}`
+        }
+
+        // Only local context is available
+        if (this.manager.attachedToAppServer) {
+            return this.context[providerKey];
+        }
+
+        const config = await this.manager.redisClient.hget(
+            this.USER_CONFIG_KEY(), providerKey
+        ) || '{}'
+        const parsedConfig = JSON.parse(config);
+        this.context[providerKey] = parsedConfig;
+        return parsedConfig;
+    }
+
+    async setProviderConfig (providerType, providerName, config, appId = null, appVersion = null) {
+        let providerKey = `${providerType}:${providerName}`
+        if (appId && appVersion) {
+            providerKey = `${appId}:${appVersion}:${providerKey}`
+        }
+        return await this.manager.redisClient.hset(
+            this.USER_CONFIG_KEY(), providerKey, JSON.stringify(config)
+        )
+    }
+
+    async deleteProviderConfig (providerType, providerName, appId = null, appVersion = null) {
+        let providerKey = `${providerType}:${providerName}`
+        if (appId && appVersion) {
+            providerKey = `${appId}:${appVersion}:${providerKey}`
+        }
+        return await this.manager.redisClient.hdel(
+            this.USER_CONFIG_KEY(), providerKey
+        )
+    }
+
+    async fetchAgentAppManifest(appId, appVersion) {
+        const agent = this.get('agent')
+        const manifest = _.find(agent.apps, a => a.appId === appId && a.version === appVersion)
+        return manifest ? manifest : null
+    }
+
+    async installApp(appId, appVersion) {
+        // Store the installed app manifest
+        const appVersionKey = this.APPID_VERSION_KEY(appId, appVersion)
+        const userAppsKey = this.USER_APPS_KEY()
+
+        await this.manager.redisClient.hset(userAppsKey, appVersionKey, 1)
+    }
+
+    async uninstallApp (appId, appVersion) {
+        const appVersionKey = this.APPID_VERSION_KEY(appId, appVersion)
+        await this.manager.redisClient.hdel(this.USER_APPS_KEY(), appVersionKey)
+    }
+
+    async isAppInstalled (appId, appVersion) {
+        if (this.manager.attachedToAppServer) {
+            return (this.get('installedApps') || []).includes(`${appId}:${appVersion}`)
+        }
+        const appVersionKey = this.APPID_VERSION_KEY(appId, appVersion)
+        return !!await this.manager.redisClient.hexists(this.USER_APPS_KEY(), appVersionKey)
+    }
+
+    async getAppVersionIds() {
+        if (this.manager.attachedToAppServer) {
+            return this.get('appVersionIds')
+        }
+        let appVersionIds
+        let agent = this.get('agent')
+        if (agent) {
+            appVersionIds = _.map(agent.apps, a => `${a.appId}:${a.version}`)
+        } else {
+            const userToAppsKey = this.USER_APPS_KEY()
+            appVersionIds = await this.manager.redisClient.hkeys(userToAppsKey)
+            if (!appVersionIds) {
+                appVersionIds = []
+            }
+        }
+        return appVersionIds
+    }
+
+    getEnabledFunctions (appId, appVersion) {
+        if (this.manager.attachedToAppServer) {
+            return this.get('enabledFunctions')[this.APPID_VERSION_KEY(appId, appVersion)] || []
+        }
+        const agent = this.get('agent')
+        const apps = agent?.apps || []
+        const appVersionKey = this.APPID_VERSION_KEY(appId, appVersion)
+        if (agent && apps.some(a => a.appId === appId && a.version === appVersion)) {
+            const enabledFunctions = agent['enabledFunctions'] || {}
+            return enabledFunctions[appVersionKey] || []
+        } else {
+            return []
+        }
+    }
+}
+
+/**
+ * A proxy class that adds vader user context to manager function calls
+ */
+class VaderUserContextProxy {
+    constructor(target, context) {
+        this.target = target;
+        this.context = context;
+
+        // Create a proxy to intercept function calls
+        return new Proxy(this, {
+            get: (obj, prop) => {
+                // If the property exists on the target and is a function
+                if (typeof this.target[prop] === 'function') {
+                    return this._wrapFunction(prop);
+                }
+                // For non-function properties, return the target's property
+                return this.target[prop];
+            }
+        });
+    }
+    _wrapFunction(functionName) {
+        return async (...args) => {
+            const vaderUserContext = new VaderUserContext(this.target);
+            await vaderUserContext.load(this.context)
+            return await this.target[functionName](
+                ...args,
+                vaderUserContext,
+            );
+        };
+    }
+}
+
+module.exports = {
+    VaderUserContext,
+    VaderUserContextProxy
+}
+

package/package.json

@@ -0,0 +1,33 @@
+{
+    "name": "@vida-global/apps-tools",
+    "version": "1.0.1",
+    "description": "Classes for creating vida apps",
+    "author": "",
+    "license": "ISC",
+    "main": "index.js",
+    "directories": {
+        "doc": "lib",
+        "test": "test"
+    },
+    "scripts": {
+        "test": "NODE_NO_WARNINGS=1 node --experimental-vm-modules node_modules/jest/bin/jest.js"
+    },
+    "jest": {
+        "collectCoverage": true,
+        "coveragePathIgnorePatterns": [
+            "/test/helpers"
+        ]
+    },
+    "dependencies": {
+        "axios": "^1.8.1",
+        "googleapis": "^146.0.0",
+        "lodash": "^4.17.21",
+        "mcp-client": "^1.13.1",
+        "swagger-ui-express": "^5.0.1",
+        "@vida-global/core": "^1.1.0"
+    },
+    "devDependencies": {
+        "jest": "^29.6.2"
+    }
+}
+