1
0
mirror of https://github.com/journey-ad/Moe-Counter.git synced 2024-11-27 17:00:51 +01:00
Moe-Counter/index.js
d1y 4daac0781f add return json data REST api
获取@name数据(json)
2020-08-22 00:24:02 +08:00

103 lines
2.2 KiB
JavaScript

'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')
});
const getCountByName = async name=> {
// console.log(name)
if (name === 'demo') return { num: '0123456789', name }
try {
const counter = await db.getNum(name) || { name, num: 0 }
const r = counter.num + 1
db.setNum(counter.name, r)
return counter
} catch (error) {
console.log("get count by name is error: ", error)
const errorDefaultCount = 0
return errorDefaultCount
}
}
// the rest api get data
// link: https://www.liaoxuefeng.com/wiki/1022910821149312/1105009634703392
app.get('/rest/@:name', async (req, res) => {
const name = req.params.name
try {
const data = await getCountByName(name)
res.send(data)
} catch (error) {
res.send({
num: 0,
name
})
}
})
// get the image
app.get('/get/@:name', async (req, res) => {
const name = req.params.name
const theme = req.query.theme || 'moebooru'
let length = PLACES, count = 0
// 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)
count = data.num
if (name === 'demo') {
res.set({
'cache-control': 'max-age=31536000'
})
length = 10
}
// Send the generated SVG as the result
const renderSvg = themify.getCountImage({ count, theme, length })
res.send(renderSvg)
})
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')
});
let port = 3000
try {
let configPort = config.app.port
if (configPort) {
port = configPort
}
} catch (error) {
throw new Error(error)
}
const listener = app.listen(port, () => {
console.log('Your app is listening on port ' + listener.address().port)
})