index.js

"use strict";
const express = require("express");
const fs = require("fs");
const path = require("path");
const nunjucks = require("nunjucks");
const bodyParser = require("body-parser");
const session = require("express-session");
const cookieParser = require("cookie-parser");
const { lstatSync, readdirSync } = require("fs");
const { join } = require("path");

const app = express();

const PORT = process.env.PORT || 8080;
const HOST = process.env.HOST || "0.0.0.0";
const USAR_AUTENTICACION = false;

nunjucks.configure("templates", {
  autoescape: true,
  express: app
});

app.use(cookieParser());
app.use(
  session({
    secret: "pecuniams123ekretsession",
    resave: false,
    saveUninitialized: true
  })
);

app.use(express.static(path.join(__dirname, "public")));
app.use(bodyParser.urlencoded({ extended: false }));
app.use(bodyParser.json());

function autenticado(req, res, next) {
  if (USAR_AUTENTICACION) {
    if (!req.session.autenticado) {
      res.send("No estás autorizado a ingresar en esta sección");
    } else {
      next();
    }
  } else {
    next();
  }
}

app.get("/", function(req, res) {
  if (req.session.autenticado) {
    res.redirect("/videos");
  } else {
    let error = req.query.error;
    res.render("index.html", { error });
  }
});

app.post("/login", function(req, res) {
  let data = req.body;

  if (data.clave === "123") {
    req.session.autenticado = true;

    req.session.save(function(err) {
      res.redirect("/videos");
    });
  } else {
    res.redirect("/?error=Clave incorrecta");
  }
});

app.get("/logout", function(req, res) {
  req.session.destroy(err => {
    if (err) {
      return console.log(err);
    }
    res.redirect("/");
  });
});

app.get("/video/:nombre", function(req, res) {
  const { nombre } = req.params;
  const es_video = filename => filename.match(/\.mp4$/) !== null;
  const quitar_extension = filename => filename.replace(/\.[^.]*$/, '');
  const ruta = `${__dirname}/videos/${nombre}`;
  const videos = fs.readdirSync(ruta).filter(es_video).map(quitar_extension);

  if (videos.length <= 1) {
    const ruta_al_video = `${nombre}/${videos[0]}`;
    res.render("video.html", { nombre, existe_video: true, ruta_al_video, progreso_id: nombre });
  } else {
    const archivos = videos.map(archivo => ({
      captura: `/captura/${nombre}/${archivo}`,
      video: `/video/${nombre}/${archivo}`,
      nombre: archivo,
      progreso_id: `${nombre}/${archivo}`,
    }));

    res.render("videos.html", { archivos });
  }
});

app.get("/video/:nombre/:video", function(req, res) {
  const { nombre, video } = req.params;
  const ruta_al_video = `${nombre}/${video}`;
  const existe_video = fs.existsSync(path.join(__dirname, 'videos', `${ruta_al_video}.mp4`));
  res.render("video.html", { nombre, existe_video, ruta_al_video, progreso_id: ruta_al_video });
});

app.get("/video-subs/:nombre/:video", autenticado, function(req, res) {
  const { nombre, video } = req.params;
  let archivo = path.join(`${__dirname}/videos/${nombre}/${video}.vtt`);

  if (fs.existsSync(archivo)) {
    res.sendFile(archivo);
  } else {
    res.sendFile(path.join(`${__dirname}/public/sin-subtitulo.vtt`));
  }
});

app.get("/video-stream/:nombre/:video", autenticado, function(req, res) {
  const { nombre, video } = req.params;
  const path = `videos/${nombre}/${video}.mp4`;
  const stat = fs.statSync(path);
  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(path, { 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(path).pipe(res);
  }
});

/* Captura para toda una colección */
app.get("/captura/:nombre", function(req, res) {
  const { nombre } = req.params;
  const archivo = obtener_archivo_de_extension(nombre, "jpg");

  if (archivo) {
    res.sendFile(`${__dirname}/videos/${nombre}/${archivo}`);
  } else {
    res.sendFile(path.join(`${__dirname}/public/sin-imagen.png`));
  }
});

/* Captura para un video específico */
app.get("/captura/:nombre/:video", function(req, res) {
  const { nombre, video } = req.params;
  const archivo = `${__dirname}/videos/${nombre}/${video}.jpg`;

  if (fs.existsSync(archivo)) {
    res.sendFile(archivo);
  } else {
    res.sendFile(path.join(`${__dirname}/public/sin-imagen.png`));
  }
});

/*
 * Recorre el directorio del enviado por parámetro y retorna
 * el primer archivo con la extensión solicitada por el segundo
 * parámetro.
 */
function obtener_archivo_de_extension(video, extension) {
  let nombre = video;
  let archivos = fs.readdirSync(`${__dirname}/videos/${nombre}/`);

  return archivos.find(a => a.toLowerCase().endsWith(`.${extension}`));
}

function es_directorio(archivo) {
  return lstatSync(archivo).isDirectory();
}

function es_visible(archivo) {
  return !archivo.startsWith(".");
}

function eliminar_prefijo_videos(nombre) {
  return nombre.replace("videos/", "");
}

function obtener_directorios(ruta) {
  return readdirSync(ruta)
    .filter(es_visible)
    .map(name => join(ruta, name))
    .filter(es_directorio)
    .map(eliminar_prefijo_videos);
}

app.get("/videos", autenticado, function(req, res) {
  const archivos = obtener_directorios("videos").map(directorio => ({
    captura: `/captura/${directorio}`,
    video: `/video/${directorio}`,
    nombre: directorio,
    progreso_id: directorio,
  }));
  res.render("videos.html", { archivos });
});

app.listen(PORT, HOST, function() {
  console.log(`Iniciando servicio en http://${HOST}:${PORT}`);
});