index.js

"use strict";

const fs = require("fs");
const config = require("config-yml");
const express = require("express");
const compression = require("compression");
const cron = require("node-cron");
const db = require("./db");
const themify = require("./utils/themify");
const axios = require("axios");
const PLACES = 7;
const urlToPing = "http://moe-counter-krfg.onrender.com/get/@Server";
const app = express();
const mysql = require("mysql2");
require("dotenv").config();
const distalk = process.env.DISTALK_DB_URL;

app.use(express.static("assets"));
app.use(compression());
app.set("view engine", "pug");

app.get("/", (req, res) => {
  const site = config.app.site || `${req.protocol}://${req.get("host")}`;
  res.render("index", { site });
});

// get the image
app.get("/get/@:name", async (req, res) => {
  const { name } = req.params;
  const { theme = "moebooru" } = req.query;
  let length = PLACES;

  // This helps with GitHub's image cache
  res.set({
    "content-type": "image/svg+xml",
    "cache-control": "max-age=0, no-cache, no-store, must-revalidate",
  });

  const data = await getCountByName(name);

  if (name === "demo") {
    res.set({
      "cache-control": "max-age=31536000",
    });
    length = 10;
  }

  // Send the generated SVG as the result
  const renderSvg = themify.getCountImage({ count: data.num, theme, length });
  res.send(renderSvg);

  console.log(
    data,
    `theme: ${theme}`,
    `ref: ${req.get("Referrer") || null}`,
    `ua: ${req.get("User-Agent") || null}`
  );
});

// JSON record
app.get("/record/@:name", async (req, res) => {
  const { name } = req.params;

  const data = await getCountByName(name);

  res.json(data);
});

app.get("/heart-beat", (req, res) => {
  res.set({
    "cache-control": "max-age=0, no-cache, no-store, must-revalidate",
  });

  res.send("alive");
  console.log("heart-beat");
});

const listener = app.listen(config.app.port || 3000, () => {
  console.log("Your app is listening on port " + listener.address().port);
});

let __cache_counter = {},
  shouldPush = false;

setInterval(() => {
  shouldPush = true;
}, 1000 * 60);

async function pushDB() {
  if (!shouldPush) return;

  try {
    shouldPush = false;
    if (Object.keys(__cache_counter).length === 0) return;

    console.log("pushDB", __cache_counter);

    const counters = Object.keys(__cache_counter).map((key) => {
      return {
        name: key,
        num: __cache_counter[key],
      };
    });

    await db.setNumMulti(counters);
    __cache_counter = {};
  } catch (error) {
    console.log("pushDB is error: ", error);
  }
}

async function getCountByName(name) {
  const defaultCount = { name, num: 0 };

  if (name === "demo") return { name, num: "0123456789" };

  try {
    if (!(name in __cache_counter)) {
      const counter = (await db.getNum(name)) || defaultCount;
      __cache_counter[name] = counter.num + 1;
    } else {
      __cache_counter[name]++;
    }

    pushDB();

    return { name, num: __cache_counter[name] };
  } catch (error) {
    console.log("get count by name is error: ", error);
    return defaultCount;
  }
}

cron.schedule("*/14 * * * *", async () => {
  try {
    const response = await axios.get(urlToPing);
    console.log(`Pinged ${urlToPing} - Status Code: ${response.status}`);
  } catch (error) {
    console.error(`Error pinging ${urlToPing}: ${error.message}`);
  }
});

async function executeMySQLQuery() {
  try {
    const connection = await mysql.createConnection(distalk);
    const [rows] = await connection.promise().query("SHOW TABLES");
    const tableNames = Object.values(rows[0]);
    console.log("MySQL Query Result - Tables:", tableNames);
    await connection.end();
  } catch (error) {
    console.error("Error executing MySQL query:", error.message);
  }
}

cron.schedule("0 0 */10 * *", async () => {
  try {
    await executeMySQLQuery();
  } catch (error) {
    console.error("Error in cron job:", error.message);
  }
});