go/ferret
1
0
Fork 0
mirror of https://github.com/MontFerret/ferret.git synced 2025-07-03 00:46:51 +02:00
Code Issues Releases Activity

build core.Functions with map

Browse Source
This commit is contained in:
3timeslazy
2019-10-15 21:06:48 +03:00
parent 55e82a9791
commit ad2ec2e817
12 changed files with 231 additions and 207 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
e2e/runner/lib.go
View File

@ -11,15 +11,19 @@ import (
)
func HTTPHelpers(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"GET": httpGet,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"GET": httpGet,
}),
)
}
func Assertions(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"EXPECT": expect,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"EXPECT": expect,
}),
)
}
func expect(_ context.Context, args ...core.Value) (core.Value, error) {

6
pkg/compiler/namespace.go
View File

@ -60,8 +60,10 @@ func (nc *NamespaceContainer) RemoveFunction(name string) {
nc.funcs.Unset(nc.makeFullName(name))
}
func (nc *NamespaceContainer) RegisterFunctions(funcs core.FunctionsMap) error {
for name, fun := range funcs {
func (nc *NamespaceContainer) RegisterFunctions(funcs *core.Functions) error {
for _, name := range funcs.Names() {
fun, _ := funcs.Get(name)
if err := nc.RegisterFunction(name, fun); err != nil {
return err
}

23
pkg/runtime/core/function.go
View File

@ -12,7 +12,7 @@ type (
Namespace interface {
Namespace(name string) Namespace
RegisterFunction(name string, fun Function) error
RegisterFunctions(funs FunctionsMap) error
RegisterFunctions(funs *Functions) error
RegisteredFunctions() []string
RemoveFunction(name string)
}
@ -37,12 +37,9 @@ func ValidateArgs(args []Value, minimum, maximum int) error {
type (
// Functions is a container for functions.
Functions struct {
functions FunctionsMap
functions map[string]Function
}
// FunctionsMap is a map of functions and their names.
FunctionsMap map[string]Function
// Function is a common interface for all functions of FQL.
Function = func(ctx context.Context, args ...Value) (Value, error)
)
@ -50,10 +47,22 @@ type (
// NewFunctions returns new empty Functions.
func NewFunctions() *Functions {
return &Functions{
functions: make(FunctionsMap),
functions: make(map[string]Function),
}
}
// NewFunctionsFromMap creates new Functions from map, where
// key is the name of the function and value is the function.
func NewFunctionsFromMap(funcs map[string]Function) *Functions {
fns := NewFunctions()
for name, fn := range funcs {
fns.Set(name, fn)
}
return fns
}
// Get returns the function with the given name. If the function
// does not exist it returns nil, false.
func (fns *Functions) Get(name string) (Function, bool) {
@ -67,7 +76,7 @@ func (fns *Functions) Set(name string, fn Function) {
// the preferred way to create Functions is NewFunctions.
// But just in case, if someone creates differently
if fns.functions == nil {
fns.functions = make(FunctionsMap, 1)
fns.functions = make(map[string]Function, 1)
}
fns.functions[strings.ToUpper(name)] = fn

49
pkg/stdlib/arrays/lib.go
View File

@ -6,30 +6,31 @@ import (
)
func RegisterLib(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"APPEND": Append,
"FIRST": First,
"FLATTEN": Flatten,
"INTERSECTION": Intersection,
"LAST": Last,
"MINUS": Minus,
"NTH": Nth,
"OUTERSECTION": Outersection,
"POP": Pop,
"POSITION": Position,
"PUSH": Push,
"REMOVE_NTH": RemoveNth,
"REMOVE_VALUE": RemoveValue,
"REMOVE_VALUES": RemoveValues,
"SHIFT": Shift,
"SLICE": Slice,
"SORTED": Sorted,
"SORTED_UNIQUE": SortedUnique,
"UNION": Union,
"UNION_DISTINCT": UnionDistinct,
"UNIQUE": Unique,
"UNSHIFT": Unshift,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"APPEND": Append,
"FIRST": First,
"FLATTEN": Flatten,
"INTERSECTION": Intersection,
"LAST": Last,
"MINUS": Minus,
"NTH": Nth,
"OUTERSECTION": Outersection,
"POP": Pop,
"POSITION": Position,
"PUSH": Push,
"REMOVE_NTH": RemoveNth,
"REMOVE_VALUE": RemoveValue,
"REMOVE_VALUES": RemoveValues,
"SHIFT": Shift,
"SLICE": Slice,
"SORTED": Sorted,
"SORTED_UNIQUE": SortedUnique,
"UNION": Union,
"UNION_DISTINCT": UnionDistinct,
"UNIQUE": Unique,
"UNSHIFT": Unshift,
}))
}
func ToUniqueArray(arr *values.Array) *values.Array {

9
pkg/stdlib/collections/lib.go
View File

@ -3,8 +3,9 @@ package collections
import "github.com/MontFerret/ferret/pkg/runtime/core"
func RegisterLib(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"LENGTH": Length,
"REVERSE": Reverse,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"LENGTH": Length,
"REVERSE": Reverse,
}))
}

4
pkg/stdlib/datetime/lib.go
View File

@ -2,8 +2,8 @@ package datetime
import "github.com/MontFerret/ferret/pkg/runtime/core"
func NewLib() core.FunctionsMap {
return core.FunctionsMap{
func NewLib() map[string]core.Function {
return map[string]core.Function{
"NOW": Now,
"DATE": Date,
"DATE_DAYOFWEEK": DateDayOfWeek,

117
pkg/stdlib/html/lib.go
View File

@ -11,64 +11,65 @@ import (
)
func RegisterLib(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"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,
})
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) {

71
pkg/stdlib/math/lib.go
View File

@ -16,41 +16,42 @@ const (
)
func RegisterLib(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"ABS": Abs,
"ACOS": Acos,
"ASIN": Asin,
"ATAN": Atan,
"ATAN2": Atan2,
"AVERAGE": Average,
"CEIL": Ceil,
"COS": Cos,
"DEGREES": Degrees,
"EXP": Exp,
"EXP2": Exp2,
"FLOOR": Floor,
"LOG": Log,
"LOG2": Log2,
"LOG10": Log10,
"MAX": Max,
"MEDIAN": Median,
"MIN": Min,
"PERCENTILE": Percentile,
"PI": Pi,
"POW": Pow,
"RADIANS": Radians,
"RAND": Rand,
"RANGE": Range,
"ROUND": Round,
"SIN": Sin,
"SQRT": Sqrt,
"STDDEV_POPULATION": StandardDeviationPopulation,
"STDDEV_SAMPLE": StandardDeviationSample,
"SUM": Sum,
"TAN": Tan,
"VARIANCE_POPULATION": PopulationVariance,
"VARIANCE_SAMPLE": SampleVariance,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"ABS": Abs,
"ACOS": Acos,
"ASIN": Asin,
"ATAN": Atan,
"ATAN2": Atan2,
"AVERAGE": Average,
"CEIL": Ceil,
"COS": Cos,
"DEGREES": Degrees,
"EXP": Exp,
"EXP2": Exp2,
"FLOOR": Floor,
"LOG": Log,
"LOG2": Log2,
"LOG10": Log10,
"MAX": Max,
"MEDIAN": Median,
"MIN": Min,
"PERCENTILE": Percentile,
"PI": Pi,
"POW": Pow,
"RADIANS": Radians,
"RAND": Rand,
"RANGE": Range,
"ROUND": Round,
"SIN": Sin,
"SQRT": Sqrt,
"STDDEV_POPULATION": StandardDeviationPopulation,
"STDDEV_SAMPLE": StandardDeviationSample,
"SUM": Sum,
"TAN": Tan,
"VARIANCE_POPULATION": PopulationVariance,
"VARIANCE_SAMPLE": SampleVariance,
}))
}
func toFloat(arg core.Value) float64 {

19
pkg/stdlib/objects/lib.go
View File

@ -3,13 +3,14 @@ package objects
import "github.com/MontFerret/ferret/pkg/runtime/core"
func RegisterLib(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"HAS": Has,
"KEYS": Keys,
"KEEP_KEYS": KeepKeys,
"MERGE": Merge,
"ZIP": Zip,
"VALUES": Values,
"MERGE_RECURSIVE": MergeRecursive,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"HAS": Has,
"KEYS": Keys,
"KEEP_KEYS": KeepKeys,
"MERGE": Merge,
"ZIP": Zip,
"VALUES": Values,
"MERGE_RECURSIVE": MergeRecursive,
}))
}

71
pkg/stdlib/strings/lib.go
View File

@ -3,39 +3,40 @@ package strings
import "github.com/MontFerret/ferret/pkg/runtime/core"
func RegisterLib(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"CONCAT": Concat,
"CONCAT_SEPARATOR": ConcatWithSeparator,
"CONTAINS": Contains,
"ESCAPE_HTML": EscapeHTML,
"DECODE_URI_COMPONENT": DecodeURIComponent,
"ENCODE_URI_COMPONENT": EncodeURIComponent,
"FIND_FIRST": FindFirst,
"FIND_LAST": FindLast,
"JSON_PARSE": JSONParse,
"JSON_STRINGIFY": JSONStringify,
"LEFT": Left,
"LIKE": Like,
"LOWER": Lower,
"LTRIM": LTrim,
"RANDOM_TOKEN": RandomToken,
"MD5": Md5,
"REGEXP_MATCH": RegexMatch,
"REGEXP_SPLIT": RegexSplit,
"REGEXP_TEST": RegexTest,
"REGEXP_REPLACE": RegexReplace,
"RIGHT": Right,
"RTRIM": RTrim,
"SHA1": Sha1,
"SHA512": Sha512,
"SPLIT": Split,
"SUBSTITUTE": Substitute,
"SUBSTRING": Substring,
"TO_BASE64": ToBase64,
"FROM_BASE64": FromBase64,
"TRIM": Trim,
"UPPER": Upper,
"FMT": Fmt,
"UNESCAPE_HTML": UnescapeHTML,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"CONCAT": Concat,
"CONCAT_SEPARATOR": ConcatWithSeparator,
"CONTAINS": Contains,
"ESCAPE_HTML": EscapeHTML,
"DECODE_URI_COMPONENT": DecodeURIComponent,
"ENCODE_URI_COMPONENT": EncodeURIComponent,
"FIND_FIRST": FindFirst,
"FIND_LAST": FindLast,
"JSON_PARSE": JSONParse,
"JSON_STRINGIFY": JSONStringify,
"LEFT": Left,
"LIKE": Like,
"LOWER": Lower,
"LTRIM": LTrim,
"RANDOM_TOKEN": RandomToken,
"MD5": Md5,
"REGEXP_MATCH": RegexMatch,
"REGEXP_SPLIT": RegexSplit,
"REGEXP_TEST": RegexTest,
"REGEXP_REPLACE": RegexReplace,
"RIGHT": Right,
"RTRIM": RTrim,
"SHA1": Sha1,
"SHA512": Sha512,
"SPLIT": Split,
"SUBSTITUTE": Substitute,
"SUBSTRING": Substring,
"TO_BASE64": ToBase64,
"FROM_BASE64": FromBase64,
"TRIM": Trim,
"UPPER": Upper,
"FMT": Fmt,
"UNESCAPE_HTML": UnescapeHTML,
}))
}

