feat: add WAKATIME_AI_STATS metric (#27)

Author: thanhhaudev

.github/workflows/ci.yml

@@ -0,0 +1,61 @@
+name: CI
+
+on:
+  push:
+    branches: [master]
+  pull_request:
+    branches: [master]
+
+permissions:
+  contents: read
+
+jobs:
+  test:
+    name: Test
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-go@v5
+        with:
+          go-version: '1.24'
+          cache: true
+
+      - name: Check gofmt
+        run: |
+          unformatted=$(gofmt -l .)
+          if [ -n "$unformatted" ]; then
+            echo "The following files are not gofmt-ed:"
+            echo "$unformatted"
+            exit 1
+          fi
+
+      - name: Vet
+        run: go vet ./...
+
+      - name: Test
+        run: go test -race ./...
+
+  lint:
+    name: Lint
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-go@v5
+        with:
+          go-version: '1.24'
+          cache: true
+
+      - uses: golangci/golangci-lint-action@v7
+        with:
+          version: v2.1
+
+  docker:
+    name: Docker build
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Build image
+        run: docker build -t github-stats:ci .

.golangci.yml

@@ -0,0 +1,18 @@
+version: "2"
+
+run:
+  timeout: 5m
+
+linters:
+  default: none
+  enable:
+    - errcheck
+    - govet
+    - ineffassign
+    - staticcheck
+    - misspell
+  exclusions:
+    rules:
+      - path: _test\.go
+        linters:
+          - errcheck

README.md

@@ -192,6 +192,27 @@ Displays all languages you use with colorful badges showing percentages.
 ![Java](https://img.shields.io/badge/Java-12.0%25-b07219?&logo=Java&labelColor=151b23)
 ![Go](https://img.shields.io/badge/Go-2.8%25-00ADD8?&logo=Go&labelColor=151b23)
 
+### 🤖 `WAKATIME_AI_STATS`
+
+Shows AI vs human coding attribution from WakaTime, aggregated across all your projects in the configured range.
+
+**Requirements:**
+- `WAKATIME_API_KEY` (required) — WakaTime must be tracking AI attribution (requires GenAI integration in your editor)
+- `WAKATIME_RANGE` (optional) — same range as `WAKATIME_SPENT_TIME`
+
+**Example output:**
+
+**🤖 My AI Footprint**
+```
+🤖 Generated by AI:        12,340 lines
+👤 Written by Hand:        8,721 lines
+📊 AI Contribution:        58.6%
+🔤 Tokens In / Out:        1.2M / 3.4M
+💬 Average Prompt:         142 chars
+```
+
+> 💡 **Note:** The block is hidden entirely when no AI activity is reported (no GenAI integration, or zero AI usage in the range), so you won't see a section full of zeros. `Avg Prompt Length` is weighted by `ai_input_tokens` across projects.
+
 ### ⏱️ `WAKATIME_SPENT_TIME`
 
 Shows detailed coding activity from WakaTime (requires WakaTime API key).
@@ -301,7 +322,7 @@ env:
   WAKATIME_API_KEY: ${{ secrets.WAKATIME_API_KEY }}
   WAKATIME_DATA: "EDITORS,LANGUAGES,PROJECTS,OPERATING_SYSTEMS"
   WAKATIME_RANGE: "last_30_days"
-  SHOW_METRICS: "COMMIT_TIMES_OF_DAY,COMMIT_DAYS_OF_WEEK,LANGUAGE_PER_REPO,LANGUAGES_AND_TOOLS,WAKATIME_SPENT_TIME,CODING_STREAK"
+  SHOW_METRICS: "COMMIT_TIMES_OF_DAY,COMMIT_DAYS_OF_WEEK,LANGUAGE_PER_REPO,LANGUAGES_AND_TOOLS,WAKATIME_SPENT_TIME,CODING_STREAK,WAKATIME_AI_STATS"
   SHOW_LAST_UPDATE: "true"
   ONLY_MAIN_BRANCH: "true"
   PROGRESS_BAR_VERSION: "2"

cmd/git_test.go

@@ -8,99 +8,99 @@ import (
 
 func TestSanitizeError(t *testing.T) {
 	tests := []struct {
-		name          string
-		err           error
-		token         string
-		owner         string
-		expectedParts []string // Parts that should be in the result
+		name           string
+		err            error
+		token          string
+		owner          string
+		expectedParts  []string // Parts that should be in the result
 		forbiddenParts []string // Parts that should NOT be in the result
 	}{
 		{
-			name:          "nil error returns nil",
-			err:           nil,
-			token:         "ghp_1234567890",
-			owner:         "testuser",
-			expectedParts: nil,
+			name:           "nil error returns nil",
+			err:            nil,
+			token:          "ghp_1234567890",
+			owner:          "testuser",
+			expectedParts:  nil,
 			forbiddenParts: nil,
 		},
 		{
-			name:          "token is redacted",
-			err:           errors.New("authentication failed with token ghp_1234567890"),
-			token:         "ghp_1234567890",
-			owner:         "testuser",
-			expectedParts: []string{"[***]"},
+			name:           "token is redacted",
+			err:            errors.New("authentication failed with token ghp_1234567890"),
+			token:          "ghp_1234567890",
+			owner:          "testuser",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"ghp_1234567890"},
 		},
 		{
-			name:          "owner/username is redacted",
-			err:           errors.New("failed to push to testuser/repo"),
-			token:         "ghp_1234567890",
-			owner:         "testuser",
-			expectedParts: []string{"[***]"},
+			name:           "owner/username is redacted",
+			err:            errors.New("failed to push to testuser/repo"),
+			token:          "ghp_1234567890",
+			owner:          "testuser",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"testuser"},
 		},
 		{
-			name:          "https URL is completely redacted",
-			err:           errors.New("failed to clone https://github.com/user/repo.git"),
-			token:         "",
-			owner:         "",
-			expectedParts: []string{"[***]"},
+			name:           "https URL is completely redacted",
+			err:            errors.New("failed to clone https://github.com/user/repo.git"),
+			token:          "",
+			owner:          "",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"https://github.com", "github.com", "/user/repo.git"},
 		},
 		{
-			name:          "http URL is completely redacted",
-			err:           errors.New("failed to fetch http://example.com/path/to/resource"),
-			token:         "",
-			owner:         "",
-			expectedParts: []string{"[***]"},
+			name:           "http URL is completely redacted",
+			err:            errors.New("failed to fetch http://example.com/path/to/resource"),
+			token:          "",
+			owner:          "",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"http://example.com", "example.com", "/path/to/resource"},
 		},
 		{
-			name:          "multiple URLs are redacted",
-			err:           errors.New("failed: https://github.com/repo1 and https://github.com/repo2"),
-			token:         "",
-			owner:         "",
-			expectedParts: []string{"[***]"},
+			name:           "multiple URLs are redacted",
+			err:            errors.New("failed: https://github.com/repo1 and https://github.com/repo2"),
+			token:          "",
+			owner:          "",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"github.com", "repo1", "repo2"},
 		},
 		{
-			name:          "URL with token in it is redacted",
-			err:           errors.New("failed to push to https://ghp_token123@github.com/user/repo.git"),
-			token:         "ghp_token123",
-			owner:         "user",
-			expectedParts: []string{"[***]"},
+			name:           "URL with token in it is redacted",
+			err:            errors.New("failed to push to https://ghp_token123@github.com/user/repo.git"),
+			token:          "ghp_token123",
+			owner:          "user",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"ghp_token123", "github.com", "user", "repo.git"},
 		},
 		{
-			name:          "complex error with token, owner, and URL",
-			err:           errors.New("git remote set-url failed for testuser: https://ghp_secret@github.com/testuser/myrepo.git"),
-			token:         "ghp_secret",
-			owner:         "testuser",
-			expectedParts: []string{"[***]"},
+			name:           "complex error with token, owner, and URL",
+			err:            errors.New("git remote set-url failed for testuser: https://ghp_secret@github.com/testuser/myrepo.git"),
+			token:          "ghp_secret",
+			owner:          "testuser",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"ghp_secret", "testuser", "github.com", "myrepo"},
 		},
 		{
-			name:          "error without sensitive data remains unchanged",
-			err:           errors.New("connection timeout"),
-			token:         "ghp_1234",
-			owner:         "user",
-			expectedParts: []string{"connection timeout"},
+			name:           "error without sensitive data remains unchanged",
+			err:            errors.New("connection timeout"),
+			token:          "ghp_1234",
+			owner:          "user",
+			expectedParts:  []string{"connection timeout"},
 			forbiddenParts: nil,
 		},
 		{
-			name:          "URL with query parameters is fully redacted",
-			err:           errors.New("failed: https://api.github.com/repos/user/repo?token=abc123"),
-			token:         "",
-			owner:         "",
-			expectedParts: []string{"[***]"},
+			name:           "URL with query parameters is fully redacted",
+			err:            errors.New("failed: https://api.github.com/repos/user/repo?token=abc123"),
+			token:          "",
+			owner:          "",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"api.github.com", "token=abc123", "user", "repo"},
 		},
 		{
-			name:          "empty token and owner still redacts URLs",
-			err:           errors.New("error with https://github.com/path"),
-			token:         "",
-			owner:         "",
-			expectedParts: []string{"[***]"},
+			name:           "empty token and owner still redacts URLs",
+			err:            errors.New("error with https://github.com/path"),
+			token:          "",
+			owner:          "",
+			expectedParts:  []string{"[***]"},
 			forbiddenParts: []string{"github.com"},
 		},
 	}
