feat: add cookie_utils

Author: ilotoki0804

src/httpc/__main__.py

@@ -9,6 +9,8 @@
 from argparse import ArgumentParser
 import typing
 
+from httpc.cookie_utils import cookie_json_to_raw
+
 from ._base import __version__, logger
 
 T = typing.TypeVar("T")
@@ -395,10 +397,7 @@ def _handle_cookies(args) -> None:
         text = Path(args.file).read_text("utf-8")
 
     cookies = json.loads(text.strip())
-    cookies_text = []
-    for cookie in cookies:
-        cookies_text.append(f'{cookie["name"]}={cookie["value"]}')
-    value = "; ".join(cookies_text)
+    value = cookie_json_to_raw(cookies)
 
     print(json.dumps(value))
 

src/httpc/cookie_utils.py

@@ -0,0 +1,25 @@
+from http.cookies import Morsel, SimpleCookie
+import typing
+
+
+def cookie_json_to_raw(cookies: dict, name: str = "name", value: str = "value") -> str:
+    cookies_text = []
+    for cookie in cookies:
+        cookies_text.append(f'{cookie[name]}={cookie[value]}')
+    return "; ".join(cookies_text)
+
+
+def cookie_dict_to_raw(cookies: dict[str, Morsel] | dict[str, str]) -> str:
+    cookies_text = []
+    for name, value in cookies.items():
+        if isinstance(value, Morsel):
+            value = value.coded_value
+        cookies_text.append(f'{name}={value}')
+    return "; ".join(cookies_text)
+
+
+def set_cookies(cookie_string: str, cookies: typing.Mapping[str, str] | dict[str, Morsel]) -> str:
+    cookie = SimpleCookie(cookie_string)
+    for name, value in cookies.items():
+        cookie[name] = value
+    return cookie_dict_to_raw(cookie)