pinclub/pinclub

var User = require('../proxy').User;
var Topic = require('../proxy').Topic;
var Reply = require('../proxy').Reply;
var TopicCollect = require('../proxy').TopicCollect;
var utility = require('utility');
var util = require('util');
var TopicModel = require('../models').Topic;
var ReplyModel = require('../models').Reply;
var Forum = require('../proxy').Forum;
var Board = require('../proxy').Board;
var tools = require('../common/tools');
var cache = require('../common/cache');
var config = require('../config');
var EventProxy = require('eventproxy');
var validator = require('validator');
var speakeasy = require('speakeasy');
var QRCode = require('qrcode');
var _ = require('lodash');

exports.index = function (req, res, next) {
    var user_name = req.params.name;
    var currentUser = req.session.user;
    User.getUserByLoginName(user_name, function (err, user) {
        if (err) {
            return next(err);
        }
        if (!user) {
            res.render404('这个用户不存在。');
            return;
        }

        var render = function (recent_topics, recent_replies, forums, boards) {
            user.url = (function () {
                if (user.url && user.url.indexOf('http') !== 0) {
                    return 'http://' + user.url;
                }
                return user.url;
            })();
            // 如果用户没有激活，那么管理员可以帮忙激活
            var token = '';
            if (!user.active && req.session.user && req.session.user.is_admin) {
                token = utility.md5(user.email + user.pass + config.session_secret);
            }
            res.render('user/index', {
                user: user,
                recent_topics: recent_topics,
                recent_replies: recent_replies,
                token: token,
                forums: forums,
                boards: boards,
                pageTitle: util.format('@%s 的个人主页', user.loginname),
            });
        };

        var proxy = new EventProxy();
        proxy.assign('recent_topics', 'recent_replies', 'forums', 'boards', render);
        proxy.fail(next);

        var query = {author: user._id, type: 'text'};
        var opt = {limit: 5, sort: '-create_at'};
        Topic.getTopicsByQuery(query, opt, proxy.done('recent_topics'));

        Reply.getRepliesByAuthorId(user._id, {limit: 20, sort: '-create_at'},
            proxy.done('recent_replies', function (replies) {

                return replies;

                // var topic_ids = replies.map(function (reply) {
                //     return reply.topic.id;
                // });
                // topic_ids = _.uniq(topic_ids).slice(0, 5); //  只显示最近5条
                //
                // var query = {_id: {'$in': topic_ids}};
                // var opt = {};
                // Topic.getTopicsByQuery(query, opt, proxy.done('recent_replies', function (recent_replies) {
                //     recent_replies = _.sortBy(recent_replies, function (topic) {
                //         return topic_ids.indexOf(topic._id.toString());
                //     });
                //     return recent_replies;
                // }));
            }));

        // 获取板块信息
        var queryForum = {};
        queryForum.type = 'public';
        queryForum.parent = {$exists: false };
        if (!!currentUser) {
            queryForum.type = {$ne: 'private'};
        }
        var forumsCacheKey = JSON.stringify(queryForum) + 'pages';
        cache.get(forumsCacheKey, proxy.done(function (forums) {
            if (forums) {
                proxy.emit('forums', forums);
            } else {
                Forum.getForumsByQuery(queryForum, {limit: 10}, proxy.done(function (forums) {
                    cache.set(forumsCacheKey, forums, 60 * 1);
                    proxy.emit('forums', forums);
                }));
            }
        }));
        // 获取用户Board
        var queryBoard = {type: 'public'};
        if (!user.is_admin) {
            queryBoard.user_id = user;
        } else if (user.is_admin || (currentUser && user.id !== currentUser.id)) {
            delete queryBoard.type;
        }
        Board.getBoardsByQuery(queryBoard, {}, function (err, boards) {
            _.forEach(boards, function(board) {
                Topic.getImagesByQuery({
                    board: board.id
                }, {
                    limit: 4
                }, function (err, topics) {
                    board.topics = topics;
                    proxy.emit('boardtopic');
                });
            });
            proxy.after('boardtopic', boards.length, function () {
                proxy.emit('boards', boards);
            });
        });
    });
};

exports.listStars = function (req, res, next) {
    User.getUsersByQuery({is_star: true}, {}, function (err, stars) {
        if (err) {
            return next(err);
        }
        res.render('user/stars', {stars: stars});
    });
};

