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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
{-# LANGUAGE RecordWildCards #-}
{-# LANGUAGE EmptyDataDecls #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE TypeFamilies #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TypeOperators #-}
module Thermoprint.Server
( thermoprintServer
, Config(..)
, module Data.Default.Class
, module Servant.Server.Internal.Enter
) where
import Data.Default.Class
import qualified Config.Dyre as Dyre
import System.IO (hPutStrLn, stderr)
import System.Exit (exitFailure)
import Control.Monad ((<=<), mapM_, liftM2)
import Prelude hiding ((.), id)
import Control.Category
import Control.Monad.Logger
import Control.Monad.Reader
import Control.Monad.Trans.Resource
import Control.Monad.Trans.Either
import Control.Monad.IO.Class
import Data.Functor.Compose
import Thermoprint.API hiding (JobId(..), DraftId(..))
import qualified Thermoprint.API as API (JobId(..), DraftId(..))
import Data.Set (Set)
import qualified Data.Set as Set
import Data.Sequence (Seq)
import qualified Data.Sequence as Seq
import Data.Map (Map)
import qualified Data.Map as Map
import qualified Network.Wai.Handler.Warp as Warp
import Network.Wai (Application)
import Servant
import Servant.Server
import Servant.Server.Internal.Enter
import Database.Persist
import Database.Persist.Sql
import Database.Persist.TH
data Config = Config { dyreError :: Maybe String
, warpSettings :: Warp.Settings
}
instance Default Config where
def = Config { dyreError = Nothing
, warpSettings = Warp.defaultSettings
}
data HandlerInput = HandlerInput { sqlPool :: ConnectionPool
}
share [mkPersist sqlSettings, mkMigrate "migrateAll"] [persistLowerCase|
Job
content Printout
Draft
title DraftTitle Maybe
content Printout
|]
thermoprintServer :: ( MonadLoggerIO m
, MonadIO m
, MonadBaseControl IO m
, MonadReader ConnectionPool m
) => (m :~> IO) -> Config -> IO ()
thermoprintServer io = Dyre.wrapMain $ Dyre.defaultParams
{ Dyre.projectName = "thermoprint-server"
, Dyre.realMain = realMain <=< handleDyreErrors
, Dyre.showError = (\cfg msg -> cfg { dyreError = Just msg })
}
where
handleDyreErrors cfg@(Config{..})
| Just msg <- dyreError = do
hPutStrLn stderr msg
exitFailure
undefined
| otherwise = return cfg
realMain (Config{..}) = enter io $ do
sqlPool <- ask
runSqlPool (runMigrationSilent migrateAll) sqlPool >>= mapM_ ($(logWarnS) "DB")
logFunc <- askLoggerIO
let
handlerInput = HandlerInput
{ sqlPool = sqlPool
}
io' :: ReaderT HandlerInput (LoggingT IO) :~> IO
io' = Nat (($ logFunc) . runLoggingT) . runReaderTNat handlerInput
liftIO . Warp.runSettings warpSettings . serve thermoprintAPI $ enter (hoistNat io') thermoprintServer'
type Handler = EitherT ServantErr (ReaderT HandlerInput (LoggingT IO))
(<||>) :: Monad m => m a -> m b -> m (a :<|> b)
(<||>) = liftM2 (:<|>)
infixr 9 <||>
thermoprintServer' :: ServerT ThermoprintAPI Handler
thermoprintServer' = listPrinters
:<|> (listJobs :<|> queueJob)
:<|> getJob <||> jobStatus <||> deleteJob
:<|> (listDrafts :<|> addDraft)
:<|> updateDraft <||> getDraft <||> deleteDraft <||> printDraft
listPrinters :: Handler (Map PrinterId PrinterStatus)
listPrinters = return $ Map.fromList [(1, Available), (7, Available), (3, Available)]
queueJob :: Maybe PrinterId -> Printout -> Handler API.JobId
queueJob = return undefined
printerStatus :: PrinterId -> Handler PrinterStatus
printerStatus = return undefined
listJobs :: Maybe PrinterId -> Maybe API.JobId -> Maybe API.JobId -> Handler (Seq (API.JobId, JobStatus))
listJobs = return undefined
getJob :: API.JobId -> Handler Printout
getJob = return undefined
jobStatus :: API.JobId -> Handler JobStatus
jobStatus = return undefined
deleteJob :: API.JobId -> Handler ()
deleteJob = return undefined
listDrafts :: Handler (Map API.DraftId (Maybe DraftTitle))
listDrafts = return undefined
addDraft :: Maybe DraftTitle -> Printout -> Handler API.DraftId
addDraft = return undefined
updateDraft :: API.DraftId -> Maybe DraftTitle -> Printout -> Handler ()
updateDraft = return undefined
getDraft :: API.DraftId -> Handler (Maybe DraftTitle, Printout)
getDraft = return undefined
deleteDraft :: API.DraftId -> Handler ()
deleteDraft = return undefined
printDraft :: API.DraftId -> Maybe PrinterId -> Handler API.JobId
printDraft = return undefined
|
