@multiplayer-app/ai-agent-node 0.1.0-beta.12 → 0.1.0-beta.15

package/dist/esm/index.d.ts

@@ -1,13 +1,40 @@
-export { redisService } from './services/RedisService';
-export { socketService } from './services/SocketService';
-export { agentStore, AgentProcessEventType } from './store/AgentStore';
+export { createProposeFormValuesTool, proposeFormValuesToolName } from './tools/proposeFormValuesTool';
+export type { Config } from './config';
 export type { AgentProcessEvent, AgentProcessListener, AgentProcess, AgentProcessEventPayload } from './store/AgentStore';
+export type { AgentLogger } from './libs/logger';
 export { ConfigStore } from './store/ConfigStore';
-export { ModelStore } from './store/ModelStore';
-export { ArtifactStore } from './store/ArtifactStore';
-export { ChatProcessor, ModelsProcessor } from './processors';
-export { startServices, stopServices } from './helpers/SetupHelper';
-export { createProposeFormValuesTool, proposeFormValuesToolName } from './tools/proposeFormValuesTool';
-export { s3 } from './libs/s3';
-export { setLogger, getLogger, logger, type AgentLogger } from './libs/logger';
+import { Config } from './config';
+import { S3Lib } from './libs/s3';
+import { Server as HTTPServer } from 'http';
+import { AgentChatRepository, AgentMessageRepository } from '@multiplayer-app/ai-agent-db';
+import { ConfigStore } from './store/ConfigStore';
+import { type AgentLogger } from './libs/logger';
+import { ArtifactStore } from './store/ArtifactStore';
+import { ChatProcessor, ModelsProcessor } from './processors';
+import { Middleware } from './services/SocketService';
+export interface AIAgentServiceOptions {
+    chatRepository: AgentChatRepository;
+    messageRepository: AgentMessageRepository;
+    setupAgents: (agentStore: ConfigStore) => void;
+    artifactStore?: ArtifactStore;
+    config?: Partial<Config>;
+    logger?: AgentLogger;
+}
+export declare class AIAgentService {
+    private chatProcessor;
+    private modelsProcessor;
+    private s3;
+    private socketService;
+    private redisService;
+    private agentStore;
+    constructor(options: AIAgentServiceOptions);
+    start(): Promise<[void, void, void]>;
+    stop(): Promise<[void]>;
+    getS3(): S3Lib;
+    stopAgentProcess(chatId: string): Promise<void>;
+    initSocketSupport(httpServer: HTTPServer, authorizationMiddleware?: Middleware): Promise<void>;
+    getChatProcessor(): ChatProcessor;
+    getModelProcessor(): ModelsProcessor;
+    private fetchModels;
+}
 //# sourceMappingURL=index.d.ts.map

package/dist/esm/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AACvE,YAAY,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,YAAY,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAC1H,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACpE,OAAO,EAAE,2BAA2B,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AACvG,OAAO,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,WAAW,EAAE,MAAM,eAAe,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,2BAA2B,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AACvG,YAAY,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AACvC,YAAY,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,YAAY,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAC1H,YAAY,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,OAAO,EAAkB,MAAM,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,MAAM,IAAI,UAAU,EAAE,MAAM,MAAM,CAAC;AAE5C,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAC3F,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAe,EAAa,KAAK,WAAW,EAAE,MAAM,eAAe,CAAC;AAEpE,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAGtD,MAAM,WAAW,qBAAqB;IAClC,cAAc,EAAE,mBAAmB,CAAC;IACpC,iBAAiB,EAAE,sBAAsB,CAAC;IAC1C,WAAW,EAAE,CAAC,UAAU,EAAE,WAAW,KAAK,IAAI,CAAC;IAC/C,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,MAAM,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACzB,MAAM,CAAC,EAAE,WAAW,CAAC;CACxB;AAED,qBAAa,cAAc;IACvB,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,EAAE,CAAQ;IAClB,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,UAAU,CAAa;gBAEnB,OAAO,EAAE,qBAAqB;IAyB1C,KAAK;IAQL,IAAI;IAIJ,KAAK;IAIC,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIrD,iBAAiB,CAAC,UAAU,EAAE,UAAU,EAAE,uBAAuB,CAAC,EAAE,UAAU;IAI9E,gBAAgB;IAIhB,iBAAiB;YAIH,WAAW;CAU5B"}

