package graph

import (
	"context"
	"strconv"

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

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

	repo := model.NewWorkgroupRepository(db)

	workgroups, err := repo.FindWorkgroups(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return workgroups, nil
}

func handleJoinWorkgroup(ctx context.Context, rawWorkgroupID string) (*model.Workgroup, error) {
	workgroupID, err := parseWorkgroupID(rawWorkgroupID)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	user, db, err := getSessionUser(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	repo := model.NewWorkgroupRepository(db)

	workgroup, err := repo.AddUserToWorkgroup(ctx, user.ID, workgroupID)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return workgroup, nil
}

func handleLeaveWorkgroup(ctx context.Context, rawWorkgroupID string) (*model.Workgroup, error) {
	workgroupID, err := parseWorkgroupID(rawWorkgroupID)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	user, db, err := getSessionUser(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	repo := model.NewWorkgroupRepository(db)

	workgroup, err := repo.RemoveUserFromWorkgroup(ctx, user.ID, workgroupID)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return workgroup, nil
}

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

	repo := model.NewWorkgroupRepository(db)

	workgroup, err := repo.CreateWorkgroup(ctx, changes)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return workgroup, nil
}

func handleCloseWorkgroup(ctx context.Context, rawWorkgroupID string) (*model.Workgroup, error) {
	workgroupID, err := parseWorkgroupID(rawWorkgroupID)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	db, err := getDB(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	repo := model.NewWorkgroupRepository(db)

	workgroup, err := repo.CloseWorkgroup(ctx, workgroupID)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return workgroup, nil
}

func handleUpdateWorkgroup(ctx context.Context, rawWorkgroupID string, changes model.WorkgroupChanges) (*model.Workgroup, error) {
	workgroupID, err := parseWorkgroupID(rawWorkgroupID)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	db, err := getDB(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	repo := model.NewWorkgroupRepository(db)

	workgroup, err := repo.UpdateWorkgroup(ctx, workgroupID, changes)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return workgroup, nil
}

func parseWorkgroupID(workgroupID string) (uint, error) {
	workgroupID64, err := strconv.ParseUint(workgroupID, 10, 32)
	if err != nil {
		return 0, errors.WithStack(err)
	}

	return uint(workgroupID64), nil
}