package html

import (
	"context"
	"time"

	"github.com/MontFerret/ferret/pkg/drivers"
	"github.com/MontFerret/ferret/pkg/runtime/core"
	"github.com/MontFerret/ferret/pkg/runtime/values"
	"github.com/MontFerret/ferret/pkg/runtime/values/types"
)

func RegisterLib(ns core.Namespace) error {
	return ns.RegisterFunctions(
		core.NewFunctionsFromMap(map[string]core.Function{
			"ATTR_GET":          AttributeGet,
			"ATTR_REMOVE":       AttributeRemove,
			"ATTR_SET":          AttributeSet,
			"BLUR":              Blur,
			"COOKIE_DEL":        CookieDel,
			"COOKIE_GET":        CookieGet,
			"COOKIE_SET":        CookieSet,
			"CLICK":             Click,
			"CLICK_ALL":         ClickAll,
			"DOCUMENT":          Open,
			"DOWNLOAD":          Download,
			"ELEMENT":           Element,
			"ELEMENT_EXISTS":    ElementExists,
			"ELEMENTS":          Elements,
			"ELEMENTS_COUNT":    ElementsCount,
			"FOCUS":             Focus,
			"HOVER":             Hover,
			"INNER_HTML":        GetInnerHTML,
			"INNER_HTML_SET":    SetInnerHTML,
			"INNER_HTML_ALL":    GetInnerHTMLAll,
			"INNER_TEXT":        GetInnerText,
			"INNER_TEXT_SET":    SetInnerText,
			"INNER_TEXT_ALL":    GetInnerTextAll,
			"INPUT":             Input,
			"INPUT_CLEAR":       InputClear,
			"MOUSE":             MouseMoveXY,
			"NAVIGATE":          Navigate,
			"NAVIGATE_BACK":     NavigateBack,
			"NAVIGATE_FORWARD":  NavigateForward,
			"PAGINATION":        Pagination,
			"PDF":               PDF,
			"SCREENSHOT":        Screenshot,
			"SCROLL":            ScrollXY,
			"SCROLL_BOTTOM":     ScrollBottom,
			"SCROLL_ELEMENT":    ScrollInto,
			"SCROLL_TOP":        ScrollTop,
			"SELECT":            Select,
			"STYLE_GET":         StyleGet,
			"STYLE_REMOVE":      StyleRemove,
			"STYLE_SET":         StyleSet,
			"WAIT_ATTR":         WaitAttribute,
			"WAIT_NO_ATTR":      WaitNoAttribute,
			"WAIT_ATTR_ALL":     WaitAttributeAll,
			"WAIT_NO_ATTR_ALL":  WaitNoAttributeAll,
			"WAIT_ELEMENT":      WaitElement,
			"WAIT_NO_ELEMENT":   WaitNoElement,
			"WAIT_CLASS":        WaitClass,
			"WAIT_NO_CLASS":     WaitNoClass,
			"WAIT_CLASS_ALL":    WaitClassAll,
			"WAIT_NO_CLASS_ALL": WaitNoClassAll,
			"WAIT_STYLE":        WaitStyle,
			"WAIT_NO_STYLE":     WaitNoStyle,
			"WAIT_STYLE_ALL":    WaitStyleAll,
			"WAIT_NO_STYLE_ALL": WaitNoStyleAll,
			"WAIT_NAVIGATION":   WaitNavigation,
			"XPATH":             XPath,
		}))
}

func OpenOrCastPage(ctx context.Context, value core.Value) (drivers.HTMLPage, bool, error) {
	err := core.ValidateType(value, drivers.HTMLPageType, types.String)

	if err != nil {
		return nil, false, err
	}

	var page drivers.HTMLPage
	var closeAfter bool

	if value.Type() == types.String {
		buf, err := Open(ctx, value, values.NewBoolean(true))

		if err != nil {
			return nil, false, err
		}

		page = buf.(drivers.HTMLPage)
		closeAfter = true
	} else {
		page = value.(drivers.HTMLPage)
	}

	return page, closeAfter, nil
}

func waitTimeout(ctx context.Context, value values.Int) (context.Context, context.CancelFunc) {
	return context.WithTimeout(
		ctx,
		time.Duration(value)*time.Millisecond,
	)
}
