Merge pull request 'review-labels' (#14) from review-labels into master

Reviewed-on: #14

.github/issue_template/ai-refinement.md

@@ -0,0 +1,19 @@
+---
+
+name: "AI refinement"
+about: "Use AI to refine ticket"
+ref: "develop"
+labels:
+- Review/AI
+
+---
+
+Describe your request:
+
+What are you trying to do and what's happening?
+
+How can it be fixed or improved?
+
+Steps to reproduce?
+
+Any other useful information? Logs, screenshots, steps to reproduce?

.github/pull_request_template.md

@@ -9,7 +9,5 @@
 
 ## Checklist
 <!-- Complete after creating PR -->
- - [ ] Linked issues
+ - [ ] Reviewed changes (or use `Review/AI` label)
- - [ ] Reviewed changes
  - [ ] Updated comments/documentation
- 

.github/workflows/build.yml

@@ -1,44 +1,45 @@
-name: Build and publish
+name: Publish Library
-run-name: Build and publish
+run-name: Publish Library
 
 on:
     push:
 
 jobs:
-    container:
+    build:
-        name: Build Container
+        name: Build NPM Project
         runs-on: ubuntu-latest
-        container: docker
+        container: node:alpine
         steps:
-            -   name: Build Container
+            - name: Clone Repository
-                run: |
+              uses: ztimson/actions/clone@develop
-                    git clone -b "${{github.ref_name}}" "$(echo ${{github.server_url}}/${{github.repository}}.git | sed s%://%://${{github.token}}@% )" .
-                    DOCKER_HUB=$([ -n "${{secrets.DOCKER_HUB_USER}}" ] && [ -n "${{secrets.DOCKER_HUB_TOKEN}}" ] && [ -n "${{secrets.DOCKER_HUB_IMAGE}}" ] && echo "true" || echo "false")
-                    REGISTRY="$(echo "${{github.server_url}}" | sed -E 's|https?://||')"
-                    VERSION="$(cat package.json | grep version | grep -Eo '[0-9][[:alnum:]\.\/\-]+')"
 
-                    docker login -u "${{github.repository_owner}}" -p "${{secrets.DEPLOY_TOKEN}}" "$REGISTRY"
+            - name: Publish Library
-                    if [ "$DOCKER_HUB" = "true" ]; then docker login -u "${{secrets.DOCKER_HUB_USER}}" -p "${{secrets.DOCKER_HUB_TOKEN}}" docker.io; fi
+              run: |
+                  if [ "${{github.ref_name}}" = "master" ]; then
+                    REGISTRY="${{github.server_url}}/api/packages/${{github.repository_owner}}/npm/"
+                    npm set registry "$REGISTRY"
+                    npm set $(echo $REGISTRY | sed s%http:%% | sed s%https:%% ):_authToken "${{secrets.DEPLOY_TOKEN}}"
+                    npm publish || echo "Failed to publish"
 
-                    docker build -t "$REGISTRY/${{github.repository}}:${{github.ref_name}}" .
+                    REGISTRY="https://registry.npmjs.org/"
-                    docker push "$REGISTRY/${{github.repository}}:${{github.ref_name}}"
+                    npm set registry "$REGISTRY"
-                    if [ "$DOCKER_HUB" = "true" ]; then
+                    npm set $(echo $REGISTRY | sed s%http:%% | sed s%https:%% ):_authToken "${{secrets.NPM_TOKEN}}"
-                      docker tag "$REGISTRY/${{github.repository}}:${{github.ref_name}}" "docker.io/${{secrets.DOCKER_HUB_IMAGE}}:${{github.ref_name}}"
+                    npm publish || echo "Failed to publish"
-                      docker push "docker.io/${{secrets.DOCKER_HUB_IMAGE}}:${{github.ref_name}}"
+                  fi
-                    fi
+    tag:
+        name: Tag Version
+        needs: build
+        if: ${{github.ref_name}} == 'release'
+        runs-on: ubuntu-latest
+        container: node:alpine
+        steps:
+            -   name: Clone Repository
+                uses: ztimson/actions/clone@develop
 
-                    if [ "${{github.ref_name}}" = "master" ]; then
+            -   name: Get Version Number
-                      docker tag "$REGISTRY/${{github.repository}}:${{github.ref_name}}" "$REGISTRY/${{github.repository}}:$VERSION"
+                run: echo "VERSION=$(cat package.json | grep version | grep -Eo ':.+' | grep -Eo '[[:alnum:]\.\/\-]+')" >> $GITHUB_ENV
-                      docker push "$REGISTRY/${{github.repository}}:$VERSION"
-                      if [ "$DOCKER_HUB" = "true" ]; then
-                        docker tag "$REGISTRY/${{github.repository}}:${{github.ref_name}}" "docker.io/${{secrets.DOCKER_HUB_IMAGE}}:$VERSION"
-                        docker push "docker.io/${{secrets.DOCKER_HUB_IMAGE}}:$VERSION"
-                      fi
 
-                      docker tag "$REGISTRY/${{github.repository}}:${{github.ref_name}}" "$REGISTRY/${{github.repository}}:latest"
+            -   name: Tag Version
-                      docker push "$REGISTRY/${{github.repository}}:latest"
+                uses: ztimson/actions/tag@develop
-                      if [ "$DOCKER_HUB" = "true" ]; then
+                with:
-                        docker tag "$REGISTRY/${{github.repository}}:${{github.ref_name}}" "docker.io/${{secrets.DOCKER_HUB_IMAGE}}:latest"
+                    tag: ${{env.VERSION}}
-                        docker push "docker.io/${{secrets.DOCKER_HUB_IMAGE}}:latest"
-                      fi
-                    fi

.github/workflows/code-review.yml

@@ -1,25 +1,22 @@
 name: Code review
+
 on:
     pull_request:
-        types: [opened, synchronize, reopened]
+        types: [opened, synchronize, reopened, labeled]
 
 jobs:
     review:
         runs-on: ubuntu-latest
+        container: node:22
         steps:
-            - name: Checkout PR
+            - name: Fetch code
-              uses: actions/checkout@v4
-              with:
-                  fetch-depth: 0
-                  ref: ${{ github.event.pull_request.head.sha }}
-
-            - name: Fetch base branch
               run: |
+                  git clone "$(echo ${{github.server_url}}/${{github.repository}}.git | sed s%://%://${{github.token}}@% )" .
+                  git checkout ${{ github.event.pull_request.head.sha }}
                   git fetch origin ${{ github.event.pull_request.base.ref }}
-                  git branch -r
 
             - name: Run AI Review
-              uses: docker://git.zakscode.com/zakscode/ai-reporter:latest
+              run: npx -y -p @ztimson/ai-agents@latest review $GITHUB_WORKSPACE
               env:
                   AI_HOST: anthropic
                   AI_MODEL: claude-sonnet-4-5

.github/workflows/ticket-refinement.yml

@@ -0,0 +1,27 @@
+name: Ticket refinement
+
+on:
+    issues:
+        types: [labeled]
+
+jobs:
+    format:
+        runs-on: ubuntu-latest
+        container: node:22
+        steps:
+            - name: Fetch code
+              run: |
+                  git clone "$(echo ${{github.server_url}}/${{github.repository}}.git | sed s%://%://${{github.token}}@% )" .
+                  git checkout ${{ github.event.repository.default_branch }}
+
+            - name: Run AI Formatter
+              run: npx -y -p @ztimson/ai-agents@latest refine
+              env:
+                  AI_HOST: anthropic
+                  AI_MODEL: claude-sonnet-4-5
+                  AI_TOKEN: ${{ secrets.ANTHROPIC_TOKEN }}
+                  GIT_HOST: ${{ github.server_url }}
+                  GIT_OWNER: ${{ github.repository_owner }}
+                  GIT_REPO: ${{ github.event.repository.name }}
+                  GIT_TOKEN: ${{ secrets.ASSISTANT_TOKEN }}
+                  TICKET: ${{ github.event.issue.number }}

Dockerfile

@@ -1,7 +0,0 @@
-FROM node:22
-
-COPY . /ai
-RUN cd /ai && npm ci && mkdir -p /github/workspace
-
-WORKDIR /github/workspace
-ENTRYPOINT ["npx", "--yes", "/ai", "review", "/github/workspace"]

README.md

@@ -3,32 +3,32 @@
   <br />
 
   <!-- Logo -->
-  <img src="https://git.zakscode.com/repo-avatars/d2c56ffd0220751c2f4a9f6fc1e8125a9a63aeed452ae4f4ab696c084330faa2" alt="Logo" width="200" height="200">
+  <img src="https://git.zakscode.com/repo-avatars/309c233243bcd1c1e9b3f359ec3f59769bb01b655e8ed7b32587781be4c8b21c" alt="Logo" width="200" height="200">
 
   <!-- Title -->
-### AI Reviewer
+### AI Agents
 
   <!-- Description -->
-Automated AI-powered code review for pull requests 🤖
+AI-powered Gitea agents for automating reviews and administration
 
   <!-- Repo badges -->
-[![Version](https://img.shields.io/badge/dynamic/json.svg?label=Version&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-reviewer/tags&query=$[0].name)](https://git.zakscode.com/ztimson/ai-reviewer/tags)
+[![Version](https://img.shields.io/badge/dynamic/json.svg?label=Version&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents/tags&query=$[0].name)](https://git.zakscode.com/ztimson/ai-agents/tags)
-[![Pull Requests](https://img.shields.io/badge/dynamic/json.svg?label=Pull%20Requests&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-reviewer&query=open_pr_counter)](https://git.zakscode.com/ztimson/ai-reviewer/pulls)
+[![Pull Requests](https://img.shields.io/badge/dynamic/json.svg?label=Pull%20Requests&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents&query=open_pr_counter)](https://git.zakscode.com/ztimson/ai-agents/pulls)
-[![Issues](https://img.shields.io/badge/dynamic/json.svg?label=Issues&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-reviewer&query=open_issues_count)](https://git.zakscode.com/ztimson/ai-reviewer/issues)
+[![Issues](https://img.shields.io/badge/dynamic/json.svg?label=Issues&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents&query=open_issues_count)](https://git.zakscode.com/ztimson/ai-agents/issues)
 
 <!-- Links -->
 ---
   <div>
-    <a href="https://git.zakscode.com/ztimson/ai-reviewer/releases" target="_blank">Release Notes</a>
+    <a href="https://git.zakscode.com/ztimson/ai-agents/releases" target="_blank">Release Notes</a>
-    • <a href="https://git.zakscode.com/ztimson/ai-reviewer/issues/new?template=.github%2fissue_template%2fbug.md" target="_blank">Report a Bug</a>
+    • <a href="https://git.zakscode.com/ztimson/ai-agents/issues/new?template=.github%2fissue_template%2fbug.md" target="_blank">Report a Bug</a>
-    • <a href="https://git.zakscode.com/ztimson/ai-reviewer/issues/new?template=.github%2fissue_template%2fenhancement.md" target="_blank">Request a Feature</a>
+    • <a href="https://git.zakscode.com/ztimson/ai-agents/issues/new?template=.github%2fissue_template%2fenhancement.md" target="_blank">Request a Feature</a>
   </div>
 
   ---
 </div>
 
 ## Table of Contents
-- [AI Reviewer](#top)
+- [AI Agents](#top)
 	- [About](#about)
 		- [Built With](#built-with)
 	- [Setup](#setup)
@@ -37,7 +37,9 @@ Automated AI-powered code review for pull requests 🤖
 
 ## About
 
-Automated code reviewer that uses AI to analyze git diffs and provide inline comments on pull requests. Supports Anthropic, OpenAI, and Ollama models with tool-based reviewing for precise feedback.
+Only supports Gitea
+
+Use LLM models from Anthropic, OpenAI, or Ollama to automate ticket refinement, code reviews, and releases.
 
 ### Built With
 [![Docker](https://img.shields.io/badge/Docker-384d54?style=for-the-badge&logo=docker)](https://docker.com/)
@@ -53,43 +55,11 @@ Automated code reviewer that uses AI to analyze git diffs and provide inline com
   </h3>
 </summary>
 
+#### Prerequisites
+- [Node.js](https://nodejs.org/en/download)
+
 #### Instructions
-1. Add the following git action:
+1. Run using npx: `npx -y @ztimson/ai-agents@latest review`
-```yaml
-name: Code review
-on:
-    pull_request:
-        types: [opened, synchronize, reopened]
-
-jobs:
-    review:
-        runs-on: ubuntu-latest
-        steps:
-            - name: Checkout PR
-              uses: actions/checkout@v4
-              with:
-                  fetch-depth: 0
-                  ref: ${{ github.event.pull_request.head.sha }}
-
-            - name: Fetch base branch
-              run: |
-                  git fetch origin ${{ github.event.pull_request.base.ref }}
-                  git branch -r
-
-            - name: Run AI Review
-              uses: docker://git.zakscode.com/zakscode/ai-reporter:latest
-              env:
-                  AI_HOST: anthropic
-                  AI_MODEL: claude-sonnet-4-5
-                  AI_TOKEN: ${{ secrets.ANTHROPIC_TOKEN }}
-                  GIT_HOST: ${{ github.server_url }}
-                  GIT_OWNER: ${{ github.repository_owner }}
-                  GIT_REPO: ${{ github.event.repository.name }}
-                  GIT_TOKEN: ${{ secrets.ASSISTANT_TOKEN }}
-                  GIT_BRANCH: origin/${{ github.event.pull_request.base.ref }}
-                  PULL_REQUEST: ${{ github.event.pull_request.number }}
-
-```
 
 </details>
 

package.json

@@ -1,17 +1,22 @@
 {
-	"name": "ai-reviewer",
+	"name": "@ztimson/ai-agents",
-	"version": "0.0.0",
+	"version": "0.1.2",
-	"scripts": {
+	"description": "AI agents",
-		"review": "node src/review.mjs"
+	"keywords": ["ai", "review"],
-	},
+	"author": "ztimson",
-	"keywords": [],
-	"author": "",
 	"license": "ISC",
-	"description": "",
+	"type": "module",
+	"bin": {
+		"refine": "./src/refine.mjs",
+		"review": "./src/review.mjs"
+	},
 	"dependencies": {
 		"@ztimson/ai-utils": "^0.2.4",
 		"@ztimson/node-utils": "^1.0.7",
 		"@ztimson/utils": "^0.28.3",
 		"dotenv": "^17.2.3"
-	}
+	},
+	"files": [
+		"src/"
+	]
 }

src/refine.mjs

@@ -0,0 +1,217 @@
+#!/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, debug: false});
+dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
+
+(async () => {
+	let p = process.argv[process.argv.length - 1];
+	if(p === 'refine' || p.endsWith('refine.mjs')) p = null;
+	if(!/^(\/|[A-Z]:)/m.test(p)) p = path.join(process.cwd(), p);
+
+	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'],
+		labelDupe = process.env['LABELS_DUPE'] || 'Review/Duplicate',
+		labelEnabled = process.env['LABEL_ENABLED'] || 'Review/AI',
+		labelsReq = process.env['LABELS_REQ'] || 'Kind/Aesthetic,Kind/Bug,Kind/DevOps,Kind/Document,Kind/Enhancement,Kind/Refactor,Kind/Security',
+		labelsOpt = process.env['LABELS_OPT'] || 'Breaking,Priority,QA',
+		token = process.env['AI_TOKEN'];
+
+	console.log(`Processing issue #${ticket}`);
+
+	// Fetch issue
+	const issueData = await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}`, {
+		headers: {'Authorization': `token ${auth}`}
+	}).then(async resp => {
+		if(resp.ok) return  resp.json();
+		else throw new Error(`${resp.status} ${await resp.text()}`);
+	});
+	if(issueData.labels?.length !== 1 || issueData.labels[0]?.name !== labelEnabled) {
+		console.log('Skipping');
+		return process.exit();
+	}
+
+	// Gather readme & template
+	let title = '', labels = [], readme = '', readmeP = path.join(process.cwd(), 'README.md');
+	if(fs.existsSync(readmeP)) readme = fs.readFileSync(readmeP, 'utf-8');
+	const template = p ? fs.readFileSync(p, 'utf-8') : `## Description
+
+A clear explanation of the request
+
+---
+
+## Current Behavior
+
+what's happening now or the current state/gap
+
+## Expected Behavior
+
+What should happen instead
+
+## Steps to Reproduce / Desired Flow
+
+1. First step
+2. Second step
+3. Third step
+
+## Additional Context
+
+Logs, screenshots, links, related issues
+
+## Acceptance Criteria
+
+- [ ] Todo requirement
+- [X] Completed requirement
+
+## Technical Notes
+
+Implementation details, constraints, dependencies, design decisions
+
+
+| Effort / Weight | Score    |
+|-----------------|----------|
+| Size            | 0-5      |
+| Complexity      | 0-5      |
+| Unknowns        | 0-5      |
+| **Total**       | **0-15** |
+`;
+
+	// Create AI
+	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(),
+		tools: [{
+			name: 'title',
+			description: 'Set the ticket title, must be called EXACTLY ONCE',
+			args: {title: {type: 'string', description: 'Ticket title, must match format: Module - Verb noun', required: true}},
+			fn: (args) => title = args.title
+		}, {
+			name: 'add_label',
+			description: 'Add a label to the ticket',
+			args: {label: {type: 'string', description: 'Label name', required: true}},
+			fn: async (args) => {
+				labels.push(args.label);
+				return await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/labels`, {
+					method: 'POST',
+					headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
+					body: `{"labels":["${args.label}"]}`
+				}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
+			}
+		}],
+		system: `Transform raw tickets into structured markdown following the template EXACTLY.
+
+**MANDATORY STEPS:**
+1. Call \`title\` tool EXACTLY ONCE in format: "[Module] - [Verb] [subject]" (example: Storage - fix file uploads)
+2. Identify one label from each group which best applies to the ticket: ${labelsReq.replaceAll(',', ', ')}
+3. Call the \`add_label\` tool ONCE FOR EVERY LABEL identified in the previous step
+4. Filter the following labels to any that apply to this ticket: ${labelsOpt.replaceAll(',', ', ')}
+5. Call the \`add_label\` tool ONCE FOR EVERY LABEL identified in the previous step
+6. Output the new ticket description in formatted markdown matching the following rules:
+
+**TEMPLATE RULES:**
+- Use ## headers (match template exactly)
+- Description: Clear summary of the request
+- Current Behavior: What's happening now (remove for Document tickets)
+- Expected Behavior: What should happen (remove for Document tickets)
+- Steps to Reproduce: Numbered list for bugs, flow for enhancements, remove if not applicable
+- Additional Context: Logs, screenshots, links provided by user
+- Acceptance Criteria: Convert to checkboxes (- [ ] format)
+- Technical Notes: Implementation approach, constraints, dependencies
+- Weight table (use exact format below):
+
+| Effort / Weight | Score    |
+|-----------------|----------|
+| Size            | 0-5      |
+| Complexity      | 0-5      |
+| Unknowns        | 0-5      |
+| **Total**       | **0-15** |
+
+**SCORING:**
+- Size: # of modules/layers/files affected
+- Complexity: Technical difficulty
+- Unknowns: Research/uncertainty needed
+
+**PROJECT README:**
+\`\`\`markdown
+${readme.trim() || 'No README available'}
+\`\`\`
+
+**TEMPLATE:**
+\`\`\`markdown
+${template.trim()}
+\`\`\`
+
+Output ONLY markdown. No explanations, labels, or extra formatting.`});
+
+	// Format ticket with AI
+	const messages = await ai.language.ask(`Title: ${issueData.title}\n\nDescription:\n${issueData.body || 'No description provided'}`).catch(() => []);
+	const body = messages?.pop()?.content;
+	if(!body) throw new Error('Invalid response from AI');
+
+	// Check for duplicates
+	const repoInfo = await fetch(`${git}/api/v1/repos/${owner}/${repo}`, {headers: {'Authorization': `token ${auth}`},}).then(resp => resp.ok ? resp.json() : null);
+	const search = await fetch(`${git}/api/v1/repos/issues/search`, {
+		method: 'POST',
+		headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
+		body: JSON.stringify({
+			owner,
+			priority_repo_id: repoInfo.id,
+			type: 'issues',
+			limit: 3,
+			q: title
+		})
+	}).then(resp => resp.ok ? resp.json() : []);
+	let dupeId = null;
+	const dupeIds = search.map(t => t.id);
+	const dupes = search.map(t => `ID: ${t.id}\nTitle: ${t.title}\n\`\`\`markdown\n${t.body}\n\`\`\``).join('\n\n');
+	const hasDuplicates = (await ai.language.ask(`ID: ${issueData.id}\nTitle: ${title}\n\`\`\`markdown\n${body}\n\`\`\``, {
+		system: `Your job is to identify duplicates. Respond ONLY with the duplicate's ID number or "NONE" if no match exists\n\n${dupes}`
+	}))?.pop()?.content;
+
+	// Handle duplicates
+	if(hasDuplicates && !hasDuplicates.toUpperCase().includes('NONE') && (dupeId = dupeIds.find(id => id == hasDuplicates.trim())) != null && dupeId != issueData.id) {
+		await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/comments`, {
+			method: 'POST',
+			headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
+			body: `{"body": "Duplicate of #${dupeId}"}`
+		}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
+		await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/labels`, {
+			method: 'POST',
+			headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
+			body: `{"labels":["${labelDupe}"]}`
+		}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
+		await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}`, {
+			method: 'PATCH',
+			headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
+			body: '{"state": "closed"}'
+		}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
+		console.log('Duplicate');
+		return process.exit();
+	}
+
+	// Update ticket
+	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})
+	}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
+
+	console.log(`Title: ${title}\nLabels: ${labels.join(', ')}\nBody:\n${body}`);
+})().catch(err => {
+	console.error(`Error: ${err.message || err.toString()}`);
+	process.exit(1);
+});

src/review.mjs

@@ -1,3 +1,5 @@
+#!/usr/bin/env node
+
 import {Ai} from '@ztimson/ai-utils';
 import {$} from '@ztimson/node-utils';
 import * as os from 'node:os';
@@ -5,26 +7,55 @@ import * as path from 'node:path';
 import * as fs from 'node:fs';
 import * as dotenv from 'dotenv';
 
-dotenv.config();
+dotenv.config({quiet: true, debug: false});
-dotenv.config({path: '.env.local', override: true});
+dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
 
 (async () => {
-	const root = process.argv[2] || process.cwd(),
+	let p = process.argv[process.argv.length - 1];
-		branch = process.env['GIT_BRANCH'] || await $`cd ${root} && git symbolic-ref refs/remotes/origin/HEAD`,
+	if(p === 'review' || p.endsWith('review.mjs')) p = null;
+
+	const root = p || process.cwd(),
 		git = process.env['GIT_HOST'],
 		owner = process.env['GIT_OWNER'],
 		repo = process.env['GIT_REPO'],
 		auth = process.env['GIT_TOKEN'],
+		labelEnabled = process.env['LABEL_ENABLED'] || 'Review/AI',
 		pr = process.env['PULL_REQUEST'],
 		host = process.env['AI_HOST'],
 		model = process.env['AI_MODEL'],
 		token = process.env['AI_TOKEN'];
 
+	console.log(`Reviewing: ${root}\n`);
+	const info = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}`)
+		.then(async resp => {
+			if(resp.ok) return resp.json();
+			throw new Error(`${resp.status} ${await resp.text()}`);
+		});
+	if(!info.labels.some(l => l.name === labelEnabled)) {
+		console.log('Skipping');
+		return process.exit();
+	}
+
+	const branch = process.env['GIT_BRANCH'] || await $`cd ${root} && git symbolic-ref refs/remotes/origin/HEAD`;
 	const comments = [];
 	const commit = await $`cd ${root} && git log -1 --pretty=format:%H`;
 	const gitDiff = await $`cd ${root} && git diff ${branch}`;
 
-	console.log(`Reviewing: ${root}\n`);
+	if(!gitDiff) {
+		console.warn('No diff found');
+		return process.exit();
+	}
+
+	let existingComments = 'Existing Comments:\n';
+	if(git && pr) {
+		const reviews = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews`, {
+			headers: {'Authorization': `token ${auth}`}
+		}).then(resp => resp.ok ? resp.json() : []);
+		const comments = await Promise.all(reviews.map(r => fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews/${r.id}/comments`, {
+			headers: {'Authorization': `token ${auth}`}
+		}).then(resp => resp.ok ? resp.json() : [])));
+		existingComments += comments.flat().map(c => `${c.path}:${c.position}\n${c.body}`).join('\n\n');
+	}
 
 	let options = {ollama: {model, host}};
 	if(host === 'anthropic') options = {anthropic: {model, token}};
@@ -33,7 +64,7 @@ dotenv.config({path: '.env.local', override: true});
 		...options,
 		model: [host, model],
 		path: process.env['path'] || os.tmpdir(),
-		system: `You are a code reviewer. Analyze the git diff and use the \`recommend\` tool for EACH issue you find. You must call \`recommend\` exactly once for every bug or improvement opportunity. After making all recommendations, provide a brief bullet point summary in markdown.`,
+		system: `You are a code reviewer. Analyze the git diff and use the \`recommend\` tool for EACH issue you find. You must call \`recommend\` exactly once for every bug or improvement opportunity directly related to changes. Ignore formatting recommendations. After making all recommendations, provide a quick 75 words or less sitrep.${existingComments}`,
 		tools: [{
 			name: 'read_file',
 			description: 'Read contents of a file',
@@ -67,12 +98,16 @@ dotenv.config({path: '.env.local', override: true});
 		}]
 	});
 
-	if(!gitDiff) {
+	const messages = await ai.language.ask(`Title: ${info.title || 'None'}
-		console.warn('No diff found');
+Description:
-		return process.exit();
+\`\`\`md
-	}
+${info.body || 'None'}
+\`\`\`
 
-	const messages = await ai.language.ask(gitDiff);
+Git Diff:
+\`\`\`
+${gitDiff}
+\`\`\``);
 	const summary = messages.pop().content;
 	if(git) {
 		const res = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews`, {
@@ -91,4 +126,7 @@ dotenv.config({path: '.env.local', override: true});
 		if(!res.ok) throw new Error(`${res.status} ${await res.text()}`);
 	}
 	console.log(comments.map(c => `${c.path}${c.new_position ? `:${c.new_position}` : ''}\n${c.body}`).join('\n\n') + '\n\n' + summary);
-})();
+})().catch(err => {
+	console.error(`Error: ${err.message || err.toString()}`);
+	process.exit(1);
+});