package store

import (
	"errors"
	"log"

	"git.bronku.xyz/bronku/cake-order-tracker/models"
)

func (s *Store) GetCake(id int) (models.Cake, error) {
	if id <= 0 {
		return models.Cake{}, errors.New("invalid cake id")
	}

	row := s.db.QueryRow("select id, name, price from cake where id = ?;", id)
	var out models.Cake
	err := row.Scan(&out.ID, &out.Name, &out.Price)
	if err != nil {
		return models.Cake{}, errors.New("cake not found")
	}
	out.Tags, err = s.GetCakeTags(id)
	return out, err
}

func (s *Store) GetCakes() ([]models.Cake, error) {
	rows, err := s.db.Query("select id, name, price from cake order by id;")
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var out []models.Cake
	for rows.Next() {
		var cake models.Cake
		if err := rows.Scan(&cake.ID, &cake.Name, &cake.Price); err != nil {
			return nil, err
		}
		tags, err := s.GetCakeTags(cake.ID)
		if err != nil {
			return nil, err
		}
		cake.Tags = tags
		out = append(out, cake)
	}
	return out, rows.Err()
}

func (s *Store) SaveCake(newCake models.Cake) (int, error) {
	if newCake.ID != 0 {
		q := "update cake set name = ?, price = ? where id = ?;"
		_, err := s.db.Exec(q, newCake.Name, newCake.Price, newCake.ID)
		if err != nil {
			return 0, err
		}
		log.Printf("UPDATE cake id=%d name=%q price=%d", newCake.ID, newCake.Name, newCake.Price)
		return newCake.ID, s.SaveCakeTags(newCake.ID, tagIDs(newCake.Tags))
	}

	q := "insert into cake(name, price) values (?, ?);"
	result, err := s.db.Exec(q, newCake.Name, newCake.Price)
	if err != nil {
		return 0, err
	}

	id, err := result.LastInsertId()
	if err != nil {
		return 0, err
	}
	newCake.ID = int(id)
	log.Printf("INSERT cake id=%d name=%q price=%d", newCake.ID, newCake.Name, newCake.Price)
	return newCake.ID, s.SaveCakeTags(newCake.ID, tagIDs(newCake.Tags))
}

func tagIDs(tags []models.Tag) []int {
	ids := make([]int, len(tags))
	for i, t := range tags {
		ids[i] = t.ID
	}
	return ids
}