package watchdtui_test

import (
	"context"
	"flag"
	"fmt"
	"os"
	"path/filepath"
	"reflect"
	"strings"
	"testing"
	"time"

	"github.com/charmbracelet/bubbles/spinner"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	"github.com/muesli/termenv"
	"github.com/stretchr/testify/require"
	"github.com/ubuntu/adsys/cmd/adwatchd/commands"
	watchdconfig "github.com/ubuntu/adsys/internal/config/watchd"
	log "github.com/ubuntu/adsys/internal/grpc/logstreamer"
	"github.com/ubuntu/adsys/internal/testutils"
	"github.com/ubuntu/adsys/internal/watchdservice"
	"github.com/ubuntu/adsys/internal/watchdtui"
	"gopkg.in/yaml.v3"
)

var (
	stdout bool
)

func TestInteractiveInput(t *testing.T) {
	binPath, err := os.Executable()
	require.NoError(t, err, "can't get executable directory")
	binDir := filepath.Dir(binPath)

	tests := map[string]struct {
		events        []tea.Msg
		existingPaths []string
		cfgToValidate string
		absPathInput  bool

		// Parameters for when we want to simulate a previous config file
		configOverride bool
		configDirs     []string
		prevConfig     string
		prevConfigDirs []string
	}{
		"Initial view": {
			events:        []tea.Msg{},
			existingPaths: []string{"foo/bar/", "foo/baz"},
		},

		// Config file input behaviors
		"Config file exists": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/baz"},
		},
		"Config file is absent and input is absolute": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			absPathInput: true,
		},
		"Config file is absent and input is relative": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
		},
		"Config file is absent and input is a dir": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/bar/"},
		},
		"Existing config file is passed in and is empty or has no directories": {
			configOverride: true,
		},
		"Existing config file is passed in and contains directories which exist on the system": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyDown}, // focus on submit button
			},
			configOverride: true,
			existingPaths:  []string{"foo/bar/", "foo/baz/"},
			configDirs:     []string{"foo/bar", "foo/baz"},
		},
		"Existing config file is passed in and contains directories, not all which exist on the system": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyDown}, // focus on bad input for the error message
			},
			configOverride: true,
			existingPaths:  []string{"foo/bar/"},
			configDirs:     []string{"foo/bar", "foo/baz"},
		},

		// Installed service behaviors
		"Found installed service, config not overridden": {
			prevConfig:     "myprevconfig.yaml",
			prevConfigDirs: []string{"foo/bar", "foo/baz"},
			configDirs:     []string{"foo/bar", "foo/qux"},
			existingPaths:  []string{"foo/bar/", "foo/baz/", "foo/qux/"},
		},
		"Found installed service, config overridden": {
			configOverride: true,
			prevConfig:     "myprevconfig.yaml",
			prevConfigDirs: []string{"foo/bar", "foo/baz"},
			configDirs:     []string{"foo/bar", "foo/qux"},
			existingPaths:  []string{"foo/bar/", "foo/baz/", "foo/qux/"},
		},

		// Directory input behaviors
		"Directory exists": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter}, // creates new line
				tea.KeyMsg{Type: tea.KeyEnter}, // removes new line and focuses on Submit
			},
			existingPaths: []string{"foo/bar/"},
		},
		"Directory does not exist, block input": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyUp},
				tea.KeyMsg{Type: tea.KeyUp},
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyTab},
				tea.KeyMsg{Type: tea.KeyTab},
				tea.KeyMsg{Type: tea.KeyShiftTab},
				tea.KeyMsg{Type: tea.KeyShiftTab},
			},
		},
		"Dot and double dot directory inputs are normalized": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar/./qux/../../baz")},
			},
		},
		"Directory is a file, block input": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyUp},
				tea.KeyMsg{Type: tea.KeyUp},
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyTab},
				tea.KeyMsg{Type: tea.KeyTab},
				tea.KeyMsg{Type: tea.KeyShiftTab},
				tea.KeyMsg{Type: tea.KeyShiftTab},
			},
			existingPaths: []string{"foo/bar"},
		},
		"Multiple existing directories, can cycle between the inputs": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/qux")},
				tea.KeyMsg{Type: tea.KeyUp},        // focus on first entry
				tea.KeyMsg{Type: tea.KeyBackspace}, // delete last char to make it invalid
				tea.KeyMsg{Type: tea.KeyDown},      // attempt to move
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("z")}, // fix entry
				tea.KeyMsg{Type: tea.KeyTab},
				tea.KeyMsg{Type: tea.KeyTab},  // back to the last entry
				tea.KeyMsg{Type: tea.KeyDown}, // focus on Submit
			},
			existingPaths: []string{"foo/bar/", "foo/baz/", "foo/qux/"},
		},
		"Multiple existing directories, can delete them": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyBackspace}, // delete current input, back to foo
				tea.KeyMsg{Type: tea.KeyBackspace},
				tea.KeyMsg{Type: tea.KeyBackspace},
				tea.KeyMsg{Type: tea.KeyBackspace},
				tea.KeyMsg{Type: tea.KeyEnter}, // delete current empty input
			},
			existingPaths: []string{"foo/bar/", "foo/baz/"},
		},
		"No directories, focus on dir input": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter}, // cannot move further with no directories
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
		},

		// Submit behaviors
		"Submit with default config": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/bar/", "foo/baz/"},
			cfgToValidate: filepath.Join(binDir, "adwatchd.yaml"),
		},
		"Submit with fresh config in current directory": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("my_config.yaml")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/bar/", "foo/baz/"},
			cfgToValidate: "my_config.yaml",
		},
		"Submit with fresh config in nested directory": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("aaa/bbb/ccc/my_config.yaml")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/bar/", "foo/baz/"},
			cfgToValidate: "aaa/bbb/ccc/my_config.yaml",
		},
		"Submit with duplicate directories": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar/../")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz/abc/../")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/bar/", "foo/baz/"},
			cfgToValidate: filepath.Join(binDir, "adwatchd.yaml"),
		},
		"Submit with directory as config input": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/bar/", "foo/baz/"},
			cfgToValidate: "foo/bar/adwatchd.yaml",
		},
		"Submit with dot directories is normalized": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("./foo/./bar/./")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune(".")}, // #ABSPATH#
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/bar/"},
			cfgToValidate: filepath.Join(binDir, "adwatchd.yaml"),
		},
		"Submit with double dot directories is normalized": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/baz/qux/asd/../..")}, // baz
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},
			},
			existingPaths: []string{"foo/baz/"},
			cfgToValidate: filepath.Join(binDir, "adwatchd.yaml"),
		},

		// Other navigation behaviors
		"Other navigation tests": {
			events: []tea.Msg{
				tea.KeyMsg{Type: tea.KeyUp},        // no up or shift+tab on config
				tea.KeyMsg{Type: tea.KeyShiftTab},  // no up or shift+tab on config
				tea.KeyMsg{Type: tea.KeyBackspace}, // no custom backspace on config
				tea.KeyMsg{Type: tea.KeyDown},
				tea.KeyMsg{Type: tea.KeyBackspace}, // backspace on first input cycles back to config
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("foo/bar")},
				tea.KeyMsg{Type: tea.KeyEnter},
				tea.KeyMsg{Type: tea.KeyEnter},     // focus on Submit
				tea.KeyMsg{Type: tea.KeyBackspace}, // backspace on Submit
				tea.KeyMsg{Type: tea.KeyEnter},     // back on Submit
				tea.KeyMsg{Type: tea.KeyDown},      // no down or tab on Submit
				tea.KeyMsg{Type: tea.KeyTab},
			},
			existingPaths: []string{"foo/bar/"},
		},
	}
	for name, tc := range tests {
		// Need to capture the absolute path of the package before changing directory
		// in the subtests, otherwise the tree will change.
		absPkgPath, _ := filepath.Abs(".")
		t.Run(name, func(t *testing.T) {
			t.Cleanup(func() { os.Remove(filepath.Join(binDir, "adwatchd.yaml")) })

			tmpdir := t.TempDir()
			testutils.Chdir(t, tmpdir)

			// Create existing directories/files
			for _, path := range tc.existingPaths {
				testutils.CreatePath(t, path)
			}

			var prevConfig string
			// Create actively used config file if needed
			if len(tc.prevConfigDirs) > 0 {
				prevConfig = filepath.Join(binDir, tc.prevConfig)
				data, err := yaml.Marshal(&watchdconfig.AppConfig{Dirs: tc.prevConfigDirs})
				require.NoError(t, err, "could not marshal config")
				err = os.WriteFile(prevConfig, data, 0600)
				require.NoError(t, err, "could not write actively used config")
				t.Cleanup(func() { os.Remove(prevConfig) })
			}

			// Create existing config file if needed
			if len(tc.configDirs) > 0 {
				data, err := yaml.Marshal(&watchdconfig.AppConfig{Dirs: tc.configDirs})
				require.NoError(t, err, "could not marshal config")
				err = os.WriteFile(filepath.Join(binDir, "adwatchd.yaml"), data, 0600)
				require.NoError(t, err, "could not write existing config")
			}

			m, _ := watchdtui.InitialModelForTests(filepath.Join(binDir, "adwatchd.yaml"), prevConfig, !tc.configOverride).Update(nil)

			for _, e := range tc.events {
				keyMsg, ok := e.(tea.KeyMsg)
				require.True(t, ok, "expected event to be a KeyMsg")

				// Did we request an absolute path? If so, we need to merge the
				// runes with the current working directory.
				if tc.absPathInput && keyMsg.Type == tea.KeyRunes {
					e = tea.KeyMsg{
						Type:  tea.KeyRunes,
						Runes: []rune(filepath.Join(tmpdir, string(keyMsg.Runes))),
					}
				}

				m = updateModel(t, m, e)
			}
			out := m.View()
			if stdout {
				fmt.Println(out)
			}

			got := normalizeOutput(t, out)

			goldPath := filepath.Join(absPkgPath, testutils.GoldenPath(t))
			want := testutils.LoadWithUpdateFromGolden(t, got, testutils.WithGoldenPath(goldPath))

			if tc.cfgToValidate != "" {
				outCfg, err := os.ReadFile(tc.cfgToValidate)
				require.NoError(t, err, "Cannot load test config file")

				goldCfgPath := filepath.Join(absPkgPath, testutils.GoldenPath(t)+".yaml")

				gotCfg := normalizeOutput(t, string(outCfg))
				wantCfg := testutils.LoadWithUpdateFromGolden(t, gotCfg, testutils.WithGoldenPath(goldCfgPath))
				require.Equal(t, normalizeGoldenFile(t, wantCfg), normalizeOutput(t, string(outCfg)), "Configs don't match")
			}

			require.Equal(t, normalizeGoldenFile(t, want), got, "Didn't get expected output")
		})
	}
}

