Implement LoadPhotos and DownloadPhoto

This commit is contained in:
Balakrishnan Balasubramanian 2024-08-06 23:14:06 -04:00
parent 5fee186dcb
commit 12befa0ac3

123
main.go
View File

@ -26,11 +26,12 @@ import (
var ( var (
imagesDir string imagesDir string
collageDir string collageDir string
photosDir string
devMode bool devMode bool
collageNameGen *nameGen collageNameGen *nameGen
imagesDirFs fs.FS imagesDirFs fs.FS
listenAddr string listenAddr string
photoPrismURL string photoPrismURL *url.URL
photoPrismToken string photoPrismToken string
//go:embed web //go:embed web
@ -38,19 +39,32 @@ var (
) )
func main() { func main() {
var ppURL string
flag.StringVar(&imagesDir, "images-dir", "images", "Sets the images dir") flag.StringVar(&imagesDir, "images-dir", "images", "Sets the images dir")
flag.StringVar(&collageDir, "collages-dir", "collages", "Sets the collages dir") flag.StringVar(&collageDir, "collages-dir", "collages", "Sets the collages dir")
flag.StringVar(&photosDir, "photos-dir", "photos", "Cache directory for downloaded photos")
flag.BoolVar(&devMode, "dev", false, "Serve local assets during development") flag.BoolVar(&devMode, "dev", false, "Serve local assets during development")
flag.StringVar(&listenAddr, "addr", "127.0.0.1:8767", "Web listen address, see https://pkg.go.dev/go.balki.me/anyhttp#readme-address-syntax") flag.StringVar(&listenAddr, "addr", "127.0.0.1:8767", "Web listen address, see https://pkg.go.dev/go.balki.me/anyhttp#readme-address-syntax")
flag.StringVar(&photoPrismURL, "pp-url", "", "Base url for photoprism") flag.StringVar(&ppURL, "pp-url", "", "Base url for photoprism")
flag.StringVar(&photoPrismToken, "pp-token", "", "API token for photoprism") flag.StringVar(&photoPrismToken, "pp-token", "", "API token for photoprism")
flag.Parse() flag.Parse()
photoPrismURL = func() *url.URL {
photoPrismURL, err := url.Parse(ppURL)
if err != nil {
panic(err)
}
return photoPrismURL
}()
copy := *photoPrismURL
fmt.Println(copy)
collageNameGen = NewNameGen() collageNameGen = NewNameGen()
imagesDirFs = os.DirFS(imagesDir) imagesDirFs = os.DirFS(imagesDir)
imagesURLPath := "images" imagesURLPath := "images"
collagesPath := "collages" collagesPath := "collages"
photosPath := "photos"
addFileServer := func(path, dir string) { addFileServer := func(path, dir string) {
httpFileServer := http.FileServer(http.Dir(dir)) httpFileServer := http.FileServer(http.Dir(dir))
@ -59,6 +73,7 @@ func main() {
addFileServer(imagesURLPath, imagesDir) addFileServer(imagesURLPath, imagesDir)
addFileServer(collagesPath, collageDir) addFileServer(collagesPath, collageDir)
addFileServer(photosPath, photosDir)
if devMode { if devMode {
httpFileServer := http.FileServer(http.Dir("web")) httpFileServer := http.FileServer(http.Dir("web"))
@ -170,41 +185,113 @@ type Album struct {
} }
func GetAlbums() ([]Album, error) { func GetAlbums() ([]Album, error) {
u, err := url.Parse(photoPrismURL) albumURL := func() string {
if err != nil { u := *photoPrismURL
return nil, err u.Path = "/api/v1/albums"
} v := url.Values{}
v.Add("count", "20")
v.Add("type", "album")
u.RawQuery = v.Encode()
return u.String()
}()
u.Path = "/api/v1/albums" req, err := http.NewRequest("GET", albumURL, nil)
v := url.Values{}
v.Add("count", "10")
v.Add("type", "album")
u.RawQuery = v.Encode()
req, err := http.NewRequest("GET", u.String(), nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", photoPrismToken)) req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", photoPrismToken))
req.Header.Set("Content-Type", "application/json")
resp, err := http.DefaultClient.Do(req) resp, err := http.DefaultClient.Do(req)
if err != nil { if err != nil {
return nil, err return nil, err
} }
albums := []Album{}
respBytes, err := io.ReadAll(resp.Body) respBytes, err := io.ReadAll(resp.Body)
if err != nil { if err != nil {
return nil, err return nil, err
} }
albums := []Album{}
err = json.Unmarshal(respBytes, &albums) err = json.Unmarshal(respBytes, &albums)
if err != nil { if err != nil {
return nil, err return nil, err
} }
// List albums
// yq -oc '[.[] | pick(["UID", "Title"])]' albums.json | pandoc -fcsv -tplain
return albums, nil return albums, nil
} }
type Photo struct {
Hash string `json:"Hash"`
FileUID string `json:"FileUID"`
}
func LoadPhotos(albumID string) ([]Photo, error) {
loadPhotosURL := func() string {
u := *photoPrismURL
u.Path = "/api/v1/photos"
v := url.Values{}
v.Add("count", "50")
v.Add("s", albumID)
v.Add("merged", "true")
v.Add("video", "false")
u.RawQuery = v.Encode()
return u.String()
}()
req, err := http.NewRequest("GET", loadPhotosURL, nil)
if err != nil {
return nil, err
}
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", photoPrismToken))
resp, err := http.DefaultClient.Do(req)
if err != nil {
return nil, err
}
respBytes, err := io.ReadAll(resp.Body)
if err != nil {
return nil, err
}
photos := []Photo{}
err = json.Unmarshal(respBytes, &photos)
if err != nil {
return nil, err
}
return photos, nil
}
func DownloadPhoto(photo *Photo) (string, error) {
photoPath := path.Join(photosDir, fmt.Sprintf("%s.jpg", photo.FileUID))
_, err := os.Stat(photoPath)
if err == nil {
return photoPath, nil
} else if !os.IsNotExist(err) {
return "", err
}
out, err := os.Create(photoPath)
if err != nil {
return "", fmt.Errorf("failed to create collage output file, err: %w", err)
}
defer out.Close()
downloadPhotoURL := func() string {
u := *photoPrismURL
u.Path = fmt.Sprintf("/api/v1/dl/%s", photo.Hash)
return u.String()
}()
req, err := http.NewRequest("GET", downloadPhotoURL, nil)
if err != nil {
return "", err
}
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", photoPrismToken))
resp, err := http.DefaultClient.Do(req)
if err != nil {
return "", err
}
_, err = io.Copy(out, resp.Body)
if err != nil {
return "", err
}
return photoPath, nil
}