package main

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

	"github.com/stretchr/testify/assert"
)

var exitCount = 0

func testExit(ret int) {
	exitCount = exitCount + 1
}
func TestValidateSourceAndGallery(t *testing.T) {
	originalExit := exit
	defer func() { exit = originalExit }()
	exit = testExit

	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	exitCountBefore := exitCount
	_, _ = validateSourceAndGallery(tempDir+"/nonexistent", tempDir+"/gallery")
	assert.EqualValues(t, exitCountBefore+1, exitCount, "validateArgs did not exit")

	exitCountBefore = exitCount
	_, _ = validateSourceAndGallery(tempDir, tempDir+"/gallery/nonexistent")
	assert.EqualValues(t, exitCountBefore+1, exitCount, "validateArgs did not exit")

	exitCountBefore = exitCount
	_, _ = validateSourceAndGallery(tempDir, tempDir+"/gallery")
	assert.EqualValues(t, exitCountBefore, exitCount, "validateArgs did not exit")
}

func TestIsDirectory(t *testing.T) {
	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	err = os.Mkdir(tempDir+"/subdir", 0755)
	if err != nil {
		t.Error("couldn't create subdirectory")
	}
	defer os.RemoveAll(tempDir + "/subdir")
	assert.True(t, isDirectory(tempDir+"/subdir"))

	err = os.Symlink(tempDir+"/subdir", tempDir+"/symlink")
	if err != nil {
		t.Error("couldn't create symlink")
	}
	defer os.RemoveAll(tempDir + "/symlink")
	assert.True(t, isDirectory(tempDir+"/symlink"))

	emptyFile, err := os.Create(tempDir + "/file")
	if err != nil {
		t.Error("couldn't create symlink")
	}
	defer emptyFile.Close()
	defer os.RemoveAll(tempDir + "/file")
	assert.False(t, isDirectory(tempDir+"/file"))

	assert.False(t, isDirectory(tempDir+"/nonexistent"))
}

func TestExists(t *testing.T) {
	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	emptyFile, err := os.Create(tempDir + "/file")
	if err != nil {
		t.Error("couldn't create symlink")
	}
	defer emptyFile.Close()
	defer os.RemoveAll(tempDir + "/file")
	assert.True(t, exists(tempDir+"/file"))

	assert.False(t, exists(tempDir+"/nonexistent"))
}

func TestDirHasMediaFiles(t *testing.T) {
	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	emptyFile, err := os.Create(tempDir + "/file.raw")
	if err != nil {
		t.Error("couldn't create symlink")
	}
	defer emptyFile.Close()
	defer os.RemoveAll(tempDir + "/file.raw")

	assert.True(t, dirHasMediafiles(tempDir, false))
}

func TestDirHasMediaFilesFailing(t *testing.T) {
	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	emptyFile, err := os.Create(tempDir + "/file.txt")
	if err != nil {
		t.Error("couldn't create symlink")
	}
	defer emptyFile.Close()
	defer os.RemoveAll(tempDir + "/file.txt")

	assert.False(t, dirHasMediafiles(tempDir, false))
}

func TestDirHasMediaFilesRecurse(t *testing.T) {
	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	err = os.Mkdir(tempDir+"/subdir", 0755)
	if err != nil {
		t.Error("couldn't create subdirectory")
	}
	defer os.RemoveAll(tempDir + "/subdir")

	emptyFile, err := os.Create(tempDir + "/subdir/file.jpg")
	if err != nil {
		t.Error("couldn't create symlink")
	}
	defer emptyFile.Close()
	defer os.RemoveAll(tempDir + "/subdir/file.jpg")

	assert.True(t, dirHasMediafiles(tempDir, false))
}

func TestDirHasMediaFilesRecurseFailing(t *testing.T) {
	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	err = os.Mkdir(tempDir+"/subdir", 0755)
	if err != nil {
		t.Error("couldn't create subdirectory")
	}
	defer os.RemoveAll(tempDir + "/subdir")

	emptyFile, err := os.Create(tempDir + "/subdir/file.txt")
	if err != nil {
		t.Error("couldn't create symlink")
	}
	defer emptyFile.Close()
	defer os.RemoveAll(tempDir + "/subdir/file.txt")

	assert.False(t, dirHasMediafiles(tempDir, false))
}

func TestIsXxxFile(t *testing.T) {
	assert.True(t, isVideoFile("test.mp4"))
	assert.False(t, isVideoFile("test.jpg"))
	assert.False(t, isVideoFile("test.txt"))
	assert.True(t, isImageFile("test.jpg"))
	assert.False(t, isImageFile("test.mp4"))
	assert.False(t, isImageFile("test.txt"))
	assert.True(t, isMediaFile("test.mp4", false))
	assert.True(t, isMediaFile("test.jpg", false))
	assert.False(t, isMediaFile("test.txt", false))
	assert.False(t, isMediaFile("test.mp4", true))
}