func TestInteractiveInstall(t *testing.T) {
	testutils.SkipUnlessRoot(t)

	// This is used for skipping race tests in Windows CI
	if os.Getenv("ADSYS_SKIP_INTEGRATION_TESTS") != "" {
		t.Skip("Integration tests skipped as requested")
	}

	svc, err := watchdservice.New(context.Background())
	require.NoError(t, err, "Cannot initialize watchd service")

	t.Cleanup(func() {
		time.Sleep(time.Second)
		err = svc.Uninstall(context.Background())
		require.NoError(t, err, "Cannot uninstall watchd service")
	})

	testutils.Chdir(t, t.TempDir())

	// Create existing directories/files
	err = os.MkdirAll("foo/bar", 0750)
	require.NoError(t, err, "can't create directories")
	err = os.MkdirAll("foo/baz", 0750)
	require.NoError(t, err, "can't create directories")

	m, _ := watchdtui.InitialModel().Update(nil)

	m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter}) // use default config

	// add directories
	for _, dir := range []string{"foo/bar", "foo/baz"} {
		m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune(dir)})
		m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})
	}

	// submit
	m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})
	m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})

	out := m.View()
	successMessage := "Service adwatchd was successfully installed and is now running.\n"
	require.Equal(t, successMessage, out, "Didn't get expected output")

	status, err := svc.Status(context.Background())
	require.NoError(t, err, "Cannot get status")
	require.Contains(t, status, "running", "Expected service to be running")
}

