mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-23 01:24:30 +02:00
43 lines
919 B
Go
43 lines
919 B
Go
package html
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/MontFerret/ferret/pkg/drivers"
|
|
"github.com/MontFerret/ferret/pkg/runtime/core"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values"
|
|
)
|
|
|
|
// AttributeGet gets single or more attribute(s) of a given element.
|
|
// @param el (HTMLElement) - Target element.
|
|
// @param names (...String) - Attribute name(s).
|
|
// @returns Object - Key-value pairs of attribute values.
|
|
func AttributeGet(ctx context.Context, args ...core.Value) (core.Value, error) {
|
|
err := core.ValidateArgs(args, 2, core.MaxArgs)
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
el, err := drivers.ToElement(args[0])
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
names := args[1:]
|
|
result := values.NewObject()
|
|
attrs := el.GetAttributes(ctx)
|
|
|
|
for _, n := range names {
|
|
name := values.NewString(n.String())
|
|
val, exists := attrs.Get(name)
|
|
|
|
if exists {
|
|
result.Set(name, val)
|
|
}
|
|
}
|
|
|
|
return result, nil
|
|
}
|