water-plant-uml/src/server.js

CLI tool for live-reloading and/or exporting PlantUML diagrams using the default PlantUML server or a local Docker server.

const http = require('http');
const livereload = require('livereload');
const { 
  buildHtmlPage,
} = require('./helpers');
const opn = require('better-opn');

const {
  LIVE_RELOAD_PORT = '8088', // Port live-reload server can be accessed from in browser
  SCRIPT_PATH,
  FILE_NAME,
} = process.env;

const liveReloadServer = livereload.createServer({
  applyImgLive: true,
  exts: ['puml'],
}, () => {
  const liveUrl = `http://localhost:${LIVE_RELOAD_PORT}`;
  console.log(`Live-reloading ${FILE_NAME} on ${liveUrl}`);
  if (process.env.OPEN) {
    opn(liveUrl);
  }
});
liveReloadServer.watch(SCRIPT_PATH);

const server = http.createServer((req, res) => {
  let html;
  try {
    html = buildHtmlPage();
  } catch (error) {
    console.error(error);
    res.writeHead(500, {'Content-Type': 'application/json'}); 
    return res.end(JSON.stringify({
      error: 'Something went wrong when trying to generate html of PlantUML diagram.',
    }));
  }
  res.writeHead(200, {
    'Content-Type': 'text/html',
    'Content-Length': html.length,
    'Expires': new Date().toUTCString(),
  });
  res.end(html);
});

server.listen(LIVE_RELOAD_PORT);