[haskell] Implemented ship

2 files changed, 22 insertions, 8 deletions

diff --git a/haskell/src/SpaceTraders/Database.hs b/haskell/src/SpaceTraders/Database.hs
index fdc32d3..ac3aa7b 100644
--- a/haskell/src/SpaceTraders/Database.hs
+++ b/haskell/src/SpaceTraders/Database.hs
@@ -14,23 +14,24 @@ migrations :: [S.Query]
 migrations = [
   [r|CREATE TABLE schema_version (
        version INTEGER NOT NULL
-     );
-  |],
+     );|],
   [r|CREATE TABLE tokens (
        id INTEGER PRIMARY KEY,
        data TEXT NOT NULL
-     );
-  |],
+     );|],
   [r|CREATE TABLE agents (
        id INTEGER PRIMARY KEY,
        data TEXT NOT NULL
-     );
-  |],
+     );|],
   [r|CREATE TABLE contracts (
        id INTEGER PRIMARY KEY,
        data TEXT NOT NULL
-     );
-  |]]
+     );|],
+  [r|CREATE TABLE ships (
+       id INTEGER PRIMARY KEY,
+       data TEXT NOT NULL
+     );|],
+  [r|CREATE UNIQUE INDEX ships_data_symbol ON ships (json_extract(data, '$.symbol'));|]]
 
 close :: S.Connection -> IO ()
 close conn = S.close conn
diff --git a/haskell/src/SpaceTraders/Database/Ships.hs b/haskell/src/SpaceTraders/Database/Ships.hs
new file mode 100644
index 0000000..81c422d
--- /dev/null
+++ b/haskell/src/SpaceTraders/Database/Ships.hs
@@ -0,0 +1,13 @@
+{-# LANGUAGE OverloadedStrings #-}
+
+module SpaceTraders.Database.Ships
+  ( addShip
+  ) where
+
+import Data.Aeson
+import qualified Database.SQLite.Simple as S
+
+import SpaceTraders.Model.Ship
+
+addShip :: S.Connection -> Ship -> IO ()
+addShip conn ship = S.execute conn "INSERT INTO ships(data) VALUES (json(?));" (S.Only (encode ship))