|
|
@@ -0,0 +1,501 @@
|
|
|
+'use strict';
|
|
|
+const util = require('util');
|
|
|
+const crypto = require('crypto');
|
|
|
+const fs = require('fs');
|
|
|
+const path = require('path');
|
|
|
+const _ = require('lodash');
|
|
|
+const readline = require('readline');
|
|
|
+const net = require('net');
|
|
|
+
|
|
|
+exports = module.exports;
|
|
|
+
|
|
|
+// control variable of func "myPrint"
|
|
|
+const isPrintFlag = true;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Check and invoke callback function
|
|
|
+ */
|
|
|
+exports.invokeCallback = function(cb) {
|
|
|
+ if (!!cb && typeof cb === 'function') {
|
|
|
+ cb.apply(null, Array.prototype.slice.call(arguments, 1));
|
|
|
+ }
|
|
|
+};
|
|
|
+
|
|
|
+exports.size = function(obj) {
|
|
|
+ if (!obj) {
|
|
|
+ return 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ let size = 0;
|
|
|
+ for (const f in obj) {
|
|
|
+ if (obj.hasOwnProperty(f)) {
|
|
|
+ size++;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return size;
|
|
|
+};
|
|
|
+
|
|
|
+// print the file name and the line number ~ begin
|
|
|
+function getStack() {
|
|
|
+ const orig = Error.prepareStackTrace;
|
|
|
+ Error.prepareStackTrace = function(_, stack) {
|
|
|
+ return stack;
|
|
|
+ };
|
|
|
+ const err = new Error();
|
|
|
+ Error.captureStackTrace(err, arguments.callee);
|
|
|
+ const stack = err.stack;
|
|
|
+ Error.prepareStackTrace = orig;
|
|
|
+ return stack;
|
|
|
+}
|
|
|
+
|
|
|
+function getFileName(stack) {
|
|
|
+ return stack[1].getFileName();
|
|
|
+}
|
|
|
+
|
|
|
+function getLineNumber(stack) {
|
|
|
+ return stack[1].getLineNumber();
|
|
|
+}
|
|
|
+
|
|
|
+exports.myPrint = function() {
|
|
|
+ if (isPrintFlag) {
|
|
|
+ const len = arguments.length;
|
|
|
+ if (len <= 0) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ const stack = getStack();
|
|
|
+ let aimStr =
|
|
|
+ "'" + getFileName(stack) + "' @" + getLineNumber(stack) + ' :\n';
|
|
|
+ for (let i = 0; i < len; ++i) {
|
|
|
+ aimStr +=
|
|
|
+ (typeof arguments[i] === 'object'
|
|
|
+ ? JSON.stringify(arguments[i])
|
|
|
+ : arguments[i]) + ' ';
|
|
|
+ }
|
|
|
+ console.log('\n' + aimStr);
|
|
|
+ }
|
|
|
+};
|
|
|
+
|
|
|
+exports.md5 = function(str) {
|
|
|
+ const hash = crypto.createHash('md5');
|
|
|
+ hash.update(str);
|
|
|
+ return hash.digest('hex');
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * 加密
|
|
|
+ */
|
|
|
+exports.cipher = function(algorithm, key, iv, data) {
|
|
|
+ key = new Buffer(key);
|
|
|
+ iv = new Buffer(iv ? iv : 0);
|
|
|
+ const cipher = crypto.createCipheriv(algorithm, key, iv);
|
|
|
+ cipher.setAutoPadding(true); // default true
|
|
|
+ let ciph = cipher.update(data, 'utf8', 'base64');
|
|
|
+ ciph += cipher.final('base64');
|
|
|
+
|
|
|
+ return ciph;
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * 解密
|
|
|
+ */
|
|
|
+exports.decipher = function(algorithm, key, iv, crypted) {
|
|
|
+ key = new Buffer(key);
|
|
|
+ iv = new Buffer(iv ? iv : '');
|
|
|
+ const decipher = crypto.createDecipheriv(algorithm, key, iv);
|
|
|
+ decipher.setAutoPadding(true);
|
|
|
+ let decoded = decipher.update(crypted, 'base64', 'utf8');
|
|
|
+ decoded += decipher.final('utf8');
|
|
|
+
|
|
|
+ return decoded;
|
|
|
+};
|
|
|
+
|
|
|
+exports.getDate = function(flag) {
|
|
|
+ if (!flag) {
|
|
|
+ flag = '';
|
|
|
+ }
|
|
|
+ const date = new Date();
|
|
|
+ const year = date.getFullYear();
|
|
|
+ const month = date.getMonth() + 1;
|
|
|
+ const day = date.getDate();
|
|
|
+ const mytimes = year + flag + month + flag + day;
|
|
|
+ return mytimes;
|
|
|
+};
|
|
|
+
|
|
|
+exports.dynamicSign = function(len) {
|
|
|
+ len = len || 5;
|
|
|
+ const chars = 'abcdefghijklmnopqrstuvwxyz';
|
|
|
+ const maxPos = chars.length;
|
|
|
+ let sign = '';
|
|
|
+ for (let i = 0; i < len; i++) {
|
|
|
+ sign += chars.charAt(Math.floor(Math.random() * maxPos));
|
|
|
+ }
|
|
|
+ return sign;
|
|
|
+};
|
|
|
+
|
|
|
+exports.toHump = function(str) {
|
|
|
+ let result = '';
|
|
|
+ const split = str.split('_');
|
|
|
+ result += split[0];
|
|
|
+ for (let i = 1; i < split.length; ++i) {
|
|
|
+ result += split[i][0].toUpperCase() + split[i].substr(1);
|
|
|
+ }
|
|
|
+ return result;
|
|
|
+};
|
|
|
+
|
|
|
+exports.tryJsonParse = function(jsonStr, cb) {
|
|
|
+ try {
|
|
|
+ cb(null, JSON.parse(jsonStr));
|
|
|
+ } catch (e) {
|
|
|
+ cb(e, null);
|
|
|
+ }
|
|
|
+};
|
|
|
+
|
|
|
+exports.mkdir = function(dirpath, dirname) {
|
|
|
+ // 判断是否是第一次调用
|
|
|
+ if (typeof dirname === 'undefined') {
|
|
|
+ if (fs.existsSync(dirpath)) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ this.mkdir(dirpath, path.dirname(dirpath));
|
|
|
+ } else {
|
|
|
+ // 判断第二个参数是否正常,避免调用时传入错误参数
|
|
|
+ if (dirname !== path.dirname(dirpath)) {
|
|
|
+ this.mkdir(dirpath);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ if (fs.existsSync(dirname)) {
|
|
|
+ fs.mkdirSync(dirpath);
|
|
|
+ } else {
|
|
|
+ this.mkdir(dirname, path.dirname(dirname));
|
|
|
+ fs.mkdirSync(dirpath);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ console.log('==> dir end', dirpath);
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * 获取目录下指定后缀的所有文件
|
|
|
+ * @param dir
|
|
|
+ * @param ext
|
|
|
+ * @return {Array}
|
|
|
+ */
|
|
|
+exports.getAllFiles = function(dir, ext) {
|
|
|
+ if (!dir) {
|
|
|
+ return [];
|
|
|
+ }
|
|
|
+
|
|
|
+ const self = this;
|
|
|
+ let extFiles = [];
|
|
|
+
|
|
|
+ const files = fs.readdirSync(dir);
|
|
|
+ files.forEach(function(file) {
|
|
|
+ const pathname = path.join(dir, file);
|
|
|
+ const stat = fs.lstatSync(pathname);
|
|
|
+
|
|
|
+ if (stat.isDirectory()) {
|
|
|
+ extFiles = extFiles.concat(self.getAllFiles(pathname, ext));
|
|
|
+ } else if (path.extname(pathname) === ext) {
|
|
|
+ extFiles.push(pathname.replace(cwd, '.'));
|
|
|
+ }
|
|
|
+ });
|
|
|
+
|
|
|
+ return extFiles;
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * 获取目录下所有文件夹
|
|
|
+ */
|
|
|
+exports.getDirs = function(dir) {
|
|
|
+ if (!dir) {
|
|
|
+ return [];
|
|
|
+ }
|
|
|
+
|
|
|
+ const components = [];
|
|
|
+ const files = fs.readdirSync(dir);
|
|
|
+ files.forEach(function(item, index) {
|
|
|
+ const stat = fs.lstatSync(dir + '/' + item);
|
|
|
+ if (stat.isDirectory() === true) {
|
|
|
+ components.push(item);
|
|
|
+ }
|
|
|
+ });
|
|
|
+
|
|
|
+ return components;
|
|
|
+};
|
|
|
+
|
|
|
+exports.fileExist = function(filePath) {
|
|
|
+ try {
|
|
|
+ return fs.statSync(filePath).isFile();
|
|
|
+ } catch (err) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+};
|
|
|
+
|
|
|
+exports.delDir = function(path) {
|
|
|
+ let files = [];
|
|
|
+ if (fs.existsSync(path)) {
|
|
|
+ files = fs.readdirSync(path);
|
|
|
+ files.forEach((file, index) => {
|
|
|
+ const curPath = path + '/' + file;
|
|
|
+ if (fs.statSync(curPath).isDirectory()) {
|
|
|
+ this.delDir(curPath); // 递归删除文件夹
|
|
|
+ } else {
|
|
|
+ fs.unlinkSync(curPath); // 删除文件
|
|
|
+ }
|
|
|
+ });
|
|
|
+ fs.rmdirSync(path);
|
|
|
+ }
|
|
|
+};
|
|
|
+
|
|
|
+exports.chmodPath = function(path, mode) {
|
|
|
+ let files = [];
|
|
|
+ if (fs.existsSync(path)) {
|
|
|
+ files = fs.readdirSync(path);
|
|
|
+ files.forEach((file, index) => {
|
|
|
+ const curPath = path + '/' + file;
|
|
|
+ if (fs.statSync(curPath).isDirectory()) {
|
|
|
+ this.chmodPath(curPath, mode); // 递归删除文件夹
|
|
|
+ } else {
|
|
|
+ fs.chmodSync(curPath, mode);
|
|
|
+ }
|
|
|
+ });
|
|
|
+ fs.chmodSync(path, mode);
|
|
|
+ }
|
|
|
+};
|
|
|
+
|
|
|
+/**
|
|
|
+ * 判断是否是同一天
|
|
|
+ * @param d1
|
|
|
+ * @param d2
|
|
|
+ * @return {boolean}
|
|
|
+ */
|
|
|
+exports.isSameDay = function(d1, d2) {
|
|
|
+ return (
|
|
|
+ d1.getFullYear() === d2.getFullYear() &&
|
|
|
+ d1.getMonth() === d2.getMonth() &&
|
|
|
+ d1.getDate() === d2.getDate()
|
|
|
+ );
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * 判断是否在有效期
|
|
|
+ */
|
|
|
+exports.isInDate = function(createDate, days) {
|
|
|
+ const diffDays = this.diffDays(createDate) + 1;
|
|
|
+ return diffDays <= days;
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * differ days
|
|
|
+ * 返回两个日期相隔的天数
|
|
|
+ * 按0点算
|
|
|
+ * 1日0点与2日0点相隔为1天
|
|
|
+ */
|
|
|
+exports.diffDays = function(createDate) {
|
|
|
+ const tmp = new Date();
|
|
|
+ tmp.setHours(0, 0, 0, 0);
|
|
|
+ createDate.setHours(0, 0, 0, 0);
|
|
|
+ return (tmp - createDate) / 24 / 3600000;
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * differ weeks
|
|
|
+ * 判断两个日期是否在同一周
|
|
|
+ * add by mumu
|
|
|
+ */
|
|
|
+exports.isSameWeek = function(old, now) {
|
|
|
+ const oneDayTime = 1000 * 60 * 60 * 24;
|
|
|
+ const old_count = parseInt(old.getTime() / oneDayTime);
|
|
|
+ const now_other = parseInt(now.getTime() / oneDayTime);
|
|
|
+ return parseInt((old_count + 4) / 7) == parseInt((now_other + 4) / 7);
|
|
|
+};
|
|
|
+
|
|
|
+/**
|
|
|
+ * 读取json文件并解析
|
|
|
+ * @param path
|
|
|
+ * @return {any}
|
|
|
+ */
|
|
|
+exports.loadJson = function(path) {
|
|
|
+ return JSON.parse(fs.readFileSync(path).toString());
|
|
|
+};
|
|
|
+
|
|
|
+exports.getRandomArrayElements = function(arr, count) {
|
|
|
+ let shuffled = arr.slice(0),
|
|
|
+ i = arr.length,
|
|
|
+ min = i - count,
|
|
|
+ temp,
|
|
|
+ index;
|
|
|
+ while (i-- > min) {
|
|
|
+ index = Math.floor((i + 1) * Math.random());
|
|
|
+ temp = shuffled[index];
|
|
|
+ shuffled[index] = shuffled[i];
|
|
|
+ shuffled[i] = temp;
|
|
|
+ }
|
|
|
+ return shuffled.slice(min);
|
|
|
+};
|
|
|
+
|
|
|
+exports.getTodayExpireTime = function() {
|
|
|
+ const endTime = new Date(new Date().setHours(23, 59, 59, 0)) / 1000;
|
|
|
+ const expiresTime = endTime - Math.floor(new Date().getTime() / 1000);
|
|
|
+ return expiresTime;
|
|
|
+};
|
|
|
+
|
|
|
+exports.objKeySort = function(obj) {
|
|
|
+ const newkey = Object.keys(obj).sort();
|
|
|
+ const newObj = {};
|
|
|
+ for (let i = 0; i < newkey.length; i++) {
|
|
|
+ newObj[newkey[i]] = obj[newkey[i]];
|
|
|
+ }
|
|
|
+ return newObj;
|
|
|
+};
|
|
|
+
|
|
|
+exports.serialize = function(obj) {
|
|
|
+ const str = [];
|
|
|
+ for (const p in obj) {
|
|
|
+ if (obj.hasOwnProperty(p)) {
|
|
|
+ str.push(p + '=' + obj[p]);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return str.join('&');
|
|
|
+};
|
|
|
+
|
|
|
+exports.keepTwoDecimalFull = function(num) {
|
|
|
+ let result = parseFloat(num);
|
|
|
+ if (isNaN(result)) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+ result = Math.round(num * 100) / 100;
|
|
|
+ let s_x = result.toString();
|
|
|
+ let pos_decimal = s_x.indexOf('.');
|
|
|
+ if (pos_decimal < 0) {
|
|
|
+ pos_decimal = s_x.length;
|
|
|
+ s_x += '.';
|
|
|
+ }
|
|
|
+ while (s_x.length <= pos_decimal + 2) {
|
|
|
+ s_x += '0';
|
|
|
+ }
|
|
|
+ return s_x;
|
|
|
+};
|
|
|
+
|
|
|
+exports.sleep = function(time = 0) {
|
|
|
+ return new Promise((resolve, reject) => {
|
|
|
+ setTimeout(() => {
|
|
|
+ resolve();
|
|
|
+ }, time);
|
|
|
+ });
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * 按行读取文件内容
|
|
|
+ * 返回:字符串数组
|
|
|
+ * 参数:fReadName:文件名路径
|
|
|
+ */
|
|
|
+exports.readFileToArr = async function(fReadName) {
|
|
|
+ const fRead = fs.createReadStream(fReadName);
|
|
|
+ const objReadline = readline.createInterface({
|
|
|
+ input: fRead,
|
|
|
+ });
|
|
|
+
|
|
|
+ return new Promise((resolve, reject) => {
|
|
|
+ const arr = [];
|
|
|
+ objReadline.on('line', function(line) {
|
|
|
+ arr.push(line);
|
|
|
+ });
|
|
|
+ objReadline.on('close', function() {
|
|
|
+ resolve(arr);
|
|
|
+ });
|
|
|
+ });
|
|
|
+};
|
|
|
+
|
|
|
+exports.compareVersion = function(version, bigVersion) {
|
|
|
+ version = version.split('.');
|
|
|
+ bigVersion = bigVersion.split('.');
|
|
|
+ for (let i = 0; i < version.length; i++) {
|
|
|
+ version[i] = +version[i];
|
|
|
+ bigVersion[i] = +bigVersion[i];
|
|
|
+ if (version[i] > bigVersion[i]) {
|
|
|
+ return false;
|
|
|
+ } else if (version[i] < bigVersion[i]) {
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return false;
|
|
|
+};
|
|
|
+
|
|
|
+exports.handleVersion = function(version) {
|
|
|
+ if (!version) return version;
|
|
|
+ version = version + '';
|
|
|
+ if (version[0] === 'v') {
|
|
|
+ return version.substr(1);
|
|
|
+ }
|
|
|
+ return version;
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * 获取本机IP地址
|
|
|
+ */
|
|
|
+exports.getIPAddress = function() {
|
|
|
+ const interfaces = require('os').networkInterfaces();
|
|
|
+ for (const devName in interfaces) {
|
|
|
+ const iface = interfaces[devName];
|
|
|
+ for (let i = 0; i < iface.length; i++) {
|
|
|
+ const alias = iface[i];
|
|
|
+ if (
|
|
|
+ alias.family === 'IPv4' &&
|
|
|
+ alias.address !== '127.0.0.1' &&
|
|
|
+ !alias.internal
|
|
|
+ ) {
|
|
|
+ return alias.address;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+};
|
|
|
+
|
|
|
+/*
|
|
|
+ * 判断IP是否在同一网段
|
|
|
+ */
|
|
|
+exports.isEqualIPAddress = function (addr1, addr2, mask = '255.255.255.0'){
|
|
|
+ if(!addr1 || !addr2 || !mask){
|
|
|
+ console.log("各参数不能为空");
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+ var
|
|
|
+ res1 = [],
|
|
|
+ res2 = [];
|
|
|
+ addr1 = addr1.split(".");
|
|
|
+ addr2 = addr2.split(".");
|
|
|
+ mask = mask.split(".");
|
|
|
+ for(var i = 0,ilen = addr1.length; i < ilen ; i += 1){
|
|
|
+ res1.push(parseInt(addr1[i]) & parseInt(mask[i]));
|
|
|
+ res2.push(parseInt(addr2[i]) & parseInt(mask[i]));
|
|
|
+ }
|
|
|
+ if(res1.join(".") == res2.join(".")){
|
|
|
+ return true;
|
|
|
+ }else{
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+/*
|
|
|
+ * 端口是否被占用
|
|
|
+ */
|
|
|
+exports.portIsOccupied = function portIsOccupied(port) {
|
|
|
+ const server = net.createServer().listen(port, '0.0.0.0');
|
|
|
+ return new Promise((resolve, reject) => {
|
|
|
+ server.on('listening', () => {
|
|
|
+ console.log(`the server is runnint on port ${port}`);
|
|
|
+ server.close();
|
|
|
+ resolve(false);
|
|
|
+ });
|
|
|
+
|
|
|
+ server.on('error', err => {
|
|
|
+ if (err.code === 'EADDRINUSE') {
|
|
|
+ resolve(true);
|
|
|
+ console.log(`this port ${port} is occupied.try another.`);
|
|
|
+ } else {
|
|
|
+ resolve(true);
|
|
|
+ }
|
|
|
+ });
|
|
|
+ });
|
|
|
+};
|