nodejs渐入佳境[30]-mongodb+express+middleware绑定用户权限

发布时间：2020-06-21 09:48:17 作者：jonson_jackson
来源：网络阅读：197

修改todos模型

增加_creator，将用户绑定起来

//模版
var Todo = mongoose.model('Todo',{
    text:{
      type:String,  //类型
      required:true, //必须要有
      minlength:1, //最小长度
      trim:true   //去除空格
    },
    completed:{
      type:Boolean,
      default:false  //默认值
    },
    completedAt:{
      type:Number,
      default:null
    },
    _creator: {
      type: mongoose.Schema.Types.ObjectId,
      required: true
    }
});

添加权限控制

添加权限控制并且存储了用户的id。
用户访问/todos 必须要有authenticate这个middleware。通过以后，说明用户是可信的。就可以保存todo到数据库中。并且将todo与创建的用户链接在了一起

//express route
app.post('/todos',authenticate,(req,res)=>{
//  console.log(req.body);

    //建立对象document
    var todo = new Todo({
        text:req.body.text,
        _creator:req.user._id
    });
    //保存
      todo.save().then((doc)=>{
      res.send(doc);
    },(e)=>{
        res.status(400).send(e);
    });

})
//获取所有属性

app.get('/todos', authenticate,(req, res) => {
  Todo.find({
      _creator:req.user._id
  }).then((todos) => {
    res.send({todos});
  }, (e) => {
    res.status(400).send(e);
  })
});

完整代码

var mongoose = require('mongoose');
var express = require('express');
var bodyParser = require('body-parser');
const {ObjectID} = require('mongodb');
var _ = require('lodash');

const validator = require('validator');


const jwt = require('jsonwebtoken');
const bcrypt = require('bcryptjs');
//app
var app = express();
const port = process.env.PORT || 3000;
//express middleware  Jonson对象与字符串转换。
app.use(bodyParser.json());

//
mongoose.Promise = global.Promise;
//连接mogodb
mongoose.connect(process.env.MONGODB_URI || 'mongodb://localhost:27017/TodoApp');

//模版
var Todo = mongoose.model('Todo',{
    text:{
      type:String,  //类型
      required:true, //必须要有
      minlength:1, //最小长度
      trim:true   //去除空格
    },
    completed:{
      type:Boolean,
      default:false  //默认值
    },
    completedAt:{
      type:Number,
      default:null
    },
    _creator: {
      type: mongoose.Schema.Types.ObjectId,
      required: true
    }
});





var UserSchema = new mongoose.Schema({
  email: {
    type: String,
    required: true,
    trim: true,
    minlength: 1,
    unique: true,
    validate: {
      validator: validator.isEmail,
      message: '{VALUE} is not a valid email'
    }
  },
  password: {
    type: String,
    require: true,
    minlength: 6
  },
  tokens: [{
    access: {
      type: String,
      required: true
    },
    token: {
      type: String,
      required: true
    }
  }]
});

UserSchema.methods.toJSON = function () {
  var user = this;
  var userObject = user.toObject();

  return _.pick(userObject, ['_id', 'email']);
};

UserSchema.methods.generateAuthToken = function () {
  var user = this;
  var access = 'auth';
  var token = jwt.sign({_id: user._id.toHexString(), access}, 'abc123').toString();

  user.tokens = user.tokens.concat([{access,token}]);

  return user.save().then(() => {
    return token;  //返回token
  });
};


UserSchema.methods.removeToken = function (token) {
  var user = this;

  return user.update({
    $pull: {
      tokens: {token}
    }
  });
};



UserSchema.statics.findByToken = function (token) {
  var User = this;
  var decoded;

  try {
    decoded = jwt.verify(token, 'abc123');
  } catch (e) {
    return Promise.reject();
  }

  return User.findOne({
    '_id': decoded._id,
    'tokens.token': token,
    'tokens.access': 'auth'
  });
};


UserSchema.statics.findByCredentials = function (email, password) {
  var User = this;

  return User.findOne({email}).then((user) => {
    if (!user) {
      return Promise.reject();
    }

    return new Promise((resolve, reject) => {
      // Use bcrypt.compare to compare password and user.password
      bcrypt.compare(password, user.password, (err, res) => {
        if (res) {
          resolve(user);
        } else {
          reject();
        }
      });
    });
  });
};


//在保存之前执行操作。

UserSchema.pre('save', function (next) {
  var user = this;

//保存时对于密码的更新
  if (user.isModified('password')) {
    bcrypt.genSalt(10, (err, salt) => {
      //密码变为了hash
      bcrypt.hash(user.password, salt, (err, hash) => {
        user.password = hash;
        next();
      });
    });
  } else {
    next();
  }
});


var User = mongoose.model('User', UserSchema);










//auth middlewire

var authenticate = (req, res, next) => {
  var token = req.header('x-auth');

  User.findByToken(token).then((user) => {
    if (!user) {
      return Promise.reject();
    }

    req.user = user;
    req.token = token;
    next();
  }).catch((e) => {
    res.status(401).send();
  });
};



