koji-tool-1.3: src/Common.hs

module Common (
  knownHubs,
  hubURL,
  commonQueryOptions,
  commonBuildQueryOptions,
  webUrl,
  getBuildState,
  lookupArch,
  Limit(..),
  maybeLimit,
  Natural,
  defaultBuildsLimit,
  defaultTasksLimit
  )
where

import Control.Applicative ((<|>))
import Data.List.Extra (dropSuffix, isPrefixOf)
import Distribution.Koji (fedoraKojiHub, Value(..), Struct, BuildState,
                          lookupStruct, readBuildState)
import Numeric.Natural (Natural)
import SimpleCmd (error')

-- mbox kojihub is locked
knownHubs :: [String]
knownHubs = ["fedora","stream","rpmfusion", "or URL"]

hubURL :: String -> String
hubURL "fedora" = fedoraKojiHub
-- later use centosKojiHub
hubURL "stream" = "https://kojihub.stream.centos.org/kojihub"
--hubURL "mbox" = "https://koji.mbox.centos.org/kojihub"
hubURL "rpmfusion" = "https://koji.rpmfusion.org/kojihub"
hubURL "fusion" = "https://koji.rpmfusion.org/kojihub"
hubURL hub =
  if "http" `isPrefixOf` hub
  then hub
  else error' $ "unknown hub: try " ++ show knownHubs

commonQueryOptions :: Maybe Int -> String -> [(String, Value)]
commonQueryOptions mlimit order =
  [("limit",ValueInt limit) | Just limit <- [mlimit]] ++
  [("order",ValueString order)]

commonBuildQueryOptions :: Maybe Int -> (String, Value)
commonBuildQueryOptions mlimit =
  ("queryOpts", ValueStruct (commonQueryOptions mlimit "-build_id"))

webUrl :: String -> String
webUrl = dropSuffix "hub"

getBuildState :: Struct -> Maybe BuildState
getBuildState st = readBuildState <$> lookup "state" st

-- noarch build tasks usually have arch
-- see also https://pagure.io/koji/issue/4098
lookupArch :: Struct -> Maybe String
lookupArch st = lookupStruct "label" st <|> lookupStruct "arch" st

data Limit =
  Nolimit | Limit Natural

maybeLimit :: Natural -> Maybe Limit -> Maybe Int
maybeLimit l Nothing = Just (fromEnum l)
maybeLimit _ (Just Nolimit) = Nothing
maybeLimit _ (Just (Limit l)) = Just (fromEnum l)

defaultBuildsLimit, defaultTasksLimit :: Natural
defaultBuildsLimit = 5
defaultTasksLimit = 20