@zz1996/dbhub-dameng 0.1.0
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/LICENSE +21 -0
- package/README.md +143 -0
- package/dbhub.dameng.toml.example +19 -0
- package/dist/chunk-2A2QF3CS.js +2463 -0
- package/dist/chunk-IPK7BYBL.js +35 -0
- package/dist/chunk-RTB262PR.js +60 -0
- package/dist/chunk-SQA2ISDE.js +503 -0
- package/dist/chunk-WVVMH6FJ.js +49 -0
- package/dist/chunk-ZNQTMARG.js +127 -0
- package/dist/dameng-DC7OP4VV.js +504 -0
- package/dist/demo/employee-sqlite/employee.sql +117 -0
- package/dist/demo/employee-sqlite/load_department.sql +10 -0
- package/dist/demo/employee-sqlite/load_dept_emp.sql +1103 -0
- package/dist/demo/employee-sqlite/load_dept_manager.sql +17 -0
- package/dist/demo/employee-sqlite/load_employee.sql +1000 -0
- package/dist/demo/employee-sqlite/load_salary1.sql +9488 -0
- package/dist/demo/employee-sqlite/load_title.sql +1470 -0
- package/dist/demo-loader-PSMTLZ2T.js +46 -0
- package/dist/index.d.ts +1 -0
- package/dist/index.js +1841 -0
- package/dist/mariadb-7F72IRB4.js +520 -0
- package/dist/mysql-A43SL7UM.js +527 -0
- package/dist/postgres-VFNLBWMF.js +564 -0
- package/dist/registry-XSX2VZTD.js +12 -0
- package/dist/sqlite-IOUAYHGE.js +412 -0
- package/dist/sqlserver-OE4SFH4B.js +587 -0
- package/docs/README.md +14 -0
- package/docs/dameng-connector-plan.md +143 -0
- package/package.json +110 -0
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
// src/utils/identifier-quoter.ts
|
|
2
|
+
function quoteIdentifier(identifier, dbType) {
|
|
3
|
+
if (/[\0\x08\x09\x1a\n\r]/.test(identifier)) {
|
|
4
|
+
throw new Error(`Invalid identifier: contains control characters: ${identifier}`);
|
|
5
|
+
}
|
|
6
|
+
if (!identifier) {
|
|
7
|
+
throw new Error("Identifier cannot be empty");
|
|
8
|
+
}
|
|
9
|
+
switch (dbType) {
|
|
10
|
+
case "postgres":
|
|
11
|
+
case "sqlite":
|
|
12
|
+
case "dameng":
|
|
13
|
+
return `"${identifier.replace(/"/g, '""')}"`;
|
|
14
|
+
case "mysql":
|
|
15
|
+
case "mariadb":
|
|
16
|
+
return `\`${identifier.replace(/`/g, "``")}\``;
|
|
17
|
+
case "sqlserver":
|
|
18
|
+
return `[${identifier.replace(/]/g, "]]")}]`;
|
|
19
|
+
default:
|
|
20
|
+
return `"${identifier.replace(/"/g, '""')}"`;
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
function quoteQualifiedIdentifier(tableName, schemaName, dbType) {
|
|
24
|
+
const quotedTable = quoteIdentifier(tableName, dbType);
|
|
25
|
+
if (schemaName) {
|
|
26
|
+
const quotedSchema = quoteIdentifier(schemaName, dbType);
|
|
27
|
+
return `${quotedSchema}.${quotedTable}`;
|
|
28
|
+
}
|
|
29
|
+
return quotedTable;
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
export {
|
|
33
|
+
quoteIdentifier,
|
|
34
|
+
quoteQualifiedIdentifier
|
|
35
|
+
};
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
// src/utils/multi-statement-result-parser.ts
|
|
2
|
+
function isMetadataObject(element) {
|
|
3
|
+
if (!element || typeof element !== "object" || Array.isArray(element)) {
|
|
4
|
+
return false;
|
|
5
|
+
}
|
|
6
|
+
return "affectedRows" in element || "insertId" in element || "fieldCount" in element || "warningStatus" in element;
|
|
7
|
+
}
|
|
8
|
+
function isMultiStatementResult(results) {
|
|
9
|
+
if (!Array.isArray(results) || results.length === 0) {
|
|
10
|
+
return false;
|
|
11
|
+
}
|
|
12
|
+
const firstElement = results[0];
|
|
13
|
+
return isMetadataObject(firstElement) || Array.isArray(firstElement);
|
|
14
|
+
}
|
|
15
|
+
function extractRowsFromMultiStatement(results) {
|
|
16
|
+
if (!Array.isArray(results)) {
|
|
17
|
+
return [];
|
|
18
|
+
}
|
|
19
|
+
const allRows = [];
|
|
20
|
+
for (const result of results) {
|
|
21
|
+
if (Array.isArray(result)) {
|
|
22
|
+
allRows.push(...result);
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
return allRows;
|
|
26
|
+
}
|
|
27
|
+
function extractAffectedRows(results) {
|
|
28
|
+
if (isMetadataObject(results)) {
|
|
29
|
+
return results.affectedRows || 0;
|
|
30
|
+
}
|
|
31
|
+
if (!Array.isArray(results)) {
|
|
32
|
+
return 0;
|
|
33
|
+
}
|
|
34
|
+
if (isMultiStatementResult(results)) {
|
|
35
|
+
let totalAffected = 0;
|
|
36
|
+
for (const result of results) {
|
|
37
|
+
if (isMetadataObject(result)) {
|
|
38
|
+
totalAffected += result.affectedRows || 0;
|
|
39
|
+
} else if (Array.isArray(result)) {
|
|
40
|
+
totalAffected += result.length;
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
return totalAffected;
|
|
44
|
+
}
|
|
45
|
+
return results.length;
|
|
46
|
+
}
|
|
47
|
+
function parseQueryResults(results) {
|
|
48
|
+
if (!Array.isArray(results)) {
|
|
49
|
+
return [];
|
|
50
|
+
}
|
|
51
|
+
if (isMultiStatementResult(results)) {
|
|
52
|
+
return extractRowsFromMultiStatement(results);
|
|
53
|
+
}
|
|
54
|
+
return results;
|
|
55
|
+
}
|
|
56
|
+
|
|
57
|
+
export {
|
|
58
|
+
extractAffectedRows,
|
|
59
|
+
parseQueryResults
|
|
60
|
+
};
|
|
@@ -0,0 +1,503 @@
|
|
|
1
|
+
// src/connectors/interface.ts
|
|
2
|
+
var _ConnectorRegistry = class _ConnectorRegistry {
|
|
3
|
+
/**
|
|
4
|
+
* Register a new connector
|
|
5
|
+
*/
|
|
6
|
+
static register(connector) {
|
|
7
|
+
_ConnectorRegistry.connectors.set(connector.id, connector);
|
|
8
|
+
}
|
|
9
|
+
/**
|
|
10
|
+
* Get a connector by ID
|
|
11
|
+
*/
|
|
12
|
+
static getConnector(id) {
|
|
13
|
+
return _ConnectorRegistry.connectors.get(id) || null;
|
|
14
|
+
}
|
|
15
|
+
/**
|
|
16
|
+
* Get connector for a DSN string
|
|
17
|
+
* Tries to find a connector that can handle the given DSN format
|
|
18
|
+
*/
|
|
19
|
+
static getConnectorForDSN(dsn) {
|
|
20
|
+
for (const connector of _ConnectorRegistry.connectors.values()) {
|
|
21
|
+
if (connector.dsnParser.isValidDSN(dsn)) {
|
|
22
|
+
return connector;
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
return null;
|
|
26
|
+
}
|
|
27
|
+
/**
|
|
28
|
+
* Get all available connector IDs
|
|
29
|
+
*/
|
|
30
|
+
static getAvailableConnectors() {
|
|
31
|
+
return Array.from(_ConnectorRegistry.connectors.keys());
|
|
32
|
+
}
|
|
33
|
+
/**
|
|
34
|
+
* Get sample DSN for a specific connector
|
|
35
|
+
*/
|
|
36
|
+
static getSampleDSN(connectorType) {
|
|
37
|
+
const connector = _ConnectorRegistry.getConnector(connectorType);
|
|
38
|
+
if (!connector) return null;
|
|
39
|
+
return connector.dsnParser.getSampleDSN();
|
|
40
|
+
}
|
|
41
|
+
/**
|
|
42
|
+
* Get all available sample DSNs
|
|
43
|
+
*/
|
|
44
|
+
static getAllSampleDSNs() {
|
|
45
|
+
const samples = {};
|
|
46
|
+
for (const [id, connector] of _ConnectorRegistry.connectors.entries()) {
|
|
47
|
+
samples[id] = connector.dsnParser.getSampleDSN();
|
|
48
|
+
}
|
|
49
|
+
return samples;
|
|
50
|
+
}
|
|
51
|
+
};
|
|
52
|
+
_ConnectorRegistry.connectors = /* @__PURE__ */ new Map();
|
|
53
|
+
var ConnectorRegistry = _ConnectorRegistry;
|
|
54
|
+
|
|
55
|
+
// src/utils/safe-url.ts
|
|
56
|
+
var SafeURL = class {
|
|
57
|
+
/**
|
|
58
|
+
* Parse a URL and handle special characters in passwords
|
|
59
|
+
* This is a safe alternative to the URL constructor
|
|
60
|
+
*
|
|
61
|
+
* @param urlString - The DSN string to parse
|
|
62
|
+
*/
|
|
63
|
+
constructor(urlString) {
|
|
64
|
+
this.protocol = "";
|
|
65
|
+
this.hostname = "";
|
|
66
|
+
this.port = "";
|
|
67
|
+
this.pathname = "";
|
|
68
|
+
this.username = "";
|
|
69
|
+
this.password = "";
|
|
70
|
+
this.searchParams = /* @__PURE__ */ new Map();
|
|
71
|
+
if (!urlString || urlString.trim() === "") {
|
|
72
|
+
throw new Error("URL string cannot be empty");
|
|
73
|
+
}
|
|
74
|
+
try {
|
|
75
|
+
const protocolSeparator = urlString.indexOf("://");
|
|
76
|
+
if (protocolSeparator !== -1) {
|
|
77
|
+
this.protocol = urlString.substring(0, protocolSeparator + 1);
|
|
78
|
+
urlString = urlString.substring(protocolSeparator + 3);
|
|
79
|
+
} else {
|
|
80
|
+
throw new Error('Invalid URL format: missing protocol (e.g., "mysql://")');
|
|
81
|
+
}
|
|
82
|
+
const questionMarkIndex = urlString.indexOf("?");
|
|
83
|
+
let queryParams = "";
|
|
84
|
+
if (questionMarkIndex !== -1) {
|
|
85
|
+
queryParams = urlString.substring(questionMarkIndex + 1);
|
|
86
|
+
urlString = urlString.substring(0, questionMarkIndex);
|
|
87
|
+
queryParams.split("&").forEach((pair) => {
|
|
88
|
+
const parts = pair.split("=");
|
|
89
|
+
if (parts.length === 2 && parts[0] && parts[1]) {
|
|
90
|
+
this.searchParams.set(parts[0], decodeURIComponent(parts[1]));
|
|
91
|
+
}
|
|
92
|
+
});
|
|
93
|
+
}
|
|
94
|
+
const atIndex = urlString.indexOf("@");
|
|
95
|
+
if (atIndex !== -1) {
|
|
96
|
+
const auth = urlString.substring(0, atIndex);
|
|
97
|
+
urlString = urlString.substring(atIndex + 1);
|
|
98
|
+
const colonIndex2 = auth.indexOf(":");
|
|
99
|
+
if (colonIndex2 !== -1) {
|
|
100
|
+
this.username = auth.substring(0, colonIndex2);
|
|
101
|
+
this.password = auth.substring(colonIndex2 + 1);
|
|
102
|
+
this.username = decodeURIComponent(this.username);
|
|
103
|
+
this.password = decodeURIComponent(this.password);
|
|
104
|
+
} else {
|
|
105
|
+
this.username = auth;
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
const pathSeparatorIndex = urlString.indexOf("/");
|
|
109
|
+
if (pathSeparatorIndex !== -1) {
|
|
110
|
+
this.pathname = urlString.substring(pathSeparatorIndex);
|
|
111
|
+
urlString = urlString.substring(0, pathSeparatorIndex);
|
|
112
|
+
}
|
|
113
|
+
const colonIndex = urlString.indexOf(":");
|
|
114
|
+
if (colonIndex !== -1) {
|
|
115
|
+
this.hostname = urlString.substring(0, colonIndex);
|
|
116
|
+
this.port = urlString.substring(colonIndex + 1);
|
|
117
|
+
} else {
|
|
118
|
+
this.hostname = urlString;
|
|
119
|
+
}
|
|
120
|
+
if (this.protocol === "") {
|
|
121
|
+
throw new Error("Invalid URL: protocol is required");
|
|
122
|
+
}
|
|
123
|
+
} catch (error) {
|
|
124
|
+
throw new Error(`Failed to parse URL: ${error instanceof Error ? error.message : String(error)}`);
|
|
125
|
+
}
|
|
126
|
+
}
|
|
127
|
+
/**
|
|
128
|
+
* Helper method to safely get a parameter from query string
|
|
129
|
+
*
|
|
130
|
+
* @param name - The parameter name to retrieve
|
|
131
|
+
* @returns The parameter value or null if not found
|
|
132
|
+
*/
|
|
133
|
+
getSearchParam(name) {
|
|
134
|
+
return this.searchParams.has(name) ? this.searchParams.get(name) : null;
|
|
135
|
+
}
|
|
136
|
+
/**
|
|
137
|
+
* Helper method to iterate over all parameters
|
|
138
|
+
*
|
|
139
|
+
* @param callback - Function to call for each parameter
|
|
140
|
+
*/
|
|
141
|
+
forEachSearchParam(callback) {
|
|
142
|
+
this.searchParams.forEach((value, key) => callback(value, key));
|
|
143
|
+
}
|
|
144
|
+
};
|
|
145
|
+
|
|
146
|
+
// src/utils/dsn-obfuscate.ts
|
|
147
|
+
function parseConnectionInfoFromDSN(dsn) {
|
|
148
|
+
if (!dsn) {
|
|
149
|
+
return null;
|
|
150
|
+
}
|
|
151
|
+
try {
|
|
152
|
+
const type = getDatabaseTypeFromDSN(dsn);
|
|
153
|
+
if (typeof type === "undefined") {
|
|
154
|
+
return null;
|
|
155
|
+
}
|
|
156
|
+
if (type === "sqlite") {
|
|
157
|
+
const prefix = "sqlite:///";
|
|
158
|
+
if (dsn.length > prefix.length) {
|
|
159
|
+
const rawPath = dsn.substring(prefix.length);
|
|
160
|
+
const firstChar = rawPath[0];
|
|
161
|
+
const isWindowsDrive = rawPath.length > 1 && rawPath[1] === ":";
|
|
162
|
+
const isSpecialPath = firstChar === ":" || firstChar === "." || firstChar === "~" || isWindowsDrive;
|
|
163
|
+
return {
|
|
164
|
+
type,
|
|
165
|
+
database: isSpecialPath ? rawPath : "/" + rawPath
|
|
166
|
+
};
|
|
167
|
+
}
|
|
168
|
+
return { type };
|
|
169
|
+
}
|
|
170
|
+
const url = new SafeURL(dsn);
|
|
171
|
+
const info = { type };
|
|
172
|
+
if (url.hostname) {
|
|
173
|
+
info.host = url.hostname;
|
|
174
|
+
}
|
|
175
|
+
if (url.port) {
|
|
176
|
+
info.port = parseInt(url.port, 10);
|
|
177
|
+
}
|
|
178
|
+
if (url.pathname && url.pathname.length > 1) {
|
|
179
|
+
info.database = url.pathname.substring(1);
|
|
180
|
+
}
|
|
181
|
+
if (url.username) {
|
|
182
|
+
info.user = url.username;
|
|
183
|
+
}
|
|
184
|
+
return info;
|
|
185
|
+
} catch {
|
|
186
|
+
return null;
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
function obfuscateDSNPassword(dsn) {
|
|
190
|
+
if (!dsn) {
|
|
191
|
+
return dsn;
|
|
192
|
+
}
|
|
193
|
+
try {
|
|
194
|
+
const type = getDatabaseTypeFromDSN(dsn);
|
|
195
|
+
if (type === "sqlite") {
|
|
196
|
+
return dsn;
|
|
197
|
+
}
|
|
198
|
+
const url = new SafeURL(dsn);
|
|
199
|
+
if (!url.password) {
|
|
200
|
+
return dsn;
|
|
201
|
+
}
|
|
202
|
+
const obfuscatedPassword = "*".repeat(Math.min(url.password.length, 8));
|
|
203
|
+
const protocol = dsn.split(":")[0];
|
|
204
|
+
let result;
|
|
205
|
+
if (url.username) {
|
|
206
|
+
result = `${protocol}://${url.username}:${obfuscatedPassword}@${url.hostname}`;
|
|
207
|
+
} else {
|
|
208
|
+
result = `${protocol}://${obfuscatedPassword}@${url.hostname}`;
|
|
209
|
+
}
|
|
210
|
+
if (url.port) {
|
|
211
|
+
result += `:${url.port}`;
|
|
212
|
+
}
|
|
213
|
+
result += url.pathname;
|
|
214
|
+
if (url.searchParams.size > 0) {
|
|
215
|
+
const params = [];
|
|
216
|
+
url.forEachSearchParam((value, key) => {
|
|
217
|
+
params.push(`${encodeURIComponent(key)}=${encodeURIComponent(value)}`);
|
|
218
|
+
});
|
|
219
|
+
result += `?${params.join("&")}`;
|
|
220
|
+
}
|
|
221
|
+
return result;
|
|
222
|
+
} catch {
|
|
223
|
+
return dsn;
|
|
224
|
+
}
|
|
225
|
+
}
|
|
226
|
+
function getDatabaseTypeFromDSN(dsn) {
|
|
227
|
+
if (!dsn) {
|
|
228
|
+
return void 0;
|
|
229
|
+
}
|
|
230
|
+
const protocol = dsn.split(":")[0];
|
|
231
|
+
return protocolToConnectorType(protocol);
|
|
232
|
+
}
|
|
233
|
+
function protocolToConnectorType(protocol) {
|
|
234
|
+
const mapping = {
|
|
235
|
+
"postgres": "postgres",
|
|
236
|
+
"postgresql": "postgres",
|
|
237
|
+
"mysql": "mysql",
|
|
238
|
+
"mariadb": "mariadb",
|
|
239
|
+
"sqlserver": "sqlserver",
|
|
240
|
+
"sqlite": "sqlite",
|
|
241
|
+
"dameng": "dameng",
|
|
242
|
+
"dm": "dameng"
|
|
243
|
+
};
|
|
244
|
+
return mapping[protocol];
|
|
245
|
+
}
|
|
246
|
+
function getDefaultPortForType(type) {
|
|
247
|
+
const ports = {
|
|
248
|
+
"postgres": 5432,
|
|
249
|
+
"mysql": 3306,
|
|
250
|
+
"mariadb": 3306,
|
|
251
|
+
"sqlserver": 1433,
|
|
252
|
+
"sqlite": void 0,
|
|
253
|
+
"dameng": 5236
|
|
254
|
+
};
|
|
255
|
+
return ports[type];
|
|
256
|
+
}
|
|
257
|
+
|
|
258
|
+
// src/utils/sql-parser.ts
|
|
259
|
+
var TokenType = { Plain: 0, Comment: 1, QuotedBlock: 2 };
|
|
260
|
+
function plainToken(i) {
|
|
261
|
+
return { type: TokenType.Plain, end: i + 1 };
|
|
262
|
+
}
|
|
263
|
+
function scanSingleLineComment(sql, i) {
|
|
264
|
+
if (sql[i] !== "-" || sql[i + 1] !== "-") {
|
|
265
|
+
return null;
|
|
266
|
+
}
|
|
267
|
+
let j = i;
|
|
268
|
+
while (j < sql.length && sql[j] !== "\n") {
|
|
269
|
+
j++;
|
|
270
|
+
}
|
|
271
|
+
return { type: TokenType.Comment, end: j };
|
|
272
|
+
}
|
|
273
|
+
function scanSingleLineCommentMySQL(sql, i) {
|
|
274
|
+
if (sql[i] !== "-" || sql[i + 1] !== "-") {
|
|
275
|
+
return null;
|
|
276
|
+
}
|
|
277
|
+
const next = sql[i + 2];
|
|
278
|
+
if (next !== void 0 && next.charCodeAt(0) > 32 && next.charCodeAt(0) !== 127) {
|
|
279
|
+
return null;
|
|
280
|
+
}
|
|
281
|
+
let j = i;
|
|
282
|
+
while (j < sql.length && sql[j] !== "\n") {
|
|
283
|
+
j++;
|
|
284
|
+
}
|
|
285
|
+
return { type: TokenType.Comment, end: j };
|
|
286
|
+
}
|
|
287
|
+
function scanMultiLineComment(sql, i) {
|
|
288
|
+
if (sql[i] !== "/" || sql[i + 1] !== "*") {
|
|
289
|
+
return null;
|
|
290
|
+
}
|
|
291
|
+
let j = i + 2;
|
|
292
|
+
while (j < sql.length && !(sql[j] === "*" && sql[j + 1] === "/")) {
|
|
293
|
+
j++;
|
|
294
|
+
}
|
|
295
|
+
if (j < sql.length) {
|
|
296
|
+
j += 2;
|
|
297
|
+
}
|
|
298
|
+
return { type: TokenType.Comment, end: j };
|
|
299
|
+
}
|
|
300
|
+
function scanMultiLineCommentMySQL(sql, i) {
|
|
301
|
+
if (sql[i] !== "/" || sql[i + 1] !== "*") {
|
|
302
|
+
return null;
|
|
303
|
+
}
|
|
304
|
+
const next = sql[i + 2];
|
|
305
|
+
const nextNext = sql[i + 3];
|
|
306
|
+
if (next === "!" || next === "M" && nextNext === "!") {
|
|
307
|
+
return null;
|
|
308
|
+
}
|
|
309
|
+
return scanMultiLineComment(sql, i);
|
|
310
|
+
}
|
|
311
|
+
function scanNestedMultiLineComment(sql, i) {
|
|
312
|
+
if (sql[i] !== "/" || sql[i + 1] !== "*") {
|
|
313
|
+
return null;
|
|
314
|
+
}
|
|
315
|
+
let j = i + 2;
|
|
316
|
+
let depth = 1;
|
|
317
|
+
while (j < sql.length && depth > 0) {
|
|
318
|
+
if (sql[j] === "/" && sql[j + 1] === "*") {
|
|
319
|
+
depth++;
|
|
320
|
+
j += 2;
|
|
321
|
+
} else if (sql[j] === "*" && sql[j + 1] === "/") {
|
|
322
|
+
depth--;
|
|
323
|
+
j += 2;
|
|
324
|
+
} else {
|
|
325
|
+
j++;
|
|
326
|
+
}
|
|
327
|
+
}
|
|
328
|
+
return { type: TokenType.Comment, end: j };
|
|
329
|
+
}
|
|
330
|
+
function scanSingleQuotedString(sql, i) {
|
|
331
|
+
if (sql[i] !== "'") {
|
|
332
|
+
return null;
|
|
333
|
+
}
|
|
334
|
+
let j = i + 1;
|
|
335
|
+
while (j < sql.length) {
|
|
336
|
+
if (sql[j] === "'" && sql[j + 1] === "'") {
|
|
337
|
+
j += 2;
|
|
338
|
+
} else if (sql[j] === "'") {
|
|
339
|
+
j++;
|
|
340
|
+
break;
|
|
341
|
+
} else {
|
|
342
|
+
j++;
|
|
343
|
+
}
|
|
344
|
+
}
|
|
345
|
+
return { type: TokenType.QuotedBlock, end: j };
|
|
346
|
+
}
|
|
347
|
+
function scanDoubleQuotedString(sql, i) {
|
|
348
|
+
if (sql[i] !== '"') {
|
|
349
|
+
return null;
|
|
350
|
+
}
|
|
351
|
+
let j = i + 1;
|
|
352
|
+
while (j < sql.length) {
|
|
353
|
+
if (sql[j] === '"' && sql[j + 1] === '"') {
|
|
354
|
+
j += 2;
|
|
355
|
+
} else if (sql[j] === '"') {
|
|
356
|
+
j++;
|
|
357
|
+
break;
|
|
358
|
+
} else {
|
|
359
|
+
j++;
|
|
360
|
+
}
|
|
361
|
+
}
|
|
362
|
+
return { type: TokenType.QuotedBlock, end: j };
|
|
363
|
+
}
|
|
364
|
+
var dollarQuoteOpenRegex = /^\$([a-zA-Z_]\w*)?\$/;
|
|
365
|
+
function scanDollarQuotedBlock(sql, i) {
|
|
366
|
+
if (sql[i] !== "$") {
|
|
367
|
+
return null;
|
|
368
|
+
}
|
|
369
|
+
const next = sql[i + 1];
|
|
370
|
+
if (next >= "0" && next <= "9") {
|
|
371
|
+
return null;
|
|
372
|
+
}
|
|
373
|
+
const remaining = sql.substring(i);
|
|
374
|
+
const m = dollarQuoteOpenRegex.exec(remaining);
|
|
375
|
+
if (!m) {
|
|
376
|
+
return null;
|
|
377
|
+
}
|
|
378
|
+
const tag = m[0];
|
|
379
|
+
const bodyStart = i + tag.length;
|
|
380
|
+
const closeIdx = sql.indexOf(tag, bodyStart);
|
|
381
|
+
const end = closeIdx !== -1 ? closeIdx + tag.length : sql.length;
|
|
382
|
+
return { type: TokenType.QuotedBlock, end };
|
|
383
|
+
}
|
|
384
|
+
function scanBacktickQuotedIdentifier(sql, i) {
|
|
385
|
+
if (sql[i] !== "`") {
|
|
386
|
+
return null;
|
|
387
|
+
}
|
|
388
|
+
let j = i + 1;
|
|
389
|
+
while (j < sql.length) {
|
|
390
|
+
if (sql[j] === "`" && sql[j + 1] === "`") {
|
|
391
|
+
j += 2;
|
|
392
|
+
} else if (sql[j] === "`") {
|
|
393
|
+
j++;
|
|
394
|
+
break;
|
|
395
|
+
} else {
|
|
396
|
+
j++;
|
|
397
|
+
}
|
|
398
|
+
}
|
|
399
|
+
return { type: TokenType.QuotedBlock, end: j };
|
|
400
|
+
}
|
|
401
|
+
function scanBracketQuotedIdentifier(sql, i) {
|
|
402
|
+
if (sql[i] !== "[") {
|
|
403
|
+
return null;
|
|
404
|
+
}
|
|
405
|
+
let j = i + 1;
|
|
406
|
+
while (j < sql.length) {
|
|
407
|
+
if (sql[j] === "]" && sql[j + 1] === "]") {
|
|
408
|
+
j += 2;
|
|
409
|
+
} else if (sql[j] === "]") {
|
|
410
|
+
j++;
|
|
411
|
+
break;
|
|
412
|
+
} else {
|
|
413
|
+
j++;
|
|
414
|
+
}
|
|
415
|
+
}
|
|
416
|
+
return { type: TokenType.QuotedBlock, end: j };
|
|
417
|
+
}
|
|
418
|
+
function scanTokenAnsi(sql, i) {
|
|
419
|
+
return scanSingleLineComment(sql, i) ?? scanMultiLineComment(sql, i) ?? scanSingleQuotedString(sql, i) ?? scanDoubleQuotedString(sql, i) ?? plainToken(i);
|
|
420
|
+
}
|
|
421
|
+
function scanTokenPostgres(sql, i) {
|
|
422
|
+
return scanSingleLineComment(sql, i) ?? scanNestedMultiLineComment(sql, i) ?? scanSingleQuotedString(sql, i) ?? scanDoubleQuotedString(sql, i) ?? scanDollarQuotedBlock(sql, i) ?? plainToken(i);
|
|
423
|
+
}
|
|
424
|
+
function scanTokenMySQL(sql, i) {
|
|
425
|
+
return scanSingleLineCommentMySQL(sql, i) ?? scanMultiLineCommentMySQL(sql, i) ?? scanSingleQuotedString(sql, i) ?? scanDoubleQuotedString(sql, i) ?? scanBacktickQuotedIdentifier(sql, i) ?? plainToken(i);
|
|
426
|
+
}
|
|
427
|
+
function scanTokenSQLite(sql, i) {
|
|
428
|
+
return scanSingleLineComment(sql, i) ?? scanMultiLineComment(sql, i) ?? scanSingleQuotedString(sql, i) ?? scanDoubleQuotedString(sql, i) ?? scanBacktickQuotedIdentifier(sql, i) ?? scanBracketQuotedIdentifier(sql, i) ?? plainToken(i);
|
|
429
|
+
}
|
|
430
|
+
function scanTokenSQLServer(sql, i) {
|
|
431
|
+
return scanSingleLineComment(sql, i) ?? scanMultiLineComment(sql, i) ?? scanSingleQuotedString(sql, i) ?? scanDoubleQuotedString(sql, i) ?? scanBracketQuotedIdentifier(sql, i) ?? plainToken(i);
|
|
432
|
+
}
|
|
433
|
+
var dialectScanners = {
|
|
434
|
+
postgres: scanTokenPostgres,
|
|
435
|
+
mysql: scanTokenMySQL,
|
|
436
|
+
mariadb: scanTokenMySQL,
|
|
437
|
+
sqlite: scanTokenSQLite,
|
|
438
|
+
sqlserver: scanTokenSQLServer,
|
|
439
|
+
dameng: scanTokenAnsi
|
|
440
|
+
};
|
|
441
|
+
function getScanner(dialect) {
|
|
442
|
+
return dialect ? dialectScanners[dialect] ?? scanTokenAnsi : scanTokenAnsi;
|
|
443
|
+
}
|
|
444
|
+
function stripCommentsAndStrings(sql, dialect) {
|
|
445
|
+
const scanToken = getScanner(dialect);
|
|
446
|
+
const parts = [];
|
|
447
|
+
let plainStart = -1;
|
|
448
|
+
let i = 0;
|
|
449
|
+
while (i < sql.length) {
|
|
450
|
+
const token = scanToken(sql, i);
|
|
451
|
+
if (token.type === TokenType.Plain) {
|
|
452
|
+
if (plainStart === -1) {
|
|
453
|
+
plainStart = i;
|
|
454
|
+
}
|
|
455
|
+
} else {
|
|
456
|
+
if (plainStart !== -1) {
|
|
457
|
+
parts.push(sql.substring(plainStart, i));
|
|
458
|
+
plainStart = -1;
|
|
459
|
+
}
|
|
460
|
+
parts.push(" ");
|
|
461
|
+
}
|
|
462
|
+
i = token.end;
|
|
463
|
+
}
|
|
464
|
+
if (plainStart !== -1) {
|
|
465
|
+
parts.push(sql.substring(plainStart));
|
|
466
|
+
}
|
|
467
|
+
return parts.join("");
|
|
468
|
+
}
|
|
469
|
+
function splitSQLStatements(sql, dialect) {
|
|
470
|
+
const scanToken = getScanner(dialect);
|
|
471
|
+
const statements = [];
|
|
472
|
+
let stmtStart = 0;
|
|
473
|
+
let i = 0;
|
|
474
|
+
while (i < sql.length) {
|
|
475
|
+
if (sql[i] === ";") {
|
|
476
|
+
const trimmed2 = sql.substring(stmtStart, i).trim();
|
|
477
|
+
if (trimmed2.length > 0) {
|
|
478
|
+
statements.push(trimmed2);
|
|
479
|
+
}
|
|
480
|
+
stmtStart = i + 1;
|
|
481
|
+
i++;
|
|
482
|
+
continue;
|
|
483
|
+
}
|
|
484
|
+
const token = scanToken(sql, i);
|
|
485
|
+
i = token.end;
|
|
486
|
+
}
|
|
487
|
+
const trimmed = sql.substring(stmtStart).trim();
|
|
488
|
+
if (trimmed.length > 0) {
|
|
489
|
+
statements.push(trimmed);
|
|
490
|
+
}
|
|
491
|
+
return statements;
|
|
492
|
+
}
|
|
493
|
+
|
|
494
|
+
export {
|
|
495
|
+
ConnectorRegistry,
|
|
496
|
+
SafeURL,
|
|
497
|
+
parseConnectionInfoFromDSN,
|
|
498
|
+
obfuscateDSNPassword,
|
|
499
|
+
getDatabaseTypeFromDSN,
|
|
500
|
+
getDefaultPortForType,
|
|
501
|
+
stripCommentsAndStrings,
|
|
502
|
+
splitSQLStatements
|
|
503
|
+
};
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
// src/utils/module-loader.ts
|
|
2
|
+
var MISSING_MODULE_RE = /Cannot find (?:package|module) '([^']+)'/;
|
|
3
|
+
function isDriverNotInstalled(err, driver) {
|
|
4
|
+
if (!(err instanceof Error) || !("code" in err) || err.code !== "ERR_MODULE_NOT_FOUND") {
|
|
5
|
+
return false;
|
|
6
|
+
}
|
|
7
|
+
const match = err.message.match(MISSING_MODULE_RE);
|
|
8
|
+
if (!match) {
|
|
9
|
+
return false;
|
|
10
|
+
}
|
|
11
|
+
const missingSpecifier = match[1];
|
|
12
|
+
return missingSpecifier === driver || missingSpecifier.startsWith(`${driver}/`);
|
|
13
|
+
}
|
|
14
|
+
function isModuleNotFound(err) {
|
|
15
|
+
if (!(err instanceof Error) || !("code" in err)) {
|
|
16
|
+
return false;
|
|
17
|
+
}
|
|
18
|
+
const code = err.code;
|
|
19
|
+
return code === "ERR_MODULE_NOT_FOUND" || code === "MODULE_NOT_FOUND";
|
|
20
|
+
}
|
|
21
|
+
async function loadConnectors(connectorModules) {
|
|
22
|
+
await Promise.all(
|
|
23
|
+
connectorModules.map(async ({ load, name, driver }) => {
|
|
24
|
+
try {
|
|
25
|
+
await load();
|
|
26
|
+
} catch (err) {
|
|
27
|
+
if (isDriverNotInstalled(err, driver)) {
|
|
28
|
+
console.error(
|
|
29
|
+
`Skipping ${name} connector: driver package "${driver}" not installed.`
|
|
30
|
+
);
|
|
31
|
+
} else if (isModuleNotFound(err)) {
|
|
32
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
33
|
+
const match = msg.match(MISSING_MODULE_RE);
|
|
34
|
+
const missing = match ? match[1] : "unknown";
|
|
35
|
+
console.error(
|
|
36
|
+
`Skipping ${name} connector: required dependency "${missing}" not installed.`
|
|
37
|
+
);
|
|
38
|
+
} else {
|
|
39
|
+
throw err;
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
})
|
|
43
|
+
);
|
|
44
|
+
}
|
|
45
|
+
|
|
46
|
+
export {
|
|
47
|
+
isDriverNotInstalled,
|
|
48
|
+
loadConnectors
|
|
49
|
+
};
|