CameraMachine/app/utils/utils.js

'use strict';

const crypto = require('crypto');
const fs = require('fs');
const path = require('path');
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);
      }
    });
  });
};