package store

import (
	"context"
	"time"

	"github.com/google/uuid"
)

type Image struct {
	ID          uuid.UUID
	BlogID      int64
	Filename    string
	ContentType string
	Size        int
	Data        []byte // only populated by ImageData
	CreatedAt   time.Time
}

func (s *Store) CreateImage(ctx context.Context, blogID int64, filename, contentType string, data []byte) (*Image, error) {
	img := &Image{ID: uuid.New(), BlogID: blogID, Filename: filename, ContentType: contentType, Size: len(data)}
	err := s.db.QueryRow(ctx, `INSERT INTO images (id, blog_id, filename, content_type, size, data) VALUES ($1,$2,$3,$4,$5,$6) RETURNING created_at`,
		img.ID, blogID, filename, contentType, len(data), data).Scan(&img.CreatedAt)
	return img, err
}

func (s *Store) ListImages(ctx context.Context, blogID int64) ([]Image, error) {
	rows, err := s.db.Query(ctx, `SELECT id, blog_id, filename, content_type, size, created_at FROM images WHERE blog_id=$1 ORDER BY created_at DESC`, blogID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	var out []Image
	for rows.Next() {
		var i Image
		if err := rows.Scan(&i.ID, &i.BlogID, &i.Filename, &i.ContentType, &i.Size, &i.CreatedAt); err != nil {
			return nil, err
		}
		out = append(out, i)
	}
	return out, rows.Err()
}

// ImageData loads an image including its bytes.
func (s *Store) ImageData(ctx context.Context, id uuid.UUID) (*Image, error) {
	var i Image
	err := s.db.QueryRow(ctx, `SELECT id, blog_id, filename, content_type, size, data, created_at FROM images WHERE id=$1`, id).
		Scan(&i.ID, &i.BlogID, &i.Filename, &i.ContentType, &i.Size, &i.Data, &i.CreatedAt)
	if err != nil {
		return nil, wrap(err)
	}
	return &i, nil
}

func (s *Store) DeleteImage(ctx context.Context, blogID int64, id uuid.UUID) error {
	_, err := s.db.Exec(ctx, `DELETE FROM images WHERE blog_id=$1 AND id=$2`, blogID, id)
	return err
}