package photoprism

import (
	"fmt"
	"github.com/jinzhu/gorm"
	"github.com/stretchr/testify/assert"
	"os"
	"testing"
)

const testDataPath = "testdata"
const testDataUrl = "https://www.dropbox.com/s/na9p9wwt98l7m5b/import.zip?dl=1"
const testDataHash = "ed3bdb2fe86ea662bc863b63e219b47b8d9a74024757007f7979887d"

var darktableCli = "/usr/bin/darktable-cli"
var testDataZip = GetExpandedFilename(testDataPath + "/import.zip")
var originalsPath = GetExpandedFilename(testDataPath + "/originals")
var thumbnailsPath = GetExpandedFilename(testDataPath + "/thumbnails")
var importPath = GetExpandedFilename(testDataPath + "/import")
var exportPath = GetExpandedFilename(testDataPath + "/export")
var databaseDriver = "mysql"
var databaseDsn = "photoprism:photoprism@tcp(database:3306)/photoprism"

func (c *Config) RemoveTestData(t *testing.T) {
	os.RemoveAll(c.ImportPath)
	os.RemoveAll(c.ExportPath)
	os.RemoveAll(c.OriginalsPath)
	os.RemoveAll(c.ThumbnailsPath)
}

func (c *Config) DownloadTestData(t *testing.T) {
	if fileExists(testDataZip) {
		hash := fileHash(testDataZip)

		if hash != testDataHash {
			os.Remove(testDataZip)
			t.Logf("Removed outdated test data zip file (fingerprint %s)\n", hash)
		}
	}

	if !fileExists(testDataZip) {
		fmt.Printf("Downloading latest test data zip file from %s\n", testDataUrl)

		if err := downloadFile(testDataZip, testDataUrl); err != nil {
			fmt.Printf("Download failed: %s\n", err.Error())
		}
	}
}

func (c *Config) UnzipTestData(t *testing.T) {
	if _, err := unzip(testDataZip, testDataPath); err != nil {
		t.Logf("Could not unzip test data: %s\n", err.Error())
	}
}

func (c *Config) InitializeTestData(t *testing.T) {
	t.Log("Initializing test data")

	c.RemoveTestData(t)

	c.DownloadTestData(t)

	c.UnzipTestData(t)
}

func NewTestConfig() *Config {
	return &Config{
		DarktableCli:   darktableCli,
		OriginalsPath:  originalsPath,
		ThumbnailsPath: thumbnailsPath,
		ImportPath:     importPath,
		ExportPath:     exportPath,
		DatabaseDriver: databaseDriver,
		DatabaseDsn:    databaseDsn,
	}
}

func TestNewConfig(t *testing.T) {
	c := NewConfig()

	assert.IsType(t, &Config{}, c)
}

func TestConfig_SetValuesFromFile(t *testing.T) {
	c := NewConfig()

	c.SetValuesFromFile(GetExpandedFilename("config.example.yml"))

	assert.Equal(t, GetExpandedFilename("photos/originals"), c.OriginalsPath)
	assert.Equal(t, GetExpandedFilename("photos/thumbnails"), c.ThumbnailsPath)
	assert.Equal(t, GetExpandedFilename("photos/import"), c.ImportPath)
	assert.Equal(t, GetExpandedFilename("photos/export"), c.ExportPath)
	assert.Equal(t, databaseDriver, c.DatabaseDriver)
	assert.Equal(t, databaseDsn, c.DatabaseDsn)
}

func TestConfig_ConnectToDatabase(t *testing.T) {
	c := NewTestConfig()

	c.ConnectToDatabase()

	db := c.GetDb()

	assert.IsType(t, &gorm.DB{}, db)
}