Skip to content

Commit 945eac1

Browse files
committed
fix: gofumpt formatting + go mod tidy
1 parent fe3b78f commit 945eac1

148 files changed

Lines changed: 1130 additions & 733 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

.github/dependabot.yml

Lines changed: 0 additions & 27 deletions
This file was deleted.

cmd/trace/cli/activity_render.go

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -454,8 +454,14 @@ func renderBrailleChart(w io.Writer, sty activityStyles, hourly []hourlyPoint, r
454454
// dot3 dot6 bit2 bit5
455455
// dot7 dot8 bit6 bit7
456456
dotBits := [8][2]int{
457-
{0, 0}, {1, 0}, {2, 0}, {0, 1},
458-
{1, 1}, {2, 1}, {3, 0}, {3, 1},
457+
{0, 0},
458+
{1, 0},
459+
{2, 0},
460+
{0, 1},
461+
{1, 1},
462+
{2, 1},
463+
{3, 0},
464+
{3, 1},
459465
}
460466

461467
for bit, offset := range dotBits {

cmd/trace/cli/agent/agent_test.go

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,10 @@ import (
88
"github.com/GrayCodeAI/trace/cmd/trace/cli/agent/types"
99
)
1010

11-
const mockAgentName types.AgentName = "mock" // Used by mock implementations
12-
const mockAgentType types.AgentType = "Mock Agent"
11+
const (
12+
mockAgentName types.AgentName = "mock" // Used by mock implementations
13+
mockAgentType types.AgentType = "Mock Agent"
14+
)
1315

1416
// mockAgent is a minimal implementation of Agent for testing interface compliance.
1517
type mockAgent struct{}
@@ -29,6 +31,7 @@ func (m *mockAgent) ReadTranscript(_ string) ([]byte, error) { return nil, nil }
2931
func (m *mockAgent) ChunkTranscript(_ context.Context, content []byte, _ int) ([][]byte, error) {
3032
return [][]byte{content}, nil
3133
}
34+
3235
func (m *mockAgent) ReassembleTranscript(chunks [][]byte) ([]byte, error) {
3336
var result []byte
3437
for _, c := range chunks {

cmd/trace/cli/agent/capabilities_test.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ func (m *mockBuiltinHookAgent) HookNames() []string { return nil }
4040
func (m *mockBuiltinHookAgent) ParseHookEvent(context.Context, string, io.Reader) (*Event, error) {
4141
return nil, nil //nolint:nilnil // test mock
4242
}
43+
4344
func (m *mockBuiltinHookAgent) InstallHooks(context.Context, bool, bool) (int, error) {
4445
return 0, nil
4546
}
@@ -57,6 +58,7 @@ func (m *mockFullAgent) DeclaredCapabilities() DeclaredCaps { return m.caps }
5758

5859
// HookSupport
5960
func (m *mockFullAgent) HookNames() []string { return nil }
61+
6062
func (m *mockFullAgent) ParseHookEvent(context.Context, string, io.Reader) (*Event, error) {
6163
return nil, nil //nolint:nilnil // test mock
6264
}
@@ -66,6 +68,7 @@ func (m *mockFullAgent) AreHooksInstalled(context.Context) bool {
6668

6769
// TranscriptAnalyzer
6870
func (m *mockFullAgent) GetTranscriptPosition(string) (int, error) { return 0, nil }
71+
6972
func (m *mockFullAgent) ExtractModifiedFilesFromOffset(string, int) ([]string, int, error) {
7073
return nil, 0, nil
7174
}
@@ -95,6 +98,7 @@ func (m *mockFullAgent) WriteHookResponse(string) error { return nil }
9598
func (m *mockFullAgent) ExtractAllModifiedFiles([]byte, int, string) ([]string, error) {
9699
return nil, nil
97100
}
101+
98102
func (m *mockFullAgent) CalculateTotalTokenUsage([]byte, int, string) (*TokenUsage, error) {
99103
return nil, nil //nolint:nilnil // test mock
100104
}

cmd/trace/cli/agent/claudecode/lifecycle.go

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,8 @@ func waitForTranscriptFlush(ctx context.Context, transcriptPath string, hookStar
218218
}
219219
fileAge := time.Since(info.ModTime())
220220
if fileAge > staleThreshold {
221-
logging.Debug(logCtx, "transcript file is stale, skipping sentinel wait",
221+
logging.Debug(
222+
logCtx, "transcript file is stale, skipping sentinel wait",
222223
slog.Duration("file_age", fileAge),
223224
)
224225
return
@@ -227,14 +228,16 @@ func waitForTranscriptFlush(ctx context.Context, transcriptPath string, hookStar
227228
deadline := time.Now().Add(maxWait)
228229
for time.Now().Before(deadline) {
229230
if checkStopSentinel(transcriptPath, tailBytes, hookStartTime, maxSkew) {
230-
logging.Debug(logCtx, "transcript flush sentinel found",
231+
logging.Debug(
232+
logCtx, "transcript flush sentinel found",
231233
slog.Duration("wait", time.Since(hookStartTime)),
232234
)
233235
return
234236
}
235237
time.Sleep(pollInterval)
236238
}
237-
logging.Warn(logCtx, "transcript flush sentinel not found within timeout, proceeding",
239+
logging.Warn(
240+
logCtx, "transcript flush sentinel not found within timeout, proceeding",
238241
slog.Duration("timeout", maxWait),
239242
)
240243
}

cmd/trace/cli/agent/claudecode/lifecycle_test.go

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ func TestParseHookEvent_SessionStart(t *testing.T) {
2020
input := `{"session_id": "test-session-123", "transcript_path": "/tmp/transcript.jsonl"}`
2121

2222
event, err := ag.ParseHookEvent(context.Background(), HookNameSessionStart, strings.NewReader(input))
23-
2423
if err != nil {
2524
t.Fatalf("unexpected error: %v", err)
2625
}
@@ -46,7 +45,6 @@ func TestParseHookEvent_SessionStart_IncludesModel(t *testing.T) {
4645
input := `{"session_id": "model-session", "transcript_path": "/tmp/t.jsonl", "model": "claude-sonnet-4-20250514"}`
4746

4847
event, err := ag.ParseHookEvent(context.Background(), HookNameSessionStart, strings.NewReader(input))
49-
5048
if err != nil {
5149
t.Fatalf("unexpected error: %v", err)
5250
}
@@ -66,7 +64,6 @@ func TestParseHookEvent_SessionStart_EmptyModel(t *testing.T) {
6664
input := `{"session_id": "no-model-session", "transcript_path": "/tmp/t.jsonl"}`
6765

6866
event, err := ag.ParseHookEvent(context.Background(), HookNameSessionStart, strings.NewReader(input))
69-
7067
if err != nil {
7168
t.Fatalf("unexpected error: %v", err)
7269
}
@@ -83,7 +80,6 @@ func TestParseHookEvent_TurnStart(t *testing.T) {
8380
input := `{"session_id": "sess-456", "transcript_path": "/tmp/t.jsonl", "prompt": "Hello world"}`
8481

8582
event, err := ag.ParseHookEvent(context.Background(), HookNameUserPromptSubmit, strings.NewReader(input))
86-
8783
if err != nil {
8884
t.Fatalf("unexpected error: %v", err)
8985
}
@@ -106,7 +102,6 @@ func TestParseHookEvent_TurnEnd(t *testing.T) {
106102
input := `{"session_id": "sess-789", "transcript_path": "/tmp/stop.jsonl"}`
107103

108104
event, err := ag.ParseHookEvent(context.Background(), HookNameStop, strings.NewReader(input))
109-
110105
if err != nil {
111106
t.Fatalf("unexpected error: %v", err)
112107
}
@@ -126,7 +121,6 @@ func TestParseHookEvent_TurnEnd_IncludesModel(t *testing.T) {
126121
input := `{"session_id": "sess-stop-model", "transcript_path": "/tmp/stop.jsonl", "model": "claude-opus-4-6"}`
127122

128123
event, err := ag.ParseHookEvent(context.Background(), HookNameStop, strings.NewReader(input))
129-
130124
if err != nil {
131125
t.Fatalf("unexpected error: %v", err)
132126
}
@@ -143,7 +137,6 @@ func TestParseHookEvent_SessionEnd(t *testing.T) {
143137
input := `{"session_id": "ending-session", "transcript_path": "/tmp/end.jsonl"}`
144138

145139
event, err := ag.ParseHookEvent(context.Background(), HookNameSessionEnd, strings.NewReader(input))
146-
147140
if err != nil {
148141
t.Fatalf("unexpected error: %v", err)
149142
}
@@ -163,7 +156,6 @@ func TestParseHookEvent_SessionEnd_IncludesModel(t *testing.T) {
163156
input := `{"session_id": "end-model", "transcript_path": "/tmp/end.jsonl", "model": "claude-sonnet-4-20250514"}`
164157

165158
event, err := ag.ParseHookEvent(context.Background(), HookNameSessionEnd, strings.NewReader(input))
166-
167159
if err != nil {
168160
t.Fatalf("unexpected error: %v", err)
169161
}
@@ -190,7 +182,6 @@ func TestParseHookEvent_SubagentStart(t *testing.T) {
190182
}
191183

192184
event, err := ag.ParseHookEvent(context.Background(), HookNamePreTask, strings.NewReader(string(inputBytes)))
193-
194185
if err != nil {
195186
t.Fatalf("unexpected error: %v", err)
196187
}
@@ -228,7 +219,6 @@ func TestParseHookEvent_SubagentEnd(t *testing.T) {
228219
}
229220

230221
event, err := ag.ParseHookEvent(context.Background(), HookNamePostTask, strings.NewReader(string(inputBytes)))
231-
232222
if err != nil {
233223
t.Fatalf("unexpected error: %v", err)
234224
}
@@ -261,7 +251,6 @@ func TestParseHookEvent_SubagentEnd_NoAgentID(t *testing.T) {
261251
}
262252

263253
event, err := ag.ParseHookEvent(context.Background(), HookNamePostTask, strings.NewReader(string(inputBytes)))
264-
265254
if err != nil {
266255
t.Fatalf("unexpected error: %v", err)
267256
}
@@ -278,7 +267,6 @@ func TestParseHookEvent_PostTodo_ReturnsNil(t *testing.T) {
278267
input := `{"session_id": "todo-session", "transcript_path": "/tmp/todo.jsonl"}`
279268

280269
event, err := ag.ParseHookEvent(context.Background(), HookNamePostTodo, strings.NewReader(input))
281-
282270
if err != nil {
283271
t.Fatalf("unexpected error: %v", err)
284272
}
@@ -294,7 +282,6 @@ func TestParseHookEvent_UnknownHook_ReturnsNil(t *testing.T) {
294282
input := `{"session_id": "unknown", "transcript_path": "/tmp/unknown.jsonl"}`
295283

296284
event, err := ag.ParseHookEvent(context.Background(), "unknown-hook-name", strings.NewReader(input))
297-
298285
if err != nil {
299286
t.Fatalf("unexpected error: %v", err)
300287
}
@@ -386,7 +373,6 @@ func TestParseHookEvent_AllHookTypes(t *testing.T) {
386373

387374
ag := &ClaudeCodeAgent{}
388375
event, err := ag.ParseHookEvent(context.Background(), tc.hookName, strings.NewReader(tc.inputTemplate))
389-
390376
if err != nil {
391377
t.Fatalf("unexpected error: %v", err)
392378
}
@@ -412,7 +398,6 @@ func TestReadAndParse_ValidInput(t *testing.T) {
412398
input := `{"session_id": "test-123", "transcript_path": "/path/to/transcript"}`
413399

414400
result, err := agent.ReadAndParseHookInput[sessionInfoRaw](strings.NewReader(input))
415-
416401
if err != nil {
417402
t.Fatalf("unexpected error: %v", err)
418403
}
@@ -458,7 +443,6 @@ func TestReadAndParse_PartialJSON(t *testing.T) {
458443
input := `{"session_id": "partial-only"}`
459444

460445
result, err := agent.ReadAndParseHookInput[sessionInfoRaw](strings.NewReader(input))
461-
462446
if err != nil {
463447
t.Fatalf("unexpected error: %v", err)
464448
}
@@ -477,7 +461,6 @@ func TestReadAndParse_ExtraFields(t *testing.T) {
477461
input := `{"session_id": "test", "transcript_path": "/t", "extra_field": "ignored", "another": 123}`
478462

479463
result, err := agent.ReadAndParseHookInput[sessionInfoRaw](strings.NewReader(input))
480-
481464
if err != nil {
482465
t.Fatalf("unexpected error: %v", err)
483466
}

cmd/trace/cli/agent/claudecode/transcript_test.go

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -699,7 +699,8 @@ func TestExtractAllModifiedFiles_IncludesSubagentFiles(t *testing.T) {
699699
)
700700

701701
// Subagent transcript: Write to helper.go + Edit to utils.go (still on disk)
702-
writeJSONLFile(t, subagentsDir+"/agent-sub1.jsonl",
702+
writeJSONLFile(
703+
t, subagentsDir+"/agent-sub1.jsonl",
703704
makeWriteToolLine(t, "sa1", "/repo/helper.go"),
704705
makeEditToolLine(t, "sa2", "/repo/utils.go"),
705706
)
@@ -748,7 +749,8 @@ func TestExtractAllModifiedFiles_DeduplicatesAcrossAgents(t *testing.T) {
748749
)
749750

750751
// Subagent transcript: Also modifies shared.go (same file as main)
751-
writeJSONLFile(t, subagentsDir+"/agent-sub1.jsonl",
752+
writeJSONLFile(
753+
t, subagentsDir+"/agent-sub1.jsonl",
752754
makeEditToolLine(t, "sa1", "/repo/shared.go"),
753755
)
754756

@@ -809,7 +811,8 @@ func TestExtractAllModifiedFiles_SubagentOnlyChanges(t *testing.T) {
809811
)
810812

811813
// Subagent transcript: Write to two files (still on disk)
812-
writeJSONLFile(t, subagentsDir+"/agent-sub1.jsonl",
814+
writeJSONLFile(
815+
t, subagentsDir+"/agent-sub1.jsonl",
813816
makeWriteToolLine(t, "sa1", "/repo/subagent_file1.go"),
814817
makeWriteToolLine(t, "sa2", "/repo/subagent_file2.go"),
815818
)

cmd/trace/cli/agent/codex/codex_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,8 @@ func TestCodexAgent_ResolveRestoredSessionFile(t *testing.T) {
101101

102102
path, err := ag.ResolveRestoredSessionFile(dir, "019d24c3-1111-2222-3333-444444444444", []byte(sampleRollout))
103103
require.NoError(t, err)
104-
require.Equal(t,
104+
require.Equal(
105+
t,
105106
filepath.Join(dir, "2026", "03", "25", "rollout-2026-03-25T11-31-10-019d24c3-1111-2222-3333-444444444444.jsonl"),
106107
path,
107108
)

0 commit comments

Comments
 (0)