Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 48 additions & 0 deletions pkg/scraper/autotag.go
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,49 @@ func autotagMatchTags(ctx context.Context, path string, tagReader models.TagAuto
return ret, nil
}

func (s autotagScraper) viaImage(ctx context.Context, _client *http.Client, image *models.Image) (*models.ScrapedImage, error) {
path := image.Path
if path == "" {
return nil, nil
}

var ret *models.ScrapedImage

// only trim extension if image is file-based
trimExt := image.PrimaryFileID != nil

// populate performers, studio and tags based on image path
if err := txn.WithReadTxn(ctx, s.txnManager, func(ctx context.Context) error {
performers, err := autotagMatchPerformers(ctx, path, s.performerReader, trimExt)
if err != nil {
return fmt.Errorf("autotag scraper viaImage: %w", err)
}
studio, err := autotagMatchStudio(ctx, path, s.studioReader, trimExt)
if err != nil {
return fmt.Errorf("autotag scraper viaImage: %w", err)
}

tags, err := autotagMatchTags(ctx, path, s.tagReader, trimExt)
if err != nil {
return fmt.Errorf("autotag scraper viaImage: %w", err)
}

if len(performers) > 0 || studio != nil || len(tags) > 0 {
ret = &models.ScrapedImage{
Performers: performers,
Studio: studio,
Tags: tags,
}
}

return nil
}); err != nil {
return nil, err
}

return ret, nil
}

func (s autotagScraper) viaScene(ctx context.Context, _client *http.Client, scene *models.Scene) (*models.ScrapedScene, error) {
var ret *models.ScrapedScene
const trimExt = false
Expand Down Expand Up @@ -181,6 +224,8 @@ func (s autotagScraper) supports(ty ScrapeContentType) bool {
return true
case ScrapeContentTypeGallery:
return true
case ScrapeContentTypeImage:
return true
}

return false
Expand All @@ -204,6 +249,9 @@ func (s autotagScraper) spec() Scraper {
Gallery: &ScraperSpec{
SupportedScrapes: supportedScrapes,
},
Image: &ScraperSpec{
SupportedScrapes: supportedScrapes,
},
}
}

Expand Down