1package cli
  2
  3import (
  4	"fmt"
  5	"os"
  6	"strings"
  7
  8	"github.com/alexflint/go-arg"
  9)
 10
 11var (
 12	Version   string
 13	BuildDate string
 14)
 15
 16// Args represents command-line arguments.
 17type Args struct {
 18	FeedsFile string `arg:"--feeds-file,required" help:"Path to file containing RSS feed URLs (one per line)"`
 19	Prompt    string `arg:"--prompt,required" help:"User prompt for summarisation/analysis"`
 20
 21	// Feed aggregation options
 22	MaxArticlesPerFeed int `arg:"--max-articles-per-feed" help:"Maximum articles to fetch from each feed" default:"10"`
 23	MaxDaysOld         int `arg:"--max-days-old" help:"Only include articles from the last N days (0 for all)" default:"7"`
 24	MaxTotalArticles   int `arg:"--max-total-articles" help:"Maximum total articles to process" default:"20"`
 25
 26	// Content filtering
 27	IncludeKeywords string `arg:"--include-keywords" help:"Comma-separated list of keywords to include (case-insensitive)"`
 28	ExcludeKeywords string `arg:"--exclude-keywords" help:"Comma-separated list of keywords to exclude (case-insensitive)"`
 29
 30	// LLM API options
 31	APIKey      string  `arg:"--api-key" help:"OpenAI-compatible API key (default: read from LLM_AGGREGATOR_API_KEY env var)"`
 32	BaseURL     string  `arg:"--base-url" help:"API base URL" default:"https://api.deepseek.com"`
 33	Model       string  `arg:"--model" help:"LLM model to use" default:"deepseek-chat"`
 34	MaxTokens   int     `arg:"--max-tokens" help:"Maximum tokens in response" default:"4000"`
 35	Temperature float64 `arg:"--temperature" help:"Sampling temperature (0.0 to 1.0)" default:"0.7"`
 36
 37	// Output options
 38	Output          string `arg:"--output" help:"Output format" default:"text" choice:"text,json,markdown"`
 39	OutputFile      string `arg:"--output-file" help:"Write output to file (default: stdout)"`
 40	IncludeArticles bool   `arg:"--include-articles" help:"Include original articles in JSON output"`
 41
 42	// System options
 43	SystemPrompt string `arg:"--system-prompt" help:"Custom system prompt for LLM"`
 44	TUI          bool   `arg:"--tui" help:"Enable TUI interface with progress bar"`
 45	Verbose      bool   `arg:"-v,--verbose" help:"Show verbose output"`
 46	ShowVersion  bool   `arg:"--version" help:"Show version"`
 47	DryRun       bool   `arg:"--dry-run" help:"Validate config, show article statistics, and exit without making LLM API calls"`
 48}
 49
 50// Version returns the version string.
 51func (Args) Version() string {
 52	return fmt.Sprintf("llm_aggregator v%s (built %s)", Version, BuildDate)
 53}
 54
 55// Description returns the program description.
 56func (Args) Description() string {
 57	return `LLM Aggregator - Aggregate RSS feeds and summarise with LLM API
 58
 59Examples:
 60  # Basic usage with prompts
 61  llm_aggregator --feeds-file feeds.txt --prompt "What are the latest trends in free software?"
 62  
 63  # Validate configuration and preview articles (no LLM API calls)
 64  llm_aggregator --feeds-file feeds.txt --prompt "Summarise news" --dry-run
 65  
 66  # With custom LLM model
 67  llm_aggregator --feeds-file feeds.txt --prompt "Summarise tech news" --model deepseek-coder
 68  
 69  # Output to JSON file
 70  llm_aggregator --feeds-file feeds.txt --prompt "Analyse AI developments" --output json --output-file analysis.json
 71  
 72  # Filter by keywords
 73  llm_aggregator --feeds-file feeds.txt --prompt "Linux news" --include-keywords linux,opensource
 74
 75Environment Variables:
 76  LLM_AGGREGATOR_API_KEY: Your LLM API key (required if not provided via --api-key)`
 77}
 78
 79// ParseKeywords parses comma-separated keywords string into list.
 80func ParseKeywords(keywordString string) []string {
 81	if keywordString == "" {
 82		return nil
 83	}
 84	keywords := strings.Split(keywordString, ",")
 85	result := make([]string, 0, len(keywords))
 86	for _, kw := range keywords {
 87		if trimmed := strings.TrimSpace(kw); trimmed != "" {
 88			result = append(result, trimmed)
 89		}
 90	}
 91	return result
 92}
 93
 94// ParseArgs parses command line arguments.
 95func ParseArgs() (*Args, error) {
 96	var args Args
 97	parser, err := arg.NewParser(arg.Config{
 98		Program: "llm_aggregator",
 99	}, &args)
100	if err != nil {
101		return nil, err
102	}
103
104	// Handle help and version flags before checking required fields
105	if len(os.Args) > 1 {
106		if os.Args[1] == "-h" || os.Args[1] == "--help" {
107			parser.WriteHelp(os.Stdout)
108			os.Exit(0)
109		}
110		if os.Args[1] == "--version" {
111			fmt.Printf("llm_aggregator v%s (built %s)", Version, BuildDate)
112			os.Exit(0)
113		}
114	}
115
116	err = parser.Parse(os.Args[1:])
117	if err != nil {
118		return nil, err
119	}
120
121	return &args, nil
122}
123
124// ToViperMap converts Args to a map for binding to Viper.
125// Only non-empty/non-zero values are included.
126func (a *Args) ToViperMap() map[string]any {
127	return map[string]any{
128		"max_articles_per_feed": a.MaxArticlesPerFeed,
129		"max_days_old":         a.MaxDaysOld,
130		"max_total_articles":   a.MaxTotalArticles,
131		"include_keywords":     a.IncludeKeywords,
132		"exclude_keywords":     a.ExcludeKeywords,
133		"api_key":              a.APIKey,
134		"base_url":             a.BaseURL,
135		"model":                a.Model,
136		"max_tokens":           a.MaxTokens,
137		"temperature":          a.Temperature,
138		"system_prompt":        a.SystemPrompt,
139		"output":               a.Output,
140		"output_file":          a.OutputFile,
141		"include_articles":     a.IncludeArticles,
142	}
143}