iu
50dff63ece
- rename Go module, binary, systemd unit, PG_DATABASE default, frontend package and docs from shared-album to galo - add theme-aware halo emblem and favicon (new ui-branding capability) - add make build-release (versioned, stripped, static binary) and run the binary from the repo root
91 lines
2.6 KiB
Go
91 lines
2.6 KiB
Go
package sharelinks
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"time"
|
|
|
|
"galo/internal/models"
|
|
"galo/internal/sqliteutil"
|
|
)
|
|
|
|
// SQLiteStore is the SQLite implementation of Repository.
|
|
type SQLiteStore struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewSQLiteStore(db *sql.DB) *SQLiteStore { return &SQLiteStore{db: db} }
|
|
|
|
func (s *SQLiteStore) Create(ctx context.Context, l *models.ShareLink) error {
|
|
var createdAt string
|
|
err := s.db.QueryRowContext(ctx,
|
|
`INSERT INTO share_links (id, album_id, token, created_by, expires_at)
|
|
VALUES ($1, $2, $3, $4, $5)
|
|
RETURNING created_at`,
|
|
l.ID, l.AlbumID, l.Token, l.CreatedBy, sqliteutil.FormatTime(l.ExpiresAt),
|
|
).Scan(&createdAt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
l.CreatedAt = sqliteutil.ParseTime(createdAt)
|
|
return nil
|
|
}
|
|
|
|
func (s *SQLiteStore) ListByAlbum(ctx context.Context, albumID string) ([]models.ShareLink, error) {
|
|
rows, err := s.db.QueryContext(ctx,
|
|
`SELECT id, album_id, token, created_by, created_at, expires_at, revoked_at
|
|
FROM share_links WHERE album_id = $1 ORDER BY created_at DESC`, albumID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
out := []models.ShareLink{}
|
|
for rows.Next() {
|
|
var l models.ShareLink
|
|
var createdAt, expiresAt string
|
|
var revokedAt sql.NullString
|
|
if err := rows.Scan(&l.ID, &l.AlbumID, &l.Token, &l.CreatedBy, &createdAt, &expiresAt, &revokedAt); err != nil {
|
|
return nil, err
|
|
}
|
|
l.CreatedAt = sqliteutil.ParseTime(createdAt)
|
|
l.ExpiresAt = sqliteutil.ParseTime(expiresAt)
|
|
l.RevokedAt = sqliteutil.ParseNullTime(revokedAt)
|
|
out = append(out, l)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func (s *SQLiteStore) GetByToken(ctx context.Context, token string) (*models.ShareLink, error) {
|
|
var l models.ShareLink
|
|
var createdAt, expiresAt string
|
|
var revokedAt sql.NullString
|
|
err := s.db.QueryRowContext(ctx,
|
|
`SELECT id, album_id, token, created_by, created_at, expires_at, revoked_at
|
|
FROM share_links WHERE token = $1`, token).
|
|
Scan(&l.ID, &l.AlbumID, &l.Token, &l.CreatedBy, &createdAt, &expiresAt, &revokedAt)
|
|
if err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil, ErrNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
l.CreatedAt = sqliteutil.ParseTime(createdAt)
|
|
l.ExpiresAt = sqliteutil.ParseTime(expiresAt)
|
|
l.RevokedAt = sqliteutil.ParseNullTime(revokedAt)
|
|
return &l, nil
|
|
}
|
|
|
|
func (s *SQLiteStore) Revoke(ctx context.Context, id, albumID string) error {
|
|
res, err := s.db.ExecContext(ctx,
|
|
`UPDATE share_links SET revoked_at = $3 WHERE id = $1 AND album_id = $2 AND revoked_at IS NULL`,
|
|
id, albumID, sqliteutil.FormatTime(time.Now()))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if n, _ := res.RowsAffected(); n == 0 {
|
|
return ErrNotFound
|
|
}
|
|
return nil
|
|
}
|