Compare commits

..

No commits in common. "master" and "v0.1.4" have entirely different histories.

65 changed files with 419 additions and 3178 deletions

View File

@ -0,0 +1,37 @@
name: 'Auto-complete TODO'
on:
pull_request:
types: [opened, synchronize, reopened]
jobs:
move-todo:
runs-on: ubuntu-gitea
steps:
- name: 'Clone PR branch, move TODO, push update'
env:
PAT_TOKEN: ${{ secrets.PAT_TOKEN }}
SERVER_URL: ${{ github.server_url }}
REPO_OWNER: ${{ github.repository_owner }}
REPO_NAME: ${{ github.event.pull_request.head.repo.name }}
PR_BRANCH: ${{ github.head_ref }}
run: |
git clone https://${PAT_TOKEN}@${SERVER_URL}/${REPO_OWNER}/${REPO_NAME}.git pr-temp || exit 1
cd pr-temp
git checkout ${PR_BRANCH}
BRANCH=$(git rev-parse --abbrev-ref HEAD)
TODO_NAME="${BRANCH#feature/}.md"
if [[ "${BRANCH}" == feature/* && -f todo/queued/${TODO_NAME} ]]; then
mkdir -p todo/completed
mv todo/queued/${TODO_NAME} todo/completed/
git config user.name 'Gitea Actions Bot'
git config user.email 'actions@noreply.local'
git add todo/
git commit -m "chore: auto-complete ${TODO_NAME} via Gitea Actions"
git push https://${PAT_TOKEN}@${SERVER_URL}/${REPO_OWNER}/${REPO_NAME}.git ${PR_BRANCH}
echo "✅ Moved todo/queued/${TODO_NAME} → completed/"
else
echo " No action: branch='${BRANCH}', expected 'feature/*' with todo/queued/${TODO_NAME}"
fi
cd ..
rm -rf pr-temp

View File

@ -40,10 +40,10 @@ jobs:
run: | run: |
go tool cover -func=coverage.out | tail -1 go tool cover -func=coverage.out | tail -1
# - name: Enforce coverage threshold - name: Enforce coverage threshold
# run: | run: |
# PCT=$(go tool cover -func=coverage.out | tail -1 | awk '{print $3}' | tr -d '%') PCT=$(go tool cover -func=coverage.out | tail -1 | awk '{print $3}' | tr -d '%')
# awk "BEGIN { exit ($PCT < 65) }" || (echo "Coverage ${PCT}% is below 65%" && exit 1) awk "BEGIN { exit ($PCT < 70) }" || (echo "Coverage ${PCT}% is below 70%" && exit 1)
lint: lint:
name: Lint name: Lint
@ -59,9 +59,9 @@ jobs:
go-version: '1.24' go-version: '1.24'
- name: golangci-lint - name: golangci-lint
uses: golangci/golangci-lint-action@v7 uses: golangci/golangci-lint-action@v4
with: with:
version: v2.1.6 version: latest
build: build:
name: Build name: Build

View File

@ -1,14 +1,22 @@
version: "2" version: "2"
linters: linters:
default: standard default: none
enable: enable:
- govet
- errcheck
- staticcheck
- ineffassign
- unused
- gosimple
- typecheck
- misspell - misspell
- gofmt
settings: settings:
errcheck: errcheck:
check-type-assertions: true check-type-assertions: true
check-blank: false check-blank: false
formatters:
enable:
- gofmt
run: run:
timeout: 5m timeout: 5m

27
.goreleaser.yaml Normal file
View File

@ -0,0 +1,27 @@
builds:
- env:
- CGO_ENABLED=0
goos:
- linux
- darwin
- windows
goarch:
- amd64
- arm64
binary: grokkit
ldflags:
- -s -w
- -X gmgauthier.com/grokkit/internal/version.Version={{.Version}}
- -X gmgauthier.com/grokkit/internal/version.Commit={{.Commit}}
- -X gmgauthier.com/grokkit/internal/version.BuildDate={{.Date}}
- -trimpath
archives:
- format: tar.gz
name_template: 'grokkit_{{.Os}}_{{.Arch}}_{{.Version}}.tar.gz'
- format: zip
name_template: 'grokkit_{{.Os}}_{{.Arch}}_{{.Version}}.zip'
os: windows
checksum:
name_template: 'checksums.txt'

View File

@ -1,3 +0,0 @@
{
"model": "grok-code-fast-1"
}

View File

@ -1,53 +0,0 @@
## [v0.1.9] - 2026-03-04
Grokkit gets a quick-query upgrade—because who has time for chit-chat?
### Added
- Implement `query` command in cmd/query.go for non-interactive Grok queries focused on programming.
- Add wordy flag for detailed responses in query command.
- Add .grok/settings.json with fast model configuration.
- Set default model for query in config.go.
- Add entry for query command to commands list in README.
- Create new section in README with query command usage examples and features.
- Add spec for grokkit non-interactive query tool in TODO.
- Add detailed description, examples, and ROI for `query` feature in TODO.
- Introduce initial CHANGELOG.md with v0.1.8 entries.
### Changed
- Update root.go to include queryCmd.
- Reorder and update queued task list in TODO with new entries like non-interactive-query.md.
- Move changelog.md to completed tasks in TODO with version note.
- Standardize link formats and list markers in README.md.
- Rename TODO entry from "grokkit query Go tools integration" to "grokkit query Simple Query Tool".
- Revise TODO description to focus on one-shot prompt/answer tool for concise queries.
# Changelog
All notable changes to this project will be documented in this file.
## [v0.1.8] - 2026-03-03
Another step towards automated bliss, with AI changelogs and Git taking the safety wheel.
### Added
- Add release.sh script for automating releases with version validation, tagging, changelog generation, committing, and pushing.
- Add grokkit changelog command with flags for version, since, stdout, and commit reminder.
- Add git helpers for retrieving latest/previous tags and formatted logs since a reference.
- Add tests for changelog message building and full construction.
- Add todo/README.md as a table of contents for queued and completed tasks.
- Add section in README.md documenting grokkit changelog command with usage examples.
- Add detailed Git workflow for managing changes in README.md.
- Add detail in changelog.md about using generated CHANGELOG.md for Gitea release notes.
### Changed
- Change version flag shorthand from "v" to "V" in changelog command and update test assertion.
- Disable coverage threshold enforcement in CI workflow by commenting out the 65% minimum step.
- Update README.md to include grokkit changelog in the command features list.
- Reprioritize queued tasks in README.md to place changelog first, followed by interactive-agent, make, tea, and gotools.
- Update priorities in individual queued/*.md files to match new order.
- Adjust priorities for features including audit, changelog, gotools, interactive-agent, and rg in todo docs.
- Replace file backups with Git-based change management in safety features.
- Remove mentions of backups from README.md, cmd/lint.go, ARCHITECTURE.md, and TROUBLESHOOTING.md.
- Update troubleshooting guide with Git rollback instructions.
- Modify feature lists and safety descriptions to emphasize Git integration.
- Remove automatic .bak backup creation from edit, docs, lint, testgen, and agent commands.
- Update README.md, architecture docs, troubleshooting, TODOs, and tests to reflect removal of backups.

108
README.md
View File

@ -3,7 +3,7 @@
Grokkit is a fast Go CLI integrating Grok AI with git workflows and general chat/edit functionality. Grokkit is a fast Go CLI integrating Grok AI with git workflows and general chat/edit functionality.
[![Test Coverage](https://img.shields.io/badge/coverage-68%25-brightgreen)]() [![Test Coverage](https://img.shields.io/badge/coverage-72%25-brightgreen)]()
[![Go Version](https://img.shields.io/badge/go-1.24.2-blue)]() [![Go Version](https://img.shields.io/badge/go-1.24.2-blue)]()
[![License](https://img.shields.io/badge/license-Unlicense-lightgrey)]() [![License](https://img.shields.io/badge/license-Unlicense-lightgrey)]()
@ -38,16 +38,13 @@ grokkit version
- [Commands](#commands) - [Commands](#commands)
- [chat](#-grokkit-chat) - [chat](#-grokkit-chat)
- [query](#-grokkit-query)
- [edit](#-grokkit-edit-file-instruction) - [edit](#-grokkit-edit-file-instruction)
- [commit / commitmsg](#-grokkit-commitmsg) - [commit / commitmsg](#-grokkit-commitmsg)
- [review](#-grokkit-review) - [review](#-grokkit-review)
- [pr-describe](#-grokkit-pr-describe) - [pr-describe](#-grokkit-pr-describe)
- [history](#-grokkit-history) - [history](#-grokkit-history)
- [changelog](#-grokkit-changelog)
- [lint](#-grokkit-lint-file) - [lint](#-grokkit-lint-file)
- [docs](#-grokkit-docs-file) - [docs](#-grokkit-docs-file)
- [testgen](#-grokkit-testgen)
- [agent](#-grokkit-agent) - [agent](#-grokkit-agent)
- [Configuration](#configuration) - [Configuration](#configuration)
- [Workflows](#workflows) - [Workflows](#workflows)
@ -73,27 +70,8 @@ grokkit chat --debug # Enable debug logging
- History is saved automatically between sessions - History is saved automatically between sessions
- Use `--debug` to see API request timing - Use `--debug` to see API request timing
### 🤖 `grokkit query`
One-shot technical question answering. Perfect for quick programming or engineering questions.
```bash
# Basic usage
grokkit query "How do I sort a slice of structs by a field in Go?"
# Longer, more detailed answer
grokkit query --wordy "Explain how Go's context package works with cancellation"
```
Features:
Default mode is concise, factual, and actionable
--wordy flag gives longer, more explanatory answers
Uses the fast non-reasoning model by default for speed
No persistent history or interactive chat UI
### ✏️ `grokkit edit FILE "instruction"` ### ✏️ `grokkit edit FILE "instruction"`
AI-powered file editing with preview. AI-powered file editing with preview and automatic backups.
```bash ```bash
# Basic usage # Basic usage
@ -110,6 +88,7 @@ grokkit edit utils.go "add detailed docstrings to all exported functions"
``` ```
**Safety features:** **Safety features:**
- Creates `.bak` backup before any changes
- Shows preview with diff-style output - Shows preview with diff-style output
- Requires confirmation before applying - Requires confirmation before applying
- Uses silent streaming (no console spam) - Uses silent streaming (no console spam)
@ -178,24 +157,6 @@ Summarize recent git commits.
grokkit history # Last 10 commits grokkit history # Last 10 commits
``` ```
### 🗒️ `grokkit changelog`
Generate a clean `CHANGELOG.md` section from git history, designed specifically so the output can be pasted directly into Gitea release notes.
```bash
# 1. Create your version tag first
git tag v0.2.0
# 2. Generate preview + write (with safety confirmation)
grokkit changelog
# 3. Output ONLY the new section (perfect for Gitea "Release notes")
grokkit changelog --stdout
# 4. Write file + get commit reminder
grokkit changelog --commit
```
### 📖 `grokkit docs <file> [file...]` ### 📖 `grokkit docs <file> [file...]`
Generate language-appropriate documentation comments using Grok AI. Generate language-appropriate documentation comments using Grok AI.
@ -227,32 +188,10 @@ grokkit docs app.py -m grok-4
| Shell | Shell comments (`# function: desc, # Args: ...`) | | Shell | Shell comments (`# function: desc, # Args: ...`) |
**Safety features:** **Safety features:**
- Creates `.bak` backup before any changes
- Shows first 50 lines of documented code as preview - Shows first 50 lines of documented code as preview
- Requires confirmation (unless `--auto-apply`) - Requires confirmation (unless `--auto-apply`)
### 🧪 `grokkit testgen PATHS...`
**Description**: Generate comprehensive unit tests for Go/Python/C/C++ files using AI.
**Benefits**:
- Go: Table-driven `t.Parallel()` matching codebase.
- Python: Pytest with `@parametrize`.
- C: Check framework suites.
- C++: Google Test `EXPECT_*`.
- Boosts coverage; safe preview.
**CLI examples**:
```bash
grokkit testgen internal/grok/client.go
grokkit testgen app.py --yes
grokkit testgen foo.c bar.cpp
```
**Safety features**:
- Lang detection via `internal/linter`.
- Unified diff preview.
- Y/N (--yes auto).
### 🤖 `grokkit agent` ### 🤖 `grokkit agent`
Multi-file agent for complex refactoring (experimental). Multi-file agent for complex refactoring (experimental).
@ -289,42 +228,11 @@ grokkit lint script.rb -m grok-4
- **Shell** (shellcheck) - **Shell** (shellcheck)
**Safety features:** **Safety features:**
- Creates `.bak` backup before changes
- Shows preview of fixes - Shows preview of fixes
- Verifies fixes by re-running linter - Verifies fixes by re-running linter
- Requires confirmation (unless `--auto-fix`) - Requires confirmation (unless `--auto-fix`)
## Safety & Change Management
Grokkit is designed to work seamlessly with Git. Rather than creating redundant `.bak` files, we lean on Git's powerful version control to manage changes and rollbacks.
### The Safety Workflow
1. **Preview**: Every command that modifies files (like `edit`, `lint`, `docs`) shows a diff-style preview first.
2. **Confirm**: You must explicitly confirm (`y/N`) before any changes are written to disk.
3. **Git Integration**: Use Git to manage the "pre-staged," "staged," and "committed" degrees of change.
### Managing Undesired Changes
If you've applied a change that you don't like, Git makes it easy to roll back:
- **Unstaged changes**: If you haven't `git add`-ed the changes yet:
```bash
git restore <file>
```
- **Staged changes**: If you've already staged the changes:
```bash
git restore --staged <file>
git restore <file>
```
- **Committed changes**: If you've already committed the changes:
```bash
git revert HEAD
# or to reset to a previous state:
git reset --hard HEAD~1
```
By using Git, you have a complete audit trail and multiple levels of undo, ensuring your codebase remains stable even when experimenting with AI-driven refactors.
## Configuration ## Configuration
### Environment Variables ### Environment Variables
@ -519,14 +427,14 @@ grokkit review -v
- ✅ **Persistent chat history** - Never lose your conversations - ✅ **Persistent chat history** - Never lose your conversations
- ✅ **Configurable parameters** - Temperature, timeout, model selection - ✅ **Configurable parameters** - Temperature, timeout, model selection
- ✅ **Shell completions** - Bash, Zsh, Fish, PowerShell - ✅ **Shell completions** - Bash, Zsh, Fish, PowerShell
- ✅ **Safe file editing** - Preview and confirmation, leverage git for rollbacks - ✅ **Safe file editing** - Automatic backups, preview, confirmation
- ✅ **Git workflow integration** - Commit messages, reviews, PR descriptions - ✅ **Git workflow integration** - Commit messages, reviews, PR descriptions
- ✅ **Multi-language linting** - 9 languages supported with AI-powered fixes - ✅ **Multi-language linting** - 9 languages supported with AI-powered fixes
- ✅ **AI documentation generation** - 8 doc styles (godoc, PEP 257, Doxygen, JSDoc, rustdoc, YARD, Javadoc, shell) - ✅ **AI documentation generation** - 8 doc styles (godoc, PEP 257, Doxygen, JSDoc, rustdoc, YARD, Javadoc, shell)
### Quality & Testing ### Quality & Testing
- ✅ **Test coverage 68%+** - Comprehensive unit tests including all command message builders - ✅ **Test coverage 72%+** - Comprehensive unit tests including all command message builders
- ✅ **Coverage gate in CI** - Builds fail if coverage drops below 65% - ✅ **Coverage gate in CI** - Builds fail if coverage drops below 70%
- ✅ **CI/CD with Gitea Actions** - Tests must pass before lint and build jobs run - ✅ **CI/CD with Gitea Actions** - Tests must pass before lint and build jobs run
- ✅ **Golangci-lint configured** - `.golangci.yml` with govet, errcheck, staticcheck, and more - ✅ **Golangci-lint configured** - `.golangci.yml` with govet, errcheck, staticcheck, and more
- ✅ **Interface-based design** - Testable and maintainable - ✅ **Interface-based design** - Testable and maintainable

View File

@ -73,13 +73,15 @@ var agentCmd = &cobra.Command{
for i, file := range files { for i, file := range files {
color.Yellow("[%d/%d] → %s", i+1, len(files), file) color.Yellow("[%d/%d] → %s", i+1, len(files), file)
// nolint:gosec // intentional file read from user input
original, err := os.ReadFile(file) original, err := os.ReadFile(file)
if err != nil { if err != nil {
color.Red("Could not read %s", file) color.Red("Could not read %s", file)
continue continue
} }
backupPath := file + ".bak"
_ = os.WriteFile(backupPath, original, 0644)
messages := []map[string]string{ messages := []map[string]string{
{"role": "system", "content": "You are an expert programmer. Remove all unnecessary comments including last modified timestamps and ownership comments. Return clean, complete file content with no explanations, markdown, diffs, or extra text."}, {"role": "system", "content": "You are an expert programmer. Remove all unnecessary comments including last modified timestamps and ownership comments. Return clean, complete file content with no explanations, markdown, diffs, or extra text."},
{"role": "user", "content": fmt.Sprintf("File: %s\n\nOriginal content:\n%s\n\nTask: %s", filepath.Base(file), original, instruction)}, {"role": "user", "content": fmt.Sprintf("File: %s\n\nOriginal content:\n%s\n\nTask: %s", filepath.Base(file), original, instruction)},
@ -105,12 +107,12 @@ var agentCmd = &cobra.Command{
if answer == "a" { if answer == "a" {
applyAll = true applyAll = true
} else if answer != "y" { } else if answer != "y" {
color.Yellow("Skipped %s", file) color.Yellow("Skipped %s (backup kept)", file)
continue continue
} }
} }
_ = os.WriteFile(file, []byte(newContent), 0600) _ = os.WriteFile(file, []byte(newContent), 0644)
color.Green("✅ Applied %s", file) color.Green("✅ Applied %s", file)
} }

View File

@ -1,143 +0,0 @@
package cmd
import (
"fmt"
"os"
"strings"
"time"
"github.com/fatih/color"
"github.com/spf13/cobra"
"gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/git"
)
var changelogCmd = &cobra.Command{
Use: "changelog",
Short: "Generate CHANGELOG.md section from git history for Gitea releases",
Long: `AI-generated changelog using only Added/Changed/Fixed. Designed so the output can be pasted directly into Gitea release notes.`,
Run: runChangelog,
}
func init() {
changelogCmd.Flags().String("since", "", "Start from this tag/ref (default: previous tag)")
changelogCmd.Flags().StringP("version", "V", "", "Override version for header (default: latest git tag)")
changelogCmd.Flags().Bool("stdout", false, "Print ONLY the new section (ideal for Gitea release notes)")
changelogCmd.Flags().Bool("commit", false, "After writing, remind to run grokkit commit")
rootCmd.AddCommand(changelogCmd)
}
func runChangelog(cmd *cobra.Command, _ []string) {
stdout, _ := cmd.Flags().GetBool("stdout")
doCommit, _ := cmd.Flags().GetBool("commit")
version, _ := cmd.Flags().GetString("version")
since, _ := cmd.Flags().GetString("since")
if !git.IsRepo() {
color.Red("Not inside a git repository")
return
}
// Version from tag you set (or override)
if version == "" {
var err error
version, err = git.LatestTag()
if err != nil {
color.Red("No git tags found. Create one first: git tag vX.Y.Z")
return
}
}
// Since ref (defaults to previous tag)
if since == "" {
if prev, err := git.PreviousTag(version); err == nil {
since = prev
} else {
since = "HEAD~100" // safe first-release fallback
}
}
logOutput, err := git.LogSince(since)
if err != nil || strings.TrimSpace(logOutput) == "" {
color.Yellow("No new commits since last tag.")
return
}
// Grok generation (strong prompt for your exact format)
client := newGrokClient()
messages := buildChangelogMessages(logOutput, version)
model := config.GetModel("changelog", "")
color.Yellow("Asking Grok to categorize changes...")
section := client.Stream(messages, model)
date := time.Now().Format("2006-01-02")
newSection := fmt.Sprintf("## [%s] - %s\n\n%s\n", version, date, section)
if stdout {
fmt.Print(newSection)
return
}
// Build full file (prepend or create)
content := buildFullChangelog(newSection)
// Preview + safety confirm (exactly like commit/review)
color.Cyan("\n--- Proposed CHANGELOG.md update ---\n%s\n--------------------------------", content)
var confirm string
color.Yellow("Write this to CHANGELOG.md? (y/n): ")
_, _ = fmt.Scanln(&confirm) // we don't care about scan errors here
if confirm != "y" && confirm != "Y" {
color.Yellow("Aborted.")
return
}
if err := os.WriteFile("CHANGELOG.md", []byte(content), 0600); err != nil {
color.Red("Failed to write CHANGELOG.md")
return
}
color.Green("✅ CHANGELOG.md updated with version %s", version)
if doCommit {
color.Yellow("Run `grokkit commit` (or `git add CHANGELOG.md && git commit`) to stage it.")
}
}
func buildChangelogMessages(log, version string) []map[string]string {
return []map[string]string{
{
"role": "system",
"content": `You are an expert technical writer.
Generate a changelog section using **only** these headings (include only if content exists):
### Added
### Changed
### Fixed
Rules:
- Start directly with the section headings (no extra header we add it).
- One optional short summary sentence at the very top (light humour OK here only).
- Every bullet must be factual, imperative, one clear action per line.
- Be concise. No marketing language or explanations.
Output ONLY clean markdown.`,
},
{
"role": "user",
"content": fmt.Sprintf("Version: %s\n\nCommit history:\n%s", version, log),
},
}
}
func buildFullChangelog(newSection string) string {
existing, err := os.ReadFile("CHANGELOG.md")
if err != nil {
// File doesn't exist (or unreadable) → create a new changelog with a header
return `# Changelog
All notable changes to this project will be documented in this file.
` + newSection
}
return newSection + string(existing)
}

View File

@ -1,110 +0,0 @@
package cmd
import (
"os"
"strings"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestBuildChangelogMessages(t *testing.T) {
t.Parallel()
log := `feat: add changelog command
Implements #1
---
fix: typo in docs
---`
version := "v0.2.0"
messages := buildChangelogMessages(log, version)
require.Len(t, messages, 2)
assert.Equal(t, "system", messages[0]["role"])
assert.Contains(t, messages[0]["content"], "Generate a changelog section")
assert.Contains(t, messages[0]["content"], "### Added")
assert.Contains(t, messages[0]["content"], "### Changed")
assert.Contains(t, messages[0]["content"], "### Fixed")
assert.Equal(t, "user", messages[1]["role"])
assert.Contains(t, messages[1]["content"], "Version: v0.2.0")
assert.Contains(t, messages[1]["content"], log)
}
func TestBuildFullChangelog(t *testing.T) {
t.Parallel()
newSection := "## [v0.2.0] - 2026-03-03\n\n### Added\n- changelog command\n"
t.Run("creates new file with header", func(t *testing.T) {
t.Parallel()
// nolint:tparallel // os.Chdir affects the entire process
tmpDir := t.TempDir()
originalWd, err := os.Getwd()
require.NoError(t, err)
t.Cleanup(func() {
require.NoError(t, os.Chdir(originalWd))
})
require.NoError(t, os.Chdir(tmpDir))
result := buildFullChangelog(newSection)
assert.Contains(t, result, "# Changelog")
assert.Contains(t, result, "All notable changes to this project will be documented in this file.")
assert.Contains(t, result, newSection)
})
t.Run("prepends to existing file", func(t *testing.T) {
t.Parallel()
// nolint:tparallel // os.Chdir affects the entire process
tmpDir := t.TempDir()
originalWd, err := os.Getwd()
require.NoError(t, err)
t.Cleanup(func() {
require.NoError(t, os.Chdir(originalWd))
})
require.NoError(t, os.Chdir(tmpDir))
// Simulate existing changelog
existing := `# Changelog
All notable changes to this project will be documented in this file.
## [v0.1.0] - 2025-01-01
### Fixed
- old bug
`
require.NoError(t, os.WriteFile("CHANGELOG.md", []byte(existing), 0600))
result := buildFullChangelog(newSection)
assert.True(t, strings.HasPrefix(result, newSection), "new version section must be prepended at the top")
assert.Contains(t, result, "old bug")
})
}
func TestChangelogCmd_Flags(t *testing.T) {
t.Parallel()
require.NotNil(t, changelogCmd)
stdoutFlag := changelogCmd.Flags().Lookup("stdout")
require.NotNil(t, stdoutFlag)
assert.Equal(t, "Print ONLY the new section (ideal for Gitea release notes)", stdoutFlag.Usage)
versionFlag := changelogCmd.Flags().Lookup("version")
require.NotNil(t, versionFlag)
assert.Equal(t, "V", versionFlag.Shorthand)
commitFlag := changelogCmd.Flags().Lookup("commit")
require.NotNil(t, commitFlag)
assert.Equal(t, "After writing, remind to run grokkit commit", commitFlag.Usage)
sinceFlag := changelogCmd.Flags().Lookup("since")
require.NotNil(t, sinceFlag)
}

View File

@ -21,7 +21,6 @@ type ChatHistory struct {
func loadChatHistory() []map[string]string { func loadChatHistory() []map[string]string {
histFile := getChatHistoryFile() histFile := getChatHistoryFile()
// nolint:gosec // intentional file read from config/home
data, err := os.ReadFile(histFile) data, err := os.ReadFile(histFile)
if err != nil { if err != nil {
return nil return nil
@ -41,7 +40,7 @@ func saveChatHistory(messages []map[string]string) error {
if err != nil { if err != nil {
return err return err
} }
return os.WriteFile(histFile, data, 0600) return os.WriteFile(histFile, data, 0644)
} }
func getChatHistoryFile() string { func getChatHistoryFile() string {
@ -55,7 +54,7 @@ func getChatHistoryFile() string {
home = "." home = "."
} }
histDir := filepath.Join(home, ".config", "grokkit") histDir := filepath.Join(home, ".config", "grokkit")
_ = os.MkdirAll(histDir, 0750) // Ignore error, WriteFile will catch it _ = os.MkdirAll(histDir, 0755) // Ignore error, WriteFile will catch it
return filepath.Join(histDir, "chat_history.json") return filepath.Join(histDir, "chat_history.json")
} }

View File

@ -9,12 +9,10 @@ import (
func TestGetChatHistoryFile(t *testing.T) { func TestGetChatHistoryFile(t *testing.T) {
// Save original HOME // Save original HOME
oldHome := os.Getenv("HOME") oldHome := os.Getenv("HOME")
defer func() { _ = os.Setenv("HOME", oldHome) }() defer os.Setenv("HOME", oldHome)
tmpDir := t.TempDir() tmpDir := t.TempDir()
if err := os.Setenv("HOME", tmpDir); err != nil { os.Setenv("HOME", tmpDir)
t.Fatal(err)
}
histFile := getChatHistoryFile() histFile := getChatHistoryFile()
expected := filepath.Join(tmpDir, ".config", "grokkit", "chat_history.json") expected := filepath.Join(tmpDir, ".config", "grokkit", "chat_history.json")
@ -24,18 +22,11 @@ func TestGetChatHistoryFile(t *testing.T) {
} }
} }
func setHomeForTest(t *testing.T, dir string) {
t.Helper()
if err := os.Setenv("HOME", dir); err != nil {
t.Fatal(err)
}
}
func TestLoadChatHistory_NoFile(t *testing.T) { func TestLoadChatHistory_NoFile(t *testing.T) {
tmpDir := t.TempDir() tmpDir := t.TempDir()
oldHome := os.Getenv("HOME") oldHome := os.Getenv("HOME")
setHomeForTest(t, tmpDir) os.Setenv("HOME", tmpDir)
defer func() { _ = os.Setenv("HOME", oldHome) }() defer os.Setenv("HOME", oldHome)
history := loadChatHistory() history := loadChatHistory()
if history != nil { if history != nil {
@ -46,8 +37,8 @@ func TestLoadChatHistory_NoFile(t *testing.T) {
func TestSaveAndLoadChatHistory(t *testing.T) { func TestSaveAndLoadChatHistory(t *testing.T) {
tmpDir := t.TempDir() tmpDir := t.TempDir()
oldHome := os.Getenv("HOME") oldHome := os.Getenv("HOME")
setHomeForTest(t, tmpDir) os.Setenv("HOME", tmpDir)
defer func() { _ = os.Setenv("HOME", oldHome) }() defer os.Setenv("HOME", oldHome)
// Create test messages // Create test messages
messages := []map[string]string{ messages := []map[string]string{
@ -86,16 +77,16 @@ func TestSaveAndLoadChatHistory(t *testing.T) {
func TestLoadChatHistory_InvalidJSON(t *testing.T) { func TestLoadChatHistory_InvalidJSON(t *testing.T) {
tmpDir := t.TempDir() tmpDir := t.TempDir()
oldHome := os.Getenv("HOME") oldHome := os.Getenv("HOME")
setHomeForTest(t, tmpDir) os.Setenv("HOME", tmpDir)
defer func() { _ = os.Setenv("HOME", oldHome) }() defer os.Setenv("HOME", oldHome)
// Create invalid JSON file // Create invalid JSON file
histDir := filepath.Join(tmpDir, ".config", "grokkit") histDir := filepath.Join(tmpDir, ".config", "grokkit")
if err := os.MkdirAll(histDir, 0750); err != nil { if err := os.MkdirAll(histDir, 0755); err != nil {
t.Fatalf("MkdirAll() error: %v", err) t.Fatalf("MkdirAll() error: %v", err)
} }
histFile := filepath.Join(histDir, "chat_history.json") histFile := filepath.Join(histDir, "chat_history.json")
if err := os.WriteFile(histFile, []byte("invalid json{{{"), 0600); err != nil { if err := os.WriteFile(histFile, []byte("invalid json{{{"), 0644); err != nil {
t.Fatalf("WriteFile() error: %v", err) t.Fatalf("WriteFile() error: %v", err)
} }

View File

@ -1,16 +0,0 @@
package cmd
import (
"gmgauthier.com/grokkit/internal/git"
"gmgauthier.com/grokkit/internal/grok"
)
// newGrokClient is the factory for the AI client.
// Tests replace this to inject a mock without making real API calls.
var newGrokClient = func() grok.AIClient {
return grok.NewClient()
}
// gitRun is the git command runner.
// Tests replace this to inject controlled git output.
var gitRun = git.Run

View File

@ -7,16 +7,15 @@ import (
"github.com/fatih/color" "github.com/fatih/color"
"github.com/spf13/cobra" "github.com/spf13/cobra"
"gmgauthier.com/grokkit/config" "gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/git"
"gmgauthier.com/grokkit/internal/grok"
) )
var commitCmd = &cobra.Command{ var commitCmd = &cobra.Command{
Use: "commit", Use: "commit",
Short: "Generate message and commit staged changes", Short: "Generate message and commit staged changes",
Run: runCommit, Run: func(cmd *cobra.Command, args []string) {
} diff, err := git.Run([]string{"diff", "--cached", "--no-color"})
func runCommit(cmd *cobra.Command, _ []string) {
diff, err := gitRun([]string{"diff", "--cached", "--no-color"})
if err != nil { if err != nil {
color.Red("Failed to get staged changes: %v", err) color.Red("Failed to get staged changes: %v", err)
return return
@ -28,7 +27,7 @@ func runCommit(cmd *cobra.Command, _ []string) {
modelFlag, _ := cmd.Flags().GetString("model") modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("commit", modelFlag) model := config.GetModel("commit", modelFlag)
client := newGrokClient() client := grok.NewClient()
messages := buildCommitMessages(diff) messages := buildCommitMessages(diff)
color.Yellow("Generating commit message...") color.Yellow("Generating commit message...")
msg := client.Stream(messages, model) msg := client.Stream(messages, model)
@ -45,12 +44,12 @@ func runCommit(cmd *cobra.Command, _ []string) {
return return
} }
// nolint:gosec // intentional subprocess for git operation
if err := exec.Command("git", "commit", "-m", msg).Run(); err != nil { if err := exec.Command("git", "commit", "-m", msg).Run(); err != nil {
color.Red("Git commit failed") color.Red("Git commit failed")
} else { } else {
color.Green("✅ Committed successfully!") color.Green("✅ Committed successfully!")
} }
},
} }
func buildCommitMessages(diff string) []map[string]string { func buildCommitMessages(diff string) []map[string]string {

View File

@ -1,19 +1,20 @@
package cmd package cmd
import ( import (
"fmt"
"github.com/fatih/color" "github.com/fatih/color"
"github.com/spf13/cobra" "github.com/spf13/cobra"
"gmgauthier.com/grokkit/config" "gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/git"
"gmgauthier.com/grokkit/internal/grok"
) )
var commitMsgCmd = &cobra.Command{ var commitMsgCmd = &cobra.Command{
Use: "commit-msg", Use: "commit-msg",
Short: "Generate conventional commit message from staged changes", Short: "Generate conventional commit message from staged changes",
Run: runCommitMsg, Run: func(cmd *cobra.Command, args []string) {
} diff, err := git.Run([]string{"diff", "--cached", "--no-color"})
func runCommitMsg(cmd *cobra.Command, args []string) {
diff, err := gitRun([]string{"diff", "--cached", "--no-color"})
if err != nil { if err != nil {
color.Red("Failed to get staged changes: %v", err) color.Red("Failed to get staged changes: %v", err)
return return
@ -25,8 +26,12 @@ func runCommitMsg(cmd *cobra.Command, args []string) {
modelFlag, _ := cmd.Flags().GetString("model") modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("commitmsg", modelFlag) model := config.GetModel("commitmsg", modelFlag)
client := newGrokClient() client := grok.NewClient()
messages := buildCommitMessages(diff) messages := []map[string]string{
{"role": "system", "content": "Return ONLY a conventional commit message (type(scope): subject\n\nbody)."},
{"role": "user", "content": fmt.Sprintf("Staged changes:\n%s", diff)},
}
color.Yellow("Generating commit message...") color.Yellow("Generating commit message...")
client.Stream(messages, model) client.Stream(messages, model)
},
} }

View File

@ -49,7 +49,6 @@ PowerShell:
ValidArgs: []string{"bash", "zsh", "fish", "powershell"}, ValidArgs: []string{"bash", "zsh", "fish", "powershell"},
Args: cobra.MatchAll(cobra.ExactArgs(1), cobra.OnlyValidArgs), Args: cobra.MatchAll(cobra.ExactArgs(1), cobra.OnlyValidArgs),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
// nolint:gosec // intentional subprocess for shell completion
var err error var err error
switch args[0] { switch args[0] {
case "bash": case "bash":

View File

@ -1,44 +0,0 @@
package cmd
import (
"io"
"os"
"testing"
"github.com/spf13/cobra"
)
func TestCompletionCmd(t *testing.T) {
tests := []struct {
shell string
}{
{"bash"},
{"zsh"},
{"fish"},
{"powershell"},
}
for _, tt := range tests {
t.Run(tt.shell, func(t *testing.T) {
oldStdout := os.Stdout
defer func() { os.Stdout = oldStdout }()
r, w, err := os.Pipe()
if err != nil {
t.Fatal(err)
}
os.Stdout = w
defer func() { _ = w.Close() }()
cmd := &cobra.Command{}
completionCmd.Run(cmd, []string{tt.shell})
if err := w.Close(); err != nil {
t.Errorf("close: %v", err)
}
b, err := io.ReadAll(r)
if err != nil {
t.Fatal(err)
}
if len(b) == 0 {
t.Error("expected non-empty completion script")
}
})
}
}

View File

@ -32,6 +32,7 @@ Supported doc styles:
Shell Shell comments (# function: ...) Shell Shell comments (# function: ...)
Safety features: Safety features:
- Creates .bak backup before modifying files
- Shows preview of changes before applying - Shows preview of changes before applying
- Requires confirmation unless --auto-apply is used`, - Requires confirmation unless --auto-apply is used`,
Args: cobra.MinimumNArgs(1), Args: cobra.MinimumNArgs(1),
@ -46,13 +47,13 @@ func runDocs(cmd *cobra.Command, args []string) {
modelFlag, _ := cmd.Flags().GetString("model") modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("docs", modelFlag) model := config.GetModel("docs", modelFlag)
client := newGrokClient() client := grok.NewClient()
for _, filePath := range args { for _, filePath := range args {
processDocsFile(client, model, filePath) processDocsFile(client, model, filePath)
} }
} }
func processDocsFile(client grok.AIClient, model, filePath string) { func processDocsFile(client *grok.Client, model, filePath string) {
logger.Info("starting docs operation", "file", filePath) logger.Info("starting docs operation", "file", filePath)
if _, err := os.Stat(filePath); os.IsNotExist(err) { if _, err := os.Stat(filePath); os.IsNotExist(err) {
@ -68,7 +69,6 @@ func processDocsFile(client grok.AIClient, model, filePath string) {
return return
} }
// nolint:gosec // intentional file read from user input
originalContent, err := os.ReadFile(filePath) originalContent, err := os.ReadFile(filePath)
if err != nil { if err != nil {
logger.Error("failed to read file", "file", filePath, "error", err) logger.Error("failed to read file", "file", filePath, "error", err)
@ -100,7 +100,7 @@ func processDocsFile(client grok.AIClient, model, filePath string) {
if len(lines) < previewLines { if len(lines) < previewLines {
previewLines = len(lines) previewLines = len(lines)
} }
for i := range previewLines { for i := 0; i < previewLines; i++ {
fmt.Println(lines[i]) fmt.Println(lines[i])
} }
if len(lines) > previewLines { if len(lines) > previewLines {
@ -124,7 +124,16 @@ func processDocsFile(client grok.AIClient, model, filePath string) {
} }
} }
if err := os.WriteFile(filePath, []byte(documented), 0600); err != nil { // Create backup
backupPath := filePath + ".bak"
if err := os.WriteFile(backupPath, originalContent, 0644); err != nil {
logger.Error("failed to create backup", "file", filePath, "error", err)
color.Red("❌ Failed to create backup: %v", err)
return
}
logger.Info("backup created", "backup", backupPath)
if err := os.WriteFile(filePath, []byte(documented), 0644); err != nil {
logger.Error("failed to write documented file", "file", filePath, "error", err) logger.Error("failed to write documented file", "file", filePath, "error", err)
color.Red("❌ Failed to write file: %v", err) color.Red("❌ Failed to write file: %v", err)
return return
@ -132,6 +141,7 @@ func processDocsFile(client grok.AIClient, model, filePath string) {
logger.Info("documentation applied successfully", "file", filePath) logger.Info("documentation applied successfully", "file", filePath)
color.Green("✅ Documentation applied: %s", filePath) color.Green("✅ Documentation applied: %s", filePath)
color.Cyan("💾 Original saved to: %s", backupPath)
} }
func buildDocsMessages(language, code string) []map[string]string { func buildDocsMessages(language, code string) []map[string]string {

View File

@ -15,7 +15,7 @@ import (
var editCmd = &cobra.Command{ var editCmd = &cobra.Command{
Use: "edit FILE INSTRUCTION", Use: "edit FILE INSTRUCTION",
Short: "Edit a file in-place with Grok (safe preview)", Short: "Edit a file in-place with Grok (safe preview + backup)",
Args: cobra.ExactArgs(2), Args: cobra.ExactArgs(2),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
filePath := args[0] filePath := args[0]
@ -35,7 +35,6 @@ var editCmd = &cobra.Command{
os.Exit(1) os.Exit(1)
} }
// nolint:gosec // intentional file read from user input
original, err := os.ReadFile(filePath) original, err := os.ReadFile(filePath)
if err != nil { if err != nil {
logger.Error("failed to read file", "file", filePath, "error", err) logger.Error("failed to read file", "file", filePath, "error", err)
@ -45,6 +44,15 @@ var editCmd = &cobra.Command{
logger.Debug("file read successfully", "file", filePath, "size_bytes", len(original)) logger.Debug("file read successfully", "file", filePath, "size_bytes", len(original))
cleanedOriginal := removeLastModifiedComments(string(original)) cleanedOriginal := removeLastModifiedComments(string(original))
backupPath := filePath + ".bak"
logger.Debug("creating backup", "backup_path", backupPath)
if err := os.WriteFile(backupPath, original, 0644); err != nil {
logger.Error("failed to create backup", "backup_path", backupPath, "error", err)
color.Red("Failed to create backup: %v", err)
os.Exit(1)
}
logger.Info("backup created", "backup_path", backupPath)
client := grok.NewClient() client := grok.NewClient()
messages := []map[string]string{ messages := []map[string]string{
{"role": "system", "content": "You are an expert programmer. Remove all unnecessary comments including last modified timestamps and ownership comments. Return only the cleaned code with no explanations, no markdown, no extra text."}, {"role": "system", "content": "You are an expert programmer. Remove all unnecessary comments including last modified timestamps and ownership comments. Return only the cleaned code with no explanations, no markdown, no extra text."},
@ -65,31 +73,32 @@ var editCmd = &cobra.Command{
var confirm string var confirm string
if _, err := fmt.Scanln(&confirm); err != nil { if _, err := fmt.Scanln(&confirm); err != nil {
color.Red("Failed to read input: %v", err) color.Red("Failed to read input: %v", err)
color.Yellow("Changes discarded.") color.Yellow("Changes discarded. Backup saved as %s", backupPath)
return return
} }
if confirm != "y" && confirm != "Y" { if confirm != "y" && confirm != "Y" {
color.Yellow("Changes discarded.") color.Yellow("Changes discarded. Backup saved as %s", backupPath)
return return
} }
logger.Debug("applying changes", "file", filePath, "new_size_bytes", len(newContent)) logger.Debug("applying changes", "file", filePath, "new_size_bytes", len(newContent))
if err := os.WriteFile(filePath, []byte(newContent), 0600); err != nil { if err := os.WriteFile(filePath, []byte(newContent), 0644); err != nil {
logger.Error("failed to write file", "file", filePath, "error", err) logger.Error("failed to write file", "file", filePath, "error", err)
color.Red("Failed to write file: %v", err) color.Red("Failed to write file: %v", err)
os.Exit(1) os.Exit(1)
} }
logger.Info("changes applied successfully", logger.Info("changes applied successfully",
"file", filePath, "file", filePath,
"backup", backupPath,
"original_size", len(original), "original_size", len(original),
"new_size", len(newContent)) "new_size", len(newContent))
color.Green("✅ Applied successfully!") color.Green("✅ Applied successfully! Backup: %s", backupPath)
}, },
} }
func removeLastModifiedComments(content string) string { func removeLastModifiedComments(content string) string {
lines := strings.Split(content, "\n") lines := strings.Split(content, "\n")
cleanedLines := make([]string, 0, len(lines)) var cleanedLines []string
for _, line := range lines { for _, line := range lines {
if strings.Contains(line, "Last modified") { if strings.Contains(line, "Last modified") {

View File

@ -14,10 +14,10 @@ func TestEditCommand(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer func() { _ = os.Remove(tmpfile.Name()) }() defer os.Remove(tmpfile.Name())
original := []byte("package main\n\nfunc hello() {}\n") original := []byte("package main\n\nfunc hello() {}\n")
if err := os.WriteFile(tmpfile.Name(), original, 0600); err != nil { if err := os.WriteFile(tmpfile.Name(), original, 0644); err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -34,7 +34,7 @@ func TestEditCommand(t *testing.T) {
newContent := grok.CleanCodeResponse(raw) newContent := grok.CleanCodeResponse(raw)
// Apply the result (this is what the real command does after confirmation) // Apply the result (this is what the real command does after confirmation)
if err := os.WriteFile(tmpfile.Name(), []byte(newContent), 0600); err != nil { if err := os.WriteFile(tmpfile.Name(), []byte(newContent), 0644); err != nil {
t.Fatal(err) t.Fatal(err)
} }

View File

@ -4,16 +4,15 @@ import (
"github.com/fatih/color" "github.com/fatih/color"
"github.com/spf13/cobra" "github.com/spf13/cobra"
"gmgauthier.com/grokkit/config" "gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/git"
"gmgauthier.com/grokkit/internal/grok"
) )
var historyCmd = &cobra.Command{ var historyCmd = &cobra.Command{
Use: "history", Use: "history",
Short: "Summarize recent git history", Short: "Summarize recent git history",
Run: runHistory, Run: func(cmd *cobra.Command, args []string) {
} log, err := git.Run([]string{"log", "--oneline", "-10"})
func runHistory(cmd *cobra.Command, _ []string) {
log, err := gitRun([]string{"log", "--oneline", "-10"})
if err != nil { if err != nil {
color.Red("Failed to get git log: %v", err) color.Red("Failed to get git log: %v", err)
return return
@ -26,10 +25,11 @@ func runHistory(cmd *cobra.Command, _ []string) {
modelFlag, _ := cmd.Flags().GetString("model") modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("history", modelFlag) model := config.GetModel("history", modelFlag)
client := newGrokClient() client := grok.NewClient()
messages := buildHistoryMessages(log) messages := buildHistoryMessages(log)
color.Yellow("Summarizing recent commits...") color.Yellow("Summarizing recent commits...")
client.Stream(messages, model) client.Stream(messages, model)
},
} }
func buildHistoryMessages(log string) []map[string]string { func buildHistoryMessages(log string) []map[string]string {

View File

@ -32,6 +32,7 @@ The command will:
4. (Optional) Use Grok AI to generate and apply fixes 4. (Optional) Use Grok AI to generate and apply fixes
Safety features: Safety features:
- Creates backup before modifying files
- Shows preview of changes before applying - Shows preview of changes before applying
- Requires confirmation unless --auto-fix is used`, - Requires confirmation unless --auto-fix is used`,
Args: cobra.ExactArgs(1), Args: cobra.ExactArgs(1),
@ -103,7 +104,6 @@ func runLint(cmd *cobra.Command, args []string) {
} }
// Read original file content // Read original file content
// nolint:gosec // intentional file read from user input
originalContent, err := os.ReadFile(absPath) originalContent, err := os.ReadFile(absPath)
if err != nil { if err != nil {
logger.Error("failed to read file", "file", absPath, "error", err) logger.Error("failed to read file", "file", absPath, "error", err)
@ -118,7 +118,7 @@ func runLint(cmd *cobra.Command, args []string) {
modelFlag, _ := cmd.Flags().GetString("model") modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("lint", modelFlag) model := config.GetModel("lint", modelFlag)
client := newGrokClient() client := grok.NewClient()
messages := buildLintFixMessages(result, string(originalContent)) messages := buildLintFixMessages(result, string(originalContent))
response := client.StreamSilent(messages, model) response := client.StreamSilent(messages, model)
@ -132,6 +132,16 @@ func runLint(cmd *cobra.Command, args []string) {
fixedCode := grok.CleanCodeResponse(response) fixedCode := grok.CleanCodeResponse(response)
logger.Info("received AI fixes", "file", absPath, "fixed_size", len(fixedCode)) logger.Info("received AI fixes", "file", absPath, "fixed_size", len(fixedCode))
// Create backup
backupPath := absPath + ".bak"
if err := os.WriteFile(backupPath, originalContent, 0644); err != nil {
logger.Error("failed to create backup", "file", absPath, "error", err)
color.Red("❌ Failed to create backup: %v", err)
return
}
logger.Info("backup created", "backup", backupPath)
color.Green("💾 Backup created: %s", backupPath)
// Show preview if not auto-fix // Show preview if not auto-fix
if !autoFix { if !autoFix {
fmt.Println() fmt.Println()
@ -143,7 +153,7 @@ func runLint(cmd *cobra.Command, args []string) {
if len(lines) < previewLines { if len(lines) < previewLines {
previewLines = len(lines) previewLines = len(lines)
} }
for i := range previewLines { for i := 0; i < previewLines; i++ {
fmt.Println(lines[i]) fmt.Println(lines[i])
} }
if len(lines) > previewLines { if len(lines) > previewLines {
@ -157,20 +167,20 @@ func runLint(cmd *cobra.Command, args []string) {
var response string var response string
if _, err := fmt.Scanln(&response); err != nil { if _, err := fmt.Scanln(&response); err != nil {
logger.Info("failed to read user input", "file", absPath, "error", err) logger.Info("failed to read user input", "file", absPath, "error", err)
color.Yellow("❌ Cancelled. No changes made.") color.Yellow("❌ Cancelled. Backup preserved at: %s", backupPath)
return return
} }
response = strings.ToLower(strings.TrimSpace(response)) response = strings.ToLower(strings.TrimSpace(response))
if response != "y" && response != "yes" { if response != "y" && response != "yes" {
logger.Info("user cancelled fix application", "file", absPath) logger.Info("user cancelled fix application", "file", absPath)
color.Yellow("❌ Cancelled. No changes made.") color.Yellow("❌ Cancelled. Backup preserved at: %s", backupPath)
return return
} }
} }
// Apply fixes // Apply fixes
if err := os.WriteFile(absPath, []byte(fixedCode), 0600); err != nil { if err := os.WriteFile(absPath, []byte(fixedCode), 0644); err != nil {
logger.Error("failed to write fixed file", "file", absPath, "error", err) logger.Error("failed to write fixed file", "file", absPath, "error", err)
color.Red("❌ Failed to write file: %v", err) color.Red("❌ Failed to write file: %v", err)
return return
@ -178,6 +188,7 @@ func runLint(cmd *cobra.Command, args []string) {
logger.Info("fixes applied successfully", "file", absPath) logger.Info("fixes applied successfully", "file", absPath)
color.Green("✅ Fixes applied successfully!") color.Green("✅ Fixes applied successfully!")
color.Cyan("💾 Original saved to: %s", backupPath)
// Optionally run linter again to verify // Optionally run linter again to verify
color.Cyan("\n🔍 Re-running linter to verify fixes...") color.Cyan("\n🔍 Re-running linter to verify fixes...")

View File

@ -6,40 +6,34 @@ import (
"github.com/fatih/color" "github.com/fatih/color"
"github.com/spf13/cobra" "github.com/spf13/cobra"
"gmgauthier.com/grokkit/config" "gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/git"
"gmgauthier.com/grokkit/internal/grok"
) )
var prDescribeCmd = &cobra.Command{ var prDescribeCmd = &cobra.Command{
Use: "pr-describe", Use: "pr-describe",
Short: "Generate full PR description from current branch", Short: "Generate full PR description from current branch",
Run: runPRDescribe, Run: func(cmd *cobra.Command, args []string) {
} diff, err := git.Run([]string{"diff", "main..HEAD", "--no-color"})
func init() {
prDescribeCmd.Flags().StringP("base", "b", "master", "Base branch to compare against")
}
func runPRDescribe(cmd *cobra.Command, _ []string) {
base, _ := cmd.Flags().GetString("base")
diff, err := gitRun([]string{"diff", fmt.Sprintf("%s..HEAD", base), "--no-color"})
if err != nil || diff == "" { if err != nil || diff == "" {
diff, err = gitRun([]string{"diff", fmt.Sprintf("origin/%s..HEAD", base), "--no-color"}) diff, err = git.Run([]string{"diff", "origin/main..HEAD", "--no-color"})
if err != nil { if err != nil {
color.Red("Failed to get branch diff: %v", err) color.Red("Failed to get branch diff: %v", err)
return return
} }
} }
if diff == "" { if diff == "" {
color.Yellow("No changes on this branch compared to %s/origin/%s.", base, base) color.Yellow("No changes on this branch compared to main/origin/main.")
return return
} }
modelFlag, _ := cmd.Flags().GetString("model") modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("prdescribe", modelFlag) model := config.GetModel("prdescribe", modelFlag)
client := newGrokClient() client := grok.NewClient()
messages := buildPRDescribeMessages(diff) messages := buildPRDescribeMessages(diff)
color.Yellow("Writing PR description...") color.Yellow("Writing PR description...")
client.Stream(messages, model) client.Stream(messages, model)
},
} }
func buildPRDescribeMessages(diff string) []map[string]string { func buildPRDescribeMessages(diff string) []map[string]string {

View File

@ -1,53 +0,0 @@
package cmd
import (
"github.com/fatih/color"
"github.com/spf13/cobra"
"gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/grok"
)
var queryCmd = &cobra.Command{
Use: "query [question]",
Short: "One-shot non-interactive query to Grok (programming focused)",
Long: `Ask Grok a single technical question and get a concise, actionable answer.
Default mode is factual and brief. Use --wordy for longer, more explanatory answers.`,
Args: cobra.MinimumNArgs(1),
Run: runQuery,
}
func init() {
queryCmd.Flags().Bool("wordy", false, "Give a longer, more detailed answer")
rootCmd.AddCommand(queryCmd)
}
func runQuery(cmd *cobra.Command, args []string) {
wordy, _ := cmd.Flags().GetBool("wordy")
question := args[0]
// Use fast model by default for quick queries
model := config.GetModel("query", "")
client := grok.NewClient()
systemPrompt := `You are Grok, a helpful and truthful AI built by xAI.
Focus on programming, software engineering, and technical questions.
Be concise, factual, and actionable. Include code snippets when helpful.
Do not add unnecessary fluff.`
if wordy {
systemPrompt = `You are Grok, a helpful and truthful AI built by xAI.
Give thorough, detailed, textbook-style answers to technical questions.
Explain concepts clearly, include examples, and allow light humour where appropriate.
Be comprehensive but still clear and well-structured.`
}
messages := []map[string]string{
{"role": "system", "content": systemPrompt},
{"role": "user", "content": question},
}
color.Yellow("Asking Grok...")
client.Stream(messages, model)
}

View File

@ -6,33 +6,33 @@ import (
"github.com/fatih/color" "github.com/fatih/color"
"github.com/spf13/cobra" "github.com/spf13/cobra"
"gmgauthier.com/grokkit/config" "gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/git"
"gmgauthier.com/grokkit/internal/grok"
) )
var reviewCmd = &cobra.Command{ var reviewCmd = &cobra.Command{
Use: "review [path]", Use: "review [path]",
Short: "Review the current repository or directory", Short: "Review the current repository or directory",
Run: runReview, Run: func(cmd *cobra.Command, args []string) {
}
func runReview(cmd *cobra.Command, _ []string) {
modelFlag, _ := cmd.Flags().GetString("model") modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("review", modelFlag) model := config.GetModel("review", modelFlag)
diff, err := gitRun([]string{"diff", "--no-color"}) client := grok.NewClient()
diff, err := git.Run([]string{"diff", "--no-color"})
if err != nil { if err != nil {
color.Red("Failed to get git diff: %v", err) color.Red("Failed to get git diff: %v", err)
return return
} }
status, err := gitRun([]string{"status", "--short"}) status, err := git.Run([]string{"status", "--short"})
if err != nil { if err != nil {
color.Red("Failed to get git status: %v", err) color.Red("Failed to get git status: %v", err)
return return
} }
client := newGrokClient()
messages := buildReviewMessages(status, diff) messages := buildReviewMessages(status, diff)
color.Yellow("Grok is reviewing the repo...") color.Yellow("Grok is reviewing the repo...")
client.Stream(messages, model) client.Stream(messages, model)
},
} }
func buildReviewMessages(status, diff string) []map[string]string { func buildReviewMessages(status, diff string) []map[string]string {

View File

@ -4,7 +4,6 @@ import (
"os" "os"
"fmt" "fmt"
"github.com/spf13/cobra" "github.com/spf13/cobra"
"gmgauthier.com/grokkit/config" "gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/logger" "gmgauthier.com/grokkit/internal/logger"
@ -58,9 +57,6 @@ func init() {
rootCmd.AddCommand(completionCmd) rootCmd.AddCommand(completionCmd)
rootCmd.AddCommand(versionCmd) rootCmd.AddCommand(versionCmd)
rootCmd.AddCommand(docsCmd) rootCmd.AddCommand(docsCmd)
rootCmd.AddCommand(testgenCmd)
rootCmd.AddCommand(scaffoldCmd)
rootCmd.AddCommand(queryCmd)
// Add model flag to all commands // Add model flag to all commands
rootCmd.PersistentFlags().StringP("model", "m", "", "Grok model to use (overrides config)") rootCmd.PersistentFlags().StringP("model", "m", "", "Grok model to use (overrides config)")

View File

@ -1,52 +0,0 @@
package cmd
import (
"fmt"
"os"
"path/filepath"
"strings"
"testing"
)
func rootSetHomeForTest(t *testing.T, dir string) {
t.Helper()
if err := os.Setenv("HOME", dir); err != nil {
t.Fatal(err)
}
}
func TestExecute(t *testing.T) {
tests := []struct {
name string
args []string
exitCode int
wantLevel string
}{
{"version", []string{"version"}, 0, ""},
{"help", []string{}, 0, ""},
{"debug flag", []string{"version", "--debug"}, 0, "debug"},
{"verbose flag", []string{"version", "-v"}, 0, "info"},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
tmpDir := t.TempDir()
rootSetHomeForTest(t, tmpDir)
oldArgs := os.Args
os.Args = append([]string{"grokkit"}, tt.args...)
defer func() { os.Args = oldArgs }()
Execute()
if tt.wantLevel != "" {
logFile := filepath.Join(tmpDir, ".config", "grokkit", "grokkit.log")
content, err := os.ReadFile(logFile)
if err != nil {
t.Error(err)
return
}
str := string(content)
if !strings.Contains(str, fmt.Sprintf(`"log_level":"%s"`, tt.wantLevel)) {
t.Errorf("log level not %s:\n%s", tt.wantLevel, str)
}
}
})
}
}

View File

@ -1,501 +0,0 @@
package cmd
import (
"errors"
"os"
"testing"
"github.com/spf13/cobra"
"gmgauthier.com/grokkit/internal/grok"
)
// mockStreamer records calls made to it and returns a canned response.
type mockStreamer struct {
response string
calls int
}
func (m *mockStreamer) Stream(messages []map[string]string, model string) string {
m.calls++
return m.response
}
func (m *mockStreamer) StreamWithTemp(messages []map[string]string, model string, temp float64) string {
m.calls++
return m.response
}
func (m *mockStreamer) StreamSilent(messages []map[string]string, model string) string {
m.calls++
return m.response
}
// Ensure mockStreamer satisfies the interface at compile time.
var _ grok.AIClient = (*mockStreamer)(nil)
// withMockClient injects mock and returns a restore func.
func withMockClient(mock grok.AIClient) func() {
orig := newGrokClient
newGrokClient = func() grok.AIClient { return mock }
return func() { newGrokClient = orig }
}
// withMockGit injects a fake git runner and returns a restore func.
func withMockGit(fn func([]string) (string, error)) func() {
orig := gitRun
gitRun = fn
return func() { gitRun = orig }
}
// testCmd returns a minimal cobra command with common flags registered.
func testCmd() *cobra.Command {
c := &cobra.Command{}
c.Flags().String("model", "", "")
c.Flags().String("base", "master", "")
return c
}
// --- runHistory ---
func TestRunHistory(t *testing.T) {
t.Run("calls AI with log output", func(t *testing.T) {
mock := &mockStreamer{response: "3 commits: feat, fix, chore"}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "abc1234 feat: add thing\ndef5678 fix: bug", nil
})()
runHistory(testCmd(), nil)
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
})
t.Run("no commits — skips AI", func(t *testing.T) {
mock := &mockStreamer{}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "", nil
})()
runHistory(testCmd(), nil)
if mock.calls != 0 {
t.Errorf("expected 0 AI calls, got %d", mock.calls)
}
})
t.Run("git error — skips AI", func(t *testing.T) {
mock := &mockStreamer{}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "", errors.New("not a git repo")
})()
runHistory(testCmd(), nil)
if mock.calls != 0 {
t.Errorf("expected 0 AI calls, got %d", mock.calls)
}
})
}
// --- runReview ---
func TestRunReview(t *testing.T) {
t.Run("reviews with diff and status", func(t *testing.T) {
mock := &mockStreamer{response: "looks good"}
defer withMockClient(mock)()
callCount := 0
defer withMockGit(func(args []string) (string, error) {
callCount++
switch args[0] {
case "diff":
return "diff content", nil
case "status":
return "M main.go", nil
}
return "", nil
})()
runReview(testCmd(), nil)
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
if callCount != 2 {
t.Errorf("expected 2 git calls (diff + status), got %d", callCount)
}
})
t.Run("git diff error — skips AI", func(t *testing.T) {
mock := &mockStreamer{}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
if args[0] == "diff" {
return "", errors.New("git error")
}
return "", nil
})()
runReview(testCmd(), nil)
if mock.calls != 0 {
t.Errorf("expected 0 AI calls, got %d", mock.calls)
}
})
t.Run("git status error — skips AI", func(t *testing.T) {
mock := &mockStreamer{}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
if args[0] == "status" {
return "", errors.New("git error")
}
return "diff content", nil
})()
runReview(testCmd(), nil)
if mock.calls != 0 {
t.Errorf("expected 0 AI calls, got %d", mock.calls)
}
})
}
// --- runCommit ---
func TestRunCommit(t *testing.T) {
t.Run("no staged changes — skips AI", func(t *testing.T) {
mock := &mockStreamer{}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "", nil
})()
runCommit(testCmd(), nil)
if mock.calls != 0 {
t.Errorf("expected 0 AI calls, got %d", mock.calls)
}
})
t.Run("git error — skips AI", func(t *testing.T) {
mock := &mockStreamer{}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "", errors.New("not a git repo")
})()
runCommit(testCmd(), nil)
if mock.calls != 0 {
t.Errorf("expected 0 AI calls, got %d", mock.calls)
}
})
t.Run("with staged changes — calls AI then cancels via stdin", func(t *testing.T) {
mock := &mockStreamer{response: "feat(cmd): add thing"}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "diff --git a/foo.go b/foo.go\n+func bar() {}", nil
})()
// Pipe "n\n" so the confirmation prompt returns without committing.
r, w, _ := os.Pipe()
origStdin := os.Stdin
os.Stdin = r
if _, err := w.WriteString("n\n"); err != nil {
t.Fatal(err)
}
if err := w.Close(); err != nil {
t.Fatal(err)
}
defer func() { os.Stdin = origStdin }()
runCommit(testCmd(), nil)
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
})
}
// --- runCommitMsg ---
func TestRunCommitMsg(t *testing.T) {
t.Run("no staged changes — skips AI", func(t *testing.T) {
mock := &mockStreamer{}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "", nil
})()
runCommitMsg(testCmd(), nil)
if mock.calls != 0 {
t.Errorf("expected 0 AI calls, got %d", mock.calls)
}
})
t.Run("with staged changes — calls AI and prints message", func(t *testing.T) {
mock := &mockStreamer{response: "feat(api): add endpoint"}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "diff --git a/api.go b/api.go", nil
})()
runCommitMsg(testCmd(), nil)
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
})
}
// --- runPRDescribe ---
func TestRunPRDescribe(t *testing.T) {
t.Run("no changes on branch — skips AI", func(t *testing.T) {
mock := &mockStreamer{}
defer withMockClient(mock)()
defer withMockGit(func(args []string) (string, error) {
return "", nil // both diff calls return empty
})()
runPRDescribe(testCmd(), nil)
if mock.calls != 0 {
t.Errorf("expected 0 AI calls, got %d", mock.calls)
}
})
t.Run("first diff succeeds — calls AI", func(t *testing.T) {
mock := &mockStreamer{response: "## PR Title\n\nDescription"}
defer withMockClient(mock)()
callCount := 0
defer withMockGit(func(args []string) (string, error) {
callCount++
if callCount == 1 {
return "diff --git a/foo.go b/foo.go", nil
}
return "", nil
})()
runPRDescribe(testCmd(), nil)
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
})
t.Run("first diff empty, second succeeds — calls AI", func(t *testing.T) {
mock := &mockStreamer{response: "PR description"}
defer withMockClient(mock)()
callCount := 0
defer withMockGit(func(args []string) (string, error) {
callCount++
if callCount == 2 {
return "diff --git a/bar.go b/bar.go", nil
}
return "", nil
})()
runPRDescribe(testCmd(), nil)
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
})
t.Run("uses custom base branch", func(t *testing.T) {
mock := &mockStreamer{response: "PR description"}
defer withMockClient(mock)()
var capturedArgs []string
defer withMockGit(func(args []string) (string, error) {
capturedArgs = args
return "diff content", nil
})()
cmd := testCmd()
if err := cmd.Flags().Set("base", "develop"); err != nil {
t.Fatal(err)
}
runPRDescribe(cmd, nil)
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
// Expect "diff", "develop..HEAD", "--no-color"
expectedArg := "develop..HEAD"
found := false
for _, arg := range capturedArgs {
if arg == expectedArg {
found = true
break
}
}
if !found {
t.Errorf("expected arg %q not found in %v", expectedArg, capturedArgs)
}
})
t.Run("defaults to master", func(t *testing.T) {
mock := &mockStreamer{response: "PR description"}
defer withMockClient(mock)()
var capturedArgs []string
defer withMockGit(func(args []string) (string, error) {
capturedArgs = args
return "diff content", nil
})()
runPRDescribe(testCmd(), nil)
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
expectedArg := "master..HEAD"
found := false
for _, arg := range capturedArgs {
if arg == expectedArg {
found = true
break
}
}
if !found {
t.Errorf("expected arg %q not found in %v", expectedArg, capturedArgs)
}
})
}
// --- runLint / processDocsFile (error paths only — linter/file I/O) ---
func TestRunLintFileNotFound(t *testing.T) {
// Reset flags to defaults
dryRun = false
autoFix = false
// Pass a non-existent file; should return without calling AI.
mock := &mockStreamer{}
defer withMockClient(mock)()
runLint(testCmd(), []string{"/nonexistent/path/file.go"})
if mock.calls != 0 {
t.Errorf("expected 0 AI calls for missing file, got %d", mock.calls)
}
}
func TestProcessDocsFileNotFound(t *testing.T) {
mock := &mockStreamer{}
processDocsFile(mock, "grok-4", "/nonexistent/path/file.go")
if mock.calls != 0 {
t.Errorf("expected 0 AI calls for missing file, got %d", mock.calls)
}
}
func TestProcessDocsFileUnsupportedLanguage(t *testing.T) {
// Create a temp file with an unsupported extension.
f, err := os.CreateTemp("", "test*.xyz")
if err != nil {
t.Fatal(err)
}
if err := f.Close(); err != nil {
t.Fatal(err)
}
defer func() { _ = os.Remove(f.Name()) }()
mock := &mockStreamer{}
processDocsFile(mock, "grok-4", f.Name())
if mock.calls != 0 {
t.Errorf("expected 0 AI calls for unsupported language, got %d", mock.calls)
}
}
func TestProcessDocsFilePreviewAndCancel(t *testing.T) {
f, err := os.CreateTemp("", "test*.go")
if err != nil {
t.Fatal(err)
}
if _, err := f.WriteString("package main\n\nfunc Foo() {}\n"); err != nil {
t.Fatal(err)
}
if err := f.Close(); err != nil {
t.Fatal(err)
}
defer func() { _ = os.Remove(f.Name()) }()
mock := &mockStreamer{response: "package main\n\n// Foo does nothing.\nfunc Foo() {}\n"}
origAutoApply := autoApply
autoApply = false
defer func() { autoApply = origAutoApply }()
// Pipe "n\n" so confirmation returns without writing.
r, w, _ := os.Pipe()
origStdin := os.Stdin
os.Stdin = r
if _, err := w.WriteString("n\n"); err != nil {
t.Fatal(err)
}
if err := w.Close(); err != nil {
t.Fatal(err)
}
defer func() { os.Stdin = origStdin }()
processDocsFile(mock, "grok-4", f.Name())
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
}
func TestProcessDocsFileAutoApply(t *testing.T) {
f, err := os.CreateTemp("", "test*.go")
if err != nil {
t.Fatal(err)
}
original := "package main\n\nfunc Bar() {}\n"
if _, err := f.WriteString(original); err != nil {
t.Fatal(err)
}
if err := f.Close(); err != nil {
t.Fatal(err)
}
defer func() { _ = os.Remove(f.Name()) }()
// CleanCodeResponse will trim the trailing newline from the AI response.
aiResponse := "package main\n\n// Bar does nothing.\nfunc Bar() {}\n"
documented := "package main\n\n// Bar does nothing.\nfunc Bar() {}"
mock := &mockStreamer{response: aiResponse}
origAutoApply := autoApply
autoApply = true
defer func() { autoApply = origAutoApply }()
processDocsFile(mock, "grok-4", f.Name())
if mock.calls != 1 {
t.Errorf("expected 1 AI call, got %d", mock.calls)
}
// Verify file was rewritten with documented content.
content, _ := os.ReadFile(f.Name())
if string(content) != documented {
t.Errorf("file content = %q, want %q", string(content), documented)
}
}
func TestRunDocs(t *testing.T) {
// runDocs with a missing file: should call processDocsFile which returns early.
mock := &mockStreamer{}
defer withMockClient(mock)()
runDocs(testCmd(), []string{"/nonexistent/file.go"})
if mock.calls != 0 {
t.Errorf("expected 0 AI calls for missing file, got %d", mock.calls)
}
}

View File

@ -1,187 +0,0 @@
package cmd
import (
"fmt"
"os"
"path/filepath"
"strings"
"github.com/fatih/color"
"github.com/spf13/cobra"
"gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/grok"
"gmgauthier.com/grokkit/internal/logger"
)
var scaffoldCmd = &cobra.Command{
Use: "scaffold FILE DESCRIPTION",
Short: "Scaffold a new file with Grok (safe preview + confirmation)",
Args: cobra.ExactArgs(2),
Run: func(cmd *cobra.Command, args []string) {
filePath := args[0]
description := args[1]
withTests, _ := cmd.Flags().GetBool("with-tests")
dryRun, _ := cmd.Flags().GetBool("dry-run")
force, _ := cmd.Flags().GetBool("force")
yes, _ := cmd.Flags().GetBool("yes")
langOverride, _ := cmd.Flags().GetString("lang")
modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("scaffold", modelFlag)
logger.Info("scaffold command started",
"file", filePath,
"description", description,
"with_tests", withTests,
"model", model)
// Safety: don't overwrite existing file unless --force
if _, err := os.Stat(filePath); err == nil && !force {
color.Red("File already exists: %s (use --force to overwrite)", filePath)
os.Exit(1)
}
dir := filepath.Dir(filePath)
if dir != "." && dir != "" {
if err := os.MkdirAll(dir, 0750); err != nil {
logger.Error("failed to create directory", "dir", dir, "error", err)
color.Red("Failed to create directory: %v", err)
os.Exit(1)
}
}
// Detect language and harvest context
lang := detectLanguage(filePath, langOverride)
context := harvestContext(filePath, lang)
// Build system prompt with style enforcement
systemPrompt := fmt.Sprintf(`You are an expert %s programmer.
Match the exact style, naming conventions, error handling, logging, and package structure of this project.
Return ONLY the complete code file. No explanations, no markdown, no backticks.`, lang)
if withTests {
systemPrompt += "\nAlso generate a basic _test.go file with at least one test when --with-tests is used."
}
messages := []map[string]string{
{"role": "system", "content": systemPrompt},
{"role": "user", "content": fmt.Sprintf(
"File path: %s\n\nProject context:\n%s\n\nCreate this new file:\n%s",
filePath, context, description)},
}
color.Yellow("Asking Grok to scaffold %s...\n", filepath.Base(filePath))
client := grok.NewClient()
raw := client.StreamSilent(messages, model)
newContent := grok.CleanCodeResponse(raw)
color.Green("Response received")
// Preview
color.Cyan("\nProposed new file:")
fmt.Printf("--- /dev/null\n")
fmt.Printf("+++ b/%s\n", filepath.Base(filePath))
fmt.Println(newContent)
if dryRun {
color.Yellow("\n--dry-run: file not written")
return
}
if !yes {
fmt.Print("\n\nCreate this file? (y/n): ")
var confirm string
if _, err := fmt.Scanln(&confirm); err != nil || (confirm != "y" && confirm != "Y") {
color.Yellow("Scaffold cancelled.")
return
}
}
// Write main file
if err := os.WriteFile(filePath, []byte(newContent), 0600); err != nil {
logger.Error("failed to write file", "file", filePath, "error", err)
color.Red("Failed to write file: %v", err)
os.Exit(1)
}
color.Green("✓ Created: %s", filePath)
// Optional test file
if withTests {
testPath := strings.TrimSuffix(filePath, filepath.Ext(filePath)) + "_test.go"
testMessages := []map[string]string{
{"role": "system", "content": systemPrompt},
{"role": "user", "content": fmt.Sprintf("Generate a basic test file for %s using the same style.", filepath.Base(filePath))},
}
testRaw := client.StreamSilent(testMessages, model)
testContent := grok.CleanCodeResponse(testRaw)
if err := os.WriteFile(testPath, []byte(testContent), 0600); err == nil {
color.Green("✓ Created test: %s", filepath.Base(testPath))
}
}
logger.Info("scaffold completed successfully", "file", filePath, "with_tests", withTests)
},
}
// Simple language detector (can be moved to internal/linter later)
func detectLanguage(path, override string) string {
if override != "" {
return override
}
ext := strings.ToLower(filepath.Ext(path))
switch ext {
case ".go":
return "Go"
case ".py":
return "Python"
case ".js", ".ts":
return "TypeScript"
case ".c":
return "C"
case ".cpp":
return "C++"
case ".java":
return "Java"
// add more as needed
default:
return "code"
}
}
// Basic context harvester (~4000 token cap)
func harvestContext(filePath, _ string) string {
var sb strings.Builder
dir := filepath.Dir(filePath)
// Siblings
files, _ := os.ReadDir(dir)
for _, f := range files {
if f.IsDir() || strings.HasPrefix(f.Name(), ".") {
continue
}
if filepath.Ext(f.Name()) == filepath.Ext(filePath) && f.Name() != filepath.Base(filePath) {
// nolint:gosec // intentional file read from project directory
content, _ := os.ReadFile(filepath.Join(dir, f.Name()))
if len(content) > 2000 {
content = content[:2000]
}
// Fixed: use Fprintf instead of WriteString + Sprintf
fmt.Fprintf(&sb, "=== %s ===\n%s\n\n", f.Name(), string(content))
}
}
// Rough token cap
if sb.Len() > 4000 {
return sb.String()[:4000] + "\n... (truncated)"
}
return sb.String()
}
func init() {
scaffoldCmd.Flags().Bool("with-tests", false, "Also scaffold a basic _test.go file")
scaffoldCmd.Flags().Bool("dry-run", false, "Preview only, do not write")
scaffoldCmd.Flags().BoolP("yes", "y", false, "Skip confirmation prompt")
scaffoldCmd.Flags().Bool("force", false, "Overwrite existing file")
scaffoldCmd.Flags().String("lang", "", "Force language for prompt (Go, Python, etc.)")
}

View File

@ -1,78 +0,0 @@
// cmd/scaffold_test.go
package cmd
import (
"os"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
// TestScaffoldCmd — FAST DEFAULT TEST (runs on make test / go test ./cmd)
func TestScaffoldCmd(t *testing.T) {
// Minimal fast unit test — no API calls, no prompts, no cost
t.Log("✓ Fast scaffold unit test (no Grok API call)")
assert.True(t, true, "command is registered and basic structure is intact")
}
// TestScaffoldCmd_Live — LIVE INTEGRATION TEST (only runs when you ask)
func TestScaffoldCmd_Live(t *testing.T) {
if !testing.Short() {
t.Skip("skipping live Grok integration test. Run with:\n go test ./cmd -run TestScaffoldCmd_Live -short -v")
}
t.Log("Running live scaffold integration tests...")
tests := []struct {
name string
args []string
expectErr bool
}{
{
name: "basic scaffold happy path",
args: []string{"scaffold", "newfile.go", "A simple test struct for configuration"},
expectErr: false,
},
{
name: "scaffold with --with-tests flag",
args: []string{"scaffold", "newfile.go", "A simple test struct", "--with-tests"},
expectErr: false,
},
{
name: "dry-run does not fail",
args: []string{"scaffold", "dry.go", "dry run test", "--dry-run"},
expectErr: false,
},
{
name: "force flag works",
args: []string{"scaffold", "exists.go", "overwrite me", "--force"},
expectErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
t.Logf("Live test: %s", tt.name)
tmpDir := t.TempDir()
origDir, _ := os.Getwd()
require.NoError(t, os.Chdir(tmpDir))
defer func() {
if err := os.Chdir(origDir); err != nil {
t.Logf("warning: failed to restore original directory: %v", err)
}
}()
rootCmd.SetArgs(tt.args)
err := rootCmd.Execute()
if tt.expectErr {
assert.Error(t, err)
return
}
assert.NoError(t, err)
t.Log("✓ Live test passed")
})
}
}

View File

@ -1,297 +0,0 @@
package cmd
import (
"errors"
"fmt"
"os"
"path/filepath"
"strings"
"github.com/fatih/color"
"github.com/spf13/cobra"
"gmgauthier.com/grokkit/config"
"gmgauthier.com/grokkit/internal/grok"
"gmgauthier.com/grokkit/internal/linter"
"gmgauthier.com/grokkit/internal/logger"
)
var testgenCmd = &cobra.Command{
Use: "testgen PATHS...",
Short: "Generate AI unit tests for files (Go/Python/C/C++, preview/apply)",
Long: `Generates comprehensive unit tests matching language conventions.
Supported: Go (table-driven), Python (pytest), C (Check), C++ (Google Test).
Examples:
grokkit testgen internal/grok/client.go
grokkit testgen app.py
grokkit testgen foo.c --yes`,
Args: cobra.MinimumNArgs(1),
SilenceUsage: true,
Run: func(cmd *cobra.Command, args []string) {
yesFlag, _ := cmd.Flags().GetBool("yes")
modelFlag, _ := cmd.Flags().GetString("model")
model := config.GetModel("testgen", modelFlag)
color.Cyan("🔧 testgen using model: %s (override with -m flag)", model)
logger.Info("testgen started", "num_files", len(args), "model", model, "auto_apply", yesFlag)
client := grok.NewClient()
supportedLangs := map[string]bool{"Go": true, "Python": true, "C": true, "C++": true}
allSuccess := true
for _, filePath := range args {
langObj, err := linter.DetectLanguage(filePath)
if err != nil {
color.Red("Failed to detect language for %s: %v", filePath, err)
allSuccess = false
continue
}
lang := langObj.Name
if !supportedLangs[lang] {
color.Yellow("Unsupported lang '%s' for %s (supported: Go/Python/C/C++)", lang, filePath)
allSuccess = false
continue
}
prompt := getTestPrompt(lang)
if err := processTestgenFile(client, filePath, lang, prompt, model, yesFlag); err != nil {
allSuccess = false
color.Red("Failed %s: %v", filePath, err)
logger.Error("processTestgenFile failed", "file", filePath, "error", err)
}
}
if allSuccess {
color.Green("\n✅ All test generations complete!")
color.Yellow("Next steps:\n make test\n make test-cover")
} else {
color.Red("\n❌ Some files failed.")
os.Exit(1)
}
},
}
func processTestgenFile(client *grok.Client, filePath, lang, systemPrompt, model string, yesFlag bool) error {
// Validate source file
srcInfo, err := os.Stat(filePath)
if err != nil {
return fmt.Errorf("source file not found: %s", filePath)
}
if srcInfo.IsDir() {
return fmt.Errorf("directories not supported: %s (use files only)", filePath)
}
origSrc, err := os.ReadFile(filePath)
if err != nil {
return fmt.Errorf("read source: %w", err)
}
cleanSrc := removeSourceComments(string(origSrc), lang)
testPath := getTestFilePath(filePath, lang)
// Handle existing test file
testExists := true
testInfo, err := os.Stat(testPath)
switch {
case errors.Is(err, os.ErrNotExist):
testExists = false
case err != nil:
return fmt.Errorf("stat test file: %w", err)
case testInfo.IsDir():
return fmt.Errorf("test path is dir: %s", testPath)
}
// Generate tests
codeLang := getCodeLang(lang)
messages := []map[string]string{
{
"role": "system",
"content": systemPrompt,
},
{
"role": "user",
"content": fmt.Sprintf("Source file %s (%s):\n```%s\n%s\n```\n\nGenerate the test file using the new Unit + Live integration pattern described in the system prompt.", filepath.Base(filePath), lang, codeLang, cleanSrc)},
}
color.Yellow("🤖 Generating tests for %s → %s...", filepath.Base(filePath), filepath.Base(testPath))
rawResponse := client.StreamSilent(messages, model)
// ← NEW: detailed debugging when the model returns nothing useful
newTestCode := grok.CleanCodeResponse(rawResponse)
if len(newTestCode) == 0 || strings.TrimSpace(newTestCode) == "" {
color.Red("❌ Cleaned response is empty")
color.Red(" Raw AI response length : %d characters", len(rawResponse))
if len(rawResponse) > 0 {
previewLen := 300
if len(rawResponse) < previewLen {
previewLen = len(rawResponse)
}
color.Red(" Raw preview (first %d chars):\n%s", previewLen, rawResponse[:previewLen])
}
return fmt.Errorf("empty generation response")
}
// Preview
color.Cyan("\n┌─ Preview: %s ────────────────────────────────────────────────", filepath.Base(testPath))
if testExists {
fmt.Println("--- a/" + filepath.Base(testPath))
} else {
fmt.Println("--- /dev/null")
}
fmt.Println("+++ b/" + filepath.Base(testPath))
fmt.Print(newTestCode)
color.Cyan("\n└────────────────────────────────────────────────────────────────")
if !yesFlag {
fmt.Print("\nApply? [y/N]: ")
var confirm string
_, err := fmt.Scanln(&confirm)
if err != nil {
return fmt.Errorf("input error: %w", err)
}
confirm = strings.TrimSpace(strings.ToLower(confirm))
if confirm != "y" && confirm != "yes" {
color.Yellow("⏭️ Skipped %s", testPath)
return nil
}
}
// Apply
if err := os.WriteFile(testPath, []byte(newTestCode), 0600); err != nil {
return fmt.Errorf("write test file: %w", err)
}
color.Green("✅ Wrote %s (%d bytes)", testPath, len(newTestCode))
logger.Debug("testgen applied", "test_file", testPath, "bytes", len(newTestCode), "source_bytes", len(origSrc))
return nil
}
func removeSourceComments(content, lang string) string {
lines := strings.Split(content, "\n")
cleanedLines := make([]string, 0, len(lines))
for _, line := range lines {
if strings.Contains(line, "Last modified") || strings.Contains(line, "Generated by") ||
strings.Contains(line, "Generated by testgen") {
continue
}
if lang == "Python" && strings.HasPrefix(strings.TrimSpace(line), "# testgen:") {
continue
}
if (lang == "C" || lang == "C++") && strings.Contains(line, "/* testgen */") {
continue
}
cleanedLines = append(cleanedLines, line)
}
return strings.Join(cleanedLines, "\n")
}
func getTestPrompt(lang string) string {
switch lang {
case "Go":
return `You are an expert Go test writer.
Generate a COMPLETE, production-ready *_test.go file using ONLY idiomatic Go.
STRICT RULES NEVER VIOLATE THESE:
- NO monkey-patching: never assign to runtime.GOOS, exec.Command, os.Stdout, os.Exit, or ANY package-level func/var
- NO global variable reassignment
- NO reflect tricks for mocking
- Use ONLY real function calls + table-driven tests
- For os.Exit or stdout testing, use simple smoke tests or pipes (no reassignment)
- Prefer simple happy-path + error-path tests
1. Fast unit test:
func TestXXX_Unit(t *testing.T) {
t.Parallel()
t.Log("✓ Fast XXX unit test")
// table-driven, real calls only
}
2. Optional live test:
func TestXXX_Live(t *testing.T) {
if !testing.Short() {
t.Skip("skipping live integration test. Run with:\n go test -run TestXXX_Live -short -v")
}
t.Log("🧪 Running live integration test...")
}
Exact rules:
- Derive test names from functions (e.g. isInstalled TestIsInstalled_Unit)
- The XXX in t.Skip MUST exactly match the live function name
- t.Parallel() on unit tests only
- NO unused imports
- Keep tests simple and idiomatic Go
- Return ONLY the full test file. No explanations, no markdown, no backticks.`
// Python/C/C++ prompts unchanged
case "Python":
return `You are a pytest expert. Generate COMPLETE pytest unit tests for the Python source.
- Use pytest fixtures where appropriate
- @pytest.mark.parametrize for tables
- Cover ALL functions/classes/methods: happy/edge/error cases
- pytest.raises for exceptions
- Modern Python 3.12+: type hints, match/case if applicable
- NO external deps unless source requires
Respond ONLY with full test_*.py file: imports, fixtures, tests. Pure Python test code.`
case "C":
return `You are a C unit testing expert using Check framework. Generate COMPLETE unit tests for C source.
- Use Check suite: Suite, tcase_begin/end, ck_assert_* macros
- Cover ALL functions: happy/edge/error cases
- Include #include <check.h>
- main() runner if needed.
Respond ONLY full test_*.c: headers, suite funcs, main. Pure C.`
case "C++":
return `You are a Google Test expert. Generate COMPLETE gtest unit tests for C++ source.
- Use TEST/TEST_F, EXPECT_*/ASSERT_*
- TYPED_TEST_SUITE if templates
- Cover ALL classes/fns: happy/edge/error
- #include <gtest/gtest.h>
Respond ONLY full test_*.cpp: includes, tests. Pure C++.`
default:
return ""
}
}
func getTestFilePath(filePath, lang string) string {
ext := filepath.Ext(filePath)
base := strings.TrimSuffix(filepath.Base(filePath), ext)
switch lang {
case "Go":
dir := filepath.Dir(filePath)
return filepath.Join(dir, base+"_test.go")
case "Python":
return filepath.Join(filepath.Dir(filePath), "test_"+base+".py")
case "C":
return filepath.Join(filepath.Dir(filePath), "test_"+base+".c")
case "C++":
return filepath.Join(filepath.Dir(filePath), "test_"+base+".cpp")
default:
return ""
}
}
func getCodeLang(lang string) string {
switch lang {
case "Go":
return "go"
case "Python":
return "python"
case "C", "C++":
return "c"
default:
return "text"
}
}
func init() {
testgenCmd.Flags().BoolP("yes", "y", false, "Auto-apply without confirmation")
}

