package core_test

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

	"git.hrafn.xyz/aether/gosick/internal/homesick/core"
	"github.com/stretchr/testify/require"
	"github.com/stretchr/testify/suite"
)

type LinkSuite struct {
	suite.Suite
	tmpDir   string
	homeDir  string
	reposDir string
	app      *core.App
}

func TestLinkSuite(t *testing.T) {
	suite.Run(t, new(LinkSuite))
}

func (s *LinkSuite) SetupTest() {
	s.tmpDir = s.T().TempDir()
	s.homeDir = filepath.Join(s.tmpDir, "home")
	s.reposDir = filepath.Join(s.homeDir, ".homesick", "repos")

	require.NoError(s.T(), os.MkdirAll(s.reposDir, 0o755))

	s.app = &core.App{
		HomeDir:  s.homeDir,
		ReposDir: s.reposDir,
		Stdout:   io.Discard,
		Stderr:   io.Discard,
	}
}

func (s *LinkSuite) createCastle(castle string) string {
	castleHome := filepath.Join(s.reposDir, castle, "home")
	require.NoError(s.T(), os.MkdirAll(castleHome, 0o755))
	return castleHome
}

func (s *LinkSuite) writeFile(path string, content string) {
	require.NoError(s.T(), os.MkdirAll(filepath.Dir(path), 0o755))
	require.NoError(s.T(), os.WriteFile(path, []byte(content), 0o644))
}

func (s *LinkSuite) TestLink_SymlinksTopLevelFiles() {
	castleHome := s.createCastle("glencairn")
	dotfile := filepath.Join(castleHome, ".vimrc")
	s.writeFile(dotfile, "set number\n")

	err := s.app.Link("glencairn")
	require.NoError(s.T(), err)

	homePath := filepath.Join(s.homeDir, ".vimrc")
	info, err := os.Lstat(homePath)
	require.NoError(s.T(), err)
	require.True(s.T(), info.Mode()&os.ModeSymlink != 0)

	target, err := os.Readlink(homePath)
	require.NoError(s.T(), err)
	require.Equal(s.T(), dotfile, target)
}

func (s *LinkSuite) TestLink_RespectsHomesickSubdir() {
	castleHome := s.createCastle("glencairn")
	configDir := filepath.Join(castleHome, ".config")
	appDir := filepath.Join(configDir, "myapp")
	s.writeFile(filepath.Join(appDir, "config.toml"), "ok=true\n")
	s.writeFile(filepath.Join(s.reposDir, "glencairn", ".homesick_subdir"), ".config\n")

	err := s.app.Link("glencairn")
	require.NoError(s.T(), err)

	configInfo, err := os.Lstat(filepath.Join(s.homeDir, ".config"))
	require.NoError(s.T(), err)
	require.False(s.T(), configInfo.Mode()&os.ModeSymlink != 0)

	homeApp := filepath.Join(s.homeDir, ".config", "myapp")
	appInfo, err := os.Lstat(homeApp)
	require.NoError(s.T(), err)
	require.True(s.T(), appInfo.Mode()&os.ModeSymlink != 0)

	target, err := os.Readlink(homeApp)
	require.NoError(s.T(), err)
	require.Equal(s.T(), appDir, target)
}

func (s *LinkSuite) TestLink_ForceReplacesExistingFile() {
	castleHome := s.createCastle("glencairn")
	dotfile := filepath.Join(castleHome, ".zshrc")
	s.writeFile(dotfile, "export EDITOR=vim\n")
	s.writeFile(filepath.Join(s.homeDir, ".zshrc"), "existing\n")

	s.app.Force = true
	err := s.app.Link("glencairn")
	require.NoError(s.T(), err)

	homePath := filepath.Join(s.homeDir, ".zshrc")
	info, err := os.Lstat(homePath)
	require.NoError(s.T(), err)
	require.True(s.T(), info.Mode()&os.ModeSymlink != 0)
}

func (s *LinkSuite) TestLink_NoForceErrorsOnConflict() {
	castleHome := s.createCastle("glencairn")
	dotfile := filepath.Join(castleHome, ".gitconfig")
	s.writeFile(dotfile, "[user]\n")
	s.writeFile(filepath.Join(s.homeDir, ".gitconfig"), "local\n")

	err := s.app.Link("glencairn")
	require.Error(s.T(), err)
}