From 7cb7870d75266ab2c35047cbd2d022457ba6c714 Mon Sep 17 00:00:00 2001
From: Michael Snoyman <michael@fpcomplete.com>
Date: Mon, 19 Oct 2015 11:33:26 +0000
Subject: [PATCH] Include package count on snapshot page

---
 Handler/StackageHome.hs        | 1 +
 Stackage/Database.hs           | 6 ++++++
 templates/stackage-home.hamlet | 2 +-
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/Handler/StackageHome.hs b/Handler/StackageHome.hs
index b69ac41..c8626e6 100644
--- a/Handler/StackageHome.hs
+++ b/Handler/StackageHome.hs
@@ -21,6 +21,7 @@ getStackageHomeR name = do
             let queryText = "" :: Text
                 exact = False
             in $(widgetFile "hoogle-form")
+    packageCount <- getPackageCount sid
     defaultLayout $ do
         setTitle $ toHtml $ snapshotTitle snapshot
         packages <- getPackages sid
diff --git a/Stackage/Database.hs b/Stackage/Database.hs
index 094a990..e94a6c3 100644
--- a/Stackage/Database.hs
+++ b/Stackage/Database.hs
@@ -37,6 +37,7 @@ module Stackage.Database
     , currentSchema
     , last5Lts5Nightly
     , snapshotsJSON
+    , getPackageCount
     ) where
 
 import Database.Sqlite (SqliteException)
@@ -735,3 +736,8 @@ snapshotsJSON = do
         "lts-" ++ show major ++ "." ++ show minor
 
     printNightly day = "nightly-" ++ tshow day
+
+getPackageCount :: GetStackageDatabase m
+                => SnapshotId
+                -> m Int
+getPackageCount sid = run $ count [SnapshotPackageSnapshot ==. sid]
diff --git a/templates/stackage-home.hamlet b/templates/stackage-home.hamlet
index 634b499..84cb05d 100644
--- a/templates/stackage-home.hamlet
+++ b/templates/stackage-home.hamlet
@@ -38,7 +38,7 @@ $newline never
     <a href=@{SnapshotR name HoogleDatabaseR}>
       Download this hoogle database
 
-    <h3>Packages
+    <h3>Packages (#{packageCount})
 
     <p>
         <a href=@{SnapshotR name DocsR}>View documentation by modules