//go:build ssr
// +build ssr

package main

import (
	"embed"
	"filething/ui"
	"io"
	"os"
	"path/filepath"
	"strings"

	"github.com/gofiber/fiber/v3"
	"github.com/gofiber/fiber/v3/middleware/proxy"
)

// func init() {
// 	initUi = func(app *fiber.App) {
// 		tmpDir, err := os.MkdirTemp("", "filething-ssr")
// 		if err != nil {
// 			panic(err)
// 		}

// 		err = copyEmbeddedFiles(ui.DistDir, ".output", tmpDir)
// 		if err != nil {
// 			panic(err)
// 		}

// 		path := filepath.Join(tmpDir, "server/index.mjs")
// 		spawnProcess("node", []string{path}, app)

// 		// target := "localhost:3000"
// 		// e.Group("/*").Use(echoMiddleware.ProxyWithConfig(echoMiddleware.ProxyConfig{
// 		// 	Balancer: echoMiddleware.NewRoundRobinBalancer([]*echoMiddleware.ProxyTarget{
// 		// 		{URL: &url.URL{
// 		// 			Scheme: "http",
// 		// 			Host:   target,
// 		// 		}},
// 		// 	}),
// 		// }))
// 		app.Get("/*", proxy.Forward("http://localhost:3000"))
// 	}
// }

func init() {
	initUi = func(app *fiber.App) {
		if !fiber.IsChild() {
			tmpDir, err := os.MkdirTemp("", "filething-ssr")
			if err != nil {
				panic(err)
			}

			err = copyEmbeddedFiles(ui.DistDir, ".output", tmpDir)
			if err != nil {
				panic(err)
			}

			path := filepath.Join(tmpDir, "server/index.mjs")

			spawnProcess("node", []string{path}, app)
		}

		target := "localhost:3000"
		app.All("/*", func(c fiber.Ctx) error {
			path := c.Path()
			if strings.HasPrefix(path, "/api") {
				return c.Next()
			}

			request := c.Request().URI()
			return proxy.Do(c, "http://"+target+string(request.RequestURI()))
		})
	}
}

func copyEmbeddedFiles(fs embed.FS, sourcePath string, targetDir string) error {
	entries, err := fs.ReadDir(sourcePath)
	if err != nil {
		return err
	}

	for _, entry := range entries {
		sourceFile := filepath.Join(sourcePath, entry.Name())
		destFile := filepath.Join(targetDir, entry.Name())

		if entry.IsDir() {
			os.Mkdir(destFile, 0755)
			err := copyEmbeddedFiles(fs, sourceFile, destFile)
			if err != nil {
				return err
			}
		} else {
			source, err := fs.Open(sourceFile)
			if err != nil {
				return err
			}
			defer source.Close()

			dest, err := os.Create(destFile)
			if err != nil {
				return err
			}
			defer dest.Close()

			_, err = io.Copy(dest, source)
			if err != nil {
				return err
			}
		}
	}
	return nil
}