Upload.hs 2.68 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
{-# LANGUAGE PatternGuards #-}
-- This is a quick hack for uploading build reports to Hackage.

module Distribution.Client.BuildReports.Upload
    ( BuildLog
    , BuildReportId
    , uploadReports
    , postBuildReport
    , putBuildLog
    ) where

import Network.Browser
13
         ( BrowserAction, request, setAllowRedirects )
14
15
16
import Network.HTTP
         ( Header(..), HeaderName(..)
         , Request(..), RequestMethod(..), Response(..) )
Duncan Coutts's avatar
Duncan Coutts committed
17
import Network.TCP (HandleStream)
18
import Network.URI (URI, uriPath, parseRelativeReference, relativeTo)
19
20

import Control.Monad
21
         ( forM_ )
22
import System.FilePath.Posix
23
         ( (</>) )
24
25
import qualified Distribution.Client.BuildReports.Anonymous as BuildReport
import Distribution.Client.BuildReports.Anonymous (BuildReport)
26
import Distribution.Text (display)
27
28
29
30

type BuildReportId = URI
type BuildLog = String

Duncan Coutts's avatar
Duncan Coutts committed
31
uploadReports :: URI -> [(BuildReport, Maybe BuildLog)]
32
              -> BrowserAction (HandleStream String) ()
Duncan Coutts's avatar
Duncan Coutts committed
33
              ->  BrowserAction (HandleStream BuildLog) ()
34
35
36
37
38
39
40
uploadReports uri reports auth = do
  auth
  forM_ reports $ \(report, mbBuildLog) -> do
     buildId <- postBuildReport uri report
     case mbBuildLog of
       Just buildLog -> putBuildLog buildId buildLog
       Nothing       -> return ()
41

Duncan Coutts's avatar
Duncan Coutts committed
42
43
postBuildReport :: URI -> BuildReport
                -> BrowserAction (HandleStream BuildLog) BuildReportId
44
45
46
postBuildReport uri buildReport = do
  setAllowRedirects False
  (_, response) <- request Request {
47
    rqURI     = uri { uriPath = "/package" </> display (BuildReport.package buildReport) </> "reports" },
48
49
50
51
52
53
54
55
56
57
58
59
60
61
    rqMethod  = POST,
    rqHeaders = [Header HdrContentType   ("text/plain"),
                 Header HdrContentLength (show (length body)),
                 Header HdrAccept        ("text/plain")],
    rqBody    = body
  }
  case rspCode response of
    (3,0,3) | [Just buildId] <- [ do rel <- parseRelativeReference location
                                     relativeTo rel uri
                                  | Header HdrLocation location <- rspHeaders response ]
              -> return $ buildId
    _         -> error "Unrecognised response from server."
  where body  = BuildReport.show buildReport

Duncan Coutts's avatar
Duncan Coutts committed
62
63
putBuildLog :: BuildReportId -> BuildLog
            -> BrowserAction (HandleStream BuildLog) ()
64
putBuildLog reportId buildLog = do
Duncan Coutts's avatar
Duncan Coutts committed
65
  --FIXME: do something if the request fails
66
  (_, response) <- request Request {
67
      rqURI     = reportId{uriPath = uriPath reportId </> "log"},
68
69
70
71
72
73
74
      rqMethod  = PUT,
      rqHeaders = [Header HdrContentType   ("text/plain"),
                   Header HdrContentLength (show (length buildLog)),
                   Header HdrAccept        ("text/plain")],
      rqBody    = buildLog
    }
  return ()