Initial commit

This commit is contained in:
Filip Znachor 2022-06-08 22:06:57 +02:00
commit 3b00c6498a
5 changed files with 238 additions and 0 deletions

1
.gitignore vendored Normal file
View file

@ -0,0 +1 @@
node_modules

140
index.ts Normal file
View file

@ -0,0 +1,140 @@
import http, { IncomingMessage, ServerResponse } from "http";
import httpProxy from 'http-proxy';
import fs from "fs";
import path from "path";
import os from "os";
import child_process, { ChildProcess } from "child_process";
import kill from "tree-kill";
class ServiceManager {
pwd!: string;
proxy = httpProxy.createProxy();
server = http.createServer((req, res) => this.listen(req, res));
services: Service[] = [];
stopped: boolean = false;
async setup() {
this.pwd = fs.mkdtempSync(path.join(os.tmpdir(), "odp_"));
this.server.listen(3000);
}
async stop() {
if(this.stopped) return;
this.stopped = true;
this.services.forEach(s => s.stop());
fs.rmdirSync(this.pwd);
}
async listen(req: IncomingMessage, res: ServerResponse) {
let oriHost = req.headers.host;
let host = oriHost.indexOf(":") ? oriHost.split(':')[0] : oriHost;
for(let i in this.services) {
if(this.services[i].hosts.indexOf(host) !== -1) {
this.proxy.web(req, res, {
target: {
socketPath: await this.services[i].start(),
host: "localhost"
}
});
return;
}
}
console.warn("Host", host, "not found!");
res.end("Host not found in host map");
}
}
class Service implements ServiceConfig {
hosts: string[];
command: string;
file: string;
child?: ChildProcess;
timeout?: number;
life_check?: NodeJS.Timeout;
last_activity: Date;
man: ServiceManager;
alive: boolean = false;
constructor(man: ServiceManager, config: ServiceConfig) {
this.man = man;
this.hosts = config.hosts;
this.command = config.command;
this.file = config.file;
this.timeout = config.timeout*1000;
man.services.push(this);
}
async start() {
this.last_activity = new Date;
if(this.timeout) {
if(this.life_check) clearTimeout(this.life_check);
this.life_check = setTimeout(() => {
if((new Date).getTime() > this.last_activity.getTime()+this.timeout) this.stop();
}, this.timeout);
}
if(!this.child) {
if(fs.existsSync(this.file)) fs.rmSync(this.file);
this.child = child_process.exec(this.command);
console.log(`Started service ${this.hosts[0]}`);
}
if(!this.alive) {
await new Promise((resolve) => {
let int = setInterval(() => {
if(fs.existsSync(this.file)) {
clearInterval(int);
resolve(null);
}
}, 100);
});
this.alive = true;
}
return this.file;
}
async stop() {
if(fs.existsSync(this.file)) fs.rmSync(this.file);
if(this.alive) {
this.alive = false;
kill(this.child.pid, "SIGUSR2");
this.child = null;
console.log(`Stopped service ${this.hosts[0]}`);
}
}
}
interface ServiceConfig {
hosts: string[];
command: string;
file: string;
timeout?: number;
};
let start = async (config: ServiceConfig[]) => {
let man = new ServiceManager;
await man.setup();
config.forEach(s => {
new Service(man, s);
});
let exit = async (e: any) => {
if(e) console.log(e);
man.stop();
process.exit(0);
};
process.on('exit', exit);
process.on('beforeExit', exit);
process.on('SIGINT', exit);
process.on('SIGUSR1', exit);
process.on('SIGUSR2', exit);
process.on('SIGTERM', exit);
process.on('uncaughtException', console.error);
};
let map = require("./map.json");
start(map);

8
map.json Normal file
View file

@ -0,0 +1,8 @@
[
{
"hosts": ["test.example.com"],
"command": "node ../http/server.js",
"file": "http-server.sock",
"timeout": 60
}
]

17
package.json Normal file
View file

@ -0,0 +1,17 @@
{
"name": "on-demand-proxy",
"version": "1.0.0",
"description": "",
"main": "index.ts",
"keywords": [],
"author": "",
"license": "MIT",
"dependencies": {
"@types/node": "^17.0.41",
"http-proxy": "^1.18.1",
"tree-kill": "^1.2.2"
},
"devDependencies": {
"@types/http-proxy": "^1.17.9"
}
}

72
tsconfig.json Normal file
View file