exports.showSetting = function (req, res, next) {
    User.getUserById(req.session.user._id, function (err, user) {
        if (err) {
            return next(err);
        }
        if (req.query.save === 'success') {
            user.success = '保存成功。';
        }
        user.error = null;
        if (!!user.is_two_factor && !!user.two_factor_base32) {
            // secret = user.two_factor_base32;
            cache.set(req.session.user._id + '_tf_secret', user.two_factor_base32, 60 * 2);
            return res.render('user/setting', user);
        } else {
            // secret = speakeasy.generateSecret({
            //     length: 32,
            //     name: user.loginname+'@Jiuyanlou',
            //     otpauth_url: true,
            //     symbols: false
            // });

            var secret = speakeasy.generateSecret({name: user.loginname+'@Jiuyanlou'});

            var url = speakeasy.otpauthURL({
                secret: secret.base32 ,
                label: user.loginname+'@Jiuyanlou',
                encoding: 'base32'
            });
            // secret = secretO.base32;
            cache.set(req.session.user._id + '_tf_secret', secret.base32, 60 * 2);
            var token =  speakeasy.totp({
                secret: secret.base32,
                encoding: 'base32'
            });
            QRCode.toDataURL(url, function(err, data_url) {
                user.two_factor_qr = data_url;
                return res.render('user/setting', user);
            });
        }
    });
};

exports.setting = function (req, res, next) {
    var ep = new EventProxy();
    ep.fail(next);

    // 显示出错或成功信息
    function showMessage(msg, data, isSuccess) {
        data = data || req.body;
        var data2 = {
            loginname: data.loginname,
            email: data.email,
            url: data.url,
            location: data.location,
            signature: data.signature,
            weibo: data.weibo,
            accessToken: data.accessToken,
            avatar_url: data.avatar_url,
            is_two_factor: data.is_two_factor,
            gender: data.gender
        };
        if (isSuccess) {
            data2.success = msg;
        } else {
            data2.error = msg;
        }
        res.render('user/setting', data2);
    }

    // post
    var action = req.body.action;
    if (action === 'change_setting') {
        var url = validator.trim(req.body.url);
        var location = validator.trim(req.body.location);
        var weibo = validator.trim(req.body.weibo);
        var signature = validator.trim(req.body.signature);
        var gender = validator.trim(req.body.gender);

        User.getUserById(req.session.user._id, ep.done(function (user) {
            user.url = url;
            user.location = location;
            user.signature = signature;
            user.weibo = weibo;
            user.gender = gender;
            user.save(function (err) {
                if (err) {
                    return next(err);
                }
                req.session.user = user.toObject({virtual: true});
                return res.redirect('/setting?save=success');
            });
        }));
    }
    if (action === 'change_password') {
        var old_pass = validator.trim(req.body.old_pass);
        var new_pass = validator.trim(req.body.new_pass);
        if (!old_pass || !new_pass) {
            return res.send('旧密码或新密码不得为空');
        }

        User.getUserById(req.session.user._id, ep.done(function (user) {
            tools.bcompare(old_pass, user.pass, ep.done(function (bool) {
                if (!bool) {
                    return showMessage('当前密码不正确。', user);
                }

                tools.bhash(new_pass, ep.done(function (passhash) {
                    user.pass = passhash;
                    user.save(function (err) {
                        if (err) {
                            return next(err);
                        }
                        return showMessage('密码已被修改。', user, true);

                    });
                }));
            }));
        }));
    }
};

exports.toggleTwoFactor = function (req, res, next) {
    var loginname = req.params.name;
    var code = req.body.tfv;

    User.getUserByLoginName(loginname, function (err, user) {
        if (err) {
            return next(err);
        }
        if (!user) {
            return next(new Error('user is not exists'));
        }
        cache.get(user._id + '_tf_secret', function (err, secret) {
            if (!err && secret) {
                let base32 = secret;
                if (!!user.is_two_factor && !!user.two_factor_base32) {
                    base32 = user.two_factor_base32;
                } else {
                    user.two_factor_base32 = base32;
                }
                var token = speakeasy.totp({
                    secret: base32,
                    encoding: 'base32'
                });
                var verified = speakeasy.totp.verify({
                    secret: base32,
                    token: code,
                    encoding: 'base32'
                });
                // var verified = speakeasy.totp.verify({
                //     secret: base32,
                //     encoding: 'base32',
                //     token: code
                // });
                if (verified) {
                    user.is_two_factor = !user.is_two_factor;
                    user.save(function (err) {
                        if (err) {
                            return next(err);
                        }
                        res.json({success: true, is_two_factor: user.is_two_factor});
                    });
                } else {
                    res.json({success: false});
                }
            } else {
                res.json({success: false});
            }
        });

    });
};

