coze-coding-dev-sdk 0.2.2 → 0.2.5
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.
- package/README.md +0 -13
- package/dist/cli/commands/chat.d.ts.map +1 -1
- package/dist/cli/commands/chat.js +0 -1
- package/dist/cli/commands/chat.js.map +1 -1
- package/dist/cli/commands/db.d.ts +3 -0
- package/dist/cli/commands/db.d.ts.map +1 -0
- package/dist/cli/commands/db.js +91 -0
- package/dist/cli/commands/db.js.map +1 -0
- package/dist/cli/commands/image.d.ts.map +1 -1
- package/dist/cli/commands/image.js +1 -2
- package/dist/cli/commands/image.js.map +1 -1
- package/dist/cli/commands/search.d.ts.map +1 -1
- package/dist/cli/commands/search.js +1 -2
- package/dist/cli/commands/search.js.map +1 -1
- package/dist/cli/commands/video.d.ts.map +1 -1
- package/dist/cli/commands/video.js +1 -2
- package/dist/cli/commands/video.js.map +1 -1
- package/dist/cli/commands/voice.d.ts.map +1 -1
- package/dist/cli/commands/voice.js +1 -2
- package/dist/cli/commands/voice.js.map +1 -1
- package/dist/cli/index.js +3 -1
- package/dist/cli/index.js.map +1 -1
- package/dist/cli/utils.d.ts +0 -1
- package/dist/cli/utils.d.ts.map +1 -1
- package/dist/cli/utils.js +0 -29
- package/dist/cli/utils.js.map +1 -1
- package/dist/core/client.d.ts +1 -5
- package/dist/core/client.d.ts.map +1 -1
- package/dist/core/client.js +1 -81
- package/dist/core/client.js.map +1 -1
- package/dist/database/client.d.ts +13 -0
- package/dist/database/client.d.ts.map +1 -0
- package/dist/database/client.js +142 -0
- package/dist/database/client.js.map +1 -0
- package/dist/database/index.d.ts +4 -0
- package/dist/database/index.d.ts.map +1 -0
- package/dist/database/index.js +20 -0
- package/dist/database/index.js.map +1 -0
- package/dist/database/migration.d.ts +7 -0
- package/dist/database/migration.d.ts.map +1 -0
- package/dist/database/migration.js +188 -0
- package/dist/database/migration.js.map +1 -0
- package/dist/database/models.d.ts +26 -0
- package/dist/database/models.d.ts.map +1 -0
- package/dist/database/models.js +15 -0
- package/dist/database/models.js.map +1 -0
- package/dist/image/client.d.ts +1 -1
- package/dist/image/client.d.ts.map +1 -1
- package/dist/image/client.js +2 -2
- package/dist/image/client.js.map +1 -1
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +2 -0
- package/dist/index.js.map +1 -1
- package/dist/s3/client.d.ts +66 -0
- package/dist/s3/client.d.ts.map +1 -0
- package/dist/s3/client.js +253 -0
- package/dist/s3/client.js.map +1 -0
- package/dist/s3/index.d.ts +3 -0
- package/dist/s3/index.d.ts.map +1 -0
- package/dist/s3/index.js +19 -0
- package/dist/s3/index.js.map +1 -0
- package/dist/s3/models.d.ts +23 -0
- package/dist/s3/models.d.ts.map +1 -0
- package/dist/s3/models.js +15 -0
- package/dist/s3/models.js.map +1 -0
- package/dist/search/client.d.ts +1 -1
- package/dist/search/client.d.ts.map +1 -1
- package/dist/search/client.js +2 -2
- package/dist/search/client.js.map +1 -1
- package/dist/video/client.d.ts +1 -1
- package/dist/video/client.d.ts.map +1 -1
- package/dist/video/client.js +2 -2
- package/dist/video/client.js.map +1 -1
- package/dist/voice/asr.d.ts +2 -2
- package/dist/voice/asr.d.ts.map +1 -1
- package/dist/voice/asr.js +3 -3
- package/dist/voice/asr.js.map +1 -1
- package/dist/voice/tts.d.ts +1 -1
- package/dist/voice/tts.d.ts.map +1 -1
- package/dist/voice/tts.js +2 -2
- package/dist/voice/tts.js.map +1 -1
- package/package.json +10 -3
|
@@ -0,0 +1,142 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Database client module
|
|
4
|
+
*/
|
|
5
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
6
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
7
|
+
};
|
|
8
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
+
exports.loadEnv = loadEnv;
|
|
10
|
+
exports.getDbUrl = getDbUrl;
|
|
11
|
+
exports.getPool = getPool;
|
|
12
|
+
exports.getDb = getDb;
|
|
13
|
+
exports.getClient = getClient;
|
|
14
|
+
exports.closePool = closePool;
|
|
15
|
+
const node_postgres_1 = require("drizzle-orm/node-postgres");
|
|
16
|
+
const pg_1 = __importDefault(require("pg"));
|
|
17
|
+
const child_process_1 = require("child_process");
|
|
18
|
+
const models_1 = require("./models");
|
|
19
|
+
let envLoaded = false;
|
|
20
|
+
function loadEnv() {
|
|
21
|
+
if (envLoaded || process.env.PGDATABASE_URL) {
|
|
22
|
+
return;
|
|
23
|
+
}
|
|
24
|
+
try {
|
|
25
|
+
try {
|
|
26
|
+
require('dotenv').config();
|
|
27
|
+
if (process.env.PGDATABASE_URL) {
|
|
28
|
+
envLoaded = true;
|
|
29
|
+
return;
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
catch {
|
|
33
|
+
// dotenv not available
|
|
34
|
+
}
|
|
35
|
+
const pythonCode = `
|
|
36
|
+
import os
|
|
37
|
+
import sys
|
|
38
|
+
try:
|
|
39
|
+
from coze_workload_identity import Client
|
|
40
|
+
client = Client()
|
|
41
|
+
env_vars = client.get_project_env_vars()
|
|
42
|
+
client.close()
|
|
43
|
+
for env_var in env_vars:
|
|
44
|
+
print(f"{env_var.key}={env_var.value}")
|
|
45
|
+
except Exception as e:
|
|
46
|
+
print(f"# Error: {e}", file=sys.stderr)
|
|
47
|
+
`;
|
|
48
|
+
const output = (0, child_process_1.execSync)(`python3 -c '${pythonCode.replace(/'/g, "'\"'\"'")}'`, {
|
|
49
|
+
encoding: 'utf-8',
|
|
50
|
+
timeout: 10000,
|
|
51
|
+
stdio: ['pipe', 'pipe', 'pipe'],
|
|
52
|
+
});
|
|
53
|
+
const lines = output.trim().split('\n');
|
|
54
|
+
for (const line of lines) {
|
|
55
|
+
if (line.startsWith('#'))
|
|
56
|
+
continue;
|
|
57
|
+
const eqIndex = line.indexOf('=');
|
|
58
|
+
if (eqIndex > 0) {
|
|
59
|
+
const key = line.substring(0, eqIndex);
|
|
60
|
+
let value = line.substring(eqIndex + 1);
|
|
61
|
+
if ((value.startsWith("'") && value.endsWith("'")) ||
|
|
62
|
+
(value.startsWith('"') && value.endsWith('"'))) {
|
|
63
|
+
value = value.slice(1, -1);
|
|
64
|
+
}
|
|
65
|
+
if (!process.env[key]) {
|
|
66
|
+
process.env[key] = value;
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
envLoaded = true;
|
|
71
|
+
}
|
|
72
|
+
catch {
|
|
73
|
+
// Silently fail
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
function getDbUrl() {
|
|
77
|
+
loadEnv();
|
|
78
|
+
let url = process.env.PGDATABASE_URL || '';
|
|
79
|
+
if (!url) {
|
|
80
|
+
throw new Error('Database URL not configured. Set PGDATABASE_URL environment variable.\n' +
|
|
81
|
+
'Did you create a database? You can create one via the Coze Coding platform.');
|
|
82
|
+
}
|
|
83
|
+
if (url.startsWith('postgres://')) {
|
|
84
|
+
url = 'postgresql://' + url.substring('postgres://'.length);
|
|
85
|
+
}
|
|
86
|
+
return url;
|
|
87
|
+
}
|
|
88
|
+
let pool = null;
|
|
89
|
+
let db = null;
|
|
90
|
+
async function createPoolWithRetry(config) {
|
|
91
|
+
const url = getDbUrl();
|
|
92
|
+
const newPool = new pg_1.default.Pool({
|
|
93
|
+
connectionString: url,
|
|
94
|
+
max: config?.maxConnections ?? models_1.DatabaseConfig.MAX_CONNECTIONS,
|
|
95
|
+
idleTimeoutMillis: config?.idleTimeoutMillis ?? models_1.DatabaseConfig.IDLE_TIMEOUT_MILLIS,
|
|
96
|
+
connectionTimeoutMillis: config?.connectionTimeoutMillis ?? models_1.DatabaseConfig.CONNECTION_TIMEOUT_MILLIS,
|
|
97
|
+
});
|
|
98
|
+
const startTime = Date.now();
|
|
99
|
+
let lastError = null;
|
|
100
|
+
while (Date.now() - startTime < models_1.DatabaseConfig.MAX_RETRY_TIME) {
|
|
101
|
+
try {
|
|
102
|
+
const client = await newPool.connect();
|
|
103
|
+
await client.query('SELECT 1');
|
|
104
|
+
client.release();
|
|
105
|
+
return newPool;
|
|
106
|
+
}
|
|
107
|
+
catch (e) {
|
|
108
|
+
lastError = e;
|
|
109
|
+
const elapsed = ((Date.now() - startTime) / 1000).toFixed(1);
|
|
110
|
+
console.warn(`Database connection failed, retrying... (elapsed: ${elapsed}s)`);
|
|
111
|
+
await new Promise((resolve) => setTimeout(resolve, Math.min(1000, models_1.DatabaseConfig.MAX_RETRY_TIME - (Date.now() - startTime))));
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
console.error(`Database connection failed after ${models_1.DatabaseConfig.MAX_RETRY_TIME / 1000}s: ${lastError?.message}`);
|
|
115
|
+
throw lastError;
|
|
116
|
+
}
|
|
117
|
+
async function getPool(config) {
|
|
118
|
+
loadEnv();
|
|
119
|
+
if (!pool) {
|
|
120
|
+
pool = await createPoolWithRetry(config);
|
|
121
|
+
}
|
|
122
|
+
return pool;
|
|
123
|
+
}
|
|
124
|
+
async function getDb(schema) {
|
|
125
|
+
if (!db) {
|
|
126
|
+
const p = await getPool();
|
|
127
|
+
db = schema ? (0, node_postgres_1.drizzle)(p, { schema }) : (0, node_postgres_1.drizzle)(p);
|
|
128
|
+
}
|
|
129
|
+
return db;
|
|
130
|
+
}
|
|
131
|
+
async function getClient() {
|
|
132
|
+
const p = await getPool();
|
|
133
|
+
return p.connect();
|
|
134
|
+
}
|
|
135
|
+
async function closePool() {
|
|
136
|
+
if (pool) {
|
|
137
|
+
await pool.end();
|
|
138
|
+
pool = null;
|
|
139
|
+
db = null;
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
//# sourceMappingURL=client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/database/client.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;AASH,0BAyDC;AAED,4BAiBC;AAsCD,0BAMC;AAED,sBAQC;AAED,8BAGC;AAED,8BAMC;AAtJD,6DAAoD;AACpD,4CAAoB;AACpB,iDAAyC;AACzC,qCAA8D;AAE9D,IAAI,SAAS,GAAG,KAAK,CAAC;AAEtB,SAAgB,OAAO;IACrB,IAAI,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QAC5C,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,IAAI,CAAC;YACH,OAAO,CAAC,QAAQ,CAAC,CAAC,MAAM,EAAE,CAAC;YAC3B,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;gBAC/B,SAAS,GAAG,IAAI,CAAC;gBACjB,OAAO;YACT,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uBAAuB;QACzB,CAAC;QAED,MAAM,UAAU,GAAG;;;;;;;;;;;;CAYtB,CAAC;QAEE,MAAM,MAAM,GAAG,IAAA,wBAAQ,EAAC,eAAe,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,GAAG,EAAE;YAC7E,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;SAChC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACxC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS;YACnC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClC,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBAChB,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBACvC,IAAI,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;gBACxC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;oBAC9C,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;oBACnD,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC7B,CAAC;gBACD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;QAED,SAAS,GAAG,IAAI,CAAC;IACnB,CAAC;IAAC,MAAM,CAAC;QACP,gBAAgB;IAClB,CAAC;AACH,CAAC;AAED,SAAgB,QAAQ;IACtB,OAAO,EAAE,CAAC;IAEV,IAAI,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC;IAE3C,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CACb,yEAAyE;YACzE,6EAA6E,CAC9E,CAAC;IACJ,CAAC;IAED,IAAI,GAAG,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,GAAG,GAAG,eAAe,GAAG,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAC9D,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED,IAAI,IAAI,GAAmB,IAAI,CAAC;AAChC,IAAI,EAAE,GAAsC,IAAI,CAAC;AAEjD,KAAK,UAAU,mBAAmB,CAAC,MAA2B;IAC5D,MAAM,GAAG,GAAG,QAAQ,EAAE,CAAC;IAEvB,MAAM,OAAO,GAAG,IAAI,YAAE,CAAC,IAAI,CAAC;QAC1B,gBAAgB,EAAE,GAAG;QACrB,GAAG,EAAE,MAAM,EAAE,cAAc,IAAI,uBAAc,CAAC,eAAe;QAC7D,iBAAiB,EAAE,MAAM,EAAE,iBAAiB,IAAI,uBAAc,CAAC,mBAAmB;QAClF,uBAAuB,EAAE,MAAM,EAAE,uBAAuB,IAAI,uBAAc,CAAC,yBAAyB;KACrG,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,IAAI,SAAS,GAAiB,IAAI,CAAC;IAEnC,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,uBAAc,CAAC,cAAc,EAAE,CAAC;QAC9D,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;YACvC,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,SAAS,GAAG,CAAU,CAAC;YACvB,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC7D,OAAO,CAAC,IAAI,CAAC,qDAAqD,OAAO,IAAI,CAAC,CAAC;YAC/E,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5B,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,uBAAc,CAAC,cAAc,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,CAC9F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,CAAC,KAAK,CAAC,oCAAoC,uBAAc,CAAC,cAAc,GAAG,IAAI,MAAM,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;IAClH,MAAM,SAAS,CAAC;AAClB,CAAC;AAEM,KAAK,UAAU,OAAO,CAAC,MAA2B;IACvD,OAAO,EAAE,CAAC;IACV,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,IAAI,GAAG,MAAM,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAEM,KAAK,UAAU,KAAK,CACzB,MAAU;IAEV,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,MAAM,CAAC,GAAG,MAAM,OAAO,EAAE,CAAC;QAC1B,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,IAAA,uBAAO,EAAC,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,IAAA,uBAAO,EAAC,CAAC,CAAC,CAAC;IACpD,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAEM,KAAK,UAAU,SAAS;IAC7B,MAAM,CAAC,GAAG,MAAM,OAAO,EAAE,CAAC;IAC1B,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;AACrB,CAAC;AAEM,KAAK,UAAU,SAAS;IAC7B,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;QACjB,IAAI,GAAG,IAAI,CAAC;QACZ,EAAE,GAAG,IAAI,CAAC;IACZ,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/database/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,aAAa,CAAC;AAC5B,cAAc,UAAU,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
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 __exportStar = (this && this.__exportStar) || function(m, exports) {
|
|
14
|
+
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
|
|
15
|
+
};
|
|
16
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
17
|
+
__exportStar(require("./client"), exports);
|
|
18
|
+
__exportStar(require("./migration"), exports);
|
|
19
|
+
__exportStar(require("./models"), exports);
|
|
20
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/database/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,8CAA4B;AAC5B,2CAAyB"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Database migration module
|
|
3
|
+
*/
|
|
4
|
+
import { GenerateModelsOptions, UpgradeOptions } from './models';
|
|
5
|
+
export declare function generateModels(options?: GenerateModelsOptions): Promise<void>;
|
|
6
|
+
export declare function upgrade(options?: UpgradeOptions): Promise<void>;
|
|
7
|
+
//# sourceMappingURL=migration.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"migration.d.ts","sourceRoot":"","sources":["../../src/database/migration.ts"],"names":[],"mappings":"AAAA;;GAEG;AAMH,OAAO,EAAE,qBAAqB,EAAE,cAAc,EAAkB,MAAM,UAAU,CAAC;AAqBjF,wBAAsB,cAAc,CAAC,OAAO,GAAE,qBAA0B,GAAG,OAAO,CAAC,IAAI,CAAC,CA2FvF;AAED,wBAAsB,OAAO,CAAC,OAAO,GAAE,cAAmB,GAAG,OAAO,CAAC,IAAI,CAAC,CA6DzE"}
|
|
@@ -0,0 +1,188 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Database migration module
|
|
4
|
+
*/
|
|
5
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
6
|
+
if (k2 === undefined) k2 = k;
|
|
7
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
8
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
9
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
10
|
+
}
|
|
11
|
+
Object.defineProperty(o, k2, desc);
|
|
12
|
+
}) : (function(o, m, k, k2) {
|
|
13
|
+
if (k2 === undefined) k2 = k;
|
|
14
|
+
o[k2] = m[k];
|
|
15
|
+
}));
|
|
16
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
17
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
18
|
+
}) : function(o, v) {
|
|
19
|
+
o["default"] = v;
|
|
20
|
+
});
|
|
21
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
22
|
+
var ownKeys = function(o) {
|
|
23
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
24
|
+
var ar = [];
|
|
25
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
26
|
+
return ar;
|
|
27
|
+
};
|
|
28
|
+
return ownKeys(o);
|
|
29
|
+
};
|
|
30
|
+
return function (mod) {
|
|
31
|
+
if (mod && mod.__esModule) return mod;
|
|
32
|
+
var result = {};
|
|
33
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
34
|
+
__setModuleDefault(result, mod);
|
|
35
|
+
return result;
|
|
36
|
+
};
|
|
37
|
+
})();
|
|
38
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
39
|
+
exports.generateModels = generateModels;
|
|
40
|
+
exports.upgrade = upgrade;
|
|
41
|
+
const child_process_1 = require("child_process");
|
|
42
|
+
const fs = __importStar(require("fs"));
|
|
43
|
+
const path = __importStar(require("path"));
|
|
44
|
+
const os = __importStar(require("os"));
|
|
45
|
+
const models_1 = require("./models");
|
|
46
|
+
const client_1 = require("./client");
|
|
47
|
+
/**
|
|
48
|
+
* Fix Drizzle bug: introspect generates numbers beyond JS precision
|
|
49
|
+
* Replace .generatedByDefaultAsIdentity({...}) with .generatedByDefaultAsIdentity()
|
|
50
|
+
*/
|
|
51
|
+
function fixGeneratedIdentity(filePath) {
|
|
52
|
+
if (!fs.existsSync(filePath))
|
|
53
|
+
return;
|
|
54
|
+
let content = fs.readFileSync(filePath, 'utf-8');
|
|
55
|
+
const fixed = content.replace(/\.generatedByDefaultAsIdentity\(\{[^}]*\}\)/g, '.generatedByDefaultAsIdentity()');
|
|
56
|
+
if (fixed !== content) {
|
|
57
|
+
fs.writeFileSync(filePath, fixed);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
async function generateModels(options = {}) {
|
|
61
|
+
const { outputPath = models_1.DatabaseConfig.DEFAULT_SCHEMA_PATH, verbose = false, } = options;
|
|
62
|
+
(0, client_1.loadEnv)();
|
|
63
|
+
const dbUrl = (0, client_1.getDbUrl)();
|
|
64
|
+
const outputDir = path.dirname(outputPath);
|
|
65
|
+
if (outputDir && !fs.existsSync(outputDir)) {
|
|
66
|
+
fs.mkdirSync(outputDir, { recursive: true });
|
|
67
|
+
}
|
|
68
|
+
// Use /tmp for drizzle-kit output, config in outputDir for module resolution
|
|
69
|
+
const tempOutDir = fs.mkdtempSync(path.join(os.tmpdir(), 'drizzle-'));
|
|
70
|
+
const tempConfigPath = path.join(outputDir, `drizzle.config.ts`);
|
|
71
|
+
const drizzleConfig = `
|
|
72
|
+
import { defineConfig } from 'drizzle-kit';
|
|
73
|
+
|
|
74
|
+
export default defineConfig({
|
|
75
|
+
dialect: 'postgresql',
|
|
76
|
+
out: '${tempOutDir}',
|
|
77
|
+
dbCredentials: {
|
|
78
|
+
url: '${dbUrl}',
|
|
79
|
+
},
|
|
80
|
+
});
|
|
81
|
+
`;
|
|
82
|
+
try {
|
|
83
|
+
fs.writeFileSync(tempConfigPath, drizzleConfig);
|
|
84
|
+
if (verbose) {
|
|
85
|
+
console.log('Introspecting database schema...');
|
|
86
|
+
}
|
|
87
|
+
const cmd = `npx drizzle-kit introspect --config=${tempConfigPath}`;
|
|
88
|
+
if (verbose) {
|
|
89
|
+
console.log(`Running: ${cmd}`);
|
|
90
|
+
}
|
|
91
|
+
(0, child_process_1.execSync)(cmd, {
|
|
92
|
+
encoding: 'utf-8',
|
|
93
|
+
stdio: verbose ? 'inherit' : 'pipe',
|
|
94
|
+
timeout: 60000,
|
|
95
|
+
});
|
|
96
|
+
// Fix Drizzle bug in generated schema
|
|
97
|
+
const generatedSchema = path.join(tempOutDir, 'schema.ts');
|
|
98
|
+
const generatedRelations = path.join(tempOutDir, 'relations.ts');
|
|
99
|
+
if (fs.existsSync(generatedSchema)) {
|
|
100
|
+
fixGeneratedIdentity(generatedSchema);
|
|
101
|
+
fs.copyFileSync(generatedSchema, outputPath);
|
|
102
|
+
if (verbose) {
|
|
103
|
+
console.log(`✓ Copied schema.ts to ${outputDir}/`);
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
if (fs.existsSync(generatedRelations)) {
|
|
107
|
+
const relationsPath = path.join(outputDir, 'relations.ts');
|
|
108
|
+
fs.copyFileSync(generatedRelations, relationsPath);
|
|
109
|
+
if (verbose) {
|
|
110
|
+
console.log(`✓ Copied relations.ts to ${outputDir}/`);
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
// Git add generated files
|
|
114
|
+
try {
|
|
115
|
+
(0, child_process_1.execSync)('git add -A', {
|
|
116
|
+
encoding: 'utf-8',
|
|
117
|
+
stdio: 'pipe',
|
|
118
|
+
cwd: process.cwd(),
|
|
119
|
+
});
|
|
120
|
+
}
|
|
121
|
+
catch {
|
|
122
|
+
// Ignore git errors (might not be in a git repo)
|
|
123
|
+
}
|
|
124
|
+
if (verbose) {
|
|
125
|
+
console.log(`✓ Schema generated successfully`);
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
finally {
|
|
129
|
+
// Cleanup temp files
|
|
130
|
+
if (fs.existsSync(tempConfigPath)) {
|
|
131
|
+
fs.unlinkSync(tempConfigPath);
|
|
132
|
+
}
|
|
133
|
+
if (fs.existsSync(tempOutDir)) {
|
|
134
|
+
fs.rmSync(tempOutDir, { recursive: true });
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
async function upgrade(options = {}) {
|
|
139
|
+
const { schemaPath = models_1.DatabaseConfig.DEFAULT_SCHEMA_PATH, projectPath = process.cwd(), verbose = false, } = options;
|
|
140
|
+
(0, client_1.loadEnv)();
|
|
141
|
+
const dbUrl = (0, client_1.getDbUrl)();
|
|
142
|
+
const fullSchemaPath = path.isAbsolute(schemaPath)
|
|
143
|
+
? schemaPath
|
|
144
|
+
: path.join(projectPath, schemaPath);
|
|
145
|
+
if (!fs.existsSync(fullSchemaPath)) {
|
|
146
|
+
throw new Error(`Schema file not found: ${fullSchemaPath}`);
|
|
147
|
+
}
|
|
148
|
+
// Config in schema directory for module resolution
|
|
149
|
+
const schemaDir = path.dirname(fullSchemaPath);
|
|
150
|
+
const tempConfigPath = path.join(schemaDir, `drizzle.config.ts`);
|
|
151
|
+
const drizzleConfig = `
|
|
152
|
+
import { defineConfig } from 'drizzle-kit';
|
|
153
|
+
|
|
154
|
+
export default defineConfig({
|
|
155
|
+
dialect: 'postgresql',
|
|
156
|
+
schema: '${fullSchemaPath}',
|
|
157
|
+
dbCredentials: {
|
|
158
|
+
url: '${dbUrl}',
|
|
159
|
+
},
|
|
160
|
+
});
|
|
161
|
+
`;
|
|
162
|
+
try {
|
|
163
|
+
fs.writeFileSync(tempConfigPath, drizzleConfig);
|
|
164
|
+
if (verbose) {
|
|
165
|
+
console.log('Pushing schema changes to database...');
|
|
166
|
+
}
|
|
167
|
+
// Use --force to skip confirmation
|
|
168
|
+
const cmd = `npx drizzle-kit push --force --config=${tempConfigPath}`;
|
|
169
|
+
if (verbose) {
|
|
170
|
+
console.log(`Running: ${cmd}`);
|
|
171
|
+
}
|
|
172
|
+
(0, child_process_1.execSync)(cmd, {
|
|
173
|
+
encoding: 'utf-8',
|
|
174
|
+
stdio: verbose ? 'inherit' : 'pipe',
|
|
175
|
+
timeout: 120000,
|
|
176
|
+
cwd: projectPath,
|
|
177
|
+
});
|
|
178
|
+
if (verbose) {
|
|
179
|
+
console.log('✅ Database migration completed successfully');
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
finally {
|
|
183
|
+
if (fs.existsSync(tempConfigPath)) {
|
|
184
|
+
fs.unlinkSync(tempConfigPath);
|
|
185
|
+
}
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
//# sourceMappingURL=migration.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"migration.js","sourceRoot":"","sources":["../../src/database/migration.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2BH,wCA2FC;AAED,0BA6DC;AAnLD,iDAAyC;AACzC,uCAAyB;AACzB,2CAA6B;AAC7B,uCAAyB;AACzB,qCAAiF;AACjF,qCAA6C;AAE7C;;;GAGG;AACH,SAAS,oBAAoB,CAAC,QAAgB;IAC5C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO;IAErC,IAAI,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAC3B,8CAA8C,EAC9C,iCAAiC,CAClC,CAAC;IAEF,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACpC,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,cAAc,CAAC,UAAiC,EAAE;IACtE,MAAM,EACJ,UAAU,GAAG,uBAAc,CAAC,mBAAmB,EAC/C,OAAO,GAAG,KAAK,GAChB,GAAG,OAAO,CAAC;IAEZ,IAAA,gBAAO,GAAE,CAAC;IACV,MAAM,KAAK,GAAG,IAAA,iBAAQ,GAAE,CAAC;IAEzB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC3C,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3C,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,6EAA6E;IAC7E,MAAM,UAAU,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC,CAAC;IACtE,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;IAEjE,MAAM,aAAa,GAAG;;;;;UAKd,UAAU;;YAER,KAAK;;;CAGhB,CAAC;IAEA,IAAI,CAAC;QACH,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QAEhD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,GAAG,GAAG,uCAAuC,cAAc,EAAE,CAAC;QACpE,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,EAAE,CAAC,CAAC;QACjC,CAAC;QAED,IAAA,wBAAQ,EAAC,GAAG,EAAE;YACZ,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;YACnC,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;QAEH,sCAAsC;QACtC,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAC3D,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;QAEjE,IAAI,EAAE,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,CAAC;YACnC,oBAAoB,CAAC,eAAe,CAAC,CAAC;YACtC,EAAE,CAAC,YAAY,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;YAC7C,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,GAAG,CAAC,yBAAyB,SAAS,GAAG,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,IAAI,EAAE,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YAC3D,EAAE,CAAC,YAAY,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC;YACnD,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,GAAG,CAAC,4BAA4B,SAAS,GAAG,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAED,0BAA0B;QAC1B,IAAI,CAAC;YACH,IAAA,wBAAQ,EAAC,YAAY,EAAE;gBACrB,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,MAAM;gBACb,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;aACnB,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,iDAAiD;QACnD,CAAC;QAED,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;YAAS,CAAC;QACT,qBAAqB;QACrB,IAAI,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YAClC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9B,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,OAAO,CAAC,UAA0B,EAAE;IACxD,MAAM,EACJ,UAAU,GAAG,uBAAc,CAAC,mBAAmB,EAC/C,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,EAC3B,OAAO,GAAG,KAAK,GAChB,GAAG,OAAO,CAAC;IAEZ,IAAA,gBAAO,GAAE,CAAC;IACV,MAAM,KAAK,GAAG,IAAA,iBAAQ,GAAE,CAAC;IAEzB,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;QAChD,CAAC,CAAC,UAAU;QACZ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IAEvC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,0BAA0B,cAAc,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,mDAAmD;IACnD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IAC/C,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;IACjE,MAAM,aAAa,GAAG;;;;;aAKX,cAAc;;YAEf,KAAK;;;CAGhB,CAAC;IAEA,IAAI,CAAC;QACH,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QAEhD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;QACvD,CAAC;QAED,mCAAmC;QACnC,MAAM,GAAG,GAAG,yCAAyC,cAAc,EAAE,CAAC;QACtE,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,EAAE,CAAC,CAAC;QACjC,CAAC;QAED,IAAA,wBAAQ,EAAC,GAAG,EAAE;YACZ,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;YACnC,OAAO,EAAE,MAAM;YACf,GAAG,EAAE,WAAW;SACjB,CAAC,CAAC;QAEH,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;YAAS,CAAC;QACT,IAAI,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YAClC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Database module types
|
|
3
|
+
*/
|
|
4
|
+
export declare class DatabaseConfig {
|
|
5
|
+
static readonly MAX_CONNECTIONS = 100;
|
|
6
|
+
static readonly IDLE_TIMEOUT_MILLIS = 30000;
|
|
7
|
+
static readonly CONNECTION_TIMEOUT_MILLIS = 30000;
|
|
8
|
+
static readonly MAX_RETRY_TIME = 20000;
|
|
9
|
+
static readonly DEFAULT_SCHEMA_PATH = "src/storage/database/shared/schema.ts";
|
|
10
|
+
}
|
|
11
|
+
export interface DatabasePoolConfig {
|
|
12
|
+
connectionString?: string;
|
|
13
|
+
maxConnections?: number;
|
|
14
|
+
idleTimeoutMillis?: number;
|
|
15
|
+
connectionTimeoutMillis?: number;
|
|
16
|
+
}
|
|
17
|
+
export interface UpgradeOptions {
|
|
18
|
+
schemaPath?: string;
|
|
19
|
+
projectPath?: string;
|
|
20
|
+
verbose?: boolean;
|
|
21
|
+
}
|
|
22
|
+
export interface GenerateModelsOptions {
|
|
23
|
+
outputPath?: string;
|
|
24
|
+
verbose?: boolean;
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=models.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/database/models.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,qBAAa,cAAc;IACzB,MAAM,CAAC,QAAQ,CAAC,eAAe,OAAO;IACtC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,SAAS;IAC5C,MAAM,CAAC,QAAQ,CAAC,yBAAyB,SAAS;IAClD,MAAM,CAAC,QAAQ,CAAC,cAAc,SAAS;IACvC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,2CAA2C;CAC/E;AAED,MAAM,WAAW,kBAAkB;IACjC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,uBAAuB,CAAC,EAAE,MAAM,CAAC;CAClC;AAED,MAAM,WAAW,cAAc;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,qBAAqB;IACpC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Database module types
|
|
4
|
+
*/
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.DatabaseConfig = void 0;
|
|
7
|
+
class DatabaseConfig {
|
|
8
|
+
}
|
|
9
|
+
exports.DatabaseConfig = DatabaseConfig;
|
|
10
|
+
DatabaseConfig.MAX_CONNECTIONS = 100;
|
|
11
|
+
DatabaseConfig.IDLE_TIMEOUT_MILLIS = 30000;
|
|
12
|
+
DatabaseConfig.CONNECTION_TIMEOUT_MILLIS = 30000;
|
|
13
|
+
DatabaseConfig.MAX_RETRY_TIME = 20000;
|
|
14
|
+
DatabaseConfig.DEFAULT_SCHEMA_PATH = 'src/storage/database/shared/schema.ts';
|
|
15
|
+
//# sourceMappingURL=models.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/database/models.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH,MAAa,cAAc;;AAA3B,wCAMC;AALiB,8BAAe,GAAG,GAAG,CAAC;AACtB,kCAAmB,GAAG,KAAK,CAAC;AAC5B,wCAAyB,GAAG,KAAK,CAAC;AAClC,6BAAc,GAAG,KAAK,CAAC;AACvB,kCAAmB,GAAG,uCAAuC,CAAC"}
|
package/dist/image/client.d.ts
CHANGED
|
@@ -3,7 +3,7 @@ import { Config } from '../core/config';
|
|
|
3
3
|
import { ImageGenerationRequest, ImageGenerationResponse, ImageGenerationResponseHelper } from './models';
|
|
4
4
|
export declare class ImageGenerationClient extends BaseClient {
|
|
5
5
|
private model;
|
|
6
|
-
constructor(config?: Config, customHeaders?: Record<string, string
|
|
6
|
+
constructor(config?: Config, customHeaders?: Record<string, string>);
|
|
7
7
|
generate(request: ImageGenerationRequest): Promise<ImageGenerationResponse>;
|
|
8
8
|
batchGenerate(requests: ImageGenerationRequest[]): Promise<ImageGenerationResponse[]>;
|
|
9
9
|
getResponseHelper(response: ImageGenerationResponse): ImageGenerationResponseHelper;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/image/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAExC,OAAO,EAEL,sBAAsB,EACtB,uBAAuB,EACvB,6BAA6B,EAE9B,MAAM,UAAU,CAAC;AAElB,qBAAa,qBAAsB,SAAQ,UAAU;IACnD,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,CAAC,EAAE,MAAM,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC
|
|
1
|
+
{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/image/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAExC,OAAO,EAEL,sBAAsB,EACtB,uBAAuB,EACvB,6BAA6B,EAE9B,MAAM,UAAU,CAAC;AAElB,qBAAa,qBAAsB,SAAQ,UAAU;IACnD,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,CAAC,EAAE,MAAM,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAK7D,QAAQ,CACZ,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,uBAAuB,CAAC;IA8B7B,aAAa,CACjB,QAAQ,EAAE,sBAAsB,EAAE,GACjC,OAAO,CAAC,uBAAuB,EAAE,CAAC;IAKrC,iBAAiB,CACf,QAAQ,EAAE,uBAAuB,GAChC,6BAA6B;CAGjC"}
|
package/dist/image/client.js
CHANGED
|
@@ -5,8 +5,8 @@ const client_1 = require("../core/client");
|
|
|
5
5
|
const exceptions_1 = require("../core/exceptions");
|
|
6
6
|
const models_1 = require("./models");
|
|
7
7
|
class ImageGenerationClient extends client_1.BaseClient {
|
|
8
|
-
constructor(config, customHeaders
|
|
9
|
-
super(config, customHeaders
|
|
8
|
+
constructor(config, customHeaders) {
|
|
9
|
+
super(config, customHeaders);
|
|
10
10
|
this.model = models_1.ImageConfig.DEFAULT_MODEL;
|
|
11
11
|
}
|
|
12
12
|
async generate(request) {
|
package/dist/image/client.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/image/client.ts"],"names":[],"mappings":";;;AAAA,2CAA4C;AAE5C,mDAA8C;AAC9C,qCAMkB;AAElB,MAAa,qBAAsB,SAAQ,mBAAU;IAGnD,YAAY,MAAe,EAAE,aAAsC
|
|
1
|
+
{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/image/client.ts"],"names":[],"mappings":";;;AAAA,2CAA4C;AAE5C,mDAA8C;AAC9C,qCAMkB;AAElB,MAAa,qBAAsB,SAAQ,mBAAU;IAGnD,YAAY,MAAe,EAAE,aAAsC;QACjE,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,GAAG,oBAAW,CAAC,aAAa,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAA+B;QAE/B,MAAM,UAAU,GAAG,IAAA,uCAA8B,EAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAC7B,MAAM,EACN,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,4BAA4B,EAClD,UAAU,CACX,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,qBAAQ,CAChB,uBAAuB,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,eAAe,EAAE,EAC9D,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EACjD,IAAI,CAAC,KAAK,CACX,CAAC;QACJ,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC7B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,MAAM,IAAI,qBAAQ,CAChB,4BAA4B,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,eAAe,EAAE,EACnE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EACjD,IAAI,CAAC,KAAK,CACX,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,QAAkC;QAElC,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QACnE,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAED,iBAAiB,CACf,QAAiC;QAEjC,OAAO,IAAI,sCAA6B,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC;CACF;AApDD,sDAoDC"}
|
package/dist/index.d.ts
CHANGED
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,cAAc,YAAY,CAAC;AAC3B,cAAc,MAAM,CAAC;AACrB,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
|
package/dist/index.js
CHANGED
|
@@ -21,6 +21,8 @@ __exportStar(require("./search"), exports);
|
|
|
21
21
|
__exportStar(require("./voice"), exports);
|
|
22
22
|
__exportStar(require("./video"), exports);
|
|
23
23
|
__exportStar(require("./llm"), exports);
|
|
24
|
+
__exportStar(require("./database"), exports);
|
|
25
|
+
__exportStar(require("./s3"), exports);
|
|
24
26
|
var version_1 = require("./version");
|
|
25
27
|
Object.defineProperty(exports, "VERSION", { enumerable: true, get: function () { return version_1.VERSION; } });
|
|
26
28
|
//# sourceMappingURL=index.js.map
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,0CAAwB;AACxB,2CAAyB;AACzB,0CAAwB;AACxB,0CAAwB;AACxB,wCAAsB;AACtB,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,0CAAwB;AACxB,2CAAyB;AACzB,0CAAwB;AACxB,0CAAwB;AACxB,wCAAsB;AACtB,6CAA2B;AAC3B,uCAAqB;AACrB,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* S3 Storage client module
|
|
3
|
+
*/
|
|
4
|
+
import { Readable } from 'stream';
|
|
5
|
+
import { S3StorageConfig, ListFilesResult } from './models';
|
|
6
|
+
export declare class S3Storage {
|
|
7
|
+
private endpointUrl;
|
|
8
|
+
private accessKey;
|
|
9
|
+
private secretKey;
|
|
10
|
+
private bucketName;
|
|
11
|
+
private region;
|
|
12
|
+
private client;
|
|
13
|
+
constructor(config?: S3StorageConfig);
|
|
14
|
+
private getClient;
|
|
15
|
+
private generateObjectKey;
|
|
16
|
+
private resolveBucket;
|
|
17
|
+
private validateFileName;
|
|
18
|
+
uploadFile(options: {
|
|
19
|
+
fileContent: Buffer;
|
|
20
|
+
fileName: string;
|
|
21
|
+
contentType?: string;
|
|
22
|
+
bucket?: string;
|
|
23
|
+
}): Promise<string>;
|
|
24
|
+
readFile(options: {
|
|
25
|
+
fileKey: string;
|
|
26
|
+
bucket?: string;
|
|
27
|
+
}): Promise<Buffer>;
|
|
28
|
+
deleteFile(options: {
|
|
29
|
+
fileKey: string;
|
|
30
|
+
bucket?: string;
|
|
31
|
+
}): Promise<boolean>;
|
|
32
|
+
fileExists(options: {
|
|
33
|
+
fileKey: string;
|
|
34
|
+
bucket?: string;
|
|
35
|
+
}): Promise<boolean>;
|
|
36
|
+
listFiles(options?: {
|
|
37
|
+
prefix?: string;
|
|
38
|
+
bucket?: string;
|
|
39
|
+
maxKeys?: number;
|
|
40
|
+
continuationToken?: string;
|
|
41
|
+
}): Promise<ListFilesResult>;
|
|
42
|
+
generatePresignedUrl(options: {
|
|
43
|
+
key: string;
|
|
44
|
+
bucket?: string;
|
|
45
|
+
expireTime?: number;
|
|
46
|
+
}): Promise<string>;
|
|
47
|
+
streamUploadFile(options: {
|
|
48
|
+
stream: Readable;
|
|
49
|
+
fileName: string;
|
|
50
|
+
contentType?: string;
|
|
51
|
+
bucket?: string;
|
|
52
|
+
}): Promise<string>;
|
|
53
|
+
uploadFromUrl(options: {
|
|
54
|
+
url: string;
|
|
55
|
+
bucket?: string;
|
|
56
|
+
timeout?: number;
|
|
57
|
+
}): Promise<string>;
|
|
58
|
+
chunkUploadFile(options: {
|
|
59
|
+
chunks: AsyncIterable<Buffer>;
|
|
60
|
+
fileName: string;
|
|
61
|
+
contentType?: string;
|
|
62
|
+
bucket?: string;
|
|
63
|
+
}): Promise<string>;
|
|
64
|
+
}
|
|
65
|
+
export default S3Storage;
|
|
66
|
+
//# sourceMappingURL=client.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/s3/client.ts"],"names":[],"mappings":"AAAA;;GAEG;AAWH,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAGlC,OAAO,EAAE,eAAe,EAAY,eAAe,EAAwB,MAAM,UAAU,CAAC;AAM5F,qBAAa,SAAS;IACpB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAyB;gBAE3B,MAAM,GAAE,eAAoB;IAQxC,OAAO,CAAC,SAAS;IA8BjB,OAAO,CAAC,iBAAiB;IAOzB,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,gBAAgB;IAclB,UAAU,CAAC,OAAO,EAAE;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;IAoBb,QAAQ,CAAC,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAoBxE,UAAU,CAAC,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAY3E,UAAU,CAAC,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAmB3E,SAAS,CAAC,OAAO,GAAE;QACvB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,iBAAiB,CAAC,EAAE,MAAM,CAAC;KACvB,GAAG,OAAO,CAAC,eAAe,CAAC;IA4B3B,oBAAoB,CAAC,OAAO,EAAE;QAClC,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,GAAG,OAAO,CAAC,MAAM,CAAC;IAmDb,gBAAgB,CAAC,OAAO,EAAE;QAC9B,MAAM,EAAE,QAAQ,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;IAqBb,aAAa,CAAC,OAAO,EAAE;QAC3B,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GAAG,OAAO,CAAC,MAAM,CAAC;IAwBb,eAAe,CAAC,OAAO,EAAE;QAC7B,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;QAC9B,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;CAsBpB;AAED,eAAe,SAAS,CAAC"}
|