package endtoend

import (
	"context"
	"regexp"
	"testing"

	"queryorchestration/internal/serviceconfig"
	"queryorchestration/internal/serviceconfig/queue"
	"queryorchestration/internal/test"
	queryapi "queryorchestration/pkg/queryAPI"

	"github.com/oapi-codegen/runtime/types"
	"github.com/stretchr/testify/assert"
)

type CollectorConfig struct {
	serviceconfig.BaseConfig
	queue.QueueConfig
}

func TestCollectorService(t *testing.T) {
	ctx := context.Background()

	cfg := &CollectorConfig{}
	test.SetCfgProviderWithBasePath(t, cfg, "../..")

	network, ncleanup := test.CreateNetwork(t, ctx)
	defer ncleanup()

	_, clean := test.CreateAWSContainer(t, ctx, &test.CreateAWSConfig{
		Cfg:     cfg,
		Network: network,
	})
	defer clean()

	test.SetQueueClient(t, ctx, cfg)
	clientsyncurl := test.CreateQueue(t, ctx, cfg, test.ClientSyncRunnerName)

	c, cleanup := test.CreateAPINetwork(t, ctx, &test.ServiceNetworkConfig{
		Cfg:     cfg,
		Network: network,
		API:     test.QueryAPI,
	})
	defer cleanup()

	client, err := queryapi.NewClientWithResponses(c.URI)
	assert.NoError(t, err)

	contextRes, err := client.CreateQueryWithResponse(ctx, queryapi.QueryCreate{
		Type: queryapi.CONTEXTFULL,
	})
	assert.NoError(t, err)

	jsoncfg := "{\"path\":\"key\"}"
	jsonRes, err := client.CreateQueryWithResponse(ctx, queryapi.QueryCreate{
		Type:   queryapi.JSONEXTRACTOR,
		Config: &jsoncfg,
		RequiredQueries: &[]types.UUID{
			contextRes.JSON201.Id,
		},
	})
	assert.NoError(t, err)

	clientRes, err := client.CreateClientWithResponse(ctx, queryapi.ClientCreate{
		Name: "example_name",
		Id:   "ID",
	})
	assert.NoError(t, err)
	id := clientRes.JSON201.Id

	collRes, err := client.GetCollectorByClientIdWithResponse(ctx, id)
	assert.NoError(t, err)
	assert.Equal(t, id, collRes.JSON200.ClientId)
	assert.Equal(t, int32(0), collRes.JSON200.ActiveVersion)
	assert.Equal(t, int32(0), collRes.JSON200.LatestVersion)
	assert.Equal(t, int64(0), collRes.JSON200.MinimumCleanerVersion)
	assert.Equal(t, int64(0), collRes.JSON200.MinimumTextVersion)
	assert.Len(t, collRes.JSON200.Fields, 0)

	av := int32(1)
	fields := []queryapi.CollectorField{
		{
			Name:    "json",
			QueryId: jsonRes.JSON201.Id,
		},
	}
	minVersion := int64(1000)
	uRes, err := client.SetCollectorByClientIdWithResponse(ctx, id, queryapi.CollectorSet{
		ActiveVersion:         &av,
		MinimumCleanerVersion: &minVersion,
		MinimumTextVersion:    &minVersion,
		Fields:                &fields,
	})
	assert.NoError(t, err)
	assert.Equal(t, 200, uRes.StatusCode())

	collRes, err = client.GetCollectorByClientIdWithResponse(ctx, id)
	assert.NoError(t, err)
	assert.Equal(t, id, collRes.JSON200.ClientId)
	assert.Equal(t, int32(1), collRes.JSON200.ActiveVersion)
	assert.Equal(t, int32(1), collRes.JSON200.LatestVersion)
	assert.Equal(t, minVersion, collRes.JSON200.MinimumCleanerVersion)
	assert.Equal(t, minVersion, collRes.JSON200.MinimumTextVersion)
	assert.Len(t, collRes.JSON200.Fields, 1)
	assert.ElementsMatch(t, fields, collRes.JSON200.Fields)

	test.AssertMessageBody(t, ctx, cfg, clientsyncurl, regexp.MustCompile(`{"id":".+"}`))
}