package/dist/esm/index.js

@@ -1,12 +1,80 @@
-export { redisService } from './services/RedisService';
-export { socketService } from './services/SocketService';
-export { agentStore, AgentProcessEventType } from './store/AgentStore';
-export { ConfigStore } from './store/ConfigStore';
-export { ModelStore } from './store/ModelStore';
-export { ArtifactStore } from './store/ArtifactStore';
-export { ChatProcessor, ModelsProcessor } from './processors';
-export { startServices, stopServices } from './helpers/SetupHelper';
 export { createProposeFormValuesTool, proposeFormValuesToolName } from './tools/proposeFormValuesTool';
-export { s3 } from './libs/s3';
-export { setLogger, getLogger, logger } from './libs/logger';
+export { ConfigStore } from './store/ConfigStore';
+import { ConfigProvider } from './config';
+import { S3Lib } from './libs/s3';
+import { SocketService } from './services/SocketService';
+import { AgentStore, AgentProcessEventType } from './store/AgentStore';
+import { ConfigStore } from './store/ConfigStore';
+import logger, { setLogger } from './libs/logger';
+import { RedisService } from './services/RedisService';
+import { ArtifactStore } from './store/ArtifactStore';
+import { ChatProcessor, ModelsProcessor } from './processors';
+import { ModelStore } from './store';
+import { ModelFetcher } from './services';
+export class AIAgentService {
+    chatProcessor;
+    modelsProcessor;
+    s3;
+    socketService;
+    redisService;
+    agentStore;
+    constructor(options) {
+        if (options.config) {
+            ConfigProvider.getInstance().setConfig(options.config);
+        }
+        this.s3 = new S3Lib(ConfigProvider.getInstance().getConfig().s3);
+        this.redisService = new RedisService(ConfigProvider.getInstance().getConfig().redis);
+        this.socketService = new SocketService(ConfigProvider.getInstance().getConfig().socket, this.redisService);
+        this.modelsProcessor = new ModelsProcessor();
+        options.setupAgents(ConfigStore.getInstance());
+        this.agentStore = new AgentStore(this.redisService);
+        if (options.logger)
+            setLogger(options.logger);
+        this.chatProcessor = new ChatProcessor({
+            agentStore: this.agentStore,
+            chatRepository: options.chatRepository,
+            messageRepository: options.messageRepository,
+            artifactStore: options.artifactStore || new ArtifactStore(),
+            s3Lib: this.s3,
+            config: ConfigProvider.getInstance().getConfig(),
+            socketService: this.socketService
+        });
+    }
+    start() {
+        return Promise.all([
+            this.agentStore.initialize(),
+            this.fetchModels(ConfigProvider.getInstance().getConfig().ai),
+            this.s3.ensureBucketExists(ConfigProvider.getInstance().getConfig().s3.bucket ?? '').catch((err) => logger.error("Unable to auto create s3 bucket")),
+        ]);
+    }
+    stop() {
+        return Promise.all([this.redisService.disconnect()]);
+    }
+    getS3() {
+        return this.s3;
+    }
+    async stopAgentProcess(chatId) {
+        await this.agentStore.shareAgentProcessEvent(chatId, { type: AgentProcessEventType.Stop, data: undefined });
+    }
+    initSocketSupport(httpServer, authorizationMiddleware) {
+        return this.socketService.initialize(httpServer, authorizationMiddleware);
+    }
+    getChatProcessor() {
+        return this.chatProcessor;
+    }
+    getModelProcessor() {
+        return this.modelsProcessor;
+    }
+    async fetchModels(config) {
+        try {
+            const modelFetcher = new ModelFetcher(config);
+            const models = await modelFetcher.fetchAllModels();
+            ModelStore.getInstance().setModels(models);
+        }
+        catch (error) {
+            logger.error('Failed to fetch models on startup:', error);
+            throw error;
+        }
+    }
+}
 //# sourceMappingURL=index.js.map

