@@ -10,18 +10,12 @@ import Colog
1010import Control.AppRequestable
1111import Control.Concurrent
1212import Control.GithubCloner
13−import Control.HasGithubStatus
1413import Control.Monad.Except
1514import Control.Monad.IO.Unlift
1615import Control.Monad.Reader
1716import Control.Task.Scheduler
18−import Crypto.PubKey.RSA.Read
1917import Data.Aeson
2018import Data.ByteString
21−import qualified Data.ByteString as BS
22−import qualified Data.ByteString.Char8 as BSC
23−import qualified Data.ByteString.Lazy.Char8 as BSCL
24−import qualified Data.Csv as Csv
2519import Data.IORef
2620import Data.Map (Map)
2721import qualified Data.Map as M
@@ -30,18 +24,14 @@ import Data.Submission.Query
3024import Data.Tasks.Build
3125import Data.Tasks.StatusUpdate
3226import qualified Data.Text as T
33−import Data.Time.Clock
3427import Data.Yaml
3528import GHC.Generics
36−import GitHub hiding (Accept)
29+import GitHub
3730import GitHub.App.Auth
3831import GitHub.App.Request
39−import GitHub.Auth (Auth (OAuth))
4032import GitHub.Data.Name
4133import GitHub.Data.Webhooks.Events
42−import GitHub.Data.Webhooks.Payload hiding (URL (..))
43−import Network.HTTP.Client.TLS (newTlsManager)
44−import Network.HTTP.Media.MediaType
34+import GitHub.Data.Webhooks.Payload
4535import Network.Wai.Handler.Warp
4636import Servant
4737import Servant.GitHub.Webhook
@@ -50,8 +40,9 @@ import Server.Config
5040import Server.Html
5141import Server.Schema
5242import Squeal.PostgreSQL
43+import System.IO
5344import Text.Blaze.Html
54−import UnliftIO.Exception (catch, displayException, handleAny, throwIO, try)
45+import UnliftIO.Exception (try)
5546
5647type API =
5748 "github-webhook"
@@ -59,8 +50,8 @@ type API =
5950 :> GitHubSignedReqBody '[JSON] InstallationEvent
6051 :> Post '[JSON] ()
6152 :<|> GitHubEvent '[ 'WebhookPushEvent]
62− :> GitHubSignedReqBody '[JSON] PushEvent
63− :> Post '[JSON] ()
53+ :> GitHubSignedReqBody '[JSON] PushEvent
54+ :> Post '[JSON] ()
6455 )
6556 :<|> "submission"
6657 :> Capture "user name" String
@@ -73,20 +64,18 @@ type API =
7364 :> Capture "sha" String
7465 :> "restart"
7566 :> Post '[HTML] Markup
76− :<|> "results" :> Capture "task name" String :> Get '[CSV, HTML] [Score]
7767
78−webhookInstallation :: WithLog env Message m => RepoWebhookEvent -> ((), InstallationEvent) -> m ()
68+webhookInstallation :: MonadIO m => RepoWebhookEvent -> ((), InstallationEvent) -> m ()
7969webhookInstallation _ ((), ev) =
80− logInfo . T.pack . show $ ev
70+ liftIO $ do
71+ print ev
72+ hFlush stdout
73+
74+pendingStatus :: NewStatus
75+pendingStatus = NewStatus StatusPending Nothing Nothing Nothing
8176
8277webhookPushEvent ::
83− ( StaticPQ m,
84− MonadReader (ServerData m) m,
85− MonadUnliftIO m,
86− HasTasks m,
87− MonadError ServerError m,
88− WithLog env Message m
89− ) =>
78+ (StaticPQ m, WithLog env Message m, MonadUnliftIO m, HasTasks m) =>
9079 RepoWebhookEvent ->
9180 ((), PushEvent) ->
9281 m ()
@@ -94,42 +83,30 @@ webhookPushEvent _ ((), ev) = do
9483 let repo = evPushRepository ev
9584 user = T.unpack . whUserLogin . evPushSender $ ev
9685 fullRepoName = whRepoFullName repo
97− (owner, repoName) <- case T.splitOn "/" fullRepoName of
98− [a, b] -> return (a, b)
99− _ -> throwError err422
100− sha <- maybe (throwError err422) return $ evPushHeadSha ev
86+ [owner, repoName] = T.splitOn "/" fullRepoName
87+ Just sha = evPushHeadSha ev
10188 tasks <- getTasks
10289 let task =
10390 M.lookupMax $
10491 M.takeWhileAntitone (repoName >) tasks
10592 case task of
10693 Just (name, TaskConfig {..}) | T.take (T.length name) repoName == name -> do
107− nowTime <- liftIO getCurrentTime
108− if nowTime > deadline
109− then do
110− deadlineHasPassedStatus (N owner) (N repoName) (N sha)
111− createSubmission $
112− Submission
113− user
114− (T.unpack fullRepoName)
115− (T.unpack name)
116− (T.unpack sha)
117− (Jsonb PastDeadline)
118− else do
119− schedulePendingStatus (N owner) (N repoName) (N sha)
120− createSubmission $
121− Submission
122− user
123− (T.unpack fullRepoName)
124− (T.unpack name)
125− (T.unpack sha)
126− (Jsonb BuildScheduled)
127− scheduleTask $ Build prebuild dockerfile (N owner) (N repoName) (N sha) timeoutMinutes
128− logInfo . T.pack . show $ ev
94+ scheduleTask $ StatusUpdate (N owner) (N repoName) (N sha) pendingStatus
95+ createSubmission $
96+ Submission
97+ user
98+ (T.unpack fullRepoName)
99+ (T.unpack name)
100+ (T.unpack sha)
101+ (Jsonb BuildScheduled)
102+ scheduleTask $ Build prebuild build (N owner) (N repoName) (N sha)
103+ liftIO $ do
104+ print ev
105+ hFlush stdout
129106 _ -> return ()
130107
131108retestSubmission ::
132− (StaticPQ m, MonadReader (ServerData m) m, MonadUnliftIO m, HasTasks m, MonadHasBaseUrl m) =>
109+ (StaticPQ m, WithLog env Message m, MonadUnliftIO m, HasTasks m, MonadHasBaseUrl m) =>
133110 String ->
134111 String ->
135112 String ->
@@ -148,9 +125,9 @@ retestSubmission owner repoName sha = do
148125 M.takeWhileAntitone (repoName' >) tasks
149126 case task of
150127 Just (name, TaskConfig {..}) | T.take (T.length name) repoName' == name -> do
151− schedulePendingStatus (N owner') (N repoName') (N sha')
128+ scheduleTask $ StatusUpdate (N owner') (N repoName') (N sha') pendingStatus
152129 updateSubmissionStatus fullRepoName sha BuildScheduled
153− scheduleTask $ Build prebuild dockerfile (N owner') (N repoName') (N sha') timeoutMinutes
130+ scheduleTask $ Build prebuild build (N owner') (N repoName') (N sha')
154131 _ -> return ()
155132 redirectToSubmission owner repoName sha
156133
@@ -168,62 +145,44 @@ newtype ServerM (schema :: SchemasType) a
168145 MonadUnliftIO
169146 )
170147
171−instance MonadError ServerError (ServerM schema) where
172− throwError = throwIO
173− catchError = catch
174−
175148server :: ServerT API (ServerM Schema)
176−server = (webhookInstallation :<|> webhookPushEvent) :<|> getSubmissionR :<|> retestSubmission :<|> getResults
149+server = (webhookInstallation :<|> webhookPushEvent) :<|> getSubmissionR :<|> retestSubmission
177150
178−runServer :: FilePath -> IO ()
179−runServer config = do
180− c@Config {..} <- getConfig config
151+runServer :: IO ()
152+runServer = do
153+ c@Config {..} <- getConfig
181154 print c
182− pem <- BS.readFile appPkPemPath
183− appPkPem <- case readRsaPem pem of
184− Right k -> return k
185− Left e -> error (show e)
186155 auth <- mkInstallationAuth (AppAuth appId appPkPem) installationId
187156 ts <- newIORef mempty
188157 pool <- createConnectionPool databseUrl 1 0.5 10
189158 _ <- forkIO $ do
190159 updateTasks ts tasksPath
191− threadDelay $ 5 * 60 * (10 ^ (6 :: Int))
160+ threadDelay $ 10 * 60 * (10 ^ (6 :: Int))
192161 let context =
193162 gitPolyHubKey (return webhookSecret)
194163 :. EmptyContext
195− serverData =
196− ServerData
197− { githubAppAuth = auth,
198− tasks = ts,
199− logger = cfilter ((logSeverity <=) . msgSeverity) richMessageAction,
200− baseUrl = baseSiteUrl,
201− dockerUrl = T.pack cfgDockerUrl,
202− context = githubContext
203− }
164+ serverData = ServerData
165+ { githubAppAuth = auth,
166+ githubUserName = GithubUserName githubUsername,
167+ githubAccessToken = GithubAccessToken personalAccessToken,
168+ tasks = ts,
169+ logger = simpleMessageAction,
170+ baseUrl = baseSiteUrl
171+ }
204172 repeatIfNotEmpty n f = f >>= \m -> do
205173 when (m == 0) $ liftIO $ threadDelay n
206174 repeatIfNotEmpty n f
207− printingErrors =
208− handleAny (logError . T.pack . displayException >=> const (return 0))
209− usingConnectionPool pool
210− . (`runReaderT` serverData)
211− . unApp
212− . handleAny (error . displayException)
213− $ makeAllRunnable
214− replicateM_ builderCount
215− $ forkIO . repeatIfNotEmpty (10 ^ (6 :: Int))
175+ _ <-
176+ forkIO . repeatIfNotEmpty (10 ^ (6 :: Int))
216177 . usingConnectionPool pool
217178 . (`runReaderT` serverData)
218179 . unApp
219− . printingErrors
220− $ runTasks @'[Build] 1
180+ $ runTasks @'[Build] 1
221181 _ <-
222182 forkIO . repeatIfNotEmpty (10 ^ (6 :: Int))
223183 . usingConnectionPool pool
224184 . (`runReaderT` serverData)
225185 . unApp
226− . printingErrors
227186 $ runTasks @'[StatusUpdate] 5
228187 run port $ serveWithContext (Proxy @API) context $
229188 hoistServerWithContext
@@ -241,11 +200,11 @@ updateTasks ref path =
241200data ServerData m
242201 = ServerData
243202 { githubAppAuth :: !InstallationAuth,
203+ githubUserName :: !GithubUserName,
204+ githubAccessToken :: !GithubAccessToken,
244205 tasks :: IORef TasksConfig,
245206 logger :: LogAction m Message,
246− baseUrl :: T.Text,
247− dockerUrl :: T.Text,
248− context :: T.Text
207+ baseUrl :: String
249208 }
250209
251210newtype PolyGitHubKey = PolyGitHubKey (forall result. GitHubKey result)
@@ -256,18 +215,11 @@ gitPolyHubKey k = PolyGitHubKey (Servant.GitHub.Webhook.gitHubKey k)
256215instance HasContextEntry '[PolyGitHubKey] (GitHubKey result) where
257216 getContextEntry (PolyGitHubKey x :. _) = x
258217
259−instance (MonadReader (ServerData n) m, MonadIO m, MonadError ServerError m) => GithubCloner m where
260− getGithubInstallationToken =
261− either
262− (\e -> throwError err500 {errBody = BSCL.pack . displayException $ e})
263− return
264− <=< runExceptT
265− $ do
266− iAuth <- asks githubAppAuth
267− manager <- newTlsManager
268− liftIO (obtainAccessToken manager iAuth) >>= either throwError return >>= \case
269− OAuth token -> return $ InstallationToken (BSC.unpack token)
270− _ -> throwError $ UserError "Not OAuth token. This should never happen."
218+instance MonadReader (ServerData n) m => GithubCloner m where
219+
220+ getGithubAccessToken = asks githubAccessToken
221+
222+ getGithubUserName = asks githubUserName
271223
272224instance (MonadIO m, MonadReader (ServerData n) m) => AppRequestable m where
273225 appRequest req = do
@@ -280,82 +232,12 @@ instance MonadReader (ServerData n) m => MonadHasBaseUrl m where
280232instance (MonadIO m, MonadReader (ServerData n) m) => HasTasks m where
281233 getTasks = asks tasks >>= liftIO . readIORef
282234
283−instance (MonadReader (ServerData m) m, StaticPQ m, MonadUnliftIO m) => HasGithubStatus m where
284− schedulePendingStatus nOwner@(N owner) nRepo@(N repo) nCommit@(N sha) = do
285− h <- asks baseUrl
286− c <- asks context
287− scheduleTask $
288− StatusUpdate
289− nOwner
290− nRepo
291− nCommit
292− ( NewStatus
293− StatusPending
294− (Just . URL $ h <> "/submission/" <> owner <> "/" <> repo <> "/" <> sha)
295− (Just "⌛")
296− (Just c)
297− )
298− scheduleTestedStatus TestResult {..} nOwner@(N owner) nRepo@(N repo) nCommit@(N sha) = do
299− h <- asks baseUrl
300− c <- asks context
301− let total = M.size tests
302− passed = M.size . M.filter id $ tests
303− description = T.pack $ show passed <> "/" <> show total <> if passed == total then " ✅" else " ❌"
304− status =
305− StatusUpdate
306− nOwner
307− nRepo
308− nCommit
309− NewStatus
310− { newStatusState =
311− if total == passed then StatusSuccess else StatusFailure,
312− newStatusTargetUrl = Just . URL $ h <> "/submission/" <> owner <> "/" <> repo <> "/" <> sha,
313− newStatusDescription = Just description,
314− newStatusContext = Just c
315− }
316− scheduleTask status
317− scheduleFailedStatus _ nOwner@(N owner) nRepo@(N repo) nCommit@(N sha) = do
318− h <- asks baseUrl
319− c <- asks context
320− let status =
321− StatusUpdate
322− nOwner
323− nRepo
324− nCommit
325− NewStatus
326− { newStatusState = StatusError,
327− newStatusTargetUrl = Just . URL $ h <> "/submission/" <> owner <> "/" <> repo <> "/" <> sha,
328− newStatusDescription = Just "Build failed ☠️",
329− newStatusContext = Just c
330− }
331− scheduleTask status
332− deadlineHasPassedStatus nOwner@(N owner) nRepo@(N repo) nCommit@(N sha) = do
333− h <- asks baseUrl
334− c <- asks context
335− let status =
336− StatusUpdate
337− nOwner
338− nRepo
339− nCommit
340− NewStatus
341− { newStatusState = StatusError,
342− newStatusTargetUrl = Just . URL $ h <> "/submission/" <> owner <> "/" <> repo <> "/" <> sha,
343− newStatusDescription = Just "The deadline has passed ⏰",
344− newStatusContext = Just c
345− }
346− scheduleTask status
347−
348−instance (MonadReader (ServerData n) m) => HasDockerInfo m where
349− getDockerUrl = asks dockerUrl
350−
351235type TasksConfig = Map T.Text TaskConfig
352236
353237data TaskConfig
354238 = TaskConfig
355239 { prebuild :: String,
356− dockerfile :: String,
357− timeoutMinutes :: Int,
358− deadline :: UTCTime
240+ build :: String
359241 }
360242 deriving stock (Show, Eq, Generic)
361243 deriving anyclass (FromJSON, ToJSON)
@@ -364,6 +246,7 @@ class HasTasks m where
364246 getTasks :: m TasksConfig
365247
366248instance HasLog (ServerData m) Message m where
249+
367250 getLogAction :: ServerData m -> LogAction m Message
368251 getLogAction = logger
369252 {-# INLINE getLogAction #-}
@@ -371,11 +254,3 @@ instance HasLog (ServerData m) Message m where
371254 setLogAction :: LogAction m Message -> ServerData m -> ServerData m
372255 setLogAction newLogAction env = env {logger = newLogAction}
373256 {-# INLINE setLogAction #-}
374−
375−data CSV
376−
377−instance Accept CSV where
378− contentType Proxy = "text" // "csv"
379−
380−instance Csv.ToRecord a => MimeRender CSV [a] where
381− mimeRender Proxy = Csv.encode