DweebUI/controllers/apps.js

import { readFileSync, readdirSync, renameSync, mkdirSync, unlinkSync, read, existsSync } from 'fs';
import { parse } from 'yaml';
import multer from 'multer';
import AdmZip from 'adm-zip';
import { json } from 'sequelize';

const upload = multer({storage: multer.diskStorage({
  destination: function (req, file, cb) { cb(null, 'templates/tmp/') },
  filename: function (req, file, cb) { cb(null, file.originalname) },
})});

let alert = '';
let templates_global = '';
let json_templates = '';

export const Apps = async (req, res) => {
  
  let page = Number(req.params.page) || 1;
  let template_param = req.params.template || 'default';

  json_templates = '';
  let json_files = readdirSync('templates/json/');
  for (let i = 0; i < json_files.length; i++) {
    if (json_files[i] != 'default.json') {
      let filename = json_files[i].split('.')[0];
      let link = `<li><a class="dropdown-item" href="/apps/1/${filename}">${filename}</a></li>`
      json_templates += link;
    }
  }

  let apps_list = '';
  let app_count = '';
  
  let list_start = (page - 1) * 28;
  let list_end = (page * 28);
  let last_page = '';

  let pages = `<li class="page-item"><a class="page-link" href="/apps/1/${template_param}">1</a></li>
                <li class="page-item"><a class="page-link" href="/apps/2/${template_param}">2</a></li>
                <li class="page-item"><a class="page-link" href="/apps/3/${template_param}">3</a></li>
                <li class="page-item"><a class="page-link" href="/apps/4/${template_param}">4</a></li>
                <li class="page-item"><a class="page-link" href="/apps/5/${template_param}">5</a></li>`


  let prev = '/apps/' + (page - 1) + '/' + template_param;
  let next = '/apps/' + (page + 1) + '/' + template_param;
  if (page == 1) { prev = '/apps/' + (page) + '/' + template_param; }
  if (page == last_page) { next = '/apps/' + (page) + '/' + template_param;}


  if (template_param == 'compose') {
    let compose_files = readdirSync('templates/compose/');
    
    app_count = compose_files.length;
    last_page = Math.ceil(compose_files.length/28);

    compose_files.forEach(file => {
      let compose = readFileSync(`templates/compose/${file}/compose.yaml`, 'utf8');
      let compose_data = parse(compose);
      let service_name = Object.keys(compose_data.services)
      let container = compose_data.services[service_name].container_name;
      let image = compose_data.services[service_name].image;

      let appCard = readFileSync('./views/partials/appCard.html', 'utf8');
      appCard = appCard.replace(/AppName/g, service_name);
      appCard = appCard.replace(/AppShortName/g, service_name);
      appCard = appCard.replace(/AppDesc/g, 'Compose File');
      appCard = appCard.replace(/AppLogo/g, `https://raw.githubusercontent.com/lllllllillllllillll/DweebUI-Icons/main/${service_name}.png`);
      appCard = appCard.replace(/AppCategories/g, '<span class="badge bg-orange-lt">Compose</span> ');
      appCard = appCard.replace(/AppType/g, 'compose');
      apps_list += appCard;
    });
  } else {

      let template_file = readFileSync(`./templates/json/${template_param}.json`);
      let templates = JSON.parse(template_file).templates;
      templates = templates.sort((a, b) => { if (a.name < b.name) { return -1; } });
      app_count = templates.length; 

      templates_global = templates;

      apps_list = '';
      for (let i = list_start; i < list_end && i < templates.length; i++) {
          let appCard = readFileSync('./views/partials/appCard.html', 'utf8');
          let name = templates[i].name || templates[i].title.toLowerCase();
          let title = templates[i].title || templates[i].name;
          let desc = templates[i].description.slice(0, 60) + "...";
          let description = templates[i].description.replaceAll(". ", ".\n") || "no description available";
          let note = templates[i].note ? templates[i].note.replaceAll(". ", ".\n") : "no notes available";
          let image = templates[i].image;
          let logo = templates[i].logo;
          let categories = '';
          // set data.catagories to 'other' if data.catagories is empty or undefined
          if (templates[i].categories == null || templates[i].categories == undefined || templates[i].categories == '') {
              templates[i].categories = ['Other'];
          }
          // loop through the categories and add the badge to the card
          for (let j = 0; j < templates[i].categories.length; j++) {
            categories += CatagoryColor(templates[i].categories[j]);
          }
          appCard = appCard.replace(/AppName/g, name);
          appCard = appCard.replace(/AppTitle/g, title);
          appCard = appCard.replace(/AppShortName/g, name);
          appCard = appCard.replace(/AppDesc/g, desc);
          appCard = appCard.replace(/AppLogo/g, logo);
          appCard = appCard.replace(/AppCategories/g, categories);
          appCard = appCard.replace(/AppType/g, 'json');
          apps_list += appCard;
      }
    }

  
    


  res.render("apps", {
    name: req.session.user,
    role: req.session.role,
    avatar: req.session.user.charAt(0).toUpperCase(),
    list_start: list_start + 1,
    list_end: list_end,
    app_count: app_count,
    prev: prev,
    next: next,
    apps_list: apps_list,
    alert: alert,
    template_list: '',
    json_templates: json_templates,
    pages: pages,
  });
  alert = '';
}




