package graph

import (
	"context"

	"forge.cadoles.com/Cadoles/daddy/internal/session"

	"forge.cadoles.com/Cadoles/daddy/internal/model"
	"github.com/pkg/errors"
)

func handleUserProfile(ctx context.Context) (*model.User, error) {
	db, err := getDB(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	userEmail, err := session.UserEmail(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	repo := model.NewUserRepository(db)

	user, err := repo.FindUserByEmail(ctx, userEmail)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return user, nil
}

func handleUpdateUserProfile(ctx context.Context, changes model.ProfileChanges) (*model.User, error) {
	db, err := getDB(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	userEmail, err := session.UserEmail(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	repo := model.NewUserRepository(db)

	userChanges := &model.User{}

	if changes.Name != nil {
		userChanges.Name = changes.Name
	}

	user, err := repo.UpdateUserByEmail(ctx, userEmail, userChanges)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return user, nil
}