AutoCommit/src/models/commitHandle.ts

/*
 * Author       : OBKoro1
 * Date         : 2019-12-30 16:59:30
 * LastEditors  : OBKoro1
 * LastEditTime : 2020-01-07 22:30:03
 * FilePath     : /autoCommit/src/models/commitHandle.ts
 * Description  : commit 具体操作
 * https://github.com/OBKoro1
 */

import { webviewMsg } from '../util/dataStatement';
import * as moment from 'moment';
import * as fs from 'fs';
import { execSync, exec } from 'child_process';
import { RandomNumber } from '../util/util';
import { getPanelWebview, outputLog, isProduction } from '../util/vscodeUtil';
import WebView from './WebView';

interface timeElement {
  value: Array<string>;
  commitNumber: number;
}

interface dayTime {
  value: string;
  commitNumber: number;
}

class CommitHandle {
  public paramsObj: any;
  public timeArr: Array<dayTime>;
  public autoCommitView: WebView;
  private userCancel: boolean;
  constructor(message: webviewMsg) {
    this.paramsObj = message.data;
    this.timeArr = [];
    this.timeHandle();
    this.autoCommitView = getPanelWebview();
    this.userCancel = false;
  }
  // 处理所有时间段
  timeHandle() {
    // 处理所有时间范围
    this.paramsObj.timeArr.forEach((item: timeElement) => {
      // 获取每个时间范围的具体日期
      let detailTimeArr = this.getAllDay(item.value[0], item.value[1]);
      // 日期去重 组织数据
      detailTimeArr.forEach(ele => {
        let index = this.timeArr.findIndex(element => {
          return element.value === ele;
        });
        // 删除重复日期
        if (index !== -1) {
          this.timeArr.splice(index, 1);
        }
        this.timeArr.push({
          value: ele,
          commitNumber: item.commitNumber
        });
      });
    });
    this.sortTime();
  }
  // 日期排序
  sortTime() {
    this.timeArr = this.timeArr.sort(
      (item1: dayTime, item2: dayTime): number => {
        const dateArr1: Array<any> = item1.value.split('-');
        const dateArr2: Array<any> = item2.value.split('-');
        if (dateArr1[0] === dateArr2[0]) {
          if (dateArr1[1] === dateArr2[1]) {
            // 日期不同就比较日期
            return dateArr1[2] - dateArr2[2];
          } else {
            // 月份不同就比较月份
            return dateArr1[1] - dateArr2[1];
          }
        } else {
          //  年份不同就比较年份
          return dateArr1[0] - dateArr2[0];
        }
      }
    );
    this.commitFn();
  }
  async commitFn() {
    await outputLog('将要commit的日期:', JSON.stringify(this.timeArr));
    let totalNum = 0; // 总commit次数
    // 遍历日期
    for (let item of this.timeArr.values()) {
      if (this.cancelCommit()) break;
      // 每个日期commit次数
      let dayCommitNumber = this.paramsObj.commitNumber;
      if (this.paramsObj.randomCommit) {
        // 随机commit次数
        dayCommitNumber = RandomNumber(1, this.paramsObj.commitNumber);
      }
      if (item.commitNumber !== 0) {
        // 如果该范围有commit次数 则用该范围的
        dayCommitNumber = item.commitNumber;
      }
      for (let i = 0; i < dayCommitNumber; i++) {
        if (this.cancelCommit()) break;
        let time = this.formatTime(item.value); // 2019-01-02 08:00
        time = moment(time).format(); // 2019-01-02T00:00:00+0800
        const commitContent = `${time} \n随机数:${RandomNumber(
          1,
          100000
        )}\n提交次数:${totalNum + 1}`;
        fs.writeFileSync(
          `${this.paramsObj.itemSrc}/${this.paramsObj.fileName}`,
          commitContent,
          'utf-8'
        );
        let commitMsg: string = '';
        const isDebug = false; // 手动更改调试模拟是否提交git
        if (!isProduction() || !isDebug) {
          try {
            // 异步执行命令 让出线程 打印日志 等
            commitMsg = await new Promise((resolve, reject) => {
              let cmd = `cd ${this.paramsObj.itemSrc} && git add . && git commit -m '${this.paramsObj.commitMsg}' --date='${time}'`;
              exec(cmd, (error, stdout, stderr) => {
                if (error) {
                  outputLog(`执行命令出错:${cmd}`);
                  outputLog(`错误信息:${error}`, stderr);
                  reject(error);
                  return;
                }
                resolve(stdout);
              });
            });
          } catch (err) {
            continue; // 错误 退出本次循环
          }
        } else {
          // 模拟git提交
          const test = await new Promise((resolve, reject) => {
            setTimeout(() => {
              outputLog('延迟一秒');
              resolve('延迟一秒');
            }, 1000);
          });
        }
        outputLog(`${totalNum + 1}commit内容`, commitContent);
        outputLog(`${totalNum + 1}commit信息`, commitMsg);
        totalNum++;
      }
    }
    if (this.cancelCommit()) {
      if (totalNum < 1) return;
      outputLog('回滚中...');
      const res = await new Promise((resolve, reject) => {
        let cmd = `cd ${this.paramsObj.itemSrc} && git reset --hard HEAD~${totalNum}`;
        exec(cmd, (error, stdout, stderr) => {
          if (error) {
            outputLog(`执行命令出错:${cmd}`);
            outputLog(`回滚失败:${error}`, stderr);
            reject(error);
            return;
          }
          outputLog('回滚成功:', stdout);
          resolve(stdout);
        });
      });
    } else {
      outputLog('提交中...');
      const res = await new Promise((resolve, reject) => {
        let cmd = `cd ${this.paramsObj.itemSrc} && git pull && git push`;
        exec(cmd, (error, stdout, stderr) => {
          if (error) {
            outputLog(`执行命令出错:${cmd}`);
            outputLog(`错误信息:${error}`, stderr);
            reject(error);
            return;
          }
          resolve(stdout);
        });
      });
      outputLog('提交信息:', res);
    }
    this.commitEnd(totalNum);
  }
  commitEnd(totalNum: number) {
    this.userCancel = false; // 重新打开终止开关
    this.autoCommitView.postMessage('commit 完成', 'commit 完成');
    outputLog('自动commit完成', `总commit次数${totalNum}`);
  }
  cancelCommit() {
    if (this.userCancel) {
      outputLog('终止自动commit');
    }
    return this.userCancel;
  }
  public closeCommit() {
    this.userCancel = true;
  }
  // 获取当天的随机时间
  formatTime(time: string) {
    const hour1 = RandomNumber(0, 2);
    let hour2 = RandomNumber(0, 9);
    if (hour1 === 2) {
      // 小时第一个数字为2 则小时第二个数字最多为3
      hour2 = RandomNumber(0, 3);
    }
    const minute = `${RandomNumber(0, 5)}${RandomNumber(0, 9)}`;
    const hour = `${hour1}${hour2}`;
    return `${time} ${hour}:${minute}`;
  }
  // 获取两个日期之间的间隔: [ '2019-02-02', '2019-02-03' ... ]
  getAllDay(begin: string, end: string) {
    const timeArr = [];
    const beginSplit: Array<string> = begin.split('-');
    const endSplit: Array<string> = end.split('-');
    const beginDate = new Date();
    beginDate.setUTCFullYear(
      Number(beginSplit[0]),
      Number(beginSplit[1]) - 1,
      Number(beginSplit[2])
    );
    const endDate = new Date();
    endDate.setUTCFullYear(
      Number(endSplit[0]),
      Number(endSplit[1]) - 1,
      Number(endSplit[2])
    );
    const beginNumber = beginDate.getTime();
    const endNumber = endDate.getTime();
    for (let k: any = beginNumber; k <= endNumber; ) {
      const day = new Date(parseInt(k));
      const dayFormat = moment(day).format('YYYY-MM-DD');
      timeArr.push(dayFormat);
      k = k + 24 * 60 * 60 * 1000;
    }
    return timeArr;
  }
  //  同步执行命令
  myExecSync(cmd: string) {
    // 除了该方法直到子进程完全关闭后才返回 执行完毕 返回
    try {
      const res = execSync(cmd, {
        encoding: 'utf8',
        cwd: undefined,
        env: undefined
      });
      return res;
    } catch (err) {
      outputLog(`执行命令出错:${cmd}`);
      outputLog(`错误信息:${err}`);
      return err;
    }
  }
}

export default CommitHandle;