export const appSearch = async (req, res) => {
  let page = Number(req.params.page) || 1;

  let template_param = req.params.template || 'default';

  let template_file = readFileSync(`./templates/json/${template_param}.json`);

  let templates = JSON.parse(template_file).templates;

  templates = templates.sort((a, b) => {
      if (a.name < b.name) { return -1; }
  });
  
  let list_start = (page-1)*28;
  let list_end = (page*28);
  let last_page = Math.ceil(templates.length/28);
  let prev = '/apps/' + (page-1);
  let next = '/apps/' + (page+1);
  if (page == 1) { prev = '/apps/' + (page); }
  if (page == last_page) { next = '/apps/' + (page); }

  let search = req.body.search;

  let apps_list = '';
  let results = [];
  let [cat_1, cat_2, cat_3] = ['','',''];

  function searchTemplates(terms) {
    terms = terms.toLowerCase();
    for (let i = 0; i < templates.length; i++) {
      if (templates[i].categories) {
        if (templates[i].categories[0]) {
          cat_1 = (templates[i].categories[0]).toLowerCase();
        }
        if (templates[i].categories[1]) {
          cat_2 = (templates[i].categories[1]).toLowerCase();
        }
        if (templates[i].categories[2]) {
          cat_3 = (templates[i].categories[2]).toLowerCase();
        }
      }
      if ((templates[i].description.includes(terms)) || (templates[i].name.includes(terms)) || (templates[i].title.includes(terms)) || (cat_1.includes(terms)) || (cat_2.includes(terms)) || (cat_3.includes(terms))){
          results.push(templates[i]);
      }
    }
  }
  searchTemplates(search);

  for (let i = 0; i < results.length; i++) {
    let appCard = readFileSync('./views/partials/appCard.html', 'utf8');
    let name = results[i].name || results[i].title.toLowerCase();
    let desc = results[i].description.slice(0, 60) + "...";
    let description = results[i].description.replaceAll(". ", ".\n") || "no description available";
    let note = results[i].note ? results[i].note.replaceAll(". ", ".\n") : "no notes available";
    let image = results[i].image;
    let logo = results[i].logo;let categories = '';
    // set data.catagories to 'other' if data.catagories is empty or undefined
    if (results[i].categories == null || results[i].categories == undefined || results[i].categories == '') {
      results[i].categories = ['Other'];
    }
    // loop through the categories and add the badge to the card
    for (let j = 0; j < results[i].categories.length; j++) {
      categories += CatagoryColor(results[i].categories[j]);
    }
    appCard = appCard.replace(/AppName/g, name);
    appCard = appCard.replace(/AppShortName/g, name);
    appCard = appCard.replace(/AppDesc/g, desc);
    appCard = appCard.replace(/AppLogo/g, logo);
    appCard = appCard.replace(/AppCategories/g, categories);

    apps_list += appCard;
  }
  res.render("apps", {
      name: req.session.user,
      role: req.session.role,
      avatar: req.session.avatar,
      list_start: list_start + 1,
      list_end: list_end,
      app_count: results.length,
      prev: prev,
      next: next,
      apps_list: apps_list,
      alert: alert,
      template_list: '',
  });
}