package/dist/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEvE,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACpE,OAAO,EAAE,2BAA2B,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AACvG,OAAO,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAoB,MAAM,eAAe,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,2BAA2B,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAIvG,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,OAAO,EAAE,cAAc,EAAU,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAEzD,OAAO,EAAE,UAAU,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEvE,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,MAAM,EAAE,EAAE,SAAS,EAAoB,MAAM,eAAe,CAAC;AACpE,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE9D,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAU1C,MAAM,OAAO,cAAc;IACf,aAAa,CAAgB;IAC7B,eAAe,CAAkB;IACjC,EAAE,CAAQ;IACV,aAAa,CAAgB;IAC7B,YAAY,CAAe;IAC3B,UAAU,CAAa;IAE/B,YAAY,OAA8B;QACtC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACjB,cAAc,CAAC,WAAW,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,EAAE,GAAG,IAAI,KAAK,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC;QACrF,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QAC3G,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC7C,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC,CAAC;QAE/C,IAAI,CAAC,UAAU,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEpD,IAAI,OAAO,CAAC,MAAM;YAAE,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAE9C,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC;YACnC,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,cAAc,EAAE,OAAO,CAAC,cAAc;YACtC,iBAAiB,EAAE,OAAO,CAAC,iBAAiB;YAC5C,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,IAAI,aAAa,EAAE;YAC3D,KAAK,EAAE,IAAI,CAAC,EAAE;YACd,MAAM,EAAE,cAAc,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE;YAChD,aAAa,EAAE,IAAI,CAAC,aAAa;SACpC,CAAC,CAAC;IACP,CAAC;IAED,KAAK;QACH,OAAO,OAAO,CAAC,GAAG,CAAC;YACjB,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE;YAC5B,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC;YAC7D,IAAI,CAAC,EAAE,CAAC,kBAAkB,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;SACrJ,CAAC,CAAA;IACJ,CAAC;IAED,IAAI;QACA,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;IACxD,CAAC;IAED,KAAK;QACD,OAAO,IAAI,CAAC,EAAE,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,MAAc;QACjC,MAAM,IAAI,CAAC,UAAU,CAAC,sBAAsB,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,qBAAqB,CAAC,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;IAChH,CAAC;IAED,iBAAiB,CAAC,UAAsB,EAAE,uBAAoC;QAC1E,OAAO,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,UAAU,EAAE,uBAAuB,CAAC,CAAC;IAC9E,CAAC;IAED,gBAAgB;QACZ,OAAO,IAAI,CAAC,aAAa,CAAC;IAC9B,CAAC;IAED,iBAAiB;QACb,OAAO,IAAI,CAAC,eAAe,CAAA;IAC/B,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,MAAoB;QAC1C,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;YAC9C,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,cAAc,EAAE,CAAC;YACnD,UAAU,CAAC,WAAW,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC1D,MAAM,KAAK,CAAC;QACd,CAAC;IACL,CAAC;CACJ"}

package/dist/esm/libs/s3/index.d.ts

@@ -1,4 +1,3 @@
 export { ObjectCannedACL } from '@aws-sdk/client-s3';
-export * as s3 from './s3.lib';
-export * from './config';
+export { S3Lib } from './s3.lib';
 //# sourceMappingURL=index.d.ts.map

package/dist/esm/libs/s3/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/libs/s3/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAA;AACpD,OAAO,KAAK,EAAE,MAAM,UAAU,CAAA;AAC9B,cAAc,UAAU,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/libs/s3/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAA;AACpD,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAA"}

package/dist/esm/libs/s3/index.js

@@ -1,4 +1,3 @@
 export { ObjectCannedACL } from '@aws-sdk/client-s3';
-export * as s3 from './s3.lib';
-export * from './config';
+export { S3Lib } from './s3.lib';
 //# sourceMappingURL=index.js.map

package/dist/esm/libs/s3/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/libs/s3/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAA;AACpD,OAAO,KAAK,EAAE,MAAM,UAAU,CAAA;AAC9B,cAAc,UAAU,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/libs/s3/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAA;AACpD,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAA"}

package/dist/esm/libs/s3/s3.lib.d.ts

@@ -1,25 +1,32 @@
 import { CompleteMultipartUploadOutput, ObjectCannedACL, GetObjectOutput, GetObjectCommandOutput, ListObjectsV2CommandOutput } from '@aws-sdk/client-s3';
 import stream from 'stream';
