--
-- Copyright 2017 Warlock <internalmike@gmail.com>
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
--     http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--

-- | At the first look, Data.Binary.Conduit.Get module is very similar with Data.Binary.Get.
-- The main differences between them are the following.
-- While the 'S.Get' from binary is a very custom monad,
-- the local 'Get' is 'ConduitM', which leads to easy integration in complicated format parsing.
-- The Data.Binary.Get module does not have a function to create custom 'S.Get' monad,
-- this module provides 'getC'.
-- Unlike 'isolate' from binary, local 'isolate' does not "cut" bytes counter.
-- While the binary's 'S.Get' is 'MonadFail', which leads to very ugly errors handling
-- in complicated cases, local 'Get' is 'MonadError'.

module Data.Conduit.Parsers.Text.TextOffset
  ( TextOffset (..)
  ) where

import qualified Data.Text as S (Text)
import qualified Data.Text as ST hiding (Text, head, last, tail, init)
import Data.Word
import Data.Conduit.Parsers
import Data.Conduit.Parsers.GetC
import Data.Conduit.Parsers.Text

data TextOffset = TextOffset Word64 Word64 Word64 deriving Show

instance DecodingState TextOffset where
  type DecodingToken TextOffset = S.Text
  decoded !i (TextOffset !o !l !c) =
    let newlines = reverse $ drop 1 $ ST.split (== '\n') i in
    TextOffset (o + fromIntegral (ST.length i)) (l + fromIntegral (length newlines)) $ case newlines of
      [] -> c + fromIntegral (ST.length i)
      (x : _) -> fromIntegral (ST.length x)
  {-# INLINE decoded #-}

instance DecodingElemsRead TextOffset where
  decodingElemsRead (TextOffset !o _ _) = o
  {-# INLINE decodingElemsRead #-}

instance DecodingLinesRead TextOffset where
  decodingLinesRead (TextOffset _ !l _) = l
  {-# INLINE decodingLinesRead #-}

instance DecodingColumnsRead TextOffset where
  decodingColumnsRead (TextOffset _ _ !c) = c
  {-# INLINE decodingColumnsRead #-}