{-| The basic data structure that ties together syntax trees making them
    composable and addressable in B9 artifacts. -}
module B9.Content.Generator where

import           Control.Parallel.Strategies
import           Data.Binary
import           Data.Data
import           Data.Hashable
import           Control.Monad.Trans (lift)
import           GHC.Generics (Generic)
#if !MIN_VERSION_base(4,8,0)
import           Control.Applicative
#endif

import           B9.B9Monad
import           Control.Monad.IO.Class
import           System.Exit
import           B9.Content.AST
import           B9.Content.ErlangPropList
import           B9.Content.StringTemplate
import           B9.Content.YamlObject
import qualified Data.ByteString.Char8 as B
import qualified Data.ByteString as BStrict

import           Test.QuickCheck
import           B9.QCUtil

import           System.Process

data Content
    = RenderErlang (AST Content ErlangPropList)
    | RenderYaml (AST Content YamlObject)
    | FromString String
    | FromTextFile SourceFile
    | FromURL String
    | FromStrictBinary BStrict.ByteString
    deriving (Read,Show,Typeable,Eq,Data,Generic)

-- | Pack bytes into a 'Content' such that a binary file will generated.
packBinaryContent :: [Word8] -> Content
packBinaryContent = FromStrictBinary . BStrict.pack

instance Hashable Content
instance Binary Content
instance NFData Content

instance Arbitrary Content where
  arbitrary = oneof [FromTextFile <$> smaller arbitrary
                    ,RenderErlang <$> smaller arbitrary
                    ,RenderYaml <$> smaller arbitrary
                    ,FromString <$> smaller arbitrary
                    ,FromURL <$> smaller arbitrary]

instance CanRender Content where
  render (RenderErlang ast) = encodeSyntax <$> fromAST ast
  render (RenderYaml ast) = encodeSyntax <$> fromAST ast
  render (FromTextFile s) = readTemplateFile s
  render (FromString str) = return (B.pack str)
  render (FromStrictBinary b) = return b
  render (FromURL url) = lift $ do
     dbgL $ "Downloading: " ++ url
     (exitcode,out,err) <- liftIO $ readProcessWithExitCode "curl" [url] ""
     if exitcode == ExitSuccess then
       do dbgL $ "Download finished. Bytes read: " ++ show (length out)
          traceL $ "Downloaded (truncated to first 4K): \n\n" ++ take 4096 out ++ "\n\n"
          return $ B.pack out
      else
        do errorL $ "Download failed: " ++ err
           liftIO $ exitWith exitcode