-export declare const uploadFile: (Key: string, Bucket: string, Body: any, Expires?: Date) => Promise<import("@aws-sdk/client-s3").PutObjectCommandOutput>;
-export declare const downloadFile: (Key: string, Bucket: string) => Promise<GetObjectOutput>;
-export declare const downloadFileAsString: (Key: string, Bucket: string) => Promise<string | undefined>;
-export declare const downloadFileAsByteArray: (Key: string, Bucket: string) => Promise<Uint8Array | undefined>;
-export declare const streamUpload: (Key: string, Bucket: string, ACL?: ObjectCannedACL) => {
-    writeStream: stream.PassThrough;
-    promise: Promise<CompleteMultipartUploadOutput>;
-};
-export declare const getPresignedUploadUrl: (Key: string, Bucket: string, expiresIn?: number) => Promise<string>;
-export declare const getPresignedDownloadUrl: (Key: string, Bucket: string) => Promise<string>;
-export declare const getDownloadUrl: (Key: string, Bucket: string) => string;
-export declare const copy: (Bucket: string, KeyFrom: string, KeyTo: string) => Promise<import("@aws-sdk/client-s3").CopyObjectCommandOutput>;
-export declare const copyBetweenBuckets: (BucketFrom: string, KeyFrom: string, BucketTo: string, KeyTo: string) => Promise<import("@aws-sdk/client-s3").CopyObjectCommandOutput>;
-export declare const getObject: (Bucket: string, Key: string) => Promise<GetObjectCommandOutput>;
-export declare const listObjectsByPrefix: (Bucket: string, Prefix: string) => Promise<ListObjectsV2CommandOutput>;
-export declare const headObject: (Bucket: string, Key: string) => Promise<GetObjectCommandOutput>;
-export declare const deleteObject: (Bucket: string, Key: string) => Promise<GetObjectCommandOutput>;
-export declare const deleteObjectsByPrefix: (Bucket: string, Prefix: string, ContinuationToken?: string) => Promise<void>;
-/**
- * Ensures a bucket exists, creating it if it doesn't
- */
-export declare const ensureBucketExists: (Bucket: string) => Promise<void>;
+import { Config } from '../../config';
+export declare class S3Lib {
+    private readonly client;
+    private readonly signer;
+    private readonly config;
+    constructor(config: Config["s3"]);
+    uploadFile(Key: string, Bucket: string, Body: any, Expires?: Date): Promise<import("@aws-sdk/client-s3").PutObjectCommandOutput>;
+    downloadFile(Key: string, Bucket: string): Promise<GetObjectOutput>;
+    downloadFileAsString(Key: string, Bucket: string): Promise<string | undefined>;
+    downloadFileAsByteArray(Key: string, Bucket: string): Promise<Uint8Array | undefined>;
+    streamUpload(Key: string, Bucket: string, ACL?: ObjectCannedACL): {
+        writeStream: stream.PassThrough;
+        promise: Promise<CompleteMultipartUploadOutput>;
+    };
+    getPresignedUploadUrl(Key: string, Bucket: string, expiresIn?: number): Promise<string>;
+    getPresignedDownloadUrl(Key: string, Bucket: string): Promise<string>;
+    getDownloadUrl(Key: string, Bucket: string): string;
+    copy(Bucket: string, KeyFrom: string, KeyTo: string): Promise<import("@aws-sdk/client-s3").CopyObjectCommandOutput>;
+    copyBetweenBuckets(BucketFrom: string, KeyFrom: string, BucketTo: string, KeyTo: string): Promise<import("@aws-sdk/client-s3").CopyObjectCommandOutput>;
+    getObject(Bucket: string, Key: string): Promise<GetObjectCommandOutput>;
+    listObjectsByPrefix(Bucket: string, Prefix: string): Promise<ListObjectsV2CommandOutput>;
+    headObject(Bucket: string, Key: string): Promise<GetObjectCommandOutput>;
+    deleteObject(Bucket: string, Key: string): Promise<GetObjectCommandOutput>;
+    deleteObjectsByPrefix(Bucket: string, Prefix: string, ContinuationToken?: string): Promise<void>;
+    /**
+     * Ensures a bucket exists, creating it if it doesn't
+     */
+    ensureBucketExists(Bucket: string): Promise<void>;
+}
 //# sourceMappingURL=s3.lib.d.ts.map

