"use strict";

const fs = require("fs");
const config = require("config-yml");
const express = require("express");
const compression = require("compression");

const db = require("./db");
const themify = require("./utils/themify");

const PLACES = 6;

const app = express();

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;
  }
}