blob: b98359a2146e7d73ff8d400e57a5ccf1bd031f09 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
{-# LANGUAGE ViewPatterns, OverloadedStrings, FlexibleContexts #-}
module Utils
( showEntity
, takeR
) where
import Data.Text (Text)
import qualified Data.Text as Text
import Data.PQueue.Prio.Max (MaxPQueue)
import qualified Data.PQueue.Prio.Max as MaxPQueue
import Data.Monoid (Monoid(..), (<>))
import Control.Monad.State.Class
import Control.Monad.Writer.Class
import Data.Set (Set)
import qualified Data.Set as Set
import Data.Sequence (Seq, ViewR(..), (|>))
import qualified Data.Sequence as Seq
import Data.CaseInsensitive ( CI )
import qualified Data.CaseInsensitive as CI
import Types
showEntity :: Entity -> Text
showEntity (Entity name number)
| (Just (show -> n)) <- number = CI.original name <> " № " <> Text.pack n
| otherwise = CI.original name
takeR :: Integer -> Seq a -> Seq a
takeR _ (Seq.viewr -> EmptyR) = Seq.empty
takeR n (Seq.viewr -> (xs :> x))
| n <= 0 = Seq.empty
| otherwise = takeR (n - 1) xs |> x
|