feat: specify host

lionelhorn · May 5, 2020 · d635ce6 · d635ce6
1 parent 377386a
commit d635ce6
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 9 deletions.
diff --git a/bin/chii.js b/bin/chii.js
@@ -10,8 +10,14 @@ program
   .command('start')
   .description('starts chi server')
   .option('-p, --port <port>', 'set the port to start on. defaults to 3000', parseInt)
-  .action(command => {
-    server.start(command.port);
+  .option('-h, --host <host>', 'set the host. defaults to 0.0.0.0')
+  .option('-d, --domain <domain>', 'set the domain. defaults to localhost:port')
+  .action(({ port, host, domain }) => {
+    server.start({
+      port,
+      host,
+      domain,
+    });
   });
 
 program

diff --git a/server/index.js b/server/index.js
@@ -3,14 +3,16 @@ const Koa = require('koa');
 const router = require('./middle/router');
 const WebSocketServer = require('./lib/WebSocketServer');
 
-function start(port = 8080) {
+function start({ port = 8080, host, domain } = {}) {
+  domain = domain || 'localhost:' + port;
+
   const app = new Koa();
   const wss = new WebSocketServer();
 
-  app.use(router(wss.channelManager, port));
+  app.use(router(wss.channelManager, domain));
 
-  console.log(`starting server at http://localhost:${port}`);
-  const server = app.listen(port);
+  console.log(`starting server at ${domain}`);
+  const server = host ? app.listen(port, host) : app.listen(port, host);
 
   wss.start(server);
 }

diff --git a/server/middle/router.js b/server/middle/router.js
@@ -7,7 +7,7 @@ const pairs = require('licia/pairs');
 const reverse = require('licia/reverse');
 const map = require('licia/map');
 
-module.exports = function (channelManager, port) {
+module.exports = function (channelManager, domain) {
   const router = new Router();
 
   router.get('/', async ctx => {
@@ -21,7 +21,8 @@ module.exports = function (channelManager, port) {
     const tpl = await readTpl('index');
     ctx.body = tpl({
       targets,
-      port,
+      protocol: ctx.protocol,
+      domain,
     });
   });
 

diff --git a/server/tpl/index.hbs b/server/tpl/index.hbs
@@ -86,7 +86,7 @@
   {{else}}
     <div class="description">
       You can use this script to inject the chii target code into your web page.<br/>
-      <a target="_blank" href="http://localhost:{{port}}/target.js">http://localhost:{{port}}/target.js</a>
+      <a target="_blank" href="{{protocol}}://{{domain}}/target.js">{{protocol}}://{{domain}}/target.js</a>
     </div>  
   {{/each}}
   <script>