@ -0,0 +1,72 @@
{
"compilerOptions": {
/* Visit https://aka.ms/tsconfig.json to read more about this file */
/* Basic Options */
// "incremental": true, /* Enable incremental compilation */
"target": "es5", /* Specify ECMAScript target version: 'ES3' (default), 'ES5', 'ES2015', 'ES2016', 'ES2017', 'ES2018', 'ES2019', 'ES2020', 'ES2021', or 'ESNEXT'. */
"module": "commonjs", /* Specify module code generation: 'none', 'commonjs', 'amd', 'system', 'umd', 'es2015', 'es2020', or 'ESNext'. */
// "lib": [], /* Specify library files to be included in the compilation. */
// "allowJs": true, /* Allow javascript files to be compiled. */
// "checkJs": true, /* Report errors in .js files. */
// "jsx": "preserve", /* Specify JSX code generation: 'preserve', 'react-native', 'react', 'react-jsx' or 'react-jsxdev'. */
// "declaration": true, /* Generates corresponding '.d.ts' file. */
// "declarationMap": true, /* Generates a sourcemap for each corresponding '.d.ts' file. */
// "sourceMap": true, /* Generates corresponding '.map' file. */
// "outFile": "./", /* Concatenate and emit output to single file. */
// "outDir": "./", /* Redirect output structure to the directory. */
// "rootDir": "./", /* Specify the root directory of input files. Use to control the output directory structure with --outDir. */
// "composite": true, /* Enable project compilation */
// "tsBuildInfoFile": "./", /* Specify file to store incremental compilation information */
// "removeComments": true, /* Do not emit comments to output. */
// "noEmit": true, /* Do not emit outputs. */
// "importHelpers": true, /* Import emit helpers from 'tslib'. */
// "downlevelIteration": true, /* Provide full support for iterables in 'for-of', spread, and destructuring when targeting 'ES5' or 'ES3'. */
// "isolatedModules": true, /* Transpile each file as a separate module (similar to 'ts.transpileModule'). */
/* Strict Type-Checking Options */
"strict": false, /* Enable all strict type-checking options. */
// "noImplicitAny": true, /* Raise error on expressions and declarations with an implied 'any' type. */
// "strictNullChecks": true, /* Enable strict null checks. */
// "strictFunctionTypes": true, /* Enable strict checking of function types. */
// "strictBindCallApply": true, /* Enable strict 'bind', 'call', and 'apply' methods on functions. */
// "strictPropertyInitialization": true, /* Enable strict checking of property initialization in classes. */
// "noImplicitThis": true, /* Raise error on 'this' expressions with an implied 'any' type. */
// "alwaysStrict": true, /* Parse in strict mode and emit "use strict" for each source file. */
/* Additional Checks */
// "noUnusedLocals": true, /* Report errors on unused locals. */
// "noUnusedParameters": true, /* Report errors on unused parameters. */
// "noImplicitReturns": true, /* Report error when not all code paths in function return a value. */
// "noFallthroughCasesInSwitch": true, /* Report errors for fallthrough cases in switch statement. */
// "noUncheckedIndexedAccess": true, /* Include 'undefined' in index signature results */
// "noImplicitOverride": true, /* Ensure overriding members in derived classes are marked with an 'override' modifier. */
// "noPropertyAccessFromIndexSignature": true, /* Require undeclared properties from index signatures to use element accesses. */
/* Module Resolution Options */
// "moduleResolution": "node", /* Specify module resolution strategy: 'node' (Node.js) or 'classic' (TypeScript pre-1.6). */
// "baseUrl": "./", /* Base directory to resolve non-absolute module names. */
// "paths": {}, /* A series of entries which re-map imports to lookup locations relative to the 'baseUrl'. */
// "rootDirs": [], /* List of root folders whose combined content represents the structure of the project at runtime. */
// "typeRoots": [], /* List of folders to include type definitions from. */
// "types": [], /* Type declaration files to be included in compilation. */
// "allowSyntheticDefaultImports": true, /* Allow default imports from modules with no default export. This does not affect code emit, just typechecking. */
"esModuleInterop": true, /* Enables emit interoperability between CommonJS and ES Modules via creation of namespace objects for all imports. Implies 'allowSyntheticDefaultImports'. */
// "preserveSymlinks": true, /* Do not resolve the real path of symlinks. */
// "allowUmdGlobalAccess": true, /* Allow accessing UMD globals from modules. */
/* Source Map Options */
// "sourceRoot": "", /* Specify the location where debugger should locate TypeScript files instead of source locations. */
// "mapRoot": "", /* Specify the location where debugger should locate map files instead of generated locations. */
// "inlineSourceMap": true, /* Emit a single file with source maps instead of having a separate file. */
// "inlineSources": true, /* Emit the source alongside the sourcemaps within a single file; requires '--inlineSourceMap' or '--sourceMap' to be set. */
/* Experimental Options */
// "experimentalDecorators": true, /* Enables experimental support for ES7 decorators. */
// "emitDecoratorMetadata": true, /* Enables experimental support for emitting type metadata for decorators. */
/* Advanced Options */
"skipLibCheck": true, /* Skip type checking of declaration files. */
"forceConsistentCasingInFileNames": true /* Disallow inconsistently-cased references to the same file. */
}
}