package repo

import (
	"context"

	"github.com/google/uuid"
	"github.com/hay-kot/homebox/backend/ent"
	"github.com/hay-kot/homebox/backend/ent/group"
	"github.com/hay-kot/homebox/backend/ent/label"
	"github.com/hay-kot/homebox/backend/internal/types"
)

type LabelRepository struct {
	db *ent.Client
}

func (r *LabelRepository) Get(ctx context.Context, ID uuid.UUID) (*ent.Label, error) {
	return r.db.Label.Query().
		Where(label.ID(ID)).
		WithGroup().
		WithItems().
		Only(ctx)
}

func (r *LabelRepository) GetAll(ctx context.Context, groupId uuid.UUID) ([]*ent.Label, error) {
	return r.db.Label.Query().
		Where(label.HasGroupWith(group.ID(groupId))).
		WithGroup().
		All(ctx)
}

func (r *LabelRepository) Create(ctx context.Context, groupdId uuid.UUID, data types.LabelCreate) (*ent.Label, error) {
	label, err := r.db.Label.Create().
		SetName(data.Name).
		SetDescription(data.Description).
		SetColor(data.Color).
		SetGroupID(groupdId).
		Save(ctx)

	label.Edges.Group = &ent.Group{ID: groupdId} // bootstrap group ID
	return label, err
}

func (r *LabelRepository) Update(ctx context.Context, data types.LabelUpdate) (*ent.Label, error) {
	_, err := r.db.Label.UpdateOneID(data.ID).
		SetName(data.Name).
		SetDescription(data.Description).
		SetColor(data.Color).
		Save(ctx)

	if err != nil {
		return nil, err
	}

	return r.Get(ctx, data.ID)
}

func (r *LabelRepository) Delete(ctx context.Context, id uuid.UUID) error {
	return r.db.Label.DeleteOneID(id).Exec(ctx)
}