69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
// search runs a /search/?q=... query and prints the first N pages of
|
|
// matches. Works anonymously for general-rated results; mature/adult
|
|
// searches require login.
|
|
//
|
|
// go run ./examples/search dragon
|
|
// go run ./examples/search "fox knight" --max=2 --rating=general --order=date
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
fa "git.anthrove.art/public/go-fa-api"
|
|
)
|
|
|
|
func main() {
|
|
maxPages := flag.Int("max", 1, "max pages to fetch")
|
|
ratingFlag := flag.String("rating", "", "comma-separated: general, mature, adult (empty = all)")
|
|
orderFlag := flag.String("order", "", "relevancy | date | popularity (empty = default)")
|
|
flag.Parse()
|
|
|
|
if flag.NArg() == 0 {
|
|
log.Fatalf("usage: %s [flags] <query>", os.Args[0])
|
|
}
|
|
query := strings.Join(flag.Args(), " ")
|
|
|
|
opts := []fa.Option{fa.WithUserAgent("go-fa-api-example/0.1")}
|
|
if a, b := os.Getenv("FA_A"), os.Getenv("FA_B"); a != "" && b != "" {
|
|
opts = []fa.Option{
|
|
fa.WithCookies(fa.Cookies{A: a, B: b}),
|
|
fa.WithCloudflare(fa.CFCookies{Clearance: os.Getenv("CF_CLEARANCE")}),
|
|
fa.WithUserAgent(envOr("FA_UA", "go-fa-api-example/0.1")),
|
|
}
|
|
}
|
|
client := fa.New(opts...)
|
|
|
|
so := fa.SearchOptions{MaxPages: *maxPages}
|
|
if *ratingFlag != "" {
|
|
for _, r := range strings.Split(*ratingFlag, ",") {
|
|
so.Ratings = append(so.Ratings, fa.ParseRating(strings.TrimSpace(r)))
|
|
}
|
|
}
|
|
if *orderFlag != "" {
|
|
so.OrderBy = fa.SearchOrder(*orderFlag)
|
|
}
|
|
|
|
count := 0
|
|
for sub, err := range client.Search(context.Background(), query, so) {
|
|
if err != nil {
|
|
log.Fatalf("Search: %v", err)
|
|
}
|
|
count++
|
|
fmt.Printf("[%d] %s %s by %s\n",
|
|
sub.ID, sub.Title, sub.Rating, sub.Author.DisplayName)
|
|
}
|
|
fmt.Printf("\n%d results for %q across up to %d page(s)\n", count, query, *maxPages)
|
|
}
|
|
|
|
func envOr(key, fallback string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|