clog-ai
Version:
Generate commit log with AI
564 lines (557 loc) • 22.6 kB
JavaScript
'use strict';
Object.defineProperty(exports, '__esModule', { value: true });
function _interopDefault (ex) { return (ex && (typeof ex === 'object') && 'default' in ex) ? ex['default'] : ex; }
var axios = _interopDefault(require('axios'));
var child_process = _interopDefault(require('child_process'));
var loading = _interopDefault(require('loading-cli'));
var fs = require('fs');
var path = require('path');
var os = require('os');
var readline = _interopDefault(require('readline'));
function _regeneratorRuntime() {
_regeneratorRuntime = function () {
return e;
};
var t,
e = {},
r = Object.prototype,
n = r.hasOwnProperty,
o = Object.defineProperty || function (t, e, r) {
t[e] = r.value;
},
i = "function" == typeof Symbol ? Symbol : {},
a = i.iterator || "@@iterator",
c = i.asyncIterator || "@@asyncIterator",
u = i.toStringTag || "@@toStringTag";
function define(t, e, r) {
return Object.defineProperty(t, e, {
value: r,
enumerable: !0,
configurable: !0,
writable: !0
}), t[e];
}
try {
define({}, "");
} catch (t) {
define = function (t, e, r) {
return t[e] = r;
};
}
function wrap(t, e, r, n) {
var i = e && e.prototype instanceof Generator ? e : Generator,
a = Object.create(i.prototype),
c = new Context(n || []);
return o(a, "_invoke", {
value: makeInvokeMethod(t, r, c)
}), a;
}
function tryCatch(t, e, r) {
try {
return {
type: "normal",
arg: t.call(e, r)
};
} catch (t) {
return {
type: "throw",
arg: t
};
}
}
e.wrap = wrap;
var h = "suspendedStart",
l = "suspendedYield",
f = "executing",
s = "completed",
y = {};
function Generator() {}
function GeneratorFunction() {}
function GeneratorFunctionPrototype() {}
var p = {};
define(p, a, function () {
return this;
});
var d = Object.getPrototypeOf,
v = d && d(d(values([])));
v && v !== r && n.call(v, a) && (p = v);
var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p);
function defineIteratorMethods(t) {
["next", "throw", "return"].forEach(function (e) {
define(t, e, function (t) {
return this._invoke(e, t);
});
});
}
function AsyncIterator(t, e) {
function invoke(r, o, i, a) {
var c = tryCatch(t[r], t, o);
if ("throw" !== c.type) {
var u = c.arg,
h = u.value;
return h && "object" == typeof h && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) {
invoke("next", t, i, a);
}, function (t) {
invoke("throw", t, i, a);
}) : e.resolve(h).then(function (t) {
u.value = t, i(u);
}, function (t) {
return invoke("throw", t, i, a);
});
}
a(c.arg);
}
var r;
o(this, "_invoke", {
value: function (t, n) {
function callInvokeWithMethodAndArg() {
return new e(function (e, r) {
invoke(t, n, e, r);
});
}
return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg();
}
});
}
function makeInvokeMethod(e, r, n) {
var o = h;
return function (i, a) {
if (o === f) throw new Error("Generator is already running");
if (o === s) {
if ("throw" === i) throw a;
return {
value: t,
done: !0
};
}
for (n.method = i, n.arg = a;;) {
var c = n.delegate;
if (c) {
var u = maybeInvokeDelegate(c, n);
if (u) {
if (u === y) continue;
return u;
}
}
if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) {
if (o === h) throw o = s, n.arg;
n.dispatchException(n.arg);
} else "return" === n.method && n.abrupt("return", n.arg);
o = f;
var p = tryCatch(e, r, n);
if ("normal" === p.type) {
if (o = n.done ? s : l, p.arg === y) continue;
return {
value: p.arg,
done: n.done
};
}
"throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg);
}
};
}
function maybeInvokeDelegate(e, r) {
var n = r.method,
o = e.iterator[n];
if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y;
var i = tryCatch(o, e.iterator, r.arg);
if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y;
var a = i.arg;
return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y);
}
function pushTryEntry(t) {
var e = {
tryLoc: t[0]
};
1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e);
}
function resetTryEntry(t) {
var e = t.completion || {};
e.type = "normal", delete e.arg, t.completion = e;
}
function Context(t) {
this.tryEntries = [{
tryLoc: "root"
}], t.forEach(pushTryEntry, this), this.reset(!0);
}
function values(e) {
if (e || "" === e) {
var r = e[a];
if (r) return r.call(e);
if ("function" == typeof e.next) return e;
if (!isNaN(e.length)) {
var o = -1,
i = function next() {
for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next;
return next.value = t, next.done = !0, next;
};
return i.next = i;
}
}
throw new TypeError(typeof e + " is not iterable");
}
return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", {
value: GeneratorFunctionPrototype,
configurable: !0
}), o(GeneratorFunctionPrototype, "constructor", {
value: GeneratorFunction,
configurable: !0
}), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) {
var e = "function" == typeof t && t.constructor;
return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name));
}, e.mark = function (t) {
return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t;
}, e.awrap = function (t) {
return {
__await: t
};
}, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () {
return this;
}), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) {
void 0 === i && (i = Promise);
var a = new AsyncIterator(wrap(t, r, n, o), i);
return e.isGeneratorFunction(r) ? a : a.next().then(function (t) {
return t.done ? t.value : a.next();
});
}, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () {
return this;
}), define(g, "toString", function () {
return "[object Generator]";
}), e.keys = function (t) {
var e = Object(t),
r = [];
for (var n in e) r.push(n);
return r.reverse(), function next() {
for (; r.length;) {
var t = r.pop();
if (t in e) return next.value = t, next.done = !1, next;
}
return next.done = !0, next;
};
}, e.values = values, Context.prototype = {
constructor: Context,
reset: function (e) {
if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t);
},
stop: function () {
this.done = !0;
var t = this.tryEntries[0].completion;
if ("throw" === t.type) throw t.arg;
return this.rval;
},
dispatchException: function (e) {
if (this.done) throw e;
var r = this;
function handle(n, o) {
return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o;
}
for (var o = this.tryEntries.length - 1; o >= 0; --o) {
var i = this.tryEntries[o],
a = i.completion;
if ("root" === i.tryLoc) return handle("end");
if (i.tryLoc <= this.prev) {
var c = n.call(i, "catchLoc"),
u = n.call(i, "finallyLoc");
if (c && u) {
if (this.prev < i.catchLoc) return handle(i.catchLoc, !0);
if (this.prev < i.finallyLoc) return handle(i.finallyLoc);
} else if (c) {
if (this.prev < i.catchLoc) return handle(i.catchLoc, !0);
} else {
if (!u) throw new Error("try statement without catch or finally");
if (this.prev < i.finallyLoc) return handle(i.finallyLoc);
}
}
}
},
abrupt: function (t, e) {
for (var r = this.tryEntries.length - 1; r >= 0; --r) {
var o = this.tryEntries[r];
if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) {
var i = o;
break;
}
}
i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null);
var a = i ? i.completion : {};
return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a);
},
complete: function (t, e) {
if ("throw" === t.type) throw t.arg;
return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y;
},
finish: function (t) {
for (var e = this.tryEntries.length - 1; e >= 0; --e) {
var r = this.tryEntries[e];
if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y;
}
},
catch: function (t) {
for (var e = this.tryEntries.length - 1; e >= 0; --e) {
var r = this.tryEntries[e];
if (r.tryLoc === t) {
var n = r.completion;
if ("throw" === n.type) {
var o = n.arg;
resetTryEntry(r);
}
return o;
}
}
throw new Error("illegal catch attempt");
},
delegateYield: function (e, r, n) {
return this.delegate = {
iterator: values(e),
resultName: r,
nextLoc: n
}, "next" === this.method && (this.arg = t), y;
}
}, e;
}
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) {
try {
var info = gen[key](arg);
var value = info.value;
} catch (error) {
reject(error);
return;
}
if (info.done) {
resolve(value);
} else {
Promise.resolve(value).then(_next, _throw);
}
}
function _asyncToGenerator(fn) {
return function () {
var self = this,
args = arguments;
return new Promise(function (resolve, reject) {
var gen = fn.apply(self, args);
function _next(value) {
asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value);
}
function _throw(err) {
asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err);
}
_next(undefined);
});
};
}
var _process$argv$;
//@ts-ignore
/* babel-plugin-inline-import './template/template.zh.md' */
var TEMPLACE_CN = "\u73B0\u6709\u4EE5\u4E0B\u7684 git diff \u8F93\u51FA\uFF1A\n<diff>\n{{diff}}\n</diff>\n\n\u89D2\u8272\uFF1A\u4F60\u662F\u4E00\u4E2A\u6839\u636E git diff \u4FE1\u606F\u751F\u6210 git commit log \u7684\u5DE5\u5177\uFF0C\u4F60\u4F1A\u4E3A\u4E00\u4E2A\u590D\u6742\u7684\u53D8\u66F4\u751F\u6210\u4E00\u6761\u7CBE\u7B80\u7684 commit log.\n\n\u4EE5\u4E0B\u662F git commit log \u7684\u4E66\u5199\u524D\u7F00\u53CA\u5176\u5BF9\u5E94\u7684\u4F7F\u7528\u573A\u666F\uFF0C\u8BF7\u5148\u6311\u9009\u4E00\u4E2A\u7B26\u5408\u672C\u6B21\u63D0\u4EA4\u7684\u524D\u7F00\uFF1A\n* \u65B0\u529F\u80FD\u5F00\u53D1: feat\n* \u4FEE\u590D bug: fix\n* \u6587\u6863\u4FEE\u6539: docs\n* \u683C\u5F0F\u5316\u4EE3\u7801: style\n* \u91CD\u6784\u4EE3\u7801: refactor\n* \u589E\u52A0\u6D4B\u8BD5: test\n* \u66F4\u65B0\u7248\u672C\u53F7\u3001\u914D\u7F6E\u6587\u4EF6\u7B49: chore\n\n\u7136\u540E\u6309\u7167\u4EE5\u4E0B\u7684\u683C\u5F0F\u8F93\u51FA\u6700\u7EC8\u7684 commit log\uFF1A\n<output>\n[\u524D\u7F00]: \u672C\u6B21\u6240\u6709\u53D8\u66F4\u7684\u7B80\u8981\u63CF\u8FF0\n\n* \u6BCF\u6761\u4FEE\u6539\u7684\u5177\u4F53\u63CF\u8FF0\uFF08\u4E0D\u8D85\u8FC73\u6761\uFF09\n</output>\n\n\u8981\u6C42\uFF1A\n* \u4F7F\u7528\u4E2D\u6587\u8F93\u51FA\n* \u5982\u679C\u6709\u591A\u79CD\u7C7B\u578B\u7684\u4FEE\u6539\uFF0C\u4E5F\u5F3A\u5236\u538B\u7F29\u5230\u4E00\u6761 commit log \u4E2D\uFF0C\u6311\u9009\u51FA\u6700\u91CD\u8981\u7684\u66F4\u6539\u8FDB\u884C\u63CF\u8FF0\n* \u63CF\u8FF0\u4FEE\u6539\u7684\u529F\u80FD\uFF0C\u4E0D\u63CF\u8FF0\u4FEE\u6539\u4E86\u4EC0\u4E48\u6587\u4EF6\uFF0C\u4E5F\u4E0D\u63CF\u8FF0\u4FEE\u6539\u4E86\u4EC0\u4E48\u4EE3\u7801\n\n\u8BF7\u6839\u636E git diff \u7684\u5185\u5BB9\u4EE5\u53CA\u4E0A\u8FF0\u7684\u8981\u6C42\uFF0C\u76F4\u63A5\u7ED9\u51FA\u6700\u7EC8\u7684 commit log\uFF0C\u52A1\u5FC5\u5C06\u6700\u7EC8\u7684\u7ED3\u679C\u5305\u88F9\u5728xml\u6807\u7B7E <output> \u548C </output> \u4E2D\u8F93\u51FA"; //@ts-ignore
/* babel-plugin-inline-import './template/template.en.md' */
var TEMPLACE_EN = "This is the git diff output:\n<diff>\n{{diff}}\n</diff>\n\nRole: You are a tool that generates git commit log based on git diff information.\n\nThe following is the prefix of the git commit log and its corresponding usage scenario:\n* feat: new feature\n* fix: fix bug\n* docs: documentation\n* style: format (code changes that do not affect the running of the code)\n* refactor: refactoring (code changes that are neither new features nor bug fixes)\n* test: add test\n* chore: changes in the build process or auxiliary tools\n\nThe final output format:\n<output>\n[prefix]: brief description of the change\n\n* description of the modified content\n* description of the modified content\n...\n</output>\n\nOther requirements:\n* Start your commit log with the agreed prefix\n* Only one commit log is required\n* The description of the modified content does not exceed 3\n\nPlease give the final commit log directly according to the above requirements, and wrap it in the <output> and </output> tags.";
var homeDir = /*#__PURE__*/os.homedir();
var configFilePath = /*#__PURE__*/path.join(homeDir, '.config/clog-ai/config.json');
var command = (_process$argv$ = process.argv[2]) == null ? void 0 : /*#__PURE__*/_process$argv$.trim();
var isVerbose = /*#__PURE__*/process.argv.includes('--verbose') || /*#__PURE__*/process.argv.includes('-v');
if (command == 'init') {
if (fs.existsSync(configFilePath)) {
console.log('config file already exists, please edit config file: ' + path.resolve(configFilePath));
process.exit(0);
}
var _config = {
language: 'zh',
datasource: 'openai',
openai_api_key: '',
azure_api_key: '',
azure_deployment_id: '',
azure_base_url: '',
azure_model: '',
azure_api_version: ''
};
var dirPath = /*#__PURE__*/path.dirname(configFilePath);
if (!fs.existsSync(dirPath)) {
fs.mkdirSync(dirPath, {
recursive: true
});
}
fs.writeFileSync(configFilePath, JSON.stringify(_config, null, 2));
console.log('init success, please edit config file: ' + path.resolve(configFilePath));
process.exit(0);
}
if (!fs.existsSync(configFilePath)) {
console.log('please run this command first: clog-ai init');
process.exit(0);
}
var config = /*#__PURE__*/getConfig();
//valid_datasource: azure openai
var valid_datasource = ['azure', 'openai'];
if (!config.datasource) {
console.log('Please specify the datasource in the configuration file: ' + path.resolve(configFilePath));
process.exit(0);
}
if (!valid_datasource.includes(config.datasource)) {
console.log('Invalid datasource. Please check the configuration file: ' + path.resolve(configFilePath));
process.exit(0);
}
if (config.datasource === 'openai' && !config.openai_api_key) {
console.log('Data source is OpenAI, but the corresponding configuration is missing. Please add it in the configuration file: ' + path.resolve(configFilePath));
process.exit(0);
}
if (config.datasource === 'azure' && (!config.azure_api_key || !config.azure_deployment_id || !config.azure_base_url || !config.azure_model || !config.azure_api_version)) {
console.log('Data source is Azure, but the corresponding configuration is missing. Please add it in the configuration file: ' + path.resolve(configFilePath));
process.exit(0);
}
function getConfig() {
var content = fs.readFileSync(configFilePath).toString();
return JSON.parse(content || '{}');
}
function gptRequestAzure(_x) {
return _gptRequestAzure.apply(this, arguments);
}
function _gptRequestAzure() {
_gptRequestAzure = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(prompt) {
var res;
return _regeneratorRuntime().wrap(function _callee2$(_context2) {
while (1) switch (_context2.prev = _context2.next) {
case 0:
_context2.next = 2;
return axios.post(config.azure_base_url + "/openai/deployments/" + config.azure_deployment_id + "/chat/completions?api-version=" + config.azure_api_version, {
model: 'gpt-3.5-turbo-16k',
messages: [{
role: 'user',
content: prompt
}],
temperature: 0,
top_p: 1,
frequency_penalty: 0,
presence_penalty: 0
}, {
headers: {
'api-key': config.azure_api_key
},
timeout: 100000
});
case 2:
res = _context2.sent;
return _context2.abrupt("return", res.data.choices[0].message.content);
case 4:
case "end":
return _context2.stop();
}
}, _callee2);
}));
return _gptRequestAzure.apply(this, arguments);
}
function gptRequestOpenai(_x2) {
return _gptRequestOpenai.apply(this, arguments);
}
function _gptRequestOpenai() {
_gptRequestOpenai = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(prompt) {
var res;
return _regeneratorRuntime().wrap(function _callee3$(_context3) {
while (1) switch (_context3.prev = _context3.next) {
case 0:
_context3.next = 2;
return axios.post("https://api.openai.com/v1/chat/completions", {
model: 'gpt-3.5-turbo-16k',
messages: [{
role: 'user',
content: prompt
}],
temperature: 0,
top_p: 1,
frequency_penalty: 0,
presence_penalty: 0
}, {
headers: {
Authorization: "Bearer " + config.openai_api_key
},
timeout: 100000
});
case 2:
res = _context3.sent;
return _context3.abrupt("return", res.data.choices[0].message.content);
case 4:
case "end":
return _context3.stop();
}
}, _callee3);
}));
return _gptRequestOpenai.apply(this, arguments);
}
var index = /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
var diff, prompt, load, _res$match, res, commitLog, rl;
return _regeneratorRuntime().wrap(function _callee$(_context) {
while (1) switch (_context.prev = _context.next) {
case 0:
// 执行 git diff，获取变更的文件内容
diff = child_process.execSync('git diff HEAD').toString().substring(0, 8000);
prompt = (config.language == 'zh' ? TEMPLACE_CN : TEMPLACE_EN).replace('{{diff}}', diff);
load = loading({
text: 'Generating commit log...',
color: 'yellow',
interval: 100,
frames: ['◰', '◳', '◲', '◱']
}).start();
_context.prev = 3;
if (isVerbose) {
console.log('--------- Input Prompt ----------');
console.log(prompt);
}
if (!(config.datasource === 'azure')) {
_context.next = 11;
break;
}
_context.next = 8;
return gptRequestAzure(prompt);
case 8:
_context.t0 = _context.sent;
_context.next = 14;
break;
case 11:
_context.next = 13;
return gptRequestOpenai(prompt);
case 13:
_context.t0 = _context.sent;
case 14:
res = _context.t0;
if (isVerbose) {
console.log('\n--------- Output ----------');
console.log(res);
console.log('-------------------');
}
commitLog = (_res$match = res.match(/<output>([\s\S]*)<\/output>/)) == null || (_res$match = _res$match[1]) == null ? void 0 : _res$match.trim();
if (commitLog) {
_context.next = 19;
break;
}
throw new Error('No commit log generated');
case 19:
load.stop();
load.succeed('Generate commit log success');
console.log('-------------------');
console.log(commitLog);
rl = readline.createInterface({
input: process.stdin,
output: process.stdout
});
console.log('-------------------');
rl.question('Submit git commit with the log? (yes/no) ', function (answer) {
if (answer.toLowerCase() === 'yes' || answer.toLowerCase() === 'y') {
child_process.execSync("git commit -m \"" + commitLog + "\"");
load.succeed('commit success');
}
rl.close();
});
_context.next = 33;
break;
case 28:
_context.prev = 28;
_context.t1 = _context["catch"](3);
load.stop();
load.fail('Generate commit log fail');
console.error(_context.t1.message);
case 33:
case "end":
return _context.stop();
}
}, _callee, null, [[3, 28]]);
}));
exports.default = index;
//# sourceMappingURL=clog-ai.cjs.development.js.map