package main

import (
	"os"
	"path/filepath"
	"time"
	vnc "github.com/amitbet/vnc2video"
	"github.com/amitbet/vnc2video/encoders"
	"github.com/amitbet/vnc2video/logger"
)

func main() {
	framerate := 10
	speedupFactor := 3.0
	fastFramerate := int(float64(framerate) * speedupFactor)

	if len(os.Args) <= 1 {
		logger.Errorf("please provide a fbs file name")
		return
	}
	if _, err := os.Stat(os.Args[1]); os.IsNotExist(err) {
		logger.Errorf("File doesn't exist", err)
		return
	}
	encs := []vnc.Encoding{
		&vnc.RawEncoding{},
		&vnc.TightEncoding{},
		&vnc.CopyRectEncoding{},
		&vnc.ZRLEEncoding{},
	}

	fbs, err := vnc.NewFbsConn(
		os.Args[1],
		encs,
	)
	if err != nil {
		logger.Error("failed to open fbs reader:", err)
		//return nil, err
	}

	//launch video encoding process:
	vcodec := &encoders.X264ImageEncoder{FFMpegBinPath: "./ffmpeg", Framerate: framerate}
	//vcodec := &encoders.DV8ImageEncoder{}
	//vcodec := &encoders.DV9ImageEncoder{}
	dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
	logger.Tracef("current dir: %s", dir)
	go vcodec.Run("./output.mp4")

	//screenImage := image.NewRGBA(image.Rect(0, 0, int(fbs.Width()), int(fbs.Height())))
	screenImage := vnc.NewVncCanvas(int(fbs.Width()), int(fbs.Height()))
	screenImage.DrawCursor = false

	for _, enc := range encs {
		myRenderer, ok := enc.(vnc.Renderer)

		if ok {
			myRenderer.SetTargetImage(screenImage)
		}
	}

	go func() {
		frameMillis := (1000.0 / float64(fastFramerate)) - 1 //a couple of millis, adjusting for time lost in software commands
		frameDuration := time.Duration(frameMillis * float64(time.Millisecond))
		//logger.Error("milis= ", frameMillis)

		for {
			timeStart := time.Now()

			vcodec.Encode(screenImage.Image)
			timeTarget := timeStart.Add(frameDuration)
			timeLeft := timeTarget.Sub(time.Now())
			//.Add(1 * time.Millisecond)
			if timeLeft > 0 {
				time.Sleep(timeLeft)
				//logger.Error("sleeping= ", timeLeft)
			}
		}
	}()

	msgReader := vnc.NewFBSPlayHelper(fbs)
	//loop over all messages, feed images to video codec:
	for {
		_, err := msgReader.ReadFbsMessage(true, speedupFactor)
		//vcodec.Encode(screenImage.Image)
		if err != nil {
			os.Exit(-1)
		}
		//vcodec.Encode(screenImage)
	}
}