go-extractor/cookiejar.go

package extractor

import (
	"net/url"
	"strings"
	"time"
)

type Cookie struct {
	Host     string
	Path     string
	Expires  time.Time
	Secure   bool
	HttpOnly bool
	Name     string
	Value    string
}

func (c Cookie) IsTargetMatch(target string) (bool, error) {
	u, err := url.Parse(target)
	if err != nil {
		return false, err
	}

	// the host of the cookie is the same as the host of the target
	// if the cookie host starts with a dot, that means it matches any subdomain
	if c.Host == u.Host || strings.HasPrefix(c.Host, ".") && strings.HasSuffix(u.Host, c.Host) {
		if c.Path != "" {
			if !strings.HasPrefix(u.Path, c.Path) {
				return false, nil
			}

			// if the cookie path is a prefix of the target path, then it's a match
			// so now these would both match:
			// cookie path: /foo
			// target path: /foo/bar
			// cookie path: /foo
			// target path: /foosball
			// because foseball is not an actual match, we need to check to see that either the path is an exact match
			// or that the next character in the target path is a slash

			if len(u.Path) > len(c.Path) && u.Path[len(c.Path)] != '/' {
				return false, nil
			}

			return true, nil
		}
	}

	return false, nil
}

type CookieJar interface {
	GetAll() ([]Cookie, error)
	Get(url string) ([]Cookie, error)
	Set(cookie Cookie) error
	Delete(cookie Cookie) error
}

// ReadOnlyCookieJar is a wrapper for CookieJar that allows only read operations on cookies, but all
// write operations are no-ops.
type ReadOnlyCookieJar struct {
	Jar CookieJar
}

func (r ReadOnlyCookieJar) GetAll() ([]Cookie, error) {
	return r.Jar.GetAll()
}

func (r ReadOnlyCookieJar) Get(url string) ([]Cookie, error) {
	return r.Jar.Get(url)
}

func (r ReadOnlyCookieJar) Set(_ Cookie) error {
	return nil
}

func (r ReadOnlyCookieJar) Delete(_ Cookie) error {
	return nil
}