Refactored, added comprehensive testing.

input/role_test.go

@@ -0,0 +1,95 @@
+package input
+
+import (
+	"testing"
+
+	"git.wisehodl.dev/jay/aicli/config"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestDetermineRole(t *testing.T) {
+	tests := []struct {
+		name     string
+		cfg      config.ConfigData
+		hasStdin bool
+		want     StdinRole
+	}{
+		{
+			name:     "no stdin returns StdinAsPrompt",
+			cfg:      config.ConfigData{},
+			hasStdin: false,
+			want:     StdinAsPrompt,
+		},
+		{
+			name:     "stdin with no flags returns StdinAsPrompt",
+			cfg:      config.ConfigData{},
+			hasStdin: true,
+			want:     StdinAsPrompt,
+		},
+		{
+			name: "stdin with -p flag returns StdinAsPrefixedContent",
+			cfg: config.ConfigData{
+				PromptFlags: []string{"analyze this"},
+			},
+			hasStdin: true,
+			want:     StdinAsPrefixedContent,
+		},
+		{
+			name: "stdin with -pf flag returns StdinAsPrefixedContent",
+			cfg: config.ConfigData{
+				PromptPaths: []string{"prompt.txt"},
+			},
+			hasStdin: true,
+			want:     StdinAsPrefixedContent,
+		},
+		{
+			name: "stdin with -F flag returns StdinAsFile",
+			cfg: config.ConfigData{
+				StdinAsFile: true,
+			},
+			hasStdin: true,
+			want:     StdinAsFile,
+		},
+		{
+			name: "stdin with -F and -p returns StdinAsFile (explicit wins)",
+			cfg: config.ConfigData{
+				StdinAsFile: true,
+				PromptFlags: []string{"analyze"},
+			},
+			hasStdin: true,
+			want:     StdinAsFile,
+		},
+		{
+			name: "stdin with file flags returns StdinAsPrompt",
+			cfg: config.ConfigData{
+				FilePaths: []string{"main.go"},
+			},
+			hasStdin: true,
+			want:     StdinAsPrompt,
+		},
+		{
+			name: "no stdin with -F returns StdinAsFile (role set but unused)",
+			cfg: config.ConfigData{
+				StdinAsFile: true,
+			},
+			hasStdin: false,
+			want:     StdinAsPrompt,
+		},
+		{
+			name: "stdin with both -p and -pf returns StdinAsPrefixedContent",
+			cfg: config.ConfigData{
+				PromptFlags: []string{"first"},
+				PromptPaths: []string{"prompt.txt"},
+			},
+			hasStdin: true,
+			want:     StdinAsPrefixedContent,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := DetermineRole(tt.cfg, tt.hasStdin)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}