// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

package console_test

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.probo.inc/probo/e2e/internal/factory"
	"go.probo.inc/probo/e2e/internal/testutil"
)

func TestTask_Assign(t *testing.T) {
	t.Parallel()
	owner := testutil.NewClient(t, testutil.RoleOwner)

	// Create measure and task
	measureID := factory.NewMeasure(owner).Create()
	taskID := factory.NewTask(owner, measureID).Create()
	peopleID := factory.NewPeople(owner).WithFullName("Task Assignee").Create()

	query := `
		mutation AssignTask($input: AssignTaskInput!) {
			assignTask(input: $input) {
				task {
					id
					assignedTo {
						id
						fullName
					}
				}
			}
		}
	`

	var result struct {
		AssignTask struct {
			Task struct {
				ID         string `json:"id"`
				AssignedTo struct {
					ID       string `json:"id"`
					FullName string `json:"fullName"`
				} `json:"assignedTo"`
			} `json:"task"`
		} `json:"assignTask"`
	}

	err := owner.Execute(query, map[string]any{
		"input": map[string]any{
			"taskId":       taskID,
			"assignedToId": peopleID,
		},
	}, &result)
	require.NoError(t, err)

	assert.Equal(t, taskID, result.AssignTask.Task.ID)
	assert.Equal(t, peopleID, result.AssignTask.Task.AssignedTo.ID)
	assert.Equal(t, "Task Assignee", result.AssignTask.Task.AssignedTo.FullName)
}

func TestTask_Unassign(t *testing.T) {
	t.Parallel()
	owner := testutil.NewClient(t, testutil.RoleOwner)

	// Create measure, task, people and assign
	measureID := factory.NewMeasure(owner).Create()
	taskID := factory.NewTask(owner, measureID).Create()
	peopleID := factory.NewPeople(owner).WithFullName("Person to Unassign").Create()

	// First assign the task
	assignQuery := `
		mutation AssignTask($input: AssignTaskInput!) {
			assignTask(input: $input) {
				task {
					id
				}
			}
		}
	`

	_, err := owner.Do(assignQuery, map[string]any{
		"input": map[string]any{
			"taskId":       taskID,
			"assignedToId": peopleID,
		},
	})
	require.NoError(t, err)

	query := `
		mutation UnassignTask($input: UnassignTaskInput!) {
			unassignTask(input: $input) {
				task {
					id
					assignedTo {
						id
					}
				}
			}
		}
	`

	var result struct {
		UnassignTask struct {
			Task struct {
				ID         string `json:"id"`
				AssignedTo *struct {
					ID string `json:"id"`
				} `json:"assignedTo"`
			} `json:"task"`
		} `json:"unassignTask"`
	}

	err = owner.Execute(query, map[string]any{
		"input": map[string]any{
			"taskId": taskID,
		},
	}, &result)
	require.NoError(t, err)

	assert.Equal(t, taskID, result.UnassignTask.Task.ID)
	assert.Nil(t, result.UnassignTask.Task.AssignedTo)
}