gdax-0.5.0.0: app/Sandbox.hs

{-# LANGUAGE OverloadedLists   #-}
{-# LANGUAGE OverloadedStrings #-}

module Main where

import           Control.Concurrent
import           Control.Monad
import           Control.Monad.IO.Class
import           Data.Aeson                 (Value)
import qualified Data.Aeson                 as Aeson
import           Data.Aeson.Encode.Pretty
import qualified Data.ByteString.Base64     as Base64
import qualified Data.ByteString.Char8      as CBS
import qualified Data.ByteString.Lazy.Char8 as CLBS
import           Network.GDAX.Explicit
import           Network.GDAX.Types.Feed
import           Network.WebSockets
import           System.Environment
import           Wuss

main :: IO ()
main = putStrLn "For use with GHCi."

withGdax :: (MonadIO m) => (Gdax -> m a) -> m a
withGdax f = do
    gAccessKey <- liftIO $ CBS.pack <$> getEnv "GDAX_KEY"
    gSecretKey <- liftIO $ Base64.decodeLenient . CBS.pack <$> getEnv "GDAX_SECRET"
    gPassphrase <- liftIO $ CBS.pack <$> getEnv "GDAX_PASSPHRASE"

    g <- mkSandboxGdax gAccessKey gSecretKey gPassphrase

    f g

printPrettyLn :: (MonadIO m) => Value -> m ()
printPrettyLn = liftIO . CLBS.putStrLn . encodePretty

subscribeSocket :: IO ()
subscribeSocket = runSecureClient "ws-feed.gdax.com" 443 "/" client

testSub :: Subscribe
testSub = Subscribe $ Subscriptions [] [ChannelSubscription ChannelHeartbeat ["BTC-USD"]]

testUnSub :: UnSubscribe
testUnSub =  UnSubscribe $ Subscriptions [] [ChannelSubscription ChannelHeartbeat ["BTC-USD"]]

client :: ClientApp ()
client conn = do
    putStrLn "Connection opened.."

    sendTextData conn (Aeson.encode testSub)
    void . forkIO . forever $ do
        msg <- receiveData conn
        let hbeat = Aeson.decode msg :: Maybe Heartbeat
        case hbeat of
            Nothing ->
                let subs = Aeson.decode msg :: Maybe Subscriptions
                in case subs of
                    Nothing -> print ("not sub"::String) >> print msg
                    Just v  -> print v
            Just v  -> print v

    threadDelay 1000000

    putStrLn "Unsubscribing..."
    sendTextData conn (Aeson.encode testUnSub)

    threadDelay 1000000
    putStrLn "Closing..."