@@ -141,9 +141,6 @@ func TestSanitizeError(t *testing.T) {
 	}
 }
 
-
-
-
 func TestSanitizeError_EdgeCases(t *testing.T) {
 	tests := []struct {
 		name     string
@@ -282,7 +279,7 @@ func TestSanitizeError_PreservesErrorStructure(t *testing.T) {
 	}
 
 	// Should still be an error type
-	var err error = sanitized
+	var err = sanitized
 	if err == nil {
 		t.Error("sanitized error should still implement error interface")
 	}
@@ -298,7 +295,6 @@ func TestSanitizeError_PreservesErrorStructure(t *testing.T) {
 	}
 }
 
-
 // TestSanitizeError_OrderOfOperations tests that sanitization happens in the correct order
 func TestSanitizeError_OrderOfOperations(t *testing.T) {
 	// Test that token is replaced before URL regex runs
@@ -410,4 +406,3 @@ func TestSanitizeError_CaseSensitivity(t *testing.T) {
 		t.Errorf("expected at least one redaction, got: %s", resultMsg)
 	}
 }
-

pkg/config/config.go

@@ -16,6 +16,7 @@ const (
 	MetricCommitTimesOfDay  = "COMMIT_TIMES_OF_DAY"
 	MetricWakaTimeSpentTime = "WAKATIME_SPENT_TIME"
 	MetricCodingStreak      = "CODING_STREAK"
+	MetricWakaTimeAIStats   = "WAKATIME_AI_STATS"
 )
 
 // Valid data types for WAKATIME_DATA
@@ -115,7 +116,8 @@ func Load() *Config {
 	}
 
 	if ttl := os.Getenv("CACHE_TTL"); ttl != "" {
-		fmt.Sscanf(ttl, "%d", &cfg.CacheTTL)
+		// parse failure leaves CacheTTL at zero; applyDefaults() then fills the fallback
+		_, _ = fmt.Sscanf(ttl, "%d", &cfg.CacheTTL)
 	}
 
 	cfg.applyDefaults()
@@ -204,6 +206,7 @@ func (c *Config) Validate() error {
 		MetricCommitTimesOfDay,
 		MetricWakaTimeSpentTime,
 		MetricCodingStreak,
+		MetricWakaTimeAIStats,
 	}
 	for _, metric := range c.ShowMetrics {
 		trimmed := strings.TrimSpace(metric)

pkg/config/config_test.go

@@ -23,11 +23,11 @@ func TestConfig_Validate(t *testing.T) {
 		{
 			name: "valid config with all settings",
 			config: &Config{
-				GitHubToken:    "ghp_test123",
-				WakaTimeAPIKey: "waka_test123",
-				WakaTimeRange:  "last_7_days",
-				WakaTimeData:   []string{"EDITORS", "LANGUAGES"},
-				ShowMetrics:    []string{"COMMIT_TIMES_OF_DAY", "LANGUAGE_PER_REPO"},
+				GitHubToken:        "ghp_test123",
+				WakaTimeAPIKey:     "waka_test123",
+				WakaTimeRange:      "last_7_days",
+				WakaTimeData:       []string{"EDITORS", "LANGUAGES"},
+				ShowMetrics:        []string{"COMMIT_TIMES_OF_DAY", "LANGUAGE_PER_REPO"},
 				ProgressBarVersion: "2",
 			},
 			wantErr: false,
@@ -133,4 +133,3 @@ func TestConfig_Validate(t *testing.T) {
 		})
 	}
 }
-