81 lines
1.5 KiB
Go
81 lines
1.5 KiB
Go
package models
|
|
|
|
import (
|
|
"database/sql"
|
|
"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"
|
|
)
|
|
|
|
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)
|
|
}
|
|
|
|
return photos, nil
|
|
}
|
|
|
|
func (p *PhotoURL) URL() string {
|
|
imageUrl, err := url.Parse(os.Getenv("API_ENDPOINT"))
|
|
if err != nil {
|
|
return path.Join("/photo", p.PhotoName)
|
|
}
|
|
imageUrl.Path = path.Join(imageUrl.Path, "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
|
|
}
|