From 3b00c6498ae6e092ca749175a1781926934267ec Mon Sep 17 00:00:00 2001 From: Filip Znachor Date: Wed, 8 Jun 2022 22:06:57 +0200 Subject: [PATCH] Initial commit --- .gitignore | 1 + index.ts | 140 ++++++++++++++++++++++++++++++++++++++++++++++++++ map.json | 8 +++ package.json | 17 ++++++ tsconfig.json | 72 ++++++++++++++++++++++++++ 5 files changed, 238 insertions(+) create mode 100644 .gitignore create mode 100644 index.ts create mode 100644 map.json create mode 100644 package.json create mode 100644 tsconfig.json diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..b512c09 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +node_modules \ No newline at end of file diff --git a/index.ts b/index.ts new file mode 100644 index 0000000..2fdf7b6 --- /dev/null +++ b/index.ts @@ -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); \ No newline at end of file diff --git a/map.json b/map.json new file mode 100644 index 0000000..4540748 --- /dev/null +++ b/map.json @@ -0,0 +1,8 @@ +[ + { + "hosts": ["test.example.com"], + "command": "node ../http/server.js", + "file": "http-server.sock", + "timeout": 60 + } +] \ No newline at end of file diff --git a/package.json b/package.json new file mode 100644 index 0000000..a3e9511 --- /dev/null +++ b/package.json @@ -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" + } +} diff --git a/tsconfig.json b/tsconfig.json new file mode 100644 index 0000000..520824f --- /dev/null +++ b/tsconfig.json @@ -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. */ + } +}