All files / sdk/models Entity.ts

81.53% Statements 53/65
64.7% Branches 22/34
77.77% Functions 7/9
81.53% Lines 53/65

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193  8x 8x 8x 8x 8x     8x     8x                     5x 5x 5x 5x 5x 5x 5x 5x       2x     2x     2x     2x                 2x 2x 1x       1x           1x         1x   2x                         4x           4x 4x 4x       4x       4x 2325x 1721x 1721x 4x 4x         4x       4x           1x 1x     1x 1x       1x             1x     1x                                             1x 1x   1x   1x                   1x 1x               1x              
import { ConnectionRequest } from "./connectedAccounts";
import {Actions} from "./actions"
import { Apps } from "./apps";
import { Integrations } from "./integrations";
import { ActiveTriggers } from "./activeTriggers";
import { ConnectedAccounts } from "./connectedAccounts";
import { ExecuteActionResDTO } from "../client";
import { BackendClient } from "./backendClient";
import { Triggers } from "./triggers";
 
 
export class Entity {
    id: string;
    backendClient: BackendClient;
    triggerModel: Triggers;
    actionsModel: Actions;
    apps: Apps;
    connectedAccounts: ConnectedAccounts;
    integrations: Integrations;
    activeTriggers: ActiveTriggers;
 
    constructor(backendClient: BackendClient, id: string = 'default') {
        this.backendClient = backendClient;
        this.id = id;
        this.triggerModel = new Triggers(this.backendClient);
        this.actionsModel = new Actions(this.backendClient);
        this.apps = new Apps(this.backendClient);
        this.connectedAccounts = new ConnectedAccounts(this.backendClient);
        this.integrations = new Integrations(this.backendClient);
        this.activeTriggers = new ActiveTriggers(this.backendClient);
    }
 
    async execute(actionName: string, params?: Record<string, any> | undefined, text?: string | undefined, connectedAccountId?: string): Promise<ExecuteActionResDTO> {
        const action = await this.actionsModel.get({
            actionName: actionName
        });
        Iif (!action) {
            throw new Error("Could not find action: " + actionName);
        }
        const app = await this.apps.get({
            appKey: action.appKey!
        });
        Iif ((app.yaml as any).no_auth) {
            return this.actionsModel.execute({
                actionName: actionName,
                requestBody: {
                    input: params,
                    appName: action.appKey
                }
            });
        }
        let connectedAccount = null;
        if (connectedAccountId) {
            connectedAccount = await this.connectedAccounts.get({
                connectedAccountId: connectedAccountId
            });
        } else {
            const connectedAccounts = await this.connectedAccounts.list({
                user_uuid: this.id,
                appNames: action.appKey,
                status: 'ACTIVE'
            });
            // @ts-ignore
            Iif (connectedAccounts?.items!.length === 0) {
                throw new Error('No connected account found');
            }
 
            // @ts-ignore
            connectedAccount = connectedAccounts.items![0];
        }
        return this.actionsModel.execute({
            actionName: actionName,
            requestBody: {
                // @ts-ignore
                connectedAccountId: connectedAccount?.id as unknown as string,
                input: params,
                appName: action.appKey,
                text: text
            }
        });
    }
 
    async getConnection(app?: string, connectedAccountId?: string): Promise<any | null> {
        Iif (connectedAccountId) {
            return await this.connectedAccounts.get({
                connectedAccountId
            });
        }
 
        let latestAccount = null;
        let latestCreationDate: Date | null = null;
        const connectedAccounts = await this.connectedAccounts.list({
            user_uuid: this.id,
        });
 
        Iif (!connectedAccounts.items || connectedAccounts.items.length === 0) {
            return null;
        }
 
        for (const connectedAccount of connectedAccounts.items!) {
            if (app === connectedAccount.appName) {
                const creationDate = new Date(connectedAccount.createdAt!);
                if ((!latestAccount || (latestCreationDate && creationDate > latestCreationDate)) && connectedAccount.status === "ACTIVE") {
                    latestCreationDate = creationDate;
                    latestAccount = connectedAccount;
                }
            }
        }
 
        Iif (!latestAccount) {
            return null;
        }
 
        return this.connectedAccounts.get({
            connectedAccountId: latestAccount.id!
        });
    }
 
    async setupTrigger(app: string, triggerName: string, config: { [key: string]: any; }): Promise<any> {
        const connectedAccount = await this.getConnection(app);
        Iif (!connectedAccount) {
            throw new Error(`Could not find a connection with app='${app}' and entity='${this.id}'`);
        }
        const trigger = await this.triggerModel.setup(connectedAccount.id!, triggerName, config);
        return trigger;
    }
 
    async disableTrigger(triggerId: string): Promise<any> {
        return ActiveTriggers.disable({ triggerId: triggerId });
    }
 
    async getConnections(){
        /**
         * Get all connections for an entity.
         */
        const connectedAccounts = await this.connectedAccounts.list({
            user_uuid: this.id
        });
        return connectedAccounts.items!;
    }
 
    async getActiveTriggers() {
        /**
         * Get all active triggers for an entity.
         */
        const connectedAccounts = await this.getConnections();
        const activeTriggers = await this.activeTriggers.list({
           connectedAccountIds: connectedAccounts!.map((account:any) => account.id!).join(",")
        });
        return activeTriggers;
    }
 
    async initiateConnection(
        appName: string,
        authMode?: any,
        authConfig?: { [key: string]: any; },
        redirectUrl?: string,
        integrationId?: string
    ): Promise<ConnectionRequest> {
 
        // Get the app details from the client
        const app = await this.apps.get({ appKey: appName });
        const timestamp = new Date().toISOString().replace(/[-:.]/g, "");
 
        let integration = integrationId ? await this.integrations.get({ integrationId: integrationId }) : null;
        // Create a new integration if not provided
        Iif (!integration && authMode) {
            integration = await this.integrations.create({
                appId: app.appId!,
                name: `integration_${timestamp}`,
                authScheme: authMode,
                authConfig: authConfig,
                useComposioAuth: false,
            });
        }
 
        if (!integration && !authMode) {
            integration = await this.integrations.create({
                appId: app.appId!,
                name: `integration_${timestamp}`,
                useComposioAuth: true,
            });
        }
        
        // Initiate the connection process
        return this.connectedAccounts.initiate({
            integrationId: integration!.id!,
            userUuid: this.id,
            redirectUri: redirectUrl,
        });
    }
}