aboutsummaryrefslogtreecommitdiff
path: root/utils/themify.js
blob: 483cdf3b32befda030eb162f3da6b388dc38f26f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
"use strict";

const fs = require("fs");
const path = require("path");
const mimeType = require("mime-types");
const sizeOf = require("image-size");

const themePath = path.resolve(__dirname, "../assets/theme");

const themeList = {};

fs.readdirSync(themePath).forEach((theme) => {
  if (!(theme in themeList)) themeList[theme] = {};
  const imgList = fs.readdirSync(path.resolve(themePath, theme));
  imgList.forEach((img) => {
    const imgPath = path.resolve(themePath, theme, img);
    const name = path.parse(img).name;
    const { width, height } = sizeOf(imgPath);

    themeList[theme][name] = {
      width,
      height,
      data: convertToDatauri(imgPath),
    };
  });
});

function convertToDatauri(path) {
  const mime = mimeType.lookup(path);
  const base64 = fs.readFileSync(path).toString("base64");

  return `data:${mime};base64,${base64}`;
}

function getCountImage({ count, theme = "moebooru", length = 7 }) {
  if (!(theme in themeList)) theme = "moebooru";

  // This is not the greatest way for generating an SVG but it'll do for now
  const countArray = count.toString().padStart(length, "0").split("");

  let x = 0,
    y = 0;
  const parts = countArray.reduce((acc, next, index) => {
    const { width, height, data } = themeList[theme][next];

    const image = `${acc}
      <image x="${x}" y="0" width="${width}" height="${height}" xlink:href="${data}" />`;

    x += width;

    if (height > y) y = height;

    return image;
  }, "");

  return `<?xml version="1.0" encoding="UTF-8"?>
<svg width="${x}" height="${y}" version="1.1" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" style="image-rendering: pixelated;">
    <title>Urushi Counter</title>
    <g>
      ${parts}
    </g>
</svg>
`;
}

module.exports = {
  getCountImage,
};