summaryrefslogtreecommitdiff
path: root/fig-utils/src/Fig/Prelude.hs
blob: 85a3ab325bae6925a8e337073181a36689419f09 (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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
{-# Language UndecidableInstances #-}

module Fig.Prelude
  ( quot, mod, rem, quotRem
  , module GHC.Generics
  , module GHC.Num
  , module GHC.Float

  , module System.IO
  , module System.FilePath.Posix

  , module Data.Kind
  , module Data.Void
  , module Data.Bool
  , module Data.Char
  , module Data.Int
  , module Data.Text
  , module Data.Text.IO
  , module Data.Text.Encoding
  , module Data.ByteString
  , module Data.Tuple
  , module Data.Maybe
  , module Data.Either
  , module Data.List
  , module Data.Function
  , module Data.Eq
  , module Data.Ord
  , module Data.Semigroup
  , module Data.Monoid
  , module Data.Functor
  , module Data.Bifunctor
  , module Data.Traversable
  , module Data.Foldable

  , module Text.Show
  , module Text.Read

  , module Control.Applicative
  , module Control.Monad
  , module Control.Monad.IO.Class
  , module Control.Monad.State.Class
  , module Control.Monad.Reader.Class
  , module Control.Exception.Safe

  , tshow
  , headMay, atMay
  , throwLeft
  , log

  , Pretty(..)
  , Fix(..), unFix
  ) where

import Prelude (quot, mod, rem, quotRem)

import GHC.Generics (Generic)
import GHC.Num (Num(..), Integer)
import GHC.Float (Double)

import System.IO (IO, stdin, stdout, stderr, FilePath, Handle)
import System.FilePath.Posix ((</>))

import Data.Kind (Type)
import Data.Void (Void)
import Data.Bool (Bool(..), otherwise, not, (&&), (||))
import Data.Char (Char, isUpper)
import Data.Int (Int)
import Data.Text (Text, pack, unpack, unwords)
import Data.Text.IO (hPutStrLn)
import Data.Text.Encoding (decodeUtf8, decodeUtf8', encodeUtf8)
import Data.ByteString (ByteString, readFile, writeFile)
import Data.Tuple (fst, snd, curry, uncurry, swap)
import Data.Maybe (Maybe(..), maybe, fromMaybe, isJust, catMaybes)
import Data.Either (Either(..))
import Data.List (take, drop, dropWhile, filter, reverse, lookup, zip, zip3, replicate, sortOn, concatMap, elemIndex)
import Data.Function (id, const, flip, ($), (&), (.))
import Data.Eq (Eq(..))
import Data.Ord (Ord(..), Down(..))
import Data.Semigroup(Semigroup(..), (<>))
import Data.Monoid (Monoid(..), mconcat)
import Data.Functor (Functor(..), (<$>), (<$), ($>))
import Data.Bifunctor (Bifunctor(..), first, second)
import Data.Traversable (Traversable(..), forM, sequence)
import Data.Foldable (Foldable(..), any, all, mapM_, forM_)
import qualified Data.Time.Clock as Time
import qualified Data.Time.Format as Time

import Text.Show (Show(..))
import Text.Read (readMaybe)

import Control.Applicative (Applicative(..), (<*), (*>))
import Control.Monad (Monad(..), join, forever, mapM, forM, foldM, void, (>>=), (=<<), (>=>), (<=<))
import Control.Monad.IO.Class (MonadIO(..))
import Control.Monad.State.Class (MonadState(..), get, put, modify)
import Control.Monad.Reader.Class (MonadReader(..), ask)
import Control.Exception.Safe (Exception, SomeException, IOException, MonadThrow, MonadCatch, MonadMask, throwM, try, catch, catchIO, bracket, bracketOnError)

tshow :: Show a => a -> Text
tshow = pack . show

headMay :: [a] -> Maybe a
headMay [] = Nothing
headMay (x:_) = Just x

atMay :: [a] -> Int -> Maybe a
atMay [] _ = Nothing
atMay (x:_) 0 = Just x
atMay (_:xs) n = atMay xs $ n - 1

throwLeft :: (Exception e, MonadThrow m) => (b -> e) -> Either b a -> m a
throwLeft f (Left x) = throwM $ f x
throwLeft _ (Right x) = pure x

log :: MonadIO m => Text -> m ()
log msg = do
  t <- liftIO Time.getCurrentTime
  let time = Time.formatTime Time.defaultTimeLocale "[%F %T] " t
  liftIO . hPutStrLn stderr $ pack time <> msg

class Pretty a where
  pretty :: a -> Text

instance Pretty Void where
  pretty _ = ""

newtype Fix f = Fix { unFix :: f (Fix f) }
unFix :: Fix f -> f (Fix f)
unFix (Fix x) = x
instance Pretty (f (Fix f)) => Pretty (Fix f) where
  pretty (Fix x) = pretty x
instance Show (f (Fix f)) => Show (Fix f) where
  show (Fix x) = show x