function CatagoryColor(category) {
  switch (category) {
    case 'Other':
      return '<span class="badge bg-blue-lt">Other</span> ';
    case 'Productivity':
      return '<span class="badge bg-blue-lt">Productivity</span> ';
    case 'Tools':
      return '<span class="badge bg-blue-lt">Tools</span> ';
    case 'Dashboard':
      return '<span class="badge bg-blue-lt">Dashboard</span> ';
    case 'Communication':
      return '<span class="badge bg-azure-lt">Communication</span> ';
    case 'Media':
      return '<span class="badge bg-azure-lt">Media</span> ';
    case 'CMS':
      return '<span class="badge bg-azure-lt">CMS</span> ';
    case 'Monitoring':
      return '<span class="badge bg-indigo-lt">Monitoring</span> ';
    case 'LDAP':
      return '<span class="badge bg-purple-lt">LDAP</span> ';
    case 'Arr':
      return '<span class="badge bg-purple-lt">Arr</span> ';
    case 'Database':
      return '<span class="badge bg-red-lt">Database</span> ';
    case 'Paid':
      return '<span class="badge bg-red-lt" title="This is a paid product or contains paid features.">Paid</span> ';
    case 'Gaming':
      return '<span class="badge bg-pink-lt">Gaming</span> ';
    case 'Finance':
      return '<span class="badge bg-orange-lt">Finance</span> ';
    case 'Networking':
      return '<span class="badge bg-yellow-lt">Networking</span> ';
    case 'Authentication':
      return '<span class="badge bg-lime-lt">Authentication</span> ';
    case 'Development':
      return '<span class="badge bg-green-lt">Development</span> ';
    case 'Media Server':
      return '<span class="badge bg-teal-lt">Media Server</span> ';
    case 'Downloaders':
      return '<span class="badge bg-cyan-lt">Downloaders</span> ';
    default:
      return ''; // default to other if the category is not recognized
  }
}

