package repositories

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

	"github.com/rs/zerolog/log"
)

func CreateComponent(ctx context.Context, queries *db.Queries, body models.Component) (models.Component, error) {
	result, err := queries.CreateComponent(ctx, *mapper.ToModelComponent(&body))
	if err != nil {
		return models.Component{}, err
	}
	return *mapper.ToDomainComponent(result), nil
}

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

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

func UpdateComponent(ctx context.Context, queries *db.Queries, body models.Component) (models.Component, error) {
	result, err := queries.UpdateComponent(ctx, *mapper.ToUpdateModelComponent(&body))
	log.Info().Any("component", result).Err(err).Msg("Updating component")
	if err != nil {
		return models.Component{}, err
	}
	return *mapper.ToDomainComponent(result), nil
}

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