codebuff 1.0.99

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (153) hide show
  1. package/README.md +70 -0
  2. package/dist/chat-storage.d.ts +29 -0
  3. package/dist/chat-storage.js +125 -0
  4. package/dist/chat-storage.js.map +1 -0
  5. package/dist/cli.d.ts +37 -0
  6. package/dist/cli.js +416 -0
  7. package/dist/cli.js.map +1 -0
  8. package/dist/client.d.ts +41 -0
  9. package/dist/client.js +443 -0
  10. package/dist/client.js.map +1 -0
  11. package/dist/code-map/languages.d.ts +12 -0
  12. package/dist/code-map/languages.js +129 -0
  13. package/dist/code-map/languages.js.map +1 -0
  14. package/dist/code-map/parse.d.ts +11 -0
  15. package/dist/code-map/parse.js +127 -0
  16. package/dist/code-map/parse.js.map +1 -0
  17. package/dist/code-map/test-langs/test.d.ts +11 -0
  18. package/dist/code-map/test-langs/test.js +23 -0
  19. package/dist/code-map/test-langs/test.js.map +1 -0
  20. package/dist/code-map/tree-sitter-queries/readme.md +23 -0
  21. package/dist/code-map/tree-sitter-queries/tree-sitter-c-tags.scm +11 -0
  22. package/dist/code-map/tree-sitter-queries/tree-sitter-c_sharp-tags.scm +13 -0
  23. package/dist/code-map/tree-sitter-queries/tree-sitter-cpp-tags.scm +17 -0
  24. package/dist/code-map/tree-sitter-queries/tree-sitter-go-tags.scm +26 -0
  25. package/dist/code-map/tree-sitter-queries/tree-sitter-java-tags.scm +19 -0
  26. package/dist/code-map/tree-sitter-queries/tree-sitter-javascript-tags.scm +15 -0
  27. package/dist/code-map/tree-sitter-queries/tree-sitter-php-tags.scm +26 -0
  28. package/dist/code-map/tree-sitter-queries/tree-sitter-python-tags.scm +9 -0
  29. package/dist/code-map/tree-sitter-queries/tree-sitter-ruby-tags.scm +58 -0
  30. package/dist/code-map/tree-sitter-queries/tree-sitter-rust-tags.scm +26 -0
  31. package/dist/code-map/tree-sitter-queries/tree-sitter-typescript-tags.scm +15 -0
  32. package/dist/code-map/tsconfig.tsbuildinfo +1 -0
  33. package/dist/common/actions.d.ts +1062 -0
  34. package/dist/common/actions.js +180 -0
  35. package/dist/common/actions.js.map +1 -0
  36. package/dist/common/billing/quota-manager.d.ts +59 -0
  37. package/dist/common/billing/quota-manager.js +200 -0
  38. package/dist/common/billing/quota-manager.js.map +1 -0
  39. package/dist/common/constants.d.ts +24 -0
  40. package/dist/common/constants.js +73 -0
  41. package/dist/common/constants.js.map +1 -0
  42. package/dist/common/db/drizzle.config.d.ts +2 -0
  43. package/dist/common/db/drizzle.config.js +17 -0
  44. package/dist/common/db/drizzle.config.js.map +1 -0
  45. package/dist/common/db/env.d.mts +1 -0
  46. package/dist/common/db/env.mjs +26 -0
  47. package/dist/common/db/env.mjs.map +1 -0
  48. package/dist/common/db/index.d.ts +6 -0
  49. package/dist/common/db/index.js +35 -0
  50. package/dist/common/db/index.js.map +1 -0
  51. package/dist/common/db/schema.d.ts +994 -0
  52. package/dist/common/db/schema.js +113 -0
  53. package/dist/common/db/schema.js.map +1 -0
  54. package/dist/common/env.d.mts +1 -0
  55. package/dist/common/env.mjs +34 -0
  56. package/dist/common/env.mjs.map +1 -0
  57. package/dist/common/project-file-tree.d.ts +9 -0
  58. package/dist/common/project-file-tree.js +157 -0
  59. package/dist/common/project-file-tree.js.map +1 -0
  60. package/dist/common/types/referral.d.ts +2 -0
  61. package/dist/common/types/referral.js +5 -0
  62. package/dist/common/types/referral.js.map +1 -0
  63. package/dist/common/types/usage.d.ts +21 -0
  64. package/dist/common/types/usage.js +12 -0
  65. package/dist/common/types/usage.js.map +1 -0
  66. package/dist/common/util/__tests__/string.test.d.ts +1 -0
  67. package/dist/common/util/__tests__/string.test.js +83 -0
  68. package/dist/common/util/__tests__/string.test.js.map +1 -0
  69. package/dist/common/util/array.d.ts +6 -0
  70. package/dist/common/util/array.js +32 -0
  71. package/dist/common/util/array.js.map +1 -0
  72. package/dist/common/util/changes.d.ts +6 -0
  73. package/dist/common/util/changes.js +72 -0
  74. package/dist/common/util/changes.js.map +1 -0
  75. package/dist/common/util/credentials.d.ts +25 -0
  76. package/dist/common/util/credentials.js +24 -0
  77. package/dist/common/util/credentials.js.map +1 -0
  78. package/dist/common/util/dates.d.ts +1 -0
  79. package/dist/common/util/dates.js +13 -0
  80. package/dist/common/util/dates.js.map +1 -0
  81. package/dist/common/util/file.d.ts +106 -0
  82. package/dist/common/util/file.js +138 -0
  83. package/dist/common/util/file.js.map +1 -0
  84. package/dist/common/util/git.d.ts +4 -0
  85. package/dist/common/util/git.js +40 -0
  86. package/dist/common/util/git.js.map +1 -0
  87. package/dist/common/util/helpers.d.ts +1 -0
  88. package/dist/common/util/helpers.js +6 -0
  89. package/dist/common/util/helpers.js.map +1 -0
  90. package/dist/common/util/object.d.ts +18 -0
  91. package/dist/common/util/object.js +91 -0
  92. package/dist/common/util/object.js.map +1 -0
  93. package/dist/common/util/patch.d.ts +1 -0
  94. package/dist/common/util/patch.js +215 -0
  95. package/dist/common/util/patch.js.map +1 -0
  96. package/dist/common/util/promise.d.ts +1 -0
  97. package/dist/common/util/promise.js +33 -0
  98. package/dist/common/util/promise.js.map +1 -0
  99. package/dist/common/util/referral.d.ts +1 -0
  100. package/dist/common/util/referral.js +6 -0
  101. package/dist/common/util/referral.js.map +1 -0
  102. package/dist/common/util/server/referral.d.ts +14 -0
  103. package/dist/common/util/server/referral.js +85 -0
  104. package/dist/common/util/server/referral.js.map +1 -0
  105. package/dist/common/util/string.d.ts +3 -0
  106. package/dist/common/util/string.js +60 -0
  107. package/dist/common/util/string.js.map +1 -0
  108. package/dist/common/util/stripe.d.ts +2 -0
  109. package/dist/common/util/stripe.js +19 -0
  110. package/dist/common/util/stripe.js.map +1 -0
  111. package/dist/common/util/tools.d.ts +2 -0
  112. package/dist/common/util/tools.js +13 -0
  113. package/dist/common/util/tools.js.map +1 -0
  114. package/dist/common/websockets/websocket-client.d.ts +40 -0
  115. package/dist/common/websockets/websocket-client.js +187 -0
  116. package/dist/common/websockets/websocket-client.js.map +1 -0
  117. package/dist/common/websockets/websocket-schema.d.ts +2530 -0
  118. package/dist/common/websockets/websocket-schema.js +55 -0
  119. package/dist/common/websockets/websocket-schema.js.map +1 -0
  120. package/dist/config.d.ts +2 -0
  121. package/dist/config.js +8 -0
  122. package/dist/config.js.map +1 -0
  123. package/dist/credentials.d.ts +3 -0
  124. package/dist/credentials.js +33 -0
  125. package/dist/credentials.js.map +1 -0
  126. package/dist/fingerprint.d.ts +1 -0
  127. package/dist/fingerprint.js +43 -0
  128. package/dist/fingerprint.js.map +1 -0
  129. package/dist/index.d.ts +2 -0
  130. package/dist/index.js +50 -0
  131. package/dist/index.js.map +1 -0
  132. package/dist/manifold-api.d.ts +8 -0
  133. package/dist/manifold-api.js +32 -0
  134. package/dist/manifold-api.js.map +1 -0
  135. package/dist/menu.d.ts +1 -0
  136. package/dist/menu.js +92 -0
  137. package/dist/menu.js.map +1 -0
  138. package/dist/project-files.d.ts +48 -0
  139. package/dist/project-files.js +324 -0
  140. package/dist/project-files.js.map +1 -0
  141. package/dist/tool-handlers.d.ts +11 -0
  142. package/dist/tool-handlers.js +131 -0
  143. package/dist/tool-handlers.js.map +1 -0
  144. package/dist/update-manicode.d.ts +1 -0
  145. package/dist/update-manicode.js +151 -0
  146. package/dist/update-manicode.js.map +1 -0
  147. package/dist/web-scraper.d.ts +3 -0
  148. package/dist/web-scraper.js +79 -0
  149. package/dist/web-scraper.js.map +1 -0
  150. package/dist/worker-script-project-context.d.ts +1 -0
  151. package/dist/worker-script-project-context.js +13 -0
  152. package/dist/worker-script-project-context.js.map +1 -0
  153. package/package.json +62 -0
