AutoCommit/src/models/commitHandle.ts

320 lines
10 KiB
TypeScript

/*
* Author : OBKoro1
* Date : 2019-12-30 16:59:30
* LastEditors : OBKoro1
* LastEditTime : 2021-06-27 16:18:59
* File : \autoCommit\src\models\commitHandle.ts
* Description : commit 具体操作
* https://github.com/OBKoro1
*/
import * as moment from 'moment';
import * as fs from 'fs';
import { exec } from 'child_process';
import { sep } from 'path';
import { WebviewMsg } from '../util/dataStatement';
import { RandomNumber } from '../util/util';
import {
getPanelWebview,
outputLog,
isProduction,
getExtensionContext,
} from '../util/vscodeUtil';
import WebView from './WebView';
interface TimeElement {
value: Array<string>;
commitNumber: number;
}
interface DayTime {
value: string;
commitNumber: number;
}
// 获取两个日期之间的间隔: [ '2019-02-02', '2019-02-03' ... ]
const getAllDay = (begin: string, end: string) => {
const timeArr = [];
const beginDate: Date = moment(begin).toDate();
const beginNumber = beginDate.getTime();
const endDate = moment(end).toDate();
const endNumber = endDate.getTime();
for (let k: any = beginNumber; k <= endNumber;) {
// eslint-disable-next-line radix
const day = new Date(parseInt(k));
const dayFormat = moment(day).format('YYYY-MM-DD');
timeArr.push(dayFormat);
k += 24 * 60 * 60 * 1000;
}
return timeArr;
};
// 获取当天的随机时间
const getTodayRandomNumber = (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}`;
};
class CommitHandle {
public paramsObj: any;
public moreObj: any;
public timeArr: Array<DayTime>;
public autoCommitView: WebView;
private userCancel: boolean;
private totalCommit: number; // 总commit次数
constructor(message: WebviewMsg) {
this.paramsObj = message.data.form;
this.moreObj = message.data.moreObj;
this.timeArr = [];
this.totalCommit = 0;
this.userCancel = false;
this.timeHandle();
this.autoCommitView = getPanelWebview();
}
// 处理所有时间段
timeHandle() {
// 处理所有时间范围
this.paramsObj.timeArr.forEach((item: TimeElement) => {
// 获取每个时间范围的具体日期
const detailTimeArr = getAllDay(item.value[0], item.value[1]);
// 日期去重 组织数据
detailTimeArr.forEach((ele) => {
const index = this.timeArr.findIndex((element) => element.value === ele);
// 删除重复日期
if (index !== -1) {
this.timeArr.splice(index, 1);
}
// 获取当天的commit次数
const dayCommitNumber = this.getDayCommitNumber(item.commitNumber);
this.totalCommit = dayCommitNumber + this.totalCommit;
this.timeArr.push({
value: ele,
commitNumber: dayCommitNumber,
});
});
});
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];
}
// 月份不同就比较月份
return dateArr1[1] - dateArr2[1];
}
// 年份不同就比较年份
return dateArr1[0] - dateArr2[0];
},
);
this.deleteDayArrDay();
this.commitFn();
}
// 随机删除日期数组中的某几天
deleteDayArrDay() {
const { noCommitDay } = this.moreObj;
const { scopeDay } = this.moreObj;
if (scopeDay < 1 || noCommitDay < 1) return; // 必须大于1
if (scopeDay > this.timeArr.length) return; // 日期不够
// 删除
for (let i = 0; i < noCommitDay; i += 1) {
const ranDomNum = Math.floor(Math.random() * this.timeArr.length); // 随机数
this.timeArr.splice(ranDomNum, 1);
}
}
async commitFn() {
await outputLog('将要commit的日期:', JSON.stringify(this.timeArr));
outputLog('总commit天数:', this.timeArr.length);
outputLog('总commit次数:', this.totalCommit);
let totalNum = 0; // 总commit次数
if (sep === '\\') {
const reg = new RegExp(/\\/g);
this.paramsObj.itemSrc = `${this.paramsObj.itemSrc.replace(reg, '/')}`;
}
// 遍历日期
for (const item of this.timeArr.values()) {
if (this.cancelCommit()) break;
// 每个日期commit次数
const dayCommitNumber = item.commitNumber;
for (let i = 0; i < dayCommitNumber; i += 1) {
if (this.cancelCommit()) break;
let time = getTodayRandomNumber(item.value); // 2019-01-02 08:00
time = moment(time).format(); // 2019-01-02T00:00:00+0800
const commitContent = this.commitFileContent(time, totalNum);
let commitMsg;
const isDebug = false; // 手动更改调试模拟是否提交git
if (!isProduction() || !isDebug) {
try {
// 显示绿点成功 但是在仓库的时间不对 github可能修改了规则 后来的commit只会在上面
// 先add
const commitCmd = 'git add .';
// 先add 和提交commit
// const commitCmd = `git add . && git commit -m '${this.paramsObj.commitMsg}'`;
commitMsg = await this.execCmd(commitCmd);
// 修改commit日期和时间
const cmd = `git commit --date='${time}' -am '${this.paramsObj.commitMsg}'`;
// set在window下可能有兼容问题
// const cmd = `git commit --amend --date='${time}' -am '${this.paramsObj.commitMsg}' && set GIT_COMMITTER_DATE='${time}' && set GIT_AUTHOR_DATE='${time}' && git commit --amend --no-edit --date '${time}'`;
// const cmd = `set GIT_COMMITTER_DATE='${time}' && set GIT_AUTHOR_DATE='${time}' && git commit --amend --no-edit --date '${time}'`;
// 修改commit的id 的提交time
// const commitId = await this.execCmd('git rev-parse HEAD');
// const cmd = `git commit --amend --date='${time}' -C ${commitId}`;
commitMsg = await this.execCmd(cmd);
} catch (err) {
outputLog(`commit出错:${err}`);
continue; // 错误 退出本次循环
}
} else {
// 模拟git提交
await new Promise((resolve) => {
setTimeout(() => {
outputLog('延迟一秒');
resolve('延迟一秒');
}, 1000);
});
}
totalNum += 1;
// commit次数小于100显示log
console.log(`${totalNum}commit内容`, commitContent);
console.log(`${totalNum}commit信息`, commitMsg);
}
}
this.pushCommitFn(totalNum);
}
async pushCommitFn(totalNum: number) {
const commitNumberBig = 100; // commit数量过大
let thinkNumber = 10000; // 考虑时间 避免运行过快导致误操作
if (totalNum > commitNumberBig) {
outputLog(`commit数量:${totalNum}`);
outputLog('commit数量超过100次,请考虑10秒钟是否需要取消commit');
} else {
thinkNumber = 2000; // 无感 考虑两秒
}
await new Promise((resolve) => {
setTimeout(resolve, thinkNumber);
});
if (this.cancelCommit()) {
if (totalNum === 0) return;
await this.resetCommit(totalNum);
this.commitEnd(totalNum, true);
} else {
outputLog('提交中...');
this.autoCommitView.postMessage('提交中...', '提交中');
const cmd = 'git pull --rebase && git push';
const res = await this.execCmd(cmd);
outputLog('提交信息:', res);
this.commitEnd(totalNum);
}
}
async resetCommit(totalNum: number) {
this.autoCommitView.postMessage('回滚', '回滚');
outputLog('回滚中...');
const cmd = `git reset --hard HEAD~${totalNum}`;
const p = await this.execCmd(cmd);
outputLog(`回滚${totalNum}次commit成功:`, p);
return p;
}
commitEnd(totalNum: number, cancel: boolean = false) {
this.userCancel = false; // 重新打开终止开关
if (!cancel) {
this.autoCommitView.postMessage('commit 完成', this.paramsObj);
getExtensionContext().globalState.update('commit-params', this.paramsObj);
outputLog('自动commit完成', `总commit次数${totalNum}`);
}
outputLog('保存参数信息');
}
cancelCommit() {
if (this.userCancel) {
outputLog('终止自动commit');
}
return this.userCancel;
}
public closeCommit() {
this.userCancel = true;
}
// 组织commit文件的内容
commitFileContent(time: string, totalNum: number) {
const commitContent = `${time}\n随机数:${RandomNumber(
1,
100000,
)}\n提交次数:${totalNum + 1}`;
// 写入内容
fs.writeFileSync(
`${this.paramsObj.itemSrc}/${this.paramsObj.fileName}`,
commitContent,
'utf-8',
);
return commitContent;
}
// 执行命令封装
execCmd(cmd: string) {
return new Promise((resolve, reject) => {
exec(
cmd,
{
cwd: this.paramsObj.itemSrc,
env: process.env,
},
(error, stdout, stderr) => {
if (error) {
outputLog(`执行命令出错:${cmd}`);
outputLog(`错误信息:${error}`, stderr);
reject(error);
return;
}
resolve(stdout);
},
);
});
}
// 获取当天的commit次数
getDayCommitNumber(commitNumber: number) {
let dayCommitNumber = this.paramsObj.commitNumber; // 固定commit次数
// 使用随机commit次数
if (this.paramsObj.randomCommit) {
dayCommitNumber = RandomNumber(1, this.paramsObj.commitNumber);
}
// 如果该时间范围有commit次数 则用该范围的
if (commitNumber !== 0) {
dayCommitNumber = commitNumber;
}
return dayCommitNumber;
}
}
export default CommitHandle;