node.js代码二

var express = require('express');
var app = express();
var server = require('http').Server(app);
var io = require('socket.io')(server);
var formidable = require('formidable');
var mysql = require('mysql');
var session = require('express-session');
var cookieParser = require('cookie-parser');

// 静态。
app.use(express.static('public'));
app.use(express.static('node_modules'));
app.use(cookieParser());
app.use(session({
    secret:'randomstring',
    resave: false,
    saveUninitialized:true,
    cookie:{
        maxAge:600000,
        hostOnly:true
    }
}));

// 配置 模版引擎。
app.set('view engine', 'ejs');
app.set('views', './views');

// 路由设置。
// 引入登录。
app.post('/doLogin', function(req, res){

    var form = new formidable.IncomingForm();

    // 表单处理。
    form.parse(req, function(error, fields, files){

        // 获取用户名和密码。
        var username = fields.username;
        var password = fields.password;
        var rememberme = fields.rememberme;

        // 查询数据库。
        var con = mysql.createConnection({
            host:'localhost',
            user:'root',
            password:'',
            database:'lamp183'
        });

        // 连接。
        con.connect();

        // 执行查询。
        con.query("SELECT * FROM user WHERE username='"+ username +"'", function(err, results, fields){

            // console.log(results[0].password);
            // res.send('ok');

            if(results.length <= 0)
            {

                res.json({'login': 0, 'data':'没有这个用户'});
                return ;
            }

            // 判断密码。
            if(results[0].password != password)
            {

                res.json({'login': 0, 'data':'密码错误'});

            }
            else
            {

                // 判断是否记住我。
                if(rememberme == 1)
                {

                    res.cookie('admin', results[0], {maxAge: 600000000, hostOnly:true});

                }

                // 将用户数据存入session
                req.session.admin = results[0];

                // 发送登录成功指令
                res.json({'login': 1, 'data':results[0]});

            }

        });

        con.end();
        return ;

    });

});

// 注册功能。
app.post('/doRegist', function(req, res){

    var form = new formidable.IncomingForm();

    // 表单处理。
    form.parse(req, function(error, fields, files){

        // console.log(fields);
        var username = fields.username;
        var password = fields.password;
        var nickname = fields.nickname;

        // 查询数据库。
        var con = mysql.createConnection({
            host:'localhost',
            user:'root',
            password:'',
            database:'lamp183'
        });

        // 连接。
        con.connect();

        // 执行添加。
        con.query("INSERT INTO user(username,password,nickname) VALUES('"+ username +"','"+ password +"','"+ nickname +"')", function(err, results, fields){

            // console.log(results);
            if(results.affectedRows > 0)
            {

                res.json({'regist':1, 'info':'注册成功,请登录'});

            }
            else
            {

                res.json({'regist':0, 'info':'注册失败'});

            }

        });

        con.end();
        return ;

    });

});

// 监听。
server.listen(8080);

// 引入聊天室页面。
app.get('/', function(req, res){

    // 判断用户是否已经登录。
    if(req.session.admin || req.cookies.admin)
    {

        res.render('index2');

        io.on('connection', function(socket){

            if(req.session.admin)
            {
        
                socket.emit('nolog', req.session.admin);

            }
            else if(req.cookies.admin)
            {

                socket.emit('nolog', req.session.admin);

            }

        });

    }
    else{

        res.render('index');

    }

    // res.render('index');
    // console.log(req.session.admin);

});

var users = [];

// 建立连接
io.on('connection', function(socket) {

    socket.on('login', function(data){

        // console.log(data.data.nickname);
        // 保存所有用户。
        var u = {nickname:data.data.nickname};
        users.push(u);

    });

    socket.on('users', function(){

        // 发送给所有客户端。
        socket.emit('users', users);

    });

    // 监听消息
    socket.on('send', function(data){
        // console.log(data);
        // 发送给所有用户。
        socket.broadcast.emit('receiver', data);
           
    });

});

// 退出登录功能。
app.get('/logout', function(req, res){

    // 清空session和cookie
    if(req.session.admin)
    {

        var name = req.session.admin.nickname
        delete(req.session.admin);

    }
    else
    {
        var name = req.session.admin.nickname
        res.cookie('admin', 1, {maxAge: -1, hostOnly:true});
    }

    // 跳转登录页面。
    res.redirect('/');
    return ;

});