@@ -0,0 +1,180 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.SERVER_ACTION_SCHEMA = exports.CLIENT_ACTION_SCHEMA = exports.ToolCallSchema = exports.CHANGES = exports.FileChangeSchema = void 0;
4
+ const zod_1 = require("zod");
5
+ const file_1 = require("./util/file");
6
+ const credentials_1 = require("./util/credentials");
7
+ const MessageContentObjectSchema = zod_1.z.union([
8
+ zod_1.z.object({
9
+ type: zod_1.z.literal('text'),
10
+ text: zod_1.z.string(),
11
+ cache_control: zod_1.z
12
+ .object({
13
+ type: zod_1.z.literal('ephemeral'),
14
+ })
15
+ .optional(),
16
+ }),
17
+ zod_1.z.object({
18
+ type: zod_1.z.literal('tool_use'),
19
+ id: zod_1.z.string(),
20
+ name: zod_1.z.string(),
21
+ input: zod_1.z.record(zod_1.z.string(), zod_1.z.any()),
22
+ cache_control: zod_1.z
23
+ .object({
24
+ type: zod_1.z.literal('ephemeral'),
25
+ })
26
+ .optional(),
27
+ }),
28
+ zod_1.z.object({
29
+ type: zod_1.z.literal('tool_result'),
30
+ tool_use_id: zod_1.z.string(),
31
+ content: zod_1.z.string(),
32
+ cache_control: zod_1.z
33
+ .object({
34
+ type: zod_1.z.literal('ephemeral'),
35
+ })
36
+ .optional(),
37
+ }),
38
+ ]);
39
+ const MessageSchema = zod_1.z.object({
40
+ role: zod_1.z.union([zod_1.z.literal('user'), zod_1.z.literal('assistant')]),
41
+ content: zod_1.z.union([zod_1.z.string(), zod_1.z.array(MessageContentObjectSchema)]),
42
+ });
43
+ exports.FileChangeSchema = zod_1.z.object({
44
+ type: zod_1.z.enum(['patch', 'file']),
45
+ filePath: zod_1.z.string(),
46
+ content: zod_1.z.string(),
47
+ });
48
+ exports.CHANGES = zod_1.z.array(exports.FileChangeSchema);
49
+ exports.ToolCallSchema = zod_1.z.object({
50
+ name: zod_1.z.string(),
51
+ id: zod_1.z.string(),
52
+ input: zod_1.z.record(zod_1.z.string(), zod_1.z.any()),
53
+ });
54
+ exports.CLIENT_ACTION_SCHEMA = zod_1.z.discriminatedUnion('type', [
55
+ zod_1.z.object({
56
+ type: zod_1.z.literal('user-input'),
57
+ fingerprintId: zod_1.z.string(),
58
+ authToken: zod_1.z.string().optional(),
59
+ userInputId: zod_1.z.string(),
60
+ messages: zod_1.z.array(MessageSchema),
61
+ fileContext: file_1.ProjectFileContextSchema,
62
+ previousChanges: exports.CHANGES,
63
+ }),
64
+ zod_1.z.object({
65
+ type: zod_1.z.literal('read-files-response'),
66
+ files: zod_1.z.record(zod_1.z.string(), zod_1.z.union([zod_1.z.string(), zod_1.z.null()])),
67
+ }),
68
+ // z.object({
69
+ // type: z.literal('run-terminal-command'),
70
+ // command: z.string(),
71
+ // }),
72
+ zod_1.z.object({
73
+ type: zod_1.z.literal('init'),
74
+ fingerprintId: zod_1.z.string(),
75
+ authToken: zod_1.z.string().optional(),
76
+ // userId: z.string().optional(),
77
+ fileContext: file_1.ProjectFileContextSchema,
78
+ }),
79
+ zod_1.z.object({
80
+ type: zod_1.z.literal('usage'),
81
+ fingerprintId: zod_1.z.string(),
82
+ authToken: zod_1.z.string().optional(),
83
+ }),
84
+ zod_1.z.object({
85
+ type: zod_1.z.literal('login-code-request'),
86
+ fingerprintId: zod_1.z.string(),
87
+ referralCode: zod_1.z.string().optional(),
88
+ }),
89
+ zod_1.z.object({
90
+ type: zod_1.z.literal('login-status-request'),
91
+ fingerprintId: zod_1.z.string(),
92
+ fingerprintHash: zod_1.z.string(),
93
+ }),
94
+ zod_1.z.object({
95
+ type: zod_1.z.literal('clear-auth-token'),
96
+ authToken: zod_1.z.string(),
97
+ fingerprintId: zod_1.z.string(),
98
+ userId: zod_1.z.string(),
99
+ // authToken: z.string().optional(),
100
+ fingerprintHash: zod_1.z.string(),
101
+ }),
102
+ zod_1.z.object({
103
+ type: zod_1.z.literal('generate-commit-message'),
104
+ fingerprintId: zod_1.z.string(),
105
+ authToken: zod_1.z.string().optional(),
106
+ stagedChanges: zod_1.z.string(),
107
+ }),
108
+ ]);
109
+ exports.SERVER_ACTION_SCHEMA = zod_1.z.discriminatedUnion('type', [
110
+ zod_1.z.object({
111
+ type: zod_1.z.literal('response-chunk'),
112
+ userInputId: zod_1.z.string(),
113
+ chunk: zod_1.z.string(),
114
+ }),
115
+ zod_1.z.object({
116
+ type: zod_1.z.literal('response-complete'),
117
+ userInputId: zod_1.z.string(),
118
+ response: zod_1.z.string(),
119
+ changes: exports.CHANGES,
120
+ addedFileVersions: zod_1.z.array(file_1.FileVersionSchema),
121
+ resetFileVersions: zod_1.z.boolean(),
122
+ usage: zod_1.z.number().optional(),
123
+ limit: zod_1.z.number().optional(),
124
+ subscription_active: zod_1.z.boolean().optional(),
125
+ referralLink: zod_1.z.string().optional(),
126
+ }),
127
+ zod_1.z.object({
128
+ type: zod_1.z.literal('read-files'),
129
+ filePaths: zod_1.z.array(zod_1.z.string()),
130
+ }),
131
+ zod_1.z.object({
132
+ type: zod_1.z.literal('tool-call'),
133
+ userInputId: zod_1.z.string(),
134
+ response: zod_1.z.string(),
135
+ data: exports.ToolCallSchema,
136
+ changes: exports.CHANGES,
137
+ addedFileVersions: zod_1.z.array(file_1.FileVersionSchema),
138
+ resetFileVersions: zod_1.z.boolean(),
139
+ }),
140
+ zod_1.z.object({
141
+ type: zod_1.z.literal('terminal-command-result'),
142
+ userInputId: zod_1.z.string(),
143
+ result: zod_1.z.string(),
144
+ }),
145
+ zod_1.z.object({
146
+ type: zod_1.z.literal('npm-version-status'),
147
+ isUpToDate: zod_1.z.boolean(),
148
+ latestVersion: zod_1.z.string(),
149
+ }),
150
+ zod_1.z.object({
151
+ type: zod_1.z.literal('init-response'),
152
+ }),
153
+ zod_1.z.object({
154
+ type: zod_1.z.literal('auth-result'),
155
+ user: credentials_1.userSchema.optional(),
156
+ message: zod_1.z.string(),
157
+ }),
158
+ zod_1.z.object({
159
+ type: zod_1.z.literal('login-code-response'),
160
+ fingerprintId: zod_1.z.string(),
161
+ fingerprintHash: zod_1.z.string(),
162
+ loginUrl: zod_1.z.string().url(),
163
+ }),
164
+ zod_1.z.object({
165
+ type: zod_1.z.literal('usage-response'),
166
+ usage: zod_1.z.number(),
167
+ limit: zod_1.z.number(),
168
+ referralLink: zod_1.z.string().optional(),
169
+ subscription_active: zod_1.z.boolean(),
170
+ }),
171
+ zod_1.z.object({
172
+ type: zod_1.z.literal('action-error'),
173
+ message: zod_1.z.string(),
174
+ }),
175
+ zod_1.z.object({
176
+ type: zod_1.z.literal('commit-message-response'),
177
+ commitMessage: zod_1.z.string(),
178
+ }),
179
+ ]);
180
+ //# sourceMappingURL=actions.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"actions.js","sourceRoot":"","sources":["../src/actions.ts"],"names":[],"mappings":";;;AAAA,6BAAuB;AACvB,sCAAyE;AACzE,oDAA+C;AAE/C,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,eAAe,EAAE,eAAO;KACzB,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,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,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC;QACpC,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;QACvB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;QACpB,OAAO,EAAE,eAAO;QAChB,iBAAiB,EAAE,OAAC,CAAC,KAAK,CAAC,wBAAiB,CAAC;QAC7C,iBAAiB,EAAE,OAAC,CAAC,OAAO,EAAE;QAC9B,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAC5B,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAC5B,mBAAmB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;QAC3C,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KACpC,CAAC;IACF,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,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,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,eAAe,CAAC;KACjC,CAAC;IACF,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,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;QACjC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;QACjB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;QACjB,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACnC,mBAAmB,EAAE,OAAC,CAAC,OAAO,EAAE;KACjC,CAAC;IACF,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"}
@@ -0,0 +1,59 @@
1
+ export interface IQuotaManager {
2
+ updateQuota(id: string): Promise<{
3
+ creditsUsed: number;
4
+ quota: number;
5
+ subscription_active: boolean;
6
+ }>;
7
+ checkQuota(id: string): Promise<{
8
+ creditsUsed: number;
9
+ quota: number;
10
+ endDate: Date;
11
+ subscription_active: boolean;
12
+ }>;
13
+ resetQuota(id: string): Promise<void>;
14
+ }
15
+ export declare class AnonymousQuotaManager implements IQuotaManager {
16
+ updateQuota(fingerprintId: string): Promise<{
17
+ creditsUsed: number;
18
+ quota: number;
19
+ subscription_active: boolean;
20
+ }>;
21
+ checkQuota(fingerprintId: string): Promise<{
22
+ creditsUsed: number;
23
+ quota: number;
24
+ endDate: Date;
25
+ subscription_active: boolean;
26
+ }>;
27
+ resetQuota(fingerprintId: string): Promise<void>;
28
+ private setQuotaExceeded;
29
+ }
30
+ export declare class AuthenticatedQuotaManager implements IQuotaManager {
31
+ updateQuota(userId: string): Promise<{
32
+ creditsUsed: number;
33
+ quota: number;
34
+ subscription_active: boolean;
35
+ }>;
36
+ checkQuota(userId: string): Promise<{
37
+ creditsUsed: number;
38
+ quota: number;
39
+ endDate: Date;
40
+ subscription_active: boolean;
41
+ }>;
42
+ resetQuota(userId: string): Promise<void>;
43
+ private setQuotaExceeded;
44
+ }
45
+ export type AuthType = 'anonymous' | 'authenticated';
46
+ export declare const getQuotaManager: (authType: AuthType, id: string) => {
47
+ updateQuota: () => Promise<{
48
+ creditsUsed: number;
49
+ quota: number;
50
+ subscription_active: boolean;
51
+ }>;
52
+ checkQuota: () => Promise<{
53
+ creditsUsed: number;
54
+ quota: number;
55
+ endDate: Date;
56
+ subscription_active: boolean;
57
+ }>;
58
+ resetQuota: () => Promise<void>;
59
+ };
@@ -0,0 +1,200 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || function (mod) {
19
+ if (mod && mod.__esModule) return mod;
20
+ var result = {};
21
+ if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
22
+ __setModuleDefault(result, mod);
23
+ return result;
24
+ };
25
+ var __importDefault = (this && this.__importDefault) || function (mod) {
26
+ return (mod && mod.__esModule) ? mod : { "default": mod };
27
+ };
28
+ Object.defineProperty(exports, "__esModule", { value: true });
29
+ exports.getQuotaManager = exports.AuthenticatedQuotaManager = exports.AnonymousQuotaManager = void 0;
30
+ const constants_1 = require("../constants");
31
+ const db_1 = __importDefault(require("../db"));
32
+ const schema = __importStar(require("../db/schema"));
33
+ const drizzle_orm_1 = require("drizzle-orm");
34
+ const dates_1 = require("../util/dates");
35
+ const ts_pattern_1 = require("ts-pattern");
36
+ class AnonymousQuotaManager {
37
+ async updateQuota(fingerprintId) {
38
+ const { creditsUsed, quota, endDate } = await this.checkQuota(fingerprintId);
39
+ if (creditsUsed >= quota) {
40
+ await this.setQuotaExceeded(fingerprintId);
41
+ }
42
+ return {
43
+ creditsUsed,
44
+ quota,
45
+ subscription_active: false,
46
+ };
47
+ }
48
+ async checkQuota(fingerprintId) {
49
+ const quota = constants_1.CREDITS_USAGE_LIMITS.ANON;
50
+ const startDate = (0, drizzle_orm_1.sql) `COALESCE(${schema.fingerprint.next_quota_reset}, now()) - INTERVAL '1 month'`;
51
+ const endDate = (0, drizzle_orm_1.sql) `COALESCE(${schema.fingerprint.next_quota_reset}, now())`;
52
+ const result = await db_1.default
53
+ .select({
54
+ creditsUsed: (0, drizzle_orm_1.sql) `SUM(COALESCE(${schema.message.credits}, 0))`,
55
+ endDate,
56
+ })
57
+ .from(schema.fingerprint)
58
+ .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)))
59
+ .where((0, drizzle_orm_1.eq)(schema.fingerprint.id, fingerprintId))
60
+ .groupBy(schema.fingerprint.next_quota_reset)
61
+ .then((rows) => {
62
+ if (rows.length > 0)
63
+ return rows[0];
64
+ return {
65
+ creditsUsed: 0,
66
+ quota,
67
+ endDate: new Date().toDateString(),
68
+ };
69
+ });
70
+ return {
71
+ creditsUsed: result.creditsUsed,
72
+ quota,
73
+ endDate: new Date(result.endDate),
74
+ subscription_active: false,
75
+ };
76
+ }
77
+ async resetQuota(fingerprintId) {
78
+ await db_1.default
79
+ .update(schema.fingerprint)
80
+ .set({
81
+ quota_exceeded: false,
82
+ next_quota_reset: null,
83
+ })
84
+ .where((0, drizzle_orm_1.eq)(schema.fingerprint.id, fingerprintId));
85
+ }
86
+ async setQuotaExceeded(fingerprintId) {
87
+ const nextQuotaReset = await db_1.default
88
+ .select({
89
+ next_quota_reset: schema.fingerprint.next_quota_reset,
90
+ })
91
+ .from(schema.fingerprint)
92
+ .where((0, drizzle_orm_1.eq)(schema.fingerprint.id, fingerprintId))
93
+ .then((fingerprints) => {
94
+ if (fingerprints.length === 1) {
95
+ return fingerprints[0].next_quota_reset;
96
+ }
97
+ return null;
98
+ });
99
+ await db_1.default
100
+ .update(schema.fingerprint)
101
+ .set({
102
+ quota_exceeded: true,
103
+ next_quota_reset: (0, dates_1.getNextQuotaReset)(nextQuotaReset),
104
+ })
105
+ .where((0, drizzle_orm_1.eq)(schema.fingerprint.id, fingerprintId));
106
+ }
107
+ }
108
+ exports.AnonymousQuotaManager = AnonymousQuotaManager;
109
+ class AuthenticatedQuotaManager {
110
+ async updateQuota(userId) {
111
+ const { creditsUsed, quota, subscription_active } = await this.checkQuota(userId);
112
+ if (creditsUsed >= quota && !subscription_active) {
113
+ await this.setQuotaExceeded(userId);
114
+ }
115
+ return {
116
+ creditsUsed,
117
+ quota,
118
+ subscription_active,
119
+ };
120
+ }
121
+ async checkQuota(userId) {
122
+ const startDate = (0, drizzle_orm_1.sql) `COALESCE(${schema.user.next_quota_reset}, now()) - INTERVAL '1 month'`;
123
+ const endDate = (0, drizzle_orm_1.sql) `COALESCE(${schema.user.next_quota_reset}, now())`;
124
+ const result = await db_1.default
125
+ .select({
126
+ quota: schema.user.quota,
127
+ stripe_customer_id: schema.user.stripe_customer_id,
128
+ stripe_price_id: schema.user.stripe_price_id,
129
+ subscription_active: schema.user.subscription_active,
130
+ endDate,
131
+ creditsUsed: (0, drizzle_orm_1.sql) `SUM(COALESCE(${schema.message.credits}, 0))`,
132
+ })
133
+ .from(schema.user)
134
+ .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)))
135
+ .where((0, drizzle_orm_1.eq)(schema.user.id, userId))
136
+ .groupBy(schema.user.quota, schema.user.stripe_customer_id, schema.user.stripe_price_id, schema.user.next_quota_reset, schema.user.subscription_active)
137
+ .then((rows) => {
138
+ if (rows.length > 0)
139
+ return rows[0];
140
+ return {
141
+ quota: 0,
142
+ stripe_customer_id: null,
143
+ stripe_price_id: null,
144
+ creditsUsed: '0',
145
+ endDate: new Date().toDateString(),
146
+ subscription_active: false,
147
+ };
148
+ });
149
+ const quota = !result?.stripe_customer_id && !result?.stripe_price_id
150
+ ? constants_1.CREDITS_USAGE_LIMITS.FREE
151
+ : result?.quota ?? constants_1.CREDITS_USAGE_LIMITS.FREE;
152
+ return {
153
+ creditsUsed: parseInt(result.creditsUsed),
154
+ quota,
155
+ endDate: new Date(result.endDate),
156
+ subscription_active: !!result.subscription_active,
157
+ };
158
+ }
159
+ async resetQuota(userId) {
160
+ await db_1.default
161
+ .update(schema.user)
162
+ .set({ quota_exceeded: false, next_quota_reset: null })
163
+ .where((0, drizzle_orm_1.eq)(schema.user.id, userId));
164
+ }
165
+ async setQuotaExceeded(userId) {
166
+ const nextQuotaReset = await db_1.default
167
+ .select({
168
+ next_quota_reset: schema.user.next_quota_reset,
169
+ })
170
+ .from(schema.user)
171
+ .where((0, drizzle_orm_1.eq)(schema.user.id, userId))
172
+ .then((users) => {
173
+ if (users.length === 1) {
174
+ return users[0].next_quota_reset;
175
+ }
176
+ return null;
177
+ });
178
+ await db_1.default
179
+ .update(schema.user)
180
+ .set({
181
+ quota_exceeded: true,
182
+ next_quota_reset: (0, dates_1.getNextQuotaReset)(nextQuotaReset),
183
+ })
184
+ .where((0, drizzle_orm_1.eq)(schema.user.id, userId));
185
+ }
186
+ }
187
+ exports.AuthenticatedQuotaManager = AuthenticatedQuotaManager;
188
+ const getQuotaManager = (authType, id) => {
189
+ const manager = (0, ts_pattern_1.match)(authType)
190
+ .with('anonymous', () => new AnonymousQuotaManager())
191
+ .with('authenticated', () => new AuthenticatedQuotaManager())
192
+ .exhaustive();
193
+ return {
194
+ updateQuota: () => manager.updateQuota(id),
195
+ checkQuota: () => manager.checkQuota(id),
196
+ resetQuota: () => manager.resetQuota(id),
197
+ };
198
+ };
199
+ exports.getQuotaManager = getQuotaManager;
200
+ //# sourceMappingURL=quota-manager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"quota-manager.js","sourceRoot":"","sources":["../../src/billing/quota-manager.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAmD;AACnD,+CAAsB;AACtB,qDAAsC;AACtC,6CAA4D;AAC5D,yCAAiD;AACjD,2CAAkC;AAiBlC,MAAa,qBAAqB;IAChC,KAAK,CAAC,WAAW,CAAC,aAAqB;QACrC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAA;QAE5E,IAAI,WAAW,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAA;QAC5C,CAAC;QACD,OAAO;YACL,WAAW;YACX,KAAK;YACL,mBAAmB,EAAE,KAAK;SAC3B,CAAA;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,aAAqB;QAMpC,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;QAEjG,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,CAAC;gBACd,KAAK;gBACL,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,YAAY,EAAE;aACnC,CAAA;QACH,CAAC,CAAC,CAAA;QAEJ,OAAO;YACL,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,KAAK;YACL,OAAO,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACjC,mBAAmB,EAAE,KAAK;SAC3B,CAAA;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,aAAqB;QACpC,MAAM,YAAE;aACL,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC;aAC1B,GAAG,CAAC;YACH,cAAc,EAAE,KAAK;YACrB,gBAAgB,EAAE,IAAI;SACvB,CAAC;aACD,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,WAAW,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC,CAAA;IACpD,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,aAAqB;QAClD,MAAM,cAAc,GAAG,MAAM,YAAE;aAC5B,MAAM,CAAC;YACN,gBAAgB,EAAE,MAAM,CAAC,WAAW,CAAC,gBAAgB;SACtD,CAAC;aACD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;aACxB,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,WAAW,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC;aAC/C,IAAI,CAAC,CAAC,YAAY,EAAE,EAAE;YACrB,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC9B,OAAO,YAAY,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAA;YACzC,CAAC;YACD,OAAO,IAAI,CAAA;QACb,CAAC,CAAC,CAAA;QAEJ,MAAM,YAAE;aACL,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC;aAC1B,GAAG,CAAC;YACH,cAAc,EAAE,IAAI;YACpB,gBAAgB,EAAE,IAAA,yBAAiB,EAAC,cAAc,CAAC;SACpD,CAAC;aACD,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,WAAW,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC,CAAA;IACpD,CAAC;CACF;AAxFD,sDAwFC;AAED,MAAa,yBAAyB;IACpC,KAAK,CAAC,WAAW,CAAC,MAAc;QAC9B,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,mBAAmB,EAAE,GAC/C,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;QAE/B,IAAI,WAAW,IAAI,KAAK,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACjD,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAA;QACrC,CAAC;QACD,OAAO;YACL,WAAW;YACX,KAAK;YACL,mBAAmB;SACpB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc;QAC7B,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;QAE1F,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,gBAAgB,EAC5B,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAChC;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,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;SAClD,CAAA;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc;QAC7B,MAAM,YAAE;aACL,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;aACnB,GAAG,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC;aACtD,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAA;IACtC,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,MAAc;QAC3C,MAAM,cAAc,GAAG,MAAM,YAAE;aAC5B,MAAM,CAAC;YACN,gBAAgB,EAAE,MAAM,CAAC,IAAI,CAAC,gBAAgB;SAC/C,CAAC;aACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;aACjB,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;aACjC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE;YACd,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAA;YAClC,CAAC;YACD,OAAO,IAAI,CAAA;QACb,CAAC,CAAC,CAAA;QAEJ,MAAM,YAAE;aACL,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;aACnB,GAAG,CAAC;YACH,cAAc,EAAE,IAAI;YACpB,gBAAgB,EAAE,IAAA,yBAAiB,EAAC,cAAc,CAAC;SACpD,CAAC;aACD,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAA;IACtC,CAAC;CACF;AAlGD,8DAkGC;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,WAAW,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QAC1C,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACxC,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;KACzC,CAAA;AACH,CAAC,CAAA;AAXY,QAAA,eAAe,mBAW3B"}
@@ -0,0 +1,24 @@
1
+ export declare const STOP_MARKER: string;
2
+ export declare const FIND_FILES_MARKER: string;
3
+ export declare const TOOL_RESULT_MARKER: string;
4
+ export declare const EXISTING_CODE_MARKER = "[[**REPLACE_WITH_EXISTING_CODE**]]";
5
+ export declare const SKIPPED_TERMINAL_COMMANDS: string[];
6
+ export declare const MAX_DATE: Date;
7
+ export type UsageLimits = 'ANON' | 'FREE' | 'PAID';
8
+ export declare const CREDITS_USAGE_LIMITS: Record<UsageLimits, number>;
9
+ export declare const CREDITS_REFERRAL_BONUS = 500;
10
+ export declare const claudeModels: {
11
+ readonly sonnet: "claude-3-5-sonnet-20241022";
12
+ readonly haiku: "claude-3-haiku-20240307";
13
+ };
14
+ export declare const openaiModels: {
15
+ readonly gpt4o: "gpt-4o-2024-08-06";
16
+ readonly gpt4omini: "gpt-4o-mini-2024-07-18";
17
+ };
18
+ export declare const models: {
19
+ gpt4o: "gpt-4o-2024-08-06";
20
+ gpt4omini: "gpt-4o-mini-2024-07-18";
21
+ sonnet: "claude-3-5-sonnet-20241022";
22
+ haiku: "claude-3-haiku-20240307";
23
+ };
24
+ export declare const TEST_USER_ID = "test-user-id";
@@ -0,0 +1,73 @@
1
+ "use strict";
2
+ // import { env } from './env.mjs'
3
+ Object.defineProperty(exports, "__esModule", { value: true });
4
+ exports.TEST_USER_ID = exports.models = exports.openaiModels = exports.claudeModels = exports.CREDITS_REFERRAL_BONUS = exports.CREDITS_USAGE_LIMITS = exports.MAX_DATE = exports.SKIPPED_TERMINAL_COMMANDS = exports.EXISTING_CODE_MARKER = exports.TOOL_RESULT_MARKER = exports.FIND_FILES_MARKER = exports.STOP_MARKER = void 0;
5
+ exports.STOP_MARKER = '[' + 'END]';
6
+ exports.FIND_FILES_MARKER = '[' + 'FIND_FILES_PLEASE]';
7
+ exports.TOOL_RESULT_MARKER = '[' + 'TOOL_RESULT]';
8
+ exports.EXISTING_CODE_MARKER = '[[**REPLACE_WITH_EXISTING_CODE**]]';
9
+ exports.SKIPPED_TERMINAL_COMMANDS = [
10
+ 'continue',
11
+ 'date',
12
+ 'head',
13
+ 'history',
14
+ 'if',
15
+ 'jobs',
16
+ 'less',
17
+ 'man',
18
+ 'more',
19
+ 'nice',
20
+ 'read',
21
+ 'set',
22
+ 'sort',
23
+ 'split',
24
+ 'tail',
25
+ 'test',
26
+ 'time',
27
+ 'top',
28
+ 'touch',
29
+ 'type',
30
+ 'unset',
31
+ 'what',
32
+ 'which',
33
+ 'who',
34
+ 'write',
35
+ 'yes',
36
+ 'help',
37
+ 'find',
38
+ 'kill',
39
+ 'add',
40
+ 'hey',
41
+ 'diff',
42
+ 'make',
43
+ 'please',
44
+ 'apply',
45
+ 'look',
46
+ ];
47
+ exports.MAX_DATE = new Date(86399999999999);
48
+ exports.CREDITS_USAGE_LIMITS = process.env.NEXT_PUBLIC_ENVIRONMENT === 'local'
49
+ ? {
50
+ ANON: 1_000_000,
51
+ FREE: 2_500_000,
52
+ PAID: 10_000_000,
53
+ }
54
+ : {
55
+ ANON: 1_000,
56
+ FREE: 2_500,
57
+ PAID: 10_000,
58
+ };
59
+ exports.CREDITS_REFERRAL_BONUS = 500;
60
+ exports.claudeModels = {
61
+ sonnet: 'claude-3-5-sonnet-20241022',
62
+ haiku: 'claude-3-haiku-20240307',
63
+ };
64
+ exports.openaiModels = {
65
+ gpt4o: 'gpt-4o-2024-08-06',
66
+ gpt4omini: 'gpt-4o-mini-2024-07-18',
67
+ };
68
+ exports.models = {
69
+ ...exports.claudeModels,
70
+ ...exports.openaiModels,
71
+ };
72
+ exports.TEST_USER_ID = 'test-user-id';
73
+ //# sourceMappingURL=constants.js.map
@@ -0,0 +1 @@
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,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,KAAK;IACL,OAAO;IACP,KAAK;IACL,MAAM;IACN,MAAM;IACN,MAAM;IACN,KAAK;IACL,KAAK;IACL,MAAM;IACN,MAAM;IACN,QAAQ;IACR,OAAO;IACP,MAAM;CACP,CAAA;AAEY,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,UAAU;KACjB;IACH,CAAC,CAAC;QACE,IAAI,EAAE,KAAK;QACX,IAAI,EAAE,KAAK;QACX,IAAI,EAAE,MAAM;KACb,CAAA;AACM,QAAA,sBAAsB,GAAG,GAAG,CAAA;AAE5B,QAAA,YAAY,GAAG;IAC1B,MAAM,EAAE,4BAA4B;IACpC,KAAK,EAAE,yBAAyB;CACxB,CAAA;AAEG,QAAA,YAAY,GAAG;IAC1B,KAAK,EAAE,mBAAmB;IAC1B,SAAS,EAAE,wBAAwB;CAC3B,CAAA;AAEG,QAAA,MAAM,GAAG;IACpB,GAAG,oBAAY;IACf,GAAG,oBAAY;CAChB,CAAA;AAEY,QAAA,YAAY,GAAG,cAAc,CAAA"}
@@ -0,0 +1,2 @@
1
+ declare const _default: import("drizzle-kit").Config;
2
+ export default _default;
@@ -0,0 +1,17 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ const drizzle_kit_1 = require("drizzle-kit");
7
+ const env_mjs_1 = require("./env.mjs");
8
+ const path_1 = __importDefault(require("path"));
9
+ exports.default = (0, drizzle_kit_1.defineConfig)({
10
+ dialect: 'postgresql',
11
+ schema: path_1.default.join('src', 'db', 'schema.ts'),
12
+ out: path_1.default.join('src', 'db', 'migrations'),
13
+ dbCredentials: {
14
+ url: env_mjs_1.env.DATABASE_URL,
15
+ },
16
+ });
17
+ //# sourceMappingURL=drizzle.config.js.map
@@ -0,0 +1 @@
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,KAAK,EAAE,IAAI,EAAE,WAAW,CAAC;IAC3C,GAAG,EAAE,cAAI,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,YAAY,CAAC;IACzC,aAAa,EAAE;QACb,GAAG,EAAE,aAAG,CAAC,YAAY;KACtB;CACF,CAAC,CAAA"}
@@ -0,0 +1 @@
1
+ export const env: any;
@@ -0,0 +1,26 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.env = void 0;
7
+ const env_core_1 = require("@t3-oss/env-core");
8
+ const zod_1 = require("zod");
9
+ const dotenv_1 = __importDefault(require("dotenv"));
10
+ dotenv_1.default.config({ path: '../stack.env' });
11
+ if (!process.env.ENVIRONMENT) {
12
+ console.error('ENVIRONMENT is not set, please check `stack.env`');
13
+ process.exit(1);
14
+ }
15
+ const DOTENV_PATH = process.env.RENDER === 'true' ? '/etc/secrets' : '..';
16
+ const path = `${DOTENV_PATH}/.env.${process.env.ENVIRONMENT}`;
17
+ console.log(`Using environment: ${process.env.ENVIRONMENT} (path: ${path})`);
18
+ dotenv_1.default.config({ path });
19
+ exports.env = (0, env_core_1.createEnv)({
20
+ server: {
21
+ ENVIRONMENT: zod_1.z.string().min(1),
22
+ DATABASE_URL: zod_1.z.string().min(1),
23
+ },
24
+ runtimeEnv: process.env,
25
+ });
26
+ //# sourceMappingURL=env.mjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"env.mjs","sourceRoot":"","sources":["../../src/db/env.mjs"],"names":[],"mappings":";;;;;;AAAA,+CAA4C;AAC5C,6BAAuB;AACvB,oDAA2B;AAE3B,gBAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAA;AACvC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;IAC7B,OAAO,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAA;IACjE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;AACjB,CAAC;AAED,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAA;AACzE,MAAM,IAAI,GAAG,GAAG,WAAW,SAAS,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAA;AAC7D,OAAO,CAAC,GAAG,CAAC,sBAAsB,OAAO,CAAC,GAAG,CAAC,WAAW,WAAW,IAAI,GAAG,CAAC,CAAA;AAC5E,gBAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAA;AAEV,QAAA,GAAG,GAAG,IAAA,oBAAS,EAAC;IAC3B,MAAM,EAAE;QACN,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9B,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;KAChC;IACD,UAAU,EAAE,OAAO,CAAC,GAAG;CACxB,CAAC,CAAA"}
@@ -0,0 +1,6 @@
1
+ import { Pool } from 'pg';
2
+ import * as schema from './schema';
3
+ declare const db: import("drizzle-orm/node-postgres").NodePgDatabase<typeof schema> & {
4
+ $client: Pool;
5
+ };
6
+ export default db;