added archive, megamillions, and powerball site logic

cookiejar.go

@@ -1,20 +1,58 @@
 package extractor
 
 import (
+	"net/url"
+	"strings"
 	"time"
 )
 
 type Cookie struct {
-	Name     string
-	Value    string
-	Domain   string
+	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
 }
@@ -29,6 +67,10 @@ 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
 }