Generate a cabal file for the initialised site

Fixes #267

1 files changed, 52 insertions, 8 deletions

diff --git a/src/Hakyll/Init.hs b/src/Hakyll/Init.hs
index d50c6f6..51ba14b 100644
--- a/src/Hakyll/Init.hs
+++ b/src/Hakyll/Init.hs
@@ -5,11 +5,15 @@ module Main
 
 
 --------------------------------------------------------------------------------
+import           Control.Arrow         (first)
 import           Control.Monad         (forM_)
-import           System.Directory      (copyFile)
+import           Data.Char             (isAlphaNum, isNumber)
+import           Data.List             (intercalate)
+import           Data.Version          (Version(..))
+import           System.Directory      (copyFile, canonicalizePath)
 import           System.Environment    (getArgs, getProgName)
 import           System.Exit           (exitFailure)
-import           System.FilePath       ((</>))
+import           System.FilePath       ((</>), splitDirectories)
 
 
 --------------------------------------------------------------------------------
@@ -26,12 +30,52 @@ main = do
     files    <- getRecursiveContents (const $ return False) srcDir
 
     case args of
-        [dstDir] -> forM_ files $ \file -> do
-            let dst = dstDir </> file
-                src = srcDir </> file
-            putStrLn $ "Creating " ++ dst
-            makeDirectories dst
-            copyFile src dst
+        [dstDir] -> do
+            forM_ files $ \file -> do
+                let dst = dstDir </> file
+                    src = srcDir </> file
+                putStrLn $ "Creating " ++ dst
+                makeDirectories dst
+                copyFile src dst
+            -- canonicalizePath is safe because the destination
+            -- directory should exist at this point
+            canonicalizePath dstDir >>= createCabal
         _ -> do
             putStrLn $ "Usage: " ++ progName ++ " <directory>"
             exitFailure
+
+
+createCabal :: FilePath -> IO ()
+createCabal dstDir = do
+        putStrLn $ "Creating " ++ name ++ ".cabal"
+        writeFile (dstDir </> name ++ ".cabal") $ unlines [
+            "name:               " ++ name
+          , "version:            0.1.0.0"
+          , "build-type:         Simple"
+          , "cabal-version:      >= 1.10"
+          , ""
+          , "executable site"
+          , "  main-is:          site.hs"
+          , "  build-depends:    base == 4.*"
+          , "                  , hakyll == " ++ version' ++ ".*"
+          , "  ghc-options:      -threaded"
+          , "  default-language: Haskell2010"
+          ]
+  where
+    -- Major hakyll version
+    version' = intercalate "." . take 2 . map show $ versionBranch version
+    -- last is safe here as the path is canonicalised and "/" is just
+    -- a very rare but possible corner case
+    name = case last (splitDirectories  dstDir) of
+        "/" -> fallbackName
+        x   -> repair (fallbackName ++) id x
+    -- Package name repair code comes from
+    -- cabal-install.Distribution.Client.Init.Heuristics
+    repair invalid valid x = case dropWhile (not . isAlphaNum) x of
+        "" -> repairComponent ""
+        x' -> let (c, r) = first repairComponent $ break (not . isAlphaNum) x'
+              in c ++ repairRest r
+      where repairComponent c | all isNumber c = invalid c
+                              | otherwise = valid c
+    repairRest = repair id ('-' :)
+    fallbackName = "site"