package repositories

import (
	"context"
	"wm-backend/internal/mapper"
	"wm-backend/internal/models"
	db "wm-backend/sqlc_gen"
)

func CreateInvoice(ctx context.Context, queries *db.Queries, body models.Invoice) (models.Invoice, error) {
	result, err := queries.CreateInvoice(ctx, *mapper.ToModelInvoice(&body))
	if err != nil {
		return models.Invoice{}, err
	}
	return *mapper.ToDomainInvoice(result), nil
}

func GetInvoiceByID(ctx context.Context, queries *db.Queries, id int64) (models.Invoice, error) {
	result, err := queries.GetInvoiceByID(ctx, id)
	if err != nil {
		return models.Invoice{}, err
	}
	return *mapper.ToDomainInvoice(result), nil
}

func ListInvoices(ctx context.Context, queries *db.Queries) ([]models.Invoice, error) {
	results, err := queries.ListInvoices(ctx)
	if err != nil {
		return nil, err
	}
	var items []models.Invoice
	for _, r := range results {
		items = append(items, *mapper.ToDomainInvoice(r))
	}
	return items, nil
}

func UpdateInvoice(ctx context.Context, queries *db.Queries, body models.Invoice) (models.Invoice, error) {
	result, err := queries.UpdateInvoice(ctx, *mapper.ToUpdateModelInvoice(&body))
	if err != nil {
		return models.Invoice{}, err
	}
	return *mapper.ToDomainInvoice(result), nil
}

func DeleteInvoice(ctx context.Context, queries *db.Queries, id int64) (int64, error) {
	rowsAffected, err := queries.DeleteInvoice(ctx, id)
	if err != nil {
		return rowsAffected, err
	}
	return rowsAffected, nil
}