package api

import (
	"context"
	"testing"

	"connectrpc.com/connect"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	apiv1 "github.com/OliveTin/OliveTin/gen/olivetin/api/v1"
	config "github.com/OliveTin/OliveTin/internal/config"
	"github.com/OliveTin/OliveTin/internal/entities"
	"github.com/OliveTin/OliveTin/internal/executor"
)

func TestGetEntitiesPaginatesAndFiltersInstances(t *testing.T) {
	entities.ClearEntitiesOfType("server")
	entities.AddEntity("server", "0", map[string]any{"name": "alpha", "hostname": "alpha.example.com", "ip": "10.0.0.1"})
	entities.AddEntity("server", "1", map[string]any{"name": "beta", "hostname": "beta.example.com", "ip": "10.0.0.2"})
	entities.AddEntity("server", "2", map[string]any{"name": "gamma", "hostname": "gamma.example.com", "ip": "10.0.0.3"})
	t.Cleanup(func() {
		entities.ClearEntitiesOfType("server")
	})

	cfg := config.DefaultConfig()
	cfg.Entities = []*config.EntityFile{
		{
			Name: "server",
			Properties: []config.EntityProperty{
				{Name: "hostname", Title: "Hostname"},
				{Name: "ip", Title: "IP"},
			},
		},
	}
	cfg.Sanitize()

	ex := executor.DefaultExecutor(cfg)
	ex.RebuildActionMap()
	ts, client := getNewTestServerAndClientWithExecutor(cfg, ex)
	defer ts.Close()

	filteredResp, err := client.GetEntities(context.Background(), connect.NewRequest(&apiv1.GetEntitiesRequest{
		EntityType: "server",
		Filter:     "beta",
		Page:       1,
		PageSize:   10,
	}))
	require.NoError(t, err)
	require.Len(t, filteredResp.Msg.EntityDefinitions, 1)
	assert.Equal(t, int32(1), filteredResp.Msg.EntityDefinitions[0].TotalInstances)
	require.Len(t, filteredResp.Msg.EntityDefinitions[0].Instances, 1)
	assert.Equal(t, "beta.example.com", filteredResp.Msg.EntityDefinitions[0].Instances[0].Fields["hostname"])

	pagedResp, err := client.GetEntities(context.Background(), connect.NewRequest(&apiv1.GetEntitiesRequest{
		EntityType: "server",
		Page:       2,
		PageSize:   1,
	}))
	require.NoError(t, err)
	require.Len(t, pagedResp.Msg.EntityDefinitions, 1)
	assert.Equal(t, int32(3), pagedResp.Msg.EntityDefinitions[0].TotalInstances)
	require.Len(t, pagedResp.Msg.EntityDefinitions[0].Instances, 1)
	assert.Equal(t, "1", pagedResp.Msg.EntityDefinitions[0].Instances[0].UniqueKey)
}

func TestGetEntityRestrictsFieldsToConfiguredProperties(t *testing.T) {
	entities.ClearEntitiesOfType("server")
	entities.AddEntity("server", "0", map[string]any{
		"name":     "alpha",
		"hostname": "alpha.example.com",
		"ip":       "10.0.0.1",
		"groups":   []string{"admins"},
	})
	t.Cleanup(func() {
		entities.ClearEntitiesOfType("server")
	})

	cfg := config.DefaultConfig()
	cfg.Entities = []*config.EntityFile{
		{
			Name: "server",
			Properties: []config.EntityProperty{
				{Name: "hostname", Title: "Hostname"},
				{Name: "ip", Title: "IP"},
			},
		},
	}
	cfg.Sanitize()

	ex := executor.DefaultExecutor(cfg)
	ex.RebuildActionMap()
	ts, client := getNewTestServerAndClientWithExecutor(cfg, ex)
	defer ts.Close()

	resp, err := client.GetEntity(context.Background(), connect.NewRequest(&apiv1.GetEntityRequest{
		Type:      "server",
		UniqueKey: "0",
	}))
	require.NoError(t, err)
	require.NotNil(t, resp.Msg)

	assert.Equal(t, "alpha.example.com", resp.Msg.Fields["hostname"])
	assert.Equal(t, "10.0.0.1", resp.Msg.Fields["ip"])
	assert.NotContains(t, resp.Msg.Fields, "groups")
	assert.NotContains(t, resp.Msg.Fields, "name")
}