kopia lustrzana https://github.com/badgen/badgen.net
80 wiersze
1.8 KiB
TypeScript
80 wiersze
1.8 KiB
TypeScript
import fs from 'fs'
|
|
import path from 'path'
|
|
import http from 'http'
|
|
import serveHandler from 'serve-handler'
|
|
|
|
import serve404 from './libs/serve-404'
|
|
import serveDocs from './endpoints/docs'
|
|
|
|
const sendError = (req, res, error) => {
|
|
res.statusCode = 500
|
|
res.end(error.message)
|
|
}
|
|
|
|
const badgeHandlers = fs.readdirSync(path.join(__dirname, 'endpoints'))
|
|
.filter(name => /\.[jt]s$/.test(name))
|
|
.map(name => name.replace(/\.[jt]s$/, ''))
|
|
|
|
const isStatic = (url) => {
|
|
if (url === '/') return true
|
|
if (url.startsWith('/_next/')) return true
|
|
if (url.startsWith('/static/')) return true
|
|
if (url.startsWith('/builder')) return true
|
|
return false
|
|
}
|
|
|
|
const serveStaticHeaders = [
|
|
{
|
|
source: "**/*",
|
|
headers: [{
|
|
key: "Cache-Control",
|
|
value: "public, max-age=86400, s-maxage=604800, stale-while-revalidate=86400"
|
|
}]
|
|
}
|
|
]
|
|
|
|
const { PUB_DIR = '.' } = process.env
|
|
const server = http.createServer(async (req, res) => {
|
|
// handle statics
|
|
if (isStatic(req.url)) {
|
|
return serveHandler(req, res, {
|
|
public: path.resolve(__dirname, PUB_DIR),
|
|
headers: serveStaticHeaders
|
|
})
|
|
}
|
|
|
|
// handle `/docs/:name`
|
|
if (req.url!.startsWith('/docs/')) {
|
|
return serveDocs(req, res)
|
|
}
|
|
|
|
// handle endpoints
|
|
const handlerName = badgeHandlers.find(h => req.url!.startsWith(`/${h}/`))
|
|
|
|
try {
|
|
if (handlerName) {
|
|
const handlerPath = path.join(__dirname, 'endpoints', handlerName)
|
|
const { default: handler } = await import(handlerPath)
|
|
return handler(req, res)
|
|
}
|
|
} catch (error) {
|
|
console.error(error)
|
|
return sendError(req, res, error)
|
|
}
|
|
|
|
return serve404(req, res)
|
|
})
|
|
|
|
// Auto run
|
|
if (require.main === module) {
|
|
const port = process.env.PORT || 3000
|
|
server.listen(port)
|
|
console.log(`Badgen listening on port ${port}`)
|
|
}
|
|
|
|
process.on('unhandledRejection', e => {
|
|
console.error(500, e)
|
|
})
|
|
|
|
export default server
|