Update the StreamK.parseBreakChunks driver

Author: adithyaov

core/src/Streamly/Internal/Data/StreamK.hs

@@ -1337,28 +1337,31 @@ parseBreakChunks parser input = do
             -- If we stop in an alternative, it will try calling the next
             -- parser, the next parser may call initial returning Partial and
             -- then immediately we have to call extract on it.
-            ParserK.Partial 0 cont1 ->
+            ParserK.SPartial (-1) cont1 ->
                  go [] cont1 nil
-            ParserK.Partial n cont1 -> do
-                let n1 = negate n
+            ParserK.SPartial m cont1 -> do
+                let n = m + 1
+                    n1 = negate n
                 assertM(n1 >= 0 && n1 <= sum (Prelude.map Array.length backBuf))
                 let (s1, backBuf1) = backTrack n1 backBuf nil
                  in go backBuf1 cont1 s1
-            ParserK.Continue 0 cont1 ->
+            ParserK.SContinue (-1) cont1 ->
                 go backBuf cont1 nil
-            ParserK.Continue n cont1 -> do
-                let n1 = negate n
+            ParserK.SContinue m cont1 -> do
+                let n = m + 1
+                    n1 = negate n
                 assertM(n1 >= 0 && n1 <= sum (Prelude.map Array.length backBuf))
                 let (s1, backBuf1) = backTrack n1 backBuf nil
                  in go backBuf1 cont1 s1
-            ParserK.Done 0 b ->
+            ParserK.SDone (-1) b ->
                 return (Right b, nil)
-            ParserK.Done n b -> do
-                let n1 = negate n
+            ParserK.SDone m b -> do
+                let n = m + 1
+                    n1 = negate n
                 assertM(n1 >= 0 && n1 <= sum (Prelude.map Array.length backBuf))
                 let (s1, _) = backTrack n1 backBuf nil
                  in return (Right b, s1)
-            ParserK.Error _ err -> do
+            ParserK.SError _ err -> do
                 let (s1, _) = backTrack maxBound backBuf nil
                 return (Left (ParseError err), s1)
 
@@ -1370,7 +1373,8 @@ parseBreakChunks parser input = do
         pRes <- parserk (ParserK.Chunk arr)
         let len = Array.length arr
         case pRes of
-            ParserK.Partial n cont1 ->
+            ParserK.SPartial m cont1 -> do
+                let n = m + 1
                 case compare n len of
                     EQ -> go [] cont1 stream
                     LT -> do
@@ -1384,7 +1388,8 @@ parseBreakChunks parser input = do
                             let (s1, _) = backTrack n1 backBuf s
                             go [] cont1 s1
                     GT -> seekErr n len
-            ParserK.Continue n cont1 ->
+            ParserK.SContinue m cont1 -> do
+                let n = m + 1
                 case compare n len of
                     EQ -> go (arr:backBuf) cont1 stream
                     LT -> do
@@ -1398,12 +1403,13 @@ parseBreakChunks parser input = do
                             let (s1, backBuf1) = backTrack n1 backBuf s
                             go backBuf1 cont1 s1
                     GT -> seekErr n len
-            ParserK.Done n b -> do
-                let n1 = len - n
+            ParserK.SDone m b -> do
+                let n = m + 1
+                    n1 = len - n
                 assertM(n1 <= sum (Prelude.map Array.length (arr:backBuf)))
                 let (s1, _) = backTrack n1 (arr:backBuf) stream
                  in return (Right b, s1)
-            ParserK.Error _ err -> do
+            ParserK.SError _ err -> do
                 let (s1, _) = backTrack maxBound (arr:backBuf) stream
                 return (Left (ParseError err), s1)