package update

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

	"github.com/fleetdm/fleet/v4/orbit/pkg/constant"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestInitializeDirectories(t *testing.T) {
	t.Parallel()

	tmpDir := t.TempDir()
	require.NoError(t, os.Chmod(tmpDir, constant.DefaultDirMode))

	opt := DefaultOptions
	opt.RootDirectory = tmpDir
	updater := Updater{opt: opt}
	err := updater.initializeDirectories()
	require.NoError(t, err)
	assertDir(t, filepath.Join(tmpDir, binDir))
}

func assertDir(t *testing.T, path string) {
	info, err := os.Stat(path)
	assert.NoError(t, err, "stat should succeed")
	assert.True(t, info.IsDir())
}

func TestMakeRepoPath(t *testing.T) {
	testCases := []struct {
		name     string
		version  string
		platform string
		expected string
	}{
		{platform: "linux", name: "osqueryd", version: "4.6.0", expected: "osqueryd/linux/4.6.0/osqueryd"},
		{platform: "linux", name: "osqueryd", version: "3.3.2", expected: "osqueryd/linux/3.3.2/osqueryd"},
		{platform: "linux-arm64", name: "osqueryd", version: "3.3.2", expected: "osqueryd/linux-arm64/3.3.2/osqueryd"},
		{platform: "macos", name: "osqueryd", version: "4.6.0", expected: "osqueryd/macos/4.6.0/osqueryd"},
		{platform: "macos", name: "osqueryd", version: "3.3.2", expected: "osqueryd/macos/3.3.2/osqueryd"},
		{platform: "macos-app", name: "osqueryd", version: "3.3.2", expected: "osqueryd/macos-app/3.3.2/osqueryd.app.tar.gz"},
		{platform: "windows", name: "osqueryd", version: "4.6.0", expected: "osqueryd/windows/4.6.0/osqueryd.exe"},
		{platform: "windows", name: "osqueryd", version: "3.3.2", expected: "osqueryd/windows/3.3.2/osqueryd.exe"},
	}

	for _, tt := range testCases {
		t.Run(tt.expected, func(t *testing.T) {
			opt := DefaultOptions

			osqueryd := opt.Targets[tt.name]
			osqueryd.Platform = tt.platform
			osqueryd.Channel = tt.version
			osqueryd.TargetFile = filepath.Base(tt.expected)
			opt.Targets[tt.name] = osqueryd

			u := Updater{opt: opt}
			repoPath, err := u.repoPath(tt.name)
			require.NoError(t, err)
			assert.Equal(t, tt.expected, repoPath)
		})
	}
}

func TestLocalTargetPaths(t *testing.T) {
	testCases := []struct {
		info         TargetInfo
		wantPath     string
		wantExecPath string
		wantDirPath  string
	}{
		{
			DesktopWindowsTarget,
			"root/bin/target/windows/stable/fleet-desktop.exe",
			"root/bin/target/windows/stable/fleet-desktop.exe",
			"",
		},
		{
			NudgeMacOSTarget,
			"root/bin/target/macos/stable/nudge.app.tar.gz",
			"root/bin/target/macos/stable/Nudge.app/Contents/MacOS/Nudge",
			"root/bin/target/macos/stable/Nudge.app",
		},
		{
			DesktopLinuxTarget,
			"root/bin/target/linux/stable/desktop.tar.gz",
			"root/bin/target/linux/stable/fleet-desktop/fleet-desktop",
			"root/bin/target/linux/stable/fleet-desktop",
		},
		{
			DesktopLinuxArm64Target,
			"root/bin/target/linux-arm64/stable/desktop.tar.gz",
			"root/bin/target/linux-arm64/stable/fleet-desktop/fleet-desktop",
			"root/bin/target/linux-arm64/stable/fleet-desktop",
		},
		{
			SwiftDialogMacOSTarget,
			"root/bin/target/macos/stable/swiftDialog.app.tar.gz",
			"root/bin/target/macos/stable/Dialog.app/Contents/MacOS/Dialog",
			"root/bin/target/macos/stable/Dialog.app",
		},
	}

	for _, tt := range testCases {
		path, execPath, dirPath := LocalTargetPaths("root", "target", tt.info)
		require.Equal(t, tt.wantPath, path)
		require.Equal(t, tt.wantExecPath, execPath)
		require.Equal(t, tt.wantDirPath, dirPath)
	}
}