package convert

import (
	"fmt"
	terminal2 "github.com/qeesung/image2ascii/terminal"
	"github.com/qeesung/image2ascii/terminal/mocks"
	"github.com/stretchr/testify/assert"
	"log"
	"testing"
)

// TestScaleImageWithFixedHeight test scale the image by fixed height
func TestScaleImageWithFixedHeight(t *testing.T) {
	handler := NewResizeHandler()
	assertions := assert.New(t)
	imageFilePath := "testdata/cat_2000x1500.jpg"
	img, err := OpenImageFile(imageFilePath)
	assertions.True(img != nil)
	assertions.True(err == nil)

	options := DefaultOptions
	options.Colored = false
	options.FixedHeight = 100

	scaledImage := handler.ScaleImage(img, &options)
	sz := scaledImage.Bounds()
	oldSz := img.Bounds()
	assertions.Equal(100, sz.Max.Y, "scaled image height should be 100")
	assertions.Equal(oldSz.Max.X, sz.Max.X, "scaled image width should be changed")
}

// TestScaleImageWithFixedWidth test scale the image by fixed width
func TestScaleImageWithFixedWidth(t *testing.T) {
	handler := NewResizeHandler()
	assertions := assert.New(t)
	imageFilePath := "testdata/cat_2000x1500.jpg"
	img, err := OpenImageFile(imageFilePath)
	assertions.True(img != nil)
	assertions.True(err == nil)

	options := DefaultOptions
	options.Colored = false
	options.FixedWidth = 200

	scaledImage := handler.ScaleImage(img, &options)
	sz := scaledImage.Bounds()
	oldSz := img.Bounds()
	assertions.Equal(oldSz.Max.Y, sz.Max.Y, "scaled image height should be changed")
	assertions.Equal(200, sz.Max.X, "scaled image width should be 200")
}

// TestScaleImageWithFixedWidthHeight test scale the image by fixed width
func TestScaleImageWithFixedWidthHeight(t *testing.T) {
	handler := NewResizeHandler()
	assertions := assert.New(t)
	imageFilePath := "testdata/cat_2000x1500.jpg"
	img, err := OpenImageFile(imageFilePath)
	assertions.True(img != nil)
	assertions.True(err == nil)

	options := DefaultOptions
	options.Colored = false
	options.FixedWidth = 200
	options.FixedHeight = 100

	scaledImage := handler.ScaleImage(img, &options)
	sz := scaledImage.Bounds()
	assertions.Equal(100, sz.Max.Y, "scaled image height should be 100")
	assertions.Equal(200, sz.Max.X, "scaled image width should be 200")
}

// TestScaleImageByRatio test scale image by ratio
func TestScaleImageByRatio(t *testing.T) {
	handler := NewResizeHandler()
	terminal := terminal2.NewTerminalAccessor()
	assertions := assert.New(t)
	imageFilePath := "testdata/cat_2000x1500.jpg"
	img, err := OpenImageFile(imageFilePath)
	assertions.True(img != nil)
	assertions.True(err == nil)

	options := DefaultOptions
	options.Colored = false
	options.Ratio = 0.5

	scaledImage := handler.ScaleImage(img, &options)
	sz := scaledImage.Bounds()
	oldSz := img.Bounds()
	expectedHeight := int(float64(oldSz.Max.Y) * 0.5 * terminal.CharWidth())
	expectedWidth := int(float64(oldSz.Max.X) * 0.5)
	assertions.Equal(expectedHeight, sz.Max.Y, fmt.Sprintf("scaled image height should be %d", expectedHeight))
	assertions.Equal(expectedWidth, sz.Max.X, fmt.Sprintf("scaled image width should be %d", expectedHeight))
}

// TestCalcFitSize test calc the fit size
func TestCalcFitSize(t *testing.T) {
	handler := ImageResizeHandler{
		terminal: terminal2.NewTerminalAccessor(),
	}
	fitSizeTests := []struct {
		width         int
		height        int
		toBeFitWidth  int
		toBeFitHeight int
		fitWidth      int
		fitHeight     int
	}{
		{width: 100, height: 80, toBeFitWidth: 50, toBeFitHeight: 120, fitWidth: 66, fitHeight: 80},
		{width: 100, height: 80, toBeFitWidth: 120, toBeFitHeight: 50, fitWidth: 100, fitHeight: 20},
	}
	for _, tt := range fitSizeTests {
		t.Run(fmt.Sprintf("%d, %d -> %d, %d",
			tt.width, tt.height, tt.toBeFitWidth, tt.toBeFitHeight), func(t *testing.T) {
			fitWidth, fitHeight := handler.CalcFitSize(
				float64(tt.width),
				float64(tt.height),
				float64(tt.toBeFitWidth),
				float64(tt.toBeFitHeight))
			if fitWidth != tt.fitWidth || fitHeight != tt.fitHeight {
				t.Errorf("%d, %d -> %d, %d should be %d, %d, but get %d, %d",
					tt.width, tt.height, tt.toBeFitWidth,
					tt.toBeFitHeight, tt.fitWidth, tt.fitHeight,
					fitWidth, fitHeight)
			}
		})
	}
}

