leetcode-cheat/scripts/utils.js at 0.3 · leetcode-pp/leetcode-cheat · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
const fs = require("fs");
const request = require("request");
const {resolve: pathResolve} = require("path");
const mkdirp = require("mkdirp");

const PROJECT_ROOT = pathResolve(__dirname, "../");

const isFun = (input) => typeof input === "function";
const COLORS = [
	"magenta",
	"red",
	"volcano",
	"orange",
	"gold",
	"lime",
	"green",
	"cyan",
	"blue",
	"geekblue",
	"purple",
];
const Utils = {
	getColor(text) {
		let ans = 0;
		for (const c of text) ans += c.charCodeAt();
		return COLORS[ans % COLORS.length];
	},
	/**
   * 获取不同语言的正则表达式
   *
   * @static
   * @param {*} lang
   * @returns  Regexp
   * @memberof Utils
   */
	genCodeRegByLang(lang) {
		return new RegExp(`(?:\`\`\`${lang})((.|[\r\n])*?)(?:\`\`\`)`, "g");
	},

	httpGet(url, {beforeRequest, onSuccess, onError} = {}) {
		return new Promise((resolve, reject) => {
			isFun(beforeRequest) && beforeRequest();
			beforeRequest;
			request(
				{
					method: "GET",
					url,
					encoding: null,
				},
				(error, _, body) => {
					if (error) {
						reject(error);
						isFun(onError) && onError();
					} else {
						isFun(onSuccess) && onSuccess();
						resolve(body);
					}
				},
			);
		});
	},

	getSatelliteDataReg() {
		return {
			pre: /(?<=## \u524d\u7f6e\u77e5\u8bc6)([\s\S]*?)(?=##)/g,
			keyPoints: /(?<=## \u5173\u952e\u70b9)([\s\S]*?)(?=##)/,
			companies: /(?<=## \u516c\u53f8)([\s\S]*?)(?=##)/,
		};
	},

	/**
   *  递归创建目录
   *
   * @static
   * @param {*} path
   * @memberof Utils
   */
	mkdirSync(path) {
		mkdirp.sync(pathResolve(PROJECT_ROOT, path));
	},

	/**
   *  关键点匹配 TODO
   *
   * @static
   * @returns
   * @memberof Utils
   */
	getKeyMatterReg() {
		return /(?<=\u5173\u952e\u70b9)([.|[\r\n]\]*?)(?:#)/g;
	},

	/**
   *  读取文件内容
   *
   * @static
   * @param {*} relativePath
   * @param {*} filename
   * @returns
   * @memberof Utils
   */
	readFileSync(relativePath, filename) {
		return fs.readFileSync(
			pathResolve(PROJECT_ROOT, relativePath, filename),
			{
				encoding: "utf8",
			},
		);
	},

	/**
   * 获取文件下所有文件的名称
   *
   * @static
   * @param {*} dir
   * @returns Array<string>
   * @memberof Utils
   */
	getDirsFileNameSync(dir) {
		return fs.readdirSync(pathResolve(PROJECT_ROOT, dir), {encoding: "utf8"});
	},

	/**
   * 写入文件
   *
   * @static
   * @param {*} dir
   * @param {*} name
   * @param {*} content
   * @memberof Utils
   */
	writeFileSync(dir, name, content) {
		fs.writeFileSync(pathResolve(PROJECT_ROOT, dir, name), content);
	},
};

module.exports = Utils;