package repositories

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

func CreateComponentCode(ctx context.Context, queries *db.Queries, body models.ComponentCode) (models.ComponentCode, error) {
	result, err := queries.CreateComponentCode(ctx, *mapper.ToModelComponentCode(&body))
	if err != nil {
		return models.ComponentCode{}, err
	}
	return *mapper.ToDomainComponentCode(result), nil
}

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

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

func UpdateComponentCode(ctx context.Context, queries *db.Queries, body models.ComponentCode) (models.ComponentCode, error) {
	result, err := queries.UpdateComponentCode(ctx, *mapper.ToUpdateModelComponentCode(&body))
	if err != nil {
		return models.ComponentCode{}, err
	}
	return *mapper.ToDomainComponentCode(result), nil
}

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