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
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
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"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values/types"
|
|
)
|
|
|
|
// CookieSet gets a cookie from a given document by name.
|
|
// @param source (HTMLDocument) - Target HTMLDocument.
|
|
// @param cookie (...HTTPCookie|String) - Cookie or cookie name to delete.
|
|
func CookieDel(ctx context.Context, args ...core.Value) (core.Value, error) {
|
|
err := core.ValidateArgs(args, 2, core.MaxArgs)
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
err = core.ValidateType(args[0], drivers.HTMLDocumentType)
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
doc := args[0].(drivers.HTMLDocument)
|
|
inputs := args[1:]
|
|
var currentCookies *values.Array
|
|
cookies := make([]drivers.HTTPCookie, 0, len(inputs))
|
|
|
|
for _, c := range inputs {
|
|
switch cookie := c.(type) {
|
|
case values.String:
|
|
if currentCookies == nil {
|
|
current, err := doc.GetCookies(ctx)
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
currentCookies = current
|
|
}
|
|
|
|
found, isFound := currentCookies.Find(func(value core.Value, _ int) bool {
|
|
cv := value.(drivers.HTTPCookie)
|
|
|
|
return cv.Name == cookie.String()
|
|
})
|
|
|
|
if isFound {
|
|
cookies = append(cookies, found.(drivers.HTTPCookie))
|
|
}
|
|
|
|
case drivers.HTTPCookie:
|
|
cookies = append(cookies, cookie)
|
|
|
|
default:
|
|
return values.None, core.TypeError(c.Type(), types.String, drivers.HTTPCookieType)
|
|
}
|
|
}
|
|
|
|
return values.None, doc.DeleteCookies(ctx, cookies...)
|
|
}
|