package model

import (
	"database/sql"
	"encoding/json"
	"errors"
	"fmt"
)

type Recipe struct {
	Id          string `json:"id"`
	Title       string `json:"title"`
	Portions    string `json:"portions"`
	Url         string `json:"url"`
	Notes       string `json:"notes"`
	Created     string `json:"created"`
	LastChanged string `json:"last_changed"`
	Steps       []Step `json:"steps"`
}

func (r Recipe) String() string {
	b, _ := json.MarshalIndent(r, "", "    ")
	return string(b)
}

func (r *Recipe) Create() error {
	tx, err := db.Begin()
	if err != nil {
		return err
	}

	cmd := `
INSERT INTO recipes
    (title, portions, url, notes, created, last_changed)
VALUES
    (?, ?, ?, ?, ?, ?)
`

	result, err := tx.Exec(cmd, r.Title, r.Portions, r.Url, r.Notes, r.Created,
		r.LastChanged)
	if err != nil {
		rollback(tx)
		return err
	}

	id, err := result.LastInsertId()
	if err != nil {
		rollback(tx)
		return err
	}

	r.Id = fmt.Sprint(id)

	err = r.CreateSteps(tx)
	if err != nil {
		rollback(tx)
		return err
	}

	return tx.Commit()
}

func (r *Recipe) Read() error {
	tx, err := db.Begin()
	if err != nil {
		return err
	}

	cmd := `
SELECT id, title, portions, url, notes, created, last_changed
FROM recipes
WHERE id = ?
`

	rows, err := tx.Query(cmd, r.Id)
	if err != nil {
		rollback(tx)
		return err
	}
	defer rows.Close()

	if !rows.Next() {
		rollback(tx)
		return sql.ErrNoRows
	}

	err = rows.Scan(
		&r.Id,
		&r.Title,
		&r.Portions,
		&r.Url,
		&r.Notes,
		&r.Created,
		&r.LastChanged,
	)
	if err != nil {
		rollback(tx)
		return err
	}

	err = r.ReadSteps(tx)
	if err != nil {
		rollback(tx)
		return err
	}

	return tx.Commit()
}

func (r *Recipe) Update() error {
	tx, err := db.Begin()
	if err != nil {
		return err
	}

	err = r.UpdateSteps(tx)
	if err != nil {
		rollback(tx)
		return err
	}

	query := `UPDATE
    recipes
SET
    title = ?,
	portions = ?,
	url = ?,
	notes = ?,
	last_changed = ?
WHERE
    id = ?`

	res, err := tx.Exec(query, r.Title, r.Portions, r.Url, r.Notes,
		r.LastChanged, r.Id)
	if err != nil {
		rollback(tx)
		return err
	}

	affected, err := res.RowsAffected()
	if err != nil {
		rollback(tx)
		return err
	}
	if affected != 1 {
		rollback(tx)
		return fmt.Errorf("Recipe update affected %d rows instead of 1", affected)
	}

	return tx.Commit()
}

func (r *Recipe) Delete() error {
	tx, err := db.Begin()
	if err != nil {
		return err
	}

	err = r.DeleteSteps(tx)
	if err != nil {
		rollback(tx)
		return err
	}

	query := `
DELETE FROM
    recipes
WHERE
    id = ?
`

	result, err := tx.Exec(query, r.Id)
	if err != nil {
		rollback(tx)
		return err
	}

	rows, err := result.RowsAffected()
	if rows != 1 {
		rollback(tx)
		return errors.New("Recipe deletion did not affect exactly one row")
	}

	return tx.Commit()
}

func RecipeTestData() []Recipe {
	return []Recipe{
		{
			Id:          "1",
			Title:       "Pancakes",
			Portions:    "4",
			Url:         "https://example.org",
			Notes:       "Very fluffy",
			Created:     "",
			LastChanged: "",
			Steps: []Step{
				{Text: "Stir the dough"},
				{Text: "Heat up pan"},
				{Text: "Make pancakes!"},
			},
		},
		{
			Id:          "2",
			Title:       "Burger",
			Portions:    "2",
			Url:         "https://xengineering.eu/git/ceres",
			Notes:       "Delicious!",
			Created:     "",
			LastChanged: "",
			Steps: []Step{},
		},
	}
}