package/dist/esm/libs/s3/s3.lib.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"s3.lib.d.ts","sourceRoot":"","sources":["../../../../src/libs/s3/s3.lib.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,6BAA6B,EAC7B,eAAe,EAAE,eAAe,EAChC,sBAAsB,EAAE,0BAA0B,EAInD,MAAM,oBAAoB,CAAA;AAK3B,OAAO,MAAM,MAAM,QAAQ,CAAA;AAoB3B,eAAO,MAAM,UAAU,GAAI,KAAK,MAAM,EAAE,QAAQ,MAAM,EAAE,MAAM,GAAG,EAAE,UAAU,IAAI,iEAOhF,CAAA;AAED,eAAO,MAAM,YAAY,GAAI,KAAK,MAAM,EAAE,QAAQ,MAAM,KAAG,OAAO,CAAC,eAAe,CAOjF,CAAA;AAED,eAAO,MAAM,oBAAoB,GAC/B,KAAK,MAAM,EACX,QAAQ,MAAM,KACb,OAAO,CAAC,MAAM,GAAG,SAAS,CAe5B,CAAA;AAED,eAAO,MAAM,uBAAuB,GAAU,KAAK,MAAM,EAAE,QAAQ,MAAM,KAAG,OAAO,CAAC,UAAU,GAAG,SAAS,CAezG,CAAA;AAED,eAAO,MAAM,YAAY,GAAI,KAAK,MAAM,EAAE,QAAQ,MAAM,EAAE,MAAM,eAAe;;aAqBjD,OAAO,CAAC,6BAA6B,CAAC;CAEnE,CAAA;AAED,eAAO,MAAM,qBAAqB,GAChC,KAAK,MAAM,EACX,QAAQ,MAAM,EACd,YAAY,MAAM,oBAenB,CAAA;AAED,eAAO,MAAM,uBAAuB,GAAU,KAAK,MAAM,EAAE,QAAQ,MAAM,oBAgBxE,CAAA;AAED,eAAO,MAAM,cAAc,GAAI,KAAK,MAAM,EAAE,QAAQ,MAAM,WAOzD,CAAA;AAED,eAAO,MAAM,IAAI,GACf,QAAQ,MAAM,EACd,SAAS,MAAM,EACf,OAAO,MAAM,kEAOd,CAAA;AAED,eAAO,MAAM,kBAAkB,GAC7B,YAAY,MAAM,EAClB,SAAS,MAAM,EACf,UAAU,MAAM,EAChB,OAAO,MAAM,kEAOd,CAAA;AAED,eAAO,MAAM,SAAS,GACpB,QAAQ,MAAM,EACd,KAAK,MAAM,KACV,OAAO,CAAC,sBAAsB,CAKhC,CAAA;AAED,eAAO,MAAM,mBAAmB,GAC9B,QAAQ,MAAM,EACd,QAAQ,MAAM,KACb,OAAO,CAAC,0BAA0B,CAOpC,CAAA;AAED,eAAO,MAAM,UAAU,GACrB,QAAQ,MAAM,EACd,KAAK,MAAM,KACV,OAAO,CAAC,sBAAsB,CAKhC,CAAA;AAED,eAAO,MAAM,YAAY,GACvB,QAAQ,MAAM,EACd,KAAK,MAAM,KACV,OAAO,CAAC,sBAAsB,CAKhC,CAAA;AAED,eAAO,MAAM,qBAAqB,GAChC,QAAQ,MAAM,EACd,QAAQ,MAAM,EACd,oBAAoB,MAAM,KACzB,OAAO,CAAC,IAAI,CAqBd,CAAA;AAED;;GAEG;AACH,eAAO,MAAM,kBAAkB,GAAU,QAAQ,MAAM,KAAG,OAAO,CAAC,IAAI,CAwBrE,CAAA"}
+{"version":3,"file":"s3.lib.d.ts","sourceRoot":"","sources":["../../../../src/libs/s3/s3.lib.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,6BAA6B,EAC7B,eAAe,EACf,eAAe,EACf,sBAAsB,EACtB,0BAA0B,EAI3B,MAAM,oBAAoB,CAAA;AAK3B,OAAO,MAAM,MAAM,QAAQ,CAAA;AAE3B,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAA;AAErC,qBAAa,KAAK;IAChB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAI;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAoB;IAE3C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAc;gBACzB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;IAehC,UAAU,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,IAAI;IASjE,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAS7D,oBAAoB,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAiB9E,uBAAuB,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;IAiB3F,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,eAAe;;iBAqBjC,OAAO,CAAC,6BAA6B,CAAC;;IAI9D,qBAAqB,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM;IAcrE,uBAAuB,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM;IAgBzD,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM;IAS1C,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAQnD,kBAAkB,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAQjF,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAOvE,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,0BAA0B,CAAC;IASxF,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAOxE,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAO1E,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,iBAAiB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBtG;;OAEG;IACG,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAyBxD"}

package/dist/esm/libs/s3/s3.lib.js

@@ -4,195 +4,200 @@ import { createRequest } from '@aws-sdk/util-create-request';
 import { formatUrl } from '@aws-sdk/util-format-url';
 import { Upload } from '@aws-sdk/lib-storage';
 import stream from 'stream';
-import { S3_HOST, S3_PRESIGNED_URL_EXPIRES, AWS_REGION, } from './config';
 import logger from '../logger';
-const S3Client = new S3({
-    ...S3_HOST ? {
-        endpoint: S3_HOST,
-        forcePathStyle: true,
-    }
-        : {},
-});
-const signer = new S3RequestPresigner({
-    ...S3Client.config,
-});
-export const uploadFile = (Key, Bucket, Body, Expires) => {
-    return S3Client.putObject({
-        Key,
-        Bucket,
-        Body,
-        Expires,
-    });
-};
-export const downloadFile = (Key, Bucket) => {
-    const command = new GetObjectCommand({
-        Bucket,
-        Key,
-    });
-    return S3Client.send(command);
-};
-export const downloadFileAsString = async (Key, Bucket) => {
-    try {
+export class S3Lib {
+    client;
+    signer;
+    config;
+    constructor(config) {
+        this.config = config;
+        this.client = new S3({
+            ...this.config.host ? {
+                endpoint: this.config.host,
+                forcePathStyle: true,
+            }
+                : {},
+        });
+        this.signer = new S3RequestPresigner({
+            ...this.client.config,
+        });
+    }
+    uploadFile(Key, Bucket, Body, Expires) {
+        return this.client.putObject({
+            Key,
+            Bucket,
+            Body,
+            Expires,
+        });
+    }
+    downloadFile(Key, Bucket) {
         const command = new GetObjectCommand({
             Bucket,
             Key,
         });
-        const response = await S3Client.send(command);
-        return response.Body?.transformToString();
+        return this.client.send(command);
     }
-    catch (err) {
-        if (err?.name === 'NoSuchKey') {
-            return undefined;
+    async downloadFileAsString(Key, Bucket) {
+        try {
+            const command = new GetObjectCommand({
+                Bucket,
+                Key,
+            });
+            const response = await this.client.send(command);
+            return response.Body?.transformToString();
+        }
+        catch (err) {
+            if (err?.name === 'NoSuchKey') {
+                return undefined;
+            }
+            throw err;
         }
-        throw err;
     }
-};
-export const downloadFileAsByteArray = async (Key, Bucket) => {
-    try {
-        const command = new GetObjectCommand({
+    async downloadFileAsByteArray(Key, Bucket) {
+        try {
+            const command = new GetObjectCommand({
+                Bucket,
+                Key,
+            });
+            const response = await this.client.send(command);
+            return response.Body?.transformToByteArray();
+        }
+        catch (err) {
+            if (err?.name === 'NoSuchKey') {
+                return undefined;
+            }
+            throw err;
+        }
+    }
+    streamUpload(Key, Bucket, ACL) {
+        const passThroughStream = new stream.PassThrough();
+        const options = {
+            client: this.client,
+            params: {
+                Bucket,
+                Key,
+                Body: passThroughStream,
+            },
+            leavePartsOnError: false,
+        };
+        if (ACL) {
+            options.params.ACL = ACL;
+        }
+        const upload = new Upload(options);
+        return {
+            writeStream: passThroughStream,
+            promise: upload.done(),
+        };
+    }
+    async getPresignedUploadUrl(Key, Bucket, expiresIn) {
+        const request = await createRequest(this.client, new PutObjectCommand({ Key, Bucket }));
+        return formatUrl(await this.signer.presign(request, {
+            expiresIn: expiresIn || this.config.presignedUrlExpires || 120,
+        }));
+    }
+    async getPresignedDownloadUrl(Key, Bucket) {
+        const request = await createRequest(this.client, new GetObjectCommand({ Key, Bucket }));
+        return formatUrl(await this.signer.presign(request, {
+            expiresIn: this.config.presignedUrlExpires || 120,
+        }));
+    }
+    getDownloadUrl(Key, Bucket) {
+        if (this.config.host) {
+            return `${this.config.host}/${Bucket}/${Key}`;
+        }
+        const regionString = this.config.region?.includes('us-east-1') ? '' : '-' + this.config.region;
+        return `https://${Bucket}.s3${regionString}.amazonaws.com/${Key}`;
+    }
+    copy(Bucket, KeyFrom, KeyTo) {
+        return this.client.copyObject({
+            CopySource: `/${Bucket}/${KeyFrom}`,
+            Bucket,
+            Key: KeyTo,
+        });
+    }
+    copyBetweenBuckets(BucketFrom, KeyFrom, BucketTo, KeyTo) {
+        return this.client.copyObject({
+            CopySource: `/${BucketFrom}/${KeyFrom}`,
+            Bucket: BucketTo,
+            Key: KeyTo,
+        });
+    }
+    async getObject(Bucket, Key) {
+        return this.client.getObject({
             Bucket,
             Key,
         });
-        const response = await S3Client.send(command);
-        return response.Body?.transformToByteArray();
     }
-    catch (err) {
-        if (err?.name === 'NoSuchKey') {
-            return undefined;
-        }
-        throw err;
-    }
-};
-export const streamUpload = (Key, Bucket, ACL) => {
-    const passThroughStream = new stream.PassThrough();
-    const options = {
-        client: S3Client,
-        params: {
+    async listObjectsByPrefix(Bucket, Prefix) {
+        return this.client.send(new ListObjectsV2Command({
+            Bucket,
+            Prefix,
+        }));
+    }
+    async headObject(Bucket, Key) {
+        return this.client.headObject({
             Bucket,
             Key,
-            Body: passThroughStream,
-        },
-        leavePartsOnError: false,
-    };
-    if (ACL) {
-        options.params.ACL = ACL;
-    }
-    const upload = new Upload(options);
-    return {
-        writeStream: passThroughStream,
-        promise: upload.done(),
-    };
-};
-export const getPresignedUploadUrl = async (Key, Bucket, expiresIn) => {
-    const request = await createRequest(S3Client, new PutObjectCommand({ Key, Bucket }));
-    const signedUrl = formatUrl(await signer.presign(request, {
-        expiresIn: expiresIn || S3_PRESIGNED_URL_EXPIRES,
-    }));
-    return signedUrl;
-};
-export const getPresignedDownloadUrl = async (Key, Bucket) => {
-    const request = await createRequest(S3Client, new GetObjectCommand({ Key, Bucket }));
-    const signedUrl = formatUrl(await signer.presign(request, {
-        expiresIn: S3_PRESIGNED_URL_EXPIRES,
-    }));
-    return signedUrl;
-};
-export const getDownloadUrl = (Key, Bucket) => {
-    if (S3_HOST) {
-        return `${S3_HOST}/${Bucket}/${Key}`;
-    }
-    const regionString = AWS_REGION.includes('us-east-1') ? '' : '-' + AWS_REGION;
-    return `https://${Bucket}.s3${regionString}.amazonaws.com/${Key}`;
-};
-export const copy = (Bucket, KeyFrom, KeyTo) => {
-    return S3Client.copyObject({
-        CopySource: `/${Bucket}/${KeyFrom}`,
-        Bucket,
-        Key: KeyTo,
-    });
-};
-export const copyBetweenBuckets = (BucketFrom, KeyFrom, BucketTo, KeyTo) => {
-    return S3Client.copyObject({
-        CopySource: `/${BucketFrom}/${KeyFrom}`,
-        Bucket: BucketTo,
-        Key: KeyTo,
-    });
-};
-export const getObject = async (Bucket, Key) => {
-    return S3Client.getObject({
-        Bucket,
-        Key,
-    });
-};
-export const listObjectsByPrefix = async (Bucket, Prefix) => {
-    return S3Client.send(new ListObjectsV2Command({
-        Bucket,
-        Prefix,
-    }));
-};
-export const headObject = async (Bucket, Key) => {
-    return S3Client.headObject({
-        Bucket,
-        Key,
-    });
-};
-export const deleteObject = async (Bucket, Key) => {
-    return S3Client.deleteObject({
-        Bucket,
-        Key,
-    });
-};
-export const deleteObjectsByPrefix = async (Bucket, Prefix, ContinuationToken) => {
-    const objects = await S3Client.listObjectsV2({ Bucket, Prefix, ContinuationToken });
-    if (!objects.Contents?.length)
-        return;
-    const deleteParams = {
-        Bucket,
-        Delete: { Objects: [] },
-    };
-    objects.Contents.forEach(({ Key }) => {
-        if (!Key)
+        });
+    }
+    async deleteObject(Bucket, Key) {
+        return this.client.deleteObject({
+            Bucket,
+            Key,
+        });
+    }
+    async deleteObjectsByPrefix(Bucket, Prefix, ContinuationToken) {
+        const objects = await this.client.listObjectsV2({ Bucket, Prefix, ContinuationToken });
+        if (!objects.Contents?.length)
             return;
-        deleteParams.Delete.Objects.push({ Key });
-    });
-    logger.info(deleteParams, '[S3] Deleting objects');
-    const deletedResp = await S3Client.deleteObjects(deleteParams);
-    if (deletedResp.Errors?.length) {
-        logger.error(deletedResp.Errors, '[S3] Error during deleteObjectsByPrefix');
-    }
-    if (objects.IsTruncated) {
-        await deleteObjectsByPrefix(Bucket, Prefix, objects.NextContinuationToken);
-    }
-};
-/**
- * Ensures a bucket exists, creating it if it doesn't
- */
-export const ensureBucketExists = async (Bucket) => {
-    try {
-        await S3Client.send(new HeadBucketCommand({ Bucket }));
-        // Bucket exists
-        return;
-    }
-    catch (error) {
-        // If bucket doesn't exist, create it
-        if (error.$metadata?.httpStatusCode === 404) {
-            try {
-                await S3Client.send(new CreateBucketCommand({ Bucket }));
-                logger.info({ Bucket }, '[S3] Created bucket');
-            }
-            catch (createError) {
-                // Ignore if bucket was created by another process
-                if (createError.$metadata?.httpStatusCode !== 409) {
-                    logger.error({ Bucket, error: createError }, '[S3] Failed to create bucket');
-                    throw createError;
+        const deleteParams = {
+            Bucket,
+            Delete: { Objects: [] },
+        };
+        objects.Contents.forEach(({ Key }) => {
+            if (!Key)
+                return;
+            deleteParams.Delete.Objects.push({ Key });
+        });
+        logger.info(deleteParams, '[S3] Deleting objects');
+        const deletedResp = await this.client.deleteObjects(deleteParams);
+        if (deletedResp.Errors?.length) {
+            logger.error(deletedResp.Errors, '[S3] Error during deleteObjectsByPrefix');
+        }
+        if (objects.IsTruncated) {
+            await this.deleteObjectsByPrefix(Bucket, Prefix, objects.NextContinuationToken);
+        }
+    }
+    /**
+     * Ensures a bucket exists, creating it if it doesn't
+     */
+    async ensureBucketExists(Bucket) {
+        try {
+            await this.client.send(new HeadBucketCommand({ Bucket }));
+            // Bucket exists
+            return;
+        }
+        catch (error) {
+            // If bucket doesn't exist, create it
+            if (error.$metadata?.httpStatusCode === 404) {
+                try {
+                    await this.client.send(new CreateBucketCommand({ Bucket }));
+                    logger.info({ Bucket }, '[S3] Created bucket');
+                }
+                catch (createError) {
+                    // Ignore if bucket was created by another process
+                    if (createError.$metadata?.httpStatusCode !== 409) {
+                        logger.error({ Bucket, error: createError }, '[S3] Failed to create bucket');
+                        throw createError;
+                    }
                 }
             }
-        }
-        else {
-            // Some other error (permissions, etc.)
-            logger.error({ Bucket, error }, '[S3] Failed to check bucket existence');
-            throw error;
+            else {
+                // Some other error (permissions, etc.)
+                logger.error({ Bucket, error }, '[S3] Failed to check bucket existence');
+                throw error;
+            }
         }
     }
-};
+}
 //# sourceMappingURL=s3.lib.js.map