mirror of
https://codeberg.org/scip/kleingebaeck.git
synced 2025-12-17 12:31:03 +01:00
Compare commits
3 Commits
v0.3.8
...
fix/linter
| Author | SHA1 | Date | |
|---|---|---|---|
| 39269d3790 | |||
| bebcd15ada | |||
| 20e6299ebd |
12
Makefile
12
Makefile
@@ -56,6 +56,14 @@ test: clean
|
|||||||
mkdir -p t/out
|
mkdir -p t/out
|
||||||
go test ./... $(ARGS)
|
go test ./... $(ARGS)
|
||||||
|
|
||||||
|
testlint: test lint
|
||||||
|
|
||||||
|
lint:
|
||||||
|
golangci-lint run
|
||||||
|
|
||||||
|
lint-full:
|
||||||
|
golangci-lint run --enable-all --exclude-use-default --disable exhaustivestruct,exhaustruct,depguard,interfacer,deadcode,golint,structcheck,scopelint,varcheck,ifshort,maligned,nosnakecase,godot,funlen,gofumpt,cyclop,noctx,gochecknoglobals,paralleltest
|
||||||
|
|
||||||
testfuzzy: clean
|
testfuzzy: clean
|
||||||
go test -fuzz ./... $(ARGS)
|
go test -fuzz ./... $(ARGS)
|
||||||
|
|
||||||
@@ -88,5 +96,5 @@ show-versions: buildlocal
|
|||||||
@echo "### go version used for building:"
|
@echo "### go version used for building:"
|
||||||
@grep -m 1 go go.mod
|
@grep -m 1 go go.mod
|
||||||
|
|
||||||
lint:
|
# lint:
|
||||||
golangci-lint run -p bugs -p unused
|
# golangci-lint run -p bugs -p unused
|
||||||
|
|||||||
6
ad.go
6
ad.go
@@ -30,7 +30,7 @@ type Index struct {
|
|||||||
type Ad struct {
|
type Ad struct {
|
||||||
Title string `goquery:"h1"`
|
Title string `goquery:"h1"`
|
||||||
Slug string
|
Slug string
|
||||||
Id string
|
ID string
|
||||||
Condition string `goquery:".addetailslist--detail--value,text"`
|
Condition string `goquery:".addetailslist--detail--value,text"`
|
||||||
Category string
|
Category string
|
||||||
CategoryTree []string `goquery:".breadcrump-link,text"`
|
CategoryTree []string `goquery:".breadcrump-link,text"`
|
||||||
@@ -46,7 +46,7 @@ func (ad *Ad) LogValue() slog.Value {
|
|||||||
return slog.GroupValue(
|
return slog.GroupValue(
|
||||||
slog.String("title", ad.Title),
|
slog.String("title", ad.Title),
|
||||||
slog.String("price", ad.Price),
|
slog.String("price", ad.Price),
|
||||||
slog.String("id", ad.Id),
|
slog.String("id", ad.ID),
|
||||||
slog.Int("imagecount", len(ad.Images)),
|
slog.Int("imagecount", len(ad.Images)),
|
||||||
slog.Int("bodysize", len(ad.Text)),
|
slog.Int("bodysize", len(ad.Text)),
|
||||||
slog.String("categorytree", strings.Join(ad.CategoryTree, "+")),
|
slog.String("categorytree", strings.Join(ad.CategoryTree, "+")),
|
||||||
@@ -76,7 +76,7 @@ func (ad *Ad) CalculateExpire() {
|
|||||||
if len(ad.Created) > 0 {
|
if len(ad.Created) > 0 {
|
||||||
ts, err := time.Parse("02.01.2006", ad.Created)
|
ts, err := time.Parse("02.01.2006", ad.Created)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
ad.Expire = ts.AddDate(0, 2, 1).Format("02.01.2006")
|
ad.Expire = ts.AddDate(0, ExpireMonths, ExpireDays).Format("02.01.2006")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
84
config.go
84
config.go
@@ -17,7 +17,6 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
@@ -35,16 +34,16 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
VERSION string = "0.3.1"
|
VERSION string = "0.3.2"
|
||||||
Baseuri string = "https://www.kleinanzeigen.de"
|
Baseuri string = "https://www.kleinanzeigen.de"
|
||||||
Listuri string = "/s-bestandsliste.html"
|
Listuri string = "/s-bestandsliste.html"
|
||||||
Defaultdir string = "."
|
Defaultdir string = "."
|
||||||
|
|
||||||
DefaultTemplate string = "Title: {{.Title}}\nPrice: {{.Price}}\nId: {{.Id}}\n" +
|
DefaultTemplate string = "Title: {{.Title}}\nPrice: {{.Price}}\nId: {{.ID}}\n" +
|
||||||
"Category: {{.Category}}\nCondition: {{.Condition}}\n" +
|
"Category: {{.Category}}\nCondition: {{.Condition}}\n" +
|
||||||
"Created: {{.Created}}\nExpire: {{.Expire}}\n\n{{.Text}}\n"
|
"Created: {{.Created}}\nExpire: {{.Expire}}\n\n{{.Text}}\n"
|
||||||
|
|
||||||
DefaultTemplateWin string = "Title: {{.Title}}\r\nPrice: {{.Price}}\r\nId: {{.Id}}\r\n" +
|
DefaultTemplateWin string = "Title: {{.Title}}\r\nPrice: {{.Price}}\r\nId: {{.ID}}\r\n" +
|
||||||
"Category: {{.Category}}\r\nCondition: {{.Condition}}\r\n" +
|
"Category: {{.Category}}\r\nCondition: {{.Condition}}\r\n" +
|
||||||
"Created: {{.Created}}\r\nExpires: {{.Expire}}\r\n\r\n{{.Text}}\r\n"
|
"Created: {{.Created}}\r\nExpires: {{.Expire}}\r\n\r\n{{.Text}}\r\n"
|
||||||
|
|
||||||
@@ -56,6 +55,14 @@ const (
|
|||||||
// for image download throttling
|
// for image download throttling
|
||||||
MinThrottle int = 2
|
MinThrottle int = 2
|
||||||
MaxThrottle int = 20
|
MaxThrottle int = 20
|
||||||
|
|
||||||
|
// we extract the slug from the uri
|
||||||
|
SlugURIPartNum int = 6
|
||||||
|
|
||||||
|
ExpireMonths int = 2
|
||||||
|
ExpireDays int = 1
|
||||||
|
|
||||||
|
WIN string = "windows"
|
||||||
)
|
)
|
||||||
|
|
||||||
const Usage string = `This is kleingebaeck, the kleinanzeigen.de backup tool.
|
const Usage string = `This is kleingebaeck, the kleinanzeigen.de backup tool.
|
||||||
@@ -107,17 +114,17 @@ func (c *Config) IncrImgs(num int) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// load commandline flags and config file
|
// load commandline flags and config file
|
||||||
func InitConfig(w io.Writer) (*Config, error) {
|
func InitConfig(output io.Writer) (*Config, error) {
|
||||||
var k = koanf.New(".")
|
var kloader = koanf.New(".")
|
||||||
|
|
||||||
// determine template based on os
|
// determine template based on os
|
||||||
template := DefaultTemplate
|
template := DefaultTemplate
|
||||||
if runtime.GOOS == "windows" {
|
if runtime.GOOS == WIN {
|
||||||
template = DefaultTemplateWin
|
template = DefaultTemplateWin
|
||||||
}
|
}
|
||||||
|
|
||||||
// Load default values using the confmap provider.
|
// Load default values using the confmap provider.
|
||||||
if err := k.Load(confmap.Provider(map[string]interface{}{
|
if err := kloader.Load(confmap.Provider(map[string]interface{}{
|
||||||
"template": template,
|
"template": template,
|
||||||
"outdir": ".",
|
"outdir": ".",
|
||||||
"loglevel": "notice",
|
"loglevel": "notice",
|
||||||
@@ -125,37 +132,39 @@ func InitConfig(w io.Writer) (*Config, error) {
|
|||||||
"adnametemplate": DefaultAdNameTemplate,
|
"adnametemplate": DefaultAdNameTemplate,
|
||||||
"useragent": DefaultUserAgent,
|
"useragent": DefaultUserAgent,
|
||||||
}, "."), nil); err != nil {
|
}, "."), nil); err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("failed to load default values into koanf: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// setup custom usage
|
// setup custom usage
|
||||||
f := flag.NewFlagSet("config", flag.ContinueOnError)
|
flagset := flag.NewFlagSet("config", flag.ContinueOnError)
|
||||||
f.Usage = func() {
|
flagset.Usage = func() {
|
||||||
fmt.Fprintln(w, Usage)
|
fmt.Fprintln(output, Usage)
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
|
|
||||||
// parse commandline flags
|
// parse commandline flags
|
||||||
f.StringP("config", "c", "", "config file")
|
flagset.StringP("config", "c", "", "config file")
|
||||||
f.StringP("outdir", "o", "", "directory where to store ads")
|
flagset.StringP("outdir", "o", "", "directory where to store ads")
|
||||||
f.IntP("user", "u", 0, "user id")
|
flagset.IntP("user", "u", 0, "user id")
|
||||||
f.IntP("limit", "l", 0, "limit ads to be downloaded (default 0, unlimited)")
|
flagset.IntP("limit", "l", 0, "limit ads to be downloaded (default 0, unlimited)")
|
||||||
f.BoolP("verbose", "v", false, "be verbose")
|
flagset.BoolP("verbose", "v", false, "be verbose")
|
||||||
f.BoolP("debug", "d", false, "enable debug log")
|
flagset.BoolP("debug", "d", false, "enable debug log")
|
||||||
f.BoolP("version", "V", false, "show program version")
|
flagset.BoolP("version", "V", false, "show program version")
|
||||||
f.BoolP("help", "h", false, "show usage")
|
flagset.BoolP("help", "h", false, "show usage")
|
||||||
f.BoolP("manual", "m", false, "show manual")
|
flagset.BoolP("manual", "m", false, "show manual")
|
||||||
f.BoolP("force", "f", false, "force")
|
flagset.BoolP("force", "f", false, "force")
|
||||||
|
|
||||||
if err := f.Parse(os.Args[1:]); err != nil {
|
if err := flagset.Parse(os.Args[1:]); err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("failed to parse program arguments: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// generate a list of config files to try to load, including the
|
// generate a list of config files to try to load, including the
|
||||||
// one provided via -c, if any
|
// one provided via -c, if any
|
||||||
var configfiles []string
|
var configfiles []string
|
||||||
configfile, _ := f.GetString("config")
|
|
||||||
|
configfile, _ := flagset.GetString("config")
|
||||||
home, _ := os.UserHomeDir()
|
home, _ := os.UserHomeDir()
|
||||||
|
|
||||||
if configfile != "" {
|
if configfile != "" {
|
||||||
configfiles = []string{configfile}
|
configfiles = []string{configfile}
|
||||||
} else {
|
} else {
|
||||||
@@ -171,31 +180,30 @@ func InitConfig(w io.Writer) (*Config, error) {
|
|||||||
for _, cfgfile := range configfiles {
|
for _, cfgfile := range configfiles {
|
||||||
if path, err := os.Stat(cfgfile); !os.IsNotExist(err) {
|
if path, err := os.Stat(cfgfile); !os.IsNotExist(err) {
|
||||||
if !path.IsDir() {
|
if !path.IsDir() {
|
||||||
if err := k.Load(file.Provider(cfgfile), toml.Parser()); err != nil {
|
if err := kloader.Load(file.Provider(cfgfile), toml.Parser()); err != nil {
|
||||||
return nil, errors.New("error loading config file: " + err.Error())
|
return nil, fmt.Errorf("error loading config file: %w", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
} // else: we ignore the file if it doesn't exists
|
||||||
// else: we ignore the file if it doesn't exists
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// env overrides config file
|
// env overrides config file
|
||||||
if err := k.Load(env.Provider("KLEINGEBAECK_", ".", func(s string) string {
|
if err := kloader.Load(env.Provider("KLEINGEBAECK_", ".", func(s string) string {
|
||||||
return strings.Replace(strings.ToLower(
|
return strings.ReplaceAll(strings.ToLower(
|
||||||
strings.TrimPrefix(s, "KLEINGEBAECK_")), "_", ".", -1)
|
strings.TrimPrefix(s, "KLEINGEBAECK_")), "_", ".")
|
||||||
}), nil); err != nil {
|
}), nil); err != nil {
|
||||||
return nil, errors.New("error loading environment: " + err.Error())
|
return nil, fmt.Errorf("error loading environment: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// command line overrides env
|
// command line overrides env
|
||||||
if err := k.Load(posflag.Provider(f, ".", k), nil); err != nil {
|
if err := kloader.Load(posflag.Provider(flagset, ".", kloader), nil); err != nil {
|
||||||
return nil, errors.New("error loading flags: " + err.Error())
|
return nil, fmt.Errorf("error loading flags: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// fetch values
|
// fetch values
|
||||||
conf := &Config{}
|
conf := &Config{}
|
||||||
if err := k.Unmarshal("", &conf); err != nil {
|
if err := kloader.Unmarshal("", &conf); err != nil {
|
||||||
return nil, errors.New("error unmarshalling: " + err.Error())
|
return nil, fmt.Errorf("error unmarshalling: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// adjust loglevel
|
// adjust loglevel
|
||||||
@@ -207,7 +215,7 @@ func InitConfig(w io.Writer) (*Config, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// are there any args left on commandline? if so threat them as adlinks
|
// are there any args left on commandline? if so threat them as adlinks
|
||||||
conf.Adlinks = f.Args()
|
conf.Adlinks = flagset.Args()
|
||||||
|
|
||||||
return conf, nil
|
return conf, nil
|
||||||
}
|
}
|
||||||
|
|||||||
20
fetch.go
20
fetch.go
@@ -19,6 +19,7 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"net/http"
|
"net/http"
|
||||||
@@ -33,10 +34,10 @@ type Fetcher struct {
|
|||||||
Cookies []*http.Cookie
|
Cookies []*http.Cookie
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewFetcher(c *Config) (*Fetcher, error) {
|
func NewFetcher(conf *Config) (*Fetcher, error) {
|
||||||
jar, err := cookiejar.New(nil)
|
jar, err := cookiejar.New(nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("failed to create a cookie jar obj: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &Fetcher{
|
return &Fetcher{
|
||||||
@@ -44,35 +45,37 @@ func NewFetcher(c *Config) (*Fetcher, error) {
|
|||||||
Transport: &loggingTransport{}, // implemented in http.go
|
Transport: &loggingTransport{}, // implemented in http.go
|
||||||
Jar: jar,
|
Jar: jar,
|
||||||
},
|
},
|
||||||
Config: c,
|
Config: conf,
|
||||||
Cookies: []*http.Cookie{},
|
Cookies: []*http.Cookie{},
|
||||||
},
|
},
|
||||||
nil
|
nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Fetcher) Get(uri string) (io.ReadCloser, error) {
|
func (f *Fetcher) Get(uri string) (io.ReadCloser, error) {
|
||||||
req, err := http.NewRequest("GET", uri, nil)
|
req, err := http.NewRequest(http.MethodGet, uri, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("failed to create a new HTTP request obj: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
req.Header.Set("User-Agent", f.Config.UserAgent)
|
req.Header.Set("User-Agent", f.Config.UserAgent)
|
||||||
|
|
||||||
if len(f.Cookies) > 0 {
|
if len(f.Cookies) > 0 {
|
||||||
uriobj, _ := url.Parse(Baseuri)
|
uriobj, _ := url.Parse(Baseuri)
|
||||||
|
|
||||||
slog.Debug("have cookies, sending them",
|
slog.Debug("have cookies, sending them",
|
||||||
"sample-cookie-name", f.Cookies[0].Name,
|
"sample-cookie-name", f.Cookies[0].Name,
|
||||||
"sample-cookie-expire", f.Cookies[0].Expires,
|
"sample-cookie-expire", f.Cookies[0].Expires,
|
||||||
)
|
)
|
||||||
|
|
||||||
f.Client.Jar.SetCookies(uriobj, f.Cookies)
|
f.Client.Jar.SetCookies(uriobj, f.Cookies)
|
||||||
}
|
}
|
||||||
|
|
||||||
res, err := f.Client.Do(req)
|
res, err := f.Client.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("failed to initiate HTTP request to %s: %w", uri, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if res.StatusCode != 200 {
|
if res.StatusCode != http.StatusOK {
|
||||||
return nil, errors.New("could not get page via HTTP")
|
return nil, errors.New("could not get page via HTTP")
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -85,12 +88,15 @@ func (f *Fetcher) Get(uri string) (io.ReadCloser, error) {
|
|||||||
// fetch an image
|
// fetch an image
|
||||||
func (f *Fetcher) Getimage(uri string) (io.ReadCloser, error) {
|
func (f *Fetcher) Getimage(uri string) (io.ReadCloser, error) {
|
||||||
slog.Debug("fetching ad image", "uri", uri)
|
slog.Debug("fetching ad image", "uri", uri)
|
||||||
|
|
||||||
body, err := f.Get(uri)
|
body, err := f.Get(uri)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if f.Config.IgnoreErrors {
|
if f.Config.IgnoreErrors {
|
||||||
slog.Info("Failed to download image, error ignored", "error", err.Error())
|
slog.Info("Failed to download image, error ignored", "error", err.Error())
|
||||||
|
|
||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
1
go.mod
1
go.mod
@@ -31,6 +31,7 @@ require (
|
|||||||
github.com/mitchellh/reflectwalk v1.0.2 // indirect
|
github.com/mitchellh/reflectwalk v1.0.2 // indirect
|
||||||
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 // indirect
|
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 // indirect
|
||||||
github.com/pelletier/go-toml v1.9.5 // indirect
|
github.com/pelletier/go-toml v1.9.5 // indirect
|
||||||
|
github.com/pkg/errors v0.9.1 // indirect
|
||||||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b // indirect
|
golang.org/x/net v0.0.0-20220722155237-a158d28d115b // indirect
|
||||||
golang.org/x/sys v0.14.0 // indirect
|
golang.org/x/sys v0.14.0 // indirect
|
||||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
|
|||||||
2
go.sum
2
go.sum
@@ -50,6 +50,8 @@ github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 h1:zYyBkD/k9seD2A7fsi6
|
|||||||
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8=
|
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8=
|
||||||
github.com/pelletier/go-toml v1.9.5 h1:4yBQzkHv+7BHq2PQUZF3Mx0IYxG7LsP222s7Agd3ve8=
|
github.com/pelletier/go-toml v1.9.5 h1:4yBQzkHv+7BHq2PQUZF3Mx0IYxG7LsP222s7Agd3ve8=
|
||||||
github.com/pelletier/go-toml v1.9.5/go.mod h1:u1nR/EPcESfeI/szUZKdtJ0xRNbUoANCkoOuaOx1Y+c=
|
github.com/pelletier/go-toml v1.9.5/go.mod h1:u1nR/EPcESfeI/szUZKdtJ0xRNbUoANCkoOuaOx1Y+c=
|
||||||
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
|
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
|
||||||
|
|||||||
37
http.go
37
http.go
@@ -19,6 +19,7 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"math"
|
"math"
|
||||||
@@ -32,17 +33,20 @@ import (
|
|||||||
// easier associated in debug output
|
// easier associated in debug output
|
||||||
var letters = []rune("ABCDEF0123456789")
|
var letters = []rune("ABCDEF0123456789")
|
||||||
|
|
||||||
func getid() string {
|
const IDLEN int = 8
|
||||||
b := make([]rune, 8)
|
|
||||||
for i := range b {
|
|
||||||
b[i] = letters[rand.Intn(len(letters))]
|
|
||||||
}
|
|
||||||
return string(b)
|
|
||||||
}
|
|
||||||
|
|
||||||
// retry after HTTP 50x errors or err!=nil
|
// retry after HTTP 50x errors or err!=nil
|
||||||
const RetryCount = 3
|
const RetryCount = 3
|
||||||
|
|
||||||
|
func getid() string {
|
||||||
|
b := make([]rune, IDLEN)
|
||||||
|
for i := range b {
|
||||||
|
b[i] = letters[rand.Intn(len(letters))]
|
||||||
|
}
|
||||||
|
|
||||||
|
return string(b)
|
||||||
|
}
|
||||||
|
|
||||||
// used to inject debug log and implement retries
|
// used to inject debug log and implement retries
|
||||||
type loggingTransport struct{}
|
type loggingTransport struct{}
|
||||||
|
|
||||||
@@ -75,6 +79,7 @@ func drainBody(resp *http.Response) {
|
|||||||
// unable to copy data? uff!
|
// unable to copy data? uff!
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
resp.Body.Close()
|
resp.Body.Close()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -82,8 +87,8 @@ func drainBody(resp *http.Response) {
|
|||||||
|
|
||||||
// the actual logging transport with retries
|
// the actual logging transport with retries
|
||||||
func (t *loggingTransport) RoundTrip(req *http.Request) (*http.Response, error) {
|
func (t *loggingTransport) RoundTrip(req *http.Request) (*http.Response, error) {
|
||||||
// just requred for debugging
|
// just required for debugging
|
||||||
id := getid()
|
requestid := getid()
|
||||||
|
|
||||||
// clone the request body, put into request on retry
|
// clone the request body, put into request on retry
|
||||||
var bodyBytes []byte
|
var bodyBytes []byte
|
||||||
@@ -92,16 +97,16 @@ func (t *loggingTransport) RoundTrip(req *http.Request) (*http.Response, error)
|
|||||||
req.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))
|
req.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))
|
||||||
}
|
}
|
||||||
|
|
||||||
slog.Debug("REQUEST", "id", id, "uri", req.URL, "host", req.Host)
|
slog.Debug("REQUEST", "id", requestid, "uri", req.URL, "host", req.Host)
|
||||||
|
|
||||||
// first try
|
// first try
|
||||||
resp, err := http.DefaultTransport.RoundTrip(req)
|
resp, err := http.DefaultTransport.RoundTrip(req)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
slog.Debug("RESPONSE", "id", id, "status", resp.StatusCode,
|
slog.Debug("RESPONSE", "id", requestid, "status", resp.StatusCode,
|
||||||
"contentlength", resp.ContentLength)
|
"contentlength", resp.ContentLength)
|
||||||
}
|
}
|
||||||
|
|
||||||
// enter retry check and loop, if first req were successfull, leave loop immediately
|
// enter retry check and loop, if first req were successful, leave loop immediately
|
||||||
retries := 0
|
retries := 0
|
||||||
for shouldRetry(err, resp) && retries < RetryCount {
|
for shouldRetry(err, resp) && retries < RetryCount {
|
||||||
time.Sleep(backoff(retries))
|
time.Sleep(backoff(retries))
|
||||||
@@ -118,12 +123,16 @@ func (t *loggingTransport) RoundTrip(req *http.Request) (*http.Response, error)
|
|||||||
resp, err = http.DefaultTransport.RoundTrip(req)
|
resp, err = http.DefaultTransport.RoundTrip(req)
|
||||||
|
|
||||||
if err == nil {
|
if err == nil {
|
||||||
slog.Debug("RESPONSE", "id", id, "status", resp.StatusCode,
|
slog.Debug("RESPONSE", "id", requestid, "status", resp.StatusCode,
|
||||||
"contentlength", resp.ContentLength, "retry", retries)
|
"contentlength", resp.ContentLength, "retry", retries)
|
||||||
}
|
}
|
||||||
|
|
||||||
retries++
|
retries++
|
||||||
}
|
}
|
||||||
|
|
||||||
return resp, err
|
if err != nil {
|
||||||
|
return resp, fmt.Errorf("failed to get HTTP response for %s: %w", req.URL, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return resp, nil
|
||||||
}
|
}
|
||||||
|
|||||||
21
image.go
21
image.go
@@ -19,6 +19,7 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"fmt"
|
||||||
"image/jpeg"
|
"image/jpeg"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"os"
|
"os"
|
||||||
@@ -33,14 +34,14 @@ type Image struct {
|
|||||||
Filename string
|
Filename string
|
||||||
Hash *goimagehash.ImageHash
|
Hash *goimagehash.ImageHash
|
||||||
Data *bytes.Buffer
|
Data *bytes.Buffer
|
||||||
Uri string
|
URI string
|
||||||
}
|
}
|
||||||
|
|
||||||
// used for logging to avoid printing Data
|
// used for logging to avoid printing Data
|
||||||
func (img *Image) LogValue() slog.Value {
|
func (img *Image) LogValue() slog.Value {
|
||||||
return slog.GroupValue(
|
return slog.GroupValue(
|
||||||
slog.String("filename", img.Filename),
|
slog.String("filename", img.Filename),
|
||||||
slog.String("uri", img.Uri),
|
slog.String("uri", img.URI),
|
||||||
slog.String("hash", img.Hash.ToString()),
|
slog.String("hash", img.Hash.ToString()),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
@@ -51,7 +52,7 @@ type Cache []*goimagehash.ImageHash
|
|||||||
func NewImage(buf *bytes.Buffer, filename string, uri string) *Image {
|
func NewImage(buf *bytes.Buffer, filename string, uri string) *Image {
|
||||||
img := &Image{
|
img := &Image{
|
||||||
Filename: filename,
|
Filename: filename,
|
||||||
Uri: uri,
|
URI: uri,
|
||||||
Data: buf,
|
Data: buf,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -62,12 +63,12 @@ func NewImage(buf *bytes.Buffer, filename string, uri string) *Image {
|
|||||||
func (img *Image) CalcHash() error {
|
func (img *Image) CalcHash() error {
|
||||||
jpgdata, err := jpeg.Decode(img.Data)
|
jpgdata, err := jpeg.Decode(img.Data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to decode JPEG image: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
hash1, err := goimagehash.DifferenceHash(jpgdata)
|
hash1, err := goimagehash.DifferenceHash(jpgdata)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to calculate diff hash of image: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
img.Hash = hash1
|
img.Hash = hash1
|
||||||
@@ -80,16 +81,18 @@ func (img *Image) Similar(hash *goimagehash.ImageHash) bool {
|
|||||||
distance, err := img.Hash.Distance(hash)
|
distance, err := img.Hash.Distance(hash)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
slog.Debug("failed to compute diff hash distance", "error", err)
|
slog.Debug("failed to compute diff hash distance", "error", err)
|
||||||
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
if distance < MaxDistance {
|
if distance < MaxDistance {
|
||||||
slog.Debug("distance computation", "image-A", img.Hash.ToString(),
|
slog.Debug("distance computation", "image-A", img.Hash.ToString(),
|
||||||
"image-B", hash.ToString(), "distance", distance)
|
"image-B", hash.ToString(), "distance", distance)
|
||||||
|
|
||||||
return true
|
return true
|
||||||
} else {
|
|
||||||
return false
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// check current image against all known hashes.
|
// check current image against all known hashes.
|
||||||
@@ -108,7 +111,7 @@ func (img *Image) SimilarExists(cache Cache) bool {
|
|||||||
func ReadImages(addir string, dont bool) (Cache, error) {
|
func ReadImages(addir string, dont bool) (Cache, error) {
|
||||||
files, err := os.ReadDir(addir)
|
files, err := os.ReadDir(addir)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("failed to read ad directory contents: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
cache := Cache{}
|
cache := Cache{}
|
||||||
@@ -122,6 +125,7 @@ func ReadImages(addir string, dont bool) (Cache, error) {
|
|||||||
ext := filepath.Ext(file.Name())
|
ext := filepath.Ext(file.Name())
|
||||||
if !file.IsDir() && (ext == ".jpg" || ext == ".jpeg" || ext == ".JPG" || ext == ".JPEG") {
|
if !file.IsDir() && (ext == ".jpg" || ext == ".jpeg" || ext == ".JPG" || ext == ".JPEG") {
|
||||||
filename := filepath.Join(addir, file.Name())
|
filename := filepath.Join(addir, file.Name())
|
||||||
|
|
||||||
data, err := ReadImage(filename)
|
data, err := ReadImage(filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
@@ -137,6 +141,5 @@ func ReadImages(addir string, dont bool) (Cache, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
//return nil, errors.New("ende")
|
|
||||||
return cache, nil
|
return cache, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -133,7 +133,7 @@
|
|||||||
.\" ========================================================================
|
.\" ========================================================================
|
||||||
.\"
|
.\"
|
||||||
.IX Title "KLEINGEBAECK 1"
|
.IX Title "KLEINGEBAECK 1"
|
||||||
.TH KLEINGEBAECK 1 "2024-01-24" "1" "User Commands"
|
.TH KLEINGEBAECK 1 "2024-01-25" "1" "User Commands"
|
||||||
.\" For nroff, turn off justification. Always turn off hyphenation; it makes
|
.\" For nroff, turn off justification. Always turn off hyphenation; it makes
|
||||||
.\" way too many mistakes in technical documents.
|
.\" way too many mistakes in technical documents.
|
||||||
.if n .ad l
|
.if n .ad l
|
||||||
@@ -182,7 +182,7 @@ Format is pretty simple:
|
|||||||
\& template = """
|
\& template = """
|
||||||
\& Title: {{.Title}}
|
\& Title: {{.Title}}
|
||||||
\& Price: {{.Price}}
|
\& Price: {{.Price}}
|
||||||
\& Id: {{.Id}}
|
\& Id: {{.ID}}
|
||||||
\& Category: {{.Category}}
|
\& Category: {{.Category}}
|
||||||
\& Condition: {{.Condition}}
|
\& Condition: {{.Condition}}
|
||||||
\& Created: {{.Created}}
|
\& Created: {{.Created}}
|
||||||
@@ -191,7 +191,7 @@ Format is pretty simple:
|
|||||||
\& """
|
\& """
|
||||||
.Ve
|
.Ve
|
||||||
.PP
|
.PP
|
||||||
Be carefull if you want to change the template. The variable is a
|
Be careful if you want to change the template. The variable is a
|
||||||
multiline string surrounded by three double quotes. You can left out
|
multiline string surrounded by three double quotes. You can left out
|
||||||
certain fields and use any formatting you like. Refer to
|
certain fields and use any formatting you like. Refer to
|
||||||
<https://pkg.go.dev/text/template> for details how to write a
|
<https://pkg.go.dev/text/template> for details how to write a
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ CONFIGURATION
|
|||||||
template = """
|
template = """
|
||||||
Title: {{.Title}}
|
Title: {{.Title}}
|
||||||
Price: {{.Price}}
|
Price: {{.Price}}
|
||||||
Id: {{.Id}}
|
Id: {{.ID}}
|
||||||
Category: {{.Category}}
|
Category: {{.Category}}
|
||||||
Condition: {{.Condition}}
|
Condition: {{.Condition}}
|
||||||
Created: {{.Created}}
|
Created: {{.Created}}
|
||||||
@@ -51,7 +51,7 @@ CONFIGURATION
|
|||||||
{{.Text}}
|
{{.Text}}
|
||||||
"""
|
"""
|
||||||
|
|
||||||
Be carefull if you want to change the template. The variable is a
|
Be careful if you want to change the template. The variable is a
|
||||||
multiline string surrounded by three double quotes. You can left out
|
multiline string surrounded by three double quotes. You can left out
|
||||||
certain fields and use any formatting you like. Refer to
|
certain fields and use any formatting you like. Refer to
|
||||||
<https://pkg.go.dev/text/template> for details how to write a template.
|
<https://pkg.go.dev/text/template> for details how to write a template.
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ Format is pretty simple:
|
|||||||
template = """
|
template = """
|
||||||
Title: {{.Title}}
|
Title: {{.Title}}
|
||||||
Price: {{.Price}}
|
Price: {{.Price}}
|
||||||
Id: {{.Id}}
|
Id: {{.ID}}
|
||||||
Category: {{.Category}}
|
Category: {{.Category}}
|
||||||
Condition: {{.Condition}}
|
Condition: {{.Condition}}
|
||||||
Created: {{.Created}}
|
Created: {{.Created}}
|
||||||
@@ -51,7 +51,7 @@ Format is pretty simple:
|
|||||||
{{.Text}}
|
{{.Text}}
|
||||||
"""
|
"""
|
||||||
|
|
||||||
Be carefull if you want to change the template. The variable is a
|
Be careful if you want to change the template. The variable is a
|
||||||
multiline string surrounded by three double quotes. You can left out
|
multiline string surrounded by three double quotes. You can left out
|
||||||
certain fields and use any formatting you like. Refer to
|
certain fields and use any formatting you like. Refer to
|
||||||
L<https://pkg.go.dev/text/template> for details how to write a
|
L<https://pkg.go.dev/text/template> for details how to write a
|
||||||
|
|||||||
44
main.go
44
main.go
@@ -22,10 +22,8 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"math/rand"
|
|
||||||
"os"
|
"os"
|
||||||
"runtime/debug"
|
"runtime/debug"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/lmittmann/tint"
|
"github.com/lmittmann/tint"
|
||||||
"github.com/tlinden/yadu"
|
"github.com/tlinden/yadu"
|
||||||
@@ -37,38 +35,43 @@ func main() {
|
|||||||
os.Exit(Main(os.Stdout))
|
os.Exit(Main(os.Stdout))
|
||||||
}
|
}
|
||||||
|
|
||||||
func Main(w io.Writer) int {
|
func Main(output io.Writer) int {
|
||||||
logLevel := &slog.LevelVar{}
|
logLevel := &slog.LevelVar{}
|
||||||
opts := &tint.Options{
|
opts := &tint.Options{
|
||||||
Level: logLevel,
|
Level: logLevel,
|
||||||
AddSource: false,
|
AddSource: false,
|
||||||
ReplaceAttr: func(groups []string, a slog.Attr) slog.Attr {
|
ReplaceAttr: func(groups []string, attr slog.Attr) slog.Attr {
|
||||||
// Remove time from the output
|
// Remove time from the output
|
||||||
if a.Key == slog.TimeKey {
|
if attr.Key == slog.TimeKey {
|
||||||
return slog.Attr{}
|
return slog.Attr{}
|
||||||
}
|
}
|
||||||
return a
|
|
||||||
|
return attr
|
||||||
},
|
},
|
||||||
NoColor: IsNoTty(),
|
NoColor: IsNoTty(),
|
||||||
}
|
}
|
||||||
|
|
||||||
logLevel.Set(LevelNotice)
|
logLevel.Set(LevelNotice)
|
||||||
handler := tint.NewHandler(w, opts)
|
|
||||||
|
handler := tint.NewHandler(output, opts)
|
||||||
logger := slog.New(handler)
|
logger := slog.New(handler)
|
||||||
|
|
||||||
slog.SetDefault(logger)
|
slog.SetDefault(logger)
|
||||||
|
|
||||||
conf, err := InitConfig(w)
|
conf, err := InitConfig(output)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return Die(err)
|
return Die(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if conf.Showversion {
|
if conf.Showversion {
|
||||||
fmt.Fprintf(w, "This is kleingebaeck version %s\n", VERSION)
|
fmt.Fprintf(output, "This is kleingebaeck version %s\n", VERSION)
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
if conf.Showhelp {
|
if conf.Showhelp {
|
||||||
fmt.Fprintln(w, Usage)
|
fmt.Fprintln(output, Usage)
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -77,6 +80,7 @@ func Main(w io.Writer) int {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return Die(err)
|
return Die(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -94,7 +98,8 @@ func Main(w io.Writer) int {
|
|||||||
}
|
}
|
||||||
|
|
||||||
logLevel.Set(slog.LevelDebug)
|
logLevel.Set(slog.LevelDebug)
|
||||||
handler := yadu.NewHandler(w, opts)
|
|
||||||
|
handler := yadu.NewHandler(output, opts)
|
||||||
debuglogger := slog.New(handler).With(
|
debuglogger := slog.New(handler).With(
|
||||||
slog.Group("program_info",
|
slog.Group("program_info",
|
||||||
slog.Int("pid", os.Getpid()),
|
slog.Int("pid", os.Getpid()),
|
||||||
@@ -118,10 +123,8 @@ func Main(w io.Writer) int {
|
|||||||
return Die(err)
|
return Die(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// randomization needed here and there
|
switch {
|
||||||
rand.Seed(time.Now().UnixNano())
|
case len(conf.Adlinks) >= 1:
|
||||||
|
|
||||||
if len(conf.Adlinks) >= 1 {
|
|
||||||
// directly backup ad listing[s]
|
// directly backup ad listing[s]
|
||||||
for _, uri := range conf.Adlinks {
|
for _, uri := range conf.Adlinks {
|
||||||
err := ScrapeAd(fetch, uri)
|
err := ScrapeAd(fetch, uri)
|
||||||
@@ -129,25 +132,27 @@ func Main(w io.Writer) int {
|
|||||||
return Die(err)
|
return Die(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else if conf.User > 0 {
|
case conf.User > 0:
|
||||||
// backup all ads of the given user (via config or cmdline)
|
// backup all ads of the given user (via config or cmdline)
|
||||||
err := ScrapeUser(fetch)
|
err := ScrapeUser(fetch)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return Die(err)
|
return Die(err)
|
||||||
}
|
}
|
||||||
} else {
|
default:
|
||||||
return Die(errors.New("invalid or no user id or no ad link specified"))
|
return Die(errors.New("invalid or no user id or no ad link specified"))
|
||||||
}
|
}
|
||||||
|
|
||||||
if conf.StatsCountAds > 0 {
|
if conf.StatsCountAds > 0 {
|
||||||
adstr := "ads"
|
adstr := "ads"
|
||||||
|
|
||||||
if conf.StatsCountAds == 1 {
|
if conf.StatsCountAds == 1 {
|
||||||
adstr = "ad"
|
adstr = "ad"
|
||||||
}
|
}
|
||||||
fmt.Fprintf(w, "Successfully downloaded %d %s with %d images to %s.\n",
|
|
||||||
|
fmt.Fprintf(output, "Successfully downloaded %d %s with %d images to %s.\n",
|
||||||
conf.StatsCountAds, adstr, conf.StatsCountImages, conf.Outdir)
|
conf.StatsCountAds, adstr, conf.StatsCountImages, conf.Outdir)
|
||||||
} else {
|
} else {
|
||||||
fmt.Fprintf(w, "No ads found.")
|
fmt.Fprintf(output, "No ads found.")
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
@@ -155,5 +160,6 @@ func Main(w io.Writer) int {
|
|||||||
|
|
||||||
func Die(err error) int {
|
func Die(err error) int {
|
||||||
slog.Error("Failure", "error", err.Error())
|
slog.Error("Failure", "error", err.Error())
|
||||||
|
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
|
|||||||
83
main_test.go
83
main_test.go
@@ -43,7 +43,7 @@ const LISTTPL string = `<!DOCTYPE html>
|
|||||||
{{ range . }}
|
{{ range . }}
|
||||||
<h2 class="text-module-begin">
|
<h2 class="text-module-begin">
|
||||||
<a class="ellipsis"
|
<a class="ellipsis"
|
||||||
href="/s-anzeige/{{ .Slug }}/{{ .Id }}">{{ .Title }}</a>
|
href="/s-anzeige/{{ .Slug }}/{{ .ID }}">{{ .Title }}</a>
|
||||||
</h2>
|
</h2>
|
||||||
{{ end }}
|
{{ end }}
|
||||||
</body>
|
</body>
|
||||||
@@ -247,7 +247,7 @@ var invalidtests = []Tests{
|
|||||||
type AdConfig struct {
|
type AdConfig struct {
|
||||||
Title string
|
Title string
|
||||||
Slug string
|
Slug string
|
||||||
Id string
|
ID string
|
||||||
Price string
|
Price string
|
||||||
Category string
|
Category string
|
||||||
Condition string
|
Condition string
|
||||||
@@ -259,7 +259,7 @@ type AdConfig struct {
|
|||||||
var adsrc = []AdConfig{
|
var adsrc = []AdConfig{
|
||||||
{
|
{
|
||||||
Title: "First Ad",
|
Title: "First Ad",
|
||||||
Id: "1", Price: "5€",
|
ID: "1", Price: "5€",
|
||||||
Category: "Klimbim",
|
Category: "Klimbim",
|
||||||
Text: "Thing to sale",
|
Text: "Thing to sale",
|
||||||
Slug: "first-ad",
|
Slug: "first-ad",
|
||||||
@@ -269,7 +269,7 @@ var adsrc = []AdConfig{
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Title: "Secnd Ad",
|
Title: "Secnd Ad",
|
||||||
Id: "2", Price: "5€",
|
ID: "2", Price: "5€",
|
||||||
Category: "Kram",
|
Category: "Kram",
|
||||||
Text: "Thing to sale",
|
Text: "Thing to sale",
|
||||||
Slug: "second-ad",
|
Slug: "second-ad",
|
||||||
@@ -279,7 +279,7 @@ var adsrc = []AdConfig{
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Title: "Third Ad",
|
Title: "Third Ad",
|
||||||
Id: "3",
|
ID: "3",
|
||||||
Price: "5€",
|
Price: "5€",
|
||||||
Category: "Kuddelmuddel",
|
Category: "Kuddelmuddel",
|
||||||
Text: "Thing to sale",
|
Text: "Thing to sale",
|
||||||
@@ -290,7 +290,7 @@ var adsrc = []AdConfig{
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Title: "Forth Ad",
|
Title: "Forth Ad",
|
||||||
Id: "4",
|
ID: "4",
|
||||||
Price: "5€",
|
Price: "5€",
|
||||||
Category: "Krempel",
|
Category: "Krempel",
|
||||||
Text: "Thing to sale",
|
Text: "Thing to sale",
|
||||||
@@ -301,7 +301,7 @@ var adsrc = []AdConfig{
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Title: "Fifth Ad",
|
Title: "Fifth Ad",
|
||||||
Id: "5",
|
ID: "5",
|
||||||
Price: "5€",
|
Price: "5€",
|
||||||
Category: "Kladderadatsch",
|
Category: "Kladderadatsch",
|
||||||
Text: "Thing to sale",
|
Text: "Thing to sale",
|
||||||
@@ -312,7 +312,7 @@ var adsrc = []AdConfig{
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Title: "Sixth Ad",
|
Title: "Sixth Ad",
|
||||||
Id: "6",
|
ID: "6",
|
||||||
Price: "5€",
|
Price: "5€",
|
||||||
Category: "Klunker",
|
Category: "Klunker",
|
||||||
Text: "Thing to sale",
|
Text: "Thing to sale",
|
||||||
@@ -334,17 +334,17 @@ type Adsource struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Render a HTML template for an adlisting or an ad
|
// Render a HTML template for an adlisting or an ad
|
||||||
func GetTemplate(l []AdConfig, a AdConfig, htmltemplate string) string {
|
func GetTemplate(adconfigs []AdConfig, adconfig AdConfig, htmltemplate string) string {
|
||||||
tmpl, err := tpl.New("template").Parse(htmltemplate)
|
tmpl, err := tpl.New("template").Parse(htmltemplate)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
var out bytes.Buffer
|
var out bytes.Buffer
|
||||||
if len(a.Id) == 0 {
|
if len(adconfig.ID) == 0 {
|
||||||
err = tmpl.Execute(&out, l)
|
err = tmpl.Execute(&out, adconfigs)
|
||||||
} else {
|
} else {
|
||||||
err = tmpl.Execute(&out, a)
|
err = tmpl.Execute(&out, adconfig)
|
||||||
}
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -391,10 +391,9 @@ func InitValidSources() []Adsource {
|
|||||||
// prepare urls for the ads
|
// prepare urls for the ads
|
||||||
for _, ad := range adsrc {
|
for _, ad := range adsrc {
|
||||||
ads = append(ads, Adsource{
|
ads = append(ads, Adsource{
|
||||||
uri: fmt.Sprintf("%s/s-anzeige/%s/%s", Baseuri, ad.Slug, ad.Id),
|
uri: fmt.Sprintf("%s/s-anzeige/%s/%s", Baseuri, ad.Slug, ad.ID),
|
||||||
content: GetTemplate(nil, ad, ADTPL),
|
content: GetTemplate(nil, ad, ADTPL),
|
||||||
})
|
})
|
||||||
//panic(GetTemplate(nil, ad, ADTPL))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return ads
|
return ads
|
||||||
@@ -447,46 +446,48 @@ func GetImage(path string) []byte {
|
|||||||
|
|
||||||
// setup httpmock
|
// setup httpmock
|
||||||
func SetIntercept(ads []Adsource) {
|
func SetIntercept(ads []Adsource) {
|
||||||
ch := http.Header{}
|
headers := http.Header{}
|
||||||
ch.Add("Set-Cookie", "session=permanent")
|
headers.Add("Set-Cookie", "session=permanent")
|
||||||
|
|
||||||
for _, ad := range ads {
|
for _, advertisement := range ads {
|
||||||
if ad.status == 0 {
|
if advertisement.status == 0 {
|
||||||
ad.status = 200
|
advertisement.status = 200
|
||||||
}
|
}
|
||||||
|
|
||||||
httpmock.RegisterResponder("GET", ad.uri,
|
httpmock.RegisterResponder("GET", advertisement.uri,
|
||||||
httpmock.NewStringResponder(ad.status, ad.content).HeaderAdd(ch))
|
httpmock.NewStringResponder(advertisement.status, advertisement.content).HeaderAdd(headers))
|
||||||
}
|
}
|
||||||
|
|
||||||
// we just use 2 images, put this here
|
// we just use 2 images, put this here
|
||||||
for _, image := range []string{"t/1.jpg", "t/2.jpg"} {
|
for _, image := range []string{"t/1.jpg", "t/2.jpg"} {
|
||||||
httpmock.RegisterResponder("GET", image,
|
httpmock.RegisterResponder("GET", image,
|
||||||
httpmock.NewBytesResponder(200, GetImage(image)).HeaderAdd(ch))
|
httpmock.NewBytesResponder(200, GetImage(image)).HeaderAdd(headers))
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func VerifyAd(ad AdConfig) error {
|
func VerifyAd(advertisement AdConfig) error {
|
||||||
body := ad.Title + ad.Price + ad.Id + "Kleinanzeigen => " +
|
body := advertisement.Title + advertisement.Price + advertisement.ID + "Kleinanzeigen => " +
|
||||||
ad.Category + ad.Condition + ad.Created
|
advertisement.Category + advertisement.Condition + advertisement.Created
|
||||||
|
|
||||||
// prepare ad dir name using DefaultAdNameTemplate
|
// prepare ad dir name using DefaultAdNameTemplate
|
||||||
c := Config{Adnametemplate: "{{ .Slug }}"}
|
c := Config{Adnametemplate: "{{ .Slug }}"}
|
||||||
adstruct := Ad{Slug: ad.Slug, Id: ad.Id}
|
adstruct := Ad{Slug: advertisement.Slug, ID: advertisement.ID}
|
||||||
|
|
||||||
addir, err := AdDirName(&c, &adstruct)
|
addir, err := AdDirName(&c, &adstruct)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
file := fmt.Sprintf("t/out/%s/Adlisting.txt", addir)
|
file := fmt.Sprintf("t/out/%s/Adlisting.txt", addir)
|
||||||
|
|
||||||
content, err := os.ReadFile(file)
|
content, err := os.ReadFile(file)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("unable to read adlisting file: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if body != strings.TrimSpace(string(content)) {
|
if body != strings.TrimSpace(string(content)) {
|
||||||
msg := fmt.Sprintf("ad content doesn't match.\nExpect: %s\n Got: %s\n", body, content)
|
msg := fmt.Sprintf("ad content doesn't match.\nExpect: %s\n Got: %s\n", body, content)
|
||||||
|
|
||||||
return errors.New(msg)
|
return errors.New(msg)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -504,20 +505,21 @@ func TestMain(t *testing.T) {
|
|||||||
SetIntercept(InitValidSources())
|
SetIntercept(InitValidSources())
|
||||||
|
|
||||||
// run commandline tests
|
// run commandline tests
|
||||||
for _, tt := range tests {
|
for _, test := range tests {
|
||||||
var buf bytes.Buffer
|
var buf bytes.Buffer
|
||||||
os.Args = strings.Split(tt.args, " ")
|
|
||||||
|
os.Args = strings.Split(test.args, " ")
|
||||||
|
|
||||||
ret := Main(&buf)
|
ret := Main(&buf)
|
||||||
|
|
||||||
if ret != tt.exitcode {
|
if ret != test.exitcode {
|
||||||
t.Errorf("%s with cmd <%s> did not exit with %d but %d",
|
t.Errorf("%s with cmd <%s> did not exit with %d but %d",
|
||||||
tt.name, tt.args, tt.exitcode, ret)
|
test.name, test.args, test.exitcode, ret)
|
||||||
}
|
}
|
||||||
|
|
||||||
if !strings.Contains(buf.String(), tt.expect) {
|
if !strings.Contains(buf.String(), test.expect) {
|
||||||
t.Errorf("%s with cmd <%s> output did not match.\nExpect: %s\n Got: %s\n",
|
t.Errorf("%s with cmd <%s> output did not match.\nExpect: %s\n Got: %s\n",
|
||||||
tt.name, tt.args, tt.expect, buf.String())
|
test.name, test.args, test.expect, buf.String())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -540,20 +542,21 @@ func TestMainInvalids(t *testing.T) {
|
|||||||
SetIntercept(InitInvalidSources())
|
SetIntercept(InitInvalidSources())
|
||||||
|
|
||||||
// run commandline tests
|
// run commandline tests
|
||||||
for _, tt := range invalidtests {
|
for _, test := range invalidtests {
|
||||||
var buf bytes.Buffer
|
var buf bytes.Buffer
|
||||||
os.Args = strings.Split(tt.args, " ")
|
|
||||||
|
os.Args = strings.Split(test.args, " ")
|
||||||
|
|
||||||
ret := Main(&buf)
|
ret := Main(&buf)
|
||||||
|
|
||||||
if ret != tt.exitcode {
|
if ret != test.exitcode {
|
||||||
t.Errorf("%s with cmd <%s> did not exit with %d but %d",
|
t.Errorf("%s with cmd <%s> did not exit with %d but %d",
|
||||||
tt.name, tt.args, tt.exitcode, ret)
|
test.name, test.args, test.exitcode, ret)
|
||||||
}
|
}
|
||||||
|
|
||||||
if !strings.Contains(buf.String(), tt.expect) {
|
if !strings.Contains(buf.String(), test.expect) {
|
||||||
t.Errorf("%s with cmd <%s> output did not match.\nExpect: %s\n Got: %s\n",
|
t.Errorf("%s with cmd <%s> output did not match.\nExpect: %s\n Got: %s\n",
|
||||||
tt.name, tt.args, tt.expect, buf.String())
|
test.name, test.args, test.expect, buf.String())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
74
scrape.go
74
scrape.go
@@ -19,10 +19,10 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -43,7 +43,9 @@ func ScrapeUser(fetch *Fetcher) error {
|
|||||||
|
|
||||||
for {
|
for {
|
||||||
var index Index
|
var index Index
|
||||||
|
|
||||||
slog.Debug("fetching page", "uri", uri)
|
slog.Debug("fetching page", "uri", uri)
|
||||||
|
|
||||||
body, err := fetch.Get(uri)
|
body, err := fetch.Get(uri)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -52,7 +54,7 @@ func ScrapeUser(fetch *Fetcher) error {
|
|||||||
|
|
||||||
err = goq.NewDecoder(body).Decode(&index)
|
err = goq.NewDecoder(body).Decode(&index)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to goquery decode HTML index body: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(index.Links) == 0 {
|
if len(index.Links) == 0 {
|
||||||
@@ -67,16 +69,16 @@ func ScrapeUser(fetch *Fetcher) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
page++
|
page++
|
||||||
uri = baseuri + "&pageNum=" + fmt.Sprintf("%d", page)
|
uri = baseuri + "&pageNum=" + strconv.Itoa(page)
|
||||||
}
|
}
|
||||||
|
|
||||||
for i, adlink := range adlinks {
|
for index, adlink := range adlinks {
|
||||||
err := ScrapeAd(fetch, Baseuri+adlink)
|
err := ScrapeAd(fetch, Baseuri+adlink)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
if fetch.Config.Limit > 0 && i == fetch.Config.Limit-1 {
|
if fetch.Config.Limit > 0 && index == fetch.Config.Limit-1 {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -86,18 +88,20 @@ func ScrapeUser(fetch *Fetcher) error {
|
|||||||
|
|
||||||
// scrape an ad. uri is the full uri of the ad, dir is the basedir
|
// scrape an ad. uri is the full uri of the ad, dir is the basedir
|
||||||
func ScrapeAd(fetch *Fetcher, uri string) error {
|
func ScrapeAd(fetch *Fetcher, uri string) error {
|
||||||
ad := &Ad{}
|
advertisement := &Ad{}
|
||||||
|
|
||||||
// extract slug and id from uri
|
// extract slug and id from uri
|
||||||
uriparts := strings.Split(uri, "/")
|
uriparts := strings.Split(uri, "/")
|
||||||
if len(uriparts) < 6 {
|
if len(uriparts) < SlugURIPartNum {
|
||||||
return errors.New("invalid uri: " + uri)
|
return fmt.Errorf("invalid uri: %s", uri)
|
||||||
}
|
}
|
||||||
ad.Slug = uriparts[4]
|
|
||||||
ad.Id = uriparts[5]
|
advertisement.Slug = uriparts[4]
|
||||||
|
advertisement.ID = uriparts[5]
|
||||||
|
|
||||||
// get the ad
|
// get the ad
|
||||||
slog.Debug("fetching ad page", "uri", uri)
|
slog.Debug("fetching ad page", "uri", uri)
|
||||||
|
|
||||||
body, err := fetch.Get(uri)
|
body, err := fetch.Get(uri)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -105,36 +109,37 @@ func ScrapeAd(fetch *Fetcher, uri string) error {
|
|||||||
defer body.Close()
|
defer body.Close()
|
||||||
|
|
||||||
// extract ad contents with goquery/goq
|
// extract ad contents with goquery/goq
|
||||||
err = goq.NewDecoder(body).Decode(&ad)
|
err = goq.NewDecoder(body).Decode(&advertisement)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to goquery decode HTML ad body: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(ad.CategoryTree) > 0 {
|
if len(advertisement.CategoryTree) > 0 {
|
||||||
ad.Category = strings.Join(ad.CategoryTree, " => ")
|
advertisement.Category = strings.Join(advertisement.CategoryTree, " => ")
|
||||||
}
|
}
|
||||||
|
|
||||||
if ad.Incomplete() {
|
if advertisement.Incomplete() {
|
||||||
slog.Debug("got ad", "ad", ad)
|
slog.Debug("got ad", "ad", advertisement)
|
||||||
return errors.New("could not extract ad data from page, got empty struct")
|
|
||||||
|
return fmt.Errorf("could not extract ad data from page, got empty struct")
|
||||||
}
|
}
|
||||||
|
|
||||||
ad.CalculateExpire()
|
advertisement.CalculateExpire()
|
||||||
|
|
||||||
// write listing
|
// write listing
|
||||||
addir, err := WriteAd(fetch.Config, ad)
|
addir, err := WriteAd(fetch.Config, advertisement)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
slog.Debug("extracted ad listing", "ad", ad)
|
slog.Debug("extracted ad listing", "ad", advertisement)
|
||||||
|
|
||||||
fetch.Config.IncrAds()
|
fetch.Config.IncrAds()
|
||||||
|
|
||||||
return ScrapeImages(fetch, ad, addir)
|
return ScrapeImages(fetch, advertisement, addir)
|
||||||
}
|
}
|
||||||
|
|
||||||
func ScrapeImages(fetch *Fetcher, ad *Ad, addir string) error {
|
func ScrapeImages(fetch *Fetcher, advertisement *Ad, addir string) error {
|
||||||
// fetch images
|
// fetch images
|
||||||
img := 1
|
img := 1
|
||||||
adpath := filepath.Join(fetch.Config.Outdir, addir)
|
adpath := filepath.Join(fetch.Config.Outdir, addir)
|
||||||
@@ -145,16 +150,17 @@ func ScrapeImages(fetch *Fetcher, ad *Ad, addir string) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
g := new(errgroup.Group)
|
egroup := new(errgroup.Group)
|
||||||
|
|
||||||
for _, imguri := range ad.Images {
|
for _, imguri := range advertisement.Images {
|
||||||
imguri := imguri
|
imguri := imguri
|
||||||
file := filepath.Join(adpath, fmt.Sprintf("%d.jpg", img))
|
file := filepath.Join(adpath, fmt.Sprintf("%d.jpg", img))
|
||||||
g.Go(func() error {
|
|
||||||
|
egroup.Go(func() error {
|
||||||
// wait a little
|
// wait a little
|
||||||
|
|
||||||
t := GetThrottleTime()
|
throttle := GetThrottleTime()
|
||||||
time.Sleep(t)
|
time.Sleep(throttle)
|
||||||
|
|
||||||
body, err := fetch.Getimage(imguri)
|
body, err := fetch.Getimage(imguri)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -164,7 +170,7 @@ func ScrapeImages(fetch *Fetcher, ad *Ad, addir string) error {
|
|||||||
buf := new(bytes.Buffer)
|
buf := new(bytes.Buffer)
|
||||||
_, err = buf.ReadFrom(body)
|
_, err = buf.ReadFrom(body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to read from image buffer: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
buf2 := buf.Bytes() // needed for image writing
|
buf2 := buf.Bytes() // needed for image writing
|
||||||
@@ -177,7 +183,8 @@ func ScrapeImages(fetch *Fetcher, ad *Ad, addir string) error {
|
|||||||
|
|
||||||
if !fetch.Config.ForceDownload {
|
if !fetch.Config.ForceDownload {
|
||||||
if image.SimilarExists(cache) {
|
if image.SimilarExists(cache) {
|
||||||
slog.Debug("similar image exists, not written", "uri", image.Uri)
|
slog.Debug("similar image exists, not written", "uri", image.URI)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -187,17 +194,18 @@ func ScrapeImages(fetch *Fetcher, ad *Ad, addir string) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
slog.Debug("wrote image", "image", image, "size", len(buf2), "throttle", t)
|
slog.Debug("wrote image", "image", image, "size", len(buf2), "throttle", throttle)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
})
|
})
|
||||||
img++
|
img++
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := g.Wait(); err != nil {
|
if err := egroup.Wait(); err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to finalize error waitgroup: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
fetch.Config.IncrImgs(len(ad.Images))
|
fetch.Config.IncrImgs(len(advertisement.Images))
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
52
store.go
52
store.go
@@ -28,30 +28,32 @@ import (
|
|||||||
tpl "text/template"
|
tpl "text/template"
|
||||||
)
|
)
|
||||||
|
|
||||||
func AdDirName(c *Config, ad *Ad) (string, error) {
|
func AdDirName(conf *Config, advertisement *Ad) (string, error) {
|
||||||
tmpl, err := tpl.New("adname").Parse(c.Adnametemplate)
|
tmpl, err := tpl.New("adname").Parse(conf.Adnametemplate)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", fmt.Errorf("failed to parse adname template: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
buf := bytes.Buffer{}
|
buf := bytes.Buffer{}
|
||||||
err = tmpl.Execute(&buf, ad)
|
|
||||||
|
err = tmpl.Execute(&buf, advertisement)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", fmt.Errorf("failed to execute adname template: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return buf.String(), nil
|
return buf.String(), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func WriteAd(c *Config, ad *Ad) (string, error) {
|
func WriteAd(conf *Config, advertisement *Ad) (string, error) {
|
||||||
// prepare ad dir name
|
// prepare ad dir name
|
||||||
addir, err := AdDirName(c, ad)
|
addir, err := AdDirName(conf, advertisement)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
|
||||||
// prepare output dir
|
// prepare output dir
|
||||||
dir := filepath.Join(c.Outdir, addir)
|
dir := filepath.Join(conf.Outdir, addir)
|
||||||
|
|
||||||
err = Mkdir(dir)
|
err = Mkdir(dir)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
@@ -59,26 +61,27 @@ func WriteAd(c *Config, ad *Ad) (string, error) {
|
|||||||
|
|
||||||
// write ad file
|
// write ad file
|
||||||
listingfile := filepath.Join(dir, "Adlisting.txt")
|
listingfile := filepath.Join(dir, "Adlisting.txt")
|
||||||
f, err := os.Create(listingfile)
|
|
||||||
if err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
defer f.Close()
|
|
||||||
|
|
||||||
if runtime.GOOS == "windows" {
|
listingfd, err := os.Create(listingfile)
|
||||||
ad.Text = strings.ReplaceAll(ad.Text, "<br/>", "\r\n")
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("failed to create Adlisting.txt: %w", err)
|
||||||
|
}
|
||||||
|
defer listingfd.Close()
|
||||||
|
|
||||||
|
if runtime.GOOS == WIN {
|
||||||
|
advertisement.Text = strings.ReplaceAll(advertisement.Text, "<br/>", "\r\n")
|
||||||
} else {
|
} else {
|
||||||
ad.Text = strings.ReplaceAll(ad.Text, "<br/>", "\n")
|
advertisement.Text = strings.ReplaceAll(advertisement.Text, "<br/>", "\n")
|
||||||
}
|
}
|
||||||
|
|
||||||
tmpl, err := tpl.New("adlisting").Parse(c.Template)
|
tmpl, err := tpl.New("adlisting").Parse(conf.Template)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", fmt.Errorf("failed to parse adlisting template: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = tmpl.Execute(f, ad)
|
err = tmpl.Execute(listingfd, advertisement)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", fmt.Errorf("failed to execute adlisting template: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
slog.Info("wrote ad listing", "listingfile", listingfile)
|
slog.Info("wrote ad listing", "listingfile", listingfile)
|
||||||
@@ -89,14 +92,14 @@ func WriteAd(c *Config, ad *Ad) (string, error) {
|
|||||||
func WriteImage(filename string, buf []byte) error {
|
func WriteImage(filename string, buf []byte) error {
|
||||||
file, err := os.Create(filename)
|
file, err := os.Create(filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to open image file: %w", err)
|
||||||
}
|
}
|
||||||
defer file.Close()
|
defer file.Close()
|
||||||
|
|
||||||
_, err = file.Write(buf)
|
_, err = file.Write(buf)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to write to image file: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
@@ -111,12 +114,12 @@ func ReadImage(filename string) (*bytes.Buffer, error) {
|
|||||||
|
|
||||||
data, err := os.ReadFile(filename)
|
data, err := os.ReadFile(filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("failed to read image file: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = buf.Write(data)
|
_, err = buf.Write(data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("failed to write image into buffer: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &buf, nil
|
return &buf, nil
|
||||||
@@ -127,5 +130,6 @@ func fileExists(filename string) bool {
|
|||||||
if os.IsNotExist(err) {
|
if os.IsNotExist(err) {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
return !info.IsDir()
|
return !info.IsDir()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -26,6 +26,8 @@ import (
|
|||||||
// doesn't show up in the coverage report for unknown reasons, so
|
// doesn't show up in the coverage report for unknown reasons, so
|
||||||
// here's a single test for it
|
// here's a single test for it
|
||||||
func TestWriteImage(t *testing.T) {
|
func TestWriteImage(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
buf := []byte{1, 2, 3, 4, 5, 6, 7, 8}
|
buf := []byte{1, 2, 3, 4, 5, 6, 7, 8}
|
||||||
file := "t/out/t.jpg"
|
file := "t/out/t.jpg"
|
||||||
|
|
||||||
@@ -33,5 +35,4 @@ func TestWriteImage(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Errorf("Could not write mock image to %s: %s", file, err.Error())
|
t.Errorf("Could not write mock image to %s: %s", file, err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
# empty config for Main() unit tests to force unit tests NOT to use an
|
# empty config for Main() unit tests to force unit tests NOT to use an
|
||||||
# eventually existing ~/.kleingebaeck!
|
# eventually existing ~/.kleingebaeck!
|
||||||
template="""
|
template="""
|
||||||
{{.Title}}{{.Price}}{{.Id}}{{.Category}}{{.Condition}}{{.Created}}
|
{{.Title}}{{.Price}}{{.ID}}{{.Category}}{{.Condition}}{{.Created}}
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|||||||
@@ -2,5 +2,5 @@ user = 1
|
|||||||
loglevel = "verbose"
|
loglevel = "verbose"
|
||||||
outdir = "t/out"
|
outdir = "t/out"
|
||||||
template="""
|
template="""
|
||||||
{{.Title}}{{.Price}}{{.Id}}{{.Category}}{{.Condition}}{{.Created}}
|
{{.Title}}{{.Price}}{{.ID}}{{.Category}}{{.Condition}}{{.Created}}
|
||||||
"""
|
"""
|
||||||
|
|||||||
10
util.go
10
util.go
@@ -20,6 +20,7 @@ package main
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"errors"
|
"errors"
|
||||||
|
"fmt"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
@@ -33,7 +34,7 @@ func Mkdir(dir string) error {
|
|||||||
if _, err := os.Stat(dir); errors.Is(err, os.ErrNotExist) {
|
if _, err := os.Stat(dir); errors.Is(err, os.ErrNotExist) {
|
||||||
err := os.Mkdir(dir, os.ModePerm)
|
err := os.Mkdir(dir, os.ModePerm)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to create directory %s: %w", dir, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -44,7 +45,8 @@ func man() error {
|
|||||||
man := exec.Command("less", "-")
|
man := exec.Command("less", "-")
|
||||||
|
|
||||||
var b bytes.Buffer
|
var b bytes.Buffer
|
||||||
b.Write([]byte(manpage))
|
|
||||||
|
b.WriteString(manpage)
|
||||||
|
|
||||||
man.Stdout = os.Stdout
|
man.Stdout = os.Stdout
|
||||||
man.Stdin = &b
|
man.Stdin = &b
|
||||||
@@ -53,7 +55,7 @@ func man() error {
|
|||||||
err := man.Run()
|
err := man.Run()
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to execute 'less': %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
@@ -61,7 +63,7 @@ func man() error {
|
|||||||
|
|
||||||
// returns TRUE if stdout is NOT a tty or windows
|
// returns TRUE if stdout is NOT a tty or windows
|
||||||
func IsNoTty() bool {
|
func IsNoTty() bool {
|
||||||
if runtime.GOOS == "windows" || !isatty.IsTerminal(os.Stdout.Fd()) {
|
if runtime.GOOS == WIN || !isatty.IsTerminal(os.Stdout.Fd()) {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user