109 lines
2.1 KiB
Go
109 lines
2.1 KiB
Go
package models
|
|
|
|
import (
|
|
"database/sql"
|
|
"log"
|
|
"net/url"
|
|
"os"
|
|
"path"
|
|
)
|
|
|
|
type Photo struct {
|
|
PhotoID int
|
|
Title string
|
|
Path string
|
|
AlbumId int
|
|
ExifId *int
|
|
}
|
|
|
|
func (p *Photo) ID() int {
|
|
return p.PhotoID
|
|
}
|
|
|
|
type PhotoPurpose string
|
|
|
|
const (
|
|
PhotoThumbnail PhotoPurpose = "thumbnail"
|
|
PhotoHighRes PhotoPurpose = "high-res"
|
|
PhotoOriginal PhotoPurpose = "original"
|
|
)
|
|
|
|
type PhotoURL struct {
|
|
UrlID int
|
|
PhotoId int
|
|
PhotoName string
|
|
Width int
|
|
Height int
|
|
Purpose PhotoPurpose
|
|
ContentType string
|
|
}
|
|
|
|
func NewPhotoFromRow(row *sql.Row) (*Photo, error) {
|
|
photo := Photo{}
|
|
|
|
if err := row.Scan(&photo.PhotoID, &photo.Title, &photo.Path, &photo.AlbumId, &photo.ExifId); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &photo, nil
|
|
}
|
|
|
|
func NewPhotosFromRows(rows *sql.Rows) ([]*Photo, error) {
|
|
photos := make([]*Photo, 0)
|
|
|
|
for rows.Next() {
|
|
var photo Photo
|
|
if err := rows.Scan(&photo.PhotoID, &photo.Title, &photo.Path, &photo.AlbumId, &photo.ExifId); err != nil {
|
|
return nil, err
|
|
}
|
|
photos = append(photos, &photo)
|
|
}
|
|
|
|
rows.Close()
|
|
|
|
return photos, nil
|
|
}
|
|
|
|
func (p *PhotoURL) URL() string {
|
|
|
|
publicUrl := os.Getenv("PUBLIC_ENDPOINT")
|
|
if publicUrl == "" {
|
|
publicUrl = os.Getenv("API_ENDPOINT")
|
|
}
|
|
|
|
imageUrl, err := url.Parse(publicUrl)
|
|
if err != nil {
|
|
log.Println("Endpoint url is not properly configured, make sure the PUBLIC_ENDPOINT AND API_ENDPOINT environment variables are set correctly")
|
|
return p.PhotoName
|
|
}
|
|
|
|
imageUrl.Path = path.Join(imageUrl.Path, "api", "photo", p.PhotoName)
|
|
return imageUrl.String()
|
|
}
|
|
|
|
func NewPhotoURLFromRow(row *sql.Row) (*PhotoURL, error) {
|
|
url := PhotoURL{}
|
|
|
|
if err := row.Scan(&url.UrlID, &url.PhotoId, &url.PhotoName, &url.Width, &url.Height, &url.Purpose, &url.ContentType); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &url, nil
|
|
}
|
|
|
|
func NewPhotoURLFromRows(rows *sql.Rows) ([]*PhotoURL, error) {
|
|
urls := make([]*PhotoURL, 0)
|
|
|
|
for rows.Next() {
|
|
var url PhotoURL
|
|
if err := rows.Scan(&url.UrlID, &url.PhotoId, &url.PhotoName, &url.Width, &url.Height, &url.Purpose, &url.ContentType); err != nil {
|
|
return nil, err
|
|
}
|
|
urls = append(urls, &url)
|
|
}
|
|
|
|
rows.Close()
|
|
|
|
return urls, nil
|
|
}
|