From 3c0ed789dfa6b67a7853da0f2ea744a70c528d3a Mon Sep 17 00:00:00 2001
From: tv <tv@shackspace.de>
Date: Sun, 28 Dec 2014 03:36:43 +0100
Subject: store Trammel String in buffers

---
 src/Trammel.hs | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/src/Trammel.hs b/src/Trammel.hs
index 36c1140..bd3cd32 100644
--- a/src/Trammel.hs
+++ b/src/Trammel.hs
@@ -4,6 +4,7 @@
 {-# LANGUAGE OverloadedLists #-}
 {-# LANGUAGE OverloadedStrings #-}
 {-# LANGUAGE RecordWildCards #-}
+{-# LANGUAGE LambdaCase #-}
 module Trammel where
 
 import Control.Applicative
@@ -214,3 +215,33 @@ pp t = renderString emptyRenderState t ""
 renderSGR :: Pm -> String
 renderSGR [] = []
 renderSGR xs = ("\ESC["++) . (++"m") . intercalate ";" $ map show xs
+
+
+trammelDrop :: Int -> Trammel String -> Trammel String
+trammelDrop n = \case
+    Append t1 t2 ->
+        case compare n (len t1) of
+            LT -> Append (trammelDrop n t1) t2
+            EQ -> t2
+            GT -> trammelDrop (n - len t1) t2
+    Plain s ->
+        Plain (drop n s)
+    SGR pm t ->
+        SGR pm (trammelDrop n t)
+    Empty ->
+        Empty
+
+
+trammelTake :: Int -> Trammel String -> Trammel String
+trammelTake n = \case
+    Append t1 t2 ->
+        case compare n (len t1) of
+            LT -> trammelTake n t1
+            EQ -> t1
+            GT -> Append t1 (trammelTake (n - len t1) t2)
+    Plain s ->
+        Plain (take n s)
+    SGR pm t ->
+        SGR pm (trammelTake n t)
+    Empty ->
+        Empty
-- 
cgit v1.2.3