screepsmod-auth
Version:
## This adds user/pass auth to the screeps private server
96 lines (86 loc) • 3.43 kB
JavaScript
const path = require('path')
const express = require('express')
const bodyParser = require('body-parser')
const cookieParser = require('cookie-parser')
const passport = require('passport')
const GitHubStrategy = require('passport-github').Strategy
const authlib = require('@screeps/backend/lib/authlib')
const auth = require('@screeps/backend/lib/game/api/auth')
const app = new express.Router()
module.exports = function (config) {
const clientId = process.env.GITHUB_CLIENT_ID || (config.auth.screepsrc.github && config.auth.screepsrc.github.clientId) || null
const clientSecret = process.env.GITHUB_CLIENT_SECRET || (config.auth.screepsrc.github && config.auth.screepsrc.github.clientSecret) || null
const enabled = !!(clientId && clientSecret)
config.auth.info.github = enabled
let registered = false
app.use(cookieParser())
app.get('/', (req, res, next) => {
let { token, returnUrl } = req.query
if (token) res.cookie('auth_token', token)
if (!registered) {
registered = true
let proto = req.get('X-Forwarded-Proto') || req.protocol || 'http'
let baseUrl = returnUrl || `${proto}://${req.get('host')}`
passport.use('github', new GitHubStrategy({
callbackURL: baseUrl + '/api/auth/github/return',
clientID: clientId,
clientSecret
}, (accessToken, refreshTokem, profile, done) => done(null, profile.id)))
}
setTimeout(next, 100)
}, passport.authenticate('github'))
app.get('/return', passport.authenticate('github', { failureRedirect: '/' }), (req, res) => {
let user = null
let token = req.cookies.auth_token
res.clearCookie('auth_token')
if (token) user = authlib.checkToken(token)
githubFindOrCreateUser(user, req.user)
.then(user => authlib.genToken(user._id))
.then(token => {
let json = JSON.stringify({ username: req.user.username, token })
res.writeHead(200, { 'Content-Type':'text/html' }).end(`<html><body><script type="text/javascript">opener.postMessage(JSON.stringify(${json}), '*');window.close();</script></body></html>`)
})
.catch(err => res.end('Failed to auth'))
})
// })
config.auth.router.use('/api/auth/github', app)
config.auth.router.post('/api/user/unlink-github', auth.tokenAuth, (req, res) => {
if (!req.user) return
config.common.storage.db.users.update({ _id: req.user._id }, { $unset: { github: true }})
res.json({ ok: 1 })
})
function githubFindOrCreateUser(user, id) {
let { db, env } = config.common.storage
if (user) {
return user.then((user)=>{
return db.users.update({ _id: user._id }, { $set: { github: { id }}})
.then(()=>user)
})
}
return db.users.findOne({ 'github.id': id })
.then((user) => {
if (user) return user
user = {
github: { id },
cpu: 100,
cpuAvailable: 0,
registeredDate: new Date(),
money: 0,
gcl: 0
};
return db.users.insert(user)
.then(result => {
user = result;
return db['users.code'].insert({
user: user._id,
modules: {main: ''},
branch: 'default',
activeWorld: true,
activeSim: true
})
})
.then(() => env.set('scrUserMemory:'+user._id, JSON.stringify({})))
.then(() => user)
})
}
}