94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/regismesquita/search-cli/internal/adapters"
|
|
"github.com/regismesquita/search-cli/internal/formatter"
|
|
)
|
|
|
|
func main() {
|
|
jsonOutput := flag.Bool("json", false, "Output in JSON format")
|
|
tavilyMode := flag.Bool("t", false, "Use Tavily as provider")
|
|
serperMode := flag.Bool("s", false, "Use Serper as provider (default)")
|
|
extractMode := flag.Bool("e", false, "Extract content from URLs (Tavily only)")
|
|
depth := flag.String("depth", "basic", "Search depth (basic or advanced) - only for Tavily")
|
|
flag.Parse()
|
|
|
|
if flag.NArg() < 1 {
|
|
fmt.Printf("Usage: %s [options] <query>\n", os.Args[0])
|
|
fmt.Println("Options:")
|
|
flag.PrintDefaults()
|
|
os.Exit(0)
|
|
}
|
|
|
|
// Validate provider flags
|
|
if *tavilyMode && *serperMode {
|
|
fmt.Fprintln(os.Stderr, "Error: Cannot use both -t and -s")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Validate extract mode
|
|
if *extractMode && *serperMode {
|
|
fmt.Fprintln(os.Stderr, "Error: Extract mode (-e) is only available with Tavily (-t)")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if *extractMode {
|
|
provider, err := adapters.NewTavilyProvider()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
urls := strings.Fields(strings.Join(flag.Args(), " "))
|
|
results, err := provider.Extract(urls)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if *jsonOutput {
|
|
json.NewEncoder(os.Stdout).Encode(results)
|
|
} else {
|
|
formatter.FormatExtractResults(results)
|
|
}
|
|
} else {
|
|
var searchProvider adapters.SearchProvider
|
|
var err error
|
|
|
|
// Default to Serper unless Tavily is explicitly requested
|
|
if *tavilyMode {
|
|
searchProvider, err = adapters.NewTavilyProvider()
|
|
} else {
|
|
searchProvider, err = adapters.NewSerperProvider()
|
|
}
|
|
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
options := map[string]string{}
|
|
if *tavilyMode {
|
|
options["depth"] = *depth
|
|
}
|
|
|
|
query := flag.Arg(0)
|
|
results, err := searchProvider.Search(query, options)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if *jsonOutput {
|
|
json.NewEncoder(os.Stdout).Encode(results)
|
|
} else {
|
|
formatter.FormatResults(results)
|
|
}
|
|
}
|
|
}
|