@aion0/bastion 0.1.13 → 0.1.14
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/classifier/model-manager.d.ts +43 -0
- package/dist/classifier/model-manager.d.ts.map +1 -0
- package/dist/classifier/model-manager.js +161 -0
- package/dist/classifier/model-manager.js.map +1 -0
- package/dist/classifier/onnx-provider.d.ts +22 -0
- package/dist/classifier/onnx-provider.d.ts.map +1 -0
- package/dist/classifier/onnx-provider.js +131 -0
- package/dist/classifier/onnx-provider.js.map +1 -0
- package/dist/cli/commands/plugins.js +3 -3
- package/dist/cli/commands/plugins.js.map +1 -1
- package/dist/plugins/builtin/pi-classifier.d.ts +22 -0
- package/dist/plugins/builtin/pi-classifier.d.ts.map +1 -0
- package/dist/plugins/builtin/pi-classifier.js +177 -0
- package/dist/plugins/builtin/pi-classifier.js.map +1 -0
- package/package.json +1 -1
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
export interface ModelInfo {
|
|
2
|
+
modelId: string;
|
|
3
|
+
dir: string;
|
|
4
|
+
ready: boolean;
|
|
5
|
+
files: string[];
|
|
6
|
+
}
|
|
7
|
+
export interface DownloadProgress {
|
|
8
|
+
file: string;
|
|
9
|
+
received: number;
|
|
10
|
+
total: number;
|
|
11
|
+
percent: number;
|
|
12
|
+
}
|
|
13
|
+
export type ProgressCallback = (progress: DownloadProgress) => void;
|
|
14
|
+
/**
|
|
15
|
+
* Resolve the local directory for a given model ID.
|
|
16
|
+
* Model IDs use '/' which gets replaced with '--' for filesystem safety.
|
|
17
|
+
*/
|
|
18
|
+
export declare function getModelDir(modelId: string): string;
|
|
19
|
+
/**
|
|
20
|
+
* Check if a model is already downloaded and complete.
|
|
21
|
+
*/
|
|
22
|
+
export declare function isModelReady(modelId: string): Promise<boolean>;
|
|
23
|
+
/**
|
|
24
|
+
* Get info about a local model.
|
|
25
|
+
*/
|
|
26
|
+
export declare function getModelInfo(modelId: string): Promise<ModelInfo>;
|
|
27
|
+
/**
|
|
28
|
+
* Download model files from Hugging Face Hub.
|
|
29
|
+
* Files: model.onnx (from onnx/ subdirectory), tokenizer.json, config.json
|
|
30
|
+
*/
|
|
31
|
+
export declare function downloadModel(modelId: string, onProgress?: ProgressCallback, options?: {
|
|
32
|
+
onnxSubdir?: string;
|
|
33
|
+
}): Promise<string>;
|
|
34
|
+
/**
|
|
35
|
+
* Ensure a model is available locally. Downloads if missing.
|
|
36
|
+
*/
|
|
37
|
+
export declare function ensureModel(modelId: string, onProgress?: ProgressCallback, options?: {
|
|
38
|
+
onnxSubdir?: string;
|
|
39
|
+
logger?: {
|
|
40
|
+
info: (msg: string, meta?: Record<string, unknown>) => void;
|
|
41
|
+
};
|
|
42
|
+
}): Promise<string>;
|
|
43
|
+
//# sourceMappingURL=model-manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-manager.d.ts","sourceRoot":"","sources":["../../src/classifier/model-manager.ts"],"names":[],"mappings":"AASA,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,OAAO,CAAC;IACf,KAAK,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,MAAM,gBAAgB,GAAG,CAAC,QAAQ,EAAE,gBAAgB,KAAK,IAAI,CAAC;AAEpE;;;GAGG;AACH,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAGnD;AAED;;GAEG;AACH,wBAAsB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAWpE;AAED;;GAEG;AACH,wBAAsB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAiBtE;AAoFD;;;GAGG;AACH,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,UAAU,CAAC,EAAE,gBAAgB,EAC7B,OAAO,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,GAChC,OAAO,CAAC,MAAM,CAAC,CA8BjB;AAED;;GAEG;AACH,wBAAsB,WAAW,CAC/B,OAAO,EAAE,MAAM,EACf,UAAU,CAAC,EAAE,gBAAgB,EAC7B,OAAO,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE;QAAE,IAAI,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAA;KAAE,CAAA;CAAE,GAC1G,OAAO,CAAC,MAAM,CAAC,CAQjB"}
|
|
@@ -0,0 +1,161 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.getModelDir = getModelDir;
|
|
7
|
+
exports.isModelReady = isModelReady;
|
|
8
|
+
exports.getModelInfo = getModelInfo;
|
|
9
|
+
exports.downloadModel = downloadModel;
|
|
10
|
+
exports.ensureModel = ensureModel;
|
|
11
|
+
const node_path_1 = require("node:path");
|
|
12
|
+
const node_os_1 = require("node:os");
|
|
13
|
+
const promises_1 = require("node:fs/promises");
|
|
14
|
+
const node_https_1 = __importDefault(require("node:https"));
|
|
15
|
+
const MODELS_DIR = (0, node_path_1.join)((0, node_os_1.homedir)(), '.bastion', 'models');
|
|
16
|
+
const REQUIRED_FILES = ['model.onnx', 'tokenizer.json', 'config.json'];
|
|
17
|
+
/**
|
|
18
|
+
* Resolve the local directory for a given model ID.
|
|
19
|
+
* Model IDs use '/' which gets replaced with '--' for filesystem safety.
|
|
20
|
+
*/
|
|
21
|
+
function getModelDir(modelId) {
|
|
22
|
+
const safeName = modelId.replace(/\//g, '--');
|
|
23
|
+
return (0, node_path_1.join)(MODELS_DIR, safeName);
|
|
24
|
+
}
|
|
25
|
+
/**
|
|
26
|
+
* Check if a model is already downloaded and complete.
|
|
27
|
+
*/
|
|
28
|
+
async function isModelReady(modelId) {
|
|
29
|
+
const dir = getModelDir(modelId);
|
|
30
|
+
for (const file of REQUIRED_FILES) {
|
|
31
|
+
try {
|
|
32
|
+
const s = await (0, promises_1.stat)((0, node_path_1.join)(dir, file));
|
|
33
|
+
if (s.size === 0)
|
|
34
|
+
return false;
|
|
35
|
+
}
|
|
36
|
+
catch {
|
|
37
|
+
return false;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
return true;
|
|
41
|
+
}
|
|
42
|
+
/**
|
|
43
|
+
* Get info about a local model.
|
|
44
|
+
*/
|
|
45
|
+
async function getModelInfo(modelId) {
|
|
46
|
+
const dir = getModelDir(modelId);
|
|
47
|
+
const files = [];
|
|
48
|
+
for (const file of REQUIRED_FILES) {
|
|
49
|
+
try {
|
|
50
|
+
await (0, promises_1.stat)((0, node_path_1.join)(dir, file));
|
|
51
|
+
files.push(file);
|
|
52
|
+
}
|
|
53
|
+
catch {
|
|
54
|
+
// file missing
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
return {
|
|
58
|
+
modelId,
|
|
59
|
+
dir,
|
|
60
|
+
ready: files.length === REQUIRED_FILES.length,
|
|
61
|
+
files,
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
/**
|
|
65
|
+
* Download a single file from Hugging Face Hub.
|
|
66
|
+
*/
|
|
67
|
+
function getHfToken() {
|
|
68
|
+
return process.env['HF_TOKEN'] ?? process.env['HUGGING_FACE_HUB_TOKEN'];
|
|
69
|
+
}
|
|
70
|
+
function downloadFile(url, destPath, onProgress, fileName) {
|
|
71
|
+
return new Promise((resolve, reject) => {
|
|
72
|
+
const request = (targetUrl) => {
|
|
73
|
+
const headers = { 'User-Agent': 'bastion-pro/0.1.0' };
|
|
74
|
+
const hfToken = getHfToken();
|
|
75
|
+
if (hfToken && targetUrl.includes('huggingface.co')) {
|
|
76
|
+
headers['Authorization'] = `Bearer ${hfToken}`;
|
|
77
|
+
}
|
|
78
|
+
node_https_1.default.get(targetUrl, { headers }, (res) => {
|
|
79
|
+
// Follow redirects (HF Hub returns 302 to CDN)
|
|
80
|
+
if (res.statusCode && res.statusCode >= 301 && res.statusCode <= 308 && res.statusCode !== 304) {
|
|
81
|
+
let location = res.headers.location;
|
|
82
|
+
if (!location)
|
|
83
|
+
return reject(new Error(`Redirect without location for ${url}`));
|
|
84
|
+
// Handle relative redirects — resolve against current URL
|
|
85
|
+
if (location.startsWith('/')) {
|
|
86
|
+
const base = new URL(targetUrl);
|
|
87
|
+
location = `${base.protocol}//${base.host}${location}`;
|
|
88
|
+
}
|
|
89
|
+
res.resume();
|
|
90
|
+
request(location);
|
|
91
|
+
return;
|
|
92
|
+
}
|
|
93
|
+
if (res.statusCode !== 200) {
|
|
94
|
+
res.resume();
|
|
95
|
+
return reject(new Error(`HTTP ${res.statusCode} downloading ${url}`));
|
|
96
|
+
}
|
|
97
|
+
const total = parseInt(res.headers['content-length'] ?? '0', 10);
|
|
98
|
+
let received = 0;
|
|
99
|
+
const chunks = [];
|
|
100
|
+
res.on('data', (chunk) => {
|
|
101
|
+
chunks.push(chunk);
|
|
102
|
+
received += chunk.length;
|
|
103
|
+
if (onProgress) {
|
|
104
|
+
onProgress({
|
|
105
|
+
file: fileName ?? destPath,
|
|
106
|
+
received,
|
|
107
|
+
total,
|
|
108
|
+
percent: total > 0 ? Math.round((received / total) * 100) : 0,
|
|
109
|
+
});
|
|
110
|
+
}
|
|
111
|
+
});
|
|
112
|
+
res.on('end', async () => {
|
|
113
|
+
try {
|
|
114
|
+
await (0, promises_1.writeFile)(destPath, Buffer.concat(chunks));
|
|
115
|
+
resolve();
|
|
116
|
+
}
|
|
117
|
+
catch (err) {
|
|
118
|
+
reject(err);
|
|
119
|
+
}
|
|
120
|
+
});
|
|
121
|
+
res.on('error', reject);
|
|
122
|
+
}).on('error', reject);
|
|
123
|
+
};
|
|
124
|
+
request(url);
|
|
125
|
+
});
|
|
126
|
+
}
|
|
127
|
+
/**
|
|
128
|
+
* Build a Hugging Face Hub download URL for a file in a model repo.
|
|
129
|
+
* Format: https://huggingface.co/{model_id}/resolve/main/{filename}
|
|
130
|
+
*/
|
|
131
|
+
function hfFileUrl(modelId, filename) {
|
|
132
|
+
return `https://huggingface.co/${modelId}/resolve/main/${filename}`;
|
|
133
|
+
}
|
|
134
|
+
/**
|
|
135
|
+
* Download model files from Hugging Face Hub.
|
|
136
|
+
* Files: model.onnx (from onnx/ subdirectory), tokenizer.json, config.json
|
|
137
|
+
*/
|
|
138
|
+
async function downloadModel(modelId, onProgress, options) {
|
|
139
|
+
const dir = getModelDir(modelId);
|
|
140
|
+
await (0, promises_1.mkdir)(dir, { recursive: true });
|
|
141
|
+
const onnxSubdir = options?.onnxSubdir ?? 'onnx';
|
|
142
|
+
// Download config.json and tokenizer.json from repo root
|
|
143
|
+
await downloadFile(hfFileUrl(modelId, 'config.json'), (0, node_path_1.join)(dir, 'config.json'), onProgress, 'config.json');
|
|
144
|
+
await downloadFile(hfFileUrl(modelId, 'tokenizer.json'), (0, node_path_1.join)(dir, 'tokenizer.json'), onProgress, 'tokenizer.json');
|
|
145
|
+
// Download model.onnx from onnx subdirectory
|
|
146
|
+
await downloadFile(hfFileUrl(modelId, `${onnxSubdir}/model.onnx`), (0, node_path_1.join)(dir, 'model.onnx'), onProgress, 'model.onnx');
|
|
147
|
+
return dir;
|
|
148
|
+
}
|
|
149
|
+
/**
|
|
150
|
+
* Ensure a model is available locally. Downloads if missing.
|
|
151
|
+
*/
|
|
152
|
+
async function ensureModel(modelId, onProgress, options) {
|
|
153
|
+
if (await isModelReady(modelId)) {
|
|
154
|
+
const dir = getModelDir(modelId);
|
|
155
|
+
options?.logger?.info('Model already downloaded, skipping', { modelId, dir });
|
|
156
|
+
return dir;
|
|
157
|
+
}
|
|
158
|
+
options?.logger?.info('Model not found locally, downloading...', { modelId });
|
|
159
|
+
return downloadModel(modelId, onProgress, { onnxSubdir: options?.onnxSubdir });
|
|
160
|
+
}
|
|
161
|
+
//# sourceMappingURL=model-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-manager.js","sourceRoot":"","sources":["../../src/classifier/model-manager.ts"],"names":[],"mappings":";;;;;AA6BA,kCAGC;AAKD,oCAWC;AAKD,oCAiBC;AAwFD,sCAkCC;AAKD,kCAYC;AAjND,yCAAiC;AACjC,qCAAkC;AAClC,+CAA0D;AAC1D,4DAA+B;AAE/B,MAAM,UAAU,GAAG,IAAA,gBAAI,EAAC,IAAA,iBAAO,GAAE,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;AAEzD,MAAM,cAAc,GAAG,CAAC,YAAY,EAAE,gBAAgB,EAAE,aAAa,CAAC,CAAC;AAkBvE;;;GAGG;AACH,SAAgB,WAAW,CAAC,OAAe;IACzC,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC9C,OAAO,IAAA,gBAAI,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;AACpC,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,YAAY,CAAC,OAAe;IAChD,MAAM,GAAG,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,MAAM,CAAC,GAAG,MAAM,IAAA,eAAI,EAAC,IAAA,gBAAI,EAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;YACtC,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,YAAY,CAAC,OAAe;IAChD,MAAM,GAAG,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACjC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,MAAM,IAAA,eAAI,EAAC,IAAA,gBAAI,EAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnB,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IACD,OAAO;QACL,OAAO;QACP,GAAG;QACH,KAAK,EAAE,KAAK,CAAC,MAAM,KAAK,cAAc,CAAC,MAAM;QAC7C,KAAK;KACN,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,UAAU;IACjB,OAAO,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;AAC1E,CAAC;AAED,SAAS,YAAY,CACnB,GAAW,EACX,QAAgB,EAChB,UAA6B,EAC7B,QAAiB;IAEjB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,OAAO,GAAG,CAAC,SAAiB,EAAE,EAAE;YACpC,MAAM,OAAO,GAA2B,EAAE,YAAY,EAAE,mBAAmB,EAAE,CAAC;YAC9E,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;YAC7B,IAAI,OAAO,IAAI,SAAS,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBACpD,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,OAAO,EAAE,CAAC;YACjD,CAAC;YACD,oBAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE;gBACxC,+CAA+C;gBAC/C,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE,CAAC;oBAC/F,IAAI,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC;oBACpC,IAAI,CAAC,QAAQ;wBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,iCAAiC,GAAG,EAAE,CAAC,CAAC,CAAC;oBAChF,0DAA0D;oBAC1D,IAAI,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;wBAC7B,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC;wBAChC,QAAQ,GAAG,GAAG,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,IAAI,GAAG,QAAQ,EAAE,CAAC;oBACzD,CAAC;oBACD,GAAG,CAAC,MAAM,EAAE,CAAC;oBACb,OAAO,CAAC,QAAQ,CAAC,CAAC;oBAClB,OAAO;gBACT,CAAC;gBAED,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE,CAAC;oBAC3B,GAAG,CAAC,MAAM,EAAE,CAAC;oBACb,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,UAAU,gBAAgB,GAAG,EAAE,CAAC,CAAC,CAAC;gBACxE,CAAC;gBAED,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;gBACjE,IAAI,QAAQ,GAAG,CAAC,CAAC;gBACjB,MAAM,MAAM,GAAa,EAAE,CAAC;gBAE5B,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;oBAC/B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACnB,QAAQ,IAAI,KAAK,CAAC,MAAM,CAAC;oBACzB,IAAI,UAAU,EAAE,CAAC;wBACf,UAAU,CAAC;4BACT,IAAI,EAAE,QAAQ,IAAI,QAAQ;4BAC1B,QAAQ;4BACR,KAAK;4BACL,OAAO,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;yBAC9D,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;oBACvB,IAAI,CAAC;wBACH,MAAM,IAAA,oBAAS,EAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;wBACjD,OAAO,EAAE,CAAC;oBACZ,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,MAAM,CAAC,GAAG,CAAC,CAAC;oBACd,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC1B,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACzB,CAAC,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,SAAS,SAAS,CAAC,OAAe,EAAE,QAAgB;IAClD,OAAO,0BAA0B,OAAO,iBAAiB,QAAQ,EAAE,CAAC;AACtE,CAAC;AAED;;;GAGG;AACI,KAAK,UAAU,aAAa,CACjC,OAAe,EACf,UAA6B,EAC7B,OAAiC;IAEjC,MAAM,GAAG,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACjC,MAAM,IAAA,gBAAK,EAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAEtC,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,MAAM,CAAC;IAEjD,yDAAyD;IACzD,MAAM,YAAY,CAChB,SAAS,CAAC,OAAO,EAAE,aAAa,CAAC,EACjC,IAAA,gBAAI,EAAC,GAAG,EAAE,aAAa,CAAC,EACxB,UAAU,EACV,aAAa,CACd,CAAC;IAEF,MAAM,YAAY,CAChB,SAAS,CAAC,OAAO,EAAE,gBAAgB,CAAC,EACpC,IAAA,gBAAI,EAAC,GAAG,EAAE,gBAAgB,CAAC,EAC3B,UAAU,EACV,gBAAgB,CACjB,CAAC;IAEF,6CAA6C;IAC7C,MAAM,YAAY,CAChB,SAAS,CAAC,OAAO,EAAE,GAAG,UAAU,aAAa,CAAC,EAC9C,IAAA,gBAAI,EAAC,GAAG,EAAE,YAAY,CAAC,EACvB,UAAU,EACV,YAAY,CACb,CAAC;IAEF,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,WAAW,CAC/B,OAAe,EACf,UAA6B,EAC7B,OAA2G;IAE3G,IAAI,MAAM,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;QAChC,MAAM,GAAG,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;QACjC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,oCAAoC,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9E,OAAO,GAAG,CAAC;IACb,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,yCAAyC,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC9E,OAAO,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;AACjF,CAAC"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { ClassifierProvider, ClassificationResult } from '../plugin-api/types.js';
|
|
2
|
+
export interface OnnxProviderConfig {
|
|
3
|
+
modelDir: string;
|
|
4
|
+
maxLength?: number;
|
|
5
|
+
}
|
|
6
|
+
export declare class OnnxClassifierProvider implements ClassifierProvider {
|
|
7
|
+
readonly name = "onnx";
|
|
8
|
+
private config;
|
|
9
|
+
private maxLength;
|
|
10
|
+
private session;
|
|
11
|
+
private tokenizer;
|
|
12
|
+
private labels;
|
|
13
|
+
private _modelName;
|
|
14
|
+
constructor(config: OnnxProviderConfig);
|
|
15
|
+
get modelName(): string;
|
|
16
|
+
get ready(): boolean;
|
|
17
|
+
initialize(): Promise<void>;
|
|
18
|
+
classify(text: string): Promise<ClassificationResult>;
|
|
19
|
+
classifyBatch(texts: string[]): Promise<ClassificationResult[]>;
|
|
20
|
+
destroy(): Promise<void>;
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=onnx-provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"onnx-provider.d.ts","sourceRoot":"","sources":["../../src/classifier/onnx-provider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAEvF,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAaD,qBAAa,sBAAuB,YAAW,kBAAkB;IAC/D,QAAQ,CAAC,IAAI,UAAU;IACvB,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,OAAO,CAA4D;IAC3E,OAAO,CAAC,SAAS,CAA4D;IAC7E,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,UAAU,CAAM;gBAEZ,MAAM,EAAE,kBAAkB;IAKtC,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,IAAI,KAAK,IAAI,OAAO,CAEnB;IAEK,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA0C3B,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,CAAC;IA6DrD,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAK/D,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAO/B"}
|
|
@@ -0,0 +1,131 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.OnnxClassifierProvider = void 0;
|
|
4
|
+
// Lazy-loaded native dependencies (only imported when initialize() is called)
|
|
5
|
+
let ort = null;
|
|
6
|
+
let TokenizerModule = null;
|
|
7
|
+
function softmax(logits) {
|
|
8
|
+
const max = logits.reduce((a, b) => Math.max(a, b), -Infinity);
|
|
9
|
+
const exps = logits.map(v => Math.exp(v - max));
|
|
10
|
+
const sum = exps.reduce((a, b) => a + b, 0);
|
|
11
|
+
return exps.map(v => v / sum);
|
|
12
|
+
}
|
|
13
|
+
class OnnxClassifierProvider {
|
|
14
|
+
name = 'onnx';
|
|
15
|
+
config;
|
|
16
|
+
maxLength;
|
|
17
|
+
session = null;
|
|
18
|
+
tokenizer = null;
|
|
19
|
+
labels = [];
|
|
20
|
+
_modelName = '';
|
|
21
|
+
constructor(config) {
|
|
22
|
+
this.config = config;
|
|
23
|
+
this.maxLength = config.maxLength ?? 512;
|
|
24
|
+
}
|
|
25
|
+
get modelName() {
|
|
26
|
+
return this._modelName;
|
|
27
|
+
}
|
|
28
|
+
get ready() {
|
|
29
|
+
return this.session !== null && this.tokenizer !== null;
|
|
30
|
+
}
|
|
31
|
+
async initialize() {
|
|
32
|
+
const { join } = await import('node:path');
|
|
33
|
+
const { readFile } = await import('node:fs/promises');
|
|
34
|
+
// Lazy-load native modules
|
|
35
|
+
if (!ort)
|
|
36
|
+
ort = await import('onnxruntime-node');
|
|
37
|
+
if (!TokenizerModule)
|
|
38
|
+
TokenizerModule = await import('@huggingface/tokenizers');
|
|
39
|
+
const modelPath = join(this.config.modelDir, 'model.onnx');
|
|
40
|
+
const tokenizerPath = join(this.config.modelDir, 'tokenizer.json');
|
|
41
|
+
const configPath = join(this.config.modelDir, 'config.json');
|
|
42
|
+
// Load model config for label mapping
|
|
43
|
+
let modelConfig = {};
|
|
44
|
+
try {
|
|
45
|
+
const configRaw = await readFile(configPath, 'utf-8');
|
|
46
|
+
modelConfig = JSON.parse(configRaw);
|
|
47
|
+
if (modelConfig.id2label) {
|
|
48
|
+
const id2label = modelConfig.id2label;
|
|
49
|
+
const maxId = Math.max(...Object.keys(id2label).map(Number));
|
|
50
|
+
this.labels = new Array(maxId + 1);
|
|
51
|
+
for (const [id, label] of Object.entries(id2label)) {
|
|
52
|
+
this.labels[Number(id)] = label;
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
this._modelName = (modelConfig._name_or_path ?? modelConfig.model_type ?? 'unknown');
|
|
56
|
+
}
|
|
57
|
+
catch {
|
|
58
|
+
this.labels = ['SAFE', 'INJECTION'];
|
|
59
|
+
this._modelName = 'unknown';
|
|
60
|
+
}
|
|
61
|
+
// Load tokenizer — @huggingface/tokenizers v0.1.x uses new Tokenizer(tokenizerJson, configJson)
|
|
62
|
+
const tokenizerRaw = await readFile(tokenizerPath, 'utf-8');
|
|
63
|
+
const tokenizerJson = JSON.parse(tokenizerRaw);
|
|
64
|
+
this.tokenizer = new TokenizerModule.Tokenizer(tokenizerJson, modelConfig);
|
|
65
|
+
// Create ONNX inference session
|
|
66
|
+
this.session = await ort.InferenceSession.create(modelPath, {
|
|
67
|
+
executionProviders: ['cpu'],
|
|
68
|
+
});
|
|
69
|
+
}
|
|
70
|
+
async classify(text) {
|
|
71
|
+
if (!this.session || !this.tokenizer) {
|
|
72
|
+
throw new Error('OnnxClassifierProvider not initialized — call initialize() first');
|
|
73
|
+
}
|
|
74
|
+
const start = performance.now();
|
|
75
|
+
// Tokenize — @huggingface/tokenizers v0.1.x returns plain object { ids, attention_mask }
|
|
76
|
+
const encoded = this.tokenizer.encode(text);
|
|
77
|
+
const inputIds = encoded.ids;
|
|
78
|
+
const attentionMask = encoded.attention_mask;
|
|
79
|
+
// Truncate to maxLength
|
|
80
|
+
const length = Math.min(inputIds.length, this.maxLength);
|
|
81
|
+
const ids = BigInt64Array.from(inputIds.slice(0, length).map((v) => BigInt(v)));
|
|
82
|
+
const mask = BigInt64Array.from(attentionMask.slice(0, length).map((v) => BigInt(v)));
|
|
83
|
+
// Create tensors
|
|
84
|
+
const inputIdsTensor = new ort.Tensor('int64', ids, [1, length]);
|
|
85
|
+
const attentionMaskTensor = new ort.Tensor('int64', mask, [1, length]);
|
|
86
|
+
// Run inference
|
|
87
|
+
const feeds = {
|
|
88
|
+
input_ids: inputIdsTensor,
|
|
89
|
+
attention_mask: attentionMaskTensor,
|
|
90
|
+
};
|
|
91
|
+
const output = await this.session.run(feeds);
|
|
92
|
+
// Get logits from first output
|
|
93
|
+
const outputName = this.session.outputNames[0];
|
|
94
|
+
const logits = output[outputName].data;
|
|
95
|
+
// Apply softmax
|
|
96
|
+
const probs = softmax(logits);
|
|
97
|
+
// Build label scores
|
|
98
|
+
const labels = [];
|
|
99
|
+
let bestIdx = 0;
|
|
100
|
+
let bestScore = 0;
|
|
101
|
+
for (let i = 0; i < probs.length; i++) {
|
|
102
|
+
const label = this.labels[i] ?? `CLASS_${i}`;
|
|
103
|
+
const score = probs[i];
|
|
104
|
+
labels.push({ label, score });
|
|
105
|
+
if (score > bestScore) {
|
|
106
|
+
bestScore = score;
|
|
107
|
+
bestIdx = i;
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
const latencyMs = Math.round(performance.now() - start);
|
|
111
|
+
return {
|
|
112
|
+
label: this.labels[bestIdx] ?? `CLASS_${bestIdx}`,
|
|
113
|
+
score: bestScore,
|
|
114
|
+
labels,
|
|
115
|
+
latencyMs,
|
|
116
|
+
};
|
|
117
|
+
}
|
|
118
|
+
async classifyBatch(texts) {
|
|
119
|
+
// Simple sequential batch — ONNX batch inference requires padding and is more complex
|
|
120
|
+
return Promise.all(texts.map(text => this.classify(text)));
|
|
121
|
+
}
|
|
122
|
+
async destroy() {
|
|
123
|
+
if (this.session) {
|
|
124
|
+
await this.session.release();
|
|
125
|
+
this.session = null;
|
|
126
|
+
}
|
|
127
|
+
this.tokenizer = null;
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
exports.OnnxClassifierProvider = OnnxClassifierProvider;
|
|
131
|
+
//# sourceMappingURL=onnx-provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"onnx-provider.js","sourceRoot":"","sources":["../../src/classifier/onnx-provider.ts"],"names":[],"mappings":";;;AAOA,8EAA8E;AAC9E,IAAI,GAAG,GAA6C,IAAI,CAAC;AACzD,IAAI,eAAe,GAAoD,IAAI,CAAC;AAE5E,SAAS,OAAO,CAAC,MAAoB;IACnC,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;IAC/D,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IAChD,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,GAAG,CAA4B,CAAC;AAC3D,CAAC;AAED,MAAa,sBAAsB;IACxB,IAAI,GAAG,MAAM,CAAC;IACf,MAAM,CAAqB;IAC3B,SAAS,CAAS;IAClB,OAAO,GAAuD,IAAI,CAAC;IACnE,SAAS,GAAuD,IAAI,CAAC;IACrE,MAAM,GAAa,EAAE,CAAC;IACtB,UAAU,GAAG,EAAE,CAAC;IAExB,YAAY,MAA0B;QACpC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,GAAG,CAAC;IAC3C,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,OAAO,KAAK,IAAI,IAAI,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAEtD,2BAA2B;QAC3B,IAAI,CAAC,GAAG;YAAE,GAAG,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACjD,IAAI,CAAC,eAAe;YAAE,eAAe,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;QAEhF,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAC3D,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAE7D,sCAAsC;QACtC,IAAI,WAAW,GAA4B,EAAE,CAAC;QAC9C,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACtD,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YACpC,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAkC,CAAC;gBAChE,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;gBAC7D,IAAI,CAAC,MAAM,GAAG,IAAI,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;gBACnC,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACnD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC;gBAClC,CAAC;YACH,CAAC;YACD,IAAI,CAAC,UAAU,GAAG,CAAC,WAAW,CAAC,aAAa,IAAI,WAAW,CAAC,UAAU,IAAI,SAAS,CAAW,CAAC;QACjG,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,MAAM,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACpC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC9B,CAAC;QAED,gGAAgG;QAChG,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAC5D,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC/C,IAAI,CAAC,SAAS,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;QAE3E,gCAAgC;QAChC,IAAI,CAAC,OAAO,GAAG,MAAM,GAAG,CAAC,gBAAgB,CAAC,MAAM,CAAC,SAAS,EAAE;YAC1D,kBAAkB,EAAE,CAAC,KAAK,CAAC;SAC5B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAAY;QACzB,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,kEAAkE,CAAC,CAAC;QACtF,CAAC;QAED,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QAEhC,yFAAyF;QACzF,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAgD,CAAC;QAC3F,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC;QAC7B,MAAM,aAAa,GAAG,OAAO,CAAC,cAAc,CAAC;QAE7C,wBAAwB;QACxB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACzD,MAAM,GAAG,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxF,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAE9F,iBAAiB;QACjB,MAAM,cAAc,GAAG,IAAI,GAAI,CAAC,MAAM,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;QAClE,MAAM,mBAAmB,GAAG,IAAI,GAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;QAExE,gBAAgB;QAChB,MAAM,KAAK,GAAsD;YAC/D,SAAS,EAAE,cAAc;YACzB,cAAc,EAAE,mBAAmB;SACpC,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAE7C,+BAA+B;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,IAAoB,CAAC;QAEvD,gBAAgB;QAChB,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAE9B,qBAAqB;QACrB,MAAM,MAAM,GAA4C,EAAE,CAAC;QAC3D,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,EAAE,CAAC;YAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YAC9B,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;gBACtB,SAAS,GAAG,KAAK,CAAC;gBAClB,OAAO,GAAG,CAAC,CAAC;YACd,CAAC;QACH,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;QAExD,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,SAAS,OAAO,EAAE;YACjD,KAAK,EAAE,SAAS;YAChB,MAAM;YACN,SAAS;SACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,KAAe;QACjC,sFAAsF;QACtF,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACtB,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IACxB,CAAC;CACF;AAzID,wDAyIC"}
|
|
@@ -119,12 +119,12 @@ function registerPluginsCommand(program) {
|
|
|
119
119
|
(0, node_fs_1.rmSync)(pluginsDest, { recursive: true, force: true });
|
|
120
120
|
(0, node_fs_1.cpSync)(pluginSource, pluginsDest, { recursive: true, filter: (src) => !src.includes('node_modules') && !src.includes('.git') });
|
|
121
121
|
}
|
|
122
|
-
// Rewrite @
|
|
122
|
+
// Rewrite @aion0/bastion-plugin-api to file: path
|
|
123
123
|
const pkgPath = (0, node_path_1.join)(pluginsDest, 'package.json');
|
|
124
124
|
const pkg = JSON.parse((0, node_fs_1.readFileSync)(pkgPath, 'utf-8'));
|
|
125
|
-
if (pkg.dependencies?.['@
|
|
125
|
+
if (pkg.dependencies?.['@aion0/bastion-plugin-api']) {
|
|
126
126
|
const relPath = (0, node_path_1.relative)(pluginsDest, pluginApiDir);
|
|
127
|
-
pkg.dependencies['@
|
|
127
|
+
pkg.dependencies['@aion0/bastion-plugin-api'] = `file:${relPath}`;
|
|
128
128
|
(0, node_fs_1.writeFileSync)(pkgPath, JSON.stringify(pkg, null, 2) + '\n');
|
|
129
129
|
console.log(`Rewrote plugin-api path to: file:${relPath}`);
|
|
130
130
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"plugins.js","sourceRoot":"","sources":["../../../src/cli/commands/plugins.ts"],"names":[],"mappings":";;;;;AA0DA,wDA0JC;AAnND,qCAAoH;AACpH,yCAAoD;AACpD,qCAAkC;AAClC,2DAA8C;AAC9C,iDAAgD;AAChD,sDAA2B;AAE3B,MAAM,WAAW,GAAG,IAAA,gBAAI,EAAC,IAAA,iBAAO,GAAE,EAAE,UAAU,CAAC,CAAC;AAChD,MAAM,WAAW,GAAG,IAAA,gBAAI,EAAC,WAAW,EAAE,aAAa,CAAC,CAAC;AACrD,MAAM,UAAU,GAAG,IAAA,gBAAI,EAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;AAE/C,SAAS,GAAG,CAAC,QAAgB;IAC3B,MAAM,EAAE,GAAG,IAAA,+BAAe,EAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7E,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,OAAO,CAAC,GAAW;IAC1B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,CAAC;QACH,KAAK,MAAM,KAAK,IAAI,IAAA,qBAAW,EAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1D,IAAI,CAAC;gBACH,MAAM,EAAE,GAAG,IAAA,kBAAQ,EAAC,IAAA,gBAAI,EAAC,GAAG,EAAE,KAAe,CAAC,CAAC,CAAC;gBAChD,IAAI,EAAE,CAAC,MAAM,EAAE;oBAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC;YACpC,CAAC;YAAC,MAAM,CAAC,CAAC,UAAU,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QAAC,OAAO,MAAM,CAAC;IAAC,CAAC;IAC1B,IAAI,KAAK,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI;QAAE,OAAO,GAAG,CAAC,KAAK,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;IACvF,OAAO,GAAG,CAAC,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;AAClD,CAAC;AAQD,SAAS,UAAU;IACjB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAA,sBAAY,EAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAC/C,OAAQ,iBAAI,CAAC,IAAI,CAAC,GAAG,CAA6B,IAAI,EAAE,CAAC;IAC3D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,MAA+B;IACjD,IAAA,mBAAS,EAAC,IAAA,mBAAO,EAAC,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACrD,IAAA,uBAAa,EAAC,WAAW,EAAE,iBAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,kBAAkB,CAAC,MAA+B;IACzD,MAAM,OAAO,GAAG,MAAM,CAAC,OAA8C,CAAC;IACtE,OAAQ,OAAO,EAAE,QAAwC,IAAI,EAAE,CAAC;AAClE,CAAC;AAED,SAAgB,sBAAsB,CAAC,OAAgB;IACrD,MAAM,GAAG,GAAG,OAAO;SAChB,OAAO,CAAC,SAAS,CAAC;SAClB,WAAW,CAAC,yBAAyB,CAAC,CAAC;IAE1C,GAAG;SACA,OAAO,CAAC,MAAM,CAAC;SACf,WAAW,CAAC,iCAAiC,CAAC;SAC9C,MAAM,CAAC,GAAG,EAAE;QACX,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;YAC9C,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;YAC3C,OAAO,CAAC,GAAG,CAAC,yEAAyE,CAAC,CAAC;YACvF,OAAO;QACT,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QACnC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,yBAAyB,CAAC;YACpF,MAAM,MAAM,GAAG,IAAA,oBAAU,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACzC,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,qCAAqC,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,OAAO,KAAK,MAAM,IAAI,GAAG,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,GAAG;SACA,OAAO,CAAC,SAAS,CAAC;SAClB,WAAW,CAAC,8BAA8B,CAAC;SAC3C,QAAQ,CAAC,QAAQ,EAAE,iCAAiC,CAAC;SACrD,MAAM,CAAC,CAAC,UAAmB,EAAE,EAAE;QAC9B,kCAAkC;QAClC,IAAI,YAAY,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,4CAA4C;YAC5C,MAAM,SAAS,GAAG,IAAA,gBAAI,EAAC,OAAO,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,oBAAoB,CAAC,CAAC;YAClE,IAAI,IAAA,oBAAU,EAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,cAAc,CAAC,CAAC,EAAE,CAAC;gBAChD,YAAY,GAAG,SAAS,CAAC;YAC3B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,YAAY,IAAI,CAAC,IAAA,oBAAU,EAAC,IAAA,gBAAI,EAAC,YAAY,EAAE,cAAc,CAAC,CAAC,EAAE,CAAC;YACrE,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;YACtD,OAAO,CAAC,KAAK,CAAC,6DAA6D,CAAC,CAAC;YAC7E,OAAO,CAAC,KAAK,CAAC,+EAA+E,CAAC,CAAC;YAC/F,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,qDAAqD;QACrD,MAAM,MAAM,GAAG,IAAA,gBAAI,EAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,IAAA,oBAAU,EAAC,IAAA,gBAAI,EAAC,MAAM,EAAE,cAAc,CAAC,CAAC,EAAE,CAAC;YAC9C,OAAO,CAAC,KAAK,CAAC,4BAA4B,MAAM,yBAAyB,CAAC,CAAC;YAC3E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,WAAW,GAAG,IAAA,gBAAI,EAAC,MAAM,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAC3D,MAAM,YAAY,GAAG,IAAA,gBAAI,EAAC,MAAM,EAAE,UAAU,EAAE,oBAAoB,CAAC,CAAC;QAEpE,OAAO,CAAC,GAAG,CAAC,qCAAqC,YAAY,EAAE,CAAC,CAAC;QAEjE,iDAAiD;QACjD,IAAA,mBAAS,EAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,6CAA6C;QAC7C,IAAI,CAAC;YACH,IAAA,6BAAQ,EAAC,kEAAkE,YAAY,OAAO,WAAW,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QACpI,CAAC;QAAC,MAAM,CAAC;YACP,4BAA4B;YAC5B,IAAA,gBAAM,EAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACtD,IAAA,gBAAM,EAAC,YAAY,EAAE,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAClI,CAAC;QAED,
|
|
1
|
+
{"version":3,"file":"plugins.js","sourceRoot":"","sources":["../../../src/cli/commands/plugins.ts"],"names":[],"mappings":";;;;;AA0DA,wDA0JC;AAnND,qCAAoH;AACpH,yCAAoD;AACpD,qCAAkC;AAClC,2DAA8C;AAC9C,iDAAgD;AAChD,sDAA2B;AAE3B,MAAM,WAAW,GAAG,IAAA,gBAAI,EAAC,IAAA,iBAAO,GAAE,EAAE,UAAU,CAAC,CAAC;AAChD,MAAM,WAAW,GAAG,IAAA,gBAAI,EAAC,WAAW,EAAE,aAAa,CAAC,CAAC;AACrD,MAAM,UAAU,GAAG,IAAA,gBAAI,EAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;AAE/C,SAAS,GAAG,CAAC,QAAgB;IAC3B,MAAM,EAAE,GAAG,IAAA,+BAAe,EAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7E,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,OAAO,CAAC,GAAW;IAC1B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,CAAC;QACH,KAAK,MAAM,KAAK,IAAI,IAAA,qBAAW,EAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1D,IAAI,CAAC;gBACH,MAAM,EAAE,GAAG,IAAA,kBAAQ,EAAC,IAAA,gBAAI,EAAC,GAAG,EAAE,KAAe,CAAC,CAAC,CAAC;gBAChD,IAAI,EAAE,CAAC,MAAM,EAAE;oBAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC;YACpC,CAAC;YAAC,MAAM,CAAC,CAAC,UAAU,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QAAC,OAAO,MAAM,CAAC;IAAC,CAAC;IAC1B,IAAI,KAAK,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI;QAAE,OAAO,GAAG,CAAC,KAAK,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;IACvF,OAAO,GAAG,CAAC,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;AAClD,CAAC;AAQD,SAAS,UAAU;IACjB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAA,sBAAY,EAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAC/C,OAAQ,iBAAI,CAAC,IAAI,CAAC,GAAG,CAA6B,IAAI,EAAE,CAAC;IAC3D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,MAA+B;IACjD,IAAA,mBAAS,EAAC,IAAA,mBAAO,EAAC,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACrD,IAAA,uBAAa,EAAC,WAAW,EAAE,iBAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,kBAAkB,CAAC,MAA+B;IACzD,MAAM,OAAO,GAAG,MAAM,CAAC,OAA8C,CAAC;IACtE,OAAQ,OAAO,EAAE,QAAwC,IAAI,EAAE,CAAC;AAClE,CAAC;AAED,SAAgB,sBAAsB,CAAC,OAAgB;IACrD,MAAM,GAAG,GAAG,OAAO;SAChB,OAAO,CAAC,SAAS,CAAC;SAClB,WAAW,CAAC,yBAAyB,CAAC,CAAC;IAE1C,GAAG;SACA,OAAO,CAAC,MAAM,CAAC;SACf,WAAW,CAAC,iCAAiC,CAAC;SAC9C,MAAM,CAAC,GAAG,EAAE;QACX,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;YAC9C,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;YAC3C,OAAO,CAAC,GAAG,CAAC,yEAAyE,CAAC,CAAC;YACvF,OAAO;QACT,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QACnC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,yBAAyB,CAAC;YACpF,MAAM,MAAM,GAAG,IAAA,oBAAU,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACzC,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,qCAAqC,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,OAAO,KAAK,MAAM,IAAI,GAAG,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,GAAG;SACA,OAAO,CAAC,SAAS,CAAC;SAClB,WAAW,CAAC,8BAA8B,CAAC;SAC3C,QAAQ,CAAC,QAAQ,EAAE,iCAAiC,CAAC;SACrD,MAAM,CAAC,CAAC,UAAmB,EAAE,EAAE;QAC9B,kCAAkC;QAClC,IAAI,YAAY,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,4CAA4C;YAC5C,MAAM,SAAS,GAAG,IAAA,gBAAI,EAAC,OAAO,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,oBAAoB,CAAC,CAAC;YAClE,IAAI,IAAA,oBAAU,EAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,cAAc,CAAC,CAAC,EAAE,CAAC;gBAChD,YAAY,GAAG,SAAS,CAAC;YAC3B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,YAAY,IAAI,CAAC,IAAA,oBAAU,EAAC,IAAA,gBAAI,EAAC,YAAY,EAAE,cAAc,CAAC,CAAC,EAAE,CAAC;YACrE,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;YACtD,OAAO,CAAC,KAAK,CAAC,6DAA6D,CAAC,CAAC;YAC7E,OAAO,CAAC,KAAK,CAAC,+EAA+E,CAAC,CAAC;YAC/F,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,qDAAqD;QACrD,MAAM,MAAM,GAAG,IAAA,gBAAI,EAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,IAAA,oBAAU,EAAC,IAAA,gBAAI,EAAC,MAAM,EAAE,cAAc,CAAC,CAAC,EAAE,CAAC;YAC9C,OAAO,CAAC,KAAK,CAAC,4BAA4B,MAAM,yBAAyB,CAAC,CAAC;YAC3E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,WAAW,GAAG,IAAA,gBAAI,EAAC,MAAM,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAC3D,MAAM,YAAY,GAAG,IAAA,gBAAI,EAAC,MAAM,EAAE,UAAU,EAAE,oBAAoB,CAAC,CAAC;QAEpE,OAAO,CAAC,GAAG,CAAC,qCAAqC,YAAY,EAAE,CAAC,CAAC;QAEjE,iDAAiD;QACjD,IAAA,mBAAS,EAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,6CAA6C;QAC7C,IAAI,CAAC;YACH,IAAA,6BAAQ,EAAC,kEAAkE,YAAY,OAAO,WAAW,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QACpI,CAAC;QAAC,MAAM,CAAC;YACP,4BAA4B;YAC5B,IAAA,gBAAM,EAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACtD,IAAA,gBAAM,EAAC,YAAY,EAAE,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAClI,CAAC;QAED,kDAAkD;QAClD,MAAM,OAAO,GAAG,IAAA,gBAAI,EAAC,WAAW,EAAE,cAAc,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAA,sBAAY,EAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QACvD,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC,2BAA2B,CAAC,EAAE,CAAC;YACpD,MAAM,OAAO,GAAG,IAAA,oBAAQ,EAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YACpD,GAAG,CAAC,YAAY,CAAC,2BAA2B,CAAC,GAAG,QAAQ,OAAO,EAAE,CAAC;YAClE,IAAA,uBAAa,EAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;YAC5D,OAAO,CAAC,GAAG,CAAC,oCAAoC,OAAO,EAAE,CAAC,CAAC;QAC7D,CAAC;QAED,sBAAsB;QACtB,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC1C,IAAA,6BAAQ,EAAC,aAAa,EAAE,EAAE,GAAG,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7D,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC3B,IAAA,6BAAQ,EAAC,eAAe,EAAE,EAAE,GAAG,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAE/D,qBAAqB;QACrB,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAA4B,CAAC;QAClE,MAAM,OAAO,GAAI,OAAO,CAAC,QAAwC,IAAI,EAAE,CAAC;QACxE,MAAM,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;QAC/E,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACtD,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC;YAC3B,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC;YACzB,UAAU,CAAC,MAAM,CAAC,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QACtC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEL,GAAG;SACA,OAAO,CAAC,WAAW,CAAC;SACpB,WAAW,CAAC,8BAA8B,CAAC;SAC3C,QAAQ,CAAC,QAAQ,EAAE,8CAA8C,EAAE,aAAa,CAAC;SACjF,MAAM,CAAC,OAAO,EAAE,+CAA+C,CAAC;SAChE,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,IAAuB,EAAE,EAAE;QACtD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAE3C,sBAAsB;QACtB,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1D,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,gCAAgC,IAAI,oBAAoB,CAAC,CAAC;YACxE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,0BAA0B;QAC1B,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC;QAChC,IAAI,IAAA,oBAAU,EAAC,SAAS,CAAC,EAAE,CAAC;YAC1B,IAAA,gBAAM,EAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,EAAE,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,0CAA0C,SAAS,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,gBAAgB;QAChB,MAAM,OAAO,GAAG,MAAM,CAAC,OAAkC,CAAC;QAC1D,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC;QACpD,UAAU,CAAC,MAAM,CAAC,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QAEnC,wBAAwB;QACxB,MAAM,SAAS,GAAG,IAAA,oBAAU,EAAC,UAAU,CAAC,IAAI,IAAA,qBAAW,EAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAC/G,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,YAAY,GAAG,IAAI,CAAC,GAAG,IAAI,KAAK,CAAC;YACrC,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,MAAM,IAAI,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;gBACjC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,qCAAqC,IAAI,OAAO,UAAU,WAAW,CAAC,CAAC;gBAChG,YAAY,GAAG,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,KAAK,CAAC;YACpD,CAAC;YACD,IAAI,YAAY,EAAE,CAAC;gBACjB,IAAA,gBAAM,EAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACrD,OAAO,CAAC,GAAG,CAAC,YAAY,UAAU,EAAE,CAAC,CAAC;YACxC,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,mBAAmB,UAAU,EAAE,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;AACP,CAAC"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { Plugin } from '../types.js';
|
|
2
|
+
import type { PluginEventBus } from '../event-bus.js';
|
|
3
|
+
import type { ClassifierProvider } from '../../plugin-api/types.js';
|
|
4
|
+
import type Database from 'better-sqlite3';
|
|
5
|
+
export interface PiClassifierPluginConfig {
|
|
6
|
+
enabled?: boolean;
|
|
7
|
+
modelId?: string;
|
|
8
|
+
onnxSubdir?: string;
|
|
9
|
+
threshold?: number;
|
|
10
|
+
action?: 'warn' | 'block';
|
|
11
|
+
scanSystem?: boolean;
|
|
12
|
+
}
|
|
13
|
+
/**
|
|
14
|
+
* Extract user message texts from a parsed request body.
|
|
15
|
+
* Covers Anthropic and OpenAI message formats, including tool_result content blocks.
|
|
16
|
+
*/
|
|
17
|
+
export declare function extractUserTexts(parsedBody: Readonly<Record<string, unknown>>, scanSystem?: boolean): string[];
|
|
18
|
+
export declare function createPiClassifierPlugin(db: Database.Database, config: PiClassifierPluginConfig, eventBus: PluginEventBus): {
|
|
19
|
+
plugin: Plugin;
|
|
20
|
+
getClassifierProvider: () => ClassifierProvider | undefined;
|
|
21
|
+
};
|
|
22
|
+
//# sourceMappingURL=pi-classifier.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pi-classifier.d.ts","sourceRoot":"","sources":["../../../src/plugins/builtin/pi-classifier.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAuC,MAAM,aAAa,CAAC;AAC/E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAKpE,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAY3C,MAAM,WAAW,wBAAwB;IACvC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;IAC1B,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,UAAU,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,OAAO,GAAG,MAAM,EAAE,CAmC9G;AAED,wBAAgB,wBAAwB,CACtC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,MAAM,EAAE,wBAAwB,EAChC,QAAQ,EAAE,cAAc,GACvB;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,qBAAqB,EAAE,MAAM,kBAAkB,GAAG,SAAS,CAAA;CAAE,CAmIjF"}
|
|
@@ -0,0 +1,177 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.extractUserTexts = extractUserTexts;
|
|
4
|
+
exports.createPiClassifierPlugin = createPiClassifierPlugin;
|
|
5
|
+
const onnx_provider_js_1 = require("../../classifier/onnx-provider.js");
|
|
6
|
+
const model_manager_js_1 = require("../../classifier/model-manager.js");
|
|
7
|
+
const plugin_events_js_1 = require("../../storage/repositories/plugin-events.js");
|
|
8
|
+
const logger_js_1 = require("../../utils/logger.js");
|
|
9
|
+
const log = (0, logger_js_1.createLogger)('pi-classifier');
|
|
10
|
+
const DEFAULT_MODEL_ID = 'protectai/deberta-v3-base-prompt-injection-v2';
|
|
11
|
+
const DEFAULT_THRESHOLD = 0.8;
|
|
12
|
+
// Labels considered safe/benign across different models
|
|
13
|
+
const SAFE_LABELS = new Set(['BENIGN', 'SAFE']);
|
|
14
|
+
// Labels considered high severity (not critical)
|
|
15
|
+
const HIGH_SEVERITY_LABELS = new Set(['JAILBREAK']);
|
|
16
|
+
/**
|
|
17
|
+
* Extract user message texts from a parsed request body.
|
|
18
|
+
* Covers Anthropic and OpenAI message formats, including tool_result content blocks.
|
|
19
|
+
*/
|
|
20
|
+
function extractUserTexts(parsedBody, scanSystem) {
|
|
21
|
+
const texts = [];
|
|
22
|
+
const messages = parsedBody.messages;
|
|
23
|
+
if (!Array.isArray(messages))
|
|
24
|
+
return texts;
|
|
25
|
+
for (const msg of messages) {
|
|
26
|
+
const role = msg?.role;
|
|
27
|
+
if (role !== 'user' && !(scanSystem && role === 'system'))
|
|
28
|
+
continue;
|
|
29
|
+
const content = msg?.content;
|
|
30
|
+
if (typeof content === 'string') {
|
|
31
|
+
texts.push(content);
|
|
32
|
+
}
|
|
33
|
+
else if (Array.isArray(content)) {
|
|
34
|
+
for (const block of content) {
|
|
35
|
+
if (typeof block === 'string') {
|
|
36
|
+
texts.push(block);
|
|
37
|
+
}
|
|
38
|
+
else if (block?.type === 'text' && typeof block.text === 'string') {
|
|
39
|
+
texts.push(block.text);
|
|
40
|
+
}
|
|
41
|
+
else if (block?.type === 'tool_result') {
|
|
42
|
+
// Anthropic tool_result: content can be string or content blocks
|
|
43
|
+
const trContent = block.content;
|
|
44
|
+
if (typeof trContent === 'string') {
|
|
45
|
+
texts.push(trContent);
|
|
46
|
+
}
|
|
47
|
+
else if (Array.isArray(trContent)) {
|
|
48
|
+
for (const inner of trContent) {
|
|
49
|
+
if (typeof inner === 'string')
|
|
50
|
+
texts.push(inner);
|
|
51
|
+
else if (inner?.type === 'text' && typeof inner.text === 'string')
|
|
52
|
+
texts.push(inner.text);
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
return texts;
|
|
60
|
+
}
|
|
61
|
+
function createPiClassifierPlugin(db, config, eventBus) {
|
|
62
|
+
const modelId = config.modelId ?? DEFAULT_MODEL_ID;
|
|
63
|
+
const threshold = config.threshold ?? DEFAULT_THRESHOLD;
|
|
64
|
+
const action = config.action ?? 'warn';
|
|
65
|
+
const scanSystem = config.scanSystem ?? false;
|
|
66
|
+
const onnxSubdir = config.onnxSubdir ?? 'onnx';
|
|
67
|
+
let provider = null;
|
|
68
|
+
let initialized = false;
|
|
69
|
+
const pluginEventsRepo = new plugin_events_js_1.PluginEventsRepository(db);
|
|
70
|
+
async function initProvider() {
|
|
71
|
+
if (initialized)
|
|
72
|
+
return;
|
|
73
|
+
initialized = true;
|
|
74
|
+
log.info('Initializing PI classifier', { modelId, threshold, action });
|
|
75
|
+
try {
|
|
76
|
+
// Ensure model is downloaded
|
|
77
|
+
let lastLoggedPercent = -10;
|
|
78
|
+
const modelDir = await (0, model_manager_js_1.ensureModel)(modelId, (progress) => {
|
|
79
|
+
if (progress.percent >= lastLoggedPercent + 10) {
|
|
80
|
+
lastLoggedPercent = progress.percent;
|
|
81
|
+
log.info('Downloading model', {
|
|
82
|
+
file: progress.file,
|
|
83
|
+
percent: `${progress.percent}%`,
|
|
84
|
+
});
|
|
85
|
+
}
|
|
86
|
+
}, { onnxSubdir, logger: log });
|
|
87
|
+
// Initialize ONNX provider
|
|
88
|
+
const onnxProvider = new onnx_provider_js_1.OnnxClassifierProvider({
|
|
89
|
+
modelDir,
|
|
90
|
+
maxLength: 512,
|
|
91
|
+
});
|
|
92
|
+
await onnxProvider.initialize();
|
|
93
|
+
provider = onnxProvider;
|
|
94
|
+
log.info('PI classifier ready', {
|
|
95
|
+
modelName: provider.modelName,
|
|
96
|
+
modelDir,
|
|
97
|
+
});
|
|
98
|
+
}
|
|
99
|
+
catch (err) {
|
|
100
|
+
log.error('Failed to initialize PI classifier', {
|
|
101
|
+
error: err.message,
|
|
102
|
+
});
|
|
103
|
+
// Non-fatal: plugin will be skipped in onRequest
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
// Start initialization eagerly (non-blocking)
|
|
107
|
+
if (config.enabled !== false) {
|
|
108
|
+
initProvider().catch(() => { });
|
|
109
|
+
}
|
|
110
|
+
const plugin = {
|
|
111
|
+
name: 'pi-classifier',
|
|
112
|
+
priority: 3,
|
|
113
|
+
version: '1.0.0',
|
|
114
|
+
apiVersion: 2,
|
|
115
|
+
source: 'builtin',
|
|
116
|
+
async onRequest(context) {
|
|
117
|
+
if (!provider?.ready)
|
|
118
|
+
return;
|
|
119
|
+
const texts = extractUserTexts(context.parsedBody, scanSystem);
|
|
120
|
+
if (texts.length === 0)
|
|
121
|
+
return;
|
|
122
|
+
let maxScore = 0;
|
|
123
|
+
let maxLabel = 'SAFE';
|
|
124
|
+
// Classify texts — use batch if available, else sequential
|
|
125
|
+
const results = provider.classifyBatch
|
|
126
|
+
? await provider.classifyBatch(texts)
|
|
127
|
+
: await Promise.all(texts.map(t => provider.classify(t)));
|
|
128
|
+
let detections = 0;
|
|
129
|
+
for (let i = 0; i < results.length; i++) {
|
|
130
|
+
const result = results[i];
|
|
131
|
+
if (!SAFE_LABELS.has(result.label) && result.score >= threshold) {
|
|
132
|
+
detections++;
|
|
133
|
+
// Persist event to DB
|
|
134
|
+
try {
|
|
135
|
+
pluginEventsRepo.insertEvent('pi-classifier', context.id, {
|
|
136
|
+
type: 'prompt-injection',
|
|
137
|
+
severity: HIGH_SEVERITY_LABELS.has(result.label) ? 'high' : 'critical',
|
|
138
|
+
rule: `pi:${result.label.toLowerCase()}`,
|
|
139
|
+
detail: `ML detected ${result.label} (score: ${result.score.toFixed(3)}, latency: ${result.latencyMs}ms)`,
|
|
140
|
+
matchedText: texts[i].slice(0, 200),
|
|
141
|
+
});
|
|
142
|
+
}
|
|
143
|
+
catch (err) {
|
|
144
|
+
log.warn('Failed to persist PI event', { error: err.message });
|
|
145
|
+
}
|
|
146
|
+
if (result.score > maxScore) {
|
|
147
|
+
maxScore = result.score;
|
|
148
|
+
maxLabel = result.label;
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
if (detections === 0)
|
|
153
|
+
return;
|
|
154
|
+
// Emit pi:detected event for threat-scorer escalation
|
|
155
|
+
eventBus.emit('pi:detected', {
|
|
156
|
+
sessionId: context.sessionId,
|
|
157
|
+
label: maxLabel,
|
|
158
|
+
score: maxScore,
|
|
159
|
+
detections,
|
|
160
|
+
});
|
|
161
|
+
if (action === 'block') {
|
|
162
|
+
return {
|
|
163
|
+
blocked: {
|
|
164
|
+
reason: `PI classifier detected ${maxLabel} (score: ${maxScore.toFixed(3)})`,
|
|
165
|
+
},
|
|
166
|
+
};
|
|
167
|
+
}
|
|
168
|
+
// 'warn' — log but don't block
|
|
169
|
+
return;
|
|
170
|
+
},
|
|
171
|
+
};
|
|
172
|
+
return {
|
|
173
|
+
plugin,
|
|
174
|
+
getClassifierProvider: () => provider ?? undefined,
|
|
175
|
+
};
|
|
176
|
+
}
|
|
177
|
+
//# sourceMappingURL=pi-classifier.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pi-classifier.js","sourceRoot":"","sources":["../../../src/plugins/builtin/pi-classifier.ts"],"names":[],"mappings":";;AAgCA,4CAmCC;AAED,4DAuIC;AAzMD,wEAA2E;AAC3E,wEAAgE;AAChE,kFAAqF;AACrF,qDAAqD;AAGrD,MAAM,GAAG,GAAG,IAAA,wBAAY,EAAC,eAAe,CAAC,CAAC;AAE1C,MAAM,gBAAgB,GAAG,+CAA+C,CAAC;AACzE,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAE9B,wDAAwD;AACxD,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;AAChD,iDAAiD;AACjD,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;AAWpD;;;GAGG;AACH,SAAgB,gBAAgB,CAAC,UAA6C,EAAE,UAAoB;IAClG,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;IACrC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;QAAE,OAAO,KAAK,CAAC;IAE3C,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,GAAG,EAAE,IAA0B,CAAC;QAC7C,IAAI,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,UAAU,IAAI,IAAI,KAAK,QAAQ,CAAC;YAAE,SAAS;QAEpE,MAAM,OAAO,GAAG,GAAG,EAAE,OAAO,CAAC;QAC7B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAClC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAC9B,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACpB,CAAC;qBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACpE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACzB,CAAC;qBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;oBACzC,iEAAiE;oBACjE,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC;oBAChC,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;wBAClC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBACxB,CAAC;yBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;wBACpC,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;4BAC9B,IAAI,OAAO,KAAK,KAAK,QAAQ;gCAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iCAC5C,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;gCAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC5F,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAgB,wBAAwB,CACtC,EAAqB,EACrB,MAAgC,EAChC,QAAwB;IAExB,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,gBAAgB,CAAC;IACnD,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,iBAAiB,CAAC;IACxD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC;IACvC,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC;IAC9C,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC;IAE/C,IAAI,QAAQ,GAA8B,IAAI,CAAC;IAC/C,IAAI,WAAW,GAAG,KAAK,CAAC;IAExB,MAAM,gBAAgB,GAAG,IAAI,yCAAsB,CAAC,EAAE,CAAC,CAAC;IAExD,KAAK,UAAU,YAAY;QACzB,IAAI,WAAW;YAAE,OAAO;QACxB,WAAW,GAAG,IAAI,CAAC;QAEnB,GAAG,CAAC,IAAI,CAAC,4BAA4B,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;QAEvE,IAAI,CAAC;YACH,6BAA6B;YAC7B,IAAI,iBAAiB,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,MAAM,IAAA,8BAAW,EAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,EAAE;gBACvD,IAAI,QAAQ,CAAC,OAAO,IAAI,iBAAiB,GAAG,EAAE,EAAE,CAAC;oBAC/C,iBAAiB,GAAG,QAAQ,CAAC,OAAO,CAAC;oBACrC,GAAG,CAAC,IAAI,CAAC,mBAAmB,EAAE;wBAC5B,IAAI,EAAE,QAAQ,CAAC,IAAI;wBACnB,OAAO,EAAE,GAAG,QAAQ,CAAC,OAAO,GAAG;qBAChC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;YAEhC,2BAA2B;YAC3B,MAAM,YAAY,GAAG,IAAI,yCAAsB,CAAC;gBAC9C,QAAQ;gBACR,SAAS,EAAE,GAAG;aACf,CAAC,CAAC;YACH,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAChC,QAAQ,GAAG,YAAY,CAAC;YAExB,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE;gBAC9B,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,KAAK,CAAC,oCAAoC,EAAE;gBAC9C,KAAK,EAAG,GAAa,CAAC,OAAO;aAC9B,CAAC,CAAC;YACH,iDAAiD;QACnD,CAAC;IACH,CAAC;IAED,8CAA8C;IAC9C,IAAI,MAAM,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;QAC7B,YAAY,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACjC,CAAC;IAED,MAAM,MAAM,GAAW;QACrB,IAAI,EAAE,eAAe;QACrB,QAAQ,EAAE,CAAC;QACX,OAAO,EAAE,OAAO;QAChB,UAAU,EAAE,CAAC;QACb,MAAM,EAAE,SAAS;QAEjB,KAAK,CAAC,SAAS,CAAC,OAAuB;YACrC,IAAI,CAAC,QAAQ,EAAE,KAAK;gBAAE,OAAO;YAE7B,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YAC/D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO;YAE/B,IAAI,QAAQ,GAAG,CAAC,CAAC;YACjB,IAAI,QAAQ,GAAG,MAAM,CAAC;YAEtB,2DAA2D;YAC3D,MAAM,OAAO,GAAG,QAAQ,CAAC,aAAa;gBACpC,CAAC,CAAC,MAAM,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC;gBACrC,CAAC,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,QAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAE7D,IAAI,UAAU,GAAG,CAAC,CAAC;YAEnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;gBAC1B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,IAAI,SAAS,EAAE,CAAC;oBAChE,UAAU,EAAE,CAAC;oBAEb,sBAAsB;oBACtB,IAAI,CAAC;wBACH,gBAAgB,CAAC,WAAW,CAAC,eAAe,EAAE,OAAO,CAAC,EAAE,EAAE;4BACxD,IAAI,EAAE,kBAAkB;4BACxB,QAAQ,EAAE,oBAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU;4BACtE,IAAI,EAAE,MAAM,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE;4BACxC,MAAM,EAAE,eAAe,MAAM,CAAC,KAAK,YAAY,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,MAAM,CAAC,SAAS,KAAK;4BACzG,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;yBACpC,CAAC,CAAC;oBACL,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,GAAG,CAAC,IAAI,CAAC,4BAA4B,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC5E,CAAC;oBAED,IAAI,MAAM,CAAC,KAAK,GAAG,QAAQ,EAAE,CAAC;wBAC5B,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC;wBACxB,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC;oBAC1B,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,UAAU,KAAK,CAAC;gBAAE,OAAO;YAE7B,sDAAsD;YACtD,QAAQ,CAAC,IAAI,CAAC,aAAa,EAAE;gBAC3B,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,KAAK,EAAE,QAAQ;gBACf,KAAK,EAAE,QAAQ;gBACf,UAAU;aACX,CAAC,CAAC;YAEH,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE;wBACP,MAAM,EAAE,0BAA0B,QAAQ,YAAY,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;qBAC7E;iBACF,CAAC;YACJ,CAAC;YAED,+BAA+B;YAC/B,OAAO;QACT,CAAC;KACF,CAAC;IAEF,OAAO;QACL,MAAM;QACN,qBAAqB,EAAE,GAAG,EAAE,CAAC,QAAQ,IAAI,SAAS;KACnD,CAAC;AACJ,CAAC"}
|