@@ -10,17 +10,12 @@ import Colog
10import Control.AppRequestable
10import Control.AppRequestable
11import Control.Concurrent
11import Control.Concurrent
12import Control.GithubCloner
12import Control.GithubCloner
13−import Control.HasGithubStatus
14import Control.Monad.Except
13import Control.Monad.Except
15import Control.Monad.IO.Unlift
14import Control.Monad.IO.Unlift
16import Control.Monad.Reader
15import Control.Monad.Reader
17import Control.Task.Scheduler
16import Control.Task.Scheduler
18−import Crypto.PubKey.RSA.Read
19import Data.Aeson
17import Data.Aeson
20import Data.ByteString
18import 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
24import qualified Data.Csv as Csv
19import qualified Data.Csv as Csv
25import Data.IORef
20import Data.IORef
26import Data.Map (Map)
21import Data.Map (Map)
@@ -30,17 +25,14 @@ import Data.Submission.Query
30import Data.Tasks.Build
25import Data.Tasks.Build
31import Data.Tasks.StatusUpdate
26import Data.Tasks.StatusUpdate
32import qualified Data.Text as T
27import qualified Data.Text as T
33−import Data.Time.Clock
34import Data.Yaml
28import Data.Yaml
35import GHC.Generics
29import GHC.Generics
36import GitHub hiding (Accept)
30import GitHub hiding (Accept)
37import GitHub.App.Auth
31import GitHub.App.Auth
38import GitHub.App.Request
32import GitHub.App.Request
39−import GitHub.Auth (Auth (OAuth))
40import GitHub.Data.Name
33import GitHub.Data.Name
41import GitHub.Data.Webhooks.Events
34import GitHub.Data.Webhooks.Events
42−import GitHub.Data.Webhooks.Payload hiding (URL (..))
35+import GitHub.Data.Webhooks.Payload
43−import Network.HTTP.Client.TLS (newTlsManager)
44import Network.HTTP.Media.MediaType
36import Network.HTTP.Media.MediaType
45import Network.Wai.Handler.Warp
37import Network.Wai.Handler.Warp
46import Servant
38import Servant
@@ -50,8 +42,9 @@ import Server.Config
50import Server.Html
42import Server.Html
51import Server.Schema
43import Server.Schema
52import Squeal.PostgreSQL
44import Squeal.PostgreSQL
53import Text.Blaze.Html
46import Text.Blaze.Html
54−import UnliftIO.Exception (catch, displayException, handleAny, throwIO, try)
47+import UnliftIO.Exception (try)
57 "github-webhook"
50 "github-webhook"
@@ -59,8 +52,8 @@ type API =
59 :> GitHubSignedReqBody '[JSON] InstallationEvent
52 :> GitHubSignedReqBody '[JSON] InstallationEvent
60 :> Post '[JSON] ()
53 :> Post '[JSON] ()
61 :<|> GitHubEvent '[ 'WebhookPushEvent]
54 :<|> GitHubEvent '[ 'WebhookPushEvent]
62− :> GitHubSignedReqBody '[JSON] PushEvent
55+ :> GitHubSignedReqBody '[JSON] PushEvent
63− :> Post '[JSON] ()
56+ :> Post '[JSON] ()
65 :<|> "submission"
58 :<|> "submission"
66 :> Capture "user name" String
59 :> Capture "user name" String
@@ -75,18 +68,17 @@ type API =
75 :> Post '[HTML] Markup
68 :> Post '[HTML] Markup
76 :<|> "results" :> Capture "task name" String :> Get '[CSV, HTML] [Score]
69 :<|> "results" :> Capture "task name" String :> Get '[CSV, HTML] [Score]
78−webhookInstallation :: WithLog env Message m => RepoWebhookEvent -> ((), InstallationEvent) -> m ()
71+webhookInstallation :: MonadIO m => RepoWebhookEvent -> ((), InstallationEvent) -> m ()
79webhookInstallation _ ((), ev) =
72webhookInstallation _ ((), ev) =
80− logInfo . T.pack . show $ ev
73+ liftIO $ do
77+pendingStatus :: NewStatus
78+pendingStatus = NewStatus StatusPending Nothing Nothing Nothing
82webhookPushEvent ::
80webhookPushEvent ::
84− MonadReader (ServerData m) m,
87− MonadError ServerError m,
88− WithLog env Message m
81+ (StaticPQ m, WithLog env Message m, MonadUnliftIO m, HasTasks m) =>
90 RepoWebhookEvent ->
82 RepoWebhookEvent ->
91 ((), PushEvent) ->
83 ((), PushEvent) ->
@@ -94,42 +86,30 @@ webhookPushEvent _ ((), ev) = do
94 let repo = evPushRepository ev
86 let repo = evPushRepository ev
95 user = T.unpack . whUserLogin . evPushSender $ ev
87 user = T.unpack . whUserLogin . evPushSender $ ev
96 fullRepoName = whRepoFullName repo
88 fullRepoName = whRepoFullName repo
97− (owner, repoName) <- case T.splitOn "/" fullRepoName of
89+ [owner, repoName] = T.splitOn "/" fullRepoName
98− [a, b] -> return (a, b)
99− _ -> throwError err422
100− sha <- maybe (throwError err422) return $ evPushHeadSha ev
90+ Just sha = evPushHeadSha ev
101 tasks <- getTasks
91 tasks <- getTasks
102 let task =
92 let task =
103 M.lookupMax $
93 M.lookupMax $
104 M.takeWhileAntitone (repoName >) tasks
94 M.takeWhileAntitone (repoName >) tasks
105 case task of
95 case task of
106 Just (name, TaskConfig {..}) | T.take (T.length name) repoName == name -> do
96 Just (name, TaskConfig {..}) | T.take (T.length name) repoName == name -> do
107− nowTime <- liftIO getCurrentTime
97+ scheduleTask $ StatusUpdate (N owner) (N repoName) (N sha) pendingStatus
108− if nowTime > deadline
110− deadlineHasPassedStatus (N owner) (N repoName) (N sha)
111− createSubmission $
98+ createSubmission $
112− Submission
99+ Submission
114− (T.unpack fullRepoName)
101+ (T.unpack fullRepoName)
115− (T.unpack name)
102+ (T.unpack name)
116− (T.unpack sha)
103+ (T.unpack sha)
117− (Jsonb PastDeadline)
119− schedulePendingStatus (N owner) (N repoName) (N sha)
123− (T.unpack fullRepoName)
126− (Jsonb BuildScheduled)
104+ (Jsonb BuildScheduled)
127− scheduleTask $ Build prebuild dockerfile (N owner) (N repoName) (N sha) timeoutMinutes
105+ scheduleTask $ Build prebuild dockerfile (N owner) (N repoName) (N sha)
128− logInfo . T.pack . show $ ev
106+ liftIO $ do
129 _ -> return ()
109 _ -> return ()
131retestSubmission ::
111retestSubmission ::
132− (StaticPQ m, MonadReader (ServerData m) m, MonadUnliftIO m, HasTasks m, MonadHasBaseUrl m) =>
112+ (StaticPQ m, WithLog env Message m, MonadUnliftIO m, HasTasks m, MonadHasBaseUrl m) =>
133 String ->
113 String ->
134 String ->
114 String ->
135 String ->
115 String ->
@@ -148,9 +128,9 @@ retestSubmission owner repoName sha = do
148 M.takeWhileAntitone (repoName' >) tasks
128 M.takeWhileAntitone (repoName' >) tasks
149 case task of
129 case task of
150 Just (name, TaskConfig {..}) | T.take (T.length name) repoName' == name -> do
130 Just (name, TaskConfig {..}) | T.take (T.length name) repoName' == name -> do
151− schedulePendingStatus (N owner') (N repoName') (N sha')
131+ scheduleTask $ StatusUpdate (N owner') (N repoName') (N sha') pendingStatus
152 updateSubmissionStatus fullRepoName sha BuildScheduled
132 updateSubmissionStatus fullRepoName sha BuildScheduled
153− scheduleTask $ Build prebuild dockerfile (N owner') (N repoName') (N sha') timeoutMinutes
133+ scheduleTask $ Build prebuild dockerfile (N owner') (N repoName') (N sha')
154 _ -> return ()
134 _ -> return ()
155 redirectToSubmission owner repoName sha
135 redirectToSubmission owner repoName sha
@@ -168,21 +148,13 @@ newtype ServerM (schema :: SchemasType) a
168 MonadUnliftIO
148 MonadUnliftIO
171−instance MonadError ServerError (ServerM schema) where
172− throwError = throwIO
175server :: ServerT API (ServerM Schema)
151server :: ServerT API (ServerM Schema)
176server = (webhookInstallation :<|> webhookPushEvent) :<|> getSubmissionR :<|> retestSubmission :<|> getResults
152server = (webhookInstallation :<|> webhookPushEvent) :<|> getSubmissionR :<|> retestSubmission :<|> getResults
178−runServer :: FilePath -> IO ()
154+runServer :: IO ()
179−runServer config = do
155+runServer = do
180− c@Config {..} <- getConfig config
156+ c@Config {..} <- getConfig
182− pem <- BS.readFile appPkPemPath
183− appPkPem <- case readRsaPem pem of
185− Left e -> error (show e)
186 auth <- mkInstallationAuth (AppAuth appId appPkPem) installationId
158 auth <- mkInstallationAuth (AppAuth appId appPkPem) installationId
187 ts <- newIORef mempty
159 ts <- newIORef mempty
188 pool <- createConnectionPool databseUrl 1 0.5 10
160 pool <- createConnectionPool databseUrl 1 0.5 10
@@ -192,38 +164,29 @@ runServer config = do
192 let context =
164 let context =
193 gitPolyHubKey (return webhookSecret)
165 gitPolyHubKey (return webhookSecret)
194 :. EmptyContext
166 :. EmptyContext
195− serverData =
167+ serverData = ServerData
197− { githubAppAuth = auth,
168+ { githubAppAuth = auth,
169+ githubUserName = GithubUserName githubUsername,
170+ githubAccessToken = GithubAccessToken personalAccessToken,
198− tasks = ts,
171+ tasks = ts,
199− logger = cfilter ((logSeverity <=) . msgSeverity) richMessageAction,
172+ logger = cfilter ((logSeverity <=) . msgSeverity) richMessageAction,
200− baseUrl = baseSiteUrl,
173+ baseUrl = baseSiteUrl,
201− dockerUrl = T.pack cfgDockerUrl,
174+ dockerUrl = T.pack cfgDockerUrl
202− context = githubContext
204 repeatIfNotEmpty n f = f >>= \m -> do
176 repeatIfNotEmpty n f = f >>= \m -> do
205 when (m == 0) $ liftIO $ threadDelay n
177 when (m == 0) $ liftIO $ threadDelay n
206 repeatIfNotEmpty n f
178 repeatIfNotEmpty n f
207− printingErrors =
179+ _ <-
208− handleAny (logError . T.pack . displayException >=> const (return 0))
209− usingConnectionPool pool
210− . (`runReaderT` serverData)
212− . handleAny (error . displayException)
214− replicateM_ builderCount
215− $ forkIO . repeatIfNotEmpty (10 ^ (6 :: Int))
180+ forkIO . repeatIfNotEmpty (10 ^ (6 :: Int))
216 . usingConnectionPool pool
181 . usingConnectionPool pool
217 . (`runReaderT` serverData)
182 . (`runReaderT` serverData)
220− $ runTasks @'[Build] 1
184+ $ runTasks @'[Build] 1
222 forkIO . repeatIfNotEmpty (10 ^ (6 :: Int))
186 forkIO . repeatIfNotEmpty (10 ^ (6 :: Int))
223 . usingConnectionPool pool
187 . usingConnectionPool pool
224 . (`runReaderT` serverData)
188 . (`runReaderT` serverData)
227 $ runTasks @'[StatusUpdate] 5
190 $ runTasks @'[StatusUpdate] 5
228 run port $ serveWithContext (Proxy @API) context $
191 run port $ serveWithContext (Proxy @API) context $
229 hoistServerWithContext
192 hoistServerWithContext
@@ -241,11 +204,12 @@ updateTasks ref path =
241data ServerData m
204data ServerData m
242 = ServerData
205 = ServerData
243 { githubAppAuth :: !InstallationAuth,
206 { githubAppAuth :: !InstallationAuth,
207+ githubUserName :: !GithubUserName,
208+ githubAccessToken :: !GithubAccessToken,
244 tasks :: IORef TasksConfig,
209 tasks :: IORef TasksConfig,
245 logger :: LogAction m Message,
210 logger :: LogAction m Message,
246− baseUrl :: T.Text,
211+ baseUrl :: String,
247− dockerUrl :: T.Text,
212+ dockerUrl :: T.Text
251newtype PolyGitHubKey = PolyGitHubKey (forall result. GitHubKey result)
215newtype PolyGitHubKey = PolyGitHubKey (forall result. GitHubKey result)
@@ -256,18 +220,11 @@ gitPolyHubKey k = PolyGitHubKey (Servant.GitHub.Webhook.gitHubKey k)
256instance HasContextEntry '[PolyGitHubKey] (GitHubKey result) where
220instance HasContextEntry '[PolyGitHubKey] (GitHubKey result) where
257 getContextEntry (PolyGitHubKey x :. _) = x
221 getContextEntry (PolyGitHubKey x :. _) = x
259−instance (MonadReader (ServerData n) m, MonadIO m, MonadError ServerError m) => GithubCloner m where
223+instance MonadReader (ServerData n) m => GithubCloner m where
260− getGithubInstallationToken =
224+
261− either
225+ getGithubAccessToken = asks githubAccessToken
262− (\e -> throwError err500 {errBody = BSCL.pack . displayException $ e})
226+
263− return
227+ getGithubUserName = asks githubUserName
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."
272instance (MonadIO m, MonadReader (ServerData n) m) => AppRequestable m where
229instance (MonadIO m, MonadReader (ServerData n) m) => AppRequestable m where
273 appRequest req = do
230 appRequest req = do
@@ -280,71 +237,6 @@ instance MonadReader (ServerData n) m => MonadHasBaseUrl m where
280instance (MonadIO m, MonadReader (ServerData n) m) => HasTasks m where
237instance (MonadIO m, MonadReader (ServerData n) m) => HasTasks m where
281 getTasks = asks tasks >>= liftIO . readIORef
238 getTasks = asks tasks >>= liftIO . readIORef
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
294− (Just . URL $ h <> "/submission/" <> owner <> "/" <> repo <> "/" <> sha)
298− scheduleTestedStatus TestResult {..} nOwner@(N owner) nRepo@(N repo) nCommit@(N sha) = do
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 " ❌"
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
317− scheduleFailedStatus _ nOwner@(N owner) nRepo@(N repo) nCommit@(N sha) = do
326− { newStatusState = StatusError,
327− newStatusTargetUrl = Just . URL $ h <> "/submission/" <> owner <> "/" <> repo <> "/" <> sha,
328− newStatusDescription = Just "Build failed ☠️",
329− newStatusContext = Just c
332− deadlineHasPassedStatus nOwner@(N owner) nRepo@(N repo) nCommit@(N sha) = do
341− { newStatusState = StatusError,
342− newStatusTargetUrl = Just . URL $ h <> "/submission/" <> owner <> "/" <> repo <> "/" <> sha,
343− newStatusDescription = Just "The deadline has passed ⏰",
344− newStatusContext = Just c
348instance (MonadReader (ServerData n) m) => HasDockerInfo m where
240instance (MonadReader (ServerData n) m) => HasDockerInfo m where
349 getDockerUrl = asks dockerUrl
241 getDockerUrl = asks dockerUrl
@@ -353,9 +245,8 @@ type TasksConfig = Map T.Text TaskConfig
353data TaskConfig
245data TaskConfig
354 = TaskConfig
246 = TaskConfig
355 { prebuild :: String,
247 { prebuild :: String,
356− dockerfile :: String,
248+ dockerfile :: String
357− timeoutMinutes :: Int,
249+ -- build :: T.Text
360 deriving stock (Show, Eq, Generic)
251 deriving stock (Show, Eq, Generic)
361 deriving anyclass (FromJSON, ToJSON)
252 deriving anyclass (FromJSON, ToJSON)
@@ -364,6 +255,7 @@ class HasTasks m where
364 getTasks :: m TasksConfig
255 getTasks :: m TasksConfig
366instance HasLog (ServerData m) Message m where
257instance HasLog (ServerData m) Message m where
367 getLogAction :: ServerData m -> LogAction m Message
259 getLogAction :: ServerData m -> LogAction m Message
368 getLogAction = logger
260 getLogAction = logger
369 {-# INLINE getLogAction #-}
261 {-# INLINE getLogAction #-}