evan/AnthoLume
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

wip 16

Browse Source
This commit is contained in:
Evan Reichard
2026-03-21 20:47:22 -04:00
parent ba919bbde4
commit 4d133994ab
55 changed files with 1901 additions and 264 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
metadata/gbooks.go
View File

@@ -1,6 +1,7 @@
package metadata
import (
"context"
"encoding/json"
"errors"
"fmt"
@@ -40,10 +41,14 @@ const GBOOKS_GBID_INFO_URL string = "https://www.googleapis.com/books/v1/volumes
const GBOOKS_GBID_COVER_URL string = "https://books.google.com/books/content/images/frontcover/%s?fife=w480-h690"
func getGBooksMetadata(metadataSearch MetadataInfo) ([]MetadataInfo, error) {
return getGBooksMetadataWithContext(context.Background(), metadataSearch)
}
func getGBooksMetadataWithContext(ctx context.Context, metadataSearch MetadataInfo) ([]MetadataInfo, error) {
var queryResults []gBooksQueryItem
if metadataSearch.ID != nil {
// Use GBID
resp, err := performGBIDRequest(*metadataSearch.ID)
resp, err := performGBIDRequestWithContext(ctx, *metadataSearch.ID)
if err != nil {
return nil, err
}
@@ -51,7 +56,7 @@ func getGBooksMetadata(metadataSearch MetadataInfo) ([]MetadataInfo, error) {
queryResults = []gBooksQueryItem{*resp}
} else if metadataSearch.ISBN13 != nil {
searchQuery := "isbn:" + *metadataSearch.ISBN13
resp, err := performSearchRequest(searchQuery)
resp, err := performSearchRequestWithContext(ctx, searchQuery)
if err != nil {
return nil, err
}
@@ -59,7 +64,7 @@ func getGBooksMetadata(metadataSearch MetadataInfo) ([]MetadataInfo, error) {
queryResults = resp.Items
} else if metadataSearch.ISBN10 != nil {
searchQuery := "isbn:" + *metadataSearch.ISBN10
resp, err := performSearchRequest(searchQuery)
resp, err := performSearchRequestWithContext(ctx, searchQuery)
if err != nil {
return nil, err
}
@@ -76,7 +81,7 @@ func getGBooksMetadata(metadataSearch MetadataInfo) ([]MetadataInfo, error) {
// Escape & Trim
searchQuery = url.QueryEscape(strings.TrimSpace(searchQuery))
resp, err := performSearchRequest(searchQuery)
resp, err := performSearchRequestWithContext(ctx, searchQuery)
if err != nil {
return nil, err
}
@@ -119,6 +124,10 @@ func getGBooksMetadata(metadataSearch MetadataInfo) ([]MetadataInfo, error) {
}
func saveGBooksCover(gbid string, coverFilePath string, overwrite bool) error {
return saveGBooksCoverWithContext(context.Background(), gbid, coverFilePath, overwrite)
}
func saveGBooksCoverWithContext(ctx context.Context, gbid string, coverFilePath string, overwrite bool) error {
// Validate File Doesn't Exists
_, err := os.Stat(coverFilePath)
if err == nil && !overwrite {
@@ -137,7 +146,14 @@ func saveGBooksCover(gbid string, coverFilePath string, overwrite bool) error {
// Download File
log.Info("Downloading Cover")
coverURL := fmt.Sprintf(GBOOKS_GBID_COVER_URL, gbid)
resp, err := http.Get(coverURL)
req, err := http.NewRequestWithContext(ctx, "GET", coverURL, nil)
if err != nil {
log.Error("Cover URL API Failure")
return errors.New("API Failure")
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
log.Error("Cover URL API Failure")
return errors.New("API Failure")
@@ -156,9 +172,20 @@ func saveGBooksCover(gbid string, coverFilePath string, overwrite bool) error {
}
func performSearchRequest(searchQuery string) (*gBooksQueryResponse, error) {
return performSearchRequestWithContext(context.Background(), searchQuery)
}
func performSearchRequestWithContext(ctx context.Context, searchQuery string) (*gBooksQueryResponse, error) {
apiQuery := fmt.Sprintf(GBOOKS_QUERY_URL, searchQuery)
log.Info("Acquiring Metadata: ", apiQuery)
resp, err := http.Get(apiQuery)
req, err := http.NewRequestWithContext(ctx, "GET", apiQuery, nil)
if err != nil {
log.Error("Google Books Query URL API Failure")
return nil, errors.New("API Failure")
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
log.Error("Google Books Query URL API Failure")
return nil, errors.New("API Failure")
@@ -166,6 +193,7 @@ func performSearchRequest(searchQuery string) (*gBooksQueryResponse, error) {
parsedResp := gBooksQueryResponse{}
err = json.NewDecoder(resp.Body).Decode(&parsedResp)
resp.Body.Close()
if err != nil {
log.Error("Google Books Query API Decode Failure")
return nil, errors.New("API Failure")
@@ -180,10 +208,21 @@ func performSearchRequest(searchQuery string) (*gBooksQueryResponse, error) {
}
func performGBIDRequest(id string) (*gBooksQueryItem, error) {
return performGBIDRequestWithContext(context.Background(), id)
}
func performGBIDRequestWithContext(ctx context.Context, id string) (*gBooksQueryItem, error) {
apiQuery := fmt.Sprintf(GBOOKS_GBID_INFO_URL, id)
log.Info("Acquiring CoverID")
resp, err := http.Get(apiQuery)
req, err := http.NewRequestWithContext(ctx, "GET", apiQuery, nil)
if err != nil {
log.Error("Cover URL API Failure")
return nil, errors.New("API Failure")
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
log.Error("Cover URL API Failure")
return nil, errors.New("API Failure")
@@ -191,6 +230,7 @@ func performGBIDRequest(id string) (*gBooksQueryItem, error) {
parsedResp := gBooksQueryItem{}
err = json.NewDecoder(resp.Body).Decode(&parsedResp)
resp.Body.Close()
if err != nil {
log.Error("Google Books ID API Decode Failure")
return nil, errors.New("API Failure")

15
metadata/metadata.go
View File

@@ -1,6 +1,7 @@
package metadata
import (
"context"
"errors"
"fmt"
"io"
@@ -45,12 +46,17 @@ type MetadataInfo struct {
// Downloads the Google Books cover file and saves it to the provided directory.
func CacheCover(gbid string, coverDir string, documentID string, overwrite bool) (*string, error) {
return CacheCoverWithContext(context.Background(), gbid, coverDir, documentID, overwrite)
}
// CacheCoverWithContext downloads the Google Books cover file and saves it to the provided directory with context support.
func CacheCoverWithContext(ctx context.Context, gbid string, coverDir string, documentID string, overwrite bool) (*string, error) {
// Get Filepath
coverFile := "." + filepath.Clean(fmt.Sprintf("/%s.jpg", documentID))
coverFilePath := filepath.Join(coverDir, coverFile)
// Save Google Books
if err := saveGBooksCover(gbid, coverFilePath, overwrite); err != nil {
if err := saveGBooksCoverWithContext(ctx, gbid, coverFilePath, overwrite); err != nil {
return nil, err
}
@@ -61,9 +67,14 @@ func CacheCover(gbid string, coverDir string, documentID string, overwrite bool)
// Searches source for metadata based on the provided information.
func SearchMetadata(s Source, metadataSearch MetadataInfo) ([]MetadataInfo, error) {
return SearchMetadataWithContext(context.Background(), s, metadataSearch)
}
// SearchMetadataWithContext searches source for metadata based on the provided information with context support.
func SearchMetadataWithContext(ctx context.Context, s Source, metadataSearch MetadataInfo) ([]MetadataInfo, error) {
switch s {
case SOURCE_GBOOK:
return getGBooksMetadata(metadataSearch)
return getGBooksMetadataWithContext(ctx, metadataSearch)
case SOURCE_OLIB:
return nil, errors.New("not implemented")
default: