mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-16 11:37:36 +02:00
acf2f13dcb
* sync with MontFerret/ferret * fix --param handling When params is converted to map it uses strings.Split, which slices a string into all substrings separated by :. * remove impossible conditions nil != nil * delete ineffectual assignments * replace '+= 1' with '++' * remove useless comparison with nil * merge variable declarations * remove bool comparison * fix imports * fix imports * delete unused file * use copy instead of loop * delete unused DummyInterface * remove unnecassary break statements * tidy modules
131 lines
2.6 KiB
Go
131 lines
2.6 KiB
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
|
|
"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 SetInDocument(ctx context.Context, doc drivers.HTMLDocument, path []core.Value, value core.Value) error {
|
|
if len(path) == 0 {
|
|
return nil
|
|
}
|
|
|
|
segment := path[0]
|
|
|
|
if segment.Type() == types.String {
|
|
segment := segment.(values.String)
|
|
|
|
switch segment {
|
|
case "url", "URL":
|
|
return doc.SetURL(ctx, values.NewString(value.String()))
|
|
case "cookies":
|
|
|
|
default:
|
|
return SetInNode(ctx, doc, path, value)
|
|
}
|
|
}
|
|
|
|
return SetInNode(ctx, doc, path, value)
|
|
}
|
|
|
|
func SetInElement(ctx context.Context, el drivers.HTMLElement, path []core.Value, value core.Value) error {
|
|
if len(path) == 0 {
|
|
return nil
|
|
}
|
|
|
|
segment := path[0]
|
|
|
|
if segment.Type() == types.String {
|
|
segment := segment.(values.String)
|
|
|
|
switch segment {
|
|
case "attributes":
|
|
if len(path) > 1 {
|
|
attrName := path[1]
|
|
|
|
return el.SetAttribute(ctx, values.NewString(attrName.String()), values.NewString(value.String()))
|
|
}
|
|
|
|
err := core.ValidateType(value, types.Object)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
curr := el.GetAttributes(ctx)
|
|
|
|
// remove all previous attributes
|
|
err = el.RemoveAttribute(ctx, curr.Keys()...)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
obj := value.(*values.Object)
|
|
obj.ForEach(func(value core.Value, key string) bool {
|
|
err = el.SetAttribute(ctx, values.NewString(key), values.NewString(value.String()))
|
|
|
|
return err == nil
|
|
})
|
|
|
|
return err
|
|
case "style":
|
|
if len(path) > 1 {
|
|
attrName := path[1]
|
|
|
|
return el.SetStyle(ctx, values.NewString(attrName.String()), value)
|
|
}
|
|
|
|
err := core.ValidateType(value, types.Object)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
styles, err := el.GetStyles(ctx)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = el.RemoveStyle(ctx, styles.Keys()...)
|
|
|
|
obj := value.(*values.Object)
|
|
obj.ForEach(func(value core.Value, key string) bool {
|
|
err = el.SetStyle(ctx, values.NewString(key), value)
|
|
|
|
return err == nil
|
|
})
|
|
|
|
return err
|
|
case "value":
|
|
if len(path) > 1 {
|
|
return core.Error(ErrInvalidPath, PathToString(path[1:]))
|
|
}
|
|
|
|
return el.SetValue(ctx, value)
|
|
}
|
|
}
|
|
|
|
return SetInNode(ctx, el, path, value)
|
|
}
|
|
|
|
func SetInNode(_ context.Context, _ drivers.HTMLNode, path []core.Value, _ core.Value) error {
|
|
if len(path) == 0 {
|
|
return nil
|
|
}
|
|
|
|
segment := path[0]
|
|
st := segment.Type()
|
|
|
|
if st == types.Int {
|
|
return core.Error(core.ErrInvalidOperation, "children are read-only")
|
|
}
|
|
|
|
return core.Error(ErrReadOnly, PathToString(path))
|
|
}
|