112 lines
2.8 KiB
JavaScript
112 lines
2.8 KiB
JavaScript
|
/*eslint-disable no-console*/
|
||
|
|
const http = require('http')
|
||
|
|
const express = require('express')
|
||
|
|
const cors = require('cors')
|
||
|
|
const morgan = require('morgan')
|
||
|
|
const unpkg = require('express-unpkg')
|
||
|
|
const { fetchStats } = require('./cloudflare')
|
||
|
|
const { logPackageRequests } = require('./logging')
|
||
|
|
|
||
|
|
const fs = require('fs')
|
||
|
|
const path = require('path')
|
||
|
|
|
||
|
|
const sendHomePage = (publicDir) => {
|
||
|
|
const html = fs.readFileSync(path.join(publicDir, 'index.html'), 'utf8')
|
||
|
|
|
||
|
|
return (req, res, next) => {
|
||
|
|
fetchStats((error, stats) => {
|
||
|
|
if (error) {
|
||
|
|
next(error)
|
||
|
|
} else {
|
||
|
|
res.set('Cache-Control', 'public, max-age=60')
|
||
|
|
res.send(
|
||
|
|
html.replace('__SERVER_DATA__', JSON.stringify({
|
||
|
|
cloudflareStats: stats
|
||
|
|
}))
|
||
|
|
)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const errorHandler = (err, req, res, next) => {
|
||
|
|
res.status(500).send('<p>Internal Server Error</p>')
|
||
|
|
console.error(err.stack)
|
||
|
|
next(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
const createServer = (config) => {
|
||
|
|
const app = express()
|
||
|
|
|
||
|
|
app.disable('x-powered-by')
|
||
|
|
|
||
|
|
if (process.env.NODE_ENV === 'development')
|
||
|
|
app.use(morgan('dev'))
|
||
|
|
|
||
|
|
app.use(errorHandler)
|
||
|
|
app.use(cors())
|
||
|
|
|
||
|
|
app.get('/', sendHomePage(config.publicDir))
|
||
|
|
|
||
|
|
app.use(express.static(config.publicDir, {
|
||
|
|
maxAge: config.maxAge
|
||
|
|
}))
|
||
|
|
|
||
|
|
if (config.redisURL)
|
||
|
|
app.use(logPackageRequests(config.redisURL))
|
||
|
|
|
||
|
|
app.use(unpkg.createRequestHandler(config))
|
||
|
|
|
||
|
|
const server = http.createServer(app)
|
||
|
|
|
||
|
|
// Heroku dynos automatically timeout after 30s. Set our
|
||
|
|
// own timeout here to force sockets to close before that.
|
||
|
|
// https://devcenter.heroku.com/articles/request-timeout
|
||
|
|
if (config.timeout) {
|
||
|
|
server.setTimeout(config.timeout, (socket) => {
|
||
|
|
const message = `Timeout of ${config.timeout}ms exceeded`
|
||
|
|
|
||
|
|
socket.end([
|
||
|
|
`HTTP/1.1 503 Service Unavailable`,
|
||
|
|
`Date: ${(new Date).toGMTString()}`,
|
||
|
|
`Content-Type: text/plain`,
|
||
|
|
`Content-Length: ${message.length}`,
|
||
|
|
`Connection: close`,
|
||
|
|
``,
|
||
|
|
message
|
||
|
|
].join(`\r\n`))
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
return server
|
||
|
|
}
|
||
|
|
|
||
|
|
const defaultServerConfig = {
|
||
|
|
id: 1,
|
||
|
|
port: parseInt(process.env.PORT, 10) || 5000,
|
||
|
|
publicDir: 'public',
|
||
|
|
timeout: parseInt(process.env.TIMEOUT, 10) || 20000,
|
||
|
|
maxAge: process.env.MAX_AGE || '365d',
|
||
|
|
|
||
|
|
// for express-unpkg
|
||
|
|
registryURL: process.env.REGISTRY_URL || 'https://registry.npmjs.org',
|
||
|
|
redirectTTL: process.env.REDIRECT_TTL || 500,
|
||
|
|
autoIndex: !process.env.DISABLE_INDEX,
|
||
|
|
redisURL: process.env.REDIS_URL,
|
||
|
|
blacklist: require('./package-blacklist').blacklist
|
||
|
|
}
|
||
|
|
|
||
|
|
const startServer = (serverConfig = {}) => {
|
||
|
|
const config = Object.assign({}, defaultServerConfig, serverConfig)
|
||
|
|
const server = createServer(config)
|
||
|
|
|
||
|
|
server.listen(config.port, () => {
|
||
|
|
console.log('Server #%s listening on port %s, Ctrl+C to stop', config.id, config.port)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = {
|
||
|
|
createServer,
|
||
|
|
startServer
|
||
|
|
}
|