@jonathanyee/devtunnel-ctl 0.0.0-stage → 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +328 -2
- package/dist/auth.d.ts +24 -0
- package/dist/auth.js +151 -0
- package/dist/auth.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.js +246 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.d.ts +22 -0
- package/dist/config.js +94 -0
- package/dist/config.js.map +1 -0
- package/dist/core.d.ts +16 -0
- package/dist/core.js +114 -0
- package/dist/core.js.map +1 -0
- package/dist/daemon.d.ts +31 -0
- package/dist/daemon.js +153 -0
- package/dist/daemon.js.map +1 -0
- package/dist/forwarding.d.ts +15 -0
- package/dist/forwarding.js +87 -0
- package/dist/forwarding.js.map +1 -0
- package/dist/host-process.d.ts +4 -0
- package/dist/host-process.js +110 -0
- package/dist/host-process.js.map +1 -0
- package/dist/host-sdk.d.ts +38 -0
- package/dist/host-sdk.js +327 -0
- package/dist/host-sdk.js.map +1 -0
- package/dist/host-service.d.ts +21 -0
- package/dist/host-service.js +193 -0
- package/dist/host-service.js.map +1 -0
- package/dist/host-worker.d.ts +1 -0
- package/dist/host-worker.js +80 -0
- package/dist/host-worker.js.map +1 -0
- package/dist/host.d.ts +43 -0
- package/dist/host.js +158 -0
- package/dist/host.js.map +1 -0
- package/dist/launchd.d.ts +15 -0
- package/dist/launchd.js +135 -0
- package/dist/launchd.js.map +1 -0
- package/dist/lock.d.ts +10 -0
- package/dist/lock.js +110 -0
- package/dist/lock.js.map +1 -0
- package/dist/log.d.ts +26 -0
- package/dist/log.js +67 -0
- package/dist/log.js.map +1 -0
- package/dist/main.d.ts +2 -0
- package/dist/main.js +5 -0
- package/dist/main.js.map +1 -0
- package/dist/paths.d.ts +4 -0
- package/dist/paths.js +39 -0
- package/dist/paths.js.map +1 -0
- package/dist/probe.d.ts +1 -0
- package/dist/probe.js +45 -0
- package/dist/probe.js.map +1 -0
- package/dist/recovery.d.ts +77 -0
- package/dist/recovery.js +265 -0
- package/dist/recovery.js.map +1 -0
- package/dist/rpc.d.ts +30 -0
- package/dist/rpc.js +208 -0
- package/dist/rpc.js.map +1 -0
- package/dist/sdk-transport.d.ts +33 -0
- package/dist/sdk-transport.js +239 -0
- package/dist/sdk-transport.js.map +1 -0
- package/npm-shrinkwrap.json +810 -0
- package/package.json +31 -4
package/dist/daemon.js
ADDED
|
@@ -0,0 +1,153 @@
|
|
|
1
|
+
import { join } from 'node:path';
|
|
2
|
+
import { randomUUID } from 'node:crypto';
|
|
3
|
+
import { Store, validateConfig } from "./config.js";
|
|
4
|
+
import { TokenProvider } from "./auth.js";
|
|
5
|
+
import { SdkTransport } from "./sdk-transport.js";
|
|
6
|
+
import { Forwarder } from "./forwarding.js";
|
|
7
|
+
import { probeSsh } from "./probe.js";
|
|
8
|
+
import { Recovery } from "./recovery.js";
|
|
9
|
+
import { RpcServer } from "./rpc.js";
|
|
10
|
+
import { Log } from "./log.js";
|
|
11
|
+
import { DaemonLock } from "./lock.js";
|
|
12
|
+
import { bounded, Fault, safeFault } from "./core.js";
|
|
13
|
+
export class Daemon {
|
|
14
|
+
store;
|
|
15
|
+
options;
|
|
16
|
+
log;
|
|
17
|
+
lock;
|
|
18
|
+
instance = randomUUID();
|
|
19
|
+
config;
|
|
20
|
+
tokens;
|
|
21
|
+
transport;
|
|
22
|
+
rpc;
|
|
23
|
+
controllers = new Map();
|
|
24
|
+
mutations = Promise.resolve();
|
|
25
|
+
queued = 0;
|
|
26
|
+
stopping = false;
|
|
27
|
+
stopPromise;
|
|
28
|
+
constructor(store, options = {}) {
|
|
29
|
+
this.store = store;
|
|
30
|
+
this.options = options;
|
|
31
|
+
this.log = new Log(store.dir);
|
|
32
|
+
this.lock = new DaemonLock(store.dir);
|
|
33
|
+
}
|
|
34
|
+
async start() {
|
|
35
|
+
await this.store.init();
|
|
36
|
+
await this.lock.acquire();
|
|
37
|
+
try {
|
|
38
|
+
await this.log.init();
|
|
39
|
+
this.config = await this.store.load();
|
|
40
|
+
this.tokens = new TokenProvider(this.config.helperPath);
|
|
41
|
+
this.transport = this.options.transport?.(this.tokens) ?? new SdkTransport(this.tokens);
|
|
42
|
+
this.rpc = new RpcServer(join(this.store.dir, 'control.sock'), req => this.handle(req), fault => this.writeLog({ event: 'control-error', message: fault.message }));
|
|
43
|
+
await this.rpc.listen();
|
|
44
|
+
for (const profile of this.config.profiles)
|
|
45
|
+
this.create(profile);
|
|
46
|
+
await this.log.write({ event: 'daemon-ready', message: 'Private control socket ready.' });
|
|
47
|
+
}
|
|
48
|
+
catch (error) {
|
|
49
|
+
await this.stop();
|
|
50
|
+
throw error;
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
writeLog(entry) {
|
|
54
|
+
void this.log.write(entry).catch(() => {
|
|
55
|
+
const fault = new Fault('internal', 'Cannot maintain bounded private service logs; stopping daemon.');
|
|
56
|
+
this.options.fatal?.(fault);
|
|
57
|
+
void this.stop().catch(stopError => this.options.fatal?.(safeFault(stopError)));
|
|
58
|
+
});
|
|
59
|
+
}
|
|
60
|
+
create(profile) {
|
|
61
|
+
const forwarder = new Forwarder(profile.localPort);
|
|
62
|
+
const controller = new Recovery(profile, {
|
|
63
|
+
transport: this.transport, forwarder, probe: signal => probeSsh(profile.localPort, signal),
|
|
64
|
+
log: status => this.logStatus(status),
|
|
65
|
+
});
|
|
66
|
+
this.controllers.set(profile.name, controller);
|
|
67
|
+
controller.start();
|
|
68
|
+
}
|
|
69
|
+
logStatus(status) {
|
|
70
|
+
this.writeLog({ event: 'state', profile: status.name, state: status.state, attempt: status.attempt,
|
|
71
|
+
relayConnected: status.relayConnected,
|
|
72
|
+
...(status.cause ? { message: status.cause.message } : {}),
|
|
73
|
+
...(status.lastHealthyAt !== undefined ? { lastHealthyAt: status.lastHealthyAt } : {}),
|
|
74
|
+
...(status.nextRetryAt !== undefined ? { nextRetryAt: status.nextRetryAt } : {}),
|
|
75
|
+
});
|
|
76
|
+
}
|
|
77
|
+
handle(req) {
|
|
78
|
+
if (this.stopping)
|
|
79
|
+
return Promise.reject(new Fault('daemon-unavailable', 'Daemon is shutting down.'));
|
|
80
|
+
if (req.command === 'status')
|
|
81
|
+
return Promise.resolve({ pid: process.pid, profiles: [...this.controllers.values()].map(c => c.status()) });
|
|
82
|
+
if (req.command === 'logs')
|
|
83
|
+
return Promise.resolve({ instance: this.instance, entries: this.log.snapshot() });
|
|
84
|
+
if (this.queued >= 32)
|
|
85
|
+
return Promise.reject(new Fault('throttled', 'Too many pending configuration changes.'));
|
|
86
|
+
this.queued++;
|
|
87
|
+
const mutation = this.mutations.then(async () => {
|
|
88
|
+
if (this.stopping)
|
|
89
|
+
throw new Fault('daemon-unavailable', 'Daemon is shutting down.');
|
|
90
|
+
const config = this.config;
|
|
91
|
+
if (req.command === 'add') {
|
|
92
|
+
const next = validateConfig({ ...config, profiles: [...config.profiles, req.profile] });
|
|
93
|
+
await this.store.save(next);
|
|
94
|
+
this.config = next;
|
|
95
|
+
if (!this.stopping)
|
|
96
|
+
this.create(req.profile);
|
|
97
|
+
return { saved: true, name: req.profile.name };
|
|
98
|
+
}
|
|
99
|
+
const profiles = config.profiles.filter(p => req.name === 'all' || p.name === req.name);
|
|
100
|
+
if (!profiles.length)
|
|
101
|
+
throw new Fault('invalid-config', 'No matching profile. Use profile list to see configured names.');
|
|
102
|
+
if (req.command !== 'reconnect') {
|
|
103
|
+
const paused = req.command === 'pause';
|
|
104
|
+
const names = new Set(profiles.map(p => p.name));
|
|
105
|
+
const next = validateConfig({ ...config, profiles: config.profiles.map(p => names.has(p.name) ? { ...p, paused } : p) });
|
|
106
|
+
await this.store.save(next);
|
|
107
|
+
this.config = next;
|
|
108
|
+
}
|
|
109
|
+
if (!this.stopping) {
|
|
110
|
+
await Promise.all(profiles.map(async (profile) => {
|
|
111
|
+
const controller = this.controllers.get(profile.name);
|
|
112
|
+
if (req.command === 'pause')
|
|
113
|
+
await controller.pause();
|
|
114
|
+
else if (req.command === 'resume')
|
|
115
|
+
controller.resume();
|
|
116
|
+
else {
|
|
117
|
+
this.tokens.invalidate(profile);
|
|
118
|
+
controller.reconnect();
|
|
119
|
+
}
|
|
120
|
+
}));
|
|
121
|
+
}
|
|
122
|
+
return { profiles: profiles.map(p => this.controllers.get(p.name).status()) };
|
|
123
|
+
});
|
|
124
|
+
this.mutations = mutation.catch(() => { });
|
|
125
|
+
return mutation.finally(() => { this.queued--; });
|
|
126
|
+
}
|
|
127
|
+
stop() {
|
|
128
|
+
if (this.stopPromise)
|
|
129
|
+
return this.stopPromise;
|
|
130
|
+
this.stopping = true;
|
|
131
|
+
this.stopPromise = (async () => {
|
|
132
|
+
const failures = [];
|
|
133
|
+
const attempt = async (action) => {
|
|
134
|
+
try {
|
|
135
|
+
await action();
|
|
136
|
+
}
|
|
137
|
+
catch (error) {
|
|
138
|
+
failures.push(safeFault(error));
|
|
139
|
+
}
|
|
140
|
+
};
|
|
141
|
+
await attempt(() => this.rpc?.close() ?? Promise.resolve());
|
|
142
|
+
await attempt(() => bounded(() => this.mutations, 2500));
|
|
143
|
+
await Promise.all([...this.controllers.values()].map(controller => attempt(() => bounded(() => controller.stop(), 3000))));
|
|
144
|
+
this.tokens?.clear();
|
|
145
|
+
await attempt(() => bounded(() => this.log.flush(), 1000));
|
|
146
|
+
await attempt(() => this.lock.release());
|
|
147
|
+
if (failures.length)
|
|
148
|
+
throw new Fault('internal', `Daemon shutdown had ${failures.length} bounded cleanup failure(s).`);
|
|
149
|
+
})();
|
|
150
|
+
return this.stopPromise;
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
//# sourceMappingURL=daemon.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"daemon.js","sourceRoot":"","sources":["../src/daemon.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAEpD,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAEzC,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAErC,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAE/B,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAMtD,MAAM,OAAO,MAAM;IACA,KAAK,CAAQ;IACb,OAAO,CAAgB;IACvB,GAAG,CAAM;IACT,IAAI,CAAa;IACjB,QAAQ,GAAG,UAAU,EAAE,CAAC;IACjC,MAAM,CAAqB;IAC3B,MAAM,CAA4B;IAClC,SAAS,CAAwB;IACjC,GAAG,CAAwB;IAClB,WAAW,GAAG,IAAI,GAAG,EAAoB,CAAC;IACnD,SAAS,GAAqB,OAAO,CAAC,OAAO,EAAE,CAAC;IAChD,MAAM,GAAG,CAAC,CAAC;IACX,QAAQ,GAAG,KAAK,CAAC;IACjB,WAAW,CAA4B;IAE/C,YAAY,KAAY,EAAE,UAAyB,EAAE;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QAAC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAC3C,IAAI,CAAC,GAAG,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAAC,IAAI,CAAC,IAAI,GAAG,IAAI,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvE,CAAC;IACD,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACxB,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAC1B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YACtC,IAAI,CAAC,MAAM,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACxD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxF,IAAI,CAAC,GAAG,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,cAAc,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EACpF,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC9E,MAAM,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;YACxB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACjE,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,+BAA+B,EAAE,CAAC,CAAC;QAC5F,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAAC,MAAM,KAAK,CAAC;QAAC,CAAC;IACrD,CAAC;IACO,QAAQ,CAAC,KAAe;QAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACpC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,UAAU,EAAE,gEAAgE,CAAC,CAAC;YACtG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC;YAC5B,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAClF,CAAC,CAAC,CAAC;IACL,CAAC;IACO,MAAM,CAAC,OAAgB;QAC7B,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,UAAU,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE;YACvC,SAAS,EAAE,IAAI,CAAC,SAAU,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC;YAC3F,GAAG,EAAE,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SACtC,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC/C,UAAU,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IACO,SAAS,CAAC,MAAc;QAC9B,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO;YAChG,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1D,GAAG,CAAC,MAAM,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtF,GAAG,CAAC,MAAM,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACjF,CAAC,CAAC;IACL,CAAC;IACO,MAAM,CAAC,GAAY;QACzB,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,oBAAoB,EAAE,0BAA0B,CAAC,CAAC,CAAC;QACtG,IAAI,GAAG,CAAC,OAAO,KAAK,QAAQ;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;QAC1I,IAAI,GAAG,CAAC,OAAO,KAAK,MAAM;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAC9G,IAAI,IAAI,CAAC,MAAM,IAAI,EAAE;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,EAAE,yCAAyC,CAAC,CAAC,CAAC;QAChH,IAAI,CAAC,MAAM,EAAE,CAAC;QACd,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;YAC9C,IAAI,IAAI,CAAC,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,EAAE,0BAA0B,CAAC,CAAC;YACrF,MAAM,MAAM,GAAG,IAAI,CAAC,MAAO,CAAC;YAC5B,IAAI,GAAG,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;gBAC1B,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACxF,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;gBACnB,IAAI,CAAC,IAAI,CAAC,QAAQ;oBAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC7C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACjD,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,KAAK,IAAI,CAAC,CAAC,IAAI,KAAK,GAAG,CAAC,IAAI,CAAC,CAAC;YACxF,IAAI,CAAC,QAAQ,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,EAAE,gEAAgE,CAAC,CAAC;YAC1H,IAAI,GAAG,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;gBAChC,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,KAAK,OAAO,CAAC;gBACvC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBACjD,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACzH,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;YACrB,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACnB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAC,OAAO,EAAC,EAAE;oBAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAE,CAAC;oBACvD,IAAI,GAAG,CAAC,OAAO,KAAK,OAAO;wBAAE,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;yBACjD,IAAI,GAAG,CAAC,OAAO,KAAK,QAAQ;wBAAE,UAAU,CAAC,MAAM,EAAE,CAAC;yBAClD,CAAC;wBAAC,IAAI,CAAC,MAAO,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;wBAAC,UAAU,CAAC,SAAS,EAAE,CAAC;oBAAC,CAAC;gBACpE,CAAC,CAAC,CAAC,CAAC;YACN,CAAC;YACD,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAE,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;QACjF,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC1C,OAAO,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC;IACD,IAAI;QACF,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO,IAAI,CAAC,WAAW,CAAC;QAC9C,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,QAAQ,GAAY,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAG,KAAK,EAAE,MAA8B,EAAE,EAAE;gBACvD,IAAI,CAAC;oBAAC,MAAM,MAAM,EAAE,CAAC;gBAAC,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;gBAAC,CAAC;YAC5E,CAAC,CAAC;YACF,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAC5D,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;YACzD,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAChE,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1D,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;YACrB,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;YAC3D,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YACzC,IAAI,QAAQ,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,UAAU,EAAE,uBAAuB,QAAQ,CAAC,MAAM,8BAA8B,CAAC,CAAC;QACzH,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;CACF"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { Duplex } from 'node:stream';
|
|
2
|
+
export type OpenStream = (signal: AbortSignal) => Promise<Duplex>;
|
|
3
|
+
export declare class Forwarder {
|
|
4
|
+
readonly port: number;
|
|
5
|
+
private readonly openTimeoutMs;
|
|
6
|
+
private servers;
|
|
7
|
+
private target;
|
|
8
|
+
private generation;
|
|
9
|
+
private readonly sockets;
|
|
10
|
+
constructor(port: number, openTimeoutMs?: number);
|
|
11
|
+
setTarget(target: OpenStream | undefined): void;
|
|
12
|
+
listen(): Promise<void>;
|
|
13
|
+
private accept;
|
|
14
|
+
close(): Promise<void>;
|
|
15
|
+
}
|
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
import net from 'node:net';
|
|
2
|
+
import { bounded, Fault, safeFault } from "./core.js";
|
|
3
|
+
export class Forwarder {
|
|
4
|
+
port;
|
|
5
|
+
openTimeoutMs;
|
|
6
|
+
servers = [];
|
|
7
|
+
target;
|
|
8
|
+
generation = 0;
|
|
9
|
+
sockets = new Set();
|
|
10
|
+
constructor(port, openTimeoutMs = 3000) { this.port = port; this.openTimeoutMs = openTimeoutMs; }
|
|
11
|
+
setTarget(target) {
|
|
12
|
+
this.generation++;
|
|
13
|
+
this.target = target;
|
|
14
|
+
for (const socket of this.sockets)
|
|
15
|
+
socket.destroy();
|
|
16
|
+
}
|
|
17
|
+
async listen() {
|
|
18
|
+
if (this.servers.length)
|
|
19
|
+
return;
|
|
20
|
+
try {
|
|
21
|
+
for (const host of ['127.0.0.1', '::1']) {
|
|
22
|
+
const server = net.createServer({ allowHalfOpen: true }, socket => this.accept(socket));
|
|
23
|
+
this.servers.push(server);
|
|
24
|
+
await new Promise((resolve, reject) => {
|
|
25
|
+
const error = (e) => { server.off('listening', listening); reject(e); };
|
|
26
|
+
const listening = () => { server.off('error', error); resolve(); };
|
|
27
|
+
server.once('error', error);
|
|
28
|
+
server.once('listening', listening);
|
|
29
|
+
server.listen({ port: this.port, host, ipv6Only: true, exclusive: true });
|
|
30
|
+
});
|
|
31
|
+
server.on('error', () => { this.setTarget(undefined); });
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
catch (error) {
|
|
35
|
+
await this.close();
|
|
36
|
+
throw safeFault(error);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
accept(socket) {
|
|
40
|
+
socket.on('error', () => { socket.destroy(); });
|
|
41
|
+
if (!this.target || this.sockets.size >= 256) {
|
|
42
|
+
socket.destroy();
|
|
43
|
+
return;
|
|
44
|
+
}
|
|
45
|
+
this.sockets.add(socket);
|
|
46
|
+
socket.setNoDelay(true);
|
|
47
|
+
const controller = new AbortController();
|
|
48
|
+
let stream;
|
|
49
|
+
socket.once('close', () => {
|
|
50
|
+
this.sockets.delete(socket);
|
|
51
|
+
controller.abort();
|
|
52
|
+
stream?.destroy();
|
|
53
|
+
});
|
|
54
|
+
const target = this.target, generation = this.generation;
|
|
55
|
+
void bounded(target, this.openTimeoutMs, controller.signal, late => { late.destroy(); }).then(result => {
|
|
56
|
+
stream = result;
|
|
57
|
+
if (socket.destroyed || generation !== this.generation) {
|
|
58
|
+
result.destroy();
|
|
59
|
+
socket.destroy();
|
|
60
|
+
return;
|
|
61
|
+
}
|
|
62
|
+
let remoteEnded = false;
|
|
63
|
+
result.once('end', () => { remoteEnded = true; });
|
|
64
|
+
result.on('error', () => { socket.destroy(); result.destroy(); });
|
|
65
|
+
result.once('close', () => {
|
|
66
|
+
// pipe() drains queued writes on EOF; destroying the socket here would truncate them.
|
|
67
|
+
if (!remoteEnded)
|
|
68
|
+
socket.destroy();
|
|
69
|
+
});
|
|
70
|
+
socket.pipe(result);
|
|
71
|
+
result.pipe(socket);
|
|
72
|
+
}, () => socket.destroy());
|
|
73
|
+
}
|
|
74
|
+
async close() {
|
|
75
|
+
this.setTarget(undefined);
|
|
76
|
+
const servers = this.servers.splice(0);
|
|
77
|
+
await bounded(() => Promise.all(servers.map(server => new Promise((resolve, reject) => {
|
|
78
|
+
server.close(error => {
|
|
79
|
+
if (error && 'code' in error && error.code !== 'ERR_SERVER_NOT_RUNNING')
|
|
80
|
+
reject(error);
|
|
81
|
+
else
|
|
82
|
+
resolve();
|
|
83
|
+
});
|
|
84
|
+
}))).then(() => { }), 1000);
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
//# sourceMappingURL=forwarding.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"forwarding.js","sourceRoot":"","sources":["../src/forwarding.ts"],"names":[],"mappings":"AAAA,OAAO,GAAG,MAAM,UAAU,CAAC;AAE3B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAItD,MAAM,OAAO,SAAS;IACX,IAAI,CAAS;IACL,aAAa,CAAS;IAC/B,OAAO,GAAiB,EAAE,CAAC;IAC3B,MAAM,CAAyB;IAC/B,UAAU,GAAG,CAAC,CAAC;IACN,OAAO,GAAG,IAAI,GAAG,EAAc,CAAC;IACjD,YAAY,IAAY,EAAE,aAAa,GAAG,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC,CAAC,CAAC;IACzG,SAAS,CAAC,MAA8B;QACtC,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;IACtD,CAAC;IACD,KAAK,CAAC,MAAM;QACV,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAO;QAChC,IAAI,CAAC;YACH,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC;gBACxC,MAAM,MAAM,GAAG,GAAG,CAAC,YAAY,CAAC,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;gBACxF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC1B,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBAC1C,MAAM,KAAK,GAAG,CAAC,CAAQ,EAAE,EAAE,GAAG,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC/E,MAAM,SAAS,GAAG,GAAG,EAAE,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;oBACnE,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;oBAC5B,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;oBACpC,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC5E,CAAC,CAAC,CAAC;gBACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAAC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YAAC,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QAAC,CAAC;IACjE,CAAC;IACO,MAAM,CAAC,MAAkB;QAC/B,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAChD,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;YAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAAC,OAAO;QAAC,CAAC;QAC3E,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACzB,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACxB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,IAAI,MAA0B,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;YACxB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC5B,UAAU,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,EAAE,OAAO,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACzD,KAAK,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,aAAa,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE;YACrG,MAAM,GAAG,MAAM,CAAC;YAChB,IAAI,MAAM,CAAC,SAAS,IAAI,UAAU,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;gBAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAAC,OAAO;YAAC,CAAC;YACvG,IAAI,WAAW,GAAG,KAAK,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,WAAW,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAClE,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;gBACxB,sFAAsF;gBACtF,IAAI,CAAC,WAAW;oBAAE,MAAM,CAAC,OAAO,EAAE,CAAC;YACrC,CAAC,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtB,CAAC,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;IAC7B,CAAC;IACD,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1F,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;gBACnB,IAAI,KAAK,IAAI,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,wBAAwB;oBAAE,MAAM,CAAC,KAAK,CAAC,CAAC;;oBAClF,OAAO,EAAE,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAC7B,CAAC;CACF"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import type { HostMessage, HostOptions } from './host.ts';
|
|
2
|
+
import { retireHostEndpoint } from './host-sdk.ts';
|
|
3
|
+
export declare function runHostWorker(options: HostOptions, emit: (message: HostMessage) => void, signal: AbortSignal, entry?: URL, retire?: typeof retireHostEndpoint): Promise<void>;
|
|
4
|
+
export declare function runHost(options: HostOptions, report: (message: HostMessage) => void, parent?: AbortSignal): Promise<void>;
|
|
@@ -0,0 +1,110 @@
|
|
|
1
|
+
import { fork } from 'node:child_process';
|
|
2
|
+
import { fileURLToPath } from 'node:url';
|
|
3
|
+
import { Fault, safeFault } from "./core.js";
|
|
4
|
+
import { parseHostMessage, superviseHost } from "./host.js";
|
|
5
|
+
import { retireHostEndpoint } from "./host-sdk.js";
|
|
6
|
+
export async function runHostWorker(options, emit, signal, entry = new URL(import.meta.url.endsWith('.ts') ? './host-worker.ts' : './host-worker.js', import.meta.url), retire = retireHostEndpoint) {
|
|
7
|
+
if (signal.aborted)
|
|
8
|
+
return;
|
|
9
|
+
let endpoint;
|
|
10
|
+
let failure;
|
|
11
|
+
await new Promise((resolve, reject) => {
|
|
12
|
+
const child = fork(fileURLToPath(entry), [], {
|
|
13
|
+
execArgv: [], stdio: ['ignore', 'ignore', 'ignore', 'ipc'],
|
|
14
|
+
env: { ...process.env, NODE_OPTIONS: '', NODE_PATH: '' },
|
|
15
|
+
});
|
|
16
|
+
let stopping = false, exited = false;
|
|
17
|
+
let terminate;
|
|
18
|
+
let kill;
|
|
19
|
+
const stop = () => {
|
|
20
|
+
if (stopping || exited)
|
|
21
|
+
return;
|
|
22
|
+
stopping = true;
|
|
23
|
+
if (child.connected)
|
|
24
|
+
child.send({ kind: 'stop' }, error => { if (error && !exited)
|
|
25
|
+
child.kill('SIGTERM'); });
|
|
26
|
+
terminate = setTimeout(() => { if (!exited)
|
|
27
|
+
child.kill('SIGTERM'); }, 2500);
|
|
28
|
+
kill = setTimeout(() => { if (!exited)
|
|
29
|
+
child.kill('SIGKILL'); }, 3000);
|
|
30
|
+
};
|
|
31
|
+
signal.addEventListener('abort', stop, { once: true });
|
|
32
|
+
child.once('spawn', () => {
|
|
33
|
+
if (signal.aborted) {
|
|
34
|
+
stop();
|
|
35
|
+
return;
|
|
36
|
+
}
|
|
37
|
+
child.send({ kind: 'start', options }, error => {
|
|
38
|
+
if (error) {
|
|
39
|
+
failure ??= new Fault('protocol');
|
|
40
|
+
stop();
|
|
41
|
+
}
|
|
42
|
+
});
|
|
43
|
+
});
|
|
44
|
+
child.on('message', value => {
|
|
45
|
+
if (stopping)
|
|
46
|
+
return;
|
|
47
|
+
const message = parseHostMessage(value);
|
|
48
|
+
if (!message) {
|
|
49
|
+
failure ??= new Fault('protocol');
|
|
50
|
+
stop();
|
|
51
|
+
return;
|
|
52
|
+
}
|
|
53
|
+
if (message.kind === 'endpoint') {
|
|
54
|
+
if (endpoint && (endpoint.id !== message.id || endpoint.clusterId !== message.clusterId)) {
|
|
55
|
+
failure ??= new Fault('protocol');
|
|
56
|
+
stop();
|
|
57
|
+
return;
|
|
58
|
+
}
|
|
59
|
+
endpoint = message;
|
|
60
|
+
child.send({ kind: 'endpoint-recorded', id: message.id }, error => {
|
|
61
|
+
if (error && !exited) {
|
|
62
|
+
failure ??= new Fault('protocol');
|
|
63
|
+
stop();
|
|
64
|
+
}
|
|
65
|
+
});
|
|
66
|
+
}
|
|
67
|
+
else {
|
|
68
|
+
if (message.kind === 'fault')
|
|
69
|
+
failure ??= new Fault(message.code, undefined, message.retryAfterMs);
|
|
70
|
+
emit(message);
|
|
71
|
+
}
|
|
72
|
+
});
|
|
73
|
+
child.once('error', () => { failure ??= new Fault('invalid-config', 'Cannot start the host worker. Check the Node runtime and installed package.'); });
|
|
74
|
+
child.once('close', () => {
|
|
75
|
+
exited = true;
|
|
76
|
+
clearTimeout(terminate);
|
|
77
|
+
clearTimeout(kill);
|
|
78
|
+
signal.removeEventListener('abort', stop);
|
|
79
|
+
if (!signal.aborted)
|
|
80
|
+
reject(failure ?? new Fault('retrying'));
|
|
81
|
+
else
|
|
82
|
+
resolve();
|
|
83
|
+
});
|
|
84
|
+
if (signal.aborted)
|
|
85
|
+
stop();
|
|
86
|
+
}).catch(error => { failure ??= safeFault(error); });
|
|
87
|
+
if (endpoint)
|
|
88
|
+
await retire(options, endpoint);
|
|
89
|
+
if (failure && !signal.aborted)
|
|
90
|
+
throw failure;
|
|
91
|
+
}
|
|
92
|
+
export async function runHost(options, report, parent) {
|
|
93
|
+
const stop = new AbortController();
|
|
94
|
+
const abort = () => stop.abort();
|
|
95
|
+
parent?.addEventListener('abort', abort, { once: true });
|
|
96
|
+
if (parent?.aborted)
|
|
97
|
+
stop.abort();
|
|
98
|
+
const signals = process.platform === 'win32' ? ['SIGINT', 'SIGTERM', 'SIGBREAK'] : ['SIGINT', 'SIGTERM'];
|
|
99
|
+
for (const signal of signals)
|
|
100
|
+
process.once(signal, abort);
|
|
101
|
+
try {
|
|
102
|
+
await superviseHost(options, { worker: runHostWorker, report }, stop.signal);
|
|
103
|
+
}
|
|
104
|
+
finally {
|
|
105
|
+
parent?.removeEventListener('abort', abort);
|
|
106
|
+
for (const signal of signals)
|
|
107
|
+
process.off(signal, abort);
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
//# sourceMappingURL=host-process.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"host-process.js","sourceRoot":"","sources":["../src/host-process.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAE5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,OAAoB,EAAE,IAAoC,EAC5F,MAAmB,EAAE,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAChI,MAAM,GAAG,kBAAkB;IAC3B,IAAI,MAAM,CAAC,OAAO;QAAE,OAAO;IAC3B,IAAI,QAAgE,CAAC;IACrE,IAAI,OAA0B,CAAC;IAC/B,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE;YAC3C,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,CAAC;YAC1D,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;SACzD,CAAC,CAAC;QACH,IAAI,QAAQ,GAAG,KAAK,EAAE,MAAM,GAAG,KAAK,CAAC;QACrC,IAAI,SAAoD,CAAC;QACzD,IAAI,IAA+C,CAAC;QACpD,MAAM,IAAI,GAAG,GAAG,EAAE;YAChB,IAAI,QAAQ,IAAI,MAAM;gBAAE,OAAO;YAC/B,QAAQ,GAAG,IAAI,CAAC;YAChB,IAAI,KAAK,CAAC,SAAS;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,GAAG,IAAI,KAAK,IAAI,CAAC,MAAM;oBAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7G,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YAC5E,IAAI,GAAG,UAAU,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QACzE,CAAC,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;YACvB,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBAAC,IAAI,EAAE,CAAC;gBAAC,OAAO;YAAC,CAAC;YACvC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,KAAK,CAAC,EAAE;gBAC7C,IAAI,KAAK,EAAE,CAAC;oBAAC,OAAO,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;oBAAC,IAAI,EAAE,CAAC;gBAAC,CAAC;YAC3D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,CAAC,EAAE;YAC1B,IAAI,QAAQ;gBAAE,OAAO;YACrB,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YACxC,IAAI,CAAC,OAAO,EAAE,CAAC;gBAAC,OAAO,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;gBAAC,IAAI,EAAE,CAAC;gBAAC,OAAO;YAAC,CAAC;YACpE,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAChC,IAAI,QAAQ,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,IAAI,QAAQ,CAAC,SAAS,KAAK,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;oBACzF,OAAO,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;oBAAC,IAAI,EAAE,CAAC;oBAAC,OAAO;gBACpD,CAAC;gBACD,QAAQ,GAAG,OAAO,CAAC;gBACnB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE;oBAChE,IAAI,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;wBAAC,OAAO,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;wBAAC,IAAI,EAAE,CAAC;oBAAC,CAAC;gBACtE,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO;oBAAE,OAAO,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;gBACnG,IAAI,CAAC,OAAO,CAAC,CAAC;YAChB,CAAC;QACH,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,OAAO,KAAK,IAAI,KAAK,CAAC,gBAAgB,EAAE,6EAA6E,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACvJ,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;YACvB,MAAM,GAAG,IAAI,CAAC;YAAC,YAAY,CAAC,SAAS,CAAC,CAAC;YAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC3D,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC1C,IAAI,CAAC,MAAM,CAAC,OAAO;gBAAE,MAAM,CAAC,OAAO,IAAI,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;;gBACzD,OAAO,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QACH,IAAI,MAAM,CAAC,OAAO;YAAE,IAAI,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,GAAG,OAAO,KAAK,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,IAAI,QAAQ;QAAE,MAAM,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC9C,IAAI,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO;QAAE,MAAM,OAAO,CAAC;AAChD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,OAAoB,EAAE,MAAsC,EAAE,MAAoB;IAC9G,MAAM,IAAI,GAAG,IAAI,eAAe,EAAE,CAAC;IACnC,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;IACjC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACzD,IAAI,MAAM,EAAE,OAAO;QAAE,IAAI,CAAC,KAAK,EAAE,CAAC;IAClC,MAAM,OAAO,GAAqB,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC3H,KAAK,MAAM,MAAM,IAAI,OAAO;QAAE,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC1D,IAAI,CAAC;QAAC,MAAM,aAAa,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAAC,CAAC;YAC7E,CAAC;QACP,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC5C,KAAK,MAAM,MAAM,IAAI,OAAO;YAAE,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC3D,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
import { Agent } from 'node:https';
|
|
2
|
+
import { TunnelRelayTunnelHost } from '@microsoft/dev-tunnels-connections';
|
|
3
|
+
import { TunnelManagementHttpClient } from '@microsoft/dev-tunnels-management';
|
|
4
|
+
import { TokenProvider } from './auth.ts';
|
|
5
|
+
import type { HostMessage, HostOptions, HostStatus } from './host.ts';
|
|
6
|
+
import { sleep } from './host.ts';
|
|
7
|
+
type HostManagement = TunnelManagementHttpClient & {
|
|
8
|
+
readonly httpsAgent: Agent;
|
|
9
|
+
};
|
|
10
|
+
export declare function hostManagement(agent: Agent): HostManagement;
|
|
11
|
+
export declare function localPort(port: number, signal: AbortSignal): Promise<void>;
|
|
12
|
+
interface ProbeDependencies {
|
|
13
|
+
local(port: number, signal: AbortSignal): Promise<void>;
|
|
14
|
+
remote(port: number, signal: AbortSignal): Promise<void>;
|
|
15
|
+
}
|
|
16
|
+
export declare class HostProbe {
|
|
17
|
+
private readonly ports;
|
|
18
|
+
private readonly deps;
|
|
19
|
+
private readonly failures;
|
|
20
|
+
private retryAt;
|
|
21
|
+
constructor(ports: number[], deps: ProbeDependencies);
|
|
22
|
+
check(ports: number[], signal: AbortSignal, age: number): Promise<{
|
|
23
|
+
status: HostStatus;
|
|
24
|
+
replace: boolean;
|
|
25
|
+
}>;
|
|
26
|
+
}
|
|
27
|
+
interface HostSdkDependencies {
|
|
28
|
+
tokens?: TokenProvider;
|
|
29
|
+
management?(agent: Agent): TunnelManagementHttpClient;
|
|
30
|
+
host?(management: HostManagement): TunnelRelayTunnelHost;
|
|
31
|
+
now?(): number;
|
|
32
|
+
wait?: typeof sleep;
|
|
33
|
+
}
|
|
34
|
+
export declare function runHostSdk(options: HostOptions, report: (message: HostMessage) => void | Promise<void>, parent: AbortSignal, deps?: HostSdkDependencies): Promise<void>;
|
|
35
|
+
export declare function retireHostEndpoint(options: HostOptions, endpoint: Extract<HostMessage, {
|
|
36
|
+
kind: 'endpoint';
|
|
37
|
+
}>, deps?: Pick<HostSdkDependencies, 'tokens' | 'management'>): Promise<void>;
|
|
38
|
+
export {};
|