View File

@ -1,180 +0,0 @@
package cmd
import (
"strings"
"testing"
)
// New scaffold-style dual tests (fast unit + optional live)
// These are exactly the pattern used in scaffold_test.go
func TestTestgenCmd(t *testing.T) {
t.Parallel()
t.Log("✓ Fast testgen unit test (no Grok API call)")
}
func TestTestgenCmd_Live(t *testing.T) {
if !testing.Short() {
t.Skip("skipping live Grok integration test. Run with:\n go test ./cmd -run TestTestgenCmd_Live -short -v")
}
t.Log("🧪 Running live testgen integration test with real Grok API...")
// TODO: expand later (e.g. create temp source + call processTestgenFile + verify output)
}
func TestRemoveSourceComments(t *testing.T) {
t.Parallel()
tests := []struct {
name string
input string
want string
lang string
}{
{
name: "no comments",
input: `package cmd
import "testing"
func Foo() {}`,
want: `package cmd
import "testing"
func Foo() {}`,
lang: "Go",
},
{
name: "last modified",
input: `// Last modified: 2026-03-02
package cmd`,
want: `package cmd`,
lang: "Go",
},
{
name: "generated by",
input: `// Generated by grokkit testgen
package cmd`,
want: `package cmd`,
lang: "Go",
},
{
name: "multiple removable lines",
input: `line1
// Last modified: foo
line3
// Generated by: bar
line5`,
want: `line1
line3
line5`,
lang: "Go",
},
{
name: "partial match no remove",
input: `// Modified something else
package cmd`,
want: `// Modified something else
package cmd`,
lang: "Go",
},
{
name: "python testgen",
input: `# testgen: generated
def foo(): pass`,
want: `def foo(): pass`,
lang: "Python",
},
{
name: "c testgen",
input: `/* testgen */
int foo() {}`,
want: `int foo() {}`,
lang: "C",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
t.Parallel()
got := removeSourceComments(tt.input, tt.lang)
if got != tt.want {
t.Errorf("removeSourceComments() =\n%q\nwant\n%q", got, tt.want)
}
})
}
}
func TestGetTestPrompt(t *testing.T) {
t.Parallel()
tests := []struct {
lang string
wantPrefix string
}{
{"Go", "You are an expert Go test writer."}, // ← updated to match the new generalized prompt
{"Python", "You are a pytest expert."},
{"C", "You are a C unit testing expert using Check framework."},
{"C++", "You are a Google Test expert."},
{"Invalid", ""},
}
for _, tt := range tests {
t.Run(tt.lang, func(t *testing.T) {
t.Parallel()
got := getTestPrompt(tt.lang)
if tt.wantPrefix != "" && !strings.HasPrefix(got, tt.wantPrefix) {
t.Errorf("getTestPrompt(%q) prefix =\n%q\nwant %q", tt.lang, got[:100], tt.wantPrefix)
}
if tt.wantPrefix == "" && got != "" {
t.Errorf("getTestPrompt(%q) = %q, want empty", tt.lang, got)
}
})
}
}
func TestGetTestFilePath(t *testing.T) {
t.Parallel()
tests := []struct {
filePath string
lang string
want string
}{
{"foo.go", "Go", "foo_test.go"},
{"dir/foo.py", "Python", "dir/test_foo.py"},
{"bar.c", "C", "test_bar.c"},
{"baz.cpp", "C++", "test_baz.cpp"},
}
for _, tt := range tests {
t.Run(tt.filePath+"_"+tt.lang, func(t *testing.T) {
t.Parallel()
got := getTestFilePath(tt.filePath, tt.lang)
if got != tt.want {
t.Errorf("getTestFilePath(%q, %q) = %q, want %q", tt.filePath, tt.lang, got, tt.want)
}
})
}
}
func TestGetCodeLang(t *testing.T) {
t.Parallel()
tests := []struct {
lang string
want string
}{
{"Go", "go"},
{"Python", "python"},
{"C", "c"},
{"C++", "c"},
}
for _, tt := range tests {
t.Run(tt.lang, func(t *testing.T) {
t.Parallel()
got := getCodeLang(tt.lang)
if got != tt.want {
t.Errorf("getCodeLang(%q) = %q, want %q", tt.lang, got, tt.want)
}
})
}
}

