generated from ztimson/template
Added ticket refinement bot
This commit is contained in:
106
src/refine.mjs
Normal file
106
src/refine.mjs
Normal file
@@ -0,0 +1,106 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
import {Ai} from '@ztimson/ai-utils';
|
||||
import * as os from 'node:os';
|
||||
import * as dotenv from 'dotenv';
|
||||
import * as fs from 'node:fs';
|
||||
import * as path from 'node:path';
|
||||
|
||||
dotenv.config({quiet: true});
|
||||
dotenv.config({path: '.env.local', override: true, quiet: true});
|
||||
|
||||
(async () => {
|
||||
let p = process.argv[process.argv.length - 1];
|
||||
if(p === 'review' || p.endsWith('review.mjs')) p = null;
|
||||
if(!/(\/|[A-Z]:)/.test(p)) p = path.join(process.cwd(), p);
|
||||
|
||||
if(!p || !fs.existsSync(p)) throw new Error('Please provide a template');
|
||||
|
||||
const git = process.env['GIT_HOST'],
|
||||
owner = process.env['GIT_OWNER'],
|
||||
repo = process.env['GIT_REPO'],
|
||||
auth = process.env['GIT_TOKEN'],
|
||||
ticket = process.env['TICKET'],
|
||||
host = process.env['AI_HOST'],
|
||||
model = process.env['AI_MODEL'],
|
||||
token = process.env['AI_TOKEN'];
|
||||
|
||||
console.log(`Processing issue #${ticket}`);
|
||||
|
||||
// Fetch issue
|
||||
const issueRes = await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}`, {
|
||||
headers: {'Authorization': `token ${auth}`}
|
||||
});
|
||||
if(!issueRes.ok) throw new Error(`${issueRes.status} ${await issueRes.text()}`);
|
||||
const issueData = await issueRes.json();
|
||||
if(!issueData.labels?.some(l => l.name === 'Review/AI')) {
|
||||
console.log('Skipping');
|
||||
return process.exit();
|
||||
}
|
||||
|
||||
let readme = '', readmeP = path.join(process.cwd(), 'README.md');
|
||||
if(fs.existsSync(readmeP)) readme = fs.readFileSync(readmeP, 'utf-8');
|
||||
const template = fs.readFileSync(p, 'utf-8');
|
||||
|
||||
let options = {ollama: {model, host}};
|
||||
if(host === 'anthropic') options = {anthropic: {model, token}};
|
||||
else if(host === 'openai') options = {openAi: {model, token}};
|
||||
const ai = new Ai({
|
||||
...options,
|
||||
model: [host, model],
|
||||
path: process.env['path'] || os.tmpdir(),
|
||||
system: `You are a ticket formatter. Transform raw issue descriptions into structured tickets.
|
||||
|
||||
**CRITICAL RULES:**
|
||||
1. Identify the ticket type (Bug, DevOps, Enhancement, Refactor, Security)
|
||||
2. Output MUST only contain the new ticket information in markdown, no extra fluff
|
||||
3. Follow the template structure EXACTLY:
|
||||
- Title format: [Module] - [Verb] [noun]
|
||||
Example: Storage - Fix file uploads
|
||||
- Fill in the identified ticket type
|
||||
- Write a clear description
|
||||
- For bugs: fill Steps to Reproduce with numbered list
|
||||
- For enhancements/refactors: REMOVE the Steps to Reproduce section entirely
|
||||
- Acceptance Criteria: convert requirements into checkboxes (- [ ])
|
||||
- Weight scoring (0-5 each):
|
||||
* Size: Number of modules, layers & files affected by change
|
||||
* Complexity: Technical difficulty to implement
|
||||
* Unknowns: Research/uncertainty in work estimation
|
||||
* Calculate Total as sum of the three
|
||||
- Remove sections that are not applicable based on ticket type
|
||||
- Use proper markdown headers (##)
|
||||
|
||||
**README:**
|
||||
\`\`\`markdown
|
||||
${readme.trim() || 'No README available'}
|
||||
\`\`\`
|
||||
|
||||
**TEMPLATE:**
|
||||
\`\`\`markdown
|
||||
${template.trim()}
|
||||
\`\`\`
|
||||
|
||||
Output ONLY the formatted ticket, no explanation.`
|
||||
});
|
||||
|
||||
const messages = await ai.language.ask(`Title: ${issueData.title}\n\nDescription:\n${issueData.body || 'No description provided'}`);
|
||||
const content = messages.pop().content;
|
||||
const title = /^# (.+)$/m.exec(content)?.[1] || issueData.title;
|
||||
const typeMatch = /^## Type:\s*(.+)$/m.exec(content);
|
||||
const type = typeMatch?.[1]?.split('/')[0]?.trim() || 'Unassigned';
|
||||
const body = content.replace(/^# .+$/m, '').replace(/^## Type:.+$/m, '').trim();
|
||||
const updateRes = await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}`, {
|
||||
method: 'PATCH',
|
||||
headers: {
|
||||
'Authorization': `token ${auth}`,
|
||||
'Content-Type': 'application/json'
|
||||
},
|
||||
body: JSON.stringify({
|
||||
title,
|
||||
body,
|
||||
labels: [`Type/${type[0].toUpperCase() + type.slice(1).toLowerCase()}`]
|
||||
})
|
||||
});
|
||||
if(!updateRes.ok) throw new Error(`${updateRes.status} ${await updateRes.text()}`);
|
||||
console.log(body);
|
||||
})();
|
||||
Reference in New Issue
Block a user