func TestCopyRootAssets(t *testing.T) {
	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	var tempGallery directory
	tempGallery.absPath = tempDir

	config := initializeConfig()

	copyRootAssets(tempGallery, false, config)

	assert.FileExists(t, tempDir+"/back.png")
	assert.FileExists(t, tempDir+"/folder.png")
	assert.FileExists(t, tempDir+"/fastgallery.css")
	assert.FileExists(t, tempDir+"/fastgallery.js")
	assert.FileExists(t, tempDir+"/feather.min.js")
	assert.FileExists(t, tempDir+"/primer.css")
}

func TestStripExtension(t *testing.T) {
	assert.Equal(t, "file", stripExtension("file.jpg"))
	assert.NotEqual(t, "file", stripExtension("file/"))
}

func TestReservedDirectory(t *testing.T) {
	myConfig := initializeConfig()

	assert.True(t, reservedDirectory(myConfig.files.thumbnailDir, myConfig))
	assert.True(t, reservedDirectory(myConfig.files.fullsizeDir, myConfig))
	assert.True(t, reservedDirectory(myConfig.files.originalDir, myConfig))
	assert.False(t, reservedDirectory("diipadaapa", myConfig))
}

func TestCreateDirectory(t *testing.T) {
	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	myConfig := initializeConfig()

	createDirectory(tempDir+"/xyz", true, myConfig.files.directoryMode)
	assert.NoDirExists(t, tempDir+"/xyz")

	createDirectory(tempDir+"/xyz", false, myConfig.files.directoryMode)
	assert.DirExists(t, tempDir+"/xyz")
	os.RemoveAll(tempDir + "/xyz")
}

func TestCreateDirectoryTree(t *testing.T) {
	myConfig := initializeConfig()

	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	// Create source directory with two files, a subdir with third file
	err = os.Mkdir(tempDir+"/source", 0755)
	if err != nil {
		t.Error("couldn't create source subdirectory")
	}
	defer os.RemoveAll(tempDir + "/source")

	emptyFile, err := os.Create(tempDir + "/source/file.jpg")
	if err != nil {
		t.Error("couldn't create file")
	}
	defer emptyFile.Close()
	defer os.RemoveAll(tempDir + "/source/file.jpg")

	emptyFile2, err := os.Create(tempDir + "/source/file2.jpg")
	if err != nil {
		t.Error("couldn't create file2")
	}
	defer emptyFile2.Close()
	defer os.RemoveAll(tempDir + "/source/file2.jpg")

	err = os.Mkdir(tempDir+"/source/subdir", 0755)
	if err != nil {
		t.Error("couldn't create source subdirectory's subdirectory")
	}
	defer os.RemoveAll(tempDir + "/source/subdir")

	emptyFile3, err := os.Create(tempDir + "/source/subdir/file.jpg")
	if err != nil {
		t.Error("couldn't create file in subdir")
	}
	defer emptyFile3.Close()
	defer os.RemoveAll(tempDir + "/source/subdir/file.jpg")

	// Create gallery subdirectory with one matching file
	err = os.Mkdir(tempDir+"/gallery", 0755)
	if err != nil {
		t.Error("couldn't create gallery subdirectory")
	}
	defer os.RemoveAll(tempDir + "/gallery")

	err = os.Mkdir(tempDir+"/gallery/"+myConfig.files.fullsizeDir, 0755)
	if err != nil {
		t.Error("couldn't create gallery subdirectory for fullsize")
	}
	defer os.RemoveAll(tempDir + "/gallery/" + myConfig.files.fullsizeDir)

	err = os.Mkdir(tempDir+"/gallery/"+myConfig.files.thumbnailDir, 0755)
	if err != nil {
		t.Error("couldn't create gallery subdirectory for thumbnail")
	}
	defer os.RemoveAll(tempDir + "/gallery/" + myConfig.files.thumbnailDir)

	err = os.Mkdir(tempDir+"/gallery/"+myConfig.files.originalDir, 0755)
	if err != nil {
		t.Error("couldn't create gallery subdirectory for original")
	}
	defer os.RemoveAll(tempDir + "/gallery/" + myConfig.files.originalDir)

	emptyFile4, err := os.Create(tempDir + "/gallery/" + myConfig.files.originalDir + "/file.jpg")
	if err != nil {
		t.Error("couldn't create original gallery file")
	}
	defer emptyFile4.Close()
	defer os.RemoveAll(tempDir + "/gallery/" + myConfig.files.originalDir + "/file.jpg")

	emptyFile5, err := os.Create(tempDir + "/gallery/" + myConfig.files.thumbnailDir + "/file.jpg")
	if err != nil {
		t.Error("couldn't create original gallery file")
	}
	defer emptyFile5.Close()
	defer os.RemoveAll(tempDir + "/gallery/" + myConfig.files.thumbnailDir + "/file.jpg")

	// Ensure thumbnail file is newer than source file
	err = os.Chtimes(tempDir+"/gallery/"+myConfig.files.thumbnailDir+"/file.jpg", time.Now(), time.Now())
	if err != nil {
		t.Error("couldn't change mtime/atime")
	}

	emptyFile6, err := os.Create(tempDir + "/gallery/" + myConfig.files.fullsizeDir + "/file.jpg")
	if err != nil {
		t.Error("couldn't create original gallery file")
	}
	defer emptyFile6.Close()
	defer os.RemoveAll(tempDir + "/gallery/" + myConfig.files.fullsizeDir + "/file.jpg")

	source := createDirectoryTree(tempDir+"/source", "", false)
	gallery := createDirectoryTree(tempDir+"/gallery", "", false)

	compareDirectoryTrees(&source, &gallery, myConfig)

	changes := countChanges(source, myConfig)

	assert.EqualValues(t, 2, changes)
}

