1 files changed, 38 insertions, 5 deletions

diff --git a/utils/themify.js b/utils/themify.js
index c440cd8..2db9886 100644
--- a/utils/themify.js
+++ b/utils/themify.js
@@ -3,6 +3,7 @@
 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')
 
@@ -12,8 +13,15 @@ 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
-    themeList[theme][name] = convertToDatauri(path.resolve(themePath, theme, img))
+    const { width, height } = sizeOf(imgPath)
+
+    themeList[theme][name] = {
+      width,
+      height,
+      data: convertToDatauri(imgPath)
+    }
   })
 })
 
@@ -24,11 +32,36 @@ function convertToDatauri(path){
   return `data:${mime};base64,${base64}`
 }
 
-function wrap(num, theme='konachan'){
-  if(!(theme in themeList)) theme = 'konachan'
-  return themeList[theme][num]
+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">
+    <title>Moe Count</title>
+    <g>
+      ${parts}
+    </g>
+</svg>
+`
 }
 
 module.exports = {
-  wrap
+  getCountImage
 }
\ No newline at end of file