diff options
Diffstat (limited to 'haskell/src/SpaceTraders/APIClient')
| -rw-r--r-- | haskell/src/SpaceTraders/APIClient/Ships.hs | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/haskell/src/SpaceTraders/APIClient/Ships.hs b/haskell/src/SpaceTraders/APIClient/Ships.hs index d396911..912f6e8 100644 --- a/haskell/src/SpaceTraders/APIClient/Ships.hs +++ b/haskell/src/SpaceTraders/APIClient/Ships.hs @@ -6,6 +6,7 @@ module SpaceTraders.APIClient.Ships ( dock , myShips , orbit + , refuel ) where import Data.Aeson.Types @@ -16,7 +17,10 @@ import Network.HTTP.Simple import SpaceTraders import SpaceTraders.APIClient.Client import SpaceTraders.APIClient.Pagination +import SpaceTraders.Database.Agents import SpaceTraders.Database.Ships +import qualified SpaceTraders.Model.Agent +import SpaceTraders.Model.Fuel import SpaceTraders.Model.Nav import SpaceTraders.Model.Ship @@ -62,3 +66,27 @@ orbit ship = if isInOrbit ship then pure (Right ship) else orbit' let s = ship{SpaceTraders.Model.Ship.nav=n} setShip s return $ Right s + +data RefuelMessage = RefuelMessage { agent :: SpaceTraders.Model.Agent.Agent + , fuel :: Fuel + } deriving (FromJSON, Generic, Show) + +refuel :: Ship -> SpaceTradersT (APIResponse Ship) +refuel ship = if overNinetyPercentFuel ship then pure (Right ship) else refuel' + where + refuel' = do + resp <- dock ship + case resp of + (Left e) -> return $ Left e + (Right ship') -> refuel'' ship' + refuel'' ship' = do + -- TODO check if we are at a marketplace, that sells fuel + resp' <- send $ setRequestPath (T.encodeUtf8 $ mconcat ["/v2/my/ships/", symbol ship, "/refuel"]) + . setRequestMethod "POST" :: SpaceTradersT (APIResponse RefuelMessage) + case resp' of + Left e -> return $ Left e + Right (RefuelMessage a f) -> do + setAgent a + let s = ship'{SpaceTraders.Model.Ship.fuel=f} + setShip s + return $ Right s |