极简权限认证必须掌握【代码+原理+建议收藏】

//针对每一个请求
app.use((req, res, next)=>{
    var url = req.originalUrl
    console.log('filter - url:',url)
    // 第一步，判断是否白名单，比如登录，退出，其他不限制的资源访问。
    if(url.startsWith('/login/')||url==='/logout'||url==='/checkToken'){
        next();
        return;
    }
    // 第二步，如果不是上面的白名单资源，那么获取令牌token，没有代令牌直接拒绝
    var t = req.header('token')
    console.log('filter - request header token:', t)
    if(!t){
        res.status(400).send("禁止越权访问!!!")
        return
    }
    // console.log('filter - current session:', req.session)
    //第三步，查询网站的用户令牌注册表，是否保护当前令牌对应的用户，否则显示令牌失效，类似拿错卡了。
    var user = getUserByToken(req)
    if(!user){
       res.status(400).send("您的token失效或者未登录，请登录！")
       return 
    }
    //最后，上面的校验都通过了，直接进行后续资源处理。
    req.session.user = user
    console.log('filter - check user is ' + user)
    next();
});
// token令牌注册表
const tokenRegisterTable = {}
// 定义一个根据令牌查询并获取用户信息的方法
const getUserByToken=(req)=>{
    var t = req.header('token')
    if(!t) return false
    return tokenRegisterTable[t]
}

//这个是登录的接口，这里省去了校验密码的过程
app.get('/login/:user', function(req, res){
    const user = req.params.user
    console.log('user is ',user)
    //从express 的session会话中查看用户是否已经登录
    if(req.session.name){
        res.send('您已登录，用户'+req.session.name)
        return;
    }
    //如果没有登录，记录用户和在相应中设置两个cookie，记录用户名和token令牌
    req.session.user = user
    res.cookie('name', user)
    res.cookie('token', getTokenFromSession(user,req.session)) //token令牌用于客户端存储。看看下面的浏览器token的截图
    res.send("登录用户=" + user)
});
// 一个生成随机令牌的简单实现
const getTokenFromSession = (user, session)=>{
   var token = 'token-' + rand()
   tokenRegisterTable[token] = user
   console.log('tokenRegisterTable=',tokenRegisterTable)
   return token
}
const rand = () => new Date().getTime()+'-'+Math.floor((Math.random()+1)*1000)

const express = require('express')
const serveStatic = require('serve-static')
const session = require('express-session')
const cookieParser = require('cookie-parser')
const app = express()
const tokenRegisterTable = {}
//价值静态页面，上图中展示登录/验证访问受限资源的页面
app.use(serveStatic('./public'));
app.use(cookieParser('雷学委Key'))
app.use(session({
  secret: '雷学委Key',
  resave: true,
  saveUninitialized: true,
  cookie: { secure: true }
}))
const getUserByToken=(req)=>{
    var t = req.header('token')
    if(!t) return false
    return tokenRegisterTable[t]
}
//请求拦截器
app.use((req, res, next)=>{
    var url = req.originalUrl
    console.log('filter - url:',url)
    if(url.startsWith('/login/')||url==='/logout'||url==='/checkToken'){
        next();
        return;
    }
    var t = req.header('token')
    console.log('filter - request header token:', t)
    if(!t){
        res.status(400).send("禁止越权访问!!!")
        return
    }
    // console.log('filter - current session:', req.session)
    var user = getUserByToken(req)
    if(!user){
       res.status(400).send("您的token失效或者未登录，请登录！")
       return 
    }
    req.session.user = user
    console.log('filter - check user is ' + user)
    next();
});
//退出用户，清楚本地的访问令牌
app.get('/logout',(req,res)=>{
    var t = req.cookies['token']
    var u = req.session.user
    req.session.destroy();
    tokenRegisterTable[t] = null
    res.clearCookie('name')
    res.clearCookie('token')
    console.log('logout - session:',req.session)
    res.send(u + ' 退出登录成功!');
});
//一个简单的受限资源，当用户合法访问时，返回“访问受限资源，当前用户某某某“
app.get('/restrictData',function(req,res){
    var user = req.session.user
    res.send('访问受限资源，当前用户'+user)
});
const rand = () => new Date().getTime()+'-'+Math.floor((Math.random()+1)*1000)
const getTokenFromSession = (user, session)=>{
   var token = 'token-' + rand()
   tokenRegisterTable[token] = user
   console.log('tokenRegisterTable=',tokenRegisterTable)
   return token
}
//单独校验token的接口
app.get('/checkToken', function(req, res){
    res.send(getUserByToken(req)&&true)
});
//登录用户的校验
app.get('/login/:user', function(req, res){
    const user = req.params.user
    console.log('user is ',user)
    if(req.session.user){
        token = req.cookies['token']
        let data = '您已登录，用户'+req.session.user
        res.send(data)
        return;
    }
    req.session.user = user
    res.cookie('name', user)
    res.cookie('token', getTokenFromSession(user,req.session))
    console.log('user login :' , user)
    res.send("登录用户=" + user)
});
PORT=12024
app.listen(PORT, function(){
    console.log('app running at port ' + PORT);
});