export const InstallModal = async (req, res) => {
  let input = req.header('hx-trigger-name');
  let type = req.header('hx-trigger');

  if (type == 'compose') {
    let compose = readFileSync(`templates/compose/${input}/compose.yaml`, 'utf8');
    let modal = readFileSync('./views/modals/compose.html', 'utf8');
    modal = modal.replace(/AppName/g, input);
    modal = modal.replace(/COMPOSE_CONTENT/g, compose);
    res.send(modal);
    return;
  } else {
      

    let result = templates_global.find(t => t.name == input);

    let name = result.name || result.title.toLowerCase();
    let short_name = name.slice(0, 25) + "...";
    let desc = result.description.replaceAll(". ", ".\n") || "no description available";
    let short_desc = desc.slice(0, 60) + "...";
    let modal_name = name.replaceAll(" ", "-");
    let form_id = name.replaceAll("-", "_");
    let note = result.note ? result.note.replaceAll(". ", ".\n") : "no notes available";
    let command = result.command ? result.command : "";
    let command_check = command ? "checked" : "";
    let privileged = result.privileged || "";
    let privileged_check = privileged ? "checked" : "";
    let repository = result.repository || "";
    let image = result.image || "";
    let net_host, net_bridge, net_docker = '';
    let net_name = 'AppBridge';
    let restart_policy = result.restart_policy || 'unless-stopped';
    
    switch (result.network) {
      case 'host':
        net_host = 'checked';
        break;
      case 'bridge':
        net_bridge = 'checked';
        net_name = result.network;
        break;
      default:
        net_docker = 'checked';
    }

    if (repository != "") {
      image = (`${repository.url}/raw/master/${repository.stackfile}`);
    }

    let [ports_data, volumes_data, env_data, label_data] = [[], [], [], []];

    for (let i = 0; i < 12; i++) {
      
      // Get port details
      try {
        let ports = result.ports[i];
        let port_check = ports ? "checked" : "";
        let port_external = ports.split(":")[0] ? ports.split(":")[0] : ports.split("/")[0];
        let port_internal = ports.split(":")[1] ? ports.split(":")[1].split("/")[0] : ports.split("/")[0];
        let port_protocol = ports.split("/")[1] ? ports.split("/")[1] : "";

        // remove /tcp or /udp from port_external if it exists
        if (port_external.includes("/")) {
          port_external = port_external.split("/")[0];
        }
        
        ports_data.push({
          check: port_check,
          external: port_external,
          internal: port_internal,
          protocol: port_protocol
        });
      } catch {
          ports_data.push({
            check: "",
            external: "",
            internal: "",
            protocol: ""
          });
      }

      // Get volume details
      try {
        let volumes = result.volumes[i];
        let volume_check = volumes ? "checked" : "";
        let volume_bind = volumes.bind ? volumes.bind : "";
        let volume_container = volumes.container ? volumes.container.split(":")[0] : "";
        let volume_readwrite = "rw";

        if (volumes.readonly == true) {
          volume_readwrite = "ro";
        }

        volumes_data.push({
          check: volume_check,
          bind: volume_bind,
          container: volume_container,
          readwrite: volume_readwrite
        });
      } catch {
        volumes_data.push({
          check: "",
          bind: "",
          container: "",
          readwrite: ""
        });
      }

      // Get environment details
      try {
        let env = result.env[i];
        let env_check = "";
        let env_default = env.default ? env.default : "";
        if (env.set) { env_default = env.set;}
        let env_description = env.description ? env.description : "";
        let env_label = env.label ? env.label : "";
        let env_name = env.name ? env.name : "";

        env_data.push({
          check: env_check,
          default: env_default,
          description: env_description,
          label: env_label,
          name: env_name
        });
      } catch {
        env_data.push({
          check: "",
          default: "",
          description: "",
          label: "",
          name: ""
        });
      }

      // Get label details
      try {
        let label = result.labels[i];
        let label_check = "";
        let label_name = label.name ? label.name : "";
        let label_value = label.value ? label.value : "";

        label_data.push({
          check: label_check,
          name: label_name,
          value: label_value
        });
      } catch {
        label_data.push({
          check: "",
          name: "",
          value: ""
        });
      }

    }
    
    let modal = readFileSync('./views/modals/json.html', 'utf8');
    modal = modal.replace(/AppName/g, name);
    modal = modal.replace(/AppNote/g, note);
    modal = modal.replace(/AppImage/g, image);
    modal = modal.replace(/RestartPolicy/g, restart_policy);
    modal = modal.replace(/NetHost/g, net_host);
    modal = modal.replace(/NetBridge/g, net_bridge);
    modal = modal.replace(/NetDocker/g, net_docker);
    modal = modal.replace(/NetName/g, net_name);
    modal = modal.replace(/ModalName/g, modal_name);
    modal = modal.replace(/FormId/g, form_id);
    modal = modal.replace(/CommandCheck/g, command_check);
    modal = modal.replace(/CommandValue/g, command);
    modal = modal.replace(/PrivilegedCheck/g, privileged_check);


    for (let i = 0; i < 12; i++) {
      modal = modal.replaceAll(`Port${i}Check`, ports_data[i].check);
      modal = modal.replaceAll(`Port${i}External`, ports_data[i].external);
      modal = modal.replaceAll(`Port${i}Internal`, ports_data[i].internal);
      modal = modal.replaceAll(`Port${i}Protocol`, ports_data[i].protocol);

      modal = modal.replaceAll(`Volume${i}Check`, volumes_data[i].check);
      modal = modal.replaceAll(`Volume${i}Bind`, volumes_data[i].bind);
      modal = modal.replaceAll(`Volume${i}Container`, volumes_data[i].container);
      modal = modal.replaceAll(`Volume${i}RW`, volumes_data[i].readwrite);
      
      modal = modal.replaceAll(`Env${i}Check`, env_data[i].check);
      modal = modal.replaceAll(`Env${i}Default`, env_data[i].default);
      modal = modal.replaceAll(`Env${i}Description`, env_data[i].description);
      modal = modal.replaceAll(`Env${i}Label`, env_data[i].label);
      modal = modal.replaceAll(`Env${i}Name`, env_data[i].name);

      modal = modal.replaceAll(`Label${i}Check`, label_data[i].check);
      modal = modal.replaceAll(`Label${i}Name`, label_data[i].name);
      modal = modal.replaceAll(`Label${i}Value`, label_data[i].value);
    }
  res.send(modal);
  }
}


