Merge pull request #13 from fffej/command-line

Command line

Author: fffej

README.md

@@ -4,8 +4,6 @@ The goal of this package is to generate arbitrary SQL Server databases (in the f
 
 Contributers more than welcome (especially if you know enough Haskell to help me simplify the code!).
 
-# Usage
-
 # Build instructions
 
 ## Linux
@@ -40,3 +38,10 @@ Assuming you're already building (if not see the instructions above) then
 Should spin up a web server on port 8888.  Fire up your web browser and try, for example, http://localhost/database/235235235?size=1 to generate a tiny example database.  Play with the size parameter with caution, the whole response is in memory first so big numbers (say > 1000) are probably a bad idea.
 
 Usage should be consistent, so going to the same URL should always generate the same value for testing purposes.
+
+If you'd rather use a command line interface then
+
+    cabal build cli
+    ./cli --help    
+
+Should produce the documentation.  Seed and size are exactly as the webserver.

cli/Main.hs

@@ -0,0 +1,31 @@
+{-# LANGUAGE DeriveDataTypeable #-}
+module Main where
+
+import System.Console.CmdArgs
+
+import Database.SqlServer.Types.Database
+
+data Arguments = Arguments
+    {
+      seed :: Int
+    , size :: Int
+    } deriving (Show,Data,Typeable)
+
+msg :: [String]
+msg =  ["More details on the github repo at " ++
+        " https://github.com/fffej/sql-server-gen"]
+
+defaultArgs :: Arguments
+defaultArgs = Arguments 
+    {
+      seed = def &= help "Seed for random number generator" &= name "s"
+    , size = 100 &= help "Size of database (optional)" &= opt (500 :: Int) &= name "n"
+    } &= summary "SQL Server Schema Generator"
+      &= help "Generate arbitrary SQL Server databases"
+      &= details msg
+
+main :: IO ()
+main = do
+  a <- cmdArgs defaultArgs
+  print $ seededDatabase (seed a) (size a) 
+  return ()

sql-server-gen.cabal

@@ -103,6 +103,15 @@ executable webserver
                         sql-server-gen,
                         scotty >= 0.10.1,
                         aeson >= 0.9.0.1,
-                        QuickCheck >= 2.8.1,
-                        pretty >= 1.1.2,
                         transformers >= 0.4.2.0
+
+executable cli
+  ghc-options:          -Wall
+  main-is:              Main.hs
+  hs-source-dirs:       cli
+  default-language:     Haskell2010
+  build-depends:        base >=4.8 && <4.9,
+                        sql-server-gen,
+                        transformers >= 0.4.2.0,
+                        cmdargs >= 0.10.13
+                        

src/Database/SqlServer/Types/Database.hs

@@ -8,9 +8,12 @@ import Database.SqlServer.Types.Procedure
 import Database.SqlServer.Types.Queue
 
 import Test.QuickCheck
+import Test.QuickCheck.Gen
+import Test.QuickCheck.Random
 import Control.Monad
 import qualified Data.Set as S
 
+
 import Text.PrettyPrint
 
 newtype TableDefinitions = TableDefinitions (S.Set TableDefinition)
@@ -111,4 +114,7 @@ dumpExamples m p = do
   writeFile p (unlines $ map (render . renderDatabaseDefinition) x)
 
 instance Show DatabaseDefinition where
-  show = render . renderDatabaseDefinition
+  show = render . renderDatabaseDefinition
+
+seededDatabase :: Int -> Int -> DatabaseDefinition
+seededDatabase seed size = unGen arbitrary (mkQCGen seed) size

web/Main.hs

@@ -6,11 +6,6 @@ module Main where
 import Web.Scotty
 import Data.Aeson hiding (json)
 import GHC.Generics
-import Test.QuickCheck
-import Test.QuickCheck.Gen
-import Test.QuickCheck.Random
-import Text.PrettyPrint (render)
-
 
 import Database.SqlServer.Types.Database;
 
@@ -28,10 +23,10 @@ generateDB seed' size' = DatabaseAsJson
     {
       seed = seed'
     , size = size'
-    , createStatement = (render . renderDatabaseDefinition) db
+    , createStatement = show db
     }
   where
-    db = unGen (arbitrary :: Gen DatabaseDefinition) (mkQCGen seed') size'
+    db = seededDatabase seed' size'
 
 main :: IO ()
 main = scotty 8888 $ do