app.delete('/users/me/token', authenticate, (req, res) => {
  req.user.removeToken(req.token).then(() => {
    res.status(200).send();
  }, () => {
    res.status(400).send();
  });
});

app.get('/users/me', authenticate, (req, res) => {
  res.send(req.user);
});

// POST /users/login {email, password}
app.post('/users/login', (req, res) => {
  var body = _.pick(req.body, ['email', 'password']);

  User.findByCredentials(body.email, body.password).then((user) => {
    return user.generateAuthToken().then((token) => {
      res.header('x-auth', token).send(user);
    });
  }).catch((e) => {
    res.status(400).send();
  });
});

// POST /users
app.post('/users', (req, res) => {
  var body = _.pick(req.body, ['email', 'password']);
  var user = new User(body);

  user.save().then(() => {
    return user.generateAuthToken(); //调用方法，产生auth token并保存。
  }).then((token) => {
    res.header('x-auth', token).send(user); //设置了响应头
  }).catch((e) => {
    res.status(400).send(e);
  })
});

//express route
app.post('/todos',authenticate,(req,res)=>{
//  console.log(req.body);

    //建立对象document
    var todo = new Todo({
        text:req.body.text,
        _creator:req.user._id
    });
    //保存
      todo.save().then((doc)=>{
      res.send(doc);
    },(e)=>{
        res.status(400).send(e);
    });

})
//获取所有属性

app.get('/todos', authenticate,(req, res) => {
  Todo.find({
      _creator:req.user._id
  }).then((todos) => {
    res.send({todos});
  }, (e) => {
    res.status(400).send(e);
  })
});


//查询id
app.get('/todos/:id', (req, res) => {
  var id = req.params.id;

  if (!ObjectID.isValid(id)) {
    return res.status(404).send();
  }

  Todo.findById(id).then((todo) => {
    if (!todo) {
      return res.status(404).send();
    }

    res.send({todo});
  }).catch((e) => {
    res.status(400).send();
  });
});


//删除
app.delete('/todos/:id', (req, res) => {
  var id = req.params.id;

  if (!ObjectID.isValid(id)) {
    return res.status(404).send();
  }

  Todo.findByIdAndRemove(id).then((todo) => {
    if (!todo) {
      return res.status(404).send();
    }

    res.send({todo});
  }).catch((e) => {
    res.status(400).send();
  });
});

//更新
app.patch('/todos/:id', (req, res) => {
  var id = req.params.id;
  var body = _.pick(req.body, ['text', 'completed']);

  if (!ObjectID.isValid(id)) {
    return res.status(404).send();
  }

  if (_.isBoolean(body.completed) && body.completed) {
    body.completedAt = new Date().getTime();
  } else {
    body.completed = false;
    body.completedAt = null;
  }

  Todo.findByIdAndUpdate(id, {$set: body}, {new: true}).then((todo) => {
    if (!todo) {
      return res.status(404).send();
    }

    res.send({todo});
  }).catch((e) => {
    res.status(400).send();
  })
});




//监听
app.listen(port,()=>{
    console.log(`Start on port ${port}`);
});
module.exports = {
   app,
   Todo
}

测试

1、打开mongoDB > ./mongod -dbpath /Users/jackson/Downloads/mongodb-data
2、运行 >node postman.js
3、打开postman 选择post 输入 >localhost:3000/users  保存user
Body中填入：
{
	"email": "zhuimengshaonian07@gmail.com",
	"password" : "123abc!"
}
返回：
{
    "_id": "5bfe716591e78c6a4ad8c164",
    "email": "zhuimengshaonian07@gmail.com"
}

header:
x-auth →eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJfaWQiOiI1YmZlNzE2NTkxZTc4YzZhNGFkOGMxNjQiLCJhY2Nlc3MiOiJhdXRoIiwiaWF0IjoxNTQzNDAxODI5fQ.wOKNzkls_w_jA5YVkCo0r9gFZ4-KtD6GarRiCDpAPr8

4、 选择post 输入 >localhost:3000/todos


Body中填入：
{
	"text": "zhuimengshaonian07@gmail.com",
}

header附带返回：
x-auth →eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJfaWQiOiI1YmZlNzE2NTkxZTc4YzZhNGFkOGMxNjQiLCJhY2Nlc3MiOiJhdXRoIiwiaWF0IjoxNTQzNDAxODI5fQ.wOKNzkls_w_jA5YVkCo0r9gFZ4-KtD6GarRiCDpAPr8

返回：
{
    "completed": false,
    "completedAt": null,
    "_id": "5c00a66978dd038d39dc4b89",
    "text": "zhuimengshaonian07@gmail.com",
    "_creator": "5c00a58c78dd038d39dc4b87",
    "__v": 0
}

本文链接： https://dreamerjonson.com/2018/11/30/node-30-bind-user-auth/
版权声明： 本博客所有文章除特别声明外，均采用 CC BY 4.0 CN协议许可协议。转载请注明出处！

nodejs渐入佳境[30]-mongodb+express+middleware绑定用户权限

nodejs渐入佳境[30]-mongodb+express+middleware绑定用户权限

修改todos模型

添加权限控制

完整代码

测试

相关阅读