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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
module Persistence.Income
( count
, list
, listv2
, 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)
data Count = Count Int
instance FromRow Count where
fromRow = Count <$> SQLite.field
count :: Query Int
count =
Query (\conn ->
(\[Count n] -> n) <$>
SQLite.query_ conn "SELECT COUNT(*) FROM income WHERE deleted_at IS NULL"
)
list :: Query [Income]
list =
Query (\conn ->
map (\(Row i) -> i) <$>
SQLite.query_ conn "SELECT * FROM income WHERE deleted_at IS NULL"
)
listv2 :: Int -> Int -> Query [Income]
listv2 page perPage =
Query (\conn ->
map (\(Row i) -> i) <$>
SQLite.query
conn
"SELECT * FROM income WHERE deleted_at IS NULL ORDER BY date DESC LIMIT ? OFFSET ?"
(perPage, (page - 1) * perPage)
)
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)
)
|
