griffinwebserver_v2/webserver.js

Basic webserver library to serve files in a folder. But can also have specific paths like /api/ that is handled by custom function.

const http = require('http');
const url = require('url');
const fs = require('fs');
const fsProm = fs.promises;
const path = require('path');

function isObject(a) {
  return !!a && a.constructor === Object;
}

function isArray(a) {
  return !!a && a.constructor === Array;
}

const mimeType = {
  '.7z': 'application/x-7z-compressed',
  '.acc': 'audio/aac',
  '.avi': 'video/x-msvideo',
  '.bz': 'application/x-bzip',
  '.bz2': 'application/x-bzip2',
  '.csv': 'text/csv',
  '.css': 'text/css',
  '.doc': 'application/msword',
  '.docx': 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
  '.eot': 'appliaction/vnd.ms-fontobject',
  '.epub': 'application/epub+zip',
  '.gif': 'image/gif',
  '.gz': 'application/gzip',
  '.htm': 'text/html',
  '.html': 'text/html',
  '.ico': 'image/x-icon',
  '.jpg': 'image/jpeg',
  '.jpeg': 'image/jpeg',
  '.js': 'text/javascript',
  '.json': 'application/json',
  '.mjs': 'text/javascript',
  '.mp3': 'audio/mpeg',
  '.mp4': 'video/mp4',
  '.mpeg': 'video/mpeg',
  '.odt': 'application/vnd.oasis.opendocument.text',
  '.ods': 'application/vnd.oasis.opendocument.spreadsheet',
  '.ogg': 'audio/ogg',
  '.ogv': 'video/ogg',
  '.pdf': 'application/pdf',
  '.png': 'image/png',
  '.ppt': 'application/vnd.ms-powerpoint',
  '.pptx': 'application/vnd.openxmlformats-officedocument.presentationml.presentation',
  '.rar': 'application/x-rar-compressed',
  '.sh': 'application/x-sh',
  '.svg': 'image/svg+xml',
  '.tar': 'application/x-tar',
  '.tgz': 'application/x-compressed',
  '.ttf': 'application/font-sfnt',
  '.txt': 'text/plain',
  '.wav': 'audio/wav',
  '.weba': 'audio/webm',
  '.webm': 'video/webm',
  '.webp': 'image/webp',
  '.woff': 'application/font-woff',
  '.woff2': 'application/font-woff2',
  '.xls': 'application/vnd.ms-excel',
  '.xlsx': 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
  '.xml': 'text/xml',
  '.zip': 'application/zip',
};

module.exports = class WebServer {
  constructor(webfolder = 'www', port = 80, handlers) {
    this.webfolder = webfolder;
    if (webfolder[0] !== '/') path.join(__dirname, this.webfolder);
    // if array just use the array, if object, just add the object within a array. default to empty array
    // todo validate handlers
    this.handlers = isArray(handlers) ? handlers : isObject(handlers) ? [handlers] : [];
    this.port = port;
    if (typeof this.port !== 'number') {
      console.log('Warning: port not number ' + this.port + ' using default port 80');
      this.port = 80;
    } else {
      this.port = Math.floor(this.port);
    }
  }

  // pathName is for instance '/api' handler need to be a function that is a normal webb handler that will receive (req,res)
  addHandler(pathName, handler) {
    const newHandler = {"pathName": pathName, "handler": handler};
    this.handlers.push(newHandler);
  }

  webHandler(req, res) {
    // there is no way in nodejs to detect protocol, and honestly, there is almost always a nginx proxy in between so would only mess things upp.
    const baseUrl = 'http://' + req.headers.host + '/';
    const parsedUrl = new URL(req.url, baseUrl);
    const webfolder = this.parent.webfolder;

    // look for custom handler. 
    // not using forEach as async will cause problems.
    const handlers = this.parent.handlers;
    
    for(let i = 0; i < handlers.length; i++) {
      const handler = handlers[i];
      if(handler?.pathName && handler.pathName === parsedUrl.pathname.slice(0, handler.pathName.length)) {
        return handler.handler(req,res);
      }
    }

    // default handle it as a local file
    let pathName = path.join(webfolder, parsedUrl.pathname);
    fsProm
      .stat(pathName)
      .then((stat) => {
        // if is a directory, then look for index.html
        if (stat.isDirectory()) {
          pathName = path.join(pathName, 'index.html');
        }
        // read file from file system
        return fsProm.readFile(pathName);
      })
      .then((data) => {
        // based on the URL path, extract the file extension. e.g. .js, .doc, ...
        const ext = path.parse(pathName).ext;
        // if the file is found, set Content-type and send data
        res.setHeader('Content-type', mimeType[ext] || 'application/octet-stream');
        res.end(data);
        console.log('served file ' + pathName);
      })
      .catch((err) => {
        res.statusCode = !err.code || err.code === 'ENOENT' ? 404 : 500;
        res.end('Error ' + res.statusCode + ': ' + err.message);
      });
  }

  start() {
    this.WebServer = new http.createServer(this.webHandler);
    this.WebServer.parent = this;
    console.log('Start webserver on port ' + this.port)
    this.WebServer.listen(this.port);

    this.WebServer.on('error', (err) => {
      console.log('Error: ' + err);
    });
  }
}