exports.toggleStar = function (req, res, next) {
    var loginname = req.params.name;
    User.getUserByLoginName(loginname, function (err, user) {
        if (err) {
            return next(err);
        }
        if (!user) {
            return next(new Error('user is not exists'));
        }
        user.is_star = !user.is_star;
        user.save(function (err) {
            if (err) {
                return next(err);
            }
            res.json({success: true});
        });
    });
};

exports.listCollectedTopics = function (req, res, next) {
    var name = req.params.name;
    var page = Number(req.query.page) || 1;
    var limit = config.list_topic_count;

    User.getUserByLoginName(name, function (err, user) {
        if (err || !user) {
            return next(err);
        }
        var pages = Math.ceil(user.collect_topic_count / limit);
        var render = function (topics) {
            res.render('user/collect_topics', {
                topics: topics,
                current_page: page,
                pages: pages,
                user: user
            });
        };

        var proxy = EventProxy.create('topics', render);
        proxy.fail(next);

        var opt = {
            skip: (page - 1) * limit,
            limit: limit
        };

        TopicCollect.getTopicCollectsByUserId(user._id, opt, proxy.done(function (docs) {
            var ids = docs.map(function (doc) {
                return String(doc.topic_id);
            });
            var query = {_id: {'$in': ids}};

            Topic.getTopicsByQuery(query, {}, proxy.done('topics', function (topics) {
                topics = _.sortBy(topics, function (topic) {
                    return ids.indexOf(String(topic._id));
                });
                return topics;
            }));
        }));
    });
};

exports.top100 = function (req, res, next) {
    var opt = {limit: 100, sort: '-score'};
    User.getUsersByQuery({is_block: false}, opt, function (err, tops) {
        if (err) {
            return next(err);
        }
        res.render('user/top100', {
            users: tops,
            pageTitle: 'top100',
        });
    });
};

exports.listTopics = function (req, res, next) {
    var user_name = req.params.name;
    var page = Number(req.query.page) || 1;
    var limit = config.list_topic_count;

    User.getUserByLoginName(user_name, function (err, user) {
        if (!user) {
            res.render404('这个用户不存在。');
            return;
        }

        var render = function (topics, pages) {
            res.render('user/topics', {
                user: user,
                topics: topics,
                current_page: page,
                pages: pages
            });
        };

        var proxy = new EventProxy();
        proxy.assign('topics', 'pages', render);
        proxy.fail(next);

        var query = {'author': user._id};
        var opt = {skip: (page - 1) * limit, limit: limit, sort: '-create_at'};
        Topic.getTopicsByQuery(query, opt, proxy.done('topics'));

        Topic.getCountByQuery(query, proxy.done(function (all_topics_count) {
            var pages = Math.ceil(all_topics_count / limit);
            proxy.emit('pages', pages);
        }));
    });
};

exports.listReplies = function (req, res, next) {
    var user_name = req.params.name;
    var page = Number(req.query.page) || 1;
    var limit = 50;

    User.getUserByLoginName(user_name, function (err, user) {
        if (!user) {
            res.render404('这个用户不存在。');
            return;
        }

        var render = function (replies, pages) {
            res.render('user/replies', {
                user: user,
                replies: replies,
                current_page: page,
                pages: pages
            });
        };

        var proxy = new EventProxy();
        proxy.assign('replies', 'pages', render);
        proxy.fail(next);

        var opt = {skip: (page - 1) * limit, limit: limit, sort: '-create_at'};
        Reply.getRepliesByAuthorId(user._id, opt, proxy.done('replies', function (replies) {
            // 获取所有有评论的主题
            return replies;
            // var topic_ids = replies.map(function (reply) {
            //     return reply.topic.id;
            // });
            // topic_ids = _.uniq(topic_ids);
            //
            // var query = {'_id': {'$in': topic_ids}};
            // Topic.getTopicsByQuery(query, {}, proxy.done('topics', function (topics) {
            //     topics = _.sortBy(topics, function (topic) {
            //         return topic_ids.indexOf(topic._id.toString());
            //     });
            //     return topics;
            // }));
        }));

        Reply.getCountByAuthorId(user._id, proxy.done('pages', function (count) {
            var pages = Math.ceil(count / limit);
            return pages;
        }));
    });
};

exports.block = function (req, res, next) {
    var loginname = req.params.name;
    var action = req.body.action;

    var ep = EventProxy.create();
    ep.fail(next);

    User.getUserByLoginName(loginname, ep.done(function (user) {
        if (!user) {
            return next(new Error('user is not exists'));
        }
        if (action === 'set_block') {
            ep.all('block_user',
                function (user) {
                    res.json({success: true, status: 'success'});
                });
            user.is_block = true;
            user.save(ep.done('block_user'));

        } else if (action === 'cancel_block') {
            user.is_block = false;
            user.save(ep.done(function () {

                res.json({success: true, status: 'success'});
            }));
        }
    }));
};

exports.deleteAll = function (req, res, next) {
    var loginname = req.params.name;

    var ep = EventProxy.create();
    ep.fail(next);

    User.getUserByLoginName(loginname, ep.done(function (user) {
        if (!user) {
            return next(new Error('user is not exists'));
        }
        ep.all('del_topics', 'del_replys', 'del_ups',
            function () {
                res.json({status: 'success'});
            });
        // 删除主题
        TopicModel.update({author: user._id}, {$set: {deleted: true}}, {multi: true}, ep.done('del_topics'));
        // 删除评论
        ReplyModel.update({author: user._id}, {$set: {deleted: true}}, {multi: true}, ep.done('del_replys'));
        // 点赞数也全部干掉
        ReplyModel.update({}, {$pull: {'ups': user._id}}, {multi: true}, ep.done('del_ups'));
    }));
};

// TODO 点击首页Get数量，进入用户Get的图片列表页面
exports.listGetImages = function (req, res, next) {
    // res.render('static/function_building', {
    //     title: 'Get 图片列表'
    // });
    var page = Number(req.query.page) || 1;
    var limit = 99999; //config.list_topic_count;
    var loginname = req.params.name;
    var ep = new EventProxy();
    ep.fail(next);

    let isCreator = false;

    User.getUserByLoginName(loginname, ep.done(function (user) {
        var query = {author: user._id};
        var opt = {skip: (page - 1) * limit, limit: limit, sort: '-create_at'};

        Topic.getImagesByQuery(query, opt, ep.done('images', function (images) {
            return images;
        }));

        Topic.getCountByQuery(query, ep.done('pages', function (count) {
            var pages = Math.ceil(count / limit);
            return pages;
        }));

        if (!!req.session.user){
            isCreator = (req.session.user.id == user.id);
        }
        ep.assign('images', 'pages', function (images, pages) {
            res.render('user/images', {
                user: user,
                images: images,
                current_page: page,
                pages: pages,
                is_creator: isCreator
            });
        });
    }));
};

// TODO 点击首页Board数量，进入用户 Board 列表页面
exports.board = function (req, res, next) {
    res.render('static/function_building', {
        title: '用户Board列表'
    });
};

// TODO 点击首页积分，进入用户 积分明细页面 的图片列表页面
exports.score = function (req, res, next) {
    res.render('static/function_building', {
        title: '积分明细页面'
    });
};

/**
 * 刷新用户统计数量
 * @param req
 * @param res
 * @param next
 */
exports.refreshCount = function (req, res, next) {
    req.checkParams({
        'id': {
            notEmpty: {
                options: [true],
                errorMessage: '用户ID不能为空'
            },
            isMongoId: {errorMessage: 'id 需为 mongoId 对象'}
        }
    });
    req.getValidationResult().then(function(result) {
        if (!result.isEmpty()) {
            return res.status(400).json({
                success: false,
                err_message: '参数验证失败',
                err: result.useFirstErrorOnly().mapped()
            }).end();
        }
        var id = req.params.id;
        var ep = new EventProxy();
        ep.fail(next);

        Board.getCountByQuery({user_id: id}, ep.done('board_count'));
        Topic.getCountByQuery({author: id, type: 'text', deleted: false}, ep.done('topic_count'));
        Topic.getCountByQuery({author: id, type: 'image', deleted: false}, ep.done('image_count'));

        ep.all('board_count', 'topic_count', 'image_count',
            function (board_count, topic_count, image_count) {
                User.getUserById(id, function (err, user) {
                    if (!!err) {
                        return next (err);
                    }
                    let changed = false;
                    if (user.board_count != board_count) {
                        user.board_count = board_count;
                        changed = true;
                        req.session.user.board_count = board_count;
                    }
                    if (user.topic_count != topic_count) {
                        user.topic_count = topic_count;
                        changed = true;
                        req.session.user.topic_count = topic_count;
                    }
                    if (user.image_count != image_count) {
                        user.image_count = image_count;
                        changed = true;
                        req.session.user.image_count = image_count;
                    }
                    if (changed) {user.save();}
                    res.send({success: true, board_count: board_count, topic_count: topic_count, image_count: image_count});
                });
            });
    });
};