// Disabled for now as Github CI's ffmpeg doesn't yet support force_divisible_by=2
func testTransformFileAndVideo(t *testing.T) {
	const videoName = "video.mp4"
	config := initializeConfig()

	tempDir, err := os.MkdirTemp("", "fastgallery-test-")
	if err != nil {
		t.Error("couldn't create temporary directory")
	}
	defer os.RemoveAll(tempDir)

	err = os.Mkdir(filepath.Join(tempDir, "source"), 0755)
	assert.NoError(t, err)
	err = os.Mkdir(filepath.Join(tempDir, "gallery"), 0755)
	assert.NoError(t, err)
	err = os.Mkdir(filepath.Join(tempDir, "gallery", config.files.fullsizeDir), 0755)
	assert.NoError(t, err)
	err = os.Mkdir(filepath.Join(tempDir, "gallery", config.files.thumbnailDir), 0755)
	assert.NoError(t, err)
	err = os.Mkdir(filepath.Join(tempDir, "gallery", config.files.originalDir), 0755)
	assert.NoError(t, err)

	cpCommand := exec.Command("cp", "-r", "../../testing/source/"+videoName, filepath.Join(tempDir, "source"))
	cpCommandOutput, err := cpCommand.CombinedOutput()
	if len(cpCommandOutput) > 0 {
		t.Error("cp produced output", string(cpCommandOutput))
	}
	if err != nil {
		t.Error("cp error", err.Error())
	}

	thumbnailFilename, fullsizeFilename := getGalleryFilenames(videoName, config)

	testJob := transformationJob{
		filename:          videoName,
		sourceFilepath:    filepath.Join(tempDir, "source", videoName),
		thumbnailFilepath: filepath.Join(tempDir, "gallery", config.files.thumbnailDir, thumbnailFilename),
		fullsizeFilepath:  filepath.Join(tempDir, "gallery", config.files.fullsizeDir, fullsizeFilename),
		originalFilepath:  filepath.Join(tempDir, "gallery", config.files.originalDir, videoName),
	}

	transformFile(testJob, nil, config)
	assert.FileExists(t, testJob.thumbnailFilepath)
	assert.FileExists(t, testJob.fullsizeFilepath)

	err = os.RemoveAll(testJob.thumbnailFilepath)
	assert.NoError(t, err)
	os.RemoveAll(testJob.fullsizeFilepath)
	assert.NoError(t, err)

	transformVideo(testJob.sourceFilepath, testJob.fullsizeFilepath, testJob.thumbnailFilepath, config)
	assert.FileExists(t, testJob.thumbnailFilepath)
	assert.FileExists(t, testJob.fullsizeFilepath)

	err = createOriginal(testJob.sourceFilepath, testJob.originalFilepath)
	assert.NoError(t, err)
	assert.FileExists(t, testJob.originalFilepath)
}

func TestGetIconSize(t *testing.T) {
	iconSize, err := getIconSize("/tmp/icon-48x48.png")
	assert.NoError(t, err)
	assert.EqualValues(t, "48x48", iconSize)

	iconSize, err = getIconSize("test192x192-apple.svg")
	assert.NoError(t, err)
	assert.EqualValues(t, "192x192", iconSize)

	iconSize, err = getIconSize("test-xicon-64x64.ico")
	assert.NoError(t, err)
	assert.EqualValues(t, "64x64", iconSize)

	iconSize, err = getIconSize("test-xicon.ico")
	assert.Error(t, err)
	assert.EqualValues(t, "", iconSize)
}

func TestGetIconType(t *testing.T) {
	iconType, err := getIconType("/tmp/icon-48x48.png")
	assert.NoError(t, err)
	assert.EqualValues(t, "image/png", iconType)

	iconType, err = getIconType("icon-48x48.png")
	assert.NoError(t, err)
	assert.EqualValues(t, "image/png", iconType)

	iconType, err = getIconType("icon-48x48.jpg")
	assert.Error(t, err)
	assert.EqualValues(t, "", iconType)
}

// TODO tests for
// isDirectory with symlinked dir
// isSymlinkDir
// createDirectoryTree("nonexistent", "")
// hasDirectoryChanged
// symlinkFile
// createHTML
// getGalleryDirectoryNames
// transformImage
// transformVideo
// createOriginal
// getGalleryFilenames
// transformFile
// transformationWorker
// createMedia
// cleanDirectory
// createGallery
//   - exists, doesn't exist, some gallery files exist / some don't
//   - thumbnail modified earlier than original or vice versa
// setupSignalHandler
// signalHandler