package server_test

import (
	"testing"
	"time"

	"github.com/stretchr/testify/require"

	"github.com/kopia/kopia/internal/apiclient"
	"github.com/kopia/kopia/internal/clock"
	"github.com/kopia/kopia/internal/serverapi"
	"github.com/kopia/kopia/internal/testlogging"
	"github.com/kopia/kopia/internal/uitask"
	"github.com/kopia/kopia/snapshot"
	"github.com/kopia/kopia/snapshot/policy"
)

func mustCreateSource(t *testing.T, cli *apiclient.KopiaAPIClient, path string, pol *policy.Policy) {
	t.Helper()

	_, err := serverapi.CreateSnapshotSource(testlogging.Context(t), cli, &serverapi.CreateSnapshotSourceRequest{
		Path:   path,
		Policy: pol,
	})
	require.NoError(t, err)
}

func mustSetPolicy(t *testing.T, cli *apiclient.KopiaAPIClient, si snapshot.SourceInfo, pol *policy.Policy) {
	t.Helper()

	require.NoError(t, serverapi.SetPolicy(testlogging.Context(t), cli, si, pol))
}

func mustListSources(t *testing.T, cli *apiclient.KopiaAPIClient, match *snapshot.SourceInfo) []*serverapi.SourceStatus {
	t.Helper()

	resp, err := serverapi.ListSources(testlogging.Context(t), cli, match)
	require.NoError(t, err)

	return resp.Sources
}

func mustGetTask(t *testing.T, cli *apiclient.KopiaAPIClient, taskID string) uitask.Info {
	t.Helper()

	resp, err := serverapi.GetTask(testlogging.Context(t), cli, taskID)
	require.NoError(t, err)

	return *resp
}

func mustListTasks(t *testing.T, cli *apiclient.KopiaAPIClient) []uitask.Info {
	t.Helper()

	resp, err := serverapi.ListTasks(testlogging.Context(t), cli)
	require.NoError(t, err)

	return resp.Tasks
}

func mustGetLatestTask(t *testing.T, cli *apiclient.KopiaAPIClient) uitask.Info {
	t.Helper()

	tl := mustListTasks(t, cli)
	require.NotEmpty(t, tl)

	return tl[0]
}

func waitForTask(t *testing.T, cli *apiclient.KopiaAPIClient, taskID string, timeout time.Duration) uitask.Info {
	t.Helper()

	var lastInfo uitask.Info

	deadline := clock.Now().Add(timeout)
	for clock.Now().Before(deadline) {
		lastInfo = mustGetTask(t, cli, taskID)

		if lastInfo.Status.IsFinished() {
			return lastInfo
		}

		time.Sleep(100 * time.Millisecond)
	}

	t.Fatalf("task %v did not complete in %v, last: %v", taskID, timeout, lastInfo)

	return lastInfo
}