package analyzers

import (
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.lsp.dev/uri"

	"github.com/karitham/thrift-ls/analyzertest"
	"github.com/karitham/thrift-ls/sema"
	"github.com/karitham/thrift-ls/store"
)

// writeThrift writes content to a .thrift file under folder. Only the
// include-shadow test needs real disk: its candidates resolve through the
// disk fallback of the view's file source.
func writeThrift(t *testing.T, folder, name, content string) string {
	t.Helper()

	p := filepath.Join(folder, name)
	require.NoError(t, os.WriteFile(p, []byte(content), 0o644))

	return p
}

func Test_IncludeShadowCheck(t *testing.T) {
	tests := []struct {
		name        string
		roots       []string // which sibling roots hold recipes/stew.thrift
		wantCount   int
		wantMessage string // required substring when wantCount > 0
	}{
		{
			name:      "include matching one include path",
			roots:     []string{"senshi"},
			wantCount: 0,
		},
		{
			name:        "include matching two include paths",
			roots:       []string{"senshi", "laios"},
			wantCount:   1,
			wantMessage: "matches multiple include paths",
		},
		{
			name:      "unresolvable include",
			roots:     nil,
			wantCount: 0,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			folder := t.TempDir()

			for _, root := range tt.roots {
				require.NoError(t, os.MkdirAll(filepath.Join(folder, root, "recipes"), 0o755))
				_ = writeThrift(t, folder, filepath.Join(root, "recipes", "stew.thrift"), "struct Monster {}\n")
			}

			content := "include \"recipes/stew.thrift\"\nstruct Party { 1: i32 members }\n"
			require.NoError(t, os.MkdirAll(filepath.Join(folder, "camp"), 0o755))
			filePath := writeThrift(t, folder, filepath.Join("camp", "main.thrift"), content)

			view := store.BuildViewForTestWithPaths([]string{
				filepath.Join(folder, "laios"),
				filepath.Join(folder, "senshi"),
			}, []*store.FileChange{
				{
					URI:     uri.File(filePath),
					Version: 0,
					Content: []byte(content),
					From:    store.FileChangeTypeDidOpen,
				},
			})

			got := analyzertest.RunOnView(t, sema.EachFile(&IncludeShadowCheck{}), view, uri.File(filePath))

			require.Len(t, got[uri.File(filePath)], tt.wantCount)

			for _, d := range got[uri.File(filePath)] {
				assert.Equal(t, sema.SeverityWarning, d.Severity)
				assert.Equal(t, sema.CodeIncludeShadow, d.Code)
				assert.Contains(t, d.Message, tt.wantMessage)
			}
		})
	}
}