import Lseed.Data
import Lseed.Data.Functions
import Lseed.DB
import Lseed.Grammar.Parse
import Lseed.Mainloop
import Lseed.Renderer.Cairo
import Control.Applicative
import Control.Monad
import Text.Printf
import System.Environment
import Data.Monoid
import Data.Maybe
import System.Random
import System.Random.Shuffle (shuffle')

randomize l = shuffle' l (length l) <$> newStdGen

getDBGarden conf = do
	dbc <- getCodeToRun conf
	gs <- randomize $ mapMaybe compileDBCode dbc
	return $ spread gs
  where spread gs = zipWith (\(u,n,g) p ->
 		 Planted ((fromIntegral p + 0.5) / l)
			 u
			 n
			 g
			 inititalPlant
		) gs [0..]
	  where l = fromIntegral (length gs)

compileDBCode dbc =
	case  parseGrammar "" (dbcCode dbc) of
		Left err          -> Nothing
		Right grammarFile -> Just (dbcUserID dbc, dbcUserName dbc, grammarFile)

dbc2genome = either (const Nothing) Just . parseGrammar "" . dbcCode

getDBUpdate conf planted = fromMaybe (genome planted) <$>
		maybe Nothing dbc2genome <$>
		getUpdatedCodeFromDB conf (plantOwner planted)

scoringObs conf = nullObserver {
	obFinished = \garden -> do
		forM_ garden $ \planted -> do
			printf "Plant from %d at %.4f: Total size %.4f\n"
				(plantOwner planted)
				(plantPosition planted)
				(plantLength (phenotype planted))
		addFinishedSeasonResults conf garden
	}

nothingNull "" = Nothing
nothingNull s  = Just s

main = do
	args <- getArgs
	case args of
	  [conf, pngfile, textfile] -> do
		obs <- cairoObserver
		let obs' = obs `mappend` scoringObs conf `mappend` pngDailyObserver pngfile
		let gs = GardenSource (getDBGarden conf)
				      (getDBUpdate conf)
                                      (nothingNull <$> readFile textfile) 
		lseedMainLoop True obs' gs 40
		obShutdown obs'
	  _ -> do
		putStrLn "L-Seed DB client application."
		putStrLn "Please pass DB configuration file, a PNG file to write, and a text file with messages on the command line."