codecane 1.0.123

package/dist/common/actions.js

@@ -0,0 +1,193 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SERVER_ACTION_SCHEMA = exports.ResponseCompleteSchema = exports.InitResponseSchema = exports.UsageReponseSchema = exports.CLIENT_ACTION_SCHEMA = exports.ToolCallSchema = exports.CHANGES = exports.FileChangeSchema = void 0;
+const zod_1 = require("zod");
+const file_1 = require("./util/file");
+const credentials_1 = require("./util/credentials");
+const constants_1 = require("./constants");
+const MessageContentObjectSchema = zod_1.z.union([
+    zod_1.z.object({
+        type: zod_1.z.literal('text'),
+        text: zod_1.z.string(),
+        cache_control: zod_1.z
+            .object({
+            type: zod_1.z.literal('ephemeral'),
+        })
+            .optional(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('tool_use'),
+        id: zod_1.z.string(),
+        name: zod_1.z.string(),
+        input: zod_1.z.record(zod_1.z.string(), zod_1.z.any()),
+        cache_control: zod_1.z
+            .object({
+            type: zod_1.z.literal('ephemeral'),
+        })
+            .optional(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('tool_result'),
+        tool_use_id: zod_1.z.string(),
+        content: zod_1.z.string(),
+        cache_control: zod_1.z
+            .object({
+            type: zod_1.z.literal('ephemeral'),
+        })
+            .optional(),
+    }),
+]);
+const MessageSchema = zod_1.z.object({
+    role: zod_1.z.union([zod_1.z.literal('user'), zod_1.z.literal('assistant')]),
+    content: zod_1.z.union([zod_1.z.string(), zod_1.z.array(MessageContentObjectSchema)]),
+});
+exports.FileChangeSchema = zod_1.z.object({
+    type: zod_1.z.enum(['patch', 'file']),
+    filePath: zod_1.z.string(),
+    content: zod_1.z.string(),
+});
+exports.CHANGES = zod_1.z.array(exports.FileChangeSchema);
+exports.ToolCallSchema = zod_1.z.object({
+    name: zod_1.z.string(),
+    id: zod_1.z.string(),
+    input: zod_1.z.record(zod_1.z.string(), zod_1.z.any()),
+});
+exports.CLIENT_ACTION_SCHEMA = zod_1.z.discriminatedUnion('type', [
+    zod_1.z.object({
+        type: zod_1.z.literal('user-input'),
+        fingerprintId: zod_1.z.string(),
+        authToken: zod_1.z.string().optional(),
+        userInputId: zod_1.z.string(),
+        messages: zod_1.z.array(MessageSchema),
+        fileContext: file_1.ProjectFileContextSchema,
+        changesAlreadyApplied: exports.CHANGES,
+        costMode: zod_1.z.enum(constants_1.costModes).optional().default('normal'),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('read-files-response'),
+        files: zod_1.z.record(zod_1.z.string(), zod_1.z.union([zod_1.z.string(), zod_1.z.null()])),
+    }),
+    // z.object({
+    //   type: z.literal('run-terminal-command'),
+    //   command: z.string(),
+    // }),
+    zod_1.z.object({
+        type: zod_1.z.literal('init'),
+        fingerprintId: zod_1.z.string(),
+        authToken: zod_1.z.string().optional(),
+        // userId: z.string().optional(),
+        fileContext: file_1.ProjectFileContextSchema,
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('usage'),
+        fingerprintId: zod_1.z.string(),
+        authToken: zod_1.z.string().optional(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('login-code-request'),
+        fingerprintId: zod_1.z.string(),
+        referralCode: zod_1.z.string().optional(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('login-status-request'),
+        fingerprintId: zod_1.z.string(),
+        fingerprintHash: zod_1.z.string(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('clear-auth-token'),
+        authToken: zod_1.z.string(),
+        fingerprintId: zod_1.z.string(),
+        userId: zod_1.z.string(),
+        // authToken: z.string().optional(),
+        fingerprintHash: zod_1.z.string(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('generate-commit-message'),
+        fingerprintId: zod_1.z.string(),
+        authToken: zod_1.z.string().optional(),
+        stagedChanges: zod_1.z.string(),
+    }),
+]);
+exports.UsageReponseSchema = zod_1.z.object({
+    type: zod_1.z.literal('usage-response'),
+    usage: zod_1.z.number(),
+    limit: zod_1.z.number(),
+    referralLink: zod_1.z.string().optional(),
+    subscription_active: zod_1.z.boolean(),
+    next_quota_reset: zod_1.z.coerce.date(),
+    session_credits_used: zod_1.z.number(),
+});
+exports.InitResponseSchema = zod_1.z
+    .object({
+    type: zod_1.z.literal('init-response'),
+})
+    .merge(exports.UsageReponseSchema.omit({
+    type: true,
+}));
+exports.ResponseCompleteSchema = zod_1.z
+    .object({
+    type: zod_1.z.literal('response-complete'),
+    userInputId: zod_1.z.string(),
+    response: zod_1.z.string(),
+    changes: exports.CHANGES,
+    changesAlreadyApplied: exports.CHANGES,
+    addedFileVersions: zod_1.z.array(file_1.FileVersionSchema),
+    resetFileVersions: zod_1.z.boolean(),
+})
+    .merge(exports.UsageReponseSchema.omit({
+    type: true,
+}).partial());
+exports.SERVER_ACTION_SCHEMA = zod_1.z.discriminatedUnion('type', [
+    zod_1.z.object({
+        type: zod_1.z.literal('response-chunk'),
+        userInputId: zod_1.z.string(),
+        chunk: zod_1.z.string(),
+    }),
+    exports.ResponseCompleteSchema,
+    zod_1.z.object({
+        type: zod_1.z.literal('read-files'),
+        filePaths: zod_1.z.array(zod_1.z.string()),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('tool-call'),
+        userInputId: zod_1.z.string(),
+        response: zod_1.z.string(),
+        data: exports.ToolCallSchema,
+        changes: exports.CHANGES,
+        changesAlreadyApplied: exports.CHANGES,
+        addedFileVersions: zod_1.z.array(file_1.FileVersionSchema),
+        resetFileVersions: zod_1.z.boolean(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('terminal-command-result'),
+        userInputId: zod_1.z.string(),
+        result: zod_1.z.string(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('npm-version-status'),
+        isUpToDate: zod_1.z.boolean(),
+        latestVersion: zod_1.z.string(),
+    }),
+    exports.InitResponseSchema,
+    zod_1.z.object({
+        type: zod_1.z.literal('auth-result'),
+        user: credentials_1.userSchema.optional(),
+        message: zod_1.z.string(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('login-code-response'),
+        fingerprintId: zod_1.z.string(),
+        fingerprintHash: zod_1.z.string(),
+        loginUrl: zod_1.z.string().url(),
+    }),
+    exports.UsageReponseSchema,
+    zod_1.z.object({
+        type: zod_1.z.literal('action-error'),
+        message: zod_1.z.string(),
+    }),
+    zod_1.z.object({
+        type: zod_1.z.literal('commit-message-response'),
+        commitMessage: zod_1.z.string(),
+    }),
+]);
+//# sourceMappingURL=actions.js.map

package/dist/common/actions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"actions.js","sourceRoot":"","sources":["../src/actions.ts"],"names":[],"mappings":";;;AAAA,6BAAuB;AACvB,sCAAyE;AACzE,oDAA+C;AAC/C,2CAAuC;AAEvC,MAAM,0BAA0B,GAAG,OAAC,CAAC,KAAK,CAAC;IACzC,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,MAAM,CAAC;QACvB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;QAChB,aAAa,EAAE,OAAC;aACb,MAAM,CAAC;YACN,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC;SAC7B,CAAC;aACD,QAAQ,EAAE;KACd,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,UAAU,CAAC;QAC3B,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE;QACd,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;QAChB,KAAK,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC;QACpC,aAAa,EAAE,OAAC;aACb,MAAM,CAAC;YACN,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC;SAC7B,CAAC;aACD,QAAQ,EAAE;KACd,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,aAAa,CAAC;QAC9B,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;QACvB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;QACnB,aAAa,EAAE,OAAC;aACb,MAAM,CAAC;YACN,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC;SAC7B,CAAC;aACD,QAAQ,EAAE;KACd,CAAC;CACH,CAAC,CAAA;AAEF,MAAM,aAAa,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7B,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC1D,OAAO,EAAE,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC,CAAC;CACpE,CAAC,CAAA;AAIW,QAAA,gBAAgB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,IAAI,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC/B,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;IACpB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;CACpB,CAAC,CAAA;AAEW,QAAA,OAAO,GAAG,OAAC,CAAC,KAAK,CAAC,wBAAgB,CAAC,CAAA;AAGnC,QAAA,cAAc,GAAG,OAAC,CAAC,MAAM,CAAC;IACrC,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;IAChB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE;IACd,KAAK,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC;CACrC,CAAC,CAAA;AAGW,QAAA,oBAAoB,GAAG,OAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE;IAC/D,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,YAAY,CAAC;QAC7B,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;QACzB,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAChC,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;QACvB,QAAQ,EAAE,OAAC,CAAC,KAAK,CAAC,aAAa,CAAC;QAChC,WAAW,EAAE,+BAAwB;QACrC,qBAAqB,EAAE,eAAO;QAC9B,QAAQ,EAAE,OAAC,CAAC,IAAI,CAAC,qBAAS,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC;KACzD,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC;QACtC,KAAK,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;KAC7D,CAAC;IACF,aAAa;IACb,6CAA6C;IAC7C,yBAAyB;IACzB,MAAM;IACN,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,MAAM,CAAC;QACvB,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;QACzB,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAChC,iCAAiC;QACjC,WAAW,EAAE,+BAAwB;KACtC,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QACxB,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;QACzB,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KACjC,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC;QACrC,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;QACzB,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KACpC,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC;QACvC,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;QACzB,eAAe,EAAE,OAAC,CAAC,MAAM,EAAE;KAC5B,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC;QACnC,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE;QACrB,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;QACzB,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE;QAClB,oCAAoC;QACpC,eAAe,EAAE,OAAC,CAAC,MAAM,EAAE;KAC5B,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,yBAAyB,CAAC;QAC1C,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;QACzB,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAChC,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;KAC1B,CAAC;CACH,CAAC,CAAA;AAGW,QAAA,kBAAkB,GAAG,OAAC,CAAC,MAAM,CAAC;IACzC,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;IACjC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;IACjB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;IACjB,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACnC,mBAAmB,EAAE,OAAC,CAAC,OAAO,EAAE;IAChC,gBAAgB,EAAE,OAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IACjC,oBAAoB,EAAE,OAAC,CAAC,MAAM,EAAE;CACjC,CAAC,CAAA;AAGW,QAAA,kBAAkB,GAAG,OAAC;KAChC,MAAM,CAAC;IACN,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,eAAe,CAAC;CACjC,CAAC;KACD,KAAK,CACJ,0BAAkB,CAAC,IAAI,CAAC;IACtB,IAAI,EAAE,IAAI;CACX,CAAC,CACH,CAAA;AAGU,QAAA,sBAAsB,GAAG,OAAC;KACpC,MAAM,CAAC;IACN,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC;IACpC,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;IACvB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;IACpB,OAAO,EAAE,eAAO;IAChB,qBAAqB,EAAE,eAAO;IAC9B,iBAAiB,EAAE,OAAC,CAAC,KAAK,CAAC,wBAAiB,CAAC;IAC7C,iBAAiB,EAAE,OAAC,CAAC,OAAO,EAAE;CAC/B,CAAC;KACD,KAAK,CACJ,0BAAkB,CAAC,IAAI,CAAC;IACtB,IAAI,EAAE,IAAI;CACX,CAAC,CAAC,OAAO,EAAE,CACb,CAAA;AAEU,QAAA,oBAAoB,GAAG,OAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE;IAC/D,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;QACjC,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;QACvB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;KAClB,CAAC;IACF,8BAAsB;IACtB,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,YAAY,CAAC;QAC7B,SAAS,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;KAC/B,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC;QAC5B,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;QACvB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;QACpB,IAAI,EAAE,sBAAc;QACpB,OAAO,EAAE,eAAO;QAChB,qBAAqB,EAAE,eAAO;QAC9B,iBAAiB,EAAE,OAAC,CAAC,KAAK,CAAC,wBAAiB,CAAC;QAC7C,iBAAiB,EAAE,OAAC,CAAC,OAAO,EAAE;KAC/B,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,yBAAyB,CAAC;QAC1C,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;QACvB,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE;KACnB,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC;QACrC,UAAU,EAAE,OAAC,CAAC,OAAO,EAAE;QACvB,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;KAC1B,CAAC;IACF,0BAAkB;IAClB,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,aAAa,CAAC;QAC9B,IAAI,EAAE,wBAAU,CAAC,QAAQ,EAAE;QAC3B,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;KACpB,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC;QACtC,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;QACzB,eAAe,EAAE,OAAC,CAAC,MAAM,EAAE;QAC3B,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE;KAC3B,CAAC;IACF,0BAAkB;IAClB,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,cAAc,CAAC;QAC/B,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;KACpB,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,yBAAyB,CAAC;QAC1C,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;KAC1B,CAAC;CACH,CAAC,CAAA"}

package/dist/common/billing/quota-manager.d.ts

@@ -0,0 +1,29 @@
+type CheckQuotaResult = Promise<{
+    creditsUsed: number;
+    quota: number;
+    endDate: Date;
+    subscription_active: boolean;
+    session_credits_used?: number;
+}>;
+export interface IQuotaManager {
+    checkQuota(id: string): CheckQuotaResult;
+    setNextQuota(id: string, quota_exceeded: boolean, next_quota_reset: Date): Promise<void>;
+}
+export declare class AnonymousQuotaManager implements IQuotaManager {
+    checkQuota(fingerprintId: string, sessionId?: string): CheckQuotaResult;
+    setNextQuota(fingerprintId: string, quota_exceeded: boolean, next_quota_reset: Date): Promise<void>;
+}
+export declare class AuthenticatedQuotaManager implements IQuotaManager {
+    getStripeSubscriptionQuota(userId: string): Promise<{
+        quota: number;
+        overageRate: number | null;
+    }>;
+    checkQuota(userId: string, sessionId?: string): CheckQuotaResult;
+    setNextQuota(userId: string, quota_exceeded: boolean, next_quota_reset: Date): Promise<void>;
+}
+export type AuthType = 'anonymous' | 'authenticated';
+export declare const getQuotaManager: (authType: AuthType, id: string) => {
+    checkQuota: (sessionId?: string) => CheckQuotaResult;
+    setNextQuota: (quota_exceeded: boolean, next_quota_reset: Date) => Promise<void>;
+};
+export {};

package/dist/common/billing/quota-manager.js

@@ -0,0 +1,213 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getQuotaManager = exports.AuthenticatedQuotaManager = exports.AnonymousQuotaManager = void 0;
+const stripe_1 = require("../util/stripe");
+const constants_1 = require("../constants");
+const db_1 = __importDefault(require("../db"));
+const schema = __importStar(require("../db/schema"));
+const drizzle_orm_1 = require("drizzle-orm");
+const ts_pattern_1 = require("ts-pattern");
+class AnonymousQuotaManager {
+    async checkQuota(fingerprintId, sessionId) {
+        const quota = constants_1.CREDITS_USAGE_LIMITS.ANON;
+        const startDate = (0, drizzle_orm_1.sql) `COALESCE(${schema.fingerprint.next_quota_reset}, now()) - INTERVAL '1 month'`;
+        const endDate = (0, drizzle_orm_1.sql) `COALESCE(${schema.fingerprint.next_quota_reset}, now())`;
+        let session_credits_used = undefined;
+        const result = await db_1.default
+            .select({
+            creditsUsed: (0, drizzle_orm_1.sql) `SUM(COALESCE(${schema.message.credits}, 0))`,
+            endDate,
+        })
+            .from(schema.fingerprint)
+            .leftJoin(schema.message, (0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema.message.fingerprint_id, fingerprintId), (0, drizzle_orm_1.between)(schema.message.finished_at, startDate, endDate)))
+            .where((0, drizzle_orm_1.eq)(schema.fingerprint.id, fingerprintId))
+            .groupBy(schema.fingerprint.next_quota_reset)
+            .then((rows) => {
+            if (rows.length > 0)
+                return rows[0];
+            return {
+                creditsUsed: '0',
+                endDate: new Date().toDateString(),
+            };
+        });
+        if (sessionId) {
+            session_credits_used = await db_1.default
+                .select({
+                client_id: schema.message.client_id,
+                fingerprint_id: schema.message.fingerprint_id,
+                sessionCreditsUsed: (0, drizzle_orm_1.sql) `SUM(COALESCE(${schema.message.credits}, 0))`,
+            })
+                .from(schema.message)
+                .where((0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema.message.client_id, sessionId), (0, drizzle_orm_1.eq)(schema.message.fingerprint_id, fingerprintId)))
+                .groupBy(schema.message.client_id, schema.message.fingerprint_id)
+                .then((rows) => {
+                if (rows.length > 0) {
+                    return parseInt(rows[0].sessionCreditsUsed);
+                }
+                return 0;
+            });
+        }
+        return {
+            creditsUsed: parseInt(result.creditsUsed),
+            quota,
+            endDate: new Date(result.endDate),
+            subscription_active: false,
+            session_credits_used,
+        };
+    }
+    async setNextQuota(fingerprintId, quota_exceeded, next_quota_reset) {
+        await db_1.default
+            .update(schema.fingerprint)
+            .set({
+            quota_exceeded,
+            next_quota_reset,
+        })
+            .where((0, drizzle_orm_1.eq)(schema.fingerprint.id, fingerprintId));
+    }
+}
+exports.AnonymousQuotaManager = AnonymousQuotaManager;
+class AuthenticatedQuotaManager {
+    async getStripeSubscriptionQuota(userId) {
+        // Get user's subscription from Stripe
+        const user = await db_1.default.query.user.findFirst({
+            where: (0, drizzle_orm_1.eq)(schema.user.id, userId),
+            columns: {
+                stripe_customer_id: true,
+                stripe_price_id: true,
+            },
+        });
+        let overageRate = null;
+        let quota = constants_1.CREDITS_USAGE_LIMITS.PAID;
+        if (user?.stripe_customer_id && user?.stripe_price_id) {
+            const subscriptions = await stripe_1.stripeServer.subscriptions.list({
+                customer: user.stripe_customer_id,
+                status: 'active',
+                limit: 1,
+            });
+            if (subscriptions.data[0]?.id) {
+                const subscription = await stripe_1.stripeServer.subscriptions.retrieve(subscriptions.data[0].id, {
+                    expand: ['items.data.price.tiers'],
+                });
+                // Get the metered price item which contains our overage rate
+                const meteredPrice = subscription.items.data.find((item) => item.price.recurring?.usage_type === 'metered');
+                if (meteredPrice?.price.tiers) {
+                    for (const tier of meteredPrice.price.tiers) {
+                        if (tier.up_to) {
+                            quota = Math.max(quota, tier.up_to);
+                        }
+                        if (tier.up_to === null && tier.unit_amount_decimal) {
+                            overageRate = parseFloat(tier.unit_amount_decimal);
+                            break;
+                        }
+                    }
+                }
+            }
+        }
+        return { quota, overageRate };
+    }
+    async checkQuota(userId, sessionId) {
+        const startDate = (0, drizzle_orm_1.sql) `COALESCE(${schema.user.next_quota_reset}, now()) - INTERVAL '1 month'`;
+        const endDate = (0, drizzle_orm_1.sql) `COALESCE(${schema.user.next_quota_reset}, now())`;
+        let session_credits_used = undefined;
+        const result = await db_1.default
+            .select({
+            quota: schema.user.quota,
+            stripe_customer_id: schema.user.stripe_customer_id,
+            stripe_price_id: schema.user.stripe_price_id,
+            subscription_active: schema.user.subscription_active,
+            endDate,
+            creditsUsed: (0, drizzle_orm_1.sql) `SUM(COALESCE(${schema.message.credits}, 0))`,
+        })
+            .from(schema.user)
+            .leftJoin(schema.message, (0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema.message.user_id, schema.user.id), (0, drizzle_orm_1.between)(schema.message.finished_at, startDate, endDate)))
+            .where((0, drizzle_orm_1.eq)(schema.user.id, userId))
+            .groupBy(schema.user.quota, schema.user.stripe_customer_id, schema.user.stripe_price_id, schema.user.subscription_active, schema.user.next_quota_reset)
+            .then((rows) => {
+            if (rows.length > 0)
+                return rows[0];
+            return {
+                quota: 0,
+                stripe_customer_id: null,
+                stripe_price_id: null,
+                creditsUsed: '0',
+                endDate: new Date().toDateString(),
+                subscription_active: false,
+            };
+        });
+        if (sessionId) {
+            session_credits_used = await db_1.default
+                .select({
+                client_id: schema.message.client_id,
+                user_id: schema.message.user_id,
+                sessionCreditsUsed: (0, drizzle_orm_1.sql) `SUM(COALESCE(${schema.message.credits}, 0))`,
+            })
+                .from(schema.message)
+                .where((0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(schema.message.client_id, sessionId), (0, drizzle_orm_1.eq)(schema.message.user_id, userId)))
+                .groupBy(schema.message.client_id, schema.message.user_id)
+                .then((rows) => {
+                if (rows.length > 0) {
+                    return parseInt(rows[0].sessionCreditsUsed);
+                }
+                return 0;
+            });
+        }
+        const quota = !result?.stripe_customer_id && !result?.stripe_price_id
+            ? constants_1.CREDITS_USAGE_LIMITS.FREE
+            : result?.quota ?? constants_1.CREDITS_USAGE_LIMITS.FREE;
+        return {
+            creditsUsed: parseInt(result.creditsUsed),
+            quota,
+            endDate: new Date(result.endDate),
+            subscription_active: !!result.subscription_active,
+            session_credits_used,
+        };
+    }
+    async setNextQuota(userId, quota_exceeded, next_quota_reset) {
+        await db_1.default
+            .update(schema.user)
+            .set({
+            quota_exceeded,
+            next_quota_reset,
+        })
+            .where((0, drizzle_orm_1.eq)(schema.user.id, userId));
+    }
+}
+exports.AuthenticatedQuotaManager = AuthenticatedQuotaManager;
+const getQuotaManager = (authType, id) => {
+    const manager = (0, ts_pattern_1.match)(authType)
+        .with('anonymous', () => new AnonymousQuotaManager())
+        .with('authenticated', () => new AuthenticatedQuotaManager())
+        .exhaustive();
+    return {
+        checkQuota: (sessionId) => manager.checkQuota(id, sessionId),
+        setNextQuota: (quota_exceeded, next_quota_reset) => manager.setNextQuota(id, quota_exceeded, next_quota_reset),
+    };
+};
+exports.getQuotaManager = getQuotaManager;
+//# sourceMappingURL=quota-manager.js.map

package/dist/common/billing/quota-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-manager.js","sourceRoot":"","sources":["../../src/billing/quota-manager.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CAA6C;AAC7C,4CAAmD;AACnD,+CAAsB;AACtB,qDAAsC;AACtC,6CAAwD;AACxD,2CAAkC;AAmBlC,MAAa,qBAAqB;IAChC,KAAK,CAAC,UAAU,CACd,aAAqB,EACrB,SAAkB;QAElB,MAAM,KAAK,GAAG,gCAAoB,CAAC,IAAI,CAAA;QACvC,MAAM,SAAS,GAAgB,IAAA,iBAAG,EAAQ,YAAY,MAAM,CAAC,WAAW,CAAC,gBAAgB,+BAA+B,CAAA;QACxH,MAAM,OAAO,GAAgB,IAAA,iBAAG,EAAQ,YAAY,MAAM,CAAC,WAAW,CAAC,gBAAgB,UAAU,CAAA;QACjG,IAAI,oBAAoB,GAAuB,SAAS,CAAA;QAExD,MAAM,MAAM,GAAG,MAAM,YAAE;aACpB,MAAM,CAAC;YACN,WAAW,EAAE,IAAA,iBAAG,EAAQ,gBAAgB,MAAM,CAAC,OAAO,CAAC,OAAO,OAAO;YACrE,OAAO;SACR,CAAC;aACD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;aACxB,QAAQ,CACP,MAAM,CAAC,OAAO,EACd,IAAA,iBAAG,EACD,IAAA,gBAAE,EAAC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,aAAa,CAAC,EAChD,IAAA,qBAAO,EAAC,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,EAAE,OAAO,CAAC,CACxD,CACF;aACA,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,WAAW,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC;aAC/C,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC;aAC5C,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACb,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC,CAAC,CAAC,CAAA;YACnC,OAAO;gBACL,WAAW,EAAE,GAAG;gBAChB,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,YAAY,EAAE;aACnC,CAAA;QACH,CAAC,CAAC,CAAA;QAEJ,IAAI,SAAS,EAAE,CAAC;YACd,oBAAoB,GAAG,MAAM,YAAE;iBAC5B,MAAM,CAAC;gBACN,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS;gBACnC,cAAc,EAAE,MAAM,CAAC,OAAO,CAAC,cAAc;gBAC7C,kBAAkB,EAAE,IAAA,iBAAG,EAAQ,gBAAgB,MAAM,CAAC,OAAO,CAAC,OAAO,OAAO;aAC7E,CAAC;iBACD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;iBACpB,KAAK,CACJ,IAAA,iBAAG,EACD,IAAA,gBAAE,EAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,EACvC,IAAA,gBAAE,EAAC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,aAAa,CAAC,CACjD,CACF;iBACA,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC;iBAChE,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;gBACb,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpB,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAA;gBAC7C,CAAC;gBACD,OAAO,CAAC,CAAA;YACV,CAAC,CAAC,CAAA;QACN,CAAC;QAED,OAAO;YACL,WAAW,EAAE,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC;YACzC,KAAK;YACL,OAAO,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACjC,mBAAmB,EAAE,KAAK;YAC1B,oBAAoB;SACrB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,aAAqB,EACrB,cAAuB,EACvB,gBAAsB;QAEtB,MAAM,YAAE;aACL,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC;aAC1B,GAAG,CAAC;YACH,cAAc;YACd,gBAAgB;SACjB,CAAC;aACD,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,WAAW,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC,CAAA;IACpD,CAAC;CACF;AA9ED,sDA8EC;AAED,MAAa,yBAAyB;IACpC,KAAK,CAAC,0BAA0B,CAC9B,MAAc;QAEd,sCAAsC;QACtC,MAAM,IAAI,GAAG,MAAM,YAAE,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;YACzC,KAAK,EAAE,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC;YACjC,OAAO,EAAE;gBACP,kBAAkB,EAAE,IAAI;gBACxB,eAAe,EAAE,IAAI;aACtB;SACF,CAAC,CAAA;QAEF,IAAI,WAAW,GAAkB,IAAI,CAAA;QACrC,IAAI,KAAK,GAAG,gCAAoB,CAAC,IAAI,CAAA;QACrC,IAAI,IAAI,EAAE,kBAAkB,IAAI,IAAI,EAAE,eAAe,EAAE,CAAC;YACtD,MAAM,aAAa,GAAG,MAAM,qBAAY,CAAC,aAAa,CAAC,IAAI,CAAC;gBAC1D,QAAQ,EAAE,IAAI,CAAC,kBAAkB;gBACjC,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,CAAC;aACT,CAAC,CAAA;YAEF,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC;gBAC9B,MAAM,YAAY,GAAG,MAAM,qBAAY,CAAC,aAAa,CAAC,QAAQ,CAC5D,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EACxB;oBACE,MAAM,EAAE,CAAC,wBAAwB,CAAC;iBACnC,CACF,CAAA;gBAED,6DAA6D;gBAC7D,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAC/C,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,KAAK,SAAS,CACzD,CAAA;gBAED,IAAI,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;oBAC9B,KAAK,MAAM,IAAI,IAAI,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;wBAC5C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;4BACf,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;wBACrC,CAAC;wBACD,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;4BACpD,WAAW,GAAG,UAAU,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;4BAClD,MAAK;wBACP,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,CAAA;IAC/B,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,SAAkB;QACjD,MAAM,SAAS,GAAgB,IAAA,iBAAG,EAAQ,YAAY,MAAM,CAAC,IAAI,CAAC,gBAAgB,+BAA+B,CAAA;QACjH,MAAM,OAAO,GAAgB,IAAA,iBAAG,EAAQ,YAAY,MAAM,CAAC,IAAI,CAAC,gBAAgB,UAAU,CAAA;QAC1F,IAAI,oBAAoB,GAAG,SAAS,CAAA;QAEpC,MAAM,MAAM,GAAG,MAAM,YAAE;aACpB,MAAM,CAAC;YACN,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK;YACxB,kBAAkB,EAAE,MAAM,CAAC,IAAI,CAAC,kBAAkB;YAClD,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,eAAe;YAC5C,mBAAmB,EAAE,MAAM,CAAC,IAAI,CAAC,mBAAmB;YACpD,OAAO;YACP,WAAW,EAAE,IAAA,iBAAG,EAAQ,gBAAgB,MAAM,CAAC,OAAO,CAAC,OAAO,OAAO;SACtE,CAAC;aACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;aACjB,QAAQ,CACP,MAAM,CAAC,OAAO,EACd,IAAA,iBAAG,EACD,IAAA,gBAAE,EAAC,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,EAC1C,IAAA,qBAAO,EAAC,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,EAAE,OAAO,CAAC,CACxD,CACF;aACA,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;aACjC,OAAO,CACN,MAAM,CAAC,IAAI,CAAC,KAAK,EACjB,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAC9B,MAAM,CAAC,IAAI,CAAC,eAAe,EAC3B,MAAM,CAAC,IAAI,CAAC,mBAAmB,EAC/B,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAC7B;aACA,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACb,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC,CAAC,CAAC,CAAA;YACnC,OAAO;gBACL,KAAK,EAAE,CAAC;gBACR,kBAAkB,EAAE,IAAI;gBACxB,eAAe,EAAE,IAAI;gBACrB,WAAW,EAAE,GAAG;gBAChB,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,YAAY,EAAE;gBAClC,mBAAmB,EAAE,KAAK;aAC3B,CAAA;QACH,CAAC,CAAC,CAAA;QAEJ,IAAI,SAAS,EAAE,CAAC;YACd,oBAAoB,GAAG,MAAM,YAAE;iBAC5B,MAAM,CAAC;gBACN,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS;gBACnC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;gBAC/B,kBAAkB,EAAE,IAAA,iBAAG,EAAQ,gBAAgB,MAAM,CAAC,OAAO,CAAC,OAAO,OAAO;aAC7E,CAAC;iBACD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;iBACpB,KAAK,CACJ,IAAA,iBAAG,EACD,IAAA,gBAAE,EAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,EACvC,IAAA,gBAAE,EAAC,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,CACnC,CACF;iBACA,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;iBACzD,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;gBACb,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpB,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAA;gBAC7C,CAAC;gBACD,OAAO,CAAC,CAAA;YACV,CAAC,CAAC,CAAA;QACN,CAAC;QAED,MAAM,KAAK,GACT,CAAC,MAAM,EAAE,kBAAkB,IAAI,CAAC,MAAM,EAAE,eAAe;YACrD,CAAC,CAAC,gCAAoB,CAAC,IAAI;YAC3B,CAAC,CAAC,MAAM,EAAE,KAAK,IAAI,gCAAoB,CAAC,IAAI,CAAA;QAEhD,OAAO;YACL,WAAW,EAAE,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC;YACzC,KAAK;YACL,OAAO,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACjC,mBAAmB,EAAE,CAAC,CAAC,MAAM,CAAC,mBAAmB;YACjD,oBAAoB;SACrB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,MAAc,EACd,cAAuB,EACvB,gBAAsB;QAEtB,MAAM,YAAE;aACL,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;aACnB,GAAG,CAAC;YACH,cAAc;YACd,gBAAgB;SACjB,CAAC;aACD,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAA;IACtC,CAAC;CACF;AAhJD,8DAgJC;AAIM,MAAM,eAAe,GAAG,CAAC,QAAkB,EAAE,EAAU,EAAE,EAAE;IAChE,MAAM,OAAO,GAAG,IAAA,kBAAK,EAAC,QAAQ,CAAC;SAC5B,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,qBAAqB,EAAE,CAAC;SACpD,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,IAAI,yBAAyB,EAAE,CAAC;SAC5D,UAAU,EAAE,CAAA;IAEf,OAAO;QACL,UAAU,EAAE,CAAC,SAAkB,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,EAAE,SAAS,CAAC;QACrE,YAAY,EAAE,CAAC,cAAuB,EAAE,gBAAsB,EAAE,EAAE,CAChE,OAAO,CAAC,YAAY,CAAC,EAAE,EAAE,cAAc,EAAE,gBAAgB,CAAC;KAC7D,CAAA;AACH,CAAC,CAAA;AAXY,QAAA,eAAe,mBAW3B"}

package/dist/common/constants.d.ts

@@ -0,0 +1,41 @@
+export declare const STOP_MARKER: string;
+export declare const FIND_FILES_MARKER: string;
+export declare const TOOL_RESULT_MARKER: string;
+export declare const EXISTING_CODE_MARKER = "[[**REPLACE_WITH_EXISTING_CODE**]]";
+export declare const DEFAULT_IGNORED_FILES: string[];
+export declare const SKIPPED_TERMINAL_COMMANDS: string[];
+export declare const REQUEST_CREDIT_SHOW_THRESHOLD = 1;
+export declare const MAX_DATE: Date;
+export type UsageLimits = 'ANON' | 'FREE' | 'PAID' | 'PRO_PLUS';
+export declare const CREDITS_USAGE_LIMITS: Record<UsageLimits, number>;
+export declare const CREDITS_REFERRAL_BONUS = 500;
+export declare const costModes: readonly ["lite", "normal", "pro"];
+export type CostMode = (typeof costModes)[number];
+export declare const getModelForMode: (costMode: CostMode, operation: "agent" | "file-requests" | "check-new-files") => "claude-3-5-sonnet-20241022" | "claude-3-5-haiku-20241022" | "gpt-4o-2024-08-06" | "gpt-4o-mini-2024-07-18";
+export declare const claudeModels: {
+    readonly sonnet: "claude-3-5-sonnet-20241022";
+    readonly haiku: "claude-3-5-haiku-20241022";
+};
+export declare const openaiModels: {
+    readonly gpt4o: "gpt-4o-2024-08-06";
+    readonly gpt4omini: "gpt-4o-mini-2024-07-18";
+    readonly o1: "o1-preview";
+};
+export declare const geminiModels: {
+    readonly gemini2flash: "gemini-2.0-flash-exp";
+};
+export declare const deepseekModels: {
+    readonly deepseekChat: "deepseek-chat";
+};
+export declare const models: {
+    readonly deepseekChat: "deepseek-chat";
+    readonly gemini2flash: "gemini-2.0-flash-exp";
+    readonly gpt4o: "gpt-4o-2024-08-06";
+    readonly gpt4omini: "gpt-4o-mini-2024-07-18";
+    readonly o1: "o1-preview";
+    readonly sonnet: "claude-3-5-sonnet-20241022";
+    readonly haiku: "claude-3-5-haiku-20241022";
+};
+export declare const OVERAGE_RATE_PRO = 0.99;
+export declare const OVERAGE_RATE_PRO_PLUS = 0.9;
+export declare const TEST_USER_ID = "test-user-id";

package/dist/common/constants.js

@@ -0,0 +1,126 @@
+"use strict";
+// import { env } from './env.mjs'
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TEST_USER_ID = exports.OVERAGE_RATE_PRO_PLUS = exports.OVERAGE_RATE_PRO = exports.models = exports.deepseekModels = exports.geminiModels = exports.openaiModels = exports.claudeModels = exports.getModelForMode = exports.costModes = exports.CREDITS_REFERRAL_BONUS = exports.CREDITS_USAGE_LIMITS = exports.MAX_DATE = exports.REQUEST_CREDIT_SHOW_THRESHOLD = exports.SKIPPED_TERMINAL_COMMANDS = exports.DEFAULT_IGNORED_FILES = exports.EXISTING_CODE_MARKER = exports.TOOL_RESULT_MARKER = exports.FIND_FILES_MARKER = exports.STOP_MARKER = void 0;
+exports.STOP_MARKER = '[' + 'END]';
+exports.FIND_FILES_MARKER = '[' + 'FIND_FILES_PLEASE]';
+exports.TOOL_RESULT_MARKER = '[' + 'TOOL_RESULT]';
+exports.EXISTING_CODE_MARKER = '[[**REPLACE_WITH_EXISTING_CODE**]]';
+exports.DEFAULT_IGNORED_FILES = [
+    '.git',
+    '.env',
+    '.env.*',
+    'env',
+    'ENV',
+    '*.min.*',
+    'node_modules',
+    'venv',
+    'virtualenv',
+    '.venv',
+    '.virtualenv',
+    '__pycache__',
+    '*.egg-info/',
+    '*.pyc',
+    '.DS_Store',
+    '.pytest_cache',
+    '.mypy_cache',
+    '.ruff_cache',
+    '.next',
+];
+exports.SKIPPED_TERMINAL_COMMANDS = [
+    'continue',
+    'date',
+    'head',
+    'history',
+    'if',
+    'jobs',
+    'less',
+    'man',
+    'more',
+    'nice',
+    'read',
+    'set',
+    'sort',
+    'split',
+    'tail',
+    'test',
+    'time',
+    'top',
+    'touch',
+    'type',
+    'unset',
+    'what',
+    'which',
+    'where',
+    'who',
+    'write',
+    'yes',
+    'help',
+    'find',
+    'add',
+    'hey',
+    'diff',
+    'make',
+    'please',
+    'apply',
+    'look',
+    'do',
+    'break',
+    'install',
+];
+exports.REQUEST_CREDIT_SHOW_THRESHOLD = 1;
+exports.MAX_DATE = new Date(86399999999999);
+exports.CREDITS_USAGE_LIMITS = process.env.NEXT_PUBLIC_ENVIRONMENT === 'local'
+    ? {
+        ANON: 1_000_000,
+        FREE: 2_500_000,
+        PAID: 5_000_000,
+        PRO_PLUS: 27_500_000,
+    }
+    : {
+        ANON: 500,
+        FREE: 1_000,
+        PAID: 5_000,
+        PRO_PLUS: 27_500,
+    };
+exports.CREDITS_REFERRAL_BONUS = 500;
+exports.costModes = ['lite', 'normal', 'pro'];
+const getModelForMode = (costMode, operation) => {
+    if (operation === 'agent') {
+        return costMode === 'lite' ? exports.claudeModels.haiku : exports.claudeModels.sonnet;
+    }
+    if (operation === 'file-requests') {
+        return costMode === 'pro' ? exports.claudeModels.sonnet : exports.claudeModels.haiku;
+    }
+    if (operation === 'check-new-files') {
+        return costMode === 'pro' ? exports.models.gpt4o : exports.models.gpt4omini;
+    }
+    throw new Error(`Unknown operation: ${operation}`);
+};
+exports.getModelForMode = getModelForMode;
+exports.claudeModels = {
+    sonnet: 'claude-3-5-sonnet-20241022',
+    haiku: 'claude-3-5-haiku-20241022',
+};
+exports.openaiModels = {
+    gpt4o: 'gpt-4o-2024-08-06',
+    gpt4omini: 'gpt-4o-mini-2024-07-18',
+    o1: 'o1-preview',
+    // o1: 'o1-2024-12-17',
+};
+exports.geminiModels = {
+    gemini2flash: 'gemini-2.0-flash-exp',
+};
+exports.deepseekModels = {
+    deepseekChat: 'deepseek-chat',
+};
+exports.models = {
+    ...exports.claudeModels,
+    ...exports.openaiModels,
+    ...exports.geminiModels,
+    ...exports.deepseekModels,
+};
+exports.OVERAGE_RATE_PRO = 0.99;
+exports.OVERAGE_RATE_PRO_PLUS = 0.9;
+exports.TEST_USER_ID = 'test-user-id';
+//# sourceMappingURL=constants.js.map

package/dist/common/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../src/constants.ts"],"names":[],"mappings":";AAAA,kCAAkC;;;AAErB,QAAA,WAAW,GAAG,GAAG,GAAG,MAAM,CAAA;AAC1B,QAAA,iBAAiB,GAAG,GAAG,GAAG,oBAAoB,CAAA;AAC9C,QAAA,kBAAkB,GAAG,GAAG,GAAG,cAAc,CAAA;AACzC,QAAA,oBAAoB,GAAG,oCAAoC,CAAA;AAE3D,QAAA,qBAAqB,GAAG;IACnC,MAAM;IACN,MAAM;IACN,QAAQ;IACR,KAAK;IACL,KAAK;IACL,SAAS;IACT,cAAc;IACd,MAAM;IACN,YAAY;IACZ,OAAO;IACP,aAAa;IACb,aAAa;IACb,aAAa;IACb,OAAO;IACP,WAAW;IACX,eAAe;IACf,aAAa;IACb,aAAa;IACb,OAAO;CACR,CAAA;AAEY,QAAA,yBAAyB,GAAG;IACvC,UAAU;IACV,MAAM;IACN,MAAM;IACN,SAAS;IACT,IAAI;IACJ,MAAM;IACN,MAAM;IACN,KAAK;IACL,MAAM;IACN,MAAM;IACN,MAAM;IACN,KAAK;IACL,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,MAAM;IACN,KAAK;IACL,OAAO;IACP,MAAM;IACN,OAAO;IACP,MAAM;IACN,OAAO;IACP,OAAO;IACP,KAAK;IACL,OAAO;IACP,KAAK;IACL,MAAM;IACN,MAAM;IACN,KAAK;IACL,KAAK;IACL,MAAM;IACN,MAAM;IACN,QAAQ;IACR,OAAO;IACP,MAAM;IACN,IAAI;IACJ,OAAO;IACP,SAAS;CACV,CAAA;AAEY,QAAA,6BAA6B,GAAG,CAAC,CAAA;AACjC,QAAA,QAAQ,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,CAAA;AAGnC,QAAA,oBAAoB,GAC/B,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,OAAO;IAC7C,CAAC,CAAC;QACE,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,SAAS;QACf,QAAQ,EAAE,UAAU;KACrB;IACH,CAAC,CAAC;QACE,IAAI,EAAE,GAAG;QACT,IAAI,EAAE,KAAK;QACX,IAAI,EAAE,KAAK;QACX,QAAQ,EAAE,MAAM;KACjB,CAAA;AACM,QAAA,sBAAsB,GAAG,GAAG,CAAA;AAE5B,QAAA,SAAS,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,KAAK,CAAU,CAAA;AAGpD,MAAM,eAAe,GAAG,CAC7B,QAAkB,EAClB,SAAwD,EACxD,EAAE;IACF,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QAC1B,OAAO,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,oBAAY,CAAC,KAAK,CAAC,CAAC,CAAC,oBAAY,CAAC,MAAM,CAAA;IACvE,CAAC;IACD,IAAI,SAAS,KAAK,eAAe,EAAE,CAAC;QAClC,OAAO,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,oBAAY,CAAC,MAAM,CAAC,CAAC,CAAC,oBAAY,CAAC,KAAK,CAAA;IACtE,CAAC;IACD,IAAI,SAAS,KAAK,iBAAiB,EAAE,CAAC;QACpC,OAAO,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,cAAM,CAAC,KAAK,CAAC,CAAC,CAAC,cAAM,CAAC,SAAS,CAAA;IAC7D,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,sBAAsB,SAAS,EAAE,CAAC,CAAA;AACpD,CAAC,CAAA;AAdY,QAAA,eAAe,mBAc3B;AAEY,QAAA,YAAY,GAAG;IAC1B,MAAM,EAAE,4BAA4B;IACpC,KAAK,EAAE,2BAA2B;CAC1B,CAAA;AAEG,QAAA,YAAY,GAAG;IAC1B,KAAK,EAAE,mBAAmB;IAC1B,SAAS,EAAE,wBAAwB;IACnC,EAAE,EAAE,YAAY;IAChB,uBAAuB;CACf,CAAA;AAEG,QAAA,YAAY,GAAG;IAC1B,YAAY,EAAE,sBAAsB;CAC5B,CAAA;AAEG,QAAA,cAAc,GAAG;IAC5B,YAAY,EAAE,eAAe;CACrB,CAAA;AAEG,QAAA,MAAM,GAAG;IACpB,GAAG,oBAAY;IACf,GAAG,oBAAY;IACf,GAAG,oBAAY;IACf,GAAG,sBAAc;CACT,CAAA;AAEG,QAAA,gBAAgB,GAAG,IAAI,CAAA;AACvB,QAAA,qBAAqB,GAAG,GAAG,CAAA;AAE3B,QAAA,YAAY,GAAG,cAAc,CAAA"}

package/dist/common/db/drizzle.config.d.ts

@@ -0,0 +1,2 @@
+declare const _default: import("drizzle-kit").Config;
+export default _default;

package/dist/common/db/drizzle.config.js

@@ -0,0 +1,17 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const drizzle_kit_1 = require("drizzle-kit");
+const env_mjs_1 = require("./env.mjs");
+const path_1 = __importDefault(require("path"));
+exports.default = (0, drizzle_kit_1.defineConfig)({
+    dialect: 'postgresql',
+    schema: path_1.default.join(__dirname, 'schema.ts').replace(/\\/g, '/'),
+    out: 'src/db/migrations',
+    dbCredentials: {
+        url: env_mjs_1.env.DATABASE_URL,
+    },
+});
+//# sourceMappingURL=drizzle.config.js.map

package/dist/common/db/drizzle.config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"drizzle.config.js","sourceRoot":"","sources":["../../src/db/drizzle.config.ts"],"names":[],"mappings":";;;;;AAAA,6CAA0C;AAC1C,uCAA+B;AAC/B,gDAAuB;AAEvB,kBAAe,IAAA,0BAAY,EAAC;IAC1B,OAAO,EAAE,YAAY;IACrB,MAAM,EAAE,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;IAC7D,GAAG,EAAE,mBAAmB;IACxB,aAAa,EAAE;QACb,GAAG,EAAE,aAAG,CAAC,YAAY;KACtB;CACF,CAAC,CAAA"}

package/dist/common/db/env.d.mts

@@ -0,0 +1 @@
+export const env: any;