export const LearnMore = async (req, res) => {
  let name = req.header('hx-trigger-name');
  let id = req.header('hx-trigger');
  let result = templates_global.find(t => t.name == name);
  
  let modal = readFileSync('./views/modals/learnmore.html', 'utf8');
  modal = modal.replace(/AppName/g, result.title);
  modal = modal.replace(/AppDesc/g, result.description);

  res.send(modal);
}


export const ImportModal = async (req, res) => {
  let modal = readFileSync('./views/modals/import.html', 'utf8');
  res.send(modal);
}


export const Upload = (req, res) => {
  upload.array('files', 10)(req, res, () => {

    alert = `<div class="alert alert-success alert-dismissible mb-0 py-2" role="alert">
              <div class="d-flex">
                <div><svg xmlns="http://www.w3.org/2000/svg" class="icon alert-icon" width="24" height="24" viewBox="0 0 24 24" stroke-width="2" stroke="currentColor" fill="none" stroke-linecap="round" stroke-linejoin="round"><path stroke="none" d="M0 0h24v24H0z" fill="none"></path><path d="M5 12l5 5l10 -10"></path></svg> </div>
                <div>Template(s) Uploaded!</div>
              </div>
              <a class="btn-close" data-bs-dismiss="alert" aria-label="close" style="padding-top: 0.5rem;"></a>
            </div>`;

            
    let exists_alert = `<div class="alert alert-danger alert-dismissible mb-0 py-2" role="alert">
              <div class="d-flex">
              <div><svg xmlns="http://www.w3.org/2000/svg" class="icon alert-icon" width="24" height="24" viewBox="0 0 24 24" stroke-width="2" stroke="currentColor" fill="none" stroke-linecap="round" stroke-linejoin="round"><path stroke="none" d="M0 0h24v24H0z" fill="none"></path><path d="M5 12l5 5l10 -10"></path></svg> </div>
              <div>Template already exists</div>
            </div>
            <a class="btn-close" data-bs-dismiss="alert" aria-label="close" style="padding-top: 0.5rem;"></a>
          </div>`;
    
    let files = readdirSync('templates/tmp/');

    for (let i = 0; i < files.length; i++) {
      if (files[i].endsWith('.zip')) {
        let zip = new AdmZip(`templates/tmp/${files[i]}`);
        zip.extractAllTo('templates/compose', true);
        unlinkSync(`templates/tmp/${files[i]}`);
      } else if (files[i].endsWith('.json')) {
        if (existsSync(`templates/json/${files[i]}`)) {
          unlinkSync(`templates/tmp/${files[i]}`);
          alert = exists_alert;
          res.redirect('/apps');
          return;
        }
        renameSync(`templates/tmp/${files[i]}`, `templates/json/${files[i]}`);
      } else if (files[i].endsWith('.yml')) {
        let compose = readFileSync(`templates/tmp/${files[i]}`, 'utf8');
        let compose_data = parse(compose);
        let service_name = Object.keys(compose_data.services);
        if (existsSync(`templates/compose/${service_name}`)) {
          unlinkSync(`templates/tmp/${files[i]}`);
          alert = exists_alert;
          res.redirect('/apps');
          return;
        }
        mkdirSync(`templates/compose/${service_name}`);
        renameSync(`templates/tmp/${files[i]}`, `templates/compose/${service_name}/compose.yaml`);
      } else if (files[i].endsWith('.yaml')) {
        let compose = readFileSync(`templates/tmp/${files[i]}`, 'utf8');
        let compose_data = parse(compose);
        let service_name = Object.keys(compose_data.services);
        if (existsSync(`templates/compose/${service_name}`)) {
          unlinkSync(`templates/tmp/${files[i]}`);
          alert = exists_alert;
          res.redirect('/apps');
          return;
        }
        mkdirSync(`templates/compose/${service_name}`);
        renameSync(`templates/tmp/${files[i]}`, `templates/compose/${service_name}/compose.yaml`);
      } else {
        // unsupported file type
        unlinkSync(`templates/tmp/${files[i]}`);
      }
    }   
    res.redirect('/apps');
  });
};