kradan/kradan.js

> The tool for helping you to broadcast your code in your speaking session.

#!/usr/bin/env node
'use strict'

const express = require('express')
const app = express()
const http = require('http').Server(app)
const io = require('socket.io')(http)
const chokidar = require('chokidar')
const helper = require('./helper.js')
const ngrok = require('ngrok')
const path = require('path')
const os = require('os')
const colors = require('colors/safe')
const minimist = require('minimist')
const version = require('./package.json').version
const argv = minimist(process.argv.slice(2))
const port = argv.port || argv.p || 1112
const currentPath = argv.dir || argv.d || '.'

const ifaces = os.networkInterfaces()

colors.setTheme({
  silly: 'rainbow',
  input: 'grey',
  verbose: 'cyan',
  prompt: 'grey',
  info: 'green',
  data: 'grey',
  help: 'cyan',
  warn: 'yellow',
  debug: 'blue',
  error: 'red'
})

if (argv.h || argv.help || argv._.length) {
  console.log(`
    Online code viewer with realtime updates.

    Usage:
      $ kradan

    Options:
      -h, --help      Display help messages
      -v, --version   Display current kradan version
      -u, --public    Genrate public URL by ngrok
      -p, --port      Specify port number
      -d, --dir       Working Dir

    Examples:
      ${colors.data('# Display help messages')}
      $ kradan --help
      ${colors.data('# Start server on port 8080')}
      $ kradan -p 8080
      ${colors.data('# Start server on port 8080')}
      $ kradan --port=8080
      ${colors.data('# Display version')}
      $ kradan --version`)
  return
}

if (argv.v || argv.version) {
  console.log(version)
  return
}

/* port validation */
if (!Number.isInteger(port) || !(port >= 1024 && port <= 65535)) {
  console.log(colors.error('Please specify a valid port number.'))
  return
}

var data = {}
data.ls = helper.getDirJson(currentPath)

data.list = helper.getDirList(currentPath)

app.all('/*', function (req, res, next) {
  res.header('Access-Control-Allow-Origin', '*')
  res.header('Access-Control-Allow-Headers', 'X-Requested-With')
  next()
})

app.use('/files', express.static(currentPath, {
  dotfiles: 'allow',
  setHeaders: function (res, path, stat) {
    res.header('Content-Type', 'text/plain')
  }
}))

var staticPath = path.join(__dirname, './dist')
app.use('/', express.static(staticPath))

app.get('/ls', function (req, res) {
  let result = helper.getDirJson(currentPath)
  res.send(result)
})

io.on('connection', function (socket) {
  io.emit('list', data.ls)
})

chokidar.watch(currentPath, {
  ignored: /[\\]\.|node_modules|\.git|\.DS_Store/,
  ignorePermissionErrors: true,
  persistent: true
}).on('all', (event, path) => {
  switch (event) {
    case 'unlink':
    case 'unlinkDir':
      data.ls = helper.getDirJson(currentPath)
      io.emit('list', data.ls)
      break
    case 'add':
      if (!data.list.find(item => item === path)) {
        data.ls = helper.getDirJson(currentPath)
        io.emit('list', data.ls)
      }
      break
    case 'addDir':
      if (!data.list.find(item => item === path + '/') && path !== currentPath) {
        data.ls = helper.getDirJson(currentPath)
        io.emit('list', data.ls)
      }
      break
    case 'change':
      io.emit('change', '/' + path)
      break
    default:
      console.log(event + ' : ' + path)
  }
})

http.listen(port, async function () {
  let ngrokUrl = null
  if (argv.u || argv.public) {
    ngrokUrl = await ngrok.connect(port)
  }
  console.log(colors.warn('Starting up kradan'))
  console.log(colors.warn('Available on:'))
  Object.keys(ifaces).forEach(function (dev) {
    ifaces[dev].forEach(function (details) {
      if (details.family === 'IPv4') {
        console.log(colors.info('  http://' + details.address + `:${port}`))
      }
    })
  })
  if (argv.u || argv.public) {
    console.log(colors.warn('Available on public:'))
    console.log('  ' + colors.info(ngrokUrl))
  }
  console.log('Hit CTRL-C to stop the server')
})

// ----------------------------------------- //

if (process.platform === 'win32') {
  require('readline').createInterface({
    input: process.stdin,
    output: process.stdout
  }).on('SIGINT', function () {
    process.emit('SIGINT')
  })
}

process.on('SIGINT', function () {
  console.log('\nhttp-server stopped.')
  process.exit()
})

process.on('SIGTERM', function () {
  console.log('\nhttp-server stopped.')
  process.exit()
})