43
pkg/stdlib/types/lib.go
View File

@ -6,27 +6,28 @@ import (
)
func RegisterLib(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"TO_BOOL": ToBool,
"TO_INT": ToInt,
"TO_FLOAT": ToFloat,
"TO_STRING": ToString,
"TO_DATETIME": ToDateTime,
"TO_ARRAY": ToArray,
"IS_NONE": IsNone,
"IS_BOOL": IsBool,
"IS_INT": IsInt,
"IS_FLOAT": IsFloat,
"IS_STRING": IsString,
"IS_DATETIME": IsDateTime,
"IS_ARRAY": IsArray,
"IS_OBJECT": IsObject,
"IS_HTML_ELEMENT": IsHTMLElement,
"IS_HTML_DOCUMENT": IsHTMLDocument,
"IS_BINARY": IsBinary,
"IS_NAN": IsNaN,
"TYPENAME": TypeName,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"TO_BOOL": ToBool,
"TO_INT": ToInt,
"TO_FLOAT": ToFloat,
"TO_STRING": ToString,
"TO_DATETIME": ToDateTime,
"TO_ARRAY": ToArray,
"IS_NONE": IsNone,
"IS_BOOL": IsBool,
"IS_INT": IsInt,
"IS_FLOAT": IsFloat,
"IS_STRING": IsString,
"IS_DATETIME": IsDateTime,
"IS_ARRAY": IsArray,
"IS_OBJECT": IsObject,
"IS_HTML_ELEMENT": IsHTMLElement,
"IS_HTML_DOCUMENT": IsHTMLDocument,
"IS_BINARY": IsBinary,
"IS_NAN": IsNaN,
"TYPENAME": TypeName,
}))
}
func isTypeof(value core.Value, ctype core.Type) core.Value {

10
pkg/stdlib/utils/lib.go
View File

@ -3,8 +3,10 @@ package utils
import "github.com/MontFerret/ferret/pkg/runtime/core"
func RegisterLib(ns core.Namespace) error {
return ns.RegisterFunctions(core.FunctionsMap{
"WAIT": Wait,
"PRINT": Print,
})
return ns.RegisterFunctions(
core.NewFunctionsFromMap(map[string]core.Function{
"WAIT": Wait,
"PRINT": Print,
}),
)
}