'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 = 7 const app = express() app.use(express.static('assets')) app.use(compression()) app.set('view engine', 'pug') app.get('/', (req, res) => { res.render('index') }); // 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 } }