package render_image

import (
	"encoding/base64"
	"fmt"
	"image"
	"image/draw"
	_ "image/jpeg"
	_ "image/png"
	"os"

	"whspbrd/pkg/resize_image"
)

func LoadImage(filePath string) (image.Image, error) {
	f, err := os.Open(filePath)
	if err != nil {
		return nil, err
	}
	defer f.Close()
	img, _, err := image.Decode(f)
	return img, err
}

func ConvertToRGBA(img image.Image) *image.RGBA {
	rgba := image.NewRGBA(img.Bounds())
	draw.Draw(rgba, rgba.Bounds(), img, image.Point{0, 0}, draw.Src)
	return rgba
}

func EncodeImageToBase64RGBA(rgba *image.RGBA) string {
	return base64.StdEncoding.EncodeToString(rgba.Pix)
}

func RenderImage(filepath string, row int, col int, width int, height int, units bool) {
	img, err := LoadImage(filepath)
	if err != nil {
		fmt.Printf("Error loading image: %v\n", err)
		return
	}

	rgba := ConvertToRGBA(img)
	if units {
		rgba, _ = resize_image.ResizeInTerminal(*rgba, width, height)
	} else {
		rgba, _ = resize_image.Resize(*rgba, width, height)
	}
	encoded := EncodeImageToBase64RGBA(rgba)

	imgWidth := rgba.Rect.Dx()
	imgHeight := rgba.Rect.Dy()

	fmt.Printf("\033[s\033[%d;%dH", row, col)

	chunkSize := 4096
	pos := 0
	first := true
	for pos < len(encoded) {
		fmt.Print("\033_G")
		if first {
			fmt.Printf("q=2,a=T,f=32,s=%d,v=%d,", imgWidth, imgHeight)
			first = false
		}
		chunkLen := len(encoded) - pos
		if chunkLen > chunkSize {
			chunkLen = chunkSize
		}
		if pos+chunkLen < len(encoded) {
			fmt.Print("m=1")
		}
		fmt.Printf(";%s\033\\", encoded[pos:pos+chunkLen])
		pos += chunkLen
	}
	fmt.Print("\033[u")
}