1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
module Persistence.Income
( list
, create
, edit
, delete
) where
import Data.Maybe (listToMaybe)
import Data.Time.Calendar (Day)
import Data.Time.Clock (getCurrentTime)
import Database.SQLite.Simple (FromRow (fromRow), Only (Only))
import qualified Database.SQLite.Simple as SQLite
import Prelude hiding (id)
import Common.Model (Income (..), IncomeId, PaymentId,
UserId)
import Model.Query (Query (Query))
newtype Row = Row Income
instance FromRow Row where
fromRow = Row <$> (Income <$>
SQLite.field <*>
SQLite.field <*>
SQLite.field <*>
SQLite.field <*>
SQLite.field <*>
SQLite.field <*>
SQLite.field)
list :: Query [Income]
list =
Query (\conn ->
map (\(Row i) -> i) <$>
SQLite.query_ conn "SELECT * FROM income WHERE deleted_at IS NULL"
)
create :: UserId -> Day -> Int -> Query Income
create userId date amount =
Query (\conn -> do
createdAt <- getCurrentTime
SQLite.execute
conn
"INSERT INTO income (user_id, date, amount, created_at) VALUES (?, ?, ?, ?)"
(userId, date, amount, createdAt)
incomeId <- SQLite.lastInsertRowId conn
return $ Income
{ _income_id = incomeId
, _income_userId = userId
, _income_date = date
, _income_amount = amount
, _income_createdAt = createdAt
, _income_editedAt = Nothing
, _income_deletedAt = Nothing
}
)
edit :: UserId -> IncomeId -> Day -> Int -> Query (Maybe Income)
edit userId incomeId incomeDate incomeAmount =
Query (\conn -> do
mbIncome <- fmap (\(Row i) -> i) . listToMaybe <$>
SQLite.query conn "SELECT * FROM income WHERE id = ?" (Only incomeId)
case mbIncome of
Just income ->
do
currentTime <- getCurrentTime
SQLite.execute
conn
"UPDATE income SET edited_at = ?, date = ?, amount = ? WHERE id = ? AND user_id = ?"
(currentTime, incomeDate, incomeAmount, incomeId, userId)
return . Just $ Income
{ _income_id = incomeId
, _income_userId = userId
, _income_date = incomeDate
, _income_amount = incomeAmount
, _income_createdAt = _income_createdAt income
, _income_editedAt = Just currentTime
, _income_deletedAt = Nothing
}
Nothing ->
return Nothing
)
delete :: UserId -> PaymentId -> Query ()
delete userId paymentId =
Query (\conn ->
SQLite.execute
conn
"UPDATE income SET deleted_at = datetime('now') WHERE id = ? AND user_id = ?"
(paymentId, userId)
)
|
