From e835103ac6cc6c66d24493d32d151b5eed52ca19 Mon Sep 17 00:00:00 2001
From: PoiScript <poiscript@gmail.com>
Date: Tue, 30 Jul 2019 21:20:39 +0800
Subject: [PATCH] feat(elements): update `Cookie` struct

Similar with org-elements-api.
---
 src/elements/cookie.rs | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/src/elements/cookie.rs b/src/elements/cookie.rs
index a12af6c..1de7e73 100644
--- a/src/elements/cookie.rs
+++ b/src/elements/cookie.rs
@@ -3,9 +3,8 @@ use memchr::{memchr, memchr2};
 #[cfg_attr(test, derive(PartialEq))]
 #[cfg_attr(feature = "serde", derive(serde::Serialize))]
 #[derive(Debug)]
-pub enum Cookie<'a> {
-    Percent(&'a str),
-    Slash(&'a str, &'a str),
+pub struct Cookie<'a> {
+    value: &'a str,
 }
 
 impl Cookie<'_> {
@@ -18,14 +17,21 @@ impl Cookie<'_> {
             memchr2(b'%', b'/', bytes).filter(|&i| bytes[1..i].iter().all(u8::is_ascii_digit))?;
 
         if bytes[num1] == b'%' && *bytes.get(num1 + 1)? == b']' {
-            Some((&src[num1 + 2..], Cookie::Percent(&src[1..num1])))
+            Some((
+                &src[num1 + 2..],
+                Cookie {
+                    value: &src[0..num1 + 2],
+                },
+            ))
         } else {
             let num2 = memchr(b']', bytes)
                 .filter(|&i| bytes[num1 + 1..i].iter().all(u8::is_ascii_digit))?;
 
             Some((
                 &src[num2 + 1..],
-                Cookie::Slash(&src[1..num1], &src[num1 + 1..num2]),
+                Cookie {
+                    value: &src[0..num2 + 1],
+                },
             ))
         }
     }
@@ -35,22 +41,25 @@ impl Cookie<'_> {
 fn parse() {
     assert_eq!(
         Cookie::parse("[1/10]"),
-        Some(("", Cookie::Slash("1", "10")))
+        Some(("", Cookie { value: "[1/10]" }))
     );
     assert_eq!(
         Cookie::parse("[1/1000]"),
-        Some(("", Cookie::Slash("1", "1000")))
+        Some(("", Cookie { value: "[1/1000]" }))
     );
-    assert_eq!(Cookie::parse("[10%]"), Some(("", Cookie::Percent("10"))));
-    assert_eq!(Cookie::parse("[%]"), Some(("", Cookie::Percent(""))));
-    assert_eq!(Cookie::parse("[/]"), Some(("", Cookie::Slash("", ""))));
+    assert_eq!(
+        Cookie::parse("[10%]"),
+        Some(("", Cookie { value: "[10%]" }))
+    );
+    assert_eq!(Cookie::parse("[%]"), Some(("", Cookie { value: "[%]" })));
+    assert_eq!(Cookie::parse("[/]"), Some(("", Cookie { value: "[/]" })));
     assert_eq!(
         Cookie::parse("[100/]"),
-        Some(("", Cookie::Slash("100", "")))
+        Some(("", Cookie { value: "[100/]" }))
     );
     assert_eq!(
         Cookie::parse("[/100]"),
-        Some(("", Cookie::Slash("", "100")))
+        Some(("", Cookie { value: "[/100]" }))
     );
 
     assert_eq!(Cookie::parse("[10% ]"), None);