refactoring

This commit is contained in:
y a t s
2026-06-19 10:42:39 -04:00
parent 74b923f06f
commit c7ccfc65bf
3 changed files with 25 additions and 33 deletions
+10 -8
View File
@@ -6,12 +6,13 @@ import (
)
// http.CookieJar interface wrapper to add convenience functions.
type jar struct {
http.CookieJar
type cookieJar struct {
jar *http.CookieJar
}
func (j *jar) GetCookie(u *url.URL, name string) *http.Cookie {
cookies := j.Cookies(u)
func (j *cookieJar) GetCookie(u *url.URL, name string) *http.Cookie {
jar := (*j.jar)
cookies := jar.Cookies(u)
for _, c := range cookies {
if c.Name == name {
return c
@@ -21,18 +22,19 @@ func (j *jar) GetCookie(u *url.URL, name string) *http.Cookie {
return nil
}
func (j *jar) SetCookie(u *url.URL, newCookie *http.Cookie) {
cookies := j.Cookies(u)
func (j *cookieJar) SetCookie(u *url.URL, newCookie *http.Cookie) {
jar := (*j.jar)
cookies := jar.Cookies(u)
for i, c := range cookies {
if c.Name == newCookie.Name {
cookies[i] = newCookie
j.SetCookies(u, cookies)
jar.SetCookies(u, cookies)
return
}
}
// Append if not already existing.
cookies = append(cookies, newCookie)
j.SetCookies(u, cookies)
jar.SetCookies(u, cookies)
}