82ded6a9c7
The adult kind scanned and browsed but could never be enriched, because no provider claimed it. ThePornDB does: it is the catalogue the Stash ecosystem uses for scenes and movies, and it is keyed the same way IGDB and TMDB are, so it registers only when VELLUM_TPDB_API_KEY is set and otherwise logs the env var at boot. The API shape came from the live OpenAPI spec rather than the docs, which is where the search parameter lives: it is `parse`, not `q` or `title`, and the endpoint is /scenes with a `data` envelope. Not verified against the live API: it needs a token I do not have, so the tests run against httptest only. The field names are taken from the published schema, which is the best available substitute. The provider also does not use ThePornDB's hash lookup, which is the more accurate way to match: it needs a file hash computed at scan time, and matching by parsed title and date works today without that.
98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package providers
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// ThePornDB matches adult scenes. It requires a free API key, so the provider
|
|
// is only registered when one is configured.
|
|
type ThePornDB struct {
|
|
APIKey string
|
|
BaseURL string
|
|
Client *http.Client
|
|
}
|
|
|
|
// NewThePornDB builds the provider. An empty key yields a provider that finds
|
|
// nothing, so a misconfiguration degrades rather than erroring.
|
|
func NewThePornDB(apiKey string) *ThePornDB {
|
|
return &ThePornDB{
|
|
APIKey: apiKey,
|
|
BaseURL: "https://api.theporndb.net",
|
|
Client: &http.Client{Timeout: 15 * time.Second},
|
|
}
|
|
}
|
|
|
|
func (p *ThePornDB) Name() string { return "theporndb" }
|
|
|
|
func (p *ThePornDB) Kinds() []string { return []string{"adult"} }
|
|
|
|
// tpdbResponse is the subset of a scene search payload we read.
|
|
type tpdbResponse struct {
|
|
Data []struct {
|
|
ID string `json:"id"`
|
|
Title string `json:"title"`
|
|
Date string `json:"date"`
|
|
Description string `json:"description"`
|
|
Image string `json:"image"`
|
|
} `json:"data"`
|
|
}
|
|
|
|
func (p *ThePornDB) Search(ctx context.Context, q Query) ([]Candidate, error) {
|
|
title := strings.TrimSpace(q.Title)
|
|
if p.APIKey == "" || title == "" {
|
|
return nil, nil
|
|
}
|
|
|
|
params := url.Values{}
|
|
params.Set("parse", title)
|
|
params.Set("per_page", "5")
|
|
if q.Year > 0 {
|
|
params.Set("year", strconv.Itoa(q.Year))
|
|
}
|
|
|
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet,
|
|
strings.TrimRight(p.BaseURL, "/")+"/scenes?"+params.Encode(), nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.Header.Set("Authorization", "Bearer "+p.APIKey)
|
|
req.Header.Set("Accept", "application/json")
|
|
|
|
resp, err := p.Client.Do(req)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("theporndb: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
if resp.StatusCode != http.StatusOK {
|
|
return nil, fmt.Errorf("theporndb: status %d", resp.StatusCode)
|
|
}
|
|
|
|
var parsed tpdbResponse
|
|
if err := json.NewDecoder(resp.Body).Decode(&parsed); err != nil {
|
|
return nil, fmt.Errorf("theporndb: decode: %w", err)
|
|
}
|
|
|
|
out := make([]Candidate, 0, len(parsed.Data))
|
|
for _, s := range parsed.Data {
|
|
if s.Title == "" {
|
|
continue
|
|
}
|
|
out = append(out, Candidate{
|
|
Provider: p.Name(),
|
|
// Namespaced so a scene id cannot collide with another provider's.
|
|
ExternalID: "scene/" + s.ID,
|
|
Title: s.Title,
|
|
Year: parseYear(s.Date),
|
|
Description: stripHTML(s.Description),
|
|
CoverURL: s.Image,
|
|
})
|
|
}
|
|
return out, nil
|
|
}
|