mrfrys-node-server/modules/stuff.js

/* ----------------------------------------------------------------------------

 Question Server
 GitLab: <https://gitlab.com/MrFry/question-node-server>

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program. If not, see <https://www.gnu.org/licenses/>.

 ------------------------------------------------------------------------- */

const express = require('express')
const bodyParser = require('body-parser')
const busboy = require('connect-busboy')
const fs = require('fs')
const app = express()
let url = ''

const logger = require('../utils/logger.js')
// const utils = require('../utils/utils.js')
// const actions = require('../utils/actions.js')

const listedFiles = './public/files'

app.set('view engine', 'ejs')
app.use(express.static('public'))
app.use(busboy({
  limits: {
    fileSize: 10000 * 1024 * 1024
  }
}))
app.use(bodyParser.json())
app.use(bodyParser.urlencoded({
  limit: '5mb',
  extended: true
}))
app.use(bodyParser.json({
  limit: '5mb'
}))

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

app.get('/*.mp4', function (req, res) {
  let p = req.url.replace(/%20/g, ' ')
  let fp = p.split('?')
  fp.pop()
  const fpath = './public/files' + fp.join('/')
  if (req.query.stream) {
    const stat = fs.statSync(fpath)
    const fileSize = stat.size
    const range = req.headers.range
    if (range) {
      const parts = range.replace(/bytes=/, '').split('-')
      const start = parseInt(parts[0], 10)
      const end = parts[1]
        ? parseInt(parts[1], 10)
        : fileSize - 1
      const chunksize = (end - start) + 1
      const file = fs.createReadStream(fpath, { start, end })
      const head = {
        'Content-Range': `bytes ${start}-${end}/${fileSize}`,
        'Accept-Ranges': 'bytes',
        'Content-Length': chunksize,
        'Content-Type': 'video/mp4'
      }
      res.writeHead(206, head)
      file.pipe(res)
    } else {
      const head = {
        'Content-Length': fileSize,
        'Content-Type': 'video/mp4'
      }
      res.writeHead(200, head)
      fs.createReadStream(fpath).pipe(res)
    }
  } else {
    let fname = p.split('/')
    fname = fname.pop()
    res.render('stuff/video', {
      path: p,
      fname,
      url
    })
  }
})

app.get('/*', function (req, res) {
  let parsedUrl = req.url.replace(/%20/g, ' ')
  let curr = listedFiles + '/' + parsedUrl.substring('/'.length, parsedUrl.length).split('?')[0]
  let relPath = curr.substring('./public/files'.length, curr.length)

  if (relPath[relPath.length - 1] !== '/') { relPath += '/' }

  let t = relPath.split('/')
  let prevDir = ''
  for (let i = 0; i < t.length - 2; i++) { prevDir += t[i] + '/' }

  // curr = curr.replace(/\//g, "/");
  // relPath = relPath.replace(/\//g, "/");

  logger.LogReq(req)

  try {
    if (fs.lstatSync(curr).isDirectory()) {
      if (curr[curr.length - 1] !== '/') { curr += '/' }

      let f = []

      fs.readdirSync(curr).forEach((item) => {
        if (item[0] !== '.') {
          let res = { name: item }
          let stats = fs.statSync(curr + '/' + item)

          let fileSizeInBytes = stats['size']
          res.size = Math.round(fileSizeInBytes / 1000000)

          res.path = relPath
          if (res.path[res.path.length - 1] !== '/') { res.path += '/' }
          res.path += item

          res.mtime = stats['mtime'].toLocaleString()

          f.push(res)
        }
      })

      res.render('stuff/folders', {
        folders: f,
        dirname: relPath,
        prevDir,
        url
      })
    } else {
      let fileStream = fs.createReadStream(curr)
      fileStream.pipe(res)
    }
  } catch (e) {
    res.render('stuff/nofile', {
      missingFile: curr,
      url
    })
  }
})

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

app.get('*', function (req, res) {
  res.render('shared/404')
  res.status(404)
  // utils.AppendToFile(logger.GetDateString() + ": " + "404 GET", logFile);
})

app.post('*', function (req, res) {
  res.status(404)
  // utils.AppendToFile(logger.GetDateString() + ": " + "404 POST", logFile);
})

exports.app = app
exports.setup = (x) => {
  url = x.url
}

logger.Log('Stuff module started', logger.GetColor('yellow'))