115 lines
2.4 KiB
Go
115 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/urfave/cli/v3"
|
|
|
|
"gitea.stevedudenhoeffer.com/steve/go-extractor"
|
|
"gitea.stevedudenhoeffer.com/steve/go-extractor/cmd/browser/pkg/browser"
|
|
"gitea.stevedudenhoeffer.com/steve/go-extractor/sites/duckduckgo"
|
|
)
|
|
|
|
type DuckDuckGoFlags []cli.Flag
|
|
|
|
var Flags = DuckDuckGoFlags{
|
|
&cli.StringFlag{
|
|
Name: "region",
|
|
Aliases: []string{"r"},
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "safesearch",
|
|
Aliases: []string{"s"},
|
|
},
|
|
}
|
|
|
|
func (f DuckDuckGoFlags) ToConfig(cmd *cli.Command) (duckduckgo.Config, error) {
|
|
var res = duckduckgo.DefaultConfig
|
|
|
|
if r := cmd.String("region"); r != "" {
|
|
res.Region = r
|
|
}
|
|
|
|
if s := cmd.String("safesearch"); s != "" {
|
|
switch s {
|
|
case "on":
|
|
res.SafeSearch = duckduckgo.SafeSearchOn
|
|
case "moderate":
|
|
res.SafeSearch = duckduckgo.SafeSearchModerate
|
|
case "off":
|
|
res.SafeSearch = duckduckgo.SafeSearchOff
|
|
default:
|
|
return res, fmt.Errorf("invalid safe search value: %s", s)
|
|
}
|
|
}
|
|
|
|
return res, nil
|
|
}
|
|
|
|
func main() {
|
|
var flags []cli.Flag
|
|
|
|
flags = append(flags, browser.Flags...)
|
|
flags = append(flags, Flags...)
|
|
|
|
cli := &cli.Command{
|
|
Name: "duckduckgo",
|
|
Usage: "Search DuckDuckGo",
|
|
Flags: flags,
|
|
Action: func(ctx context.Context, command *cli.Command) error {
|
|
c, err := Flags.ToConfig(command)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
query := strings.TrimSpace(strings.Join(command.Args().Slice(), " "))
|
|
|
|
if query == "" {
|
|
return cli.Exit("usage: duckduckgo <query>", 1)
|
|
}
|
|
|
|
b, err := browser.FromCommand(ctx, command)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create browser: %w", err)
|
|
}
|
|
defer extractor.DeferClose(b)
|
|
|
|
search, err := c.OpenSearch(ctx, b, query)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open search: %w", err)
|
|
}
|
|
|
|
defer extractor.DeferClose(search)
|
|
|
|
res, err := search.GetResults()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get results: %w", err)
|
|
}
|
|
fmt.Println("Results:", res)
|
|
|
|
err = search.LoadMore()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to load more: %w", err)
|
|
}
|
|
|
|
time.Sleep(2 * time.Second)
|
|
res, err = search.GetResults()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get results: %w", err)
|
|
}
|
|
fmt.Println("Results:", res)
|
|
|
|
return nil
|
|
},
|
|
}
|
|
|
|
if err := cli.Run(context.Background(), os.Args); err != nil {
|
|
fmt.Fprintf(os.Stderr, "error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|