filter_example_test.go

package kooky_test

import (
	"context"
	"fmt"
	"net/http"
	"regexp"

	"github.com/browserutils/kooky"
)

// example regex matching base64 strings
var reBase64 = regexp.MustCompile(`^(?:[A-Za-z0-9+/]{4})*(?:[A-Za-z0-9+/]{2}==|[A-Za-z0-9+/]{3}=|[A-Za-z0-9+/]{4})$`)

func ExampleFilter_regex() {
	var cookies = []*kooky.Cookie{{Cookie: http.Cookie{Name: `test`, Value: `dGVzdA==`}}}

	ctx := context.Background()
	cookies = kooky.FilterCookies(
		ctx,
		cookies,
		ValueRegexMatch(reBase64), // filter cookies with the regex filter
		// kooky.Debug,            // print cookies after applying the regex filter
	).Collect(ctx)

	for _, cookie := range cookies {
		fmt.Println(cookie.Value)
		break // only first element
	}

	// Output: dGVzdA==
}

func ValueRegexMatch(re *regexp.Regexp) kooky.Filter {
	return kooky.FilterFunc(func(cookie *kooky.Cookie) bool {
		return cookie != nil && re != nil && re.Match([]byte(cookie.Value))
	})
}