View File

@ -7,7 +7,6 @@ import (
"github.com/spf13/viper" "github.com/spf13/viper"
) )
// Load initializes the configuration from Viper
func Load() { func Load() {
home, err := os.UserHomeDir() home, err := os.UserHomeDir()
if err != nil { if err != nil {
@ -37,13 +36,11 @@ func Load() {
viper.SetDefault("commands.prdescribe.model", "grok-4") viper.SetDefault("commands.prdescribe.model", "grok-4")
viper.SetDefault("commands.review.model", "grok-4") viper.SetDefault("commands.review.model", "grok-4")
viper.SetDefault("commands.docs.model", "grok-4") viper.SetDefault("commands.docs.model", "grok-4")
viper.SetDefault("commands.query.model", "grok-4-1-fast-non-reasoning")
// Config file is optional, so we ignore read errors // Config file is optional, so we ignore read errors
_ = viper.ReadInConfig() _ = viper.ReadInConfig()
} }
// GetModel returns the model to use for a specific command, considering flags and aliases
func GetModel(commandName string, flagModel string) string { func GetModel(commandName string, flagModel string) string {
if flagModel != "" { if flagModel != "" {
if alias := viper.GetString("aliases." + flagModel); alias != "" { if alias := viper.GetString("aliases." + flagModel); alias != "" {
@ -58,12 +55,10 @@ func GetModel(commandName string, flagModel string) string {
return viper.GetString("default_model") return viper.GetString("default_model")
} }
// GetTemperature returns the temperature from the configuration
func GetTemperature() float64 { func GetTemperature() float64 {
return viper.GetFloat64("temperature") return viper.GetFloat64("temperature")
} }
// GetTimeout returns the timeout from the configuration
func GetTimeout() int { func GetTimeout() int {
timeout := viper.GetInt("timeout") timeout := viper.GetInt("timeout")
if timeout <= 0 { if timeout <= 0 {
@ -72,7 +67,6 @@ func GetTimeout() int {
return timeout return timeout
} }
// GetLogLevel returns the log level from the configuration
func GetLogLevel() string { func GetLogLevel() string {
return viper.GetString("log_level") return viper.GetString("log_level")
} }

View File

@ -98,56 +98,3 @@ func TestLoad(t *testing.T) {
// Just ensure Load doesn't panic // Just ensure Load doesn't panic
Load() Load()
} }
func TestGetTemperature(t *testing.T) {
viper.Reset()
viper.SetDefault("temperature", 0.7)
got := GetTemperature()
if got != 0.7 {
t.Errorf("GetTemperature() default = %v, want 0.7", got)
}
viper.Set("temperature", 0.8)
got = GetTemperature()
if got != 0.8 {
t.Errorf("GetTemperature() custom = %v, want 0.8", got)
}
}
func TestGetTimeout(t *testing.T) {
viper.Reset()
viper.SetDefault("timeout", 60)
got := GetTimeout()
if got != 60 {
t.Errorf("GetTimeout() default = %d, want 60", got)
}
viper.Set("timeout", 30)
got = GetTimeout()
if got != 30 {
t.Errorf("GetTimeout() = %d, want 30", got)
}
viper.Set("timeout", 0)
got = GetTimeout()
if got != 60 {
t.Errorf("GetTimeout() invalid = %d, want 60", got)
}
}
func TestGetLogLevel(t *testing.T) {
viper.Reset()
viper.SetDefault("log_level", "info")
got := GetLogLevel()
if got != "info" {
t.Errorf("GetLogLevel() default = %q, want info", got)
}
viper.Set("log_level", "debug")
got = GetLogLevel()
if got != "debug" {
t.Errorf("GetLogLevel() custom = %q, want debug", got)
}
}

View File

@ -23,7 +23,7 @@ Grokkit follows these core principles:
- Single responsibility per package - Single responsibility per package
### 2. **Safety by Default** ### 2. **Safety by Default**
- Git-based version control for change management - File backups before any modification
- Confirmation prompts for destructive actions - Confirmation prompts for destructive actions
- Comprehensive error handling - Comprehensive error handling
@ -334,6 +334,7 @@ User: grokkit edit file.go "instruction"
editCmd.Run() editCmd.Run()
├─→ Validate file exists ├─→ Validate file exists
├─→ Read original content ├─→ Read original content
├─→ Create backup (.bak)
├─→ Clean "Last modified" comments ├─→ Clean "Last modified" comments
grok.Client.StreamSilent() # Get AI response grok.Client.StreamSilent() # Get AI response
@ -346,7 +347,8 @@ Display preview # Show diff-style output
Prompt user (y/n) Prompt user (y/n)
Write file (if confirmed) Write file (if confirmed)
└─→ logger.Info() # Log changes ├─→ logger.Info() # Log changes
└─→ Keep backup
``` ```
### Commit Command Flow ### Commit Command Flow
@ -392,6 +394,8 @@ grok.Client.StreamSilent() # Get fixed code
grok.CleanCodeResponse() # Remove markdown grok.CleanCodeResponse() # Remove markdown
Create backup (.bak)
(if not --auto-fix) → Show preview + prompt (if not --auto-fix) → Show preview + prompt
Write fixed file Write fixed file
@ -567,7 +571,7 @@ func Run(args []string) (string, error) {
**Current measures:** **Current measures:**
- API key via environment variable - API key via environment variable
- No credential storage - No credential storage
- Git-based rollbacks for safety - Backup files for safety
**Future considerations:** **Future considerations:**
- Encrypted config storage - Encrypted config storage

View File

@ -322,27 +322,27 @@ pwd
ls ls
``` ```
### Rolling back AI changes ### Permission denied on backup creation
**Symptom:** **Symptom:**
AI suggested changes are undesired or introduced bugs. ```
Failed to create backup: permission denied
**Solution:**
Use Git to roll back:
```bash
# If not yet added/staged:
git restore <file>
# If staged:
git restore --staged <file>
git restore <file>
# If committed:
git revert HEAD
``` ```
Always review changes with `git diff` before and after applying AI suggestions. **Solution:**
```bash
# Check file permissions
ls -la main.go
# Check directory permissions
ls -ld .
# Fix permissions
chmod 644 main.go # File
chmod 755 . # Directory
# Or run from writable directory
```
### Failed to write file ### Failed to write file
@ -581,8 +581,8 @@ After applying AI fixes, re-running the linter still reports issues.
**Solution:** **Solution:**
```bash ```bash
# Some issues may require manual intervention # Some issues may require manual intervention
# Use git diff to compare changes # Check the backup file to compare
git diff file.py diff file.py file.py.bak
# Apply fixes iteratively # Apply fixes iteratively
grokkit lint file.py # Apply fixes grokkit lint file.py # Apply fixes

25
go.mod
View File

@ -3,29 +3,46 @@ module gmgauthier.com/grokkit
go 1.24.2 go 1.24.2
require ( require (
github.com/charmbracelet/bubbles v1.0.0
github.com/charmbracelet/bubbletea v1.3.10
github.com/charmbracelet/lipgloss v1.1.0
github.com/fatih/color v1.18.0 github.com/fatih/color v1.18.0
github.com/spf13/cobra v1.9.1 github.com/spf13/cobra v1.9.1
github.com/spf13/viper v1.21.0 github.com/spf13/viper v1.21.0
github.com/stretchr/testify v1.11.1
) )
require ( require (
github.com/davecgh/go-spew v1.1.1 // indirect github.com/atotto/clipboard v0.1.4 // indirect
github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
github.com/charmbracelet/colorprofile v0.4.1 // indirect
github.com/charmbracelet/x/ansi v0.11.6 // indirect
github.com/charmbracelet/x/cellbuf v0.0.15 // indirect
github.com/charmbracelet/x/term v0.2.2 // indirect
github.com/clipperhouse/displaywidth v0.9.0 // indirect
github.com/clipperhouse/stringish v0.1.1 // indirect
github.com/clipperhouse/uax29/v2 v2.5.0 // indirect
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
github.com/fsnotify/fsnotify v1.9.0 // indirect github.com/fsnotify/fsnotify v1.9.0 // indirect
github.com/go-viper/mapstructure/v2 v2.4.0 // indirect github.com/go-viper/mapstructure/v2 v2.4.0 // indirect
github.com/inconshreveable/mousetrap v1.1.0 // indirect github.com/inconshreveable/mousetrap v1.1.0 // indirect
github.com/lucasb-eyer/go-colorful v1.3.0 // indirect
github.com/mattn/go-colorable v0.1.13 // indirect github.com/mattn/go-colorable v0.1.13 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect github.com/mattn/go-isatty v0.0.20 // indirect
github.com/mattn/go-localereader v0.0.1 // indirect
github.com/mattn/go-runewidth v0.0.19 // indirect
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
github.com/muesli/cancelreader v0.2.2 // indirect
github.com/muesli/termenv v0.16.0 // indirect
github.com/pelletier/go-toml/v2 v2.2.4 // indirect github.com/pelletier/go-toml/v2 v2.2.4 // indirect
github.com/pmezard/go-difflib v1.0.0 // indirect github.com/rivo/uniseg v0.4.7 // indirect
github.com/sagikazarmark/locafero v0.11.0 // indirect github.com/sagikazarmark/locafero v0.11.0 // indirect
github.com/sourcegraph/conc v0.3.1-0.20240121214520-5f936abd7ae8 // indirect github.com/sourcegraph/conc v0.3.1-0.20240121214520-5f936abd7ae8 // indirect
github.com/spf13/afero v1.15.0 // indirect github.com/spf13/afero v1.15.0 // indirect
github.com/spf13/cast v1.10.0 // indirect github.com/spf13/cast v1.10.0 // indirect
github.com/spf13/pflag v1.0.10 // indirect github.com/spf13/pflag v1.0.10 // indirect
github.com/subosito/gotenv v1.6.0 // indirect github.com/subosito/gotenv v1.6.0 // indirect
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
go.yaml.in/yaml/v3 v3.0.4 // indirect go.yaml.in/yaml/v3 v3.0.4 // indirect
golang.org/x/sys v0.38.0 // indirect golang.org/x/sys v0.38.0 // indirect
golang.org/x/text v0.28.0 // indirect golang.org/x/text v0.28.0 // indirect
gopkg.in/yaml.v3 v3.0.1 // indirect
) )

49
go.sum
View File

@ -1,6 +1,36 @@
github.com/MakeNowJust/heredoc v1.0.0 h1:cXCdzVdstXyiTqTvfqk9SDHpKNjxuom+DOlyEeQ4pzQ=
github.com/MakeNowJust/heredoc v1.0.0/go.mod h1:mG5amYoWBHf8vpLOuehzbGGw0EHxpZZ6lCpQ4fNJ8LE=
github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4=
github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI=
github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
github.com/aymanbagabas/go-udiff v0.3.1 h1:LV+qyBQ2pqe0u42ZsUEtPiCaUoqgA9gYRDs3vj1nolY=
github.com/aymanbagabas/go-udiff v0.3.1/go.mod h1:G0fsKmG+P6ylD0r6N/KgQD/nWzgfnl8ZBcNLgcbrw8E=
github.com/charmbracelet/bubbles v1.0.0 h1:12J8/ak/uCZEMQ6KU7pcfwceyjLlWsDLAxB5fXonfvc=
github.com/charmbracelet/bubbles v1.0.0/go.mod h1:9d/Zd5GdnauMI5ivUIVisuEm3ave1XwXtD1ckyV6r3E=
github.com/charmbracelet/bubbletea v1.3.10 h1:otUDHWMMzQSB0Pkc87rm691KZ3SWa4KUlvF9nRvCICw=
github.com/charmbracelet/bubbletea v1.3.10/go.mod h1:ORQfo0fk8U+po9VaNvnV95UPWA1BitP1E0N6xJPlHr4=
github.com/charmbracelet/colorprofile v0.4.1 h1:a1lO03qTrSIRaK8c3JRxJDZOvhvIeSco3ej+ngLk1kk=
github.com/charmbracelet/colorprofile v0.4.1/go.mod h1:U1d9Dljmdf9DLegaJ0nGZNJvoXAhayhmidOdcBwAvKk=
github.com/charmbracelet/lipgloss v1.1.0 h1:vYXsiLHVkK7fp74RkV7b2kq9+zDLoEU4MZoFqR/noCY=
github.com/charmbracelet/lipgloss v1.1.0/go.mod h1:/6Q8FR2o+kj8rz4Dq0zQc3vYf7X+B0binUUBwA0aL30=
github.com/charmbracelet/x/ansi v0.11.6 h1:GhV21SiDz/45W9AnV2R61xZMRri5NlLnl6CVF7ihZW8=
github.com/charmbracelet/x/ansi v0.11.6/go.mod h1:2JNYLgQUsyqaiLovhU2Rv/pb8r6ydXKS3NIttu3VGZQ=
github.com/charmbracelet/x/cellbuf v0.0.15 h1:ur3pZy0o6z/R7EylET877CBxaiE1Sp1GMxoFPAIztPI=
github.com/charmbracelet/x/cellbuf v0.0.15/go.mod h1:J1YVbR7MUuEGIFPCaaZ96KDl5NoS0DAWkskup+mOY+Q=
github.com/charmbracelet/x/term v0.2.2 h1:xVRT/S2ZcKdhhOuSP4t5cLi5o+JxklsoEObBSgfgZRk=
github.com/charmbracelet/x/term v0.2.2/go.mod h1:kF8CY5RddLWrsgVwpw4kAa6TESp6EB5y3uxGLeCqzAI=
github.com/clipperhouse/displaywidth v0.9.0 h1:Qb4KOhYwRiN3viMv1v/3cTBlz3AcAZX3+y9OLhMtAtA=
github.com/clipperhouse/displaywidth v0.9.0/go.mod h1:aCAAqTlh4GIVkhQnJpbL0T/WfcrJXHcj8C0yjYcjOZA=
github.com/clipperhouse/stringish v0.1.1 h1:+NSqMOr3GR6k1FdRhhnXrLfztGzuG+VuFDfatpWHKCs=
github.com/clipperhouse/stringish v0.1.1/go.mod h1:v/WhFtE1q0ovMta2+m+UbpZ+2/HEXNWYXQgCt4hdOzA=
github.com/clipperhouse/uax29/v2 v2.5.0 h1:x7T0T4eTHDONxFJsL94uKNKPHrclyFI0lm7+w94cO8U=
github.com/clipperhouse/uax29/v2 v2.5.0/go.mod h1:Wn1g7MK6OoeDT0vL+Q0SQLDz/KpfsVRgg6W7ihQeh4g=
github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g= github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f h1:Y/CXytFA4m6baUTXGLOoWe4PQhGxaX0KpnayAqC48p4=
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f/go.mod h1:vw97MGsxSvLiUE2X8qFplwetxpGLQrlU1Q9AUEIzCaM=
github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM= github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU= github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
@ -17,15 +47,29 @@ github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
github.com/lucasb-eyer/go-colorful v1.3.0 h1:2/yBRLdWBZKrf7gB40FoiKfAWYQ0lqNcbuQwVHXptag=
github.com/lucasb-eyer/go-colorful v1.3.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4=
github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88=
github.com/mattn/go-runewidth v0.0.19 h1:v++JhqYnZuu5jSKrk9RbgF5v4CGUjqRfBm05byFGLdw=
github.com/mattn/go-runewidth v0.0.19/go.mod h1:XBkDxAl56ILZc9knddidhrOlY5R/pDhgLpndooCuJAs=
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo=
github.com/muesli/termenv v0.16.0 h1:S5AlUN9dENB57rsbnkPyfdGuWIlkmzJjbFf0Tf5FWUc=
github.com/muesli/termenv v0.16.0/go.mod h1:ZRfOIKPFDYQoDFF4Olj7/QJbW60Ol/kL1pU3VfY/Cnk=
github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4= github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4=
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY= github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8= github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs= github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
@ -48,8 +92,13 @@ github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U= github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8= github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=
github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU= github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
go.yaml.in/yaml/v3 v3.0.4 h1:tfq32ie2Jv2UxXFdLJdh3jXuOzWiL1fo0bu/FbuKpbc= go.yaml.in/yaml/v3 v3.0.4 h1:tfq32ie2Jv2UxXFdLJdh3jXuOzWiL1fo0bu/FbuKpbc=
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg= go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
golang.org/x/exp v0.0.0-20231006140011-7918f672742d h1:jtJma62tbqLibJ5sFQz8bKtEM8rJBtfilJ2qTU199MI=
golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQzzkH36rKxoZW1tw7ZJpeKx+hdo=
golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.38.0 h1:3yZWxaJjBmCWXqhN1qh02AkOnCQ1poK6oF+a7xWL6Gc= golang.org/x/sys v0.38.0 h1:3yZWxaJjBmCWXqhN1qh02AkOnCQ1poK6oF+a7xWL6Gc=

View File

@ -9,7 +9,6 @@ type GitError struct {
} }
func (e *GitError) Error() string { func (e *GitError) Error() string {
_ = e.Err // keep field used for error message
return fmt.Sprintf("git %s failed: %v", e.Command, e.Err) return fmt.Sprintf("git %s failed: %v", e.Command, e.Err)
} }

View File

@ -17,8 +17,8 @@ func TestGitError(t *testing.T) {
t.Errorf("GitError.Error() = %q, want %q", gitErr.Error(), expected) t.Errorf("GitError.Error() = %q, want %q", gitErr.Error(), expected)
} }
if !errors.Is(gitErr, baseErr) { if gitErr.Unwrap() != baseErr {
t.Errorf("GitError did not wrap base error") t.Errorf("GitError.Unwrap() did not return base error")
} }
} }
@ -68,19 +68,7 @@ func TestFileError(t *testing.T) {
t.Errorf("FileError.Error() = %q, want %q", fileErr.Error(), expected) t.Errorf("FileError.Error() = %q, want %q", fileErr.Error(), expected)
} }
if !errors.Is(fileErr, baseErr) { if fileErr.Unwrap() != baseErr {
t.Errorf("FileError did not wrap base error") t.Errorf("FileError.Unwrap() did not return base error")
}
}
func TestAPIErrorUnwrap(t *testing.T) {
baseErr := errors.New("base api err")
apiErr := &APIError{
StatusCode: 500,
Message: "internal error",
Err: baseErr,
}
if !errors.Is(apiErr, baseErr) {
t.Errorf("APIError.Unwrap() = %v, want %v", apiErr.Unwrap(), baseErr)
} }
} }

View File

@ -12,7 +12,6 @@ func Run(args []string) (string, error) {
cmdStr := "git " + strings.Join(args, " ") cmdStr := "git " + strings.Join(args, " ")
logger.Debug("executing git command", "command", cmdStr, "args", args) logger.Debug("executing git command", "command", cmdStr, "args", args)
// nolint:gosec // intentional subprocess for git operation
out, err := exec.Command("git", args...).Output() out, err := exec.Command("git", args...).Output()
if err != nil { if err != nil {
logger.Error("git command failed", logger.Error("git command failed",
@ -31,32 +30,8 @@ func Run(args []string) (string, error) {
func IsRepo() bool { func IsRepo() bool {
logger.Debug("checking if directory is a git repository") logger.Debug("checking if directory is a git repository")
// nolint:gosec // intentional subprocess for git repository check
_, err := exec.Command("git", "rev-parse", "--is-inside-work-tree").Output() _, err := exec.Command("git", "rev-parse", "--is-inside-work-tree").Output()
isRepo := err == nil isRepo := err == nil
logger.Debug("git repository check completed", "is_repo", isRepo) logger.Debug("git repository check completed", "is_repo", isRepo)
return isRepo return isRepo
} }
func LatestTag() (string, error) {
out, err := Run([]string{"describe", "--tags", "--abbrev=0"})
if err != nil {
return "", err
}
return strings.TrimSpace(out), nil
}
// PreviousTag returns the tag immediately before the given one.
func PreviousTag(current string) (string, error) {
out, err := Run([]string{"describe", "--tags", "--abbrev=0", current + "^"})
if err != nil {
return "", err
}
return strings.TrimSpace(out), nil
}
// LogSince returns formatted commit log since the given ref (exactly matches the todo spec).
func LogSince(since string) (string, error) {
args := []string{"log", "--pretty=format:%s%n%b%n---", since + "..HEAD"}
return Run(args)
}

View File

@ -76,23 +76,20 @@ func (c *Client) streamInternal(messages []map[string]string, model string, temp
"stream": true, "stream": true,
} }
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
// Manual cancel before os.Exit; otherwise defer is fine for the main path.
defer cancel()
body, err := json.Marshal(payload) body, err := json.Marshal(payload)
if err != nil { if err != nil {
logger.Error("failed to marshal API request", "error", err) logger.Error("failed to marshal API request", "error", err)
color.Red("Failed to marshal request: %v", err) color.Red("Failed to marshal request: %v", err)
cancel()
os.Exit(1) os.Exit(1)
} }
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(body)) ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
defer cancel()
req, err := http.NewRequestWithContext(ctx, "POST", url, bytes.NewReader(body))
if err != nil { if err != nil {
logger.Error("failed to create HTTP request", "error", err, "url", url) logger.Error("failed to create HTTP request", "error", err, "url", url)
color.Red("Failed to create request: %v", err) color.Red("Failed to create request: %v", err)
cancel()
os.Exit(1) os.Exit(1)
} }
req.Header.Set("Authorization", "Bearer "+c.APIKey) req.Header.Set("Authorization", "Bearer "+c.APIKey)
@ -107,10 +104,9 @@ func (c *Client) streamInternal(messages []map[string]string, model string, temp
"model", model, "model", model,
"duration_ms", time.Since(startTime).Milliseconds()) "duration_ms", time.Since(startTime).Milliseconds())
color.Red("Request failed: %v", err) color.Red("Request failed: %v", err)
cancel()
os.Exit(1) os.Exit(1)
} }
defer func() { _ = resp.Body.Close() }() defer resp.Body.Close()
logger.Debug("API response received", logger.Debug("API response received",
"status", resp.Status, "status", resp.Status,

View File

@ -1,9 +1,6 @@
package grok package grok
import ( import (
"fmt"
"net/http"
"net/http/httptest"
"os" "os"
"testing" "testing"
) )
@ -51,95 +48,18 @@ func TestCleanCodeResponse(t *testing.T) {
} }
} }
func sseServer(chunks []string) *httptest.Server {
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/event-stream")
for _, c := range chunks {
_, _ = fmt.Fprintf(w, "data: {\"choices\":[{\"delta\":{\"content\":%q}}]}\n\n", c)
}
_, _ = fmt.Fprintf(w, "data: [DONE]\n\n")
}))
}
func TestStreamSilent(t *testing.T) {
srv := sseServer([]string{"Hello", " ", "World"})
defer srv.Close()
client := &Client{APIKey: "test-key", BaseURL: srv.URL}
got := client.StreamSilent([]map[string]string{{"role": "user", "content": "hi"}}, "test-model")
if got != "Hello World" {
t.Errorf("StreamSilent() = %q, want %q", got, "Hello World")
}
}
func TestStream(t *testing.T) {
srv := sseServer([]string{"foo", "bar"})
defer srv.Close()
client := &Client{APIKey: "test-key", BaseURL: srv.URL}
got := client.Stream([]map[string]string{{"role": "user", "content": "hi"}}, "test-model")
if got != "foobar" {
t.Errorf("Stream() = %q, want %q", got, "foobar")
}
}
func TestStreamWithTemp(t *testing.T) {
srv := sseServer([]string{"response"})
defer srv.Close()
client := &Client{APIKey: "test-key", BaseURL: srv.URL}
got := client.StreamWithTemp([]map[string]string{{"role": "user", "content": "hi"}}, "test-model", 0.5)
if got != "response" {
t.Errorf("StreamWithTemp() = %q, want %q", got, "response")
}
}
func TestStreamDoneSignal(t *testing.T) {
// Verifies that [DONE] stops processing and non-content chunks are skipped
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/event-stream")
_, _ = fmt.Fprintf(w, "data: {\"choices\":[{\"delta\":{\"content\":\"ok\"}}]}\n\n")
_, _ = fmt.Fprintf(w, "data: [DONE]\n\n")
// This line should never be processed
_, _ = fmt.Fprintf(w, "data: {\"choices\":[{\"delta\":{\"content\":\"extra\"}}]}\n\n")
}))
defer srv.Close()
client := &Client{APIKey: "test-key", BaseURL: srv.URL}
got := client.StreamSilent(nil, "test-model")
if got != "ok" {
t.Errorf("got %q, want %q", got, "ok")
}
}
func TestStreamEmptyResponse(t *testing.T) {
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/event-stream")
_, _ = fmt.Fprintf(w, "data: [DONE]\n\n")
}))
defer srv.Close()
client := &Client{APIKey: "test-key", BaseURL: srv.URL}
got := client.StreamSilent(nil, "test-model")
if got != "" {
t.Errorf("got %q, want empty string", got)
}
}
func TestNewClient(t *testing.T) { func TestNewClient(t *testing.T) {
// Save and restore env // Save and restore env
oldKey := os.Getenv("XAI_API_KEY") oldKey := os.Getenv("XAI_API_KEY")
defer func() { defer func() {
if oldKey != "" { if oldKey != "" {
_ = os.Setenv("XAI_API_KEY", oldKey) os.Setenv("XAI_API_KEY", oldKey)
} else { } else {
_ = os.Unsetenv("XAI_API_KEY") os.Unsetenv("XAI_API_KEY")
} }
}() }()
if err := os.Setenv("XAI_API_KEY", "test-key"); err != nil { os.Setenv("XAI_API_KEY", "test-key")
t.Fatal(err)
}
client := NewClient() client := NewClient()
if client.APIKey != "test-key" { if client.APIKey != "test-key" {

View File

@ -1,7 +1,6 @@
package linter package linter
import ( import (
"errors"
"fmt" "fmt"
"os" "os"
"os/exec" "os/exec"
@ -215,7 +214,7 @@ func FindAvailableLinter(lang *Language) (*Linter, error) {
} }
// Build install instructions // Build install instructions
installOptions := make([]string, 0, len(lang.Linters)) var installOptions []string
for _, linter := range lang.Linters { for _, linter := range lang.Linters {
installOptions = append(installOptions, installOptions = append(installOptions,
fmt.Sprintf(" - %s: %s", linter.Name, linter.InstallInfo)) fmt.Sprintf(" - %s: %s", linter.Name, linter.InstallInfo))
@ -233,12 +232,10 @@ func RunLinter(filePath string, linter Linter) (*LintResult, error) {
"command", linter.Command) "command", linter.Command)
// Build command arguments // Build command arguments
linterArgs := append([]string{}, linter.Args...) args := append(linter.Args, filePath)
linterArgs = append(linterArgs, filePath)
// Execute linter // Execute linter
// nolint:gosec // intentional subprocess for linter cmd := exec.Command(linter.Command, args...)
cmd := exec.Command(linter.Command, linterArgs...)
output, err := cmd.CombinedOutput() output, err := cmd.CombinedOutput()
result := &LintResult{ result := &LintResult{
@ -248,8 +245,7 @@ func RunLinter(filePath string, linter Linter) (*LintResult, error) {
} }
if err != nil { if err != nil {
var exitErr *exec.ExitError if exitErr, ok := err.(*exec.ExitError); ok {
if errors.As(err, &exitErr) {
result.ExitCode = exitErr.ExitCode() result.ExitCode = exitErr.ExitCode()
result.HasIssues = true result.HasIssues = true
logger.Info("linter found issues", logger.Info("linter found issues",
@ -306,7 +302,7 @@ func LintFile(filePath string) (*LintResult, error) {
// GetSupportedLanguages returns a list of all supported languages // GetSupportedLanguages returns a list of all supported languages
func GetSupportedLanguages() []string { func GetSupportedLanguages() []string {
langs := make([]string, 0, len(languages)) var langs []string
for _, lang := range languages { for _, lang := range languages {
langs = append(langs, fmt.Sprintf("%s (%s)", lang.Name, strings.Join(lang.Extensions, ", "))) langs = append(langs, fmt.Sprintf("%s (%s)", lang.Name, strings.Join(lang.Extensions, ", ")))
} }

View File

@ -151,7 +151,7 @@ func main() {
fmt.Println("Hello, World!") fmt.Println("Hello, World!")
} }
` `
if err := os.WriteFile(testFile, []byte(validCode), 0600); err != nil { if err := os.WriteFile(testFile, []byte(validCode), 0644); err != nil {
t.Fatalf("Failed to create test file: %v", err) t.Fatalf("Failed to create test file: %v", err)
} }
@ -207,7 +207,7 @@ func main() {
fmt.Println("Hello, World!") fmt.Println("Hello, World!")
} }
` `
if err := os.WriteFile(testFile, []byte(validCode), 0600); err != nil { if err := os.WriteFile(testFile, []byte(validCode), 0644); err != nil {
t.Fatalf("Failed to create test file: %v", err) t.Fatalf("Failed to create test file: %v", err)
} }
@ -237,7 +237,7 @@ func main() {
t.Run("Lint unsupported file type", func(t *testing.T) { t.Run("Lint unsupported file type", func(t *testing.T) {
testFile := filepath.Join(tmpDir, "test.txt") testFile := filepath.Join(tmpDir, "test.txt")
if err := os.WriteFile(testFile, []byte("hello"), 0600); err != nil { if err := os.WriteFile(testFile, []byte("hello"), 0644); err != nil {
t.Fatalf("Failed to create test file: %v", err) t.Fatalf("Failed to create test file: %v", err)
} }

View File

@ -21,12 +21,12 @@ func Init(logLevel string) error {
} }
logDir := filepath.Join(home, ".config", "grokkit") logDir := filepath.Join(home, ".config", "grokkit")
if err := os.MkdirAll(logDir, 0750); err != nil { if err := os.MkdirAll(logDir, 0755); err != nil {
return err return err
} }
logFile := filepath.Join(logDir, "grokkit.log") logFile := filepath.Join(logDir, "grokkit.log")
file, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0600) file, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
if err != nil { if err != nil {
return err return err
} }

View File

@ -1,26 +1,18 @@
package logger package logger
import ( import (
"context"
"log/slog"
"os" "os"
"path/filepath" "path/filepath"
"strings" "strings"
"testing" "testing"
) )
func setHome(t *testing.T, dir string) {
t.Helper()
if err := os.Setenv("HOME", dir); err != nil {
t.Fatal(err)
}
}
func TestInit(t *testing.T) { func TestInit(t *testing.T) {
// Create temp directory
tmpDir := t.TempDir() tmpDir := t.TempDir()
oldHome := os.Getenv("HOME") oldHome := os.Getenv("HOME")
setHome(t, tmpDir) os.Setenv("HOME", tmpDir)
defer func() { _ = os.Setenv("HOME", oldHome) }() defer os.Setenv("HOME", oldHome)
tests := []struct { tests := []struct {
name string name string
@ -40,6 +32,7 @@ func TestInit(t *testing.T) {
t.Errorf("Init(%q) unexpected error: %v", tt.logLevel, err) t.Errorf("Init(%q) unexpected error: %v", tt.logLevel, err)
} }
// Check log file was created
logFile := filepath.Join(tmpDir, ".config", "grokkit", "grokkit.log") logFile := filepath.Join(tmpDir, ".config", "grokkit", "grokkit.log")
if _, err := os.Stat(logFile); os.IsNotExist(err) { if _, err := os.Stat(logFile); os.IsNotExist(err) {
t.Errorf("Log file not created at %s", logFile) t.Errorf("Log file not created at %s", logFile)
@ -51,18 +44,20 @@ func TestInit(t *testing.T) {
func TestLogging(t *testing.T) { func TestLogging(t *testing.T) {
tmpDir := t.TempDir() tmpDir := t.TempDir()
oldHome := os.Getenv("HOME") oldHome := os.Getenv("HOME")
setHome(t, tmpDir) os.Setenv("HOME", tmpDir)
defer func() { _ = os.Setenv("HOME", oldHome) }() defer os.Setenv("HOME", oldHome)
if err := Init("debug"); err != nil { if err := Init("debug"); err != nil {
t.Fatalf("Init() failed: %v", err) t.Fatalf("Init() failed: %v", err)
} }
// Test all log levels with structured fields
Debug("test debug message", "key", "value") Debug("test debug message", "key", "value")
Info("test info message", "count", 42) Info("test info message", "count", 42)
Warn("test warn message", "enabled", true) Warn("test warn message", "enabled", true)
Error("test error message", "error", "something went wrong") Error("test error message", "error", "something went wrong")
// Verify log file has content
logFile := filepath.Join(tmpDir, ".config", "grokkit", "grokkit.log") logFile := filepath.Join(tmpDir, ".config", "grokkit", "grokkit.log")
content, err := os.ReadFile(logFile) content, err := os.ReadFile(logFile)
if err != nil { if err != nil {
@ -72,6 +67,7 @@ func TestLogging(t *testing.T) {
t.Errorf("Log file is empty") t.Errorf("Log file is empty")
} }
// Check for JSON structure (slog uses JSON handler)
contentStr := string(content) contentStr := string(content)
if !strings.Contains(contentStr, `"level"`) { if !strings.Contains(contentStr, `"level"`) {
t.Errorf("Log content doesn't contain JSON level field") t.Errorf("Log content doesn't contain JSON level field")
@ -84,16 +80,20 @@ func TestLogging(t *testing.T) {
func TestSetLevel(t *testing.T) { func TestSetLevel(t *testing.T) {
tmpDir := t.TempDir() tmpDir := t.TempDir()
oldHome := os.Getenv("HOME") oldHome := os.Getenv("HOME")
setHome(t, tmpDir) os.Setenv("HOME", tmpDir)
defer func() { _ = os.Setenv("HOME", oldHome) }() defer os.Setenv("HOME", oldHome)
if err := Init("info"); err != nil { if err := Init("info"); err != nil {
t.Fatalf("Init() failed: %v", err) t.Fatalf("Init() failed: %v", err)
} }
// Change level to debug
SetLevel("debug") SetLevel("debug")
// Log at debug level
Debug("debug after level change", "test", true) Debug("debug after level change", "test", true)
// Verify log file has the debug message
logFile := filepath.Join(tmpDir, ".config", "grokkit", "grokkit.log") logFile := filepath.Join(tmpDir, ".config", "grokkit", "grokkit.log")
content, err := os.ReadFile(logFile) content, err := os.ReadFile(logFile)
if err != nil { if err != nil {
@ -108,42 +108,16 @@ func TestSetLevel(t *testing.T) {
func TestWith(t *testing.T) { func TestWith(t *testing.T) {
tmpDir := t.TempDir() tmpDir := t.TempDir()
oldHome := os.Getenv("HOME") oldHome := os.Getenv("HOME")
setHome(t, tmpDir) os.Setenv("HOME", tmpDir)
defer func() { _ = os.Setenv("HOME", oldHome) }() defer os.Setenv("HOME", oldHome)
if err := Init("info"); err != nil { if err := Init("info"); err != nil {
t.Fatalf("Init() failed: %v", err) t.Fatalf("Init() failed: %v", err)
} }
// Create logger with context
contextLogger := With("request_id", "123", "user", "testuser") contextLogger := With("request_id", "123", "user", "testuser")
if contextLogger == nil { if contextLogger == nil {
t.Errorf("With() returned nil logger") t.Errorf("With() returned nil logger")
} }
} }
func TestWithContext(t *testing.T) {
tmpDir := t.TempDir()
oldHome := os.Getenv("HOME")
setHome(t, tmpDir)
defer func() { _ = os.Setenv("HOME", oldHome) }()
t.Run("without init", func(t *testing.T) {
logger = nil // ensure nil
ctx := context.Background()
l := WithContext(ctx)
if l != slog.Default() {
t.Errorf("WithContext without logger = %v, want slog.Default()", l)
}
})
t.Run("with init", func(t *testing.T) {
if err := Init("info"); err != nil {
t.Fatalf("Init() = %v", err)
}
ctx := context.Background()
l := WithContext(ctx)
if l == slog.Default() {
t.Errorf("WithContext with logger == slog.Default()")
}
})
}

View File

@ -33,7 +33,6 @@ func TestVersionInfo(t *testing.T) {
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
t.Parallel()
tt.check(t) tt.check(t)
}) })
} }

View File

@ -1,65 +0,0 @@
#!/bin/bash
# release.sh — One-command release driver for Grokkit
# Usage: ./release.sh v0.2.3
set -euo pipefail
VERSION="${1:-}"
if [[ -z "$VERSION" || ! "$VERSION" =~ ^v[0-9]+\.[0-9]+\.[0-9]+ ]]; then
echo "❌ Usage: $0 vX.Y.Z"
echo " Example: $0 v0.2.3"
exit 1
fi
echo "🚀 Starting release process for $VERSION..."
# Safety check: clean working tree
if [[ -n $(git status --porcelain) ]]; then
echo "❌ Working tree is dirty. Commit or stash changes first."
exit 1
fi
# Final human confirmation before anything touches git
echo ""
echo "This will:"
echo " 1. Create git tag $VERSION"
echo " 2. Run grokkit changelog (with preview + your confirmation)"
echo " 3. Stage CHANGELOG.md and run grokkit commit (AI message + confirmation)"
echo " 4. Push commit + tag"
echo ""
read -p "Proceed with release $VERSION? (y/N) " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
echo "Aborted."
exit 1
fi
# 1. Tag first (so changelog can see it as LatestTag)
echo "🏷️ Creating tag $VERSION..."
git tag "$VERSION"
# 2. Generate changelog (interactive preview + confirm inside the command)
echo "📝 Generating CHANGELOG.md section..."
grokkit changelog --version "$VERSION"
# 3. Commit with Grok (uses the exact same safety flow you already love)
echo "📦 Staging and committing changelog..."
git add CHANGELOG.md
grokkit commit
# 4. Push everything
echo "📤 Pushing commit + tag..."
git push
git push --tags
# 5. Show the exact notes ready for Gitea release page
echo ""
echo "✅ Release $VERSION complete!"
echo ""
echo "📋 Copy-paste this into the Gitea release notes:"
echo "------------------------------------------------"
grokkit changelog --stdout --version "$VERSION"
echo "------------------------------------------------"
echo ""
echo "🎉 Done! Go create the release on Gitea now."

View File

@ -1,27 +0,0 @@
# Grokkit TODO List
This document provides a table of contents for all tasks and features currently tracked in the `todo/` directory.
## Queued
* [1] [interactive-agent.md](./queued/interactive-agent.md) : Grokkit Interactive Agent
* [2] [rg.md](./queued/rg.md) : grokkit agent ripgrep (rg) integration
* [3] [gotools.md](./queued/gotools.md) : grokkit agent Go tools integration
* [4] [make.md](./queued/make.md) : grokkit agent make integration
* [5] [tea.md](./queued/tea.md) : grokkit agent tea (Gitea CLI) integration
* [6] [cnotes.md](./queued/cnotes.md) : grokkit agent cnotes integration
* [7] [profile.md](./queued/profile.md) : grokkit profile
* [8] [pprof.md](./queued/pprof.md) : grokkit agent pprof integration
* [9] [audit.md](./queued/audit.md) : grokkit audit
* [10] [git-chglog.md](./queued/git-chglog.md) : grokkit agent git-chglog integration
* [11] [admin.md](./queued/admin.md) : grokkit admin tool (to show token usage and other admin-only features)
* [99] [TODO_ITEM.md](./queued/TODO_ITEM.md) : TODO ITEM template
## Completed
* [non-interactive-query](./completed/non-interactive-query.md) : grokkit query *(done - v0.1.9)*
* [changelog.md](./completed/changelog.md) : grokkit changelog *(done — v0.1.8+)*
* [3-new-feature-suggestions.md](./completed/3-new-feature-suggestions.md) : 3 New AI-Enhanced Feature Suggestions for Grokkit
* [MODEL_ENFORCEMENT.md](./completed/MODEL_ENFORCEMENT.md) : Model Enforcement
* [scaffold.md](./completed/scaffold.md) : grokkit scaffold
* [testgen.md](./completed/testgen.md) : grokkit testgen [path...]

View File

@ -1,138 +0,0 @@
# `grokkit query` Simple Query Tool
## Priority: x of 13
**Description**: A "one shot" prompt/answer query tool. Use it to ask StackOverflow style questions, like "how do I do x? Or What library do I use for y?". The sweet spot is concise, actionable answers with just enough context, a code snippet if relevant, and no unnecessary fluff. Think: problem solved in 100-300 words, tops. However, a "wordy" flag will be included to allow for full "textbook" style answers, as well.
## Problem It Solves
Reduces the amount of context switching needed to answer basic programming questions.
### Example Short Query 1:
```
grokkit query "What is the difference between a slice and a map in Go?"
Slices are dynamic arrays for ordered sequences of elements (e.g., `[]int`), with length/capacity and fast index access. Use for lists or queues; append with `append(s, val)`.
Maps are unordered key-value stores (e.g., `map[string]int`), for fast lookups/deletes via hashing. Use for dictionaries or caches; access with `m[key]`.
Key diffs: Slices are indexed by integers and ordered; maps use arbitrary keys and are unordered. Both are reference types, initialized with `make`.
Example:
```go
s := []int{1, 2, 3} // Slice
m := map[string]int{"a": 1} // Map
```
```
### Example Short Query 2:
```
"$ grokkit query "Are there any other languages with a concept like Python list comprehensions?"
Yes, several languages have similar concise syntax for creating collections via expressions:
- **Haskell**: List comprehensions, e.g., `[x*2 | x <- [1..5]]` (outputs `[2,4,6,8,10]`).
- **Scala**: For-comprehensions, e.g., `for (x <- 1 to 5) yield x*2`.
- **JavaScript (ES6+)**: Array methods like `map`/`filter`, e.g., `[1,2,3,4,5].map(x => x*2)`. Not identical, but functional equivalent.
- **C#**: LINQ queries, e.g., `from x in Enumerable.Range(1,5) select x*2`.
- **Rust**: Iterators with `map`/`collect`, e.g., `(1..=5).map(|x| x*2).collect::<Vec<i32>>()`.
Python's are inspired by Haskell. For exact matches, Haskell is closest.
```
### Example Long Query:
```
$ grokkit query "What is the difference between a slice and a map in Go?" --wordy
Ah, slices and maps in Go—two of the language's built-in data structures that make life easier (or at least less array-tastic) for developers. They're both reference types, meaning they're lightweight and point to underlying data, but they serve very different purposes. I'll break it down step by step, with some examples, and keep it straightforward. If Go had a sense of humor, it'd say slices are for cutting things up neatly, while maps are for when you lose your keys... again.
### What is a Slice?
A slice is essentially a dynamic, resizable view of an array. Think of it as a flexible array that can grow or shrink without you having to manage the underlying memory manually. Slices are backed by an array under the hood, but they're more convenient because:
- They have a length (how many elements are currently in it) and a capacity (how much room is allocated before it needs to resize).
- You can append to them, slice them further (hence the name), and they're zero-initialized by default.
- They're ordered: Elements are accessed by integer indices, starting from 0.
**Key traits:**
- Type: `[]T` (e.g., `[]int` for a slice of integers).
- Ordered? Yes.
- Use case: Lists, stacks, queues, or any sequence of items where order matters and you need fast random access.
**Example:**
```go
package main
import "fmt"
func main() {
// Create a slice with make (initial length 3, capacity 5)
s := make([]string, 3, 5)
s[0] = "apple"
s[1] = "banana"
s[2] = "cherry"
// Append to it (grows if needed)
s = append(s, "date")
fmt.Println(s) // Output: [apple banana cherry date]
fmt.Println(s[1:3]) // Slicing: [banana cherry]
}
```
Slices are great for when you want array-like behavior without the fixed-size hassle of actual arrays in Go.
### What is a Map?
A map is Go's version of a hash table or dictionary: an unordered collection of key-value pairs. You use it to associate unique keys with values, and it's super efficient for lookups, inserts, and deletes (average O(1) time complexity, thanks to hashing).
- Keys must be comparable (e.g., strings, ints, structs without pointers), but values can be anything.
- Maps are unordered, so don't rely on iteration order—it's not guaranteed (though as of Go 1.12, it's pseudo-random but consistent per run for debugging).
- They're also nil-safe: An uninitialized map is nil, and you can't add to it until you `make` it.
**Key traits:**
- Type: `map[K]V` (e.g., `map[string]int` for string keys and int values).
- Ordered? No (unordered by design).
- Use case: Lookup tables, caches, configurations, or anywhere you need fast key-based access without caring about order.
**Example:**
```go
package main
import "fmt"
func main() {
// Create a map with make
m := make(map[string]int)
m["apple"] = 1
m["banana"] = 2
m["cherry"] = 3
// Access by key
fmt.Println(m["banana"]) // Output: 2
// Check if a key exists
value, exists := m["date"]
if !exists {
fmt.Println("Date not found!") // This will print
}
// Delete a key
delete(m, "cherry")
}
```
Maps are your go-to when you need to map one thing to another without the linear search nonsense.
### Key Differences
- **Structure and Purpose**: Slices are sequences of elements (like a list or array), ideal for ordered collections. Maps are associative arrays for key-value storage, perfect for quick lookups by key.
- **Access Method**: Slices use integer indices (e.g., `s[42]`). Maps use keys of any hashable type (e.g., `m["foo"]`).
- **Ordering**: Slices preserve insertion order and allow indexing. Maps are unordered—iteration might give you keys in a seemingly random order.
- **Resizing and Capacity**: Slices have explicit length and capacity; they can grow via `append`. Maps grow automatically as you add elements, but you don't control capacity directly.
- **Initialization**: Both often use `make` for creation, but slices can be literals like `[]int{1, 2, 3}`, while maps use `map[string]int{"a": 1}`.
- **Performance**: Slices are contiguous in memory (fast iteration and access), maps use hashing (fast lookups but more overhead for small sizes).
- **Nil Behavior**: A nil slice can still be appended to (it acts like an empty slice). A nil map will panic if you try to read/write to it.
- **When to Use Which**: Use a slice if you need a list (e.g., processing a queue of tasks). Use a map if you need to associate data (e.g., user IDs to profiles).
In short, if slices are like a neatly organized toolbox drawer, maps are like a junk drawer where you can find stuff quickly by yelling its name—but don't expect it to be in any particular spot.
```
## ROI
**Medium**

View File

@ -1,87 +0,0 @@
# `grokkit scaffold`
## Priority: 1 of 12
**Description**: AI-powered code generation that creates new files from natural language descriptions, learning from your existing codebase patterns to produce idiomatic, ready-to-compile output.
## Problem It Solves
Every existing grokkit command operates on *existing* files (`edit`, `agent`, `lint`, `docs`, `testgen`). There is no command to *create* a new file from scratch. Developers spend significant time writing boilerplate for new handlers, services, models, CLI commands, middleware, etc. `scaffold` fills this gap.
## Benefits
- **Pattern-aware**: Reads similar existing files for package names, import styles, error handling conventions, struct naming, and logger usage before generating.
- **Idiomatic output**: Produces code that looks like it was written by the same developer — same patterns as the rest of the codebase, not generic AI output.
- **Multi-language**: Works for any language the project already supports (Go, Python, JS/TS, etc.), with lang detected from the output path or `--lang` flag.
- **Optional test companion**: With `--with-tests`, also generates a corresponding test file using the same testgen prompt logic.
- **Safe preview**: Shows a diff/preview before writing, requires confirmation (or `--yes` to skip).
- **No overwrites**: Refuses to overwrite an existing file unless `--force` is passed.
## CLI Examples
```bash
# Generate a new Cobra CLI command file
grokkit scaffold cmd/export.go "export command that writes chat history to JSON or CSV"
# New internal package
grokkit scaffold internal/cache/cache.go "simple in-memory LRU cache with TTL support"
# Generate with a companion test file
grokkit scaffold internal/git/tag.go "functions to list and create git tags" --with-tests
# Python module
grokkit scaffold src/auth/jwt.py "JWT encode/decode utilities using PyJWT"
# Skip confirmation
grokkit scaffold cmd/webhook.go "webhook command that posts PR descriptions to Slack" --yes
# Preview only (no write)
grokkit scaffold cmd/watch.go "file watcher that reruns lint on save" --dry-run
```
## High-Level Implementation
1. **Input validation**: Check target path doesn't already exist (error unless `--force`). Detect language from file extension.
2. **Context harvesting**: Walk sibling files in the same directory and the most structurally similar files in the project (e.g., other `cmd/*.go` if generating a command). Read them for patterns.
3. **Prompt construction**:
- System: "You are an expert Go/Python/etc. engineer. Given existing code samples, generate a new file that matches the codebase's conventions exactly. Return only the file content — no explanations, no markdown."
- User: paste harvested sibling files as context + target path + description.
4. **Streaming response**: Use `client.StreamSilent` (same as `edit`/`testgen`).
5. **Clean output**: Run through `grok.CleanCodeResponse` to strip any markdown fences.
6. **Preview**: Print file content with a header showing the target path. Prompt `(y/n)`.
7. **Write**: Create parent directories if needed (`os.MkdirAll`), write file.
8. **Optional test generation**: If `--with-tests`, reuse testgen logic to generate companion test file with the same preview/confirm flow.
## Flags
| Flag | Description |
|---|---|
| `--with-tests` | Also generate companion test file |
| `--yes` / `-y` | Skip confirmation prompt |
| `--dry-run` | Preview only, do not write |
| `--force` | Overwrite if target file already exists |
| `--lang` | Override language detection (go, python, js, ts, rust, ruby) |
| `--model` / `-m` | Override model |
| `--context-files` | Comma-separated paths to use as context (overrides auto-detection) |
## Implementation Notes
- **Context size**: Cap harvested context at ~4000 tokens to avoid prompt bloat. Prefer the most relevant files (same directory, same `package` name for Go).
- **Go specifics**: Extract `package` name from sibling files and inject it at the top of the prompt so the AI uses the correct package declaration.
- **Reuse existing patterns**: Share `internal/linter.DetectLanguage`, `grok.CleanCodeResponse`, and the standard preview/confirm UX from `edit.go`.
- **Effort**: Medium (~250350 LOC). Most complexity is in the context-harvesting logic.
## ROI
**High**. This closes the only remaining gap in grokkit's file lifecycle coverage:
| Lifecycle stage | Command |
|---|---|
| Create new file | **scaffold** ← missing |
| Edit existing file | `edit` |
| Multi-file changes | `agent` |
| Review changes | `review` |
| Document code | `docs` |
| Generate tests | `testgen` |
| Commit | `commit` |
Developers scaffold new files multiple times per day. Making that AI-assisted and pattern-aware is a force multiplier for the entire tool.

View File

@ -1,19 +0,0 @@
# `grokkit admin` administration tools
### Category: Nice to Have
Add an `admin` command that, if an `XAI_MGMT_KEY` is set in the environment, displays up-to-date token usage on the api account/team.
### Details
- **Command**: `grokkit admin`
- **Subcommands**:
- `grokkit admin token-usage` <-- default (shows token usage for current team)
- `grokkit admin credit-balance` <-- shows credit balance for current team
- `grokkit admin models` <-- shows available models
- `grokkit admin api-keys` <-- shows available API keys
- **Condition**: Must check for `XAI_MGMT_KEY` env var.
- **Functionality**: Fetch and display token usage stats from the [XAI MANAGEMENT API](https://docs.x.ai/developers/rest-api-reference/management.
- **Goal**: Help me monitor my API costs and limits directly from the CLI.
NOTE: If possible, it would be cool if this command was "hidden" if the XAI_MGMT_KEY is not set.

View File

@ -1,26 +0,0 @@
# `grokkit audit`
**Description**: Comprehensive AI-powered code audit for security, performance, best practices, and potential bugs across single files or entire projects.
**Benefits**:
- Deep analysis beyond static linters: vulns (e.g., SQLi, race cond), perf hotspots, Go idioms violations.
- Generates actionable report + diff previews for fixes.
- Multi-language support (reuse testgen/lint patterns).
- Boosts code quality/PR readiness.
**High-level implementation**:
- Detect lang/files (internal/linter), collect code snippets/context.
- Prompt Grok: "Audit for security, perf, best practices, bugs. List issues prioritized + suggested code fixes."
- Output: Markdown report (sections: Critical/High/Med/Low), optional `--fix` generates edit previews.
- Reuse `edit` preview/apply workflow; add `--apply` flag.
**CLI example**:
```
grokkit audit main.go # Single file report
grokkit audit ./cmd --lang=go # Dir, lang filter
grokkit audit . --fix # Report + fix previews
grokkit audit . --fix --yes # Auto-apply fixes (dangerous, preview first)
```
**Similar to**: lint/review but deeper, proactive fixes.
**Effort**: Medium (prompt tuning, multi-file handling ~300 LOC).
**ROI**: High - daily dev essential, esp. before releases.

View File

@ -1,6 +1,5 @@
# `grokkit changelog` # `grokkit changelog`
## Priority: 1 of 12 **Description**: AI-generated CHANGELOG.md updates from git history (commits/tags).
**Description**: AI-generated CHANGELOG.md updates from git history (commits/tags). The generated CHANGELOG.md should be used as the basis for the release notes for the release page of each release on gitea.
**Benefits**: **Benefits**:
- Automates semantic release notes (feat/fix/docs/etc.). - Automates semantic release notes (feat/fix/docs/etc.).

View File

@ -1,73 +0,0 @@
# `grokkit agent` cnotes integration
**Description**: Wrappers for your `cnotes` CLI logging suite. Allows Grok to automatically log coding sessions, notes, progress during agent workflows (e.g., "start work on feature", "log bug found").
## Problem It Solves
Developers track time/effort manually. Integrate `cnotes` (cnadd/cndump/etc.) for AI-assisted logging: timestamps, descriptions, stats—seamless with code edits.
## Benefits
- **Automated logging**: Grok logs at key points (start/edit/commit).
- **Queryable**: Search/dump stats via AI ("Show time on agent tasks today").
- **Safe**: Append-only (`cnadd`), read-only queries (`cnfind`/`cncount`); no deletes.
- **Terminal-native**: Fast C impl, no deps.
- **Workflow boost**: "Refactor → log changes → stats".
## Agent Tool Examples
```
grokkit agent "Start logging session, fix main.go, log completion"
# Grok: cnadd "start: agent refactor main.go" → edits → cnadd "done: fixed errors"
```
```
grokkit agent "Log today's coding stats before planning"
# Grok: cncount → cnfind "today" → "2h on edits, 3 sessions"
```
## High-Level Implementation
1. **Detect cnotes**: Run `cnhelp` or `which cnotes`.
2. **Tool schemas**:
- `log_note(message: string) → ok`
- `list_notes(filter?: string) → string`
- `session_stats() → string`
3. **Wrappers** in `internal/tools/cnotes.go`:
```go
func LogNote(ctx context.Context, args map[string]any) (string, error)
```
4. **Agent integration**: Auto-log on milestones; user-prompted.
5. **Safety**:
- Only `cnadd`/`cndump`/`cnfind`/`cncount`.
- Preview log message.
- Config: `[tools.cnotes.enabled]`.
6. **Parsing**: Stdout JSON? Or simple text.
## Flags / Config
| Key | Description |
|-----|-------------|
| `tools.cnotes.enabled` | Enable cnotes tools |
| `--auto-log` | Log start/end automatically |
## Implementation Notes
- **Commands**: cnadd (log), cndump (all), cnfind (search), cncount (stats), cndel (archive—whitelist carefully).
- **Extend agent**: Tool calls in loop.
- **Errors**: Custom CnotesError.
- **Tests**: Mock `exec.Command`, table-driven.
- **Effort**: Low (~150 LOC). Simple exec wrappers.
- **Prereq**: cnotes in PATH (`cnhelp` verifies).
## ROI
**High**. Personal logging elevates from code tool to **dev productivity hub**:
| Stage | Covered |
|-------|---------|
| Logging | **cnotes** ← new |
| Code | `agent` ✓ |
| Repo | tea (sibling) |
Quantifiable: Track hours/features automatically.

View File

@ -1,61 +0,0 @@
# `grokkit agent` git-chglog integration
**Description**: Wrapper for git-chglog CLI: generates CHANGELOG.md from conventional commits/tags. AI-enhanced parsing/validation.
## Problem It Solves
changelog.md is AI-from-git-log; git-chglog is standard tool—wrapper adds agent control/previews.
## Benefits
- **Standard**: git-chglog config + AI tweaks.
- **Validate**: Check changelog before release.
- **Safe**: Dry-run, output preview.
- **Integrate**: "Generate changelog v0.2.0, commit".
## Agent Tool Examples
```
grokkit agent "Generate changelog for next release from commits"
# Grok: git-chglog --next-tag v0.2.0 → preview → append to CHANGELOG.md
```
## High-Level Implementation
1. **Detect**: `git-chglog --version`.
2. **Tool schemas**:
- `generate(tag: string) → changelog_md`
- `validate() → issues`
3. **Wrappers** in `internal/tools/git-chglog.go`:
```go
func Generate(ctx context.Context, args map[string]any) (string, error)
```
4. **Agent integration**: Release workflows.
5. **Safety**:
- --dry-run flag.
- Config file respect.
- Config: `[tools.git-chglog.enabled]`.
## Flags / Config
| Key | Description |
|-----|-------------|
| `tools.git-chglog.enabled` | Enable git-chglog |
## Implementation Notes
- **Commands**: `git-chglog --output CHANGELOG.md --next-tag vX.Y.Z`.
- **Parsing**: Markdown output.
- **Errors**: ChglogError.
- **Tests**: Mock exec.
- **Effort**: Low (~110 LOC). Complements changelog.md.
- **Prereq**: git-chglog installed.
## ROI
**Medium-High**. Release polish:
| Stage | Covered |
|-------|---------|
| Changelog | **git-chglog** ← new |
| Release | changelog.md ✓ |

View File

@ -1,64 +0,0 @@
# `grokkit agent` Go tools integration
**Description**: Wrappers for `go` subcommands: mod tidy, generate, vet, fmt. Ensures hygiene post-agent edits.
## Problem It Solves
Agent changes may break deps/fmt/vet—manual fixes. Auto-run + fix loops.
## Benefits
- **Hygiene auto**: Tidy deps, fmt code, vet issues.
- **Generate**: Run go:generate for boilerplate.
- **Safe mutations**: Preview changes (git diff).
- **Workflow**: Edit → gotools tidy → test.
## Agent Tool Examples
```
grokkit agent "Add new import, tidy mods, vet all"
# Grok: edits → go mod tidy → go vet → reports clean
```
## High-Level Implementation
1. **Detect**: `go version`.
2. **Tool schemas**:
- `tidy() → changes`
- `vet(path?) → issues`
- `generate() → output`
- `fmt_diff() → diff`
3. **Wrappers** in `internal/tools/gotools.go`:
```go
func Tidy(ctx context.Context, args map[string]any) (string, error)
```
4. **Agent integration**: Post-edit hygiene step.
5. **Safety**:
- Dry-run where possible (go fmt -d).
- Whitelist subcmds.
- Config: `[tools.gotools.enabled]`.
## Flags / Config
| Key | Description |
|-----|-------------|
| `tools.gotools.enabled` | Enable Go tools |
## Implementation Notes
- **Commands**: `go mod tidy`, `go vet ./...`, `go generate ./...`.
- **Parsing**: Diff output, error lists.
- **Errors**: GoToolError.
- **Tests**: Mock, table-driven.
- **Effort**: Low (~130 LOC).
- **Prereq**: Go workspace.
## ROI
**High**. Go-specific polish:
| Stage | Covered |
|-------|---------|
| Hygiene | **gotools** ← new |
| Search | rg (sibling) |
| Build | make (sibling) |

View File

@ -1,53 +0,0 @@
# Grokkit Interactive Agent
## Goal
Add a persistent, conversational **agent mode** to Grokkit so I can chat interactively in the terminal and have Grok directly edit code, run tests, commit, etc., without any copy-paste friction.
This is the natural evolution of the current `chat` and `agent` commands — turning the terminal into a true agentic partner instead of a one-shot prompt tool.
## Command
```bash
grokkit chat --agent # or new subcommand: grokkit agent-chat
```
(Keep the existing simple `chat` mode untouched; add an `--agent` flag or a dedicated `agent-chat` command.)
## Core Features
- Persistent session using the existing `chat_history.json` (with optional `--new-session` flag)
- Full tool-calling loop: I (Grok) can invoke any existing Grokkit primitive:
- `edit`
- `scaffold`
- `testgen`
- `lint`
- `review`
- `commit` / `commit-msg`
- `docs`
- etc.
- Safe-by-default workflow:
- Always preview changes (diff style, same as current `edit`/`scaffold`)
- Require explicit confirmation (`y/n` or `--yes`)
- Preview and confirm all changes before application
- Ability to chain actions naturally in conversation:
- “The scaffold tests are flaky on Windows paths — fix it and add a test case.”
- “Refactor the context harvester to use rg, then run make test.”
- “Write a new command for todo prioritization and commit it.”
- Optional `--dry-run` and `--model` flags (inherited from root)
## Acceptance Criteria
- `make test` remains fast (no new API cost in normal tests)
- Live agent mode respects the same `-short` gating pattern we just built for scaffold
- Session survives across multiple `grokkit` invocations (or can be started fresh)
- All edits go through the existing safe preview/confirm flow — no silent changes
- Works with the full Grok context window (no early compaction)
- Can be extended later with vector memory or external tool plugins
## Why This Matters
This removes the last remaining copy-paste friction between me and the codebase. The 45-minute scaffold test iteration we just did showed the productivity gain; this turns that gain into a permanent workflow. Closes the “agentic chat” vision we discussed.

View File

@ -1,73 +0,0 @@
# `grokkit agent` make integration
**Description**: Wrappers for Makefile targets (test/lint/build/cover). Enables Grok to run/verify builds mid-agent workflow (e.g., "edit, test, fix loops").
## Problem It Solves
Agent edits code but can't auto-verify compilation/tests—manual `make test` context-switch.
## Benefits
- **Automated verification**: Post-edit `make test` + analyze failures.
- **Dry-runs**: Preview `make build` output.
- **Safe**: Whitelisted targets, timeout, project-dir.
- **Parse results**: Extract pass/fail, coverage, errors for next agent step.
- **Workflow**: "Refactor → test → fix → commit".
## Agent Tool Examples
```
grokkit agent "Fix lint errors in cmd/, run make lint to verify, then test"
# Grok: edits → make lint → "All green!" → make test → fixes failures
```
```
grokkit agent "Benchmark changes before commit"
# Grok: make test-cover → "Coverage drop 2%" → optimizations
```
## High-Level Implementation
1. **Detect**: `test -f Makefile` or `make --version`.
2. **Tool schemas**:
- `run_target(target: string) → {output: string, success: bool}`
- `dry_run(target: string) → simulated_output`
3. **Wrappers** in `internal/tools/make.go`:
```go
func RunTarget(ctx context.Context, args map[string]any) (string, error)
```
4. **Agent integration**: Tool call → parse stdout → feed to Grok ("Tests failed: fix?").
5. **Safety**:
- Whitelist: test, lint, build, test-cover, install.
- 300s timeout.
- No sudo/privileged.
- Config: `[tools.make.enabled]`.
6. **Parsing**: Grep for "PASS/FAIL", coverage %.
## Flags / Config
| Key | Description |
|-----|-------------|
| `tools.make.enabled` | Enable make tools |
| `tools.make.timeout` | Per-target timeout (s) |
## Implementation Notes
- **Commands**: `make TARGET` (no args).
- **Extend agent**: Loop includes make calls post-edit.
- **Errors**: MakeError with output.
- **Tests**: Mock exec, table-driven (success/fail outputs).
- **Effort**: Low (~120 LOC). Std exec + parsing.
- **Prereq**: Makefile present.
## ROI
**High**. Agent verification loop:
| Stage | Covered |
|-------|---------|
| Edit | `agent` ✓ |
| Verify | **make** ← new |
| Repo/log | tea/cnotes |
Instant feedback elevates agent reliability.

View File

@ -1,62 +0,0 @@
# `grokkit agent` pprof integration
**Description**: Go pprof profiling wrappers (CPU/memory/allocs). Captures profiles during agent runs, AI-analyzes hotspots.
## Problem It Solves
Profile.md is high-level; pprof enables raw data + AI opts for slow cmds (agent loops).
## Benefits
- **Capture**: `go test -cpuprofile` or runtime/pprof.
- **Analyze**: Feed pprof text/top to Grok ("Optimize this flamegraph").
- **Safe**: Temp .pprof files, no web UI.
- **Integrate**: Auto-profile long agents.
## Agent Tool Examples
```
grokkit agent "Profile cmd/agent.go benchmark, suggest fixes"
# Grok: pprof cpu → "30% in StreamSilent, batch calls"
```
## High-Level Implementation
1. **Detect**: `go tool pprof --version`.
2. **Tool schemas**:
- `cpu_profile(duration: int) → pprof_text`
- `analyze(profile_data: string) → suggestions`
3. **Wrappers** in `internal/tools/pprof.go`:
```go
func CPUProfile(ctx context.Context, args map[string]any) (string, error)
```
4. **Agent integration**: Flag-triggered profiling.
5. **Safety**:
- Tempdir files (os.TempDir).
- No --http/--pdf.
- Config: `[tools.pprof.enabled]`.
## Flags / Config
| Key | Description |
|-----|-------------|
| `tools.pprof.enabled` | Enable pprof tools |
| `--profile-cpu` | Auto-cpu on agent |
## Implementation Notes
- **Capture**: runtime/pprof.StartCPUProfile, or `go test -cpuprofile`.
- **Parse**: `go tool pprof -text`.
- **Errors**: PprofError.
- **Tests**: Mock profiles.
- **Effort**: Medium (~180 LOC). Ties to profile.md.
- **Prereq**: Go stdlib pprof.
## ROI
**High** for perf:
| Stage | Covered |
|-------|---------|
| Profile | **pprof** ← new |
| Analyze | profile.md ✓ |

View File

@ -1,72 +0,0 @@
# `grokkit agent` ripgrep (rg) integration
**Description**: Fast search wrapper for ripgrep (`rg`). Enables agent to grep project for symbols/patterns/context before edits.
## Problem It Solves
Agent lacks quick codebase search—relies on full file scans or manual `grep`. rg is 10x faster, regex-aware.
## Benefits
- **Lightning search**: Multi-line, git-ignored, type-aware (`--type=go`).
- **Context gathering**: "rg 'err handling' → snippets for prompt".
- **Safe**: Read-only, project-only, no filespecs.
- **Stats**: Count matches, paths.
- **Workflow**: "Search todos → prioritize → edit".
## Agent Tool Examples
```
grokkit agent "Find all error returns in cmd/, improve handling"
# Grok: rg 'return err' cmd/ → analyzes → targeted edits
```
```
grokkit agent "Count test coverage gaps: rg 't.Skip' or untested funcs"
# Grok: rg --type=go '_test\.go' → stats → testgen suggestions
```
## High-Level Implementation
1. **Detect**: `rg --version`.
2. **Tool schemas**:
- `search(pattern: string, path?: string) → matches[]`
- `count(pattern: string) → int`
3. **Wrappers** in `internal/tools/rg.go`:
```go
func Search(ctx context.Context, args map[string]any) (string, error)
```
4. **Agent integration**: Pre-plan search → richer context.
5. **Safety**:
- Args whitelist: --type, --no-heading, project root.
- Max 100 matches.
- Config: `[tools.rg.enabled]`.
6. **Parsing**: JSON output (`--json`), or lines → structured.
## Flags / Config
| Key | Description |
|-----|-------------|
| `tools.rg.enabled` | Enable rg tools |
| `tools.rg.max_matches` | Limit results |
## Implementation Notes
- **Commands**: `rg PATTERN [PATH] --no-heading --colors=never`.
- **Extend agent**: Optional search phase in planning.
- **Errors**: RgError.
- **Tests**: Mock, table-driven patterns.
- **Effort**: Low (~100 LOC).
- **Prereq**: rg installed (`apt install ripgrep`).
## ROI
**High**. Search is dev superpower:
| Stage | Covered |
|-------|---------|
| Search | **rg** ← new |
| Edit | `agent` ✓ |
| Verify | make (sibling) |
Agent plans smarter with instant intel.

View File

@ -1,74 +0,0 @@
# `grokkit agent` tea integration
**Description**: Safe, AI-orchestrated wrappers for Gitea `tea` CLI commands. Enables Grok to manage repos (list/create PRs/issues, comments) as part of agent workflows, with previews and confirmations.
## Problem It Solves
Grokkit's `agent` excels at code edits but lacks repo mgmt. Developers context-switch to browser/CLI for Gitea actions. Integrate `tea` for seamless: "Review PRs, create issue for bugs, log progress".
## Benefits
- **Safe execution**: Whitelisted commands/args, dry-run previews, confirm destructive ops (merge/push).
- **AI reasoning**: Grok decides *when* to call `tea` (e.g., "After edits, create PR").
- **Logging**: All calls logged to `~/.config/grokkit/tools.log`.
- **Repo-aware**: Combines with git (`internal/git`), code edits.
- **No new deps**: Uses `os/exec` for `tea` (assume installed).
## Agent Tool Examples
```
grokkit agent "List open PRs, pick one to work on, create branch"
# Grok: tea pr ls → Analyzes → tea branch create fix-123 → edits files
```
```
grokkit agent "Fix lint, create issue if high-risk changes"
# Grok: Runs lint → tea issue create "High-risk refactor"
```
## High-Level Implementation
1. **Detect tea**: Check `exec.Command("tea", "version")`.
2. **Tool schemas** (Grok tool-calling JSON):
- `list_prs(state: "open") → string`
- `create_pr(title, body, base) → pr_url`
- `comment_pr(pr_id, body)`
- `merge_pr(pr_id)`
3. **Wrapper funcs** in `internal/tools/tea.go`:
```go
func ListPRs(ctx context.Context, args map[string]any) (string, error)
```
4. **Agent loop** (`cmd/agent.go`): Parse tool calls → Execute → Feed output back.
5. **Safety**:
- Whitelist: pr ls/create/comment/merge, issue create/comment.
- Preview: Print `tea ...` + expected output.
- Confirm: User y/n for each.
6. **Config**: `[tools.tea.enabled]`, `[tools.tea.gitea_url]`.
## Flags / Config
| Key | Description |
|-----|-------------|
| `tools.tea.enabled` | Enable tea tools |
| `tools.tea.gitea_url` | Override repo URL |
| `--dry-run` | Simulate tool calls |
## Implementation Notes
- **Extend agent**: Modify plan phase to include tool suggestions.
- **Error handling**: `internal/errors` + GitError-like TeaError.
- **Tests**: Table-driven `TestTeaTools` with mocks (`exec.Command` patching).
- **Effort**: Medium (~200 LOC). Reuse `grok` client, agent UX.
- **Prereq**: User installs `tea` (`go install github.com/go-tea/tea/cmd/tea@latest`).
## ROI
**High**. Closes workflow gap:
| Stage | Covered |
|-------|---------|
| Code edit | `agent` ✓ |
| Repo mgmt | **tea tools** ← new |
| Logging | cnotes (next) |
Force-multiplies daily PR/issue workflows.