@nmakarov/cli-toolkit 0.14.2 → 0.18.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/dist/args.cjs +49 -9
- package/dist/args.cjs.map +1 -1
- package/dist/args.js +49 -9
- package/dist/args.js.map +1 -1
- package/dist/cli-runner.cjs +5006 -0
- package/dist/cli-runner.cjs.map +1 -0
- package/dist/cli-runner.js +4989 -0
- package/dist/cli-runner.js.map +1 -0
- package/dist/db.cjs +9 -2
- package/dist/db.cjs.map +1 -1
- package/dist/db.js +9 -2
- package/dist/db.js.map +1 -1
- package/dist/errors.cjs +17 -0
- package/dist/errors.cjs.map +1 -1
- package/dist/errors.js +15 -0
- package/dist/errors.js.map +1 -1
- package/dist/filedatabase.cjs +110 -37
- package/dist/filedatabase.cjs.map +1 -1
- package/dist/filedatabase.js +110 -36
- package/dist/filedatabase.js.map +1 -1
- package/dist/http-client.cjs +44 -34
- package/dist/http-client.cjs.map +1 -1
- package/dist/http-client.js +44 -34
- package/dist/http-client.js.map +1 -1
- package/dist/http-client2.cjs +1728 -0
- package/dist/http-client2.cjs.map +1 -0
- package/dist/http-client2.js +1690 -0
- package/dist/http-client2.js.map +1 -0
- package/dist/index.cjs +1456 -112
- package/dist/index.cjs.map +1 -1
- package/dist/index.js +1436 -110
- package/dist/index.js.map +1 -1
- package/dist/init.cjs +199 -82
- package/dist/init.cjs.map +1 -1
- package/dist/init.js +199 -82
- package/dist/init.js.map +1 -1
- package/dist/logger.cjs +28 -42
- package/dist/logger.cjs.map +1 -1
- package/dist/logger.js +28 -42
- package/dist/logger.js.map +1 -1
- package/dist/mock-server.cjs +205 -359
- package/dist/mock-server.cjs.map +1 -1
- package/dist/mock-server.js +203 -359
- package/dist/mock-server.js.map +1 -1
- package/dist/params.cjs +114 -15
- package/dist/params.cjs.map +1 -1
- package/dist/params.js +114 -15
- package/dist/params.js.map +1 -1
- package/dist/tasks.cjs +2295 -0
- package/dist/tasks.cjs.map +1 -0
- package/dist/tasks.js +2240 -0
- package/dist/tasks.js.map +1 -0
- package/dist/utils.cjs +15 -2
- package/dist/utils.cjs.map +1 -1
- package/dist/utils.js +12 -1
- package/dist/utils.js.map +1 -1
- package/package.json +18 -3
package/dist/tasks.cjs
ADDED
|
@@ -0,0 +1,2295 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __create = Object.create;
|
|
3
|
+
var __defProp = Object.defineProperty;
|
|
4
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
5
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
6
|
+
var __getProtoOf = Object.getPrototypeOf;
|
|
7
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
8
|
+
var __export = (target, all) => {
|
|
9
|
+
for (var name in all)
|
|
10
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
11
|
+
};
|
|
12
|
+
var __copyProps = (to, from, except, desc) => {
|
|
13
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
14
|
+
for (let key of __getOwnPropNames(from))
|
|
15
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
16
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
17
|
+
}
|
|
18
|
+
return to;
|
|
19
|
+
};
|
|
20
|
+
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
|
|
21
|
+
// If the importer is in node compatibility mode or this is not an ESM
|
|
22
|
+
// file that has been converted to a CommonJS file using a Babel-
|
|
23
|
+
// compatible transform (i.e. "__esModule" has not been set), then set
|
|
24
|
+
// "default" to the CommonJS "module.exports" for node compatibility.
|
|
25
|
+
isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
|
|
26
|
+
mod
|
|
27
|
+
));
|
|
28
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
29
|
+
|
|
30
|
+
// src/tasks.ts
|
|
31
|
+
var tasks_exports = {};
|
|
32
|
+
__export(tasks_exports, {
|
|
33
|
+
TaskMaster: () => TaskMaster,
|
|
34
|
+
TaskPing: () => TaskPing,
|
|
35
|
+
TaskSampleProcess: () => TaskSampleProcess,
|
|
36
|
+
TaskShellCommand: () => TaskShellCommand,
|
|
37
|
+
TaskStopRunner: () => TaskStopRunner,
|
|
38
|
+
TaskSumAB: () => TaskSumAB,
|
|
39
|
+
TaskSystemInfo: () => TaskSystemInfo,
|
|
40
|
+
TasksManager: () => TasksManager,
|
|
41
|
+
TasksRegistry: () => TasksRegistry,
|
|
42
|
+
appendTaskIpcLog: () => appendTaskIpcLog,
|
|
43
|
+
defaultTasksRegistry: () => defaultTasksRegistry,
|
|
44
|
+
enqueueStopTask: () => enqueueStopTask,
|
|
45
|
+
enqueueTask: () => enqueueTask,
|
|
46
|
+
ensureTaskTables: () => ensureTaskTables,
|
|
47
|
+
queueToTableNames: () => queueToTableNames,
|
|
48
|
+
runNodeTaskScript: () => runNodeTaskScript,
|
|
49
|
+
runTasksLoop: () => runTasksLoop,
|
|
50
|
+
updateTaskProgress: () => updateTaskProgress,
|
|
51
|
+
waitForTaskResult: () => waitForTaskResult
|
|
52
|
+
});
|
|
53
|
+
module.exports = __toCommonJS(tasks_exports);
|
|
54
|
+
|
|
55
|
+
// src/utils/date-utils.ts
|
|
56
|
+
function isTimestampFolder(folderName) {
|
|
57
|
+
const isoRegex = /^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(Z|\.\d{3}Z)$/;
|
|
58
|
+
if (!isoRegex.test(folderName)) {
|
|
59
|
+
return false;
|
|
60
|
+
}
|
|
61
|
+
const date = new Date(folderName);
|
|
62
|
+
return !isNaN(date.getTime()) && date.getTime() > 0;
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
// src/utils/fs-utils.ts
|
|
66
|
+
var import_fs = __toESM(require("fs"), 1);
|
|
67
|
+
var import_path = __toESM(require("path"), 1);
|
|
68
|
+
async function ensurePath(...pathParts) {
|
|
69
|
+
const fullPath = import_path.default.resolve(...pathParts);
|
|
70
|
+
if (!import_fs.default.existsSync(fullPath)) {
|
|
71
|
+
await import_fs.default.promises.mkdir(fullPath, { recursive: true });
|
|
72
|
+
}
|
|
73
|
+
return fullPath;
|
|
74
|
+
}
|
|
75
|
+
function getFileExtension(dataType) {
|
|
76
|
+
switch (dataType) {
|
|
77
|
+
case "json-array":
|
|
78
|
+
case "json-object":
|
|
79
|
+
return "json";
|
|
80
|
+
case "text":
|
|
81
|
+
return "txt";
|
|
82
|
+
case "xml":
|
|
83
|
+
return "xml";
|
|
84
|
+
default:
|
|
85
|
+
return "json";
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
|
|
89
|
+
// src/utils/os-utils.ts
|
|
90
|
+
var import_fs2 = __toESM(require("fs"), 1);
|
|
91
|
+
var import_path2 = __toESM(require("path"), 1);
|
|
92
|
+
var import_child_process = require("child_process");
|
|
93
|
+
function getFreeDiskSpace(targetPath) {
|
|
94
|
+
try {
|
|
95
|
+
let pathToCheck = targetPath;
|
|
96
|
+
if (!import_fs2.default.existsSync(targetPath)) {
|
|
97
|
+
const parentDir = import_path2.default.dirname(targetPath);
|
|
98
|
+
if (import_fs2.default.existsSync(parentDir)) {
|
|
99
|
+
pathToCheck = parentDir;
|
|
100
|
+
} else {
|
|
101
|
+
pathToCheck = process.platform === "win32" ? "C:\\" : "/";
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
if (process.platform === "win32") {
|
|
105
|
+
return null;
|
|
106
|
+
} else {
|
|
107
|
+
const stdout = (0, import_child_process.execSync)(`df -k "${pathToCheck}"`, { encoding: "utf8" });
|
|
108
|
+
const lines = stdout.trim().split("\n");
|
|
109
|
+
const parts = lines[1].split(/\s+/);
|
|
110
|
+
const freeKb = parseInt(parts[3], 10);
|
|
111
|
+
return freeKb * 1024;
|
|
112
|
+
}
|
|
113
|
+
} catch (error) {
|
|
114
|
+
return null;
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
|
|
118
|
+
// src/utils/format-utils.ts
|
|
119
|
+
function bytesToHumanReadable(bytes) {
|
|
120
|
+
if (bytes === 0) return "0 B";
|
|
121
|
+
const k = 1024;
|
|
122
|
+
const sizes = ["B", "KB", "MB", "GB", "TB", "PB"];
|
|
123
|
+
const i = Math.floor(Math.log(bytes) / Math.log(k));
|
|
124
|
+
return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + " " + sizes[i];
|
|
125
|
+
}
|
|
126
|
+
|
|
127
|
+
// src/utils/core-utils.ts
|
|
128
|
+
function sleepMs(ms) {
|
|
129
|
+
return new Promise((resolve) => setTimeout(resolve, ms));
|
|
130
|
+
}
|
|
131
|
+
function toJsonColumn(value) {
|
|
132
|
+
if (value === void 0 || value === null) return null;
|
|
133
|
+
return JSON.stringify(value);
|
|
134
|
+
}
|
|
135
|
+
|
|
136
|
+
// src/tasks/taskUtils.ts
|
|
137
|
+
var import_node_crypto = require("crypto");
|
|
138
|
+
function getDb(context) {
|
|
139
|
+
const db = context.db;
|
|
140
|
+
if (!db) {
|
|
141
|
+
throw new Error("Tasks component requires context.db. Initialize DB first and attach to context.");
|
|
142
|
+
}
|
|
143
|
+
return db;
|
|
144
|
+
}
|
|
145
|
+
function queueToTableNames(queue) {
|
|
146
|
+
if (!/^[a-zA-Z_][a-zA-Z0-9_]*$/.test(queue)) {
|
|
147
|
+
throw new Error(`Invalid queue name "${queue}". Use letters, numbers, underscore only.`);
|
|
148
|
+
}
|
|
149
|
+
return {
|
|
150
|
+
tasksTable: queue,
|
|
151
|
+
historyTable: `${queue}_history`
|
|
152
|
+
};
|
|
153
|
+
}
|
|
154
|
+
async function ensureTaskTables(context, options = {}) {
|
|
155
|
+
const queue = options.queue ?? "tasks";
|
|
156
|
+
const recreate = options.recreate ?? false;
|
|
157
|
+
const db = getDb(context);
|
|
158
|
+
const { tasksTable, historyTable } = queueToTableNames(queue);
|
|
159
|
+
const needsTasks = recreate ? true : !await db.tableExists(tasksTable);
|
|
160
|
+
const needsHistory = recreate ? true : !await db.tableExists(historyTable);
|
|
161
|
+
if (recreate) {
|
|
162
|
+
await db.schema.dropTableIfExists(historyTable);
|
|
163
|
+
await db.schema.dropTableIfExists(tasksTable);
|
|
164
|
+
}
|
|
165
|
+
if (needsTasks) {
|
|
166
|
+
await db.raw(`CREATE EXTENSION IF NOT EXISTS "uuid-ossp"`);
|
|
167
|
+
await db.schema.createTable(tasksTable, (t) => {
|
|
168
|
+
t.uuid("id").primary().defaultTo(db.raw("uuid_generate_v4()"));
|
|
169
|
+
t.timestamp("created_at").notNullable().defaultTo(db.fn.now());
|
|
170
|
+
t.timestamp("started_at");
|
|
171
|
+
t.timestamp("completed_at");
|
|
172
|
+
t.integer("priority").notNullable().defaultTo(0);
|
|
173
|
+
t.text("schedule");
|
|
174
|
+
t.timestamp("past_due").defaultTo(null);
|
|
175
|
+
t.text("target").notNullable();
|
|
176
|
+
t.text("task").notNullable();
|
|
177
|
+
t.json("params");
|
|
178
|
+
t.text("opid");
|
|
179
|
+
t.timestamp("paused_at").defaultTo(null);
|
|
180
|
+
t.text("progress");
|
|
181
|
+
t.boolean("success");
|
|
182
|
+
t.json("results");
|
|
183
|
+
});
|
|
184
|
+
await db.schema.alterTable(tasksTable, (t) => {
|
|
185
|
+
t.index(["target", "started_at", "created_at"], `${tasksTable}_target_started_created_idx`);
|
|
186
|
+
t.index(["target", "past_due", "priority", "created_at"], `${tasksTable}_target_past_due_priority_created_idx`);
|
|
187
|
+
t.index(["target", "task"], `${tasksTable}_target_task_idx`);
|
|
188
|
+
});
|
|
189
|
+
}
|
|
190
|
+
const tasksHasOpid = await db.schema.hasColumn(tasksTable, "opid");
|
|
191
|
+
if (!tasksHasOpid) {
|
|
192
|
+
await db.schema.alterTable(tasksTable, (t) => {
|
|
193
|
+
t.text("opid");
|
|
194
|
+
});
|
|
195
|
+
}
|
|
196
|
+
const tasksHasPausedAt = await db.schema.hasColumn(tasksTable, "paused_at");
|
|
197
|
+
if (!tasksHasPausedAt) {
|
|
198
|
+
await db.schema.alterTable(tasksTable, (t) => {
|
|
199
|
+
t.timestamp("paused_at").defaultTo(null);
|
|
200
|
+
});
|
|
201
|
+
}
|
|
202
|
+
if (needsHistory) {
|
|
203
|
+
await db.schema.createTable(historyTable, (t) => {
|
|
204
|
+
t.uuid("id").notNullable();
|
|
205
|
+
t.timestamp("created_at").notNullable();
|
|
206
|
+
t.timestamp("started_at");
|
|
207
|
+
t.timestamp("completed_at");
|
|
208
|
+
t.integer("priority").notNullable().defaultTo(0);
|
|
209
|
+
t.text("schedule");
|
|
210
|
+
t.timestamp("past_due").defaultTo(null);
|
|
211
|
+
t.text("target").notNullable();
|
|
212
|
+
t.text("task").notNullable();
|
|
213
|
+
t.json("params");
|
|
214
|
+
t.text("opid");
|
|
215
|
+
t.text("progress");
|
|
216
|
+
t.boolean("success");
|
|
217
|
+
t.json("results");
|
|
218
|
+
});
|
|
219
|
+
await db.schema.alterTable(historyTable, (t) => {
|
|
220
|
+
t.index(["target", "created_at"], `${historyTable}_target_created_idx`);
|
|
221
|
+
t.index(["task", "created_at"], `${historyTable}_task_created_idx`);
|
|
222
|
+
});
|
|
223
|
+
}
|
|
224
|
+
const historyHasOpid = await db.schema.hasColumn(historyTable, "opid");
|
|
225
|
+
if (!historyHasOpid) {
|
|
226
|
+
await db.schema.alterTable(historyTable, (t) => {
|
|
227
|
+
t.text("opid");
|
|
228
|
+
});
|
|
229
|
+
}
|
|
230
|
+
}
|
|
231
|
+
async function enqueueTask(context, options) {
|
|
232
|
+
const db = getDb(context);
|
|
233
|
+
const queue = options.queue ?? "tasks";
|
|
234
|
+
const { tasksTable } = queueToTableNames(queue);
|
|
235
|
+
const id = (0, import_node_crypto.randomUUID)();
|
|
236
|
+
await db(tasksTable).insert({
|
|
237
|
+
id,
|
|
238
|
+
target: options.target,
|
|
239
|
+
task: options.task,
|
|
240
|
+
params: toJsonColumn(options.params ?? null),
|
|
241
|
+
opid: options.opid ?? null,
|
|
242
|
+
priority: options.priority ?? 0,
|
|
243
|
+
schedule: options.schedule ?? null
|
|
244
|
+
});
|
|
245
|
+
return id;
|
|
246
|
+
}
|
|
247
|
+
async function updateTaskProgress(context, tasksTable, taskId, progress) {
|
|
248
|
+
const db = getDb(context);
|
|
249
|
+
await db(tasksTable).where({ id: taskId }).update({
|
|
250
|
+
progress: typeof progress === "string" ? progress : JSON.stringify(progress)
|
|
251
|
+
});
|
|
252
|
+
}
|
|
253
|
+
|
|
254
|
+
// src/filedatabase/index.ts
|
|
255
|
+
var import_fs3 = __toESM(require("fs"), 1);
|
|
256
|
+
var import_path3 = __toESM(require("path"), 1);
|
|
257
|
+
|
|
258
|
+
// src/filedatabase/serializers.ts
|
|
259
|
+
function detectDataType(data) {
|
|
260
|
+
if (Array.isArray(data)) {
|
|
261
|
+
return "json-array";
|
|
262
|
+
} else if (typeof data === "object" && data !== null) {
|
|
263
|
+
return "json-object";
|
|
264
|
+
} else if (typeof data === "string") {
|
|
265
|
+
const trimmed = data.trim();
|
|
266
|
+
if (trimmed.startsWith("<?xml") || trimmed.startsWith("<")) {
|
|
267
|
+
return "xml";
|
|
268
|
+
}
|
|
269
|
+
return "text";
|
|
270
|
+
} else {
|
|
271
|
+
return "text";
|
|
272
|
+
}
|
|
273
|
+
}
|
|
274
|
+
function serializeData(data) {
|
|
275
|
+
const dataType = detectDataType(data);
|
|
276
|
+
if (dataType === "json-array" || dataType === "json-object") {
|
|
277
|
+
return JSON.stringify(data, null, 4);
|
|
278
|
+
} else {
|
|
279
|
+
return String(data);
|
|
280
|
+
}
|
|
281
|
+
}
|
|
282
|
+
function deserializeData(rawData, dataType) {
|
|
283
|
+
if (dataType === "json-array" || dataType === "json-object") {
|
|
284
|
+
return JSON.parse(rawData);
|
|
285
|
+
} else {
|
|
286
|
+
return rawData;
|
|
287
|
+
}
|
|
288
|
+
}
|
|
289
|
+
|
|
290
|
+
// src/errors.ts
|
|
291
|
+
var FrameworkError = class extends Error {
|
|
292
|
+
constructor(message) {
|
|
293
|
+
super(message);
|
|
294
|
+
this.name = "FrameworkError";
|
|
295
|
+
}
|
|
296
|
+
};
|
|
297
|
+
var ParamError = class extends FrameworkError {
|
|
298
|
+
constructor(message) {
|
|
299
|
+
super(message);
|
|
300
|
+
this.name = "ParamError";
|
|
301
|
+
}
|
|
302
|
+
};
|
|
303
|
+
var FileDatabaseError = class extends FrameworkError {
|
|
304
|
+
constructor(message) {
|
|
305
|
+
super(message);
|
|
306
|
+
this.name = "FileDatabaseError";
|
|
307
|
+
}
|
|
308
|
+
};
|
|
309
|
+
|
|
310
|
+
// src/filedatabase/index.ts
|
|
311
|
+
var FileDatabase = class _FileDatabase {
|
|
312
|
+
basePath;
|
|
313
|
+
namespace;
|
|
314
|
+
tableName = null;
|
|
315
|
+
versioned;
|
|
316
|
+
maxVersions;
|
|
317
|
+
pageSize;
|
|
318
|
+
useMetadata;
|
|
319
|
+
freeSpaceThreshold;
|
|
320
|
+
logger;
|
|
321
|
+
// Current operation state
|
|
322
|
+
currentVersion = null;
|
|
323
|
+
currentVersionFolder = null;
|
|
324
|
+
currentFileNumber = 0;
|
|
325
|
+
currentRecord = 0;
|
|
326
|
+
hasReadFirstPage = false;
|
|
327
|
+
lastFileData = null;
|
|
328
|
+
metadata;
|
|
329
|
+
// Synopsis calculation functions
|
|
330
|
+
fileSynopsisFunction = null;
|
|
331
|
+
versionSynopsisFunction = null;
|
|
332
|
+
/**
|
|
333
|
+
* Constructor - accepts context as first parameter (new pattern)
|
|
334
|
+
* or config object (legacy pattern for backward compatibility)
|
|
335
|
+
*/
|
|
336
|
+
constructor(contextOrConfig, options) {
|
|
337
|
+
let config;
|
|
338
|
+
if (contextOrConfig && typeof contextOrConfig === "object" && "params" in contextOrConfig) {
|
|
339
|
+
const context = contextOrConfig;
|
|
340
|
+
const opts = options || {};
|
|
341
|
+
const defs = {
|
|
342
|
+
basePath: "string default ./data",
|
|
343
|
+
namespace: "string default default",
|
|
344
|
+
tableName: "string",
|
|
345
|
+
maxVersions: "number default 5",
|
|
346
|
+
pageSize: "number default 5000"
|
|
347
|
+
};
|
|
348
|
+
const discovered = context.params.getAllForModule(defs);
|
|
349
|
+
config = { ...discovered, ...opts, logger: context.logger };
|
|
350
|
+
} else {
|
|
351
|
+
config = contextOrConfig;
|
|
352
|
+
}
|
|
353
|
+
if (!config.basePath) {
|
|
354
|
+
throw new ParamError("[FileDatabase] basePath is required");
|
|
355
|
+
}
|
|
356
|
+
this.basePath = config.basePath;
|
|
357
|
+
this.namespace = config.namespace || "default";
|
|
358
|
+
this.tableName = config.tableName || null;
|
|
359
|
+
this.versioned = config.versioned ?? true;
|
|
360
|
+
this.maxVersions = config.maxVersions || 5;
|
|
361
|
+
this.pageSize = config.pageSize || 5e3;
|
|
362
|
+
this.useMetadata = config.useMetadata !== false;
|
|
363
|
+
this.freeSpaceThreshold = config.freeSpaceThreshold || 100 * 1024 * 1024;
|
|
364
|
+
this.logger = config.logger || console;
|
|
365
|
+
this.metadata = this.getDefaultMetadata();
|
|
366
|
+
}
|
|
367
|
+
/**
|
|
368
|
+
* Initialize FileDatabase from context and options.
|
|
369
|
+
* Params are read via getAllForModule("filedatabase", defs) for --showUsedParams grouping.
|
|
370
|
+
*/
|
|
371
|
+
static init(context, options) {
|
|
372
|
+
return new _FileDatabase(context, options ?? {});
|
|
373
|
+
}
|
|
374
|
+
/**
|
|
375
|
+
* Get default metadata structure
|
|
376
|
+
*/
|
|
377
|
+
getDefaultMetadata() {
|
|
378
|
+
return {
|
|
379
|
+
version: this.currentVersion || null,
|
|
380
|
+
files: [],
|
|
381
|
+
createdAt: (/* @__PURE__ */ new Date()).toISOString(),
|
|
382
|
+
modifiedAt: (/* @__PURE__ */ new Date()).toISOString(),
|
|
383
|
+
totalRecords: 0,
|
|
384
|
+
synopsis: null,
|
|
385
|
+
dataType: null
|
|
386
|
+
};
|
|
387
|
+
}
|
|
388
|
+
/**
|
|
389
|
+
* Get the destination path (basePath/namespace/tableName[/version])
|
|
390
|
+
*/
|
|
391
|
+
getDestinationPath(version) {
|
|
392
|
+
const errors = ["basePath", "namespace", "tableName"].filter((prop) => !this[prop]).map((prop) => `${prop} is not set`);
|
|
393
|
+
if (errors.length) {
|
|
394
|
+
throw new FileDatabaseError(`[FileDatabase] ${errors.join("; ")}`);
|
|
395
|
+
}
|
|
396
|
+
let parts = [this.basePath, this.namespace];
|
|
397
|
+
if (this.tableName) {
|
|
398
|
+
parts.push(...this.tableName.split("/"));
|
|
399
|
+
}
|
|
400
|
+
if (this.versioned && version) {
|
|
401
|
+
parts.push(version);
|
|
402
|
+
}
|
|
403
|
+
return import_path3.default.resolve(...parts);
|
|
404
|
+
}
|
|
405
|
+
/**
|
|
406
|
+
* Set current version and version folder
|
|
407
|
+
*/
|
|
408
|
+
async setCurrentVersion(version) {
|
|
409
|
+
this.currentVersion = version;
|
|
410
|
+
this.currentVersionFolder = await ensurePath(this.getDestinationPath(), version);
|
|
411
|
+
}
|
|
412
|
+
/**
|
|
413
|
+
* Create a new version folder with comprehensive timestamp logic
|
|
414
|
+
* Only works in versioned mode
|
|
415
|
+
*/
|
|
416
|
+
async makeNewVersion() {
|
|
417
|
+
if (!this.versioned) {
|
|
418
|
+
throw new FileDatabaseError("makeNewVersion() only works in versioned mode");
|
|
419
|
+
}
|
|
420
|
+
this.metadata = this.getDefaultMetadata();
|
|
421
|
+
const existingVersions = await this.getVersions();
|
|
422
|
+
let versionName;
|
|
423
|
+
if (existingVersions.length > 0) {
|
|
424
|
+
const maxTimestamp = existingVersions.reduce((max, version) => {
|
|
425
|
+
const versionDate = new Date(version.replace("Z", ""));
|
|
426
|
+
const maxDate2 = new Date(max.replace("Z", ""));
|
|
427
|
+
return versionDate > maxDate2 ? version : max;
|
|
428
|
+
});
|
|
429
|
+
const maxDate = new Date(maxTimestamp.replace("Z", ""));
|
|
430
|
+
const nextDate = new Date(maxDate.getTime() + 1e3);
|
|
431
|
+
versionName = nextDate.toISOString().split(".")[0] + "Z";
|
|
432
|
+
} else {
|
|
433
|
+
const now = /* @__PURE__ */ new Date();
|
|
434
|
+
versionName = now.toISOString().split(".")[0] + "Z";
|
|
435
|
+
}
|
|
436
|
+
await this.setCurrentVersion(versionName);
|
|
437
|
+
this.currentFileNumber = 0;
|
|
438
|
+
const versions = await this.getVersions();
|
|
439
|
+
while (versions.length > this.maxVersions) {
|
|
440
|
+
const versionToDelete = import_path3.default.resolve(this.getDestinationPath(), versions.shift());
|
|
441
|
+
this.logger.silly?.(`[FileDatabase] Deleting old version: ${versionToDelete}`);
|
|
442
|
+
await import_fs3.default.promises.rm(versionToDelete, { recursive: true, force: true });
|
|
443
|
+
}
|
|
444
|
+
return versionName;
|
|
445
|
+
}
|
|
446
|
+
/**
|
|
447
|
+
* Get list of all versions (sorted chronologically)
|
|
448
|
+
* Only works in versioned mode
|
|
449
|
+
*/
|
|
450
|
+
async getVersions() {
|
|
451
|
+
if (!this.versioned) {
|
|
452
|
+
return [];
|
|
453
|
+
}
|
|
454
|
+
const destPath = this.getDestinationPath();
|
|
455
|
+
try {
|
|
456
|
+
await ensurePath(destPath);
|
|
457
|
+
const items = await import_fs3.default.promises.readdir(destPath);
|
|
458
|
+
const versions = items.filter((item) => {
|
|
459
|
+
const itemPath = import_path3.default.join(destPath, item);
|
|
460
|
+
const stat = import_fs3.default.statSync(itemPath);
|
|
461
|
+
return stat.isDirectory() && isTimestampFolder(item);
|
|
462
|
+
});
|
|
463
|
+
return versions.sort();
|
|
464
|
+
} catch (error) {
|
|
465
|
+
return [];
|
|
466
|
+
}
|
|
467
|
+
}
|
|
468
|
+
/**
|
|
469
|
+
* Get the latest version (most recent timestamp)
|
|
470
|
+
* Only works in versioned mode
|
|
471
|
+
* @returns Latest version string or null if no versions
|
|
472
|
+
*/
|
|
473
|
+
async getLatestVersion() {
|
|
474
|
+
if (!this.versioned) {
|
|
475
|
+
throw new FileDatabaseError("getLatestVersion() only works in versioned mode");
|
|
476
|
+
}
|
|
477
|
+
const versions = await this.getVersions();
|
|
478
|
+
if (versions.length === 0) {
|
|
479
|
+
return null;
|
|
480
|
+
}
|
|
481
|
+
return versions[versions.length - 1];
|
|
482
|
+
}
|
|
483
|
+
/**
|
|
484
|
+
* Check if any data exists in this table
|
|
485
|
+
* Works for both versioned and non-versioned modes
|
|
486
|
+
* @returns true if data exists
|
|
487
|
+
*/
|
|
488
|
+
async hasData() {
|
|
489
|
+
const tablePath = this.getDestinationPath();
|
|
490
|
+
if (!import_fs3.default.existsSync(tablePath)) {
|
|
491
|
+
return false;
|
|
492
|
+
}
|
|
493
|
+
if (this.versioned) {
|
|
494
|
+
const versions = await this.getVersions();
|
|
495
|
+
return versions.length > 0;
|
|
496
|
+
} else {
|
|
497
|
+
const items = await import_fs3.default.promises.readdir(tablePath);
|
|
498
|
+
return items.some(
|
|
499
|
+
(item) => item === "metadata.json" || item.match(/^\d{6}\.(json|txt|xml)$/) || item.endsWith(".json")
|
|
500
|
+
);
|
|
501
|
+
}
|
|
502
|
+
}
|
|
503
|
+
/**
|
|
504
|
+
* Auto-detect the data format in this table
|
|
505
|
+
* Used when reading existing data
|
|
506
|
+
* @returns Format detection result
|
|
507
|
+
*/
|
|
508
|
+
async detectDataFormat() {
|
|
509
|
+
const tablePath = this.getDestinationPath();
|
|
510
|
+
if (!import_fs3.default.existsSync(tablePath)) {
|
|
511
|
+
return { versioned: false, hasMetadata: false, dataType: null };
|
|
512
|
+
}
|
|
513
|
+
const items = await import_fs3.default.promises.readdir(tablePath);
|
|
514
|
+
if (items.includes("metadata.json")) {
|
|
515
|
+
const metadata = JSON.parse(
|
|
516
|
+
await import_fs3.default.promises.readFile(import_path3.default.join(tablePath, "metadata.json"), "utf8")
|
|
517
|
+
);
|
|
518
|
+
return {
|
|
519
|
+
versioned: false,
|
|
520
|
+
hasMetadata: true,
|
|
521
|
+
dataType: metadata.dataType || null
|
|
522
|
+
};
|
|
523
|
+
}
|
|
524
|
+
const versionFolders = items.filter((item) => {
|
|
525
|
+
const itemPath = import_path3.default.join(tablePath, item);
|
|
526
|
+
const stat = import_fs3.default.statSync(itemPath);
|
|
527
|
+
return stat.isDirectory() && isTimestampFolder(item);
|
|
528
|
+
});
|
|
529
|
+
if (versionFolders.length > 0) {
|
|
530
|
+
const latestVersion = versionFolders.sort().pop();
|
|
531
|
+
const versionMetadataPath = import_path3.default.join(tablePath, latestVersion, "metadata.json");
|
|
532
|
+
return {
|
|
533
|
+
versioned: true,
|
|
534
|
+
hasMetadata: import_fs3.default.existsSync(versionMetadataPath),
|
|
535
|
+
dataType: null
|
|
536
|
+
};
|
|
537
|
+
}
|
|
538
|
+
const dataFiles = items.filter((f) => f.match(/^\d{6}\.(json|txt|xml)$/));
|
|
539
|
+
if (dataFiles.length > 0) {
|
|
540
|
+
return {
|
|
541
|
+
versioned: false,
|
|
542
|
+
hasMetadata: false,
|
|
543
|
+
dataType: null
|
|
544
|
+
};
|
|
545
|
+
}
|
|
546
|
+
return { versioned: false, hasMetadata: false, dataType: null };
|
|
547
|
+
}
|
|
548
|
+
/**
|
|
549
|
+
* Load metadata from JSON file
|
|
550
|
+
*/
|
|
551
|
+
async loadMetadataJson(version) {
|
|
552
|
+
const metadataFile = import_path3.default.join(this.getDestinationPath(), version, "metadata.json");
|
|
553
|
+
if (import_fs3.default.existsSync(metadataFile)) {
|
|
554
|
+
try {
|
|
555
|
+
const rawData = await import_fs3.default.promises.readFile(metadataFile, "utf8");
|
|
556
|
+
return JSON.parse(rawData);
|
|
557
|
+
} catch (e) {
|
|
558
|
+
throw new FileDatabaseError(`Failed to read metadata for version "${version}": ${e.message}`);
|
|
559
|
+
}
|
|
560
|
+
}
|
|
561
|
+
return null;
|
|
562
|
+
}
|
|
563
|
+
/**
|
|
564
|
+
* Build metadata by scanning files in a version folder (backward compatibility)
|
|
565
|
+
* Reads all files to get accurate counts - used when synopsis calculation is needed
|
|
566
|
+
*/
|
|
567
|
+
async figureMetadataFromVersionFiles(version) {
|
|
568
|
+
const versionPath = import_path3.default.join(this.getDestinationPath(), version);
|
|
569
|
+
if (!import_fs3.default.existsSync(versionPath)) {
|
|
570
|
+
return this.getDefaultMetadata();
|
|
571
|
+
}
|
|
572
|
+
const files = (await import_fs3.default.promises.readdir(versionPath)).filter((file) => file !== "metadata.json" && !file.startsWith(".")).sort();
|
|
573
|
+
const metadata = this.getDefaultMetadata();
|
|
574
|
+
metadata.version = version;
|
|
575
|
+
metadata.files = [];
|
|
576
|
+
let totalRecords = 0;
|
|
577
|
+
let detectedDataType = null;
|
|
578
|
+
for (let i = 0; i < files.length; i++) {
|
|
579
|
+
const fileName = files[i];
|
|
580
|
+
const filePath = import_path3.default.join(versionPath, fileName);
|
|
581
|
+
try {
|
|
582
|
+
const rawData = await import_fs3.default.promises.readFile(filePath, "utf8");
|
|
583
|
+
const extension = import_path3.default.extname(fileName).toLowerCase();
|
|
584
|
+
let dataType = "text";
|
|
585
|
+
if (extension === ".json") {
|
|
586
|
+
dataType = "json-array";
|
|
587
|
+
} else if (extension === ".xml") {
|
|
588
|
+
dataType = "xml";
|
|
589
|
+
}
|
|
590
|
+
const fileData = deserializeData(rawData, dataType);
|
|
591
|
+
const recordsCount = Array.isArray(fileData) ? fileData.length : 1;
|
|
592
|
+
if (detectedDataType === null) {
|
|
593
|
+
detectedDataType = detectDataType(fileData);
|
|
594
|
+
}
|
|
595
|
+
const fileInfo = {
|
|
596
|
+
number: i + 1,
|
|
597
|
+
recordsCount,
|
|
598
|
+
fileName
|
|
599
|
+
};
|
|
600
|
+
metadata.files.push(fileInfo);
|
|
601
|
+
totalRecords += recordsCount;
|
|
602
|
+
} catch (error) {
|
|
603
|
+
this.logger.error?.(`[FileDatabase] Failed to read file ${fileName}: ${error.message}`);
|
|
604
|
+
}
|
|
605
|
+
}
|
|
606
|
+
metadata.totalRecords = totalRecords;
|
|
607
|
+
metadata.dataType = detectedDataType;
|
|
608
|
+
return metadata;
|
|
609
|
+
}
|
|
610
|
+
/**
|
|
611
|
+
* Build metadata optimized - only reads first and last files
|
|
612
|
+
* Assumes all middle files have the same record count as the first file
|
|
613
|
+
* Much faster for large datasets with many files
|
|
614
|
+
*/
|
|
615
|
+
async buildMetadataOptimized(version) {
|
|
616
|
+
const versionPath = import_path3.default.join(this.getDestinationPath(), version);
|
|
617
|
+
if (!import_fs3.default.existsSync(versionPath)) {
|
|
618
|
+
return this.getDefaultMetadata();
|
|
619
|
+
}
|
|
620
|
+
const files = (await import_fs3.default.promises.readdir(versionPath)).filter((file) => file !== "metadata.json" && !file.startsWith(".")).sort();
|
|
621
|
+
if (files.length === 0) {
|
|
622
|
+
return this.getDefaultMetadata();
|
|
623
|
+
}
|
|
624
|
+
const metadata = this.getDefaultMetadata();
|
|
625
|
+
metadata.version = version;
|
|
626
|
+
metadata.files = files.map((fileName, index) => ({
|
|
627
|
+
number: index + 1,
|
|
628
|
+
recordsCount: 0,
|
|
629
|
+
fileName
|
|
630
|
+
}));
|
|
631
|
+
const firstFile = metadata.files[0];
|
|
632
|
+
const firstFilePath = import_path3.default.join(versionPath, firstFile.fileName);
|
|
633
|
+
const firstFileRaw = await import_fs3.default.promises.readFile(firstFilePath, "utf8");
|
|
634
|
+
let firstFileData;
|
|
635
|
+
try {
|
|
636
|
+
firstFileData = JSON.parse(firstFileRaw);
|
|
637
|
+
} catch (e) {
|
|
638
|
+
firstFileData = firstFileRaw;
|
|
639
|
+
}
|
|
640
|
+
metadata.dataType = detectDataType(firstFileData);
|
|
641
|
+
if (metadata.dataType === "json-array") {
|
|
642
|
+
const firstFileCount = Array.isArray(firstFileData) ? firstFileData.length : 1;
|
|
643
|
+
firstFile.recordsCount = firstFileCount;
|
|
644
|
+
for (let i = 1; i < metadata.files.length - 1; i++) {
|
|
645
|
+
metadata.files[i].recordsCount = firstFileCount;
|
|
646
|
+
}
|
|
647
|
+
if (files.length > 1) {
|
|
648
|
+
const lastFile = metadata.files[metadata.files.length - 1];
|
|
649
|
+
const lastFilePath = import_path3.default.join(versionPath, lastFile.fileName);
|
|
650
|
+
const lastFileRaw = await import_fs3.default.promises.readFile(lastFilePath, "utf8");
|
|
651
|
+
const lastFileData = deserializeData(lastFileRaw, metadata.dataType);
|
|
652
|
+
lastFile.recordsCount = Array.isArray(lastFileData) ? lastFileData.length : 1;
|
|
653
|
+
}
|
|
654
|
+
metadata.totalRecords = metadata.files.reduce((sum, file) => sum + file.recordsCount, 0);
|
|
655
|
+
} else {
|
|
656
|
+
metadata.files.forEach((file) => {
|
|
657
|
+
file.recordsCount = 1;
|
|
658
|
+
});
|
|
659
|
+
metadata.totalRecords = files.length;
|
|
660
|
+
}
|
|
661
|
+
return metadata;
|
|
662
|
+
}
|
|
663
|
+
/**
|
|
664
|
+
* Figure out metadata - tries JSON first, then builds from files
|
|
665
|
+
* Uses optimized building when no synopsis calculation is needed
|
|
666
|
+
*/
|
|
667
|
+
async figureMetadata(version, useOptimized = true) {
|
|
668
|
+
if (this.useMetadata) {
|
|
669
|
+
const metadata = await this.loadMetadataJson(version);
|
|
670
|
+
if (metadata) {
|
|
671
|
+
return metadata;
|
|
672
|
+
}
|
|
673
|
+
}
|
|
674
|
+
if (useOptimized && !this.fileSynopsisFunction && !this.versionSynopsisFunction) {
|
|
675
|
+
return await this.buildMetadataOptimized(version);
|
|
676
|
+
}
|
|
677
|
+
return await this.figureMetadataFromVersionFiles(version);
|
|
678
|
+
}
|
|
679
|
+
/**
|
|
680
|
+
* Load version metadata (main entry point for loading)
|
|
681
|
+
*/
|
|
682
|
+
async loadVersionMetadata(version) {
|
|
683
|
+
const metadata = await this.figureMetadata(version);
|
|
684
|
+
this.metadata = metadata;
|
|
685
|
+
return metadata;
|
|
686
|
+
}
|
|
687
|
+
/**
|
|
688
|
+
* Save version metadata to file
|
|
689
|
+
*/
|
|
690
|
+
async saveVersionMetadata(metadata) {
|
|
691
|
+
if (!this.useMetadata) {
|
|
692
|
+
return;
|
|
693
|
+
}
|
|
694
|
+
const metadataToSave = metadata || this.metadata;
|
|
695
|
+
let metadataFile;
|
|
696
|
+
if (this.versioned) {
|
|
697
|
+
if (!this.currentVersion) {
|
|
698
|
+
return;
|
|
699
|
+
}
|
|
700
|
+
metadataFile = import_path3.default.join(this.getDestinationPath(), this.currentVersion, "metadata.json");
|
|
701
|
+
} else {
|
|
702
|
+
metadataFile = import_path3.default.join(this.getDestinationPath(), "metadata.json");
|
|
703
|
+
}
|
|
704
|
+
await import_fs3.default.promises.writeFile(metadataFile, JSON.stringify(metadataToSave, null, 4), "utf8");
|
|
705
|
+
}
|
|
706
|
+
/**
|
|
707
|
+
* Create a new file entry in metadata
|
|
708
|
+
*/
|
|
709
|
+
makeNewFile() {
|
|
710
|
+
this.currentFileNumber = (this.currentFileNumber || 0) + 1;
|
|
711
|
+
const dataType = this.metadata.dataType || "json-array";
|
|
712
|
+
const fileEntry = {
|
|
713
|
+
number: this.currentFileNumber,
|
|
714
|
+
recordsCount: 0,
|
|
715
|
+
fileName: `${this.currentFileNumber.toString().padStart(6, "0")}.${getFileExtension(dataType)}`
|
|
716
|
+
};
|
|
717
|
+
this.metadata.files.push(fileEntry);
|
|
718
|
+
this.lastFileData = null;
|
|
719
|
+
this.logger.silly?.(`[FileDatabase] Created new file: ${fileEntry.fileName}, fileNumber: ${this.currentFileNumber}`);
|
|
720
|
+
}
|
|
721
|
+
/**
|
|
722
|
+
* Figure out what data to write and which file to use (for pagination)
|
|
723
|
+
* @param data - Data to write
|
|
724
|
+
* @param targetFileIndex - Optional index of existing file to overwrite (when customMetadata matches)
|
|
725
|
+
* @param forceNewFile - If true, always create a new file (when customMetadata provided but no match)
|
|
726
|
+
*/
|
|
727
|
+
figureOutDataAndFileToWrite(data, targetFileIndex = null, forceNewFile = false) {
|
|
728
|
+
let dataToWrite;
|
|
729
|
+
let dataLeftOver;
|
|
730
|
+
const incomingDataType = detectDataType(data);
|
|
731
|
+
if (this.metadata.dataType !== incomingDataType) {
|
|
732
|
+
this.metadata.dataType = incomingDataType;
|
|
733
|
+
}
|
|
734
|
+
if (targetFileIndex !== null && targetFileIndex < this.metadata.files.length) {
|
|
735
|
+
const targetFile = this.metadata.files[targetFileIndex];
|
|
736
|
+
if (!Array.isArray(data)) {
|
|
737
|
+
dataToWrite = data;
|
|
738
|
+
dataLeftOver = null;
|
|
739
|
+
return { dataToWrite, dataLeftOver, fileName: targetFile.fileName };
|
|
740
|
+
} else {
|
|
741
|
+
dataToWrite = data.slice(0, this.pageSize);
|
|
742
|
+
dataLeftOver = data.slice(this.pageSize);
|
|
743
|
+
this.lastFileData = dataToWrite;
|
|
744
|
+
return { dataToWrite, dataLeftOver, fileName: targetFile.fileName };
|
|
745
|
+
}
|
|
746
|
+
}
|
|
747
|
+
let newlyCreatedFileIndex = null;
|
|
748
|
+
if (forceNewFile) {
|
|
749
|
+
const filesBeforeCreate = this.metadata.files.length;
|
|
750
|
+
this.makeNewFile();
|
|
751
|
+
newlyCreatedFileIndex = filesBeforeCreate;
|
|
752
|
+
this.logger.silly?.(`[FileDatabase] Creating new file for unique custom metadata combination, fileNumber: ${this.currentFileNumber}`);
|
|
753
|
+
} else if (this.metadata.files.length === 0) {
|
|
754
|
+
this.makeNewFile();
|
|
755
|
+
}
|
|
756
|
+
const lastFile = this.metadata.files[this.metadata.files.length - 1];
|
|
757
|
+
const lastFileRecordsCount = lastFile.recordsCount;
|
|
758
|
+
if (forceNewFile && newlyCreatedFileIndex !== null) {
|
|
759
|
+
const newlyCreatedFile = this.metadata.files[newlyCreatedFileIndex];
|
|
760
|
+
if (newlyCreatedFile && newlyCreatedFile.fileName !== lastFile.fileName) {
|
|
761
|
+
this.logger.warn?.(`[FileDatabase] Warning: Newly created file ${newlyCreatedFile.fileName} doesn't match last file ${lastFile.fileName}`);
|
|
762
|
+
}
|
|
763
|
+
}
|
|
764
|
+
if (!Array.isArray(data) && !forceNewFile) {
|
|
765
|
+
const lastFileExtension = import_path3.default.extname(lastFile.fileName);
|
|
766
|
+
const expectedExtension = `.${getFileExtension(incomingDataType)}`;
|
|
767
|
+
if (lastFileExtension !== expectedExtension) {
|
|
768
|
+
if (lastFileRecordsCount > 0) {
|
|
769
|
+
this.makeNewFile();
|
|
770
|
+
} else {
|
|
771
|
+
lastFile.fileName = `${this.currentFileNumber.toString().padStart(6, "0")}.${getFileExtension(incomingDataType)}`;
|
|
772
|
+
}
|
|
773
|
+
}
|
|
774
|
+
} else if (!Array.isArray(data) && forceNewFile) {
|
|
775
|
+
const lastFileExtension = import_path3.default.extname(lastFile.fileName);
|
|
776
|
+
const expectedExtension = `.${getFileExtension(incomingDataType)}`;
|
|
777
|
+
if (lastFileExtension !== expectedExtension) {
|
|
778
|
+
lastFile.fileName = `${this.currentFileNumber.toString().padStart(6, "0")}.${getFileExtension(incomingDataType)}`;
|
|
779
|
+
}
|
|
780
|
+
}
|
|
781
|
+
if (Array.isArray(data)) {
|
|
782
|
+
if (forceNewFile) {
|
|
783
|
+
dataToWrite = data.slice(0, this.pageSize);
|
|
784
|
+
dataLeftOver = data.slice(this.pageSize);
|
|
785
|
+
this.lastFileData = dataToWrite;
|
|
786
|
+
} else if (lastFileRecordsCount < this.pageSize) {
|
|
787
|
+
dataToWrite = [...this.lastFileData || [], ...data.slice(0, this.pageSize - lastFileRecordsCount)];
|
|
788
|
+
dataLeftOver = data.slice(this.pageSize - lastFileRecordsCount);
|
|
789
|
+
this.lastFileData = dataToWrite;
|
|
790
|
+
} else {
|
|
791
|
+
this.makeNewFile();
|
|
792
|
+
dataToWrite = data.slice(0, this.pageSize);
|
|
793
|
+
dataLeftOver = data.slice(this.pageSize);
|
|
794
|
+
this.lastFileData = dataToWrite;
|
|
795
|
+
}
|
|
796
|
+
} else {
|
|
797
|
+
dataToWrite = data;
|
|
798
|
+
dataLeftOver = null;
|
|
799
|
+
}
|
|
800
|
+
const fileName = this.metadata.files[this.metadata.files.length - 1].fileName;
|
|
801
|
+
this.logger.silly?.(
|
|
802
|
+
`[FileDatabase] figureOutDataAndFileToWrite: filename=${fileName}, forceNewFile=${forceNewFile}, targetFileIndex=${targetFileIndex}, dataToWrite.length=${Array.isArray(dataToWrite) ? dataToWrite.length : "N/A"}, lastFileRecordsCount=${lastFileRecordsCount}`
|
|
803
|
+
);
|
|
804
|
+
return { dataToWrite, dataLeftOver, fileName };
|
|
805
|
+
}
|
|
806
|
+
/**
|
|
807
|
+
* Calculate file-level synopsis if function is set
|
|
808
|
+
*/
|
|
809
|
+
calculateFileSynopsis(data, fileIndex = this.metadata.files.length - 1) {
|
|
810
|
+
if (!this.fileSynopsisFunction) {
|
|
811
|
+
return;
|
|
812
|
+
}
|
|
813
|
+
const fileInfo = this.metadata.files[fileIndex];
|
|
814
|
+
const enhancedFileInfo = this.fileSynopsisFunction(fileInfo, data);
|
|
815
|
+
this.metadata.files[fileIndex] = enhancedFileInfo;
|
|
816
|
+
}
|
|
817
|
+
/**
|
|
818
|
+
* Calculate version-level synopsis if function is set
|
|
819
|
+
*/
|
|
820
|
+
calculateVersionSynopsis() {
|
|
821
|
+
if (!this.versionSynopsisFunction) {
|
|
822
|
+
return;
|
|
823
|
+
}
|
|
824
|
+
const enhancedMetadata = this.versionSynopsisFunction(this.metadata);
|
|
825
|
+
this.metadata = enhancedMetadata;
|
|
826
|
+
}
|
|
827
|
+
/**
|
|
828
|
+
* Update metadata after writing data
|
|
829
|
+
*/
|
|
830
|
+
updateMetadata(dataToWrite, fileName, customMetadata) {
|
|
831
|
+
let currentFile;
|
|
832
|
+
if (fileName) {
|
|
833
|
+
const foundFile = this.metadata.files.find((file) => file.fileName === fileName);
|
|
834
|
+
if (!foundFile) {
|
|
835
|
+
this.logger.warn?.(`[FileDatabase] File ${fileName} not found in metadata, using last file`);
|
|
836
|
+
currentFile = this.metadata.files[this.metadata.files.length - 1];
|
|
837
|
+
} else {
|
|
838
|
+
currentFile = foundFile;
|
|
839
|
+
}
|
|
840
|
+
} else {
|
|
841
|
+
currentFile = this.metadata.files[this.metadata.files.length - 1];
|
|
842
|
+
}
|
|
843
|
+
const recordsCount = Array.isArray(dataToWrite) ? dataToWrite.length : 1;
|
|
844
|
+
currentFile.recordsCount = recordsCount;
|
|
845
|
+
if (customMetadata) {
|
|
846
|
+
Object.assign(currentFile, customMetadata);
|
|
847
|
+
}
|
|
848
|
+
const fileIndex = this.metadata.files.indexOf(currentFile);
|
|
849
|
+
if (fileIndex !== -1) {
|
|
850
|
+
this.calculateFileSynopsis(dataToWrite, fileIndex);
|
|
851
|
+
}
|
|
852
|
+
this.metadata.version = this.currentVersion;
|
|
853
|
+
this.metadata.modifiedAt = (/* @__PURE__ */ new Date()).toISOString();
|
|
854
|
+
this.metadata.dataType = detectDataType(dataToWrite);
|
|
855
|
+
this.metadata.totalRecords = this.metadata.files.reduce((sum, file) => sum + (file.recordsCount || 0), 0);
|
|
856
|
+
this.logger.silly?.(
|
|
857
|
+
`[FileDatabase] Updated metadata for file ${currentFile.fileName}: recordsCount=${recordsCount}, totalRecords=${this.metadata.totalRecords}`
|
|
858
|
+
);
|
|
859
|
+
}
|
|
860
|
+
/**
|
|
861
|
+
* Safe write with disk space check
|
|
862
|
+
*/
|
|
863
|
+
async safeWrite(filePath, data) {
|
|
864
|
+
const serializedData = serializeData(data);
|
|
865
|
+
const dir = import_path3.default.dirname(filePath);
|
|
866
|
+
const requiredBytes = Buffer.byteLength(serializedData, "utf8");
|
|
867
|
+
const freeBytes = getFreeDiskSpace(dir);
|
|
868
|
+
if (freeBytes !== null) {
|
|
869
|
+
if (freeBytes < requiredBytes) {
|
|
870
|
+
throw new FileDatabaseError(
|
|
871
|
+
`Not enough disk space. Required: ${bytesToHumanReadable(requiredBytes)}, Free: ${bytesToHumanReadable(freeBytes)}`
|
|
872
|
+
);
|
|
873
|
+
}
|
|
874
|
+
if (freeBytes < this.freeSpaceThreshold) {
|
|
875
|
+
this.logger.warn?.(`Low disk space warning: only ${bytesToHumanReadable(freeBytes)} left`);
|
|
876
|
+
}
|
|
877
|
+
}
|
|
878
|
+
try {
|
|
879
|
+
await import_fs3.default.promises.writeFile(filePath, serializedData, "utf8");
|
|
880
|
+
this.logger.silly?.(`[FileDatabase] Wrote ${bytesToHumanReadable(requiredBytes)} to ${filePath}`);
|
|
881
|
+
} catch (error) {
|
|
882
|
+
throw new FileDatabaseError(`Failed to write file ${filePath}: ${error.message}`);
|
|
883
|
+
}
|
|
884
|
+
}
|
|
885
|
+
/**
|
|
886
|
+
* Prepare the instance for read or write operations
|
|
887
|
+
* This discovers state and sets up internal members based on mode and current data
|
|
888
|
+
*/
|
|
889
|
+
async prepare({ write, read, version }) {
|
|
890
|
+
if (write) {
|
|
891
|
+
if (this.versioned) {
|
|
892
|
+
if (this.currentVersion === null) {
|
|
893
|
+
await this.makeNewVersion();
|
|
894
|
+
this.metadata = this.getDefaultMetadata();
|
|
895
|
+
this.metadata.version = this.currentVersion;
|
|
896
|
+
this.makeNewFile();
|
|
897
|
+
} else {
|
|
898
|
+
if (!this.metadata.files.length) {
|
|
899
|
+
this.metadata = await this.figureMetadata(this.currentVersion);
|
|
900
|
+
if (this.metadata.files && this.metadata.files.length > 0) {
|
|
901
|
+
this.currentFileNumber = Math.max(...this.metadata.files.map((f) => f.number || 0));
|
|
902
|
+
} else {
|
|
903
|
+
this.currentFileNumber = 0;
|
|
904
|
+
}
|
|
905
|
+
}
|
|
906
|
+
}
|
|
907
|
+
} else {
|
|
908
|
+
await ensurePath(this.getDestinationPath());
|
|
909
|
+
if (this.useMetadata === true) {
|
|
910
|
+
const metadataPath = import_path3.default.join(this.getDestinationPath(), "metadata.json");
|
|
911
|
+
if (import_fs3.default.existsSync(metadataPath)) {
|
|
912
|
+
try {
|
|
913
|
+
const rawData = await import_fs3.default.promises.readFile(metadataPath, "utf8");
|
|
914
|
+
this.metadata = JSON.parse(rawData);
|
|
915
|
+
if (this.metadata.files && this.metadata.files.length > 0) {
|
|
916
|
+
this.currentFileNumber = Math.max(...this.metadata.files.map((f) => f.number || 0));
|
|
917
|
+
} else {
|
|
918
|
+
this.currentFileNumber = 0;
|
|
919
|
+
}
|
|
920
|
+
} catch (e) {
|
|
921
|
+
this.metadata = this.getDefaultMetadata();
|
|
922
|
+
this.currentFileNumber = 0;
|
|
923
|
+
}
|
|
924
|
+
} else {
|
|
925
|
+
this.metadata = this.getDefaultMetadata();
|
|
926
|
+
this.currentFileNumber = 0;
|
|
927
|
+
}
|
|
928
|
+
} else {
|
|
929
|
+
this.metadata = this.getDefaultMetadata();
|
|
930
|
+
this.currentFileNumber = 0;
|
|
931
|
+
}
|
|
932
|
+
}
|
|
933
|
+
} else if (read) {
|
|
934
|
+
if (this.versioned) {
|
|
935
|
+
const versions = await this.getVersions();
|
|
936
|
+
if (versions.length === 0) {
|
|
937
|
+
throw new FileDatabaseError("[FileDatabase] No versions found, cannot read");
|
|
938
|
+
}
|
|
939
|
+
if (version) {
|
|
940
|
+
if (!versions.includes(version)) {
|
|
941
|
+
throw new FileDatabaseError(`[FileDatabase] Version "${version}" not found`);
|
|
942
|
+
}
|
|
943
|
+
await this.setCurrentVersion(version);
|
|
944
|
+
} else {
|
|
945
|
+
await this.setCurrentVersion(versions[versions.length - 1]);
|
|
946
|
+
}
|
|
947
|
+
if (!this.metadata.files.length) {
|
|
948
|
+
this.metadata = await this.figureMetadata(this.currentVersion);
|
|
949
|
+
if (this.metadata.files && this.metadata.files.length > 0) {
|
|
950
|
+
this.currentFileNumber = Math.max(...this.metadata.files.map((f) => f.number || 0));
|
|
951
|
+
} else {
|
|
952
|
+
this.currentFileNumber = 0;
|
|
953
|
+
}
|
|
954
|
+
}
|
|
955
|
+
} else {
|
|
956
|
+
this.currentVersion = null;
|
|
957
|
+
if (this.useMetadata === void 0) {
|
|
958
|
+
const format = await this.detectDataFormat();
|
|
959
|
+
this.useMetadata = format.hasMetadata;
|
|
960
|
+
}
|
|
961
|
+
if (this.useMetadata) {
|
|
962
|
+
const destPath = this.getDestinationPath();
|
|
963
|
+
const metadataPath = import_path3.default.join(destPath, "metadata.json");
|
|
964
|
+
if (import_fs3.default.existsSync(metadataPath)) {
|
|
965
|
+
try {
|
|
966
|
+
const rawData = await import_fs3.default.promises.readFile(metadataPath, "utf8");
|
|
967
|
+
this.metadata = JSON.parse(rawData);
|
|
968
|
+
if (this.metadata.files && this.metadata.files.length > 0) {
|
|
969
|
+
this.currentFileNumber = Math.max(...this.metadata.files.map((f) => f.number || 0));
|
|
970
|
+
} else {
|
|
971
|
+
this.currentFileNumber = 0;
|
|
972
|
+
}
|
|
973
|
+
} catch (e) {
|
|
974
|
+
throw new FileDatabaseError(`Failed to read metadata: ${e.message}`);
|
|
975
|
+
}
|
|
976
|
+
} else {
|
|
977
|
+
throw new FileDatabaseError(
|
|
978
|
+
`[FileDatabase] No metadata found in non-versioned mode. Looked for: ${metadataPath} (table path: ${destPath})`
|
|
979
|
+
);
|
|
980
|
+
}
|
|
981
|
+
} else {
|
|
982
|
+
this.metadata = await this.figureMetadataFromVersionFiles("");
|
|
983
|
+
if (this.metadata.files && this.metadata.files.length > 0) {
|
|
984
|
+
this.currentFileNumber = Math.max(...this.metadata.files.map((f) => f.number || 0));
|
|
985
|
+
} else {
|
|
986
|
+
this.currentFileNumber = 0;
|
|
987
|
+
}
|
|
988
|
+
}
|
|
989
|
+
}
|
|
990
|
+
}
|
|
991
|
+
}
|
|
992
|
+
/**
|
|
993
|
+
* Write data to the file database
|
|
994
|
+
*/
|
|
995
|
+
async write(data, options = {}) {
|
|
996
|
+
if (options.filename) {
|
|
997
|
+
const destPath2 = this.getDestinationPath();
|
|
998
|
+
await ensurePath(destPath2);
|
|
999
|
+
const filePath = import_path3.default.join(destPath2, options.filename);
|
|
1000
|
+
await this.safeWrite(filePath, data);
|
|
1001
|
+
return;
|
|
1002
|
+
}
|
|
1003
|
+
if (options.forceNewVersion && !this.versioned) {
|
|
1004
|
+
throw new FileDatabaseError("Cannot use forceNewVersion in non-versioned mode");
|
|
1005
|
+
}
|
|
1006
|
+
await this.prepare({ write: true });
|
|
1007
|
+
const incomingDataType = detectDataType(data);
|
|
1008
|
+
this.metadata.dataType = incomingDataType;
|
|
1009
|
+
if (options.forceNewVersion) {
|
|
1010
|
+
await this.makeNewVersion();
|
|
1011
|
+
this.metadata = this.getDefaultMetadata();
|
|
1012
|
+
this.metadata.version = this.currentVersion;
|
|
1013
|
+
this.metadata.dataType = incomingDataType;
|
|
1014
|
+
this.makeNewFile();
|
|
1015
|
+
}
|
|
1016
|
+
let targetFileIndex = null;
|
|
1017
|
+
const hasCustomMetadata = options.customMetadata && Object.keys(options.customMetadata).length > 0;
|
|
1018
|
+
if (hasCustomMetadata) {
|
|
1019
|
+
for (let i = 0; i < this.metadata.files.length; i++) {
|
|
1020
|
+
const fileEntry = this.metadata.files[i];
|
|
1021
|
+
const matches = Object.keys(options.customMetadata).every((key) => {
|
|
1022
|
+
return key in fileEntry && fileEntry[key] === options.customMetadata[key];
|
|
1023
|
+
});
|
|
1024
|
+
if (matches) {
|
|
1025
|
+
targetFileIndex = i;
|
|
1026
|
+
this.logger.silly?.(`[FileDatabase] Found existing file with matching custom metadata: ${fileEntry.fileName}, metadata: ${JSON.stringify(options.customMetadata)}`);
|
|
1027
|
+
break;
|
|
1028
|
+
} else {
|
|
1029
|
+
this.logger.silly?.(`[FileDatabase] File ${fileEntry.fileName} does not match custom metadata: ${JSON.stringify(options.customMetadata)}`);
|
|
1030
|
+
}
|
|
1031
|
+
}
|
|
1032
|
+
if (targetFileIndex === null) {
|
|
1033
|
+
this.logger.silly?.(`[FileDatabase] No existing file found with custom metadata: ${JSON.stringify(options.customMetadata)}, will create new file`);
|
|
1034
|
+
}
|
|
1035
|
+
} else {
|
|
1036
|
+
this.logger.silly?.(`[FileDatabase] No custom metadata provided, will create new file`);
|
|
1037
|
+
}
|
|
1038
|
+
if (targetFileIndex !== null) {
|
|
1039
|
+
const targetFile = this.metadata.files[targetFileIndex];
|
|
1040
|
+
this.currentFileNumber = targetFile.number;
|
|
1041
|
+
this.lastFileData = null;
|
|
1042
|
+
this.currentRecord = 0;
|
|
1043
|
+
this.hasReadFirstPage = false;
|
|
1044
|
+
}
|
|
1045
|
+
const forceNewFile = hasCustomMetadata && targetFileIndex === null;
|
|
1046
|
+
let { dataToWrite, dataLeftOver, fileName } = this.figureOutDataAndFileToWrite(data, targetFileIndex, forceNewFile);
|
|
1047
|
+
const destPath = this.getDestinationPath(this.currentVersion || void 0);
|
|
1048
|
+
await this.safeWrite(import_path3.default.join(destPath, fileName), dataToWrite);
|
|
1049
|
+
this.updateMetadata(dataToWrite, fileName, options.customMetadata);
|
|
1050
|
+
while (dataLeftOver && dataLeftOver.length > 0 && targetFileIndex === null) {
|
|
1051
|
+
const writeContext = this.figureOutDataAndFileToWrite(dataLeftOver);
|
|
1052
|
+
await this.safeWrite(import_path3.default.join(destPath, writeContext.fileName), writeContext.dataToWrite);
|
|
1053
|
+
this.updateMetadata(writeContext.dataToWrite, writeContext.fileName, options.customMetadata);
|
|
1054
|
+
dataLeftOver = writeContext.dataLeftOver;
|
|
1055
|
+
}
|
|
1056
|
+
this.calculateVersionSynopsis();
|
|
1057
|
+
if (this.useMetadata) {
|
|
1058
|
+
await this.saveVersionMetadata(this.metadata);
|
|
1059
|
+
}
|
|
1060
|
+
}
|
|
1061
|
+
/**
|
|
1062
|
+
* Read data from the file database
|
|
1063
|
+
*/
|
|
1064
|
+
async read(options = {}) {
|
|
1065
|
+
const { version, nextPage = false, pageSize, filename } = options;
|
|
1066
|
+
if (filename) {
|
|
1067
|
+
const destPath = this.getDestinationPath(version);
|
|
1068
|
+
const filePath = import_path3.default.join(destPath, filename);
|
|
1069
|
+
try {
|
|
1070
|
+
const rawData = await import_fs3.default.promises.readFile(filePath, "utf8");
|
|
1071
|
+
return JSON.parse(rawData);
|
|
1072
|
+
} catch (error) {
|
|
1073
|
+
throw new FileDatabaseError(`Failed to read file ${filename}: ${error.message}`);
|
|
1074
|
+
}
|
|
1075
|
+
}
|
|
1076
|
+
await this.prepare({ read: true, version });
|
|
1077
|
+
const isNonPaginatedData = this.metadata.dataType === "text" || this.metadata.dataType === "xml" || this.metadata.dataType === "json-object";
|
|
1078
|
+
if (isNonPaginatedData) {
|
|
1079
|
+
const file = this.metadata.files[0];
|
|
1080
|
+
const filePath = import_path3.default.join(this.getDestinationPath(this.currentVersion || void 0), file.fileName);
|
|
1081
|
+
try {
|
|
1082
|
+
const rawData = await import_fs3.default.promises.readFile(filePath, "utf8");
|
|
1083
|
+
return deserializeData(rawData, this.metadata.dataType);
|
|
1084
|
+
} catch (error) {
|
|
1085
|
+
throw new FileDatabaseError(`Failed to read file ${file.fileName}: ${error.message}`);
|
|
1086
|
+
}
|
|
1087
|
+
}
|
|
1088
|
+
let effectivePageSize;
|
|
1089
|
+
if (nextPage && this.hasReadFirstPage) {
|
|
1090
|
+
effectivePageSize = pageSize || this.pageSize;
|
|
1091
|
+
this.currentRecord += effectivePageSize;
|
|
1092
|
+
} else if (!nextPage) {
|
|
1093
|
+
effectivePageSize = pageSize !== void 0 ? pageSize : this.metadata.totalRecords;
|
|
1094
|
+
this.currentRecord = 0;
|
|
1095
|
+
} else {
|
|
1096
|
+
effectivePageSize = pageSize || this.pageSize;
|
|
1097
|
+
}
|
|
1098
|
+
if (this.currentRecord >= this.metadata.totalRecords) {
|
|
1099
|
+
return [];
|
|
1100
|
+
}
|
|
1101
|
+
const result = [];
|
|
1102
|
+
let recordsRead = 0;
|
|
1103
|
+
let currentFileIndex = 0;
|
|
1104
|
+
let currentFileOffset = 0;
|
|
1105
|
+
let totalRecords = 0;
|
|
1106
|
+
for (let i = 0; i < this.metadata.files.length; i++) {
|
|
1107
|
+
const file = this.metadata.files[i];
|
|
1108
|
+
if (this.currentRecord < totalRecords + file.recordsCount) {
|
|
1109
|
+
currentFileIndex = i;
|
|
1110
|
+
currentFileOffset = totalRecords;
|
|
1111
|
+
break;
|
|
1112
|
+
}
|
|
1113
|
+
totalRecords += file.recordsCount;
|
|
1114
|
+
}
|
|
1115
|
+
let cumulativeRecords = currentFileOffset;
|
|
1116
|
+
for (let i = currentFileIndex; i < this.metadata.files.length && recordsRead < effectivePageSize; i++) {
|
|
1117
|
+
const file = this.metadata.files[i];
|
|
1118
|
+
const filePath = import_path3.default.join(this.getDestinationPath(this.currentVersion || void 0), file.fileName);
|
|
1119
|
+
try {
|
|
1120
|
+
const rawData = await import_fs3.default.promises.readFile(filePath, "utf8");
|
|
1121
|
+
const fileData = deserializeData(rawData, this.metadata.dataType);
|
|
1122
|
+
let startIndex = 0;
|
|
1123
|
+
if (i === currentFileIndex) {
|
|
1124
|
+
startIndex = this.currentRecord - cumulativeRecords;
|
|
1125
|
+
}
|
|
1126
|
+
const endIndex = Math.min(startIndex + (effectivePageSize - recordsRead), fileData.length);
|
|
1127
|
+
const recordsFromThisFile = fileData.slice(startIndex, endIndex);
|
|
1128
|
+
result.push(...recordsFromThisFile);
|
|
1129
|
+
recordsRead += recordsFromThisFile.length;
|
|
1130
|
+
cumulativeRecords += file.recordsCount;
|
|
1131
|
+
} catch (error) {
|
|
1132
|
+
throw new FileDatabaseError(`Failed to read file ${file.fileName}: ${error.message}`);
|
|
1133
|
+
}
|
|
1134
|
+
}
|
|
1135
|
+
if (result.length > 0) {
|
|
1136
|
+
if (nextPage || pageSize !== void 0 && pageSize < this.metadata.totalRecords) {
|
|
1137
|
+
this.hasReadFirstPage = true;
|
|
1138
|
+
}
|
|
1139
|
+
}
|
|
1140
|
+
return result;
|
|
1141
|
+
}
|
|
1142
|
+
/**
|
|
1143
|
+
* Set the starting record for pagination (1-based index)
|
|
1144
|
+
*/
|
|
1145
|
+
setStartRecord(startRecord) {
|
|
1146
|
+
this.currentRecord = startRecord - 1;
|
|
1147
|
+
this.hasReadFirstPage = false;
|
|
1148
|
+
}
|
|
1149
|
+
/**
|
|
1150
|
+
* Reset read pagination state
|
|
1151
|
+
*/
|
|
1152
|
+
resetPagination() {
|
|
1153
|
+
this.currentRecord = 0;
|
|
1154
|
+
this.hasReadFirstPage = false;
|
|
1155
|
+
}
|
|
1156
|
+
/**
|
|
1157
|
+
* List filenames in the table directory.
|
|
1158
|
+
* For catalog/key-value usage (files written with { filename }).
|
|
1159
|
+
* Returns data file names (.json, .txt, .xml) excluding metadata.json.
|
|
1160
|
+
*/
|
|
1161
|
+
async listFilenames() {
|
|
1162
|
+
const destPath = this.versioned && this.currentVersion ? import_path3.default.join(this.getDestinationPath(), this.currentVersion) : this.getDestinationPath();
|
|
1163
|
+
try {
|
|
1164
|
+
const entries = await import_fs3.default.promises.readdir(destPath, { withFileTypes: true });
|
|
1165
|
+
return entries.filter((e) => e.isFile() && e.name !== "metadata.json" && /\.(json|txt|xml)$/i.test(e.name)).map((e) => e.name);
|
|
1166
|
+
} catch (err) {
|
|
1167
|
+
if (err?.code === "ENOENT") return [];
|
|
1168
|
+
throw new FileDatabaseError(`Failed to list files: ${err.message}`);
|
|
1169
|
+
}
|
|
1170
|
+
}
|
|
1171
|
+
/**
|
|
1172
|
+
* Remove a file from the table directory (catalog mode).
|
|
1173
|
+
* Use with listFilenames() to manage individual files.
|
|
1174
|
+
*/
|
|
1175
|
+
async removeFile(filename) {
|
|
1176
|
+
const destPath = this.versioned && this.currentVersion ? import_path3.default.join(this.getDestinationPath(), this.currentVersion) : this.getDestinationPath();
|
|
1177
|
+
const filePath = import_path3.default.join(destPath, filename);
|
|
1178
|
+
try {
|
|
1179
|
+
await import_fs3.default.promises.unlink(filePath);
|
|
1180
|
+
} catch (err) {
|
|
1181
|
+
if (err?.code === "ENOENT") return;
|
|
1182
|
+
throw new FileDatabaseError(`Failed to remove file ${filename}: ${err.message}`);
|
|
1183
|
+
}
|
|
1184
|
+
}
|
|
1185
|
+
/**
|
|
1186
|
+
* Remove a file and its metadata entry (non-versioned mode with useMetadata).
|
|
1187
|
+
* Use with findData() to get fileName, then call removeFileEntry to delete.
|
|
1188
|
+
*/
|
|
1189
|
+
async removeFileEntry(filename) {
|
|
1190
|
+
if (this.versioned) {
|
|
1191
|
+
throw new FileDatabaseError("removeFileEntry is only supported in non-versioned mode");
|
|
1192
|
+
}
|
|
1193
|
+
await this.prepare({ read: true });
|
|
1194
|
+
const idx = this.metadata.files.findIndex((f) => f.fileName === filename);
|
|
1195
|
+
if (idx === -1) {
|
|
1196
|
+
throw new FileDatabaseError(`File entry ${filename} not found in metadata`);
|
|
1197
|
+
}
|
|
1198
|
+
const entry = this.metadata.files[idx];
|
|
1199
|
+
const recordsCount = entry.recordsCount || 0;
|
|
1200
|
+
this.metadata.files.splice(idx, 1);
|
|
1201
|
+
this.metadata.totalRecords = Math.max(0, (this.metadata.totalRecords || 0) - recordsCount);
|
|
1202
|
+
const destPath = this.getDestinationPath();
|
|
1203
|
+
const filePath = import_path3.default.join(destPath, filename);
|
|
1204
|
+
try {
|
|
1205
|
+
await import_fs3.default.promises.unlink(filePath);
|
|
1206
|
+
} catch (err) {
|
|
1207
|
+
if (err?.code === "ENOENT") {
|
|
1208
|
+
this.logger.warn?.(`[FileDatabase] File ${filename} already missing on disk`);
|
|
1209
|
+
} else {
|
|
1210
|
+
throw new FileDatabaseError(`Failed to remove file ${filename}: ${err.message}`);
|
|
1211
|
+
}
|
|
1212
|
+
}
|
|
1213
|
+
if (this.useMetadata) {
|
|
1214
|
+
await this.saveVersionMetadata(this.metadata);
|
|
1215
|
+
}
|
|
1216
|
+
}
|
|
1217
|
+
/**
|
|
1218
|
+
* Set file-level synopsis calculation function
|
|
1219
|
+
*/
|
|
1220
|
+
setFileSynopsisFunction(fn) {
|
|
1221
|
+
this.fileSynopsisFunction = fn;
|
|
1222
|
+
}
|
|
1223
|
+
/**
|
|
1224
|
+
* Set version-level synopsis calculation function
|
|
1225
|
+
*/
|
|
1226
|
+
setVersionSynopsisFunction(fn) {
|
|
1227
|
+
this.versionSynopsisFunction = fn;
|
|
1228
|
+
}
|
|
1229
|
+
/**
|
|
1230
|
+
* Get current version name
|
|
1231
|
+
*/
|
|
1232
|
+
getCurrentVersion() {
|
|
1233
|
+
return this.currentVersion;
|
|
1234
|
+
}
|
|
1235
|
+
/**
|
|
1236
|
+
* Get current metadata
|
|
1237
|
+
*/
|
|
1238
|
+
getMetadata() {
|
|
1239
|
+
return { ...this.metadata };
|
|
1240
|
+
}
|
|
1241
|
+
/**
|
|
1242
|
+
* Find data by custom metadata fields
|
|
1243
|
+
* Searches through all versions and files to find entries matching the search criteria
|
|
1244
|
+
*
|
|
1245
|
+
* @param searchCriteria - Object with field names and values to search for (e.g., { ListingKey: "123", id: "456" })
|
|
1246
|
+
* @returns Array of found entries with their file paths and metadata
|
|
1247
|
+
*/
|
|
1248
|
+
async findData(searchCriteria) {
|
|
1249
|
+
const results = [];
|
|
1250
|
+
if (!this.versioned) {
|
|
1251
|
+
await this.prepare({ read: true });
|
|
1252
|
+
const metadata = this.getMetadata();
|
|
1253
|
+
for (const fileEntry of metadata.files) {
|
|
1254
|
+
const matches = Object.keys(searchCriteria).every((key) => {
|
|
1255
|
+
return fileEntry[key] === searchCriteria[key];
|
|
1256
|
+
});
|
|
1257
|
+
if (matches) {
|
|
1258
|
+
const destPath = this.getDestinationPath();
|
|
1259
|
+
const filePath = import_path3.default.join(destPath, fileEntry.fileName);
|
|
1260
|
+
const fileData = await import_fs3.default.promises.readFile(filePath, "utf8");
|
|
1261
|
+
const data = deserializeData(fileData, metadata.dataType || "json-object");
|
|
1262
|
+
results.push({
|
|
1263
|
+
filePath,
|
|
1264
|
+
fileName: fileEntry.fileName,
|
|
1265
|
+
version: null,
|
|
1266
|
+
metadata: fileEntry,
|
|
1267
|
+
data
|
|
1268
|
+
});
|
|
1269
|
+
}
|
|
1270
|
+
}
|
|
1271
|
+
} else {
|
|
1272
|
+
const versions = await this.getVersions();
|
|
1273
|
+
for (const version of versions) {
|
|
1274
|
+
await this.prepare({ read: true, version });
|
|
1275
|
+
const metadata = this.getMetadata();
|
|
1276
|
+
for (const fileEntry of metadata.files) {
|
|
1277
|
+
const matches = Object.keys(searchCriteria).every((key) => {
|
|
1278
|
+
return fileEntry[key] === searchCriteria[key];
|
|
1279
|
+
});
|
|
1280
|
+
if (matches) {
|
|
1281
|
+
const destPath = this.getDestinationPath(version);
|
|
1282
|
+
const filePath = import_path3.default.join(destPath, fileEntry.fileName);
|
|
1283
|
+
const fileData = await import_fs3.default.promises.readFile(filePath, "utf8");
|
|
1284
|
+
const data = deserializeData(fileData, metadata.dataType || "json-object");
|
|
1285
|
+
results.push({
|
|
1286
|
+
filePath,
|
|
1287
|
+
fileName: fileEntry.fileName,
|
|
1288
|
+
version,
|
|
1289
|
+
metadata: fileEntry,
|
|
1290
|
+
data
|
|
1291
|
+
});
|
|
1292
|
+
}
|
|
1293
|
+
}
|
|
1294
|
+
}
|
|
1295
|
+
}
|
|
1296
|
+
return results;
|
|
1297
|
+
}
|
|
1298
|
+
};
|
|
1299
|
+
|
|
1300
|
+
// src/tasks/taskLogs.ts
|
|
1301
|
+
function getLogsState(context) {
|
|
1302
|
+
const holder = context;
|
|
1303
|
+
if (holder.__tasksLogsState) return holder.__tasksLogsState;
|
|
1304
|
+
const basePath = holder.params?.get?.("tasksLogsBasePath") || "./data";
|
|
1305
|
+
const namespace = holder.params?.get?.("tasksLogsNamespace") || "tasks-logs";
|
|
1306
|
+
const tableName = holder.params?.get?.("tasksLogsTable") || "runner";
|
|
1307
|
+
const errorTableName = holder.params?.get?.("tasksErrorLogsTable") || `${tableName}-errors`;
|
|
1308
|
+
const maxVersionsRaw = Number(holder.params?.get?.("tasksLogsMaxVersions"));
|
|
1309
|
+
const pageSizeRaw = Number(holder.params?.get?.("tasksLogsPageSize"));
|
|
1310
|
+
const errorDb = new FileDatabase({
|
|
1311
|
+
basePath,
|
|
1312
|
+
namespace,
|
|
1313
|
+
tableName: errorTableName,
|
|
1314
|
+
versioned: true,
|
|
1315
|
+
useMetadata: true,
|
|
1316
|
+
maxVersions: Number.isFinite(maxVersionsRaw) && maxVersionsRaw > 0 ? maxVersionsRaw : 20,
|
|
1317
|
+
pageSize: Number.isFinite(pageSizeRaw) && pageSizeRaw > 0 ? pageSizeRaw : 2e3,
|
|
1318
|
+
logger: holder.logger
|
|
1319
|
+
});
|
|
1320
|
+
const enabledRaw = holder.params?.get?.("tasksLogsEnabled");
|
|
1321
|
+
const enabled = enabledRaw === void 0 ? true : !!enabledRaw;
|
|
1322
|
+
if (!enabled) {
|
|
1323
|
+
const disabledState = {
|
|
1324
|
+
db: null,
|
|
1325
|
+
errorDb,
|
|
1326
|
+
queue: Promise.resolve(),
|
|
1327
|
+
initialized: true,
|
|
1328
|
+
errorInitialized: false
|
|
1329
|
+
};
|
|
1330
|
+
holder.__tasksLogsState = disabledState;
|
|
1331
|
+
return disabledState;
|
|
1332
|
+
}
|
|
1333
|
+
const db = new FileDatabase({
|
|
1334
|
+
basePath,
|
|
1335
|
+
namespace,
|
|
1336
|
+
tableName,
|
|
1337
|
+
versioned: true,
|
|
1338
|
+
useMetadata: true,
|
|
1339
|
+
maxVersions: Number.isFinite(maxVersionsRaw) && maxVersionsRaw > 0 ? maxVersionsRaw : 20,
|
|
1340
|
+
pageSize: Number.isFinite(pageSizeRaw) && pageSizeRaw > 0 ? pageSizeRaw : 2e3,
|
|
1341
|
+
logger: holder.logger
|
|
1342
|
+
});
|
|
1343
|
+
const state = {
|
|
1344
|
+
db,
|
|
1345
|
+
errorDb,
|
|
1346
|
+
queue: Promise.resolve(),
|
|
1347
|
+
initialized: false,
|
|
1348
|
+
errorInitialized: false
|
|
1349
|
+
};
|
|
1350
|
+
holder.__tasksLogsState = state;
|
|
1351
|
+
return state;
|
|
1352
|
+
}
|
|
1353
|
+
function isErrorPayload(payload) {
|
|
1354
|
+
if (!payload) return false;
|
|
1355
|
+
if (typeof payload === "object") {
|
|
1356
|
+
const level = typeof payload.level === "string" ? payload.level.toLowerCase() : "";
|
|
1357
|
+
if (level === "error" || level === "fatal") return true;
|
|
1358
|
+
if (typeof payload.message === "string" && /\berror\b/i.test(payload.message)) return true;
|
|
1359
|
+
return false;
|
|
1360
|
+
}
|
|
1361
|
+
if (typeof payload === "string") {
|
|
1362
|
+
return /\berror\b/i.test(payload);
|
|
1363
|
+
}
|
|
1364
|
+
return false;
|
|
1365
|
+
}
|
|
1366
|
+
function buildLogRecord(task, payload) {
|
|
1367
|
+
const params = task.params && typeof task.params === "object" ? task.params : {};
|
|
1368
|
+
return {
|
|
1369
|
+
ts: (/* @__PURE__ */ new Date()).toISOString(),
|
|
1370
|
+
opid: task.opid ?? null,
|
|
1371
|
+
taskId: task.id,
|
|
1372
|
+
taskName: task.task,
|
|
1373
|
+
target: task.target,
|
|
1374
|
+
source: typeof params.source === "string" ? params.source : null,
|
|
1375
|
+
resource: typeof params.resource === "string" ? params.resource : null,
|
|
1376
|
+
payload
|
|
1377
|
+
};
|
|
1378
|
+
}
|
|
1379
|
+
function appendTaskIpcLog(context, task, payload) {
|
|
1380
|
+
const state = getLogsState(context);
|
|
1381
|
+
if (!state.db && !state.errorDb) return;
|
|
1382
|
+
const record = buildLogRecord(task, payload);
|
|
1383
|
+
state.queue = state.queue.then(async () => {
|
|
1384
|
+
if (state.db) {
|
|
1385
|
+
await state.db.write([record], { forceNewVersion: !state.initialized });
|
|
1386
|
+
state.initialized = true;
|
|
1387
|
+
}
|
|
1388
|
+
if (state.errorDb && isErrorPayload(payload)) {
|
|
1389
|
+
await state.errorDb.write([record], { forceNewVersion: !state.errorInitialized });
|
|
1390
|
+
state.errorInitialized = true;
|
|
1391
|
+
}
|
|
1392
|
+
}).catch((error) => {
|
|
1393
|
+
context.logger.warn?.("[tasks] failed to persist IPC log entry:", error);
|
|
1394
|
+
});
|
|
1395
|
+
}
|
|
1396
|
+
|
|
1397
|
+
// src/tasks/time-matcher.ts
|
|
1398
|
+
var RANGES = ["0-59", "0-59", "0-23", "1-31", "1-12", "0-6"];
|
|
1399
|
+
function resolveAsterisks(field, range) {
|
|
1400
|
+
return field.includes("*") ? field.replace("*", range) : field;
|
|
1401
|
+
}
|
|
1402
|
+
function resolveRanges(field) {
|
|
1403
|
+
const regex = /(\d+)-(\d+)/;
|
|
1404
|
+
let current = field;
|
|
1405
|
+
while (true) {
|
|
1406
|
+
const match = regex.exec(current);
|
|
1407
|
+
if (!match) break;
|
|
1408
|
+
const raw = match[0];
|
|
1409
|
+
let first = Number(match[1]);
|
|
1410
|
+
let last = Number(match[2]);
|
|
1411
|
+
if (last < first) {
|
|
1412
|
+
[first, last] = [last, first];
|
|
1413
|
+
}
|
|
1414
|
+
const values = [];
|
|
1415
|
+
for (let i = first; i <= last; i += 1) {
|
|
1416
|
+
values.push(i);
|
|
1417
|
+
}
|
|
1418
|
+
current = current.replace(raw, values.join(","));
|
|
1419
|
+
}
|
|
1420
|
+
return current;
|
|
1421
|
+
}
|
|
1422
|
+
function resolveSteps(field) {
|
|
1423
|
+
const match = /^(.+)\/(\d+)$/.exec(field);
|
|
1424
|
+
if (!match) return field;
|
|
1425
|
+
const base = match[1];
|
|
1426
|
+
const step = Number(match[2]);
|
|
1427
|
+
if (!Number.isFinite(step) || step <= 0) return field;
|
|
1428
|
+
return base.split(",").map((v) => Number(v)).filter((v) => Number.isFinite(v) && v % step === 0).join(",");
|
|
1429
|
+
}
|
|
1430
|
+
function convertPattern(pattern) {
|
|
1431
|
+
const parts = pattern.trim().split(/\s+/);
|
|
1432
|
+
if (parts.length !== 6) {
|
|
1433
|
+
throw new Error(`Invalid schedule "${pattern}". Expected 6 fields: sec min hour day month weekday`);
|
|
1434
|
+
}
|
|
1435
|
+
return parts.map((field, idx) => resolveAsterisks(field, RANGES[idx])).map((field) => resolveRanges(field)).map((field) => resolveSteps(field));
|
|
1436
|
+
}
|
|
1437
|
+
function fieldMatches(field, value) {
|
|
1438
|
+
const allowed = field.split(",").map((v) => Number(v));
|
|
1439
|
+
return allowed.includes(value);
|
|
1440
|
+
}
|
|
1441
|
+
function timeMatcher(pattern, date = /* @__PURE__ */ new Date()) {
|
|
1442
|
+
const parsed = convertPattern(pattern);
|
|
1443
|
+
return fieldMatches(parsed[0], date.getSeconds()) && fieldMatches(parsed[1], date.getMinutes()) && fieldMatches(parsed[2], date.getHours()) && fieldMatches(parsed[3], date.getDate()) && fieldMatches(parsed[4], date.getMonth() + 1) && fieldMatches(parsed[5], date.getDay());
|
|
1444
|
+
}
|
|
1445
|
+
|
|
1446
|
+
// src/tasks/TaskMaster.ts
|
|
1447
|
+
var TaskMaster = class {
|
|
1448
|
+
context;
|
|
1449
|
+
task;
|
|
1450
|
+
constructor(context, task) {
|
|
1451
|
+
this.context = context;
|
|
1452
|
+
this.task = task;
|
|
1453
|
+
}
|
|
1454
|
+
cantRunReason() {
|
|
1455
|
+
return false;
|
|
1456
|
+
}
|
|
1457
|
+
requestStop(_allowanceMs) {
|
|
1458
|
+
}
|
|
1459
|
+
};
|
|
1460
|
+
|
|
1461
|
+
// src/tasks/coreTasks/TaskPing.ts
|
|
1462
|
+
var TaskPing = class extends TaskMaster {
|
|
1463
|
+
async run() {
|
|
1464
|
+
this.context.logger.info?.(`[TaskPing] pong (${this.task.id})`);
|
|
1465
|
+
return { success: true, results: "pong" };
|
|
1466
|
+
}
|
|
1467
|
+
};
|
|
1468
|
+
|
|
1469
|
+
// src/tasks/coreTasks/TaskSampleProcess.ts
|
|
1470
|
+
var TaskSampleProcess = class extends TaskMaster {
|
|
1471
|
+
stopRequested = false;
|
|
1472
|
+
stopAllowanceMs = 0;
|
|
1473
|
+
stopDecisionLogged = false;
|
|
1474
|
+
requestStop(allowanceMs) {
|
|
1475
|
+
this.stopRequested = true;
|
|
1476
|
+
this.stopAllowanceMs = Number.isFinite(allowanceMs) && allowanceMs > 0 ? allowanceMs : 0;
|
|
1477
|
+
this.context.logger.warn?.(
|
|
1478
|
+
`[TaskSampleProcess] stop signal received (${this.task.id}), allowanceMs=${this.stopAllowanceMs}`
|
|
1479
|
+
);
|
|
1480
|
+
}
|
|
1481
|
+
async run(reportProgress) {
|
|
1482
|
+
const totalRaw = this.task?.params?.total ?? 10;
|
|
1483
|
+
const delayRaw = this.task?.params?.delay ?? 1e3;
|
|
1484
|
+
const nameRaw = this.task?.params?.name;
|
|
1485
|
+
const total = Number(totalRaw);
|
|
1486
|
+
const delay = Number(delayRaw);
|
|
1487
|
+
const name = typeof nameRaw === "string" && nameRaw.trim() ? nameRaw.trim() : "sampleProcess";
|
|
1488
|
+
const errors = [];
|
|
1489
|
+
if (!Number.isInteger(total) || total <= 0) {
|
|
1490
|
+
errors.push('param "total" must be a positive integer');
|
|
1491
|
+
}
|
|
1492
|
+
if (!Number.isInteger(delay) || delay < 0) {
|
|
1493
|
+
errors.push('param "delay" must be an integer >= 0');
|
|
1494
|
+
}
|
|
1495
|
+
if (errors.length > 0) {
|
|
1496
|
+
return {
|
|
1497
|
+
success: false,
|
|
1498
|
+
results: {
|
|
1499
|
+
error: `Validation failed: ${errors.join(", ")}`,
|
|
1500
|
+
received: { total: totalRaw, delay: delayRaw, name: nameRaw }
|
|
1501
|
+
}
|
|
1502
|
+
};
|
|
1503
|
+
}
|
|
1504
|
+
const startedAt = Date.now();
|
|
1505
|
+
for (let i = 1; i <= total; i += 1) {
|
|
1506
|
+
if (this.stopRequested) {
|
|
1507
|
+
const remainingMs = Math.max(0, (total - i + 1) * delay);
|
|
1508
|
+
if (remainingMs <= this.stopAllowanceMs) {
|
|
1509
|
+
if (!this.stopDecisionLogged) {
|
|
1510
|
+
this.stopDecisionLogged = true;
|
|
1511
|
+
this.context.logger.warn?.(
|
|
1512
|
+
`[TaskSampleProcess] continue to finish (${this.task.id}): remainingMs=${remainingMs} <= allowanceMs=${this.stopAllowanceMs}`
|
|
1513
|
+
);
|
|
1514
|
+
}
|
|
1515
|
+
} else {
|
|
1516
|
+
this.context.logger.warn?.(
|
|
1517
|
+
`[TaskSampleProcess] stopping gracefully at iteration ${i}/${total} (${this.task.id}), remainingMs=${remainingMs} > allowanceMs=${this.stopAllowanceMs}`
|
|
1518
|
+
);
|
|
1519
|
+
return {
|
|
1520
|
+
success: false,
|
|
1521
|
+
results: {
|
|
1522
|
+
message: `Stopped before completion at iteration ${i}/${total}`,
|
|
1523
|
+
completed: i - 1,
|
|
1524
|
+
total,
|
|
1525
|
+
name,
|
|
1526
|
+
remainingMs,
|
|
1527
|
+
allowanceMs: this.stopAllowanceMs
|
|
1528
|
+
}
|
|
1529
|
+
};
|
|
1530
|
+
}
|
|
1531
|
+
}
|
|
1532
|
+
const elapsed = Date.now() - startedAt;
|
|
1533
|
+
const remaining = Math.max(0, (total - i) * delay);
|
|
1534
|
+
const progress = {
|
|
1535
|
+
name,
|
|
1536
|
+
count: i,
|
|
1537
|
+
total,
|
|
1538
|
+
elapsedMs: elapsed,
|
|
1539
|
+
remainingMs: remaining,
|
|
1540
|
+
status: `running ${name}: ${i}/${total}`
|
|
1541
|
+
};
|
|
1542
|
+
this.context.logger.progress("running", {
|
|
1543
|
+
prefix: name,
|
|
1544
|
+
count: i,
|
|
1545
|
+
total
|
|
1546
|
+
});
|
|
1547
|
+
await reportProgress(progress);
|
|
1548
|
+
await sleepMs(delay);
|
|
1549
|
+
}
|
|
1550
|
+
return {
|
|
1551
|
+
success: true,
|
|
1552
|
+
results: {
|
|
1553
|
+
message: `Completed ${total} iterations`,
|
|
1554
|
+
total,
|
|
1555
|
+
delay,
|
|
1556
|
+
name
|
|
1557
|
+
}
|
|
1558
|
+
};
|
|
1559
|
+
}
|
|
1560
|
+
};
|
|
1561
|
+
|
|
1562
|
+
// src/tasks/coreTasks/TaskShellCommand.ts
|
|
1563
|
+
var import_node_child_process = require("child_process");
|
|
1564
|
+
function runShellCommand(command, cwd) {
|
|
1565
|
+
return new Promise((resolve, reject) => {
|
|
1566
|
+
const child = (0, import_node_child_process.spawn)(command, {
|
|
1567
|
+
shell: true,
|
|
1568
|
+
cwd: cwd || process.cwd(),
|
|
1569
|
+
stdio: ["ignore", "pipe", "pipe"]
|
|
1570
|
+
});
|
|
1571
|
+
let output = "";
|
|
1572
|
+
let stderr = "";
|
|
1573
|
+
child.stdout.on("data", (chunk) => {
|
|
1574
|
+
output += String(chunk);
|
|
1575
|
+
});
|
|
1576
|
+
child.stderr.on("data", (chunk) => {
|
|
1577
|
+
stderr += String(chunk);
|
|
1578
|
+
});
|
|
1579
|
+
child.on("error", (error) => {
|
|
1580
|
+
reject(error);
|
|
1581
|
+
});
|
|
1582
|
+
child.on("close", (exitCode, signal) => {
|
|
1583
|
+
resolve({
|
|
1584
|
+
exitCode,
|
|
1585
|
+
output: output.trim(),
|
|
1586
|
+
stderr: stderr.trim(),
|
|
1587
|
+
signal
|
|
1588
|
+
});
|
|
1589
|
+
});
|
|
1590
|
+
});
|
|
1591
|
+
}
|
|
1592
|
+
var TaskShellCommand = class extends TaskMaster {
|
|
1593
|
+
async run() {
|
|
1594
|
+
const params = this.task?.params;
|
|
1595
|
+
const commandRaw = typeof params === "string" ? params : params?.command;
|
|
1596
|
+
const cwdRaw = typeof params === "string" ? void 0 : params?.cwd;
|
|
1597
|
+
const command = typeof commandRaw === "string" ? commandRaw.trim() : "";
|
|
1598
|
+
const cwd = typeof cwdRaw === "string" && cwdRaw.trim() ? cwdRaw.trim() : void 0;
|
|
1599
|
+
if (!command) {
|
|
1600
|
+
return {
|
|
1601
|
+
success: false,
|
|
1602
|
+
results: {
|
|
1603
|
+
error: 'Validation failed: param "command" must be a non-empty string',
|
|
1604
|
+
received: this.task?.params ?? null
|
|
1605
|
+
}
|
|
1606
|
+
};
|
|
1607
|
+
}
|
|
1608
|
+
try {
|
|
1609
|
+
const result = await runShellCommand(command, cwd);
|
|
1610
|
+
const success = result.exitCode === 0;
|
|
1611
|
+
this.context.logger.info?.(
|
|
1612
|
+
`[TaskShellCommand] command="${command}" exitCode=${String(result.exitCode)} (${this.task.id})`
|
|
1613
|
+
);
|
|
1614
|
+
return {
|
|
1615
|
+
success,
|
|
1616
|
+
results: {
|
|
1617
|
+
command,
|
|
1618
|
+
cwd: cwd ?? process.cwd(),
|
|
1619
|
+
output: result.output,
|
|
1620
|
+
stderr: result.stderr,
|
|
1621
|
+
exitCode: result.exitCode,
|
|
1622
|
+
signal: result.signal
|
|
1623
|
+
}
|
|
1624
|
+
};
|
|
1625
|
+
} catch (error) {
|
|
1626
|
+
return {
|
|
1627
|
+
success: false,
|
|
1628
|
+
results: {
|
|
1629
|
+
command,
|
|
1630
|
+
cwd: cwd ?? process.cwd(),
|
|
1631
|
+
output: "",
|
|
1632
|
+
stderr: "",
|
|
1633
|
+
exitCode: null,
|
|
1634
|
+
error: error?.message ?? String(error)
|
|
1635
|
+
}
|
|
1636
|
+
};
|
|
1637
|
+
}
|
|
1638
|
+
}
|
|
1639
|
+
};
|
|
1640
|
+
|
|
1641
|
+
// src/tasks/coreTasks/TaskSystemInfo.ts
|
|
1642
|
+
var import_node_os = __toESM(require("os"), 1);
|
|
1643
|
+
var import_promises = __toESM(require("fs/promises"), 1);
|
|
1644
|
+
function toGb(valueBytes) {
|
|
1645
|
+
return `${(valueBytes / 1024 ** 3).toFixed(2)} GB`;
|
|
1646
|
+
}
|
|
1647
|
+
function toMb(valueBytes) {
|
|
1648
|
+
return `${(valueBytes / 1024 ** 2).toFixed(2)} MB`;
|
|
1649
|
+
}
|
|
1650
|
+
async function getDiskStats() {
|
|
1651
|
+
const stats = await import_promises.default.statfs("/");
|
|
1652
|
+
const total = Number(stats.bsize) * Number(stats.blocks);
|
|
1653
|
+
const free = Number(stats.bsize) * Number(stats.bavail);
|
|
1654
|
+
const used = total - free;
|
|
1655
|
+
return {
|
|
1656
|
+
total: toGb(total),
|
|
1657
|
+
used: toGb(used),
|
|
1658
|
+
free: toGb(free)
|
|
1659
|
+
};
|
|
1660
|
+
}
|
|
1661
|
+
var TaskSystemInfo = class extends TaskMaster {
|
|
1662
|
+
async run() {
|
|
1663
|
+
try {
|
|
1664
|
+
const totalMemory = import_node_os.default.totalmem();
|
|
1665
|
+
const freeMemory = import_node_os.default.freemem();
|
|
1666
|
+
const usedMemory = totalMemory - freeMemory;
|
|
1667
|
+
const cpus = import_node_os.default.cpus();
|
|
1668
|
+
const cpuUtilization = cpus.map((cpu) => {
|
|
1669
|
+
const total = Object.values(cpu.times).reduce((acc, time) => acc + time, 0);
|
|
1670
|
+
const usage = (total - cpu.times.idle) / total * 100;
|
|
1671
|
+
return Number(usage.toFixed(2));
|
|
1672
|
+
});
|
|
1673
|
+
const processMemory = process.memoryUsage();
|
|
1674
|
+
const disk = await getDiskStats();
|
|
1675
|
+
const results = {
|
|
1676
|
+
memory: {
|
|
1677
|
+
total: toGb(totalMemory),
|
|
1678
|
+
used: toGb(usedMemory),
|
|
1679
|
+
free: toGb(freeMemory)
|
|
1680
|
+
},
|
|
1681
|
+
processMemory: {
|
|
1682
|
+
rss: toMb(processMemory.rss),
|
|
1683
|
+
heapTotal: toMb(processMemory.heapTotal),
|
|
1684
|
+
heapUsed: toMb(processMemory.heapUsed),
|
|
1685
|
+
external: toMb(processMemory.external)
|
|
1686
|
+
},
|
|
1687
|
+
disk,
|
|
1688
|
+
cpu: {
|
|
1689
|
+
cores: cpuUtilization.length,
|
|
1690
|
+
utilization: cpuUtilization
|
|
1691
|
+
},
|
|
1692
|
+
runtime: {
|
|
1693
|
+
platform: import_node_os.default.platform(),
|
|
1694
|
+
arch: import_node_os.default.arch(),
|
|
1695
|
+
uptimeSec: import_node_os.default.uptime(),
|
|
1696
|
+
hostname: import_node_os.default.hostname()
|
|
1697
|
+
}
|
|
1698
|
+
};
|
|
1699
|
+
this.context.logger.info?.(`[TaskSystemInfo] collected system metrics (${this.task.id})`);
|
|
1700
|
+
return { success: true, results };
|
|
1701
|
+
} catch (error) {
|
|
1702
|
+
return {
|
|
1703
|
+
success: false,
|
|
1704
|
+
results: {
|
|
1705
|
+
error: "Can't collect system stats",
|
|
1706
|
+
message: error?.message ?? String(error)
|
|
1707
|
+
}
|
|
1708
|
+
};
|
|
1709
|
+
}
|
|
1710
|
+
}
|
|
1711
|
+
};
|
|
1712
|
+
|
|
1713
|
+
// src/tasks/coreTasks/TaskSumAB.ts
|
|
1714
|
+
var TaskSumAB = class extends TaskMaster {
|
|
1715
|
+
async run() {
|
|
1716
|
+
const a = this.task?.params?.a;
|
|
1717
|
+
const b = this.task?.params?.b;
|
|
1718
|
+
if (typeof a !== "number" || Number.isNaN(a)) {
|
|
1719
|
+
return {
|
|
1720
|
+
success: false,
|
|
1721
|
+
results: {
|
|
1722
|
+
error: 'Validation failed: param "a" must be a valid number',
|
|
1723
|
+
received: { a, b }
|
|
1724
|
+
}
|
|
1725
|
+
};
|
|
1726
|
+
}
|
|
1727
|
+
if (typeof b !== "number" || Number.isNaN(b)) {
|
|
1728
|
+
return {
|
|
1729
|
+
success: false,
|
|
1730
|
+
results: {
|
|
1731
|
+
error: 'Validation failed: param "b" must be a valid number',
|
|
1732
|
+
received: { a, b }
|
|
1733
|
+
}
|
|
1734
|
+
};
|
|
1735
|
+
}
|
|
1736
|
+
const sum = a + b;
|
|
1737
|
+
this.context.logger.info?.(`[TaskSumAB] ${a} + ${b} = ${sum} (${this.task.id})`);
|
|
1738
|
+
return {
|
|
1739
|
+
success: true,
|
|
1740
|
+
results: { a, b, sum }
|
|
1741
|
+
};
|
|
1742
|
+
}
|
|
1743
|
+
};
|
|
1744
|
+
|
|
1745
|
+
// src/tasks/coreTasks/TaskStopRunner.ts
|
|
1746
|
+
var TaskStopRunner = class extends TaskMaster {
|
|
1747
|
+
async run() {
|
|
1748
|
+
const allowanceMs = Number(this.task?.params?.allowanceMs ?? 5e3);
|
|
1749
|
+
this.context.logger.warn?.(`[TaskStopRunner] stop requested (allowanceMs=${allowanceMs})`);
|
|
1750
|
+
return {
|
|
1751
|
+
success: true,
|
|
1752
|
+
results: {
|
|
1753
|
+
stopRunner: true,
|
|
1754
|
+
allowanceMs,
|
|
1755
|
+
message: "Runner stop requested"
|
|
1756
|
+
}
|
|
1757
|
+
};
|
|
1758
|
+
}
|
|
1759
|
+
};
|
|
1760
|
+
|
|
1761
|
+
// src/tasks/TasksRegistry.ts
|
|
1762
|
+
var TasksRegistry = class _TasksRegistry {
|
|
1763
|
+
map = {};
|
|
1764
|
+
constructor(initial) {
|
|
1765
|
+
if (initial) {
|
|
1766
|
+
this.addMany(initial);
|
|
1767
|
+
}
|
|
1768
|
+
}
|
|
1769
|
+
static withCoreTasks() {
|
|
1770
|
+
return new _TasksRegistry().add("ping", TaskPing).add("sampleProcess", TaskSampleProcess).add("shellCommand", TaskShellCommand).add("systemInfo", TaskSystemInfo).add("taskSumAB", TaskSumAB).add("stopRunner", TaskStopRunner).add("stop", TaskStopRunner);
|
|
1771
|
+
}
|
|
1772
|
+
add(taskName, taskClass) {
|
|
1773
|
+
this.map[taskName] = taskClass;
|
|
1774
|
+
return this;
|
|
1775
|
+
}
|
|
1776
|
+
addMany(entries) {
|
|
1777
|
+
for (const [name, klass] of Object.entries(entries)) {
|
|
1778
|
+
this.add(name, klass);
|
|
1779
|
+
}
|
|
1780
|
+
return this;
|
|
1781
|
+
}
|
|
1782
|
+
get(taskName) {
|
|
1783
|
+
return this.map[taskName];
|
|
1784
|
+
}
|
|
1785
|
+
listSupportedTasks() {
|
|
1786
|
+
return Object.keys(this.map).sort();
|
|
1787
|
+
}
|
|
1788
|
+
toObject() {
|
|
1789
|
+
return { ...this.map };
|
|
1790
|
+
}
|
|
1791
|
+
};
|
|
1792
|
+
|
|
1793
|
+
// src/tasks/taskScriptRunner.ts
|
|
1794
|
+
var import_node_child_process2 = require("child_process");
|
|
1795
|
+
function toCliArgs(args = []) {
|
|
1796
|
+
return args.filter((a) => typeof a === "string" && a.length > 0);
|
|
1797
|
+
}
|
|
1798
|
+
function formatChildLogPrefix(task) {
|
|
1799
|
+
return `${task.task}:${task.id.slice(0, 8)}${task.opid ? `:${task.opid}` : ""}`;
|
|
1800
|
+
}
|
|
1801
|
+
async function runNodeTaskScript(context, options) {
|
|
1802
|
+
const cliArgs = toCliArgs(["--route=ipc", "--mode=json", ...options.args || []]);
|
|
1803
|
+
const inheritedExecArgs = Array.isArray(process.execArgv) ? [...process.execArgv] : [];
|
|
1804
|
+
const hasTsRuntimeInParent = inheritedExecArgs.some((arg) => /tsx|ts-node/i.test(arg));
|
|
1805
|
+
const nodeArgs = hasTsRuntimeInParent ? [...inheritedExecArgs, options.scriptPath, ...cliArgs] : ["--import", "tsx", options.scriptPath, ...cliArgs];
|
|
1806
|
+
const child = (0, import_node_child_process2.spawn)(
|
|
1807
|
+
process.execPath,
|
|
1808
|
+
nodeArgs,
|
|
1809
|
+
{
|
|
1810
|
+
cwd: options.cwd || process.cwd(),
|
|
1811
|
+
stdio: ["ignore", "pipe", "pipe", "ipc"],
|
|
1812
|
+
env: {
|
|
1813
|
+
...process.env,
|
|
1814
|
+
TASK_ID: options.task.id,
|
|
1815
|
+
TASK_NAME: options.task.task,
|
|
1816
|
+
TASK_OPID: options.task.opid || ""
|
|
1817
|
+
}
|
|
1818
|
+
}
|
|
1819
|
+
);
|
|
1820
|
+
let stdout = "";
|
|
1821
|
+
let stderr = "";
|
|
1822
|
+
let workerResult = null;
|
|
1823
|
+
let hadErrorMessage = false;
|
|
1824
|
+
const prefix = formatChildLogPrefix(options.task);
|
|
1825
|
+
const db = context.db;
|
|
1826
|
+
const tasksTable = context.params?.get?.("table") || "tasks";
|
|
1827
|
+
let progressWriteChain = Promise.resolve();
|
|
1828
|
+
let progressCallbackChain = Promise.resolve();
|
|
1829
|
+
const updateProgress = (text) => {
|
|
1830
|
+
if (!db || !text || !text.trim()) return;
|
|
1831
|
+
progressWriteChain = progressWriteChain.then(async () => {
|
|
1832
|
+
await db(tasksTable).where({ id: options.task.id }).update({ progress: text.slice(0, 4e3) });
|
|
1833
|
+
}).catch((error) => {
|
|
1834
|
+
context.logger.warn?.(
|
|
1835
|
+
`[tasks] failed to update progress for task=${options.task.id}: ${error?.message ?? String(error)}`
|
|
1836
|
+
);
|
|
1837
|
+
});
|
|
1838
|
+
if (options.onProgress) {
|
|
1839
|
+
progressCallbackChain = progressCallbackChain.then(async () => {
|
|
1840
|
+
await options.onProgress?.(text.slice(0, 4e3));
|
|
1841
|
+
}).catch((error) => {
|
|
1842
|
+
context.logger.warn?.(
|
|
1843
|
+
`[tasks] reportProgress callback failed for task=${options.task.id}: ${error?.message ?? String(error)}`
|
|
1844
|
+
);
|
|
1845
|
+
});
|
|
1846
|
+
}
|
|
1847
|
+
};
|
|
1848
|
+
const payloadToProgressText = (payload) => {
|
|
1849
|
+
if (!payload) return "";
|
|
1850
|
+
if (typeof payload === "string") return payload;
|
|
1851
|
+
if (typeof payload.message === "string" && payload.level === "progress" && payload.count !== void 0 && payload.total !== void 0) {
|
|
1852
|
+
const pfx = payload.prefix ? `${payload.prefix} ` : "";
|
|
1853
|
+
return `${pfx}${payload.message} ${payload.count}/${payload.total}`;
|
|
1854
|
+
}
|
|
1855
|
+
if (typeof payload.message === "string") return payload.message;
|
|
1856
|
+
if (payload.level === "progress" && payload.count !== void 0 && payload.total !== void 0) {
|
|
1857
|
+
const pfx = payload.prefix ? `${payload.prefix} ` : "";
|
|
1858
|
+
return `${pfx}${payload.count}/${payload.total}`;
|
|
1859
|
+
}
|
|
1860
|
+
return "";
|
|
1861
|
+
};
|
|
1862
|
+
child.stdout.on("data", (chunk) => {
|
|
1863
|
+
const text = String(chunk);
|
|
1864
|
+
stdout += text;
|
|
1865
|
+
if (text.trim()) {
|
|
1866
|
+
context.logger.info?.(`[child:${prefix}] ${text.trimEnd()}`);
|
|
1867
|
+
updateProgress(text.trim().replace(/\s+/g, " ").slice(0, 400));
|
|
1868
|
+
}
|
|
1869
|
+
});
|
|
1870
|
+
child.stderr.on("data", (chunk) => {
|
|
1871
|
+
const text = String(chunk);
|
|
1872
|
+
stderr += text;
|
|
1873
|
+
if (text.trim()) {
|
|
1874
|
+
context.logger.warn?.(`[child:${prefix}] ${text.trimEnd()}`);
|
|
1875
|
+
}
|
|
1876
|
+
});
|
|
1877
|
+
child.on("message", (message) => {
|
|
1878
|
+
if (message && typeof message === "object" && "__taskWorkerResult" in message) {
|
|
1879
|
+
workerResult = message.__taskWorkerResult;
|
|
1880
|
+
return;
|
|
1881
|
+
}
|
|
1882
|
+
if (message && typeof message === "object") {
|
|
1883
|
+
const level = typeof message.level === "string" ? message.level.toLowerCase() : "";
|
|
1884
|
+
if (level === "error" || level === "fatal") {
|
|
1885
|
+
hadErrorMessage = true;
|
|
1886
|
+
}
|
|
1887
|
+
}
|
|
1888
|
+
appendTaskIpcLog(context, options.task, message);
|
|
1889
|
+
const progressText = payloadToProgressText(message);
|
|
1890
|
+
if (progressText) {
|
|
1891
|
+
updateProgress(progressText);
|
|
1892
|
+
if (typeof message === "object" && message?.level === "progress" && message.count !== void 0 && message.total !== void 0) {
|
|
1893
|
+
const countNum = Number(String(message.count).trim());
|
|
1894
|
+
const totalNum = Number(message.total);
|
|
1895
|
+
if (Number.isFinite(countNum) && Number.isFinite(totalNum) && totalNum > 0) {
|
|
1896
|
+
context.logger.progress(message.message || "progress", {
|
|
1897
|
+
prefix: message.prefix || prefix,
|
|
1898
|
+
count: countNum,
|
|
1899
|
+
total: totalNum
|
|
1900
|
+
});
|
|
1901
|
+
} else {
|
|
1902
|
+
context.logger.info?.(`[child:${prefix}] ${progressText}`);
|
|
1903
|
+
}
|
|
1904
|
+
} else {
|
|
1905
|
+
context.logger.info?.(`[child:${prefix}] ${progressText}`);
|
|
1906
|
+
}
|
|
1907
|
+
}
|
|
1908
|
+
});
|
|
1909
|
+
return await new Promise((resolve, reject) => {
|
|
1910
|
+
child.on("error", (error) => reject(error));
|
|
1911
|
+
child.on("close", (exitCode, signal) => {
|
|
1912
|
+
Promise.allSettled([progressWriteChain, progressCallbackChain]).finally(() => {
|
|
1913
|
+
resolve({
|
|
1914
|
+
exitCode,
|
|
1915
|
+
signal,
|
|
1916
|
+
stdout: stdout.trim(),
|
|
1917
|
+
stderr: stderr.trim(),
|
|
1918
|
+
workerResult,
|
|
1919
|
+
hadErrorMessage
|
|
1920
|
+
});
|
|
1921
|
+
});
|
|
1922
|
+
});
|
|
1923
|
+
});
|
|
1924
|
+
}
|
|
1925
|
+
|
|
1926
|
+
// src/tasks/index.ts
|
|
1927
|
+
var LOCKED_BY_ERROR_MESSAGE = "locked by error";
|
|
1928
|
+
var defaultTasksRegistry = TasksRegistry.withCoreTasks();
|
|
1929
|
+
function getDb2(context) {
|
|
1930
|
+
const db = context.db;
|
|
1931
|
+
if (!db) {
|
|
1932
|
+
throw new Error("Tasks component requires context.db. Initialize DB first and attach to context.");
|
|
1933
|
+
}
|
|
1934
|
+
return db;
|
|
1935
|
+
}
|
|
1936
|
+
function normalizeRegistry(registry) {
|
|
1937
|
+
if (!registry) return defaultTasksRegistry;
|
|
1938
|
+
if (registry instanceof TasksRegistry) return registry;
|
|
1939
|
+
return new TasksRegistry().addMany(registry);
|
|
1940
|
+
}
|
|
1941
|
+
function normalizeAllowedTasks(value) {
|
|
1942
|
+
if (!value) return void 0;
|
|
1943
|
+
if (Array.isArray(value)) {
|
|
1944
|
+
const out2 = value.map((v) => String(v).trim()).filter(Boolean);
|
|
1945
|
+
return out2.length ? out2 : void 0;
|
|
1946
|
+
}
|
|
1947
|
+
const out = String(value).split(",").map((v) => v.trim()).filter(Boolean);
|
|
1948
|
+
return out.length ? out : void 0;
|
|
1949
|
+
}
|
|
1950
|
+
async function enqueueStopTask(context, target, queue = "tasks", allowanceMs = 5e3) {
|
|
1951
|
+
return enqueueTask(context, {
|
|
1952
|
+
queue,
|
|
1953
|
+
target,
|
|
1954
|
+
task: "stopRunner",
|
|
1955
|
+
params: { allowanceMs },
|
|
1956
|
+
priority: 1e6
|
|
1957
|
+
});
|
|
1958
|
+
}
|
|
1959
|
+
async function signalRunningTasksStop(context, runningTaskInstances, allowanceMs) {
|
|
1960
|
+
context.logger.warn?.(`[tasks] signaling ${runningTaskInstances.size} running task(s) to stop`);
|
|
1961
|
+
for (const [, taskInstance] of runningTaskInstances) {
|
|
1962
|
+
if (typeof taskInstance.requestStop === "function") {
|
|
1963
|
+
try {
|
|
1964
|
+
await taskInstance.requestStop(allowanceMs);
|
|
1965
|
+
} catch (error) {
|
|
1966
|
+
context.logger.warn?.("[tasks] task requestStop failed:", error);
|
|
1967
|
+
}
|
|
1968
|
+
}
|
|
1969
|
+
}
|
|
1970
|
+
context.emitter.emit("stop", allowanceMs);
|
|
1971
|
+
}
|
|
1972
|
+
async function executeClaimedTask(context, tasksTable, historyTable, row, registry, runningTaskInstances) {
|
|
1973
|
+
const db = getDb2(context);
|
|
1974
|
+
const taskName = row.task;
|
|
1975
|
+
const TaskClass = registry.get(taskName);
|
|
1976
|
+
const { paused_at: _pausedAt, ...rowForHistory } = row;
|
|
1977
|
+
if (!TaskClass) {
|
|
1978
|
+
const err = { message: `Unknown task "${taskName}"` };
|
|
1979
|
+
await db(historyTable).insert({
|
|
1980
|
+
...rowForHistory,
|
|
1981
|
+
completed_at: /* @__PURE__ */ new Date(),
|
|
1982
|
+
success: false,
|
|
1983
|
+
params: toJsonColumn(row.params),
|
|
1984
|
+
results: toJsonColumn(err)
|
|
1985
|
+
});
|
|
1986
|
+
if (row.schedule) {
|
|
1987
|
+
await db(tasksTable).where({ id: row.id }).update({
|
|
1988
|
+
started_at: null,
|
|
1989
|
+
completed_at: /* @__PURE__ */ new Date(),
|
|
1990
|
+
success: false,
|
|
1991
|
+
results: toJsonColumn(err),
|
|
1992
|
+
past_due: null,
|
|
1993
|
+
paused_at: db.fn.now(),
|
|
1994
|
+
progress: LOCKED_BY_ERROR_MESSAGE
|
|
1995
|
+
});
|
|
1996
|
+
} else {
|
|
1997
|
+
await db(tasksTable).where({ id: row.id }).delete();
|
|
1998
|
+
}
|
|
1999
|
+
return { stopRunnerRequested: false, stopAllowanceMs: 0 };
|
|
2000
|
+
}
|
|
2001
|
+
let success = false;
|
|
2002
|
+
let results = null;
|
|
2003
|
+
let taskInstance = null;
|
|
2004
|
+
try {
|
|
2005
|
+
taskInstance = new TaskClass(context, row);
|
|
2006
|
+
runningTaskInstances.set(row.id, taskInstance);
|
|
2007
|
+
const runResult = await taskInstance.run((progress) => updateTaskProgress(context, tasksTable, row.id, progress));
|
|
2008
|
+
success = !!runResult?.success;
|
|
2009
|
+
results = runResult?.results ?? null;
|
|
2010
|
+
} catch (error) {
|
|
2011
|
+
success = false;
|
|
2012
|
+
results = {
|
|
2013
|
+
message: error?.message ?? String(error),
|
|
2014
|
+
name: error?.name ?? "Error",
|
|
2015
|
+
stack: error?.stack ?? null
|
|
2016
|
+
};
|
|
2017
|
+
} finally {
|
|
2018
|
+
runningTaskInstances.delete(row.id);
|
|
2019
|
+
}
|
|
2020
|
+
await db(historyTable).insert({
|
|
2021
|
+
...rowForHistory,
|
|
2022
|
+
completed_at: /* @__PURE__ */ new Date(),
|
|
2023
|
+
success,
|
|
2024
|
+
params: toJsonColumn(row.params),
|
|
2025
|
+
results: toJsonColumn(results)
|
|
2026
|
+
});
|
|
2027
|
+
if (!success) {
|
|
2028
|
+
const dbName = String(context?.params?.get?.("dbName") || "local");
|
|
2029
|
+
const tableName = String(context?.params?.get?.("table") || "tasks");
|
|
2030
|
+
const fallbackRecoverCommand = [
|
|
2031
|
+
"npx",
|
|
2032
|
+
"tsx",
|
|
2033
|
+
"examples/tasks/recover-task.ts",
|
|
2034
|
+
`--dbName='${dbName.replace(/'/g, `'\\''`)}'`,
|
|
2035
|
+
`--table='${tableName.replace(/'/g, `'\\''`)}'`,
|
|
2036
|
+
`--id='${String(row.id).replace(/'/g, `'\\''`)}'`
|
|
2037
|
+
].join(" ");
|
|
2038
|
+
const rerunCommand = results && typeof results === "object" && results.rerunCommand ? results.rerunCommand : fallbackRecoverCommand;
|
|
2039
|
+
appendTaskIpcLog(context, row, {
|
|
2040
|
+
level: "error",
|
|
2041
|
+
message: `[tasks] task failed: ${row.task} id=${row.id}. Once problem is fixed, re-run: ${rerunCommand}`,
|
|
2042
|
+
details: results
|
|
2043
|
+
});
|
|
2044
|
+
}
|
|
2045
|
+
if (row.schedule) {
|
|
2046
|
+
if (success) {
|
|
2047
|
+
await db(tasksTable).where({ id: row.id }).update({
|
|
2048
|
+
started_at: null,
|
|
2049
|
+
completed_at: /* @__PURE__ */ new Date(),
|
|
2050
|
+
success,
|
|
2051
|
+
results: toJsonColumn(results),
|
|
2052
|
+
progress: null,
|
|
2053
|
+
past_due: null
|
|
2054
|
+
});
|
|
2055
|
+
} else {
|
|
2056
|
+
await db(tasksTable).where({ id: row.id }).update({
|
|
2057
|
+
started_at: null,
|
|
2058
|
+
completed_at: /* @__PURE__ */ new Date(),
|
|
2059
|
+
success,
|
|
2060
|
+
results: toJsonColumn(results),
|
|
2061
|
+
paused_at: db.fn.now(),
|
|
2062
|
+
progress: LOCKED_BY_ERROR_MESSAGE,
|
|
2063
|
+
past_due: null
|
|
2064
|
+
});
|
|
2065
|
+
}
|
|
2066
|
+
} else {
|
|
2067
|
+
await db(tasksTable).where({ id: row.id }).delete();
|
|
2068
|
+
}
|
|
2069
|
+
const stopRunnerRequested = !!(results && typeof results === "object" && results.stopRunner === true);
|
|
2070
|
+
const stopAllowanceMs = stopRunnerRequested ? Number(results.allowanceMs ?? 5e3) : 0;
|
|
2071
|
+
return { stopRunnerRequested, stopAllowanceMs };
|
|
2072
|
+
}
|
|
2073
|
+
async function claimNextRunnableTask(context, tasksTable, target, registry, scanLimit, taskNames) {
|
|
2074
|
+
const db = getDb2(context);
|
|
2075
|
+
let query = db(tasksTable).whereNull("started_at").whereNull("paused_at").where({ target }).orderByRaw("CASE WHEN past_due IS NULL THEN 1 ELSE 0 END ASC").orderBy([{ column: "priority", order: "desc" }]).orderByRaw("CASE WHEN completed_at IS NULL THEN 0 ELSE 1 END ASC").orderBy([{ column: "completed_at", order: "asc" }, { column: "created_at", order: "asc" }]).limit(scanLimit);
|
|
2076
|
+
if (taskNames && taskNames.length > 0) {
|
|
2077
|
+
query = query.whereIn("task", taskNames);
|
|
2078
|
+
}
|
|
2079
|
+
const candidates = await query;
|
|
2080
|
+
for (const row of candidates) {
|
|
2081
|
+
if (!row.past_due && row.schedule && !timeMatcher(row.schedule)) {
|
|
2082
|
+
continue;
|
|
2083
|
+
}
|
|
2084
|
+
const TaskClass = registry.get(row.task);
|
|
2085
|
+
if (TaskClass) {
|
|
2086
|
+
const taskInstance = new TaskClass(context, row);
|
|
2087
|
+
const reason = taskInstance.cantRunReason ? await taskInstance.cantRunReason() : null;
|
|
2088
|
+
if (reason) {
|
|
2089
|
+
if (!row.past_due) {
|
|
2090
|
+
await db(tasksTable).where({ id: row.id }).update({
|
|
2091
|
+
past_due: db.fn.now(),
|
|
2092
|
+
progress: String(reason)
|
|
2093
|
+
});
|
|
2094
|
+
}
|
|
2095
|
+
continue;
|
|
2096
|
+
}
|
|
2097
|
+
}
|
|
2098
|
+
const updated = await db(tasksTable).where({ id: row.id }).whereNull("started_at").whereNull("paused_at").update({ started_at: db.fn.now() }).returning("*");
|
|
2099
|
+
const claimed = Array.isArray(updated) ? updated[0] : null;
|
|
2100
|
+
if (claimed) return claimed;
|
|
2101
|
+
}
|
|
2102
|
+
return null;
|
|
2103
|
+
}
|
|
2104
|
+
async function runTasksLoop(context, options) {
|
|
2105
|
+
const queue = options.queue ?? "tasks";
|
|
2106
|
+
const target = options.target;
|
|
2107
|
+
const pollMs = options.pollMs ?? 1e3;
|
|
2108
|
+
const maxParallel = options.maxParallel ?? 1;
|
|
2109
|
+
const scanLimit = options.scanLimit ?? 100;
|
|
2110
|
+
const allowedTasks = normalizeAllowedTasks(options.allowedTasks);
|
|
2111
|
+
const registry = normalizeRegistry(options.registry);
|
|
2112
|
+
const { tasksTable, historyTable } = queueToTableNames(queue);
|
|
2113
|
+
if (!target) throw new Error("runTasksLoop: target is required");
|
|
2114
|
+
const runningPromises = /* @__PURE__ */ new Set();
|
|
2115
|
+
const runningTaskInstances = /* @__PURE__ */ new Map();
|
|
2116
|
+
let runningStopControlPromise = null;
|
|
2117
|
+
let stopRequested = false;
|
|
2118
|
+
let stopAllowanceMs = 5e3;
|
|
2119
|
+
context.__tasksRunnerStop = false;
|
|
2120
|
+
while (!context.isStop() && !stopRequested && !context.__tasksRunnerStop) {
|
|
2121
|
+
if (!runningStopControlPromise) {
|
|
2122
|
+
const claimedStopTask = await claimNextRunnableTask(
|
|
2123
|
+
context,
|
|
2124
|
+
tasksTable,
|
|
2125
|
+
target,
|
|
2126
|
+
registry,
|
|
2127
|
+
10,
|
|
2128
|
+
["stopRunner", "stop"]
|
|
2129
|
+
);
|
|
2130
|
+
if (claimedStopTask) {
|
|
2131
|
+
runningStopControlPromise = executeClaimedTask(
|
|
2132
|
+
context,
|
|
2133
|
+
tasksTable,
|
|
2134
|
+
historyTable,
|
|
2135
|
+
claimedStopTask,
|
|
2136
|
+
registry,
|
|
2137
|
+
runningTaskInstances
|
|
2138
|
+
).then(async (outcome) => {
|
|
2139
|
+
if (outcome.stopRunnerRequested && !stopRequested) {
|
|
2140
|
+
stopRequested = true;
|
|
2141
|
+
stopAllowanceMs = outcome.stopAllowanceMs || 5e3;
|
|
2142
|
+
context.__tasksRunnerStop = true;
|
|
2143
|
+
await signalRunningTasksStop(context, runningTaskInstances, stopAllowanceMs);
|
|
2144
|
+
}
|
|
2145
|
+
}).finally(() => {
|
|
2146
|
+
runningStopControlPromise = null;
|
|
2147
|
+
});
|
|
2148
|
+
}
|
|
2149
|
+
}
|
|
2150
|
+
while (runningPromises.size < maxParallel) {
|
|
2151
|
+
const claimed = await claimNextRunnableTask(
|
|
2152
|
+
context,
|
|
2153
|
+
tasksTable,
|
|
2154
|
+
target,
|
|
2155
|
+
registry,
|
|
2156
|
+
scanLimit,
|
|
2157
|
+
allowedTasks
|
|
2158
|
+
);
|
|
2159
|
+
if (!claimed) break;
|
|
2160
|
+
const p = executeClaimedTask(context, tasksTable, historyTable, claimed, registry, runningTaskInstances).then(async (outcome) => {
|
|
2161
|
+
if (outcome.stopRunnerRequested && !stopRequested) {
|
|
2162
|
+
stopRequested = true;
|
|
2163
|
+
stopAllowanceMs = outcome.stopAllowanceMs || 5e3;
|
|
2164
|
+
context.__tasksRunnerStop = true;
|
|
2165
|
+
await signalRunningTasksStop(context, runningTaskInstances, stopAllowanceMs);
|
|
2166
|
+
}
|
|
2167
|
+
}).finally(() => {
|
|
2168
|
+
runningPromises.delete(p);
|
|
2169
|
+
});
|
|
2170
|
+
runningPromises.add(p);
|
|
2171
|
+
}
|
|
2172
|
+
await sleepMs(pollMs);
|
|
2173
|
+
}
|
|
2174
|
+
if (context.isStop() && !stopRequested) {
|
|
2175
|
+
await signalRunningTasksStop(context, runningTaskInstances, 5e3);
|
|
2176
|
+
}
|
|
2177
|
+
if (runningPromises.size > 0) {
|
|
2178
|
+
if (stopRequested) {
|
|
2179
|
+
await Promise.race([
|
|
2180
|
+
Promise.allSettled(Array.from(runningPromises)),
|
|
2181
|
+
sleepMs(stopAllowanceMs).then(() => {
|
|
2182
|
+
context.logger.warn?.(
|
|
2183
|
+
`[tasks] stop allowance (${stopAllowanceMs}ms) reached; ${runningPromises.size} task(s) still running`
|
|
2184
|
+
);
|
|
2185
|
+
})
|
|
2186
|
+
]);
|
|
2187
|
+
} else {
|
|
2188
|
+
await Promise.allSettled(Array.from(runningPromises));
|
|
2189
|
+
}
|
|
2190
|
+
}
|
|
2191
|
+
}
|
|
2192
|
+
async function waitForTaskResult(context, taskId, options = {}) {
|
|
2193
|
+
const db = getDb2(context);
|
|
2194
|
+
const queue = options.queue ?? "tasks";
|
|
2195
|
+
const timeoutMs = options.timeoutMs ?? 6e4;
|
|
2196
|
+
const pollMs = options.pollMs ?? 500;
|
|
2197
|
+
const { tasksTable, historyTable } = queueToTableNames(queue);
|
|
2198
|
+
const deadline = Date.now() + timeoutMs;
|
|
2199
|
+
while (Date.now() <= deadline) {
|
|
2200
|
+
const done = await db(historyTable).where({ id: taskId }).orderBy("created_at", "desc").first();
|
|
2201
|
+
if (done) return done;
|
|
2202
|
+
const pending = await db(tasksTable).where({ id: taskId }).first();
|
|
2203
|
+
if (!pending) {
|
|
2204
|
+
const maybeDone = await db(historyTable).where({ id: taskId }).orderBy("created_at", "desc").first();
|
|
2205
|
+
return maybeDone ?? null;
|
|
2206
|
+
}
|
|
2207
|
+
await sleepMs(pollMs);
|
|
2208
|
+
}
|
|
2209
|
+
return null;
|
|
2210
|
+
}
|
|
2211
|
+
var TasksManager = class _TasksManager {
|
|
2212
|
+
context;
|
|
2213
|
+
queue;
|
|
2214
|
+
target;
|
|
2215
|
+
recreateTaskTables;
|
|
2216
|
+
pollMs;
|
|
2217
|
+
maxParallel;
|
|
2218
|
+
scanLimit;
|
|
2219
|
+
allowedTasks;
|
|
2220
|
+
registry;
|
|
2221
|
+
constructor(context, options = {}) {
|
|
2222
|
+
this.context = context;
|
|
2223
|
+
this.queue = options.queue ?? "tasks";
|
|
2224
|
+
this.target = options.target ?? "localRunner";
|
|
2225
|
+
this.recreateTaskTables = options.recreateTaskTables ?? false;
|
|
2226
|
+
this.pollMs = options.pollMs ?? 1e3;
|
|
2227
|
+
this.maxParallel = options.maxParallel ?? 1;
|
|
2228
|
+
this.scanLimit = options.scanLimit ?? 100;
|
|
2229
|
+
this.allowedTasks = normalizeAllowedTasks(options.allowedTasks);
|
|
2230
|
+
this.registry = normalizeRegistry(options.registry);
|
|
2231
|
+
}
|
|
2232
|
+
static init(context, options = {}) {
|
|
2233
|
+
const defs = {
|
|
2234
|
+
table: "string default tasks",
|
|
2235
|
+
target: "string default localRunner",
|
|
2236
|
+
recreateTaskTables: "boolean default false",
|
|
2237
|
+
pollMs: "number default 1000",
|
|
2238
|
+
maxParallel: "number default 1",
|
|
2239
|
+
scanLimit: "number default 100",
|
|
2240
|
+
allowedTasks: "string"
|
|
2241
|
+
};
|
|
2242
|
+
const discovered = context.params.getAllForModule(defs);
|
|
2243
|
+
const resolved = {
|
|
2244
|
+
queue: discovered.table,
|
|
2245
|
+
target: discovered.target,
|
|
2246
|
+
recreateTaskTables: discovered.recreateTaskTables,
|
|
2247
|
+
pollMs: discovered.pollMs,
|
|
2248
|
+
maxParallel: discovered.maxParallel,
|
|
2249
|
+
scanLimit: discovered.scanLimit,
|
|
2250
|
+
allowedTasks: discovered.allowedTasks,
|
|
2251
|
+
...options
|
|
2252
|
+
};
|
|
2253
|
+
return new _TasksManager(context, resolved);
|
|
2254
|
+
}
|
|
2255
|
+
async ensureTaskTables(options = {}) {
|
|
2256
|
+
await ensureTaskTables(this.context, {
|
|
2257
|
+
queue: this.queue,
|
|
2258
|
+
recreate: options.recreate ?? this.recreateTaskTables
|
|
2259
|
+
});
|
|
2260
|
+
}
|
|
2261
|
+
async runTasksLoop(options = {}) {
|
|
2262
|
+
await runTasksLoop(this.context, {
|
|
2263
|
+
queue: options.queue ?? this.queue,
|
|
2264
|
+
target: options.target ?? this.target,
|
|
2265
|
+
pollMs: options.pollMs ?? this.pollMs,
|
|
2266
|
+
maxParallel: options.maxParallel ?? this.maxParallel,
|
|
2267
|
+
scanLimit: options.scanLimit ?? this.scanLimit,
|
|
2268
|
+
allowedTasks: options.allowedTasks ?? this.allowedTasks,
|
|
2269
|
+
registry: options.registry ?? this.registry
|
|
2270
|
+
});
|
|
2271
|
+
}
|
|
2272
|
+
};
|
|
2273
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
2274
|
+
0 && (module.exports = {
|
|
2275
|
+
TaskMaster,
|
|
2276
|
+
TaskPing,
|
|
2277
|
+
TaskSampleProcess,
|
|
2278
|
+
TaskShellCommand,
|
|
2279
|
+
TaskStopRunner,
|
|
2280
|
+
TaskSumAB,
|
|
2281
|
+
TaskSystemInfo,
|
|
2282
|
+
TasksManager,
|
|
2283
|
+
TasksRegistry,
|
|
2284
|
+
appendTaskIpcLog,
|
|
2285
|
+
defaultTasksRegistry,
|
|
2286
|
+
enqueueStopTask,
|
|
2287
|
+
enqueueTask,
|
|
2288
|
+
ensureTaskTables,
|
|
2289
|
+
queueToTableNames,
|
|
2290
|
+
runNodeTaskScript,
|
|
2291
|
+
runTasksLoop,
|
|
2292
|
+
updateTaskProgress,
|
|
2293
|
+
waitForTaskResult
|
|
2294
|
+
});
|
|
2295
|
+
//# sourceMappingURL=tasks.cjs.map
|