func TestInteractiveUpdate(t *testing.T) {
	testutils.SkipUnlessRoot(t)

	// This is used for skipping race tests in CI
	if os.Getenv("ADSYS_SKIP_INTEGRATION_TESTS") != "" {
		t.Skip("Integration tests skipped as requested")
	}

	tests := map[string]struct {
		prevDirs  []string
		dirsToAdd []string

		changeConfigPath bool
	}{
		"Change directories, same config file": {
			prevDirs:  []string{"foo/bar", "foo/baz"},
			dirsToAdd: []string{"foo/qux"},
		},
		"Change directories, different config file": {
			prevDirs:         []string{"foo/bar", "foo/baz"},
			dirsToAdd:        []string{"foo/qux"},
			changeConfigPath: true,
		},
	}
	for name, tc := range tests {
		t.Run(name, func(t *testing.T) {
			tmpdir := t.TempDir()
			absPrevDirs := make([]string, len(tc.prevDirs))
			for i, dir := range tc.prevDirs {
				absPrevDirs[i] = filepath.Join(tmpdir, dir)
				err := os.MkdirAll(absPrevDirs[i], 0750)
				require.NoError(t, err, "Cannot create previous directories")
			}

			absDirsToAdd := make([]string, len(tc.dirsToAdd))
			for i, dir := range tc.dirsToAdd {
				absDirsToAdd[i] = filepath.Join(tmpdir, dir)
				err := os.MkdirAll(absDirsToAdd[i], 0750)
				require.NoError(t, err, "Cannot create new directories")
			}

			prevConfigPath := filepath.Join(tmpdir, "prevconfig.yaml")
			err := watchdconfig.WriteConfig(prevConfigPath, absPrevDirs)
			require.NoError(t, err, "Cannot write previous config file")

			svc, err := watchdservice.New(context.Background(), watchdservice.WithConfig(prevConfigPath))
			require.NoError(t, err, "Cannot initialize watchd service")

			t.Cleanup(func() {
				time.Sleep(time.Second)
				err = svc.Uninstall(context.Background())
				require.NoError(t, err, "Cannot uninstall watchd service")
			})

			// Start with an installed service
			err = svc.Install(context.Background())
			require.NoError(t, err, "Cannot install watchd service")

			configPath := prevConfigPath
			if tc.changeConfigPath {
				configPath = filepath.Join(tmpdir, "newconfig.yaml")
			}

			// Using the TUI, update the watched directory
			m, _ := watchdtui.InitialModelWithPrevConfig(configPath, prevConfigPath, true).Update(nil)
			if tc.changeConfigPath {
				// Change to the new config file
				m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune(configPath)})
			}

			// Move to the end of the directory list
			m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})
			m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})
			m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})
			m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})

			// Add directories
			for _, dir := range absDirsToAdd {
				m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune(dir)})
				m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})
			}

			// Sleep to ensure the service is properly installed before we hit
			// submit on the TUI and trigger the reinstall.
			time.Sleep(time.Second)

			// Submit
			m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})
			m = updateModel(t, m, tea.KeyMsg{Type: tea.KeyEnter})

			// Check service status
			status, err := svc.Status(context.Background())
			require.NoError(t, err, "Cannot get status")
			require.Contains(t, status, configPath, fmt.Sprintf("Expected config file to be correct. TUI view: %q", m.View()))

			for _, dir := range append(absPrevDirs, absDirsToAdd...) {
				require.Contains(t, status, dir, "Expected directory to be watched")
			}
		})
	}
}

