summaryrefslogtreecommitdiff
path: root/haskell/src/SpaceTraders/Database/Systems.hs
blob: 5312fd05d002f2af5f99da445acbb36a135d63c9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
{-# LANGUAGE OverloadedStrings #-}

module SpaceTraders.Database.Systems
  ( addSystems
  , getSystems
  ) where

import Data.Aeson
import Data.Maybe
import qualified Database.SQLite.Simple as S

import Data.Text.Encoding (encodeUtf8Builder)
import Data.ByteString.Builder(toLazyByteString)

import SpaceTraders.Model.System

addSystems :: S.Connection -> [System] -> IO ()
addSystems conn systems = S.withTransaction conn $ S.executeMany conn "INSERT INTO systems(data) VALUES (json(?));" $ S.Only <$> map encode systems

getSystems :: S.Connection -> IO [System]
getSystems conn = do
  ret <- S.query_ conn "SELECT data from systems;"
  return . catMaybes $ map (decode . toLazyByteString . encodeUtf8Builder . head) ret