-type Domain = String
-type Username = String
-type Count = Int
-
--- | A wrapper around a (domain, count) pair to keep things type-safe.
-data DomainCount = DomainCount Domain Count
-
-instance Show DomainCount where
- -- | Display a DomainCount in the form \"domain: count\".
- --
- -- Examples:
- --
- -- >>> let dc = DomainCount "example.com" 100
- -- >>> show dc
- -- "example.com: 100"
- --
- show (DomainCount domain count) = domain ++ ": " ++ (show count)
-
-
--- | A wrapper around a (domain, user) pair to keep things type-safe.
-data DomainUser =
- DomainUser Domain Username
- deriving (Show)
-
-
--- | In the detailed report, we store the usernames as a map from a
--- domain name to a list of usernames. This type synonym is the type
--- of that map.
-type DomainUserMap = Map.Map Domain [Username]
-
-
--- | Convert a list of [SqlByteString, SqlInt32] to 'DomainCount's. If
--- the conversion doesn't work for some reason (bad data, not enough
--- columns, etc.), we return 'Nothing'.
---
--- Examples:
---
--- >>> import Database.HDBC ( iToSql, toSql )
---
--- >>> list_to_domain_count [toSql "example.com", iToSql 100]
--- Just example.com: 100
---
--- >>> list_to_domain_count [toSql "example.com"]
--- Nothing
---
--- >>> list_to_domain_count [toSql "example.com", toSql "example.com"]
--- Nothing
---
-list_to_domain_count :: [SqlValue] -> Maybe DomainCount
-list_to_domain_count (domain:count:_)
- | Right d <- safeFromSql domain,
- Right c <- safeFromSql count = Just $ DomainCount d c
-list_to_domain_count _ =
- Nothing
-
-
--- | Convert a list of [SqlByteString, SqlByteString] to 'DomainUser's. If
--- the conversion doesn't work for some reason (bad data, not enough
--- columns, etc.), we return 'Nothing'.
---
--- Examples:
---
--- >>> import Database.HDBC ( toSql )
--- >>>
--- >>> list_to_domain_user [toSql "example.com", toSql "user1"]
--- Just (DomainUser "example.com" "user1")
---
--- >>> list_to_domain_user [toSql "example.com"]
--- Nothing
---
-list_to_domain_user :: [SqlValue] -> Maybe DomainUser
-list_to_domain_user (domain:user:_)
- | Right d <- safeFromSql domain,
- Right u <- safeFromSql user = Just $ DomainUser d u
-list_to_domain_user _ =
- Nothing
-
-
-report_summary :: IConnection a => a -> IO String
-report_summary conn = do
- list_rows <- quickQuery conn query []
- let maybe_domain_counts = map list_to_domain_count list_rows
- let domain_counts = catMaybes maybe_domain_counts
- return $ header ++ (concatMap show domain_counts)
- where
- header = "mailbox-count summary report" ++
- "----------------------------"
-
- query = "SELECT domain,COUNT(username) " ++
- "FROM mailbox " ++
- "GROUP BY domain "++
- "ORDER BY domain;"