// updateModel calls Update() on the model and executes returned commands.
// It will reexecute Update() until there are no more returned commands.
func updateModel(t *testing.T, m tea.Model, msg tea.Msg) tea.Model {
	t.Helper()

	m, cmd := m.Update(msg)
	if cmd == nil {
		return m
	}

	messageCandidates := cmd()

	batchMsgType := reflect.TypeOf(tea.Batch(func() tea.Msg { return tea.Msg(struct{}{}) })())

	// executes all messages on batched messages, which is a slice underlying it.
	if reflect.TypeOf(messageCandidates) == batchMsgType {
		if reflect.TypeOf(messageCandidates).Kind() != reflect.Slice {
			t.Fatalf("expected batched messages to be a slice but it's not: %v", reflect.TypeOf(messageCandidates).Kind())
		}

		v := reflect.ValueOf(messageCandidates)
		for i := 0; i < v.Len(); i++ {
			messages := v.Index(i).Call(nil)
			// Call update on all returned messages, which can itself reenter Update()
			for _, msgValue := range messages {
				// if this is a Tick message, ignore it (to avoid endless loop as we will always have the next tick available)
				// and our function is reentrant, not a queue of message. Thus, install is never called.
				if _, ok := msgValue.Interface().(spinner.TickMsg); ok {
					continue
				}

				msg, ok := msgValue.Interface().(tea.Msg)
				if !ok {
					t.Fatalf("expected message to be a tea.Msg, but got: %T", msg)
				}
				m = updateModel(t, m, msg)
			}
		}
		return m
	}

	// We only got one message, call Update() on it
	return updateModel(t, m, messageCandidates)
}

// normalizeOutput normalizes the output of the view function in order to ensure
// tests work on both Linux and Windows.
func normalizeOutput(t *testing.T, out string) string {
	t.Helper()

	cwd, err := os.Getwd()
	require.NoError(t, err, "can't get current directory")

	binPath, err := os.Executable()
	require.NoError(t, err, "can't get executable directory")

	// Replace executable directory with a deterministic placeholder
	out = strings.ReplaceAll(out, filepath.Dir(binPath), "#BINDIR#")

	// Normalize backslashes to slashes
	out = strings.ReplaceAll(out, "\\", "/")

	// Replace cwd with a deterministic placeholder
	cwd = filepath.ToSlash(cwd)
	out = strings.ReplaceAll(out, cwd, "#ABSPATH#")

	return out
}

// normalizeGoldenOutput normalizes the golden file content in order to ensure
// Linux/Windows compatibility with a single set of golden files.
func normalizeGoldenFile(t *testing.T, out string) string {
	t.Helper()

	// Strip carriage returns
	out = strings.ReplaceAll(out, "\r", "")

	return out
}

func TestMain(m *testing.M) {
	// Running real command mock from service manager
	if len(os.Args) > 0 && os.Args[1] == "run" {
		app := commands.New()
		err := app.Run()
		if err != nil {
			log.Error(context.Background(), err)
			os.Exit(1)
		}
		os.Exit(0)
	}

	// Simulate a color terminal
	lipgloss.SetColorProfile(termenv.ANSI256)

	flag.BoolVar(&stdout, "stdout", false, "print output to stdout for debugging purposes")
	flag.Parse()

	m.Run()
}