func TestFitTheTerminalScreenSize(t *testing.T) {
	assertions := assert.New(t)
	// mock
	terminalMock := mocks.Terminal{}
	terminalMock.On("IsWindows").Return(false)
	terminalMock.On("CharWidth").Return(0.5)
	terminalMock.On("ScreenSize").Return(100, 80, nil)
	handler := ImageResizeHandler{
		terminal: &terminalMock,
	}

	imageFilePath := "testdata/cat_2000x1500.jpg"
	img, err := OpenImageFile(imageFilePath)
	if err != nil {
		log.Fatal("open image file " + imageFilePath + " failed")
	}

	assertions.False(terminalMock.IsWindows())
	assertions.Equal(terminalMock.CharWidth(), 0.5)
	screenWidth, screenHeight, err := terminalMock.ScreenSize()
	assertions.Equal(screenWidth, 100)
	assertions.Equal(screenHeight, 80)
	assertions.True(err == nil)

	options := DefaultOptions
	options.Colored = false
	options.FitScreen = true
	scaledImage := handler.ScaleImage(img, &options)
	sz := scaledImage.Bounds()
	assertions.Equal(100, sz.Max.X)
	assertions.Equal(37, sz.Max.Y)
}

func TestStretchTheTerminalScreenSize(t *testing.T) {
	assertions := assert.New(t)
	// mock
	terminalMock := mocks.Terminal{}
	terminalMock.On("IsWindows").Return(false)
	terminalMock.On("CharWidth").Return(0.5)
	terminalMock.On("ScreenSize").Return(100, 80, nil)
	handler := ImageResizeHandler{
		terminal: &terminalMock,
	}

	imageFilePath := "testdata/cat_2000x1500.jpg"
	img, err := OpenImageFile(imageFilePath)
	if err != nil {
		log.Fatal("open image file " + imageFilePath + " failed")
	}

	assertions.False(terminalMock.IsWindows())
	assertions.Equal(terminalMock.CharWidth(), 0.5)
	screenWidth, screenHeight, err := terminalMock.ScreenSize()
	assertions.Equal(screenWidth, 100)
	assertions.Equal(screenHeight, 80)
	assertions.True(err == nil)

	options := DefaultOptions
	options.Colored = false
	options.StretchedScreen = true
	options.FitScreen = false
	scaledImage := handler.ScaleImage(img, &options)
	sz := scaledImage.Bounds()
	assertions.Equal(100, sz.Max.X)
	assertions.Equal(80, sz.Max.Y)
}

// ExampleScaleImage is scale image example
func ExampleScaleImage() {
	handler := NewResizeHandler()
	imageFilePath := "testdata/cat_2000x1500.jpg"
	img, err := OpenImageFile(imageFilePath)
	if err != nil {
		log.Fatal("open image file " + imageFilePath + " failed")
	}

	options := DefaultOptions
	options.Colored = false
	options.FixedWidth = 200
	options.FixedHeight = 100

	scaledImage := handler.ScaleImage(img, &options)
	sz := scaledImage.Bounds()
	fmt.Print(sz.Max.X, sz.Max.Y)
	// output: 200 100
}

// BenchmarkScaleImage benchmark scale big image
func BenchmarkScaleBigImage(b *testing.B) {
	handler := NewResizeHandler()
	imageFilePath := "testdata/cat_2000x1500.jpg"
	img, err := OpenImageFile(imageFilePath)
	if err != nil {
		log.Fatalf("Open image file %s failed", imageFilePath)
	}

	options := DefaultOptions
	options.Colored = false
	options.FitScreen = false
	options.FixedHeight = 100
	options.FixedWidth = 100

	for i := 0; i < b.N; i++ {
		_ = handler.ScaleImage(img, &options)
	}
}

// BenchmarkScaleSmallImage benchmark scale small image
func BenchmarkScaleSmallImage(b *testing.B) {
	handler := NewResizeHandler()
	imageFilePath := "testdata/husky_200x200.jpg"
	img, err := OpenImageFile(imageFilePath)
	if err != nil {
		log.Fatalf("Open image file %s failed : %s", imageFilePath, err.Error())
	}

	options := DefaultOptions
	options.Colored = false
	options.FitScreen = false
	options.FixedHeight = 100
	options.FixedWidth = 100

	for i := 0; i < b.N; i++ {
		_ = handler.ScaleImage(img, &options)
	}
}