From c1525ebc69b68f7203b71b4338a8ee30f947bd94 Mon Sep 17 00:00:00 2001
From: Xhofe <i@nn.ci>
Date: Mon, 28 Mar 2022 21:10:20 +0800
Subject: [PATCH] feat: cookie operate util

---
 utils/cookie/cookie.go | 59 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 utils/cookie/cookie.go

diff --git a/utils/cookie/cookie.go b/utils/cookie/cookie.go
new file mode 100644
index 00000000..8a6ca859
--- /dev/null
+++ b/utils/cookie/cookie.go
@@ -0,0 +1,59 @@
+package cookie
+
+import (
+	"net/http"
+	"strings"
+)
+
+func Parse(str string) []*http.Cookie {
+	header := http.Header{}
+	header.Add("Cookie", str)
+	request := http.Request{Header: header}
+	return request.Cookies()
+}
+
+func ToString(cookies []*http.Cookie) string {
+	if cookies == nil {
+		return ""
+	}
+	cookieStrings := make([]string, len(cookies))
+	for i, cookie := range cookies {
+		cookieStrings[i] = cookie.String()
+	}
+	return strings.Join(cookieStrings, ";")
+}
+
+func SetCookie(cookies []*http.Cookie, name, value string) []*http.Cookie {
+	for i, cookie := range cookies {
+		if cookie.Name == name {
+			cookies[i].Value = value
+			return cookies
+		}
+	}
+	cookies = append(cookies, &http.Cookie{Name: name, Value: value})
+	return cookies
+}
+
+func GetCookie(cookies []*http.Cookie, name string) *http.Cookie {
+	for _, cookie := range cookies {
+		if cookie.Name == name {
+			return cookie
+		}
+	}
+	return nil
+}
+
+func SetStr(cookiesStr, name, value string) string {
+	cookies := Parse(cookiesStr)
+	cookies = SetCookie(cookies, name, value)
+	return ToString(cookies)
+}
+
+func GetStr(cookiesStr, name string) string {
+	cookies := Parse(cookiesStr)
+	cookie := GetCookie(cookies, name)
+	if cookie == nil {
+		return ""
+	}
+	return cookie.Value
+}