<% if (success) { %>
  <div><%= success %></div>
<% } %>
<% if (error) { %>
  <div><%= error %> </div>
<% } %>
修改 index.js ，將 app.get('/') 修改如下：

app.get('/', function (req, res) {
  res.render('index', {
    title: '主頁',
    user: req.session.user,
    success: req.flash('success').toString(),
    error: req.flash('error').toString()
  });
});

將 app.get('reg') 修改如下：

app.get('/reg', function (req, res) {
  res.render('reg', {
    title: '注冊',
    user: req.session.user,
    success: req.flash('success').toString(),
    error: req.flash('error').toString()
  });
});

現(xiàn)在運(yùn)行我們的博客，注冊成功后顯示如下：

http://wiki.jikexueyuan.com/project/express-mongodb-setup-blog/images/1.17.jpg" alt="" />

我們通過對 session 的使用實現(xiàn)了對用戶狀態(tài)的檢測，再根據(jù)不同的用戶狀態(tài)顯示不同的導(dǎo)航信息。 簡單解釋一下流程：用戶在注冊成功后，把用戶信息存入 session ，頁面跳轉(zhuǎn)到主頁顯示 注冊成功！ 的字樣。同時把 session 中的用戶信息賦給變量 user ，在渲染 index.ejs 文件時通過檢測 user 判斷用戶是否在線，根據(jù)用戶狀態(tài)的不同顯示不同的導(dǎo)航信息。

success: req.flash('success').toString() 的意思是將成功的信息賦值給變量 success， error: req.flash('error').toString() 的意思是將錯誤的信息賦值給變量 error ，然后我們在渲染 ejs 模版文件時傳遞這兩個變量來進(jìn)行檢測并顯示通知。

登錄與登出響應(yīng)

現(xiàn)在我們來實現(xiàn)用戶登錄的功能。

打開 index.js ，將 app.post('/login') 修改如下：

app.post('/login', function (req, res) {
  //生成密碼的 md5 值
  var md5 = crypto.createHash('md5'),
      password = md5.update(req.body.password).digest('hex');
  //檢查用戶是否存在
  User.get(req.body.name, function (err, user) {
    if (!user) {
      req.flash('error', '用戶不存在!'); 
      return res.redirect('/login');//用戶不存在則跳轉(zhuǎn)到登錄頁
    }
    //檢查密碼是否一致
    if (user.password != password) {
      req.flash('error', '密碼錯誤!'); 
      return res.redirect('/login');//密碼錯誤則跳轉(zhuǎn)到登錄頁
    }
    //用戶名密碼都匹配后，將用戶信息存入 session
    req.session.user = user;
    req.flash('success', '登陸成功!');
    res.redirect('/');//登陸成功后跳轉(zhuǎn)到主頁
  });
});

將 app.get('/login') 修改如下：

app.get('/login', function (req, res) {
    res.render('login', {
        title: '登錄',
        user: req.session.user,
        success: req.flash('success').toString(),
        error: req.flash('error').toString()});
});

(這樣就不會出現(xiàn) 'user is not defined' 的錯誤了)

接下來我們實現(xiàn)登出響應(yīng)。修改 app.get('/logout') 如下：

app.get('/logout', function (req, res) {
  req.session.user = null;
  req.flash('success', '登出成功!');
  res.redirect('/');//登出成功后跳轉(zhuǎn)到主頁
});

注意：通過把 req.session.user 賦值 null 丟掉 session 中用戶的信息，實現(xiàn)用戶的退出。

登錄后頁面顯示如下：

http://wiki.jikexueyuan.com/project/express-mongodb-setup-blog/images/1.18.jpg" alt="" />

登出后頁面顯示如下：

http://wiki.jikexueyuan.com/project/express-mongodb-setup-blog/images/1.19.jpg" alt="" />

至此，我們實現(xiàn)了用戶注冊與登陸的功能，并且根據(jù)用戶登錄狀態(tài)顯示不同的導(dǎo)航。

頁面權(quán)限控制

我們雖然已經(jīng)完成了用戶注冊與登陸的功能，但并不能阻止比如已經(jīng)登陸的用戶訪問 localhost:3000/reg 頁面，讀者可親自嘗試下。為此，我們需要為頁面設(shè)置訪問權(quán)限。即注冊和登陸頁面應(yīng)該阻止已登陸的用戶訪問，登出及后面我們將要實現(xiàn)的發(fā)表頁只對已登錄的用戶開放。如何實現(xiàn)頁面權(quán)限的控制呢？我們可以把用戶登錄狀態(tài)的檢查放到路由中間件中，在每個路徑前增加路由中間件，即可實現(xiàn)頁面權(quán)限控制。我們添加 checkNotLogin 和 checkLogin 函數(shù)來實現(xiàn)這個功能。

function checkLogin(req, res, next) {
  if (!req.session.user) {
    req.flash('error', '未登錄!'); 
    res.redirect('/login');
  }
  next();
}

function checkNotLogin(req, res, next) {
  if (req.session.user) {
    req.flash('error', '已登錄!'); 
    res.redirect('back');//返回之前的頁面
  }
  next();
}

checkNotLogin 和 checkLogin 用來檢測是否登陸，并通過 next() 轉(zhuǎn)移控制權(quán)，檢測到未登錄則跳轉(zhuǎn)到登錄頁，檢測到已登錄則跳轉(zhuǎn)到前一個頁面。

最終 index.js 代碼如下：

var crypto = require('crypto'),
    User = require('../models/user.js');

module.exports = function(app) {
  app.get('/', function (req, res) {
    res.render('index', {
      title: '主頁',
      user: req.session.user,
      success: req.flash('success').toString(),
      error: req.flash('error').toString()
    });
  });

  app.get('/reg', checkNotLogin);
  app.get('/reg', function (req, res) {
    res.render('reg', {
      title: '注冊',
      user: req.session.user,
      success: req.flash('success').toString(),
      error: req.flash('error').toString()
    });
  });

  app.post('/reg', checkNotLogin);
  app.post('/reg', function (req, res) {
    var name = req.body.name,
        password = req.body.password,
        password_re = req.body['password-repeat'];
    if (password_re != password) {
      req.flash('error', '兩次輸入的密碼不一致!'); 
      return res.redirect('/reg');
    }
    var md5 = crypto.createHash('md5'),
        password = md5.update(req.body.password).digest('hex');
    var newUser = new User({
        name: name,
        password: password,
        email: req.body.email
    });
    User.get(newUser.name, function (err, user) {
      if (err) {
        req.flash('error', err);
        return res.redirect('/');