Calculation/app/service/login_logging.js

'use strict';

/**
 * 登录日志-数据模型
 *
 * @author lanjianrong
 * @date 2020/8/31
 * @version
 */
const UAParser = require('ua-parser-js');

module.exports = app => {
    class LoginLogging extends app.BaseService {
        constructor(ctx) {
            super(ctx);
            this.tableName = 'login_logging';
        }

        /**
         * 创建记录
         * @param {Object} payload - 载荷
         */
        async createLog(payload) {
            const transaction = await this.db.beginTransaction();
            try {
                transaction.insert(this.tableName, payload);
                await transaction.commit();
            } catch (error) {
                await transaction.rollback();
                throw error;
            }
        }

        /**
         * 创建登录日志
         * @return {Boolean} 日志是否创建成功
         * @param {Number} type - 登录类型
         * @param {Number} status - 是否显示记录
         */
        async addLoginLog(type, status) {
            const { ctx } = this;
            const ip = ctx.request.ip ? ctx.request.ip : '';
            const ipInfo = await this.getIpInfoFromApi(ip);
            const parser = new UAParser(ctx.header['user-agent']);
            const osInfo = parser.getOS();
            const cpuInfo = parser.getCPU();
            const browserInfo = parser.getBrowser();
            const payload = {
                os: `${osInfo.name} ${osInfo.version} ${cpuInfo.architecture}`,
                browser: `${browserInfo.name} ${browserInfo.version}`,
                ip,
                address: ipInfo,
                uid: ctx.session.sessionUser.accountId,
                pid: ctx.session.sessionProject.id,
                type,
                show: status,
            };
            return await this.createLog(payload);
        }

        /**
         * 根据ip请求获取详细地址
         * @param {String} a_ip - ip地址
         * @return {String} 详细地址
         */
        async getIpInfoFromApi(a_ip = '') {
            try {
                if (!a_ip) return '';
                if (a_ip === '127.0.0.1' || a_ip === '::1' || a_ip.indexOf('192.168') !== -1) return '服务器本机访问';
                const { ip = '', region = '', city = '', isp = '' } = await this.sendRequest(a_ip);
                let address = '';
                region && (address += region + '省');
                city && (address += city + '市 ');
                isp && (address += isp + ' ');
                ip && (address += `(${ip})`);
                return address;
            } catch (error) {
                return '';
            }

        }

        /**
         * 发送请求获取详细地址
         * @param {String} ip - ip地址
         * @return {Object} the result of request
         * @private
         */
        async sendRequest(ip) {
            return new Promise((resolve, reject) => {
                this.ctx.curl(`https://api01.aliyun.venuscn.com/ip?ip=${ip}`, {
                    dateType: 'json',
                    encoding: 'utf8',
                    timeout: 2000,
                    headers: {
                        Authorization: 'APPCODE 85c64bffe70445c4af9df7ae31c7bfcc',
                    },
                }).then(({ status, data }) => {
                    if (status === 200) {
                        const result = JSON.parse(data.toString()).data;
                        if (!result.ip) {
                            resolve();
                        } else {
                            resolve(result);
                        }
                    } else {
                        resolve();
                    }
                }).catch(error => {
                    reject(error);
                });
            });
        }

        /**
         * 获取登录日志
         * @param {Number} pid - 项目id
         * @param {Number} uid - 用户id
         * @return {Promise<Array>} 日志数组
         */
        async getLoginLogs(pid, uid) {
            return this.db.select(this.tableName, {
                where: { pid, uid, show: 0 },
                orders: [['create_time', 'desc']],
                columns: ['browser', 'create_time', 'ip', 'os', 'address'],
                limit: 10, offset: 0,
            });
        }
    }
    return LoginLogging;
};