index.js

const path = require('path')
const fs = require('fs')
const url = require('url')
const mime = require('mime-types').lookup
const { SnowTransfer } = require('snowtransfer')
const bot = require('./bot.js')
const connectionHandler = require('./connectionHandler.js')
const sharp = require("sharp")
const sanitizer = require("path-sanitizer")

const options = {}

process.on("unhandledRejection", (err) => console.log(err));

try { // Use HTTPS if keys are available
  options.key = fs.readFileSync('secrets/key.pem')
  options.cert = fs.readFileSync('secrets/cert.pem')
  console.log('Found keys - using HTTPS!')
} catch (err) {
  console.log('No keys found - using HTTP!')
}

const usinghttps = !!options.key
const http = usinghttps ? require('https') : require('http')

var auth = require('./authentication.js')
auth.setHTTPS(usinghttps) // Determines whether cookies have the Secure; option

var indexpage = require('./pages/index.js');
var loginpage = require('./pages/login.js');
//var guestpage = require('./pages/guest.js');
var registerpage = require('./pages/register.js');
var forgotpage = require('./pages/forgot.js');
var channelpage = require('./pages/channel.js');
var serverpage = require('./pages/server.js');
var sendpage = require('./pages/send.js');
var { toggleTheme } = require('./pages/themeToggle.js')
var { imageProxy } = require('./pages/imageProxy.js')
var { fileProxy } = require('./pages/fileProxy.js')
var { toggleImages } = require('./pages/toggleImages.js')

bot.startBot();

function strReplace(string, needle, replacement) {
  return string.split(needle).join(replacement || '')
};
// https://stackoverflow.com/questions/1967119/why-does-javascript-replace-only-first-instance-when-using-replace

// create a server object:
const server = http.createServer(options)
connectionHandler.startWsServer(server)

async function servePage(filename, res, type, textToReplace, replacement) { // textToReplace and replacement allow for dynamic pages (not used anymore as far as I know)
  if (!type) {
    type = mime(filename)
  }
  if (filename.endsWith('/')) {
    filename += 'index.html'
  }
  fs.readFile(filename, function (err, data) {
    if (err) {
      //try to find something
      if (filename.endsWith('index.html')) {
        res.writeHead(404, { 'Content-Type': type })
        return res.end('404 Not Found')
      } else {
        servePage(filename + '/index.html', res)
        return
      }
    }
    res.writeHead(200, { 'Content-Type': type })
    if (textToReplace && replacement) {
      res.write(data.toString().replace(textToReplace, replacement))
    } else {
      res.write(data)
    }
    return res.end()
  })
}

server.on('request', async (req, res) => {
  if (req.method === 'POST') {
    let body = '' // https://itnext.io/how-to-handle-the-post-request-body-in-node-js-without-using-a-framework-cd2038b93190
    req.on('data', chunk => {
      body += chunk.toString() // convert Buffer to string
    })
    req.on('end', () => {
      const parsedurl = url.parse(req.url, true).pathname
      if (parsedurl == "/switchtheme") {
        toggleTheme(req, res)
      } else if (parsedurl == "/toggleImages") {
        toggleImages(req, res)
      } else {
        auth.handleLoginRegister(req, res, body)
      }
    })
  } else {
    const parsedurl = url.parse(req.url, true)

    const args = strReplace(parsedurl.pathname, '?', '/').split('/') // Split by / or ?
    if (args[1] === 'send') {
      const discordID = await auth.checkAuth(req, res)
      if (discordID) {
        await sendpage.sendMessage(bot, req, res, args, discordID)
      }
    } else if (args[1] === 'logout') {
      const discordID = await auth.checkAuth(req, res, true) // True = no redirect to login page
      if (discordID) {
        /*if (typeof discordID !== "object")*/ auth.logout(discordID)
      }
      res.writeHead(302, { Location: '/'/*, 'Set-Cookie': 'sessionID=' + "-" + '; SameSite=Strict; ' + (usinghttps ? 'Secure;' : '') + ' Expires=' + new Date() */ })
      res.end()
    } else if (args[1] === 'server') {
      const discordID = await auth.checkAuth(req, res)
      if (discordID) {
        await serverpage.processServer(bot, req, res, args, discordID)
      }
    } else if (args[1] === 'channels') {
      const discordID = await auth.checkAuth(req, res)
      if (discordID) {
        await channelpage.processChannel(bot, req, res, args, discordID)
      }
    } else if (args[1] === 'login.html') {
      await loginpage.processLogin(bot, req, res, args)
    } /*else if (args[1] === 'guest.html') {
      guestpage.processGuestLogin(bot, req, res, args)
    }*/ else if (args[1] === 'register.html') {
      await registerpage.processRegister(bot, req, res, args)
    } else if (args[1] === 'forgot.html') {
      await forgotpage.processForgot(bot, req, res, args)
    } else if (args[1] === 'index.html' || parsedurl.pathname === '/') {
      await indexpage.processIndex(bot, req, res, args)
    } else if (args[1] === 'longpoll.js' || args[1] === 'longpoll-xhr' || args[1] === 'api.js') { // Connection
      connectionHandler.processRequest(req, res)
    } else if (args[1] === "discord") {
      const discordID = await auth.checkAuth(req, res);
      if (discordID) {
        try {
          const query = new URLSearchParams(req.url.split("?")[1]);
          const oauthClient = new SnowTransfer(`${query.get("token_type")} ${query.get("access_token")}`);
          const user = await oauthClient.user.getSelf();
          if (user && (user.id === discordID)) {
            const guilds = await oauthClient.user.getGuilds();
            const filteredServers = guilds.filter(e => bot.client.guilds.cache.has(e.id));
            const readyServers = filteredServers.map(function (e) {
              return { serverID: e.id, discordID: discordID, icon: e.icon }
            });
            //insert onto sqlite
            auth.insertServers(readyServers);
            //GIFs
            for (const server of readyServers) {
              if (server.icon) {
                await fs.promises.mkdir(path.resolve(`pages/static/ico/server`, sanitizer(server.serverID)), { recursive: true });
                if (server.icon.startsWith("a_")) {
                  fs.promises.writeFile(path.resolve(`pages/static/ico/server`, sanitizer(`${server.serverID}/${server.icon.substring(2)}.gif`)), Buffer.from(await (await fetch(`https://cdn.discordapp.com/icons/${server.serverID}/${server.icon}.gif?size=128`)).arrayBuffer()));
                } else {
                  await sharp(await (await fetch(`https://cdn.discordapp.com/icons/${server.serverID}/${server.icon}.png?size=128`)).arrayBuffer()).toFile(path.resolve(`pages/static/ico/server/`, sanitizer(`${server.serverID}/${server.icon}.gif`)))
                }
              }
            }
            res.writeHead(302, { Location: "/server/" });
            res.write("");
            res.end();
          } else {
            res.writeHead(302, { Location: "/" });
            res.write("");
            res.end();
          }
        } catch {
          res.writeHead(302, { Location: "/" });
          res.write("");
          res.end();
        }
      }
    } else if (args[1] === 'imageProxy') {
      const fullImageUrl = `https://cdn.discordapp.com/${args[2] == "emoji" ? "emojis" : "attachments"}/${args[2] == "emoji" ? parsedurl.path.slice(18) : parsedurl.path.slice(12)}`
      await imageProxy(res, fullImageUrl);
    } else if (args[1] === 'fileProxy') {
      const filePath = parsedurl.path.slice(11)
      const fullFileUrl = `https://cdn.discordapp.com/attachments/${filePath}`
      await fileProxy(res, fullFileUrl);
    } else {
      const filename = path.resolve("pages/static", sanitizer(parsedurl.pathname));
      await servePage(filename, res)
    }
  }
})

server.listen(4000)