browser.go

package headless

import (
	"os"
	"path/filepath"

	"github.com/go-rod/rod"
	lua "github.com/yuin/gopher-lua"
)

var browserMethods = map[string]lua.LGFunction{
	"page":  browserPage,
	"close": browserClose,
}

func newBrowser() lua.LGFunction {
	return func(L *lua.LState) int {
		temp := os.TempDir()
		os.RemoveAll(filepath.Join(temp, "rod"))

		browser := rod.New()
		err := browser.Connect()

		if err != nil {
			L.Push(lua.LNil)
			L.Push(lua.LString(err.Error()))
			return 2
		}

		ud := L.NewUserData()
		ud.Value = browser
		L.SetMetatable(ud, L.GetTypeMetatable("browser"))

		L.Push(ud)
		L.Push(lua.LNil)
		return 2
	}
}

func registerBrowserType(L *lua.LState) {
	mt := L.NewTypeMetatable("browser")
	L.SetField(mt, "__index", L.SetFuncs(L.NewTable(), browserMethods))
}

func checkBrowser(L *lua.LState) *rod.Browser {
	ud := L.CheckUserData(1)
	if v, ok := ud.Value.(*rod.Browser); ok {
		return v
	}

	L.ArgError(1, "browser expected")
	return nil
}

func browserPage(L *lua.LState) int {
	browser := checkBrowser(L)
	url := L.ToString(2)

	p := browser.MustPage(url)

	ud := L.NewUserData()
	ud.Value = p
	L.SetMetatable(ud, L.GetTypeMetatable("browserPage"))

	L.Push(ud)
	return 1
}

func browserClose(L *lua.LState) int {
	browser := checkBrowser(L)
	err := browser.Close()

	if err != nil {
		L.Push(lua.LNil)
		L.Push(lua.LString(err.Error()))
		return 2
	}

	// L.Push(lua.LNil)
	L.Push(lua.LString("browser closed"))
	return 1
}