设置成功，点击调整到选择环境页面
'; 10 | return; 11 | } 12 | ctx.body = { 13 | ec: 0, 14 | clientId: getClientId(ctx), 15 | followerIp: ctx.envMgr.getFollower(ctx), 16 | }; 17 | }; 18 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/getEnv.js: -------------------------------------------------------------------------------- 1 | 2 | const { getClientId } = require('../../util'); 3 | 4 | module.exports = (ctx) => { 5 | const curEnv = ctx.envMgr.getEnv(ctx) || null; 6 | ctx.body = { 7 | ec: 0, 8 | curEnv, 9 | clientIp: ctx.ip, 10 | clientId: getClientId(ctx), 11 | }; 12 | }; 13 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/list.js: -------------------------------------------------------------------------------- 1 | const Limiter = require('async-limiter'); 2 | const { gzip } = require('zlib'); 3 | const LRU = require('lru-cache'); 4 | 5 | const limiter = new Limiter({ concurrency: 10 }); 6 | const cache = new LRU({ max: 2 }); 7 | 8 | const compress = (body) => { 9 | let promise = cache.get(body); 10 | if (!promise) { 11 | promise = new Promise((resolve, reject) => { 12 | limiter.push((done) => { 13 | gzip(body, (err, buf) => { 14 | done(); 15 | if (err) { 16 | delete cache[body]; 17 | reject(err); 18 | } else { 19 | resolve(buf); 20 | } 21 | }); 22 | }); 23 | }); 24 | cache.set(body, promise); 25 | } 26 | return promise; 27 | }; 28 | 29 | module.exports = async (ctx) => { 30 | const curEnv = ctx.envMgr.getEnv(ctx); 31 | const list = ctx.accountMgr.getAccountList(ctx.request.query.parsed); 32 | let body = JSON.stringify({ 33 | admin: ctx.admin, 34 | ec: 0, 35 | baseUrl: ctx.baseUrl, 36 | curEnv, 37 | list, 38 | }); 39 | body = await compress(body); 40 | ctx.set('content-encoding', 'gzip'); 41 | ctx.set('content-type', 'application/json'); 42 | ctx.body = body; 43 | }; 44 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/patterns.js: -------------------------------------------------------------------------------- 1 | 2 | module.exports = (ctx) => { 3 | const { 4 | patterns, 5 | preRules, 6 | postRules, 7 | } = ctx.accountMgr; 8 | ctx.body = { 9 | ec: 0, 10 | patterns, 11 | preRules, 12 | postRules, 13 | }; 14 | }; 15 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/pluginRules.js: -------------------------------------------------------------------------------- 1 | const fs = require('fs'); 2 | const path = require('path'); 3 | 4 | const resolveValue = (text) => { 5 | return text.substring(text.indexOf('```')); 6 | }; 7 | const INJECT_HTML = resolveValue(fs.readFileSync(path.join(__dirname, '../../../_rules.txt'), { encoding: 'utf8' })); // eslint-disable-line 8 | const INJECT_RULE = 'htmlPrepend://{whistle.nohost/inject.html}'; 9 | const WHISTLE_HOST = 'local.whistlejs.com'; 10 | let curBaseUrl; 11 | let curPatterns; 12 | let curRules; 13 | 14 | module.exports = (ctx) => { 15 | const { 16 | patterns, 17 | getBaseUrl, 18 | getDomainList, 19 | preRules, 20 | postRules, 21 | } = ctx.accountMgr; 22 | const curRuntimeId = ctx.get('x-whistle-runtime-id'); 23 | const isSelf = curRuntimeId === ctx.runtimeId; 24 | const baseUrl = !isSelf && getBaseUrl(); 25 | const domainList = getDomainList().concat([WHISTLE_HOST]); 26 | if (baseUrl) { 27 | if (curBaseUrl !== baseUrl || curPatterns !== patterns) { 28 | curBaseUrl = baseUrl; 29 | curPatterns = patterns; 30 | let ignorePatterns = []; 31 | const injectRule = [`internal-proxy://${baseUrl} ${INJECT_RULE} enable://clientId|multiClient|safeHtml`]; 32 | const normalRule = [`internal-proxy://${baseUrl} enable://clientId|multiClient`]; 33 | curRules = []; 34 | patterns.forEach((item) => { 35 | if (item.ignore) { 36 | ignorePatterns.push(`excludeFilter://${item.pattern}`); 37 | } else if (item.button) { 38 | injectRule.push(item.pattern); 39 | } else { 40 | normalRule.push(item.pattern); 41 | } 42 | }); 43 | ignorePatterns = ignorePatterns.join(' '); 44 | if (normalRule.length > 1) { 45 | normalRule.push(ignorePatterns); 46 | curRules.push(normalRule.join(' ')); 47 | } 48 | if (injectRule.length > 1) { 49 | injectRule.push(ignorePatterns); 50 | curRules.push(injectRule.join(' ')); 51 | } 52 | curRules.unshift(`internal-proxy://${baseUrl} ignore://rule|html enable://hide|proxyHost|clientId|multiClient ${domainList.join(' ')}`); 53 | curRules.push(`*/.whistle-path.5b6af7b9884e1165/ ignore://-proxy|-host reqHeaders://whistleInternalHost=${WHISTLE_HOST} enable://proxyTunnel`); 54 | curRules.push('\n', INJECT_HTML); 55 | curRules = preRules.concat(curRules).concat(postRules).join('\n'); 56 | } 57 | ctx.body = curRules; 58 | } else { 59 | ctx.body = ''; 60 | } 61 | }; 62 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/proxy.js: -------------------------------------------------------------------------------- 1 | const LRU = require('lru-cache'); 2 | const { transformWhistle, changeFilter } = require('../../util'); 3 | const whistleMgr = require('../../whistleMgr'); 4 | 5 | const tempCache = new LRU({ maxAge: 5000 }); 6 | 7 | module.exports = async (ctx, next) => { 8 | let { params: { name }, account } = ctx; 9 | if (!account) { 10 | account = ctx.accountMgr.getAccount(name); 11 | if (!account) { 12 | await next(); 13 | return; 14 | } 15 | } else { 16 | name = account.name; 17 | } 18 | const { req } = ctx; 19 | req.url = req.url.substring(`/account/${name}`.length); 20 | changeFilter(req, ctx.envMgr.getFollower(ctx)); 21 | req.headers.host = 'local.wproxy.org'; 22 | req.headers['x-forwarded-for'] = ctx.ip || '127.0.0.1'; 23 | const { 24 | port, 25 | remoteAddrHead, 26 | remotePortHead, 27 | } = await whistleMgr.fork(account); 28 | const remoteAddr = ctx.get('x-whistle-remote-address'); 29 | if (remoteAddr) { 30 | req.headers[remoteAddrHead] = remoteAddr; 31 | req.headers[remotePortHead] = ctx.get('x-whistle-remote-port'); 32 | } 33 | try { 34 | await transformWhistle(ctx, port); 35 | } catch (err) { 36 | if (err.code === 'ECONNREFUSED' && !tempCache.get(name)) { 37 | tempCache.set(name, 1); 38 | whistleMgr.kill(name); 39 | } 40 | throw err; 41 | } 42 | }; 43 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/proxyNetwork.js: -------------------------------------------------------------------------------- 1 | const { transformWhistle, changeFilter } = require('../../util'); 2 | const initNetwork = require('../network'); 3 | 4 | module.exports = async (ctx) => { 5 | const { req } = ctx; 6 | req.url = req.url.replace('/network', ''); 7 | req.headers.host = 'local.wproxy.org'; 8 | req.headers['x-forwarded-for'] = ctx.ip || '127.0.0.1'; 9 | const port = await initNetwork(); 10 | changeFilter(req); 11 | await transformWhistle(ctx, port); 12 | }; 13 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/redirect.js: -------------------------------------------------------------------------------- 1 | const { getClientId } = require('../../util'); 2 | 3 | module.exports = (ctx) => { 4 | const { url, name, env } = ctx.request.query; 5 | const account = ctx.accountMgr.getAccount(name); 6 | if (account) { 7 | ctx.envMgr.setEnv(getClientId(ctx), name, env); 8 | } 9 | ctx.redirect(url || './'); 10 | }; 11 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/selectEnv.js: -------------------------------------------------------------------------------- 1 | const { COOKIE_NAME, ENV_MAX_AGE, decodeURIComponentSafe, getClientId } = require('../../util'); 2 | 3 | module.exports = (ctx) => { 4 | const { name, envId, redirect } = ctx.request.query; 5 | const envName = decodeURIComponentSafe(envId); 6 | const env = ctx.envMgr.setEnv(getClientId(ctx), name, envName); 7 | let value = ''; 8 | if (env) { 9 | value = encodeURIComponent(`${name}/${env.envName}`); 10 | } 11 | ctx.cookies.set(COOKIE_NAME, value, { 12 | path: '/', 13 | expires: new Date(Date.now() + (ENV_MAX_AGE * 1000)), 14 | }); 15 | if (redirect && typeof redirect === 'string') { 16 | ctx.redirect(redirect); 17 | } else { 18 | ctx.body = { ec: 0 }; 19 | } 20 | }; 21 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/cgi/unfollow.js: -------------------------------------------------------------------------------- 1 | module.exports = (ctx) => { 2 | ctx.envMgr.unfollow(ctx); 3 | ctx.body = { ec: 0 }; 4 | }; 5 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/index.js: -------------------------------------------------------------------------------- 1 | const Koa = require('koa'); 2 | const onerror = require('koa-onerror'); 3 | const serve = require('koa-static'); 4 | const path = require('path'); 5 | const router = require('koa-router')(); 6 | const setupRouter = require('./router'); 7 | const accountMgr = require('../accountMgr'); 8 | const envMgr = require('../envMgr'); 9 | const whistleMgr = require('../whistleMgr'); 10 | 11 | const MAX_AGE = 1000 * 60 * 5; 12 | const error = function (em, ec) { 13 | em = em || '请求失败，请刷新页面重试'; 14 | ec = ec || 2; 15 | this.body = { ec, em }; 16 | }; 17 | 18 | module.exports = (server, options) => { 19 | const { username, password } = options.data; 20 | const { runtimeId } = options.config; 21 | const app = new Koa(); 22 | app.proxy = true; 23 | app.silent = true; 24 | if (process.env.PFORK_MODE === 'bind') { 25 | onerror(app); 26 | } 27 | app.use(async (ctx, next) => { 28 | const baseUrl = `http://${accountMgr.getBaseUrl()}/`; 29 | ctx.baseUrl = baseUrl; 30 | ctx.success = function () { 31 | this.body = { ec: 0, baseUrl }; 32 | }; 33 | ctx.error = error; 34 | ctx.accountMgr = accountMgr; 35 | ctx.envMgr = envMgr; 36 | ctx.config = options.config; 37 | ctx.updateRules = options.updateRules; 38 | ctx.whistleMgr = whistleMgr; 39 | ctx.admin = { username, password }; 40 | ctx.runtimeId = runtimeId; 41 | ctx.set('x-nohost-base-url', baseUrl); 42 | const { path: pathname } = ctx; 43 | if (pathname === '/' || pathname === '/index.html') { 44 | ctx.req.url = '/select.html'; 45 | } else if (pathname === '/js/nohost.js') { 46 | ctx.req.url = '/button.js'; 47 | } 48 | const origin = ctx.get('origin'); 49 | if (origin) { 50 | ctx.set('Access-Control-Allow-Origin', origin); 51 | ctx.set('Access-Control-Allow-Credentials', true); 52 | } 53 | await next(); 54 | }); 55 | setupRouter(router); 56 | app.use(router.routes()); 57 | app.use(router.allowedMethods()); 58 | app.use(serve(path.join(__dirname, '../../../../../public/'), { maxage: MAX_AGE })); 59 | server.on('request', app.callback()); 60 | }; 61 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/network.js: -------------------------------------------------------------------------------- 1 | 2 | const startWhistle = require('whistle'); 3 | const path = require('path'); 4 | const { getNohostPluginsPath, getPort, PLUGINS_DIR } = require('../util'); 5 | 6 | let result; 7 | const projectPluginsPath = [ 8 | path.join(__dirname, '../../../account'), 9 | ]; 10 | const customPluginsPath = [ 11 | path.join(getNohostPluginsPath(), 'worker_plugins'), 12 | ]; 13 | const pluginsPath = [ 14 | path.join(PLUGINS_DIR, 'whistle.nohost/lib/node_modules'), 15 | path.join(PLUGINS_DIR, 'whistle.nohost/node_modules'), 16 | ]; 17 | 18 | module.exports = () => { 19 | if (!result) { 20 | result = new Promise((resolve) => { 21 | getPort((port) => { 22 | const proxy = startWhistle({ 23 | globalData: 'Network', 24 | cmdName: 'n2', 25 | port, 26 | host: '127.0.0.1', 27 | encrypted: true, 28 | storage: 'nohost_network_mode', 29 | mode: 'plugins|disableUpdateTips|hideLeftMenu', 30 | projectPluginsPath, 31 | customPluginsPath, 32 | pluginsPath, 33 | }, () => resolve(port, proxy)); 34 | }); 35 | }); 36 | } 37 | return result; 38 | }; 39 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/uiServer/openAPI.js: -------------------------------------------------------------------------------- 1 | const proxy = require('./cgi/proxy'); 2 | const accountMgr = require('../accountMgr'); 3 | const { AUTH_KEY } = require('../util'); 4 | 5 | const CGI_MAP = { 6 | 'add-env': 'rules/add', 7 | 'add-top-env': 'rules/project', 8 | 'add-top-rules': 'rules/project', 9 | 'add-rules': 'rules/add', 10 | 'modify-env': 'rules/add', 11 | 'modify-rules': 'rules/add', 12 | 'remove-env': 'rules/remove', 13 | 'remove-rules': 'rules/remove', 14 | 'rename-env': 'rules/rename', 15 | 'rename-rules': 'rules/rename', 16 | list: 'rules/list', 17 | }; 18 | 19 | const getRules = (options) => { 20 | const { account, name, env, envName } = options; 21 | let { rules = '', headers } = accountMgr.getRules(account || name, env || envName); 22 | if (headers) { 23 | headers = JSON.stringify(headers); 24 | if (headers.length > 2) { 25 | rules += `\n* reqHeaders://(${headers})`; 26 | } 27 | } 28 | return rules; 29 | }; 30 | 31 | module.exports = async (ctx, next) => { 32 | let { cgiName } = ctx.params; 33 | const authKey = ctx.accountMgr.getAuthKey(); 34 | if (cgiName === 'rules') { 35 | const { query } = ctx.request; 36 | if (!authKey || authKey !== query.authKey) { 37 | ctx.status = 403; 38 | return; 39 | } 40 | ctx.body = getRules(query); 41 | return; 42 | } 43 | cgiName = CGI_MAP[cgiName]; 44 | const name = ctx.get('x-nohost-account-name'); 45 | const account = cgiName && ctx.accountMgr.getAccount(name); 46 | if (!account) { 47 | return; 48 | } 49 | const curAuthKey = authKey && ctx.get('x-nohost-auth-key'); 50 | if (!curAuthKey || (authKey !== curAuthKey && curAuthKey !== ctx.accountMgr.defaultAuthKey)) { 51 | ctx.status = 403; 52 | return; 53 | } 54 | const { req } = ctx; 55 | req.headers['x-whistle-auth-key'] = AUTH_KEY; 56 | let { url } = req; 57 | const index = url.indexOf('?'); 58 | url = index === -1 ? '' : url.substring(index); 59 | req.url = `/account/${name}/cgi-bin/${cgiName}${url}`; 60 | ctx.account = account; 61 | await proxy(ctx, next); 62 | }; 63 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/lib/whistleMgr.js: -------------------------------------------------------------------------------- 1 | const p = require('pfork'); 2 | const path = require('path'); 3 | const accountMgr = require('./accountMgr'); 4 | const { 5 | CONFIG_DATA_TYPE, 6 | AUTH_KEY, 7 | pluginConfig, 8 | } = require('./util'); 9 | 10 | const { realPort, realHost } = pluginConfig; 11 | const GUEST_AUTH = { guestName: '-' }; 12 | const WHISTLE_WORKER = path.join(__dirname, 'whistle.js'); 13 | const DELAY = 6000; 14 | const UPDATE_INTERVAL = 5000; 15 | const cache = {}; 16 | 17 | exports.fork = (account) => { 18 | if (!account) { 19 | return; 20 | } 21 | const { name, password } = account; 22 | if (cache[name]) { 23 | return cache[name]; 24 | } 25 | const getAccountData = () => { 26 | const data = { 27 | type: CONFIG_DATA_TYPE, 28 | username: account.name, 29 | password: account.password, 30 | guest: accountMgr.isEnableGuest() ? GUEST_AUTH : null, 31 | defaultRules: accountMgr.accountRules, 32 | }; 33 | return data; 34 | }; 35 | cache[name] = new Promise((resolve, reject) => { 36 | const guestName = accountMgr.isEnableGuest() ? '-' : undefined; 37 | p.fork({ 38 | authKey: AUTH_KEY, 39 | password, 40 | guestName, 41 | defaultRules: accountMgr.accountRules, 42 | storage: `whistle.nohost/${name}`, 43 | script: WHISTLE_WORKER, 44 | value: name, 45 | realPort, 46 | realHost, 47 | storageServer: accountMgr.storageServer, 48 | dnsServer: accountMgr.dnsServer, 49 | accountPluginPath: accountMgr.accountPluginPath, 50 | }, (err, result, child) => { 51 | if (err) { 52 | delete cache[name]; 53 | return reject(err); 54 | } 55 | resolve(result); 56 | accountMgr.addEnvList(name, result.envList); 57 | let timer; 58 | const updateAuth = () => { 59 | account = account && accountMgr.getAccount(account.name); 60 | if (!account) { 61 | return; 62 | } 63 | child.sendData(getAccountData()); 64 | timer = setTimeout(updateAuth, UPDATE_INTERVAL); 65 | }; 66 | timer = setTimeout(updateAuth, UPDATE_INTERVAL); 67 | child.on('exit', () => { 68 | clearTimeout(timer); 69 | delete cache[name]; 70 | }); 71 | child.on('data', (data) => { 72 | const type = data && data.type; 73 | if (type !== CONFIG_DATA_TYPE) { 74 | return; 75 | } 76 | account = account && accountMgr.getAccount(account.name); 77 | if (!account) { 78 | return; 79 | } 80 | const accountData = getAccountData(); 81 | accountData.index = data.index; 82 | child.sendData(accountData); 83 | accountMgr.addEnvList(name, data.envList); 84 | }); 85 | }); 86 | }); 87 | return cache[name]; 88 | }; 89 | exports.kill = (name) => { 90 | p.kill({ 91 | script: WHISTLE_WORKER, 92 | value: name, 93 | }, DELAY); 94 | }; 95 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/package.json: -------------------------------------------------------------------------------- 1 | { 2 | "name": "whistle.nohost", 3 | "version": "1.0.0", 4 | "whistleConfig": { 5 | "priority": 100 6 | } 7 | } 8 | -------------------------------------------------------------------------------- /lib/plugins/whistle.nohost/rules.txt: -------------------------------------------------------------------------------- 1 | * whistle.nohost://none includeFilter://reqH:x-whistle-nohost-env=/^\$\d+$/ 2 | * whistle.nohost://`${reqH.x-whistle-nohost-policy}` includeFilter://reqH:x-whistle-nohost-policy 3 | @whistle.nohost/cgi-bin/rules -------------------------------------------------------------------------------- /lib/service/cgi/export.js: -------------------------------------------------------------------------------- 1 | const fs = require('fs'); 2 | const path = require('path'); 3 | const fse = require('fs-extra'); 4 | const Limiter = require('async-limiter'); 5 | const { gzip } = require('zlib'); 6 | const { getSessionsDir, getDate, getAccessCode } = require('./util'); 7 | 8 | const promises = {}; 9 | const limiter = new Limiter({ concurrency: 10 }); 10 | const ROUTE_RE = /^[\w.:/=+-]{1,100}$/; 11 | 12 | const writeFile = (filepath, data) => { 13 | return new Promise((resolve, reject) => { 14 | fs.writeFile(filepath, data, { 15 | flag: 'wx', 16 | }, (err) => { 17 | if (err) { 18 | reject(err); 19 | } else { 20 | resolve(); 21 | } 22 | }); 23 | }); 24 | }; 25 | 26 | const createDir = (dir) => { 27 | let p = promises[dir]; 28 | if (!p) { 29 | p = fse.ensureDir(dir); 30 | promises[dir] = p; 31 | } 32 | return p; 33 | }; 34 | 35 | const compressSessions = (sessions) => { 36 | return new Promise((resolve, reject) => { 37 | limiter.push((done) => { 38 | gzip(sessions, (err, buf) => { 39 | done(); 40 | if (err) { 41 | reject(err); 42 | } else { 43 | resolve(buf); 44 | } 45 | }); 46 | }); 47 | }); 48 | }; 49 | 50 | module.exports = async (ctx) => { 51 | let { query: { username, route }, body: { name, sessions, code } } = ctx.request; 52 | if (!name || !sessions || typeof sessions !== 'string') { 53 | return; 54 | } 55 | code = getAccessCode(code); 56 | const date = getDate(); 57 | const dir = getSessionsDir(username, date, code); 58 | if (!dir) { 59 | return; 60 | } 61 | await createDir(dir); 62 | try { 63 | sessions = await compressSessions(sessions); 64 | name = encodeURIComponent(name); 65 | if (code) { 66 | name = `${name}[${code}]`; 67 | } 68 | await writeFile(path.join(dir, `${name}[gz]`), sessions); 69 | ctx.body = { ec: 0, date, username, route: route && ROUTE_RE.test(route) ? route : undefined }; 70 | } catch (e) { 71 | ctx.body = { ec: e.code === 'EEXIST' ? 1 : 2 }; 72 | } 73 | if (ctx.get('origin')) { 74 | ctx.set('Access-Control-Allow-Origin', '*'); 75 | } 76 | }; 77 | -------------------------------------------------------------------------------- /lib/service/cgi/import.js: -------------------------------------------------------------------------------- 1 | const fs = require('fs'); 2 | const path = require('path'); 3 | const { getSessionsDir, getAccessCode } = require('./util'); 4 | 5 | const isGzipFile = (gzFilePath) => { 6 | return new Promise((resolve) => { 7 | fs.stat(gzFilePath, (err, stat) => { 8 | resolve(!err && stat.isFile()); 9 | }); 10 | }); 11 | }; 12 | 13 | module.exports = async (ctx) => { 14 | let { username, name, date, code } = ctx.request.query; 15 | if (!name || typeof name !== 'string') { 16 | return; 17 | } 18 | code = getAccessCode(code); 19 | const dir = getSessionsDir(username, date, code); 20 | if (!dir) { 21 | return; 22 | } 23 | name = encodeURIComponent(name); 24 | if (code) { 25 | name = `${name}[${code}]`; 26 | } 27 | const gzFilePath = path.join(dir, `${name}[gz]`); 28 | const isGFile = await isGzipFile(gzFilePath); 29 | if (isGFile) { 30 | ctx.set('content-encoding', 'gzip'); 31 | } 32 | ctx.body = fs.createReadStream(isGFile ? gzFilePath : path.join(dir, name)); 33 | }; 34 | -------------------------------------------------------------------------------- /lib/service/cgi/util.js: -------------------------------------------------------------------------------- 1 | const os = require('os'); 2 | const path = require('path'); 3 | 4 | const USERNAME_RE = /^[a-z\d.]{1,64}$/; 5 | const DATE_RE = /^\d{8}$/; 6 | const ACCESS_CODE_RE = /^[a-z\d]{4}$/i; 7 | const leftPad = num => (num > 9 ? num : `0${num}`); 8 | let nohostPath; 9 | 10 | const getSessionsPath = () => { 11 | nohostPath = nohostPath || process.env.NOHOST_PATH || path.join(os.homedir(), '.NohostAppData'); 12 | return nohostPath; 13 | }; 14 | 15 | const getDate = () => { 16 | const now = new Date(); 17 | return `${now.getFullYear()}${leftPad(now.getMonth() + 1)}${leftPad(now.getDate())}`; 18 | }; 19 | 20 | exports.getDate = getDate; 21 | 22 | const checkUsername = (username) => { 23 | return username == null || username === '' || (typeof username === 'string' && USERNAME_RE.test(username)); 24 | }; 25 | 26 | exports.getSessionsDir = (username, date, encrypted) => { 27 | if (!DATE_RE.test(date) || !checkUsername(username)) { 28 | return; 29 | } 30 | const dir = getSessionsPath(); 31 | date = path.join(date, encrypted ? 'encrypted' : 'sessions'); 32 | return username ? path.join(dir, username, date) : path.join(dir, date); 33 | }; 34 | 35 | exports.getAccessCode = code => (ACCESS_CODE_RE.test(code) ? code.toLowerCase() : ''); 36 | -------------------------------------------------------------------------------- /lib/service/index.js: -------------------------------------------------------------------------------- 1 | const { fork } = require('pfork'); 2 | const script = require('path').join(__dirname, 'server.js'); 3 | const getPort = require('../util/getPort'); 4 | 5 | const workers = []; 6 | const getCreator = () => { 7 | let promise; 8 | return () => { 9 | if (!promise) { 10 | promise = new Promise((resolve, reject) => { 11 | getPort((port) => { 12 | fork({ value: `${port}`, script }, (err, _, child) => { 13 | if (err) { 14 | promise = null; 15 | reject(err); 16 | } else { 17 | child.once('close', () => { 18 | promise = null; 19 | }); 20 | resolve(port); 21 | } 22 | }); 23 | }); 24 | }); 25 | } 26 | return promise; 27 | }; 28 | }; 29 | 30 | module.exports = (index) => { 31 | let createServer = workers[index]; 32 | if (!createServer) { 33 | createServer = getCreator(); 34 | workers[index] = getCreator(); 35 | } 36 | return createServer(); 37 | }; 38 | -------------------------------------------------------------------------------- /lib/service/router.js: -------------------------------------------------------------------------------- 1 | const exportSessions = require('./cgi/export'); 2 | const importSessions = require('./cgi/import'); 3 | 4 | module.exports = (router) => { 5 | router.post('/cgi-bin/sessions/export', exportSessions); 6 | router.get('/cgi-bin/sessions/import', importSessions); 7 | }; 8 | -------------------------------------------------------------------------------- /lib/service/server.js: -------------------------------------------------------------------------------- 1 | 2 | const { createServer } = require('http'); 3 | const Koa = require('koa'); 4 | const onerror = require('koa-onerror'); 5 | const router = require('koa-router')(); 6 | const bodyParser = require('koa-bodyparser'); 7 | const setupRouter = require('./router'); 8 | 9 | 10 | module.exports = ({ value: port }, callback) => { 11 | const server = createServer(); 12 | const app = new Koa(); 13 | app.proxy = true; 14 | app.silent = true; 15 | if (process.env.PFORK_MODE === 'bind') { 16 | onerror(app); 17 | } 18 | setupRouter(router); 19 | app.use(bodyParser({ formLimit: '8mb' })); 20 | app.use(router.routes()); 21 | app.use(router.allowedMethods()); 22 | server.on('request', app.callback()); 23 | server.on('error', callback); 24 | server.listen(port, '127.0.0.1', callback); 25 | }; 26 | -------------------------------------------------------------------------------- /lib/util/address.js: -------------------------------------------------------------------------------- 1 | const os = require('os'); 2 | const net = require('net'); 3 | 4 | const LOCALHOST = '127.0.0.1'; 5 | let addressList = []; 6 | let serverIp; 7 | 8 | (function updateSystyemInfo() { 9 | const interfaces = os.networkInterfaces(); 10 | addressList = []; 11 | Object.keys(interfaces).forEach((name) => { 12 | const list = interfaces[name]; 13 | if (Array.isArray(list)) { 14 | list.forEach((info) => { 15 | if (!info.internal && (info.family === 'IPv4' || info.family === 4)) { 16 | serverIp = info.address; 17 | } 18 | addressList.push(info.address.toLowerCase()); 19 | }); 20 | // 支持多网卡时，以环境变量指定 serverIp 21 | const envServerIp = process.env.NOHOST_SERVER_IP; 22 | if (net.isIP(envServerIp) && addressList.includes(envServerIp)) { 23 | serverIp = envServerIp; 24 | } 25 | } 26 | }); 27 | setTimeout(updateSystyemInfo, 30000); 28 | }()); 29 | 30 | exports.getAddressList = () => addressList; 31 | 32 | const isLocalHost = (ip) => { 33 | if (!ip || typeof ip !== 'string') { 34 | return true; 35 | } 36 | return ip.length < 7 || ip === LOCALHOST; 37 | }; 38 | 39 | const isLocalAddress = (address) => { 40 | if (isLocalHost(address)) { 41 | return true; 42 | } 43 | if (address === '0:0:0:0:0:0:0:1') { 44 | return true; 45 | } 46 | address = address.toLowerCase(); 47 | if (address[0] === '[') { 48 | address = address.slice(1, -1); 49 | } 50 | return addressList.indexOf(address) !== -1; 51 | }; 52 | 53 | exports.isLocalAddress = isLocalAddress; 54 | 55 | exports.getServerIp = () => serverIp; 56 | -------------------------------------------------------------------------------- /lib/util/getPort.js: -------------------------------------------------------------------------------- 1 | const http = require('http'); 2 | 3 | let curPort = 30013; 4 | 5 | const getPort = (callback) => { 6 | const server = http.createServer(); 7 | server.on('error', () => { 8 | if (++curPort % 5 === 0) { 9 | ++curPort; 10 | } 11 | getPort(callback); 12 | }); 13 | server.listen(curPort, '127.0.0.1', () => { 14 | server.removeAllListeners(); 15 | server.close(() => callback(curPort)); 16 | }); 17 | }; 18 | 19 | module.exports = getPort; 20 | -------------------------------------------------------------------------------- /lib/util/login.js: -------------------------------------------------------------------------------- 1 | const crypto = require('crypto'); 2 | const getAuth = require('basic-auth'); 3 | 4 | const ENV_MAX_AGE = 60 * 60 * 24 * 3; 5 | 6 | const shasum = (str) => { 7 | if (typeof str !== 'string') { 8 | str = ''; 9 | } 10 | const result = crypto.createHash('sha1'); 11 | result.update(str); 12 | return result.digest('hex'); 13 | }; 14 | exports.shasum = shasum; 15 | 16 | const getLoginKey = (ctx, username, password) => { 17 | const ip = ctx.ip || '127.0.0.1'; 18 | return shasum(`${username || ''}\n${password || ''}\n${ip}`); 19 | }; 20 | 21 | exports.checkLogin = (ctx, authConf) => { 22 | const { 23 | username, 24 | password, 25 | nameKey, 26 | authKey, 27 | } = authConf; 28 | 29 | if (!username || !password) { 30 | return true; 31 | } 32 | const curName = ctx.cookies.get(nameKey); 33 | const lkey = ctx.cookies.get(authKey); 34 | const correctKey = getLoginKey(ctx, username, password); 35 | if (curName === username && correctKey === lkey) { 36 | return true; 37 | } 38 | 39 | const { name, pass } = getAuth(ctx.req) || {}; 40 | if (name === username && shasum(pass) === password) { 41 | const options = { 42 | expires: new Date(Date.now() + (ENV_MAX_AGE * 1000)), 43 | path: '/', 44 | }; 45 | ctx.cookies.set(nameKey, username, options); 46 | ctx.cookies.set(authKey, correctKey, options); 47 | return true; 48 | } 49 | 50 | ctx.status = 401; 51 | ctx.set('WWW-Authenticate', ' Basic realm=User Login'); 52 | ctx.set('Content-Type', 'text/html; charset=utf8'); 53 | ctx.body = 'Access denied, please try again.'; 54 | return false; 55 | }; 56 | -------------------------------------------------------------------------------- /lib/util/parseDomain.js: -------------------------------------------------------------------------------- 1 | 2 | const isBaseUrl = d => /^[\w-]+(?:\.[\w-]+){1,}$/.test(d); 3 | const SEP_RE = /\s*[,\s]\s*/; 4 | let domainList = []; 5 | let curDomain; 6 | 7 | const parseDomain = (domainStr) => { 8 | if (domainStr !== curDomain) { 9 | curDomain = domainStr; 10 | domainList = curDomain.trim().split(SEP_RE).filter(isBaseUrl); 11 | } 12 | return domainList; 13 | }; 14 | 15 | module.exports = (str) => { 16 | return !str || typeof str !== 'string' ? [] : parseDomain(str); 17 | }; 18 | -------------------------------------------------------------------------------- /lib/whistle.js: -------------------------------------------------------------------------------- 1 | 2 | const path = require('path'); 3 | const startWhistle = require('whistle'); 4 | const { homedir } = require('os'); 5 | const { getWhistlePath } = require('whistle'); 6 | const { getServerIp } = require('./util/address'); 7 | const parseDomain = require('./util/parseDomain'); 8 | const getPort = require('./util/getPort'); 9 | 10 | const getShadowRules = (port, domain) => { 11 | domain = parseDomain(domain); 12 | domain.unshift(`//${getServerIp()}:${port}`); 13 | return domain.map((d) => { 14 | return `${d} http://127.0.0.1:${port} enable://capture`; 15 | }).join('\n'); 16 | }; 17 | 18 | module.exports = (options, callback) => { 19 | getPort((port) => { 20 | const { 21 | domain, 22 | username, 23 | password, 24 | realPort, 25 | realHost, 26 | authKey, 27 | mainAuthKey, 28 | debugMode, 29 | storageServer, 30 | dnsServer, 31 | globalPluginPath, 32 | accountPluginPath, 33 | } = options; 34 | const projectPluginPath = [path.join(__dirname, 'plugins')]; 35 | const notUninstallPluginPath = [path.join(__dirname, '../node_modules')]; 36 | const addon = globalPluginPath ? projectPluginPath.concat(globalPluginPath) : projectPluginPath; 37 | let { baseDir } = options; 38 | if (/^~\//.test(baseDir)) { 39 | baseDir = path.join(homedir(), baseDir.substring(2)); 40 | } else if (baseDir && /^[\w.-]+$/.test(baseDir)) { 41 | baseDir = path.join(homedir(), '.nohost', baseDir); 42 | } 43 | if (baseDir) { 44 | process.env.NOHOST_BADE_DIR = baseDir; 45 | } 46 | const mode = 'proxyServer|master|x-forwarded-proto'; 47 | const proxy = startWhistle({ 48 | port, 49 | baseDir, 50 | projectPluginPath, 51 | notUninstallPluginPath, 52 | customPluginPath: path.join(getWhistlePath(), 'nohost_plugins/main_plugins'), 53 | addon, 54 | encrypted: true, 55 | username, 56 | password, 57 | realPort, 58 | realHost, 59 | cmdName: 'n2 -g', 60 | host: '127.0.0.1', 61 | shadowRules: getShadowRules(realPort, domain), 62 | dnsServer, 63 | authKey: mainAuthKey, 64 | pluginsDataMap: { 65 | nohost: { 66 | username, 67 | password, 68 | domain, 69 | realPort, 70 | realHost, 71 | authKey, 72 | storageServer, 73 | dnsServer, 74 | accountPluginPath, 75 | }, 76 | }, 77 | mode: debugMode ? mode : `${mode}|strict|rules|disableUpdateTips|proxifier|notAllowedDisablePlugins`, 78 | }, () => { 79 | const { 80 | REMOTE_ADDR_HEAD: remoteAddrHead, 81 | REMOTE_PORT_HEAD: remotePortHead, 82 | } = proxy.config; 83 | callback(null, { remoteAddrHead, remotePortHead, port }); 84 | }); 85 | }); 86 | }; 87 | -------------------------------------------------------------------------------- /packages/admin/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tencent/nohost/d8994d9709eedd980ab6aeb359224896e67e09cd/packages/admin/package.json -------------------------------------------------------------------------------- /packages/gateway/.npmignore: -------------------------------------------------------------------------------- 1 | # Logs 2 | logs 3 | *.log 4 | npm-debug.log* 5 | 6 | # Runtime data 7 | pids 8 | *.pid 9 | *.seed 10 | 11 | # Directory for instrumented libs generated by jscoverage/JSCover 12 | lib-cov 13 | 14 | # Coverage directory used by tools like istanbul 15 | coverage 16 | 17 | # nyc test coverage 18 | .nyc_output 19 | 20 | # Grunt intermediate storage (http://gruntjs.com/creating-plugins#storing-task-files) 21 | .grunt 22 | 23 | # node-waf configuration 24 | .lock-wscript 25 | 26 | # Compiled binary addons (http://nodejs.org/api/addons.html) 27 | build/Release 28 | 29 | # Dependency directories 30 | node_modules 31 | jspm_packages 32 | 33 | # Optional npm cache directory 34 | .npm 35 | 36 | # Optional REPL history 37 | .node_repl_history 38 | 39 | # test 40 | test 41 | jest 42 | 43 | # Hidden file 44 | /.* 45 | /jest.* 46 | 47 | -------------------------------------------------------------------------------- /packages/gateway/LICENSE: -------------------------------------------------------------------------------- 1 | MIT License 2 | 3 | Copyright (c) 2021 nohosts 4 | 5 | Permission is hereby granted, free of charge, to any person obtaining a copy 6 | of this software and associated documentation files (the "Software"), to deal 7 | in the Software without restriction, including without limitation the rights 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 9 | copies of the Software, and to permit persons to whom the Software is 10 | furnished to do so, subject to the following conditions: 11 | 12 | The above copyright notice and this permission notice shall be included in all 13 | copies or substantial portions of the Software. 14 | 15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 21 | SOFTWARE. 22 | -------------------------------------------------------------------------------- /packages/gateway/README.md: -------------------------------------------------------------------------------- 1 | # gateway 2 | Nohost 网关是一个用于集群部署的请求接入层，可以接收以下 5 种请求： 3 | 1. HTTP 代理请求 4 | 2. HTTPS 代理请求 5 | 3. Socks v5 代理请求 6 | 4. HTTP & HTTPS 直接请求 7 | 5. Nginx 网关反向代理请求等 8 | 9 | 并支持： 10 | 1. 异步加载证书将 HTTPS 请求转成普通 HTTP 请求 11 | 2. 设置 Whistle 规则 12 | 3. 将请求转到指定代理 13 | 14 | 支持以下几种代理服务： 15 | 1. 普通代理，如 Whistle、Fiddler、Charles 等 16 | 2. Nohost 单实例（单机部署） 17 | 3. Nohost Cluster（多机部署） 18 | 19 | 也可以直接将请求转到后台 Server 等。 20 | 21 |  22 | 23 | # 用法 24 | 在项目中引入 npm 包： 25 | ``` sh 26 | npm i --save @nohost/gateway 27 | ``` 28 | 29 | 自定义启动文件 `dispatch.js`： 30 | ``` js 31 | const startGateway = require('@nohost/gateway'); 32 | 33 | startGateway({ 34 | sniCallback, //【可选】加载证书方法的路径 35 | port, //【可选】代理端口，默认 8080 端口，也可以通过 `"host:port"` 指定网卡 36 | workers, //【可选】启动的 Worker 数，默认为系统 CPU 核数 37 | socksPort, //【可选】设置 Socks v5 代理端口，设置后会自动启一个 Socks v5 服务，默认关闭，也可以通过 `"host:port"` 指定网卡 38 | httpsPort, //【可选】设置 HTTPS Server 端口，设置后会自动启一个 HTTPS 服务，默认关闭，也可以通过 `"host:port"` 指定网卡（用户无需通过代理请求） 39 | httpPort, //【可选】设置 HTTP Server 端口，设置后会自动启一个 HTTP 服务，默认关闭，也可以通过 `"host:port"` 指定网卡（用户无需通过代理请求） 40 | mode, //【可选】同 https://github.com/avwo/whistle/blob/master/lib/config.js#L720 41 | handleRequest(req, res), //【可选】处理 HTTP 请求 42 | handleUpgrade(req, socket), //【可选】处理 WebSocket 请求 43 | handleConnect(req, socket), //【可选】处理隧道代理（Tunnel）请求 44 | }); 45 | ``` 46 | 47 | 以上参数也可以通过命令行和环境变量 `process.env` 设置，优先级 `命令行 > 环境变量 > 代码参数`: 48 | 49 | ### 命令行 50 | 例子（修改启动端口）： 51 | ``` sh 52 | node dispatch --port 8888 53 | ``` 54 | 55 | 支持以下参数： 56 | 1. `--sniCallback [sniCallback]`：同 `sniCallback` 参数 57 | 2. `--port [port]`：同 `port` 参数 58 | 3. `--workers [workers]`：同 `workers` 参数 59 | 4. `--socksPort [socksPort]`：同 `socksPort` 参数 60 | 5. `--httpsPort [httpsPort]`：同 `httpsPort` 参数 61 | 6. `--httpPort [httpPort]`：同 `httpPort` 参数 62 | 7. `--mode [mode]`：同 `mode` 参数 63 | 8. `--debug`：**启用调试模式（调试模式下只会启一个 Worker）可以查看抓包，默认关闭** 64 | > 命令行参数多了 `--debug` 选项，其它类型参数不支持 65 | 66 | ### 环境变量 67 | 支持以下环境变量： 68 | 1. `process.env.NOHOST_SNI_CALLBACK`：同 `sniCallback` 参数 69 | 2. `process.env.NOHOST_PORT`：同 `port` 参数 70 | 3. `process.env.NOHOST_WORKERS`：同 `workers` 参数 71 | 4. `process.env.NOHOST_SOCKS_PORT`：同 `socksPort` 参数 72 | 5. `process.env.NOHOST_HTTPS_PORT`：同 `httpsPort` 参数 73 | 6. `process.env.NOHOST_HTTP_PORT`：同 `httpPort` 参数 74 | 7. `process.env.NOHOST_MODE`：同 `mode` 参数 75 | 76 | 77 | # 例子 78 | 1. 代理到 Whistle 79 | 2. 代理到 Nohost 80 | 3. 代理到 Nohost 集群 81 | 82 | ### 代理到 Whistle 83 | 1. 部署 Whistle 84 | - 安装 Whistle：`npm i -g whistle` 85 | - 启动 Whistle：`w2 start`（默认端口 `8899`） 86 | - 如果用 Docker，可以通过 `w2 run -M prod` 启动 87 | 2. 转发普通请求：[test/whistle/gateway/dispatch.js](test/whistle/gateway/dispatch.js) 88 | 3. 查看抓包请求：[test/whistle/admin/dispatch.js](test/whistle/admin/dispatch.js) 89 | 90 | ### 代理到 Nohost 91 | 1. 部署 Nohost 92 | - 安装 Nohost`npm i -g nohost` 93 | - 启动 Nohost`n2 start`（默认端口 `8080`） 94 | - 如果用 Docker，可以通过 `n2 run -M prod` 启动 95 | 2. 转发普通请求：[test/nohost/gateway/dispatch.js](test/whistle/gateway/dispatch.js) 96 | 3. 查看抓包请求：[test/nohost/admin/dispatch.js](test/whistle/admin/dispatch.js) 97 | 98 | ### 代理到 Nohost 集群 99 | 1. 部署 Nohost 集群 100 | - 安装 Nohost`npm i -g nohost` 101 | - 启动 Nohost`n2 start`（默认端口 `8080`，需要在多个机器上起多个实例） 102 | - 如果用 Docker，可以通过 `n2 run -M prod` 启动 103 | 2. 转发普通请求：[test/cluster/gateway/dispatch.js](test/whistle/gateway/dispatch.js) 104 | 3. 查看抓包请求：[test/cluster/admin/dispatch.js](test/whistle/admin/dispatch.js) 105 | 106 | # License 107 | [MIT](./LICENSE) 108 | -------------------------------------------------------------------------------- /packages/gateway/index.js: -------------------------------------------------------------------------------- 1 | const startWhistle = require('whistle'); 2 | const Router = require('@nohost/router'); 3 | const { parseArgv } = require('./lib/util'); 4 | 5 | module.exports = (options) => { 6 | return new Promise((resolve) => { 7 | startWhistle(parseArgv(options), function(proxy) { 8 | resolve(proxy, this); 9 | }); 10 | }); 11 | }; 12 | 13 | module.exports.Router = Router; 14 | -------------------------------------------------------------------------------- /packages/gateway/lib/whistle.gateway/index.js: -------------------------------------------------------------------------------- 1 | const Router = require('@nohost/router'); 2 | const { notEmptyStr } = require('../util'); 3 | 4 | const getBody = (body) => { 5 | if (body == null) { 6 | return ''; 7 | } 8 | return Buffer.isBuffer(body) ? body : `${body}`; 9 | }; 10 | 11 | const handleResponse = async (res, svrRes, type) => { 12 | if (type === 'ws' || type === 'tunnel') { 13 | return; 14 | } 15 | let done; 16 | res.writeHead(svrRes.statusCode, svrRes.headers); 17 | svrRes.on('error', (err) => { 18 | if (!done) { 19 | done = true; 20 | res.emit('error', err); 21 | } 22 | }); 23 | svrRes.pipe(res); 24 | }; 25 | 26 | module.exports = async (server, { handlers }) => { 27 | const handleRequest = async (req, res, type) => { 28 | const handle = handlers[type] || handlers.http; 29 | const { 30 | headers, 31 | remoteAddress, 32 | remotePort, 33 | fullUrl, 34 | } = req.originalReq; 35 | req.headers = headers; 36 | // 让 router 转发后保留 https 协议 37 | if (req.isHttps) { 38 | req.url = fullUrl[0] === 'w' ? fullUrl.replace('ws', 'http') : fullUrl; 39 | } 40 | const result = handle && (await handle(req, res)); 41 | if (!result) { 42 | return req.passThrough(); 43 | } 44 | const { 45 | statusCode, 46 | headers: resHeaders, 47 | body, 48 | router, 49 | host, 50 | port, 51 | isUIRequest, 52 | } = result; 53 | let svrRes; 54 | if (statusCode > 0) { 55 | res.writeHead(statusCode, resHeaders); 56 | if (typeof result.pipe === 'function') { 57 | svrRes = result; 58 | } else if (typeof body?.pipe === 'function') { 59 | svrRes = body; 60 | } 61 | if (svrRes) { 62 | svrRes.on('error', err => res.emit('error', err)); 63 | return svrRes.pipe(res); 64 | } 65 | return res.end(getBody(body)); 66 | } 67 | if (notEmptyStr(host) && port > 0) { 68 | req.isUIRequest = isUIRequest; 69 | svrRes = await Router.proxy(req, res, result); 70 | await handleResponse(res, svrRes, type); 71 | return; 72 | } 73 | if (typeof router?.proxy !== 'function') { 74 | return req.passThrough(); 75 | } 76 | headers['x-whistle-remote-address'] = remoteAddress; 77 | headers['x-whistle-remote-port'] = remotePort; 78 | svrRes = await router.proxy(req, res, result); 79 | await handleResponse(res, svrRes, type); 80 | }; 81 | server.on('request', handleRequest); 82 | server.on('upgrade', (req, socket) => { 83 | handleRequest(req, socket, 'ws'); 84 | }); 85 | server.on('connect', (req, socket) => { 86 | handleRequest(req, socket, 'tunnel'); 87 | }); 88 | }; 89 | -------------------------------------------------------------------------------- /packages/gateway/lib/whistle.gateway/initial.js: -------------------------------------------------------------------------------- 1 | const loadModule = require; 2 | 3 | const getHandler = (fn) => { 4 | fn = fn && loadModule(fn); 5 | return typeof fn === 'function' ? fn : undefined; 6 | }; 7 | 8 | module.exports = async (options) => { 9 | const { 10 | handleUpgrade, 11 | handleConnect, 12 | handleRequest, 13 | } = options.config.globalData || {}; 14 | options.handlers = { 15 | http: getHandler(handleRequest), 16 | ws: getHandler(handleUpgrade), 17 | tunnel: getHandler(handleConnect), 18 | }; 19 | }; 20 | -------------------------------------------------------------------------------- /packages/gateway/lib/whistle.gateway/package.json: -------------------------------------------------------------------------------- 1 | { 2 | "name": "whistle.gateway", 3 | "version": "1.0.0", 4 | "description": "获取证书的插件" 5 | } 6 | -------------------------------------------------------------------------------- /packages/gateway/lib/whistle.gateway/rules.txt: -------------------------------------------------------------------------------- 1 | * gateway:// 2 | -------------------------------------------------------------------------------- /packages/gateway/lib/whistle.snicallback/index.js: -------------------------------------------------------------------------------- 1 | exports.sniCallback = (req, { sniCallback }) => { 2 | if (sniCallback) { 3 | return sniCallback(req.originalReq.servername, req); 4 | } 5 | return false; 6 | }; 7 | -------------------------------------------------------------------------------- /packages/gateway/lib/whistle.snicallback/initial.js: -------------------------------------------------------------------------------- 1 | const loadModule = require; 2 | 3 | module.exports = (options) => { 4 | let { sniCallback } = options.config.globalData || {}; 5 | if (sniCallback && typeof sniCallback === 'string') { 6 | sniCallback = loadModule(sniCallback); 7 | if (typeof sniCallback === 'function') { 8 | options.sniCallback = sniCallback; 9 | } 10 | } 11 | }; 12 | -------------------------------------------------------------------------------- /packages/gateway/lib/whistle.snicallback/package.json: -------------------------------------------------------------------------------- 1 | { 2 | "name": "whistle.snicallback", 3 | "version": "1.0.0", 4 | "description": "获取证书的插件" 5 | } 6 | -------------------------------------------------------------------------------- /packages/gateway/lib/whistle.snicallback/rules.txt: -------------------------------------------------------------------------------- 1 | 2 | * sniCallback://snicallback enable://capture 3 | -------------------------------------------------------------------------------- /packages/gateway/package.json: -------------------------------------------------------------------------------- 1 | { 2 | "name": "@nohost/gateway", 3 | "version": "1.0.0", 4 | "description": "Nohost cluster gateway", 5 | "scripts": { 6 | "lint": "eslint *.js lib test", 7 | "lintfix": "eslint --fix *.js lib test" 8 | }, 9 | "registry": "https://registry.npmjs.org/@nohost/gateway", 10 | "repository": { 11 | "type": "git", 12 | "url": "git+https://github.com/Tencent/nohost.git" 13 | }, 14 | "keywords": [ 15 | "whistle", 16 | "nohost", 17 | "cluster" 18 | ], 19 | "author": "avwu