Steve Dudenhoeffer
e5e0db85e8
- Fix archive cmd passing only archive-specific Flags instead of the merged flags variable that includes browser flags (#8) - Move defer DeferClose() after error checks in 6 locations to prevent calling Close on nil values (#19): - sites/duckduckgo/cmd/duckduckgo/main.go - sites/duckduckgo/duckduckgo.go - sites/google/cmd/google/main.go - sites/wegmans/cmd/wegmans/main.go - sites/wegmans/wegmans.go - sites/aislegopher/aislegopher.go Closes #8, #19 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package aislegopher
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"gitea.stevedudenhoeffer.com/steve/go-extractor"
|
|
)
|
|
|
|
type Config struct {
|
|
}
|
|
|
|
var DefaultConfig = Config{}
|
|
|
|
var (
|
|
ErrInvalidURL = errors.New("invalid url")
|
|
)
|
|
|
|
type Item struct {
|
|
ID int
|
|
Name string
|
|
Price float64
|
|
}
|
|
|
|
func GetItemFromURL(ctx context.Context, b extractor.Browser, u *url.URL) (Item, error) {
|
|
return DefaultConfig.GetItemFromURL(ctx, b, u)
|
|
}
|
|
|
|
func (c Config) GetItemFromURL(ctx context.Context, b extractor.Browser, u *url.URL) (Item, error) {
|
|
res := Item{}
|
|
|
|
// the url will be in the format of aislegopher.com/p/slug/id
|
|
// we need to parse the slug and id from the url
|
|
a := strings.Split(u.Path, "/")
|
|
if len(a) != 4 {
|
|
return res, ErrInvalidURL
|
|
}
|
|
|
|
if a[1] != "p" {
|
|
return res, ErrInvalidURL
|
|
}
|
|
|
|
if u.Host != "aislegopher.com" && u.Host != "www.aislegopher.com" {
|
|
return res, ErrInvalidURL
|
|
}
|
|
|
|
res.ID, _ = strconv.Atoi(a[3])
|
|
|
|
doc, err := b.Open(ctx, u.String(), extractor.OpenPageOptions{})
|
|
if err != nil {
|
|
return res, fmt.Errorf("failed to open page: %w", err)
|
|
}
|
|
defer extractor.DeferClose(doc)
|
|
|
|
names := doc.Select(".h4")
|
|
|
|
if len(names) > 0 {
|
|
res.Name, _ = names[0].Text()
|
|
}
|
|
|
|
prices := doc.Select(".h2")
|
|
|
|
if len(prices) > 0 {
|
|
priceStr, _ := prices[0].Text()
|
|
priceStr = strings.ReplaceAll(priceStr, "$", "")
|
|
priceStr = strings.TrimSpace(priceStr)
|
|
res.Price, _ = strconv.ParseFloat(priceStr, 64)
|
|
}
|
|
|
|
return res, nil
|
|
}
|