scip/kleingebaeck
1
0
Fork 0
mirror of https://codeberg.org/scip/kleingebaeck.git synced 2025-12-16 12:01:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

properly check image format for storing and distance hashing

Browse Source
This commit is contained in:
Thomas von Dein
2025-01-06 10:19:34 +01:00
committed by T.v.Dein
parent 31b27beee5
commit 2239a83f76
5 changed files with 65 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
scrape.go
View File

@@ -170,7 +170,9 @@ func ScrapeImages(fetch *Fetcher, advertisement *Ad, addir string) error {
for _, imguri := range advertisement.Images {
imguri := imguri
file := filepath.Join(adpath, fmt.Sprintf("%d.jpg", img))
// we append the suffix later in NewImage() based on image format
file := filepath.Join(adpath, fmt.Sprintf("%d", img))
egroup.Go(func() error {
// wait a little
@@ -192,7 +194,11 @@ func ScrapeImages(fetch *Fetcher, advertisement *Ad, addir string) error {
reader := bytes.NewReader(buf.Bytes())
image := NewImage(reader, file, imguri)
image, err := NewImage(reader, file, imguri)
if err != nil {
return err
}
err = image.CalcHash()
if err != nil {
return err
@@ -211,7 +217,7 @@ func ScrapeImages(fetch *Fetcher, advertisement *Ad, addir string) error {
return fmt.Errorf("failed to seek(0) on image reader: %w", err)
}
err = WriteImage(file, reader)
err = WriteImage(image.Filename, reader)
if err != nil {
return err
}