package status

import (
	"context"
	"errors"
	"fmt"
	"log"
	"madsky.ru/go-finance/internal/database"
	"madsky.ru/go-finance/internal/status"
)

type Repository interface {
	Find(ctx context.Context) ([]*status.Status, error)
	FindOne(ctx context.Context, id uint64) (*status.Status, error)
	Create(ctx context.Context, dto *status.CreateStatusDTO) (*status.Status, error)
	Update(ctx context.Context, id uint64, issue *status.Status) error
	Remove(ctx context.Context, id uint64) (uint64, error)
}

type repository struct {
	client database.Client
}

func NewRepository(client database.Client) Repository {
	return &repository{
		client: client,
	}
}

func (r *repository) Find(ctx context.Context) ([]*status.Status, error) {
	query := "select id, name, description, position from statuses order by id asc"

	rows, err := r.client.Query(ctx, query)
	if err != nil {
		log.Println("rows", err)
		return nil, err
	}

	statuses := make([]*status.Status, 0)

	for rows.Next() {
		var n status.Status

		err = rows.Scan(&n.ID, &n.Name, &n.Description, &n.Position)
		if err != nil {
			log.Println("scan", err)
			return nil, err
		}
		statuses = append(statuses, &n)
	}

	if err = rows.Err(); err != nil {
		log.Println("rows err", err)
		return nil, err
	}

	return statuses, nil
}

func (r *repository) FindOne(ctx context.Context, id uint64) (*status.Status, error) {
	query := "select id, name, description, position from statuses where id = $1"

	var s status.Status
	if err := r.client.QueryRow(ctx, query, id).Scan(&s.ID, &s.Name, &s.Description, &s.Position); err != nil {
		fmt.Println(err)
		return nil, err
	}

	return &s, nil
}

func (r *repository) Create(ctx context.Context, dto *status.CreateStatusDTO) (*status.Status, error) {
	q := "insert into statuses (name, description) values ($1, $2) returning id, name, description, position"

	var s status.Status

	if err := r.client.QueryRow(ctx, q, dto.Name, dto.Description).Scan(&s.ID, &s.Name, &s.Description, &s.Position); err != nil {
		fmt.Println(fmt.Sprintf("error %v", err))
		return nil, err
	}
	return &s, nil
}

func (r *repository) Update(ctx context.Context, id uint64, issue *status.Status) error {
	//TODO implement me
	panic("implement me")
}

func (r *repository) Remove(ctx context.Context, id uint64) (uint64, error) {
	q := "delete from statuses where id=$1"

	tag, err := r.client.Exec(ctx, q, id)
	if err != nil {
		log.Println("exec error", err)
		return 0, err
	}

	rowsAffected := tag.RowsAffected()

	if rowsAffected == 0 {
		return 0, errors.New("status not found")
	}

	return uint64(rowsAffected), nil
}