@@ -1,16 +1,13 @@
11{-# LANGUAGE
2− DataKinds,
3− DeriveGeneric,
2+ DataKinds ,
3+ DeriveGeneric ,
44 OverloadedLabels,
5− OverloadedStrings,
6− TypeApplications,
5+ OverloadedStrings ,
6+ TypeApplications ,
77 TypeOperators,
88 PartialTypeSignatures,
99 FlexibleInstances,
10− KindSignatures,
11− DeriveAnyClass,
12− RecordWildCards,
13− ScopedTypeVariables #-}
10+ KindSignatures #-}
1411
1512{-# OPTIONS_GHC -Wno-partial-type-signatures #-}
1613
@@ -19,7 +16,7 @@ module Data.Query (
1916 Schema,
2017 createTables,
2118 getPosts,
22− -- getUser,
19+ getUser,
2320 getUsers,
2421 getAllUsers,
2522 getLastPosts,
@@ -43,31 +40,8 @@ module Data.Query (
4340 createGroupChat,
4441 updateGroupChat,
4542 leaveGroupChat,
46− getGroupChatForUser,
47− getFaculty,
48− updateFaculty,
49− getFacultyFromQuery,
50− getUserWithEmail,
51− createToken,
52− validateToken,
53− invalidateToken,
54− tryVerifyToken,
55− getVerifiedToken,
56− UserCreation(..),
57− createUser,
58− tryActivateToken,
59− deactivateToken,
60− searchUsers,
61− searchTags,
62− updateUser,
63− UserUpdateRow(..),
64− killAllTokens,
65− channelCountForUser,
66− pruneAuth,
67− searchChannels,
68− isValidFaculty,
69− groupChatIsValidForUser
70− ) where
43+ getGroupChatForUser
44+) where
7145
7246import Squeal.PostgreSQL
7347import Data.User
@@ -93,23 +67,14 @@ import Servant.Docs
9367import Data.Proxy
9468import Data.Vector (Vector, toList, fromList)
9569import Data.Text (Text)
96−import qualified Data.Text as T
9770import Data.Channel.NamedChannel
9871import Data.Channel.AnonymousChannel
99−import Data.Maybe (fromMaybe, catMaybes, isJust)
72+import Data.Maybe (fromMaybe)
10073import Squeal.PostgreSQL.Render
10174import Data.GroupChat
10275import Data.Message
10376import Server.Query.Dialog
10477import qualified Server.Query.Pagination as P
105−import Data.Faculty
106−import Data.ByteString (ByteString)
107−import Data.Foldable (fold)
108−import Data.Char
109−import qualified Data.Vector as V
110−import Server.Auth.Token
111−import Server.Error
112−import Data.Text.Validator
11378
11479-- MARK: Documentation
11580
@@ -172,76 +137,31 @@ getPostQ oTe te = select (
172137-- let rows = M.fromList $ map (tupleBy $ quoteRowId . head) . groupBy ((==) `on` quoteRowId) . sortBy (compare `on` quoteRowId) $ rows'
173138-- return $ M.lookup qId rows >>= rowsToQuote rows
174139
175−getUserWithEmailQ :: Query Schema (TuplePG (Only UserEmail)) (RowPG (User FacultyUrl))
176−getUserWithEmailQ = select (
177− #userRowId `as` #userId :*
178− #userRowFirstName `as` #firstName :*
179− #userRowMiddleName `as` #middleName :*
180− #userRowLastName `as` #lastName :*
181− #userRowFacultyUrl `as` #userFaculty :*
182− #userRowEmail `as` #userEmail
183− ) (from (table #users) &
184− where_ (#userRowEmail .== param @1) &
185− where_ (notNull #userRowIsValidated))
140+getUserQ :: Query Schema (TuplePG (Only UserId)) (RowPG User)
141+getUserQ = selectStar $ from (table #users) & where_ (#userId .== param @1)
186142
187143verifiedUserIdsQ :: [UserId] -> Maybe (Query Schema '[] (RowPG (Only UserId)))
188144verifiedUserIdsQ uIds = case idsToColumn uIds of
189− Just idQ -> Just $ select (#users ! #userRowId `as` #fromOnly) $ from (subquery (idQ `As` #ids) &
145+ Just idQ -> Just $ select (#users ! #userId `as` #fromOnly) $ from (subquery (idQ `As` #ids) &
190146 innerJoin (table #users)
191− (#ids ! #id .== #users ! #userRowId))
147+ (#ids ! #id .== #users ! #userId))
192148 Nothing -> Nothing
193149
194−getUsersQ :: [UserId] -> Maybe (Query Schema '[] (RowPG UserRow))
150+
151+getUsersQ :: [UserId] -> Maybe (Query Schema '[] (RowPG User))
195152getUsersQ uIds = case idsToColumn uIds of
196− Just idQ -> Just $ select (
197− #users ! #userRowId :*
198− #users ! #userRowFirstName :*
199− #users ! #userRowMiddleName :*
200− #users ! #userRowLastName :*
201− #users ! #userRowFacultyUrl :*
202− #users ! #userRowEmail :*
203− #users ! #userRowIsValidated :*
204− #faculties ! #facultyName `as` #userRowFacultyName :*
205− #faculties ! #facultyUrl `as` #userRowFacultyURL :*
206− #faculties ! #facultyPath `as` #userRowFacultyPath :*
207− #faculties ! #facultyCampusName `as` #userRowFacultyCampusName :*
208− #faculties ! #facultyCampusCode `as` #userRowFacultyCampusCode :*
209− #faculties ! #facultyTags `as` #userRowFacultyTags :*
210− #faculties ! #facultyAddress `as` #userRowFacultyAddress
211− ) (from (subquery (idQ `As` #ids) &
153+ Just idQ -> Just $ selectDotStar #users $ from (subquery (idQ `As` #ids) &
212154 innerJoin (table #users)
213− (#ids ! #id .== #users ! #userRowId) &
214− innerJoin (table #faculties)
215− (#users ! #userRowFacultyUrl .== #faculties ! #facultyUrl)) &
216− where_ (notNull $ #users ! #userRowIsValidated))
155+ (#ids ! #id .== #users ! #userId))
217156 Nothing -> Nothing
218157
219−getAllUsersQ :: Query Schema '[] (RowPG UserRow)
220−getAllUsersQ = select (
221− #users ! #userRowId :*
222− #users ! #userRowFirstName :*
223− #users ! #userRowMiddleName :*
224− #users ! #userRowLastName :*
225− #users ! #userRowFacultyUrl :*
226− #users ! #userRowEmail :*
227− #users ! #userRowIsValidated :*
228− #faculties ! #facultyName `as` #userRowFacultyName :*
229− #faculties ! #facultyUrl `as` #userRowFacultyURL :*
230− #faculties ! #facultyPath `as` #userRowFacultyPath :*
231− #faculties ! #facultyCampusName `as` #userRowFacultyCampusName :*
232− #faculties ! #facultyCampusCode `as` #userRowFacultyCampusCode :*
233− #faculties ! #facultyTags `as` #userRowFacultyTags :*
234− #faculties ! #facultyAddress `as` #userRowFacultyAddress
235− ) (from ((table #users) &
236− innerJoin (table #faculties)
237− (#users ! #userRowFacultyUrl .== #faculties ! #facultyUrl))&
238− where_ (notNull $ #users ! #userRowIsValidated))
158+getAllUsersQ :: Query Schema '[] (RowPG User)
159+getAllUsersQ = selectStar $ from $ table #users
239160
240161getPostsByIdQ :: [PostId] -> Maybe (Query Schema '[] (RowPG PostRowResponse))
241162getPostsByIdQ pIds = case idsToColumn pIds of
242163 Just idQ -> Just $ selectDotStar #posts $ from (subquery (idQ `As` #ids) &
243− innerJoin (subquery $ getPostQ
244− (orderBy [#postElements ! #rowElementId & Desc, #postElements ! #rowElementOrd & Asc]) id `As` #posts)
164+ innerJoin (subquery $ getPostQ (orderBy [#postElements ! #rowElementOrd & Asc] . orderBy [#posts ! #postRowId & Desc]) id `As` #posts)
245165 (#ids ! #id .== #posts ! #postRowId))
246166 Nothing -> Nothing
247167
@@ -297,58 +217,12 @@ getChannelForUserQ = selectStar $
297217 from (table #channels) &
298218 where_ (#namedChannelFullOwner .== param @1 .&& #namedChannelFullId .== param @2)
299219
300−updateUserQ :: Manipulation Schema (TuplePG (UserUpdateRow)) (RowPG (Only UserId))
301−updateUserQ = update #users (
302− Same `as` #userRowId :*
303− Set (param @2) `as` #userRowFirstName :*
304− Set (param @3) `as` #userRowMiddleName :*
305− Set (param @4) `as` #userRowLastName :*
306− Set (param @5) `as` #userRowFacultyUrl :*
307− Same `as` #userRowEmail :*
308− Same `as` #userRowIsValidated)
309− (#userRowId .== param @1)
310− (Returning $ #userRowId `as` #fromOnly)
311−
312220getAllChannelsForUserQ :: Query Schema (TuplePG (Only UserId)) (RowPG NamedChannelFull)
313221getAllChannelsForUserQ = selectStar $
314222 from (table #channels) &
315223 where_ (#namedChannelFullOwner .== param @1) &
316224 orderBy [#namedChannelFullId & Desc]
317225
318−searchChannelsForUserQ :: Query Schema (TuplePG (UserId, Text)) (RowPG NamedChannelFull)
319−searchChannelsForUserQ = selectStar $
320− from (table #channels) &
321− where_ (#namedChannelFullOwner .== param @1) &
322− orderBy [
323− tsRankCd vector1 query & Desc,
324− tsRankCd vector query & Desc
325− ]
326− where
327− vector = tsVector' (textArrayToText [
328− #namedChannelFullName,
329− unsafeFunction "unnest" #namedChannelFullTags
330− ])
331− vector1 = tsVector' #namedChannelFullName
332− query = tsQuery' (param @2)
333−
334−updateFacultyQ :: Manipulation Schema (TuplePG Faculty) '[]
335−updateFacultyQ = insertRow #faculties
336− (
337− Set (param @1) `as` #facultyName :*
338− Set (param @2) `as` #facultyUrl :*
339− Set (param @3) `as` #facultyPath :*
340− Set (param @4) `as` #facultyCampusName :*
341− Set (param @5) `as` #facultyCampusCode :*
342− Set (param @6) `as` #facultyTags :*
343− Set (param @7) `as` #facultyAddress
344− )
345− (OnConflictDoNothing)
346− (Returning Nil)
347−
348−getFacultyQ :: Query Schema (TuplePG (Only Text)) (RowPG Faculty)
349−getFacultyQ = selectStar (from (table #faculties) &
350− where_ (#facultyUrl .== param @1))
351−
352226createNamedChannelQ :: Manipulation Schema (TuplePG NamedChannelCreation) (RowPG (Only NamedChannelId))
353227createNamedChannelQ = insertRow #channels
354228 (
@@ -361,11 +235,6 @@ createNamedChannelQ = insertRow #channels
361235 OnConflictDoRaise
362236 (Returning $ #namedChannelFullId `as` #fromOnly)
363237
364−countChannelsQ :: Query Schema (TuplePG (Only UserId)) (RowPG (Only Int64))
365−countChannelsQ = select (countStar `as` #fromOnly) (from (table #channels) &
366− where_ (#namedChannelFullOwner .== param @1) &
367− groupBy #namedChannelFullOwner)
368−
369238deleteNamedChannelQ :: Manipulation Schema (TuplePG (Only NamedChannelId)) '[]
370239deleteNamedChannelQ = deleteFrom_ #channels (#namedChannelFullId .== param @1)
371240
@@ -387,7 +256,7 @@ getChannelPostsQ
387256 -> Vector Text
388257 -> Query
389258 Schema
390− (TuplePG (Only (Maybe PostId))) -- post id from which to start
259+ (TuplePG (Only (Maybe PostId))) -- ^ post id from which to start
391260 (RowPG PostRowResponse)
392261getChannelPostsQ lim dir uIds tags = getLastPostsQ
393262 (where_ $ jsonbLit tags .?| #postRowTags .|| isUser #postRowAuthorId)
@@ -398,7 +267,7 @@ getChannelPostsQ lim dir uIds tags = getLastPostsQ
398267 Just uIdsQ -> (`in_` uIdsQ)
399268 Nothing -> const $ false
400269
401−getTagsQ :: Query Schema params (RowPG (Only String))
270+getTagsQ :: Query Schema '[] (RowPG (Only String))
402271getTagsQ = selectDistinct (unsafeFunction "unnest" #postRowTags `as` #fromOnly) $
403272 from (table #posts)
404273
@@ -409,26 +278,15 @@ createMessageQ :: Manipulation
409278createMessageQ = insertRow #messages
410279 (
411280 Default `as` #messageStorageId :*
412− Set (notNull $ param @1) `as` #messageStorageAuthorId :*
281+ Set (param @1) `as` #messageStorageAuthorId :*
413282 Set (param @2) `as` #messageStorageDestinationGroupId :*
414283 Set (param @3) `as` #messageStorageDestinationUserId :*
415− Set (notNull $ param @4) `as` #messageStorageBody :*
284+ Set (param @4) `as` #messageStorageBody :*
416285 Set currentTimestamp `as` #messageStorageTime
417286 )
418287 OnConflictDoRaise
419288 (Returning $ #messageStorageId `as` #fromOnly)
420289
421−createPhantomMessageQ :: Manipulation Schema (TuplePG (Only GroupChatId)) '[]
422−createPhantomMessageQ = insertRow_ #messages
423− (
424− Default `as` #messageStorageId :*
425− Set null_ `as` #messageStorageAuthorId :*
426− Set (notNull $ param @1) `as` #messageStorageDestinationGroupId :*
427− Set null_ `as` #messageStorageDestinationUserId :*
428− Set null_ `as` #messageStorageBody :*
429− Set currentTimestamp `as` #messageStorageTime
430− )
431−
432290createGroupChatQ :: Manipulation Schema (TuplePG GroupChatCreation) (RowPG (Only GroupChatId))
433291createGroupChatQ = insertRow #groupChats
434292 (
@@ -461,7 +319,7 @@ type ConditionOp schema from grouping params nullity0 nullity1 (ty :: PGType)
461319
462320type ChatQueryParams = TuplePG
463321 ( UserId
464− , Maybe String -- Just UserId as String
322+ , Maybe String -- ^ Just UserId as String
465323 , Maybe MessageId
466324 )
467325
@@ -504,7 +362,7 @@ getChatsQ lim dir = select (
504362 (ifThenElse
505363 (#messageStorageAuthorId .== param @1)
506364 #messageStorageDestinationUserId
507− #messageStorageAuthorId) `as` #messageUserId) $
365+ (notNull #messageStorageAuthorId)) `as` #messageUserId) $
508366 from (table #messages) &
509367 where_ (isNotNull #messageStorageDestinationUserId .&&
510368 (#messageStorageAuthorId .== param @1 .||
@@ -568,272 +426,18 @@ getGroupChatForUserQ = selectStar $
568426 where_ (#groupChatId .== param @2) &
569427 where_ (#groupChatUsers .? param @1)
570428
571−getFacultySearchResultQ :: Query Schema (TuplePG (Only Text)) (RowPG Faculty)
572−getFacultySearchResultQ = selectStar (from (table #faculties) &
573− where_ (vector @@ query) &
574− orderBy [
575− tsRankCd vector1 query & Desc,
576− tsRankCd vector2 query & Desc,
577− tsRankCd vector query & Desc
578− ])
579− where
580− vector = tsVector "russian" (textArrayToText [
581− #facultyName,
582− #facultyPath,
583− #facultyCampusName,
584− (arrayToText #facultyTags)])
585− vector1 = tsVector "russian" #facultyName
586− vector2 = tsVector "russian" (textArrayToText [
587− #facultyName,
588− #facultyPath])
589− query = tsQuery "russian" (param @1)
590−
591−searchUsersQ :: Query Schema (TuplePG (Only Text)) (RowPG UserRow)
592−searchUsersQ = select (
593− #users ! #userRowId :*
594− #users ! #userRowFirstName :*
595− #users ! #userRowMiddleName :*
596− #users ! #userRowLastName :*
597− #users ! #userRowFacultyUrl :*
598− #users ! #userRowEmail :*
599− #users ! #userRowIsValidated :*
600− #faculties ! #facultyName `as` #userRowFacultyName :*
601− #faculties ! #facultyUrl `as` #userRowFacultyURL :*
602− #faculties ! #facultyPath `as` #userRowFacultyPath :*
603− #faculties ! #facultyCampusName `as` #userRowFacultyCampusName :*
604− #faculties ! #facultyCampusCode `as` #userRowFacultyCampusCode :*
605− #faculties ! #facultyTags `as` #userRowFacultyTags :*
606− #faculties ! #facultyAddress `as` #userRowFacultyAddress
607− ) (from ((table #users) &
608− innerJoin (table #faculties)
609− (#users ! #userRowFacultyUrl .== #faculties ! #facultyUrl)) &
610− where_ (notNull $ #users ! #userRowIsValidated) &
611− where_ (vector @@ query) &
612− orderBy [tsRankCd vector query & Desc] &
613− limit 100)
614− where
615− vector = tsVector' (textArrayToText [
616− #users ! #userRowFirstName,
617− #users ! #userRowMiddleName,
618− #users ! #userRowLastName
619− ])
620− query = tsQuery' (param @1)
621−
622−tagsByFrquencyQ :: Query Schema params '["tag" ::: 'NotNull (PG Text), "count" ::: 'NotNull 'PGint8]
623−tagsByFrquencyQ = select
624− (#fromOnly `as` #tag :* count #fromOnly `as` #count)
625− (from (subquery $ getTagsQ `As` #tt) &
626− groupBy #fromOnly)
627−
628−getNpopularTagsQ :: Limit -> Query Schema param (RowPG (Only Text))
629−getNpopularTagsQ lim = select (#tag `as` #fromOnly)
630− (from (subquery $ tagsByFrquencyQ `As` #tt) &
631− orderBy [#count & Desc] &
632− limit lim)
633−
634−
635−searchTagsQ :: Query Schema (TuplePG (Only Text)) (RowPG (Only Text))
636−searchTagsQ = select (#tag `as` #fromOnly) $
637− from (subquery (tagsByFrquencyQ `As` #tt)) &
638− where_ (vector @@ query) &
639− orderBy [#count & Desc] &
640− limit 100
641− where
642− query = tsQuery' (param @1)
643− vector = tsVector' #tag
644−
645−tsRankCd
646− :: Expression schema from grouping params ('NotNull 'PGtext)
647− -> Expression schema from grouping params ('NotNull 'PGtext)
648− -> Expression schema from grouping params ('NotNull 'PGfloat4)
649−tsRankCd vector query = UnsafeExpression $
650− "ts_rank_cd" <> parenthesized (commaSeparated ["ARRAY[1,1,1,1]", renderExpression vector, renderExpression query, "2|4|8|16|32"])
651−
652−tsVector
653− :: ByteString -- ^ Language
654− -> Expression schema from grouping params ('NotNull 'PGtext)
655− -> Expression schema from grouping params ('NotNull 'PGtext)
656−tsVector lang x = UnsafeExpression $
657− "to_tsvector" <> parenthesized (commaSeparated [singleQuotedUtf8 lang, renderExpression x])
658−
659−tsVector'
660− :: Expression schema from grouping params ('NotNull 'PGtext)
661− -> Expression schema from grouping params ('NotNull 'PGtext)
662−tsVector' x = UnsafeExpression $
663− "to_tsvector" <> parenthesized (renderExpression x)
664−
665−tsQuery
666− :: ByteString -- ^ Language
667− -> Expression schema from grouping params ('NotNull 'PGtext)
668− -> Expression schema from grouping params ('NotNull 'PGtext)
669−tsQuery lang x = UnsafeExpression $
670− "to_tsquery" <> parenthesized (commaSeparated [singleQuotedUtf8 lang, renderExpression x])
671−
672−tsQuery'
673− :: Expression schema from grouping params ('NotNull 'PGtext)
674− -> Expression schema from grouping params ('NotNull 'PGtext)
675−tsQuery' x = UnsafeExpression $
676− "to_tsquery" <> parenthesized (renderExpression x)
677−
678−arrayToText
679− :: Expression schema from grouping params ('NotNull ('PGvararray ('NotNull 'PGtext)))
680− -> Expression schema from grouping params ('NotNull 'PGtext)
681−arrayToText x = UnsafeExpression $
682− "array_to_string" <> parenthesized (commaSeparated [renderExpression x, "' '"])
683−
684−(@@)
685− :: Expression schema from grouping params (nullity0 'PGtext)
686− -> Expression schema from grouping params (nullity1 'PGtext)
687− -> Expression schema from grouping params (nullity2 'PGbool)
688−(@@) = unsafeBinaryOp "@@"
689−
690−space :: Expression schema from grouping params ('NotNull 'PGtext)
691−space = UnsafeExpression "' '"
692−
693−textArrayToText
694− :: [Expression schema from grouping params ('NotNull 'PGtext)]
695− -> Expression schema from grouping params ('NotNull 'PGtext)
696−textArrayToText = fold . L.intersperse space
697−
698−createUserQ :: Manipulation Schema (TuplePG UserCreation) (RowPG (Only UserId))
699−createUserQ = insertRow #users (
700− Default `as` #userRowId :*
701− Set (param @1) `as` #userRowFirstName :*
702− Set (param @2) `as` #userRowMiddleName :*
703− Set (param @3) `as` #userRowLastName :*
704− Set (param @4) `as` #userRowFacultyUrl :*
705− Set (param @5) `as` #userRowEmail :*
706− Set false `as` #userRowIsValidated)
707− OnConflictDoRaise
708− (Returning (#userRowId `as` #fromOnly))
709−
710−removeUnverifiedUserQ :: Manipulation Schema (TuplePG (Only UserEmail)) '[]
711−removeUnverifiedUserQ = deleteFrom_ #users
712− (#userRowEmail .== param @1 .&&
713− (not_ . notNull $ #userRowIsValidated))
714−
715−
716−createTokenQ :: Manipulation Schema (TuplePG Token) '[]
717−createTokenQ = insertRow_ #tokens (
718− Set (param @1) `as` #tokenUserId :*
719− Set (param @2) `as` #tokenValue :*
720− Set (param @3) `as` #tokenExpiryDate :*
721− Set (param @4) `as` #tokenVerificationCode :*
722− Set (param @5) `as` #tokenActivationTriesLeft :*
723− Set (param @6) `as` #tokenUserAgent :*
724− Set (param @7) `as` #tokenActivationCode :*
725− Set (param @8) `as` #tokenDeactivationCode
726− )
727−
728−updateTokenQ :: Manipulation Schema (TuplePG Token) '[]
729−updateTokenQ = update_ #tokens (
730− Same `as` #tokenUserId :*
731− Same `as` #tokenValue :*
732− Set (param @3) `as` #tokenExpiryDate :*
733− Set (param @4) `as` #tokenVerificationCode :*
734− Set (param @5) `as` #tokenActivationTriesLeft :*
735− Set (param @6) `as` #tokenUserAgent :*
736− Set (param @7) `as` #tokenActivationCode :*
737− Set (param @8) `as` #tokenDeactivationCode)
738− (#tokenUserId .== param @1 .&& #tokenValue .== param @2)
739−
740−deleteTokenQ :: Manipulation Schema (TuplePG (UserId, ByteString)) '[]
741−deleteTokenQ = deleteFrom_ #tokens
742− (#tokenUserId .== param @1 .&& #tokenValue .== param @2)
743−
744−validateUserQ :: Manipulation Schema (TuplePG (Only UserId)) '[]
745−validateUserQ = update_ #users (
746− Same `as` #userRowId :*
747− Same `as` #userRowFirstName :*
748− Same `as` #userRowMiddleName :*
749− Same `as` #userRowLastName :*
750− Same `as` #userRowFacultyUrl :*
751− Same `as` #userRowEmail :*
752− Set true `as` #userRowIsValidated)
753− (#userRowId .== param @1)
754−
755−getTokenQ :: Query Schema (TuplePG (UserId, ByteString)) (RowPG Token)
756−getTokenQ = selectStar (from (table #tokens) &
757− where_ (#tokenUserId .== param @1 .&&
758− #tokenValue .== param @2))
759−
760−getTokenActivationQ :: Query Schema (TuplePG (UserId, ByteString)) (RowPG Token)
761−getTokenActivationQ = selectStar (from (table #tokens) &
762− where_ (#tokenUserId .== param @1 .&&
763− #tokenActivationCode .== param @2))
764−
765−deactivateTokenQ :: Manipulation Schema (TuplePG (UserId, ByteString)) (RowPG (Only Text))
766−deactivateTokenQ = deleteFrom #tokens
767− (#tokenUserId .== param @1 .&&
768− #tokenDeactivationCode .== param @2)
769− (Returning (#tokenUserAgent `as` #fromOnly))
770−
771−getVerifiedTokenQ :: Query Schema (TuplePG (UserId, ByteString)) (RowPG Token)
772−getVerifiedTokenQ = selectStar (from (table #tokens) &
773− where_ (#tokenUserId .== param @1 .&&
774− #tokenValue .== param @2 .&&
775− isNull #tokenVerificationCode .&&
776− currentTimestamp .< #tokenExpiryDate))
777−
778−killAllTokensQ :: Manipulation Schema (TuplePG (Only UserId)) '[]
779−killAllTokensQ = deleteFrom_ #tokens
780− (#tokenUserId .== param @1)
781−
782−pruneDeadTokensQ :: Manipulation Schema '[] '[]
783−pruneDeadTokensQ = deleteFrom_ #tokens
784− (#tokenExpiryDate .< currentTimestamp)
785−
786−pruneGhostUsersQ :: Manipulation Schema '[] '[]
787−pruneGhostUsersQ = deleteFrom_ #users
788− ((not_ . in_ #userRowId $ select (#tokens ! #tokenUserId `as` #id) (from $ table #tokens)) .&&
789− (not_ . notNull $ #userRowIsValidated))
790429
791430-- MARK: FrontEnd Data Structures
792431
793−instance HasValidatableText UserCreation where
794− validateText UserCreation{..} = all validateText [
795− userCreationFirstName,
796− userCreationMiddleName,
797− userCreationLastName,
798− userCreationUserFaculty,
799− userCreationUserEmail]
800−
801−data UserCreation = UserCreation {
802− userCreationFirstName :: Text,
803− userCreationMiddleName :: Text,
804− userCreationLastName :: Text,
805− userCreationUserFaculty :: FacultyUrl,
806− userCreationUserEmail :: Text
807−} deriving (GHC.Generic, SOP.Generic, SOP.HasDatatypeInfo)
808−
809−instance ToJSON UserCreation where
810− toJSON (UserCreation fName mName lName faculty email) = object [
811− "firstName" .= fName,
812− "middleName" .= mName,
813− "lastName" .= lName,
814− "faculty" .= faculty,
815− "email" .= email
816− ]
817−
818−instance FromJSON UserCreation where
819− parseJSON = withObject "user creation" $ \e ->
820− UserCreation <$> e .: "firstName" <*> e .: "middleName" <*> e .: "lastName" <*> e .: "faculty" <*> e .: "email"
821−
822−data UserUpdateRow = UserUpdateRow {
823− userUpdateRowId :: UserId,
824− userUpdateRowFirstName :: Text,
825− userUpdateRowMiddleName :: Text,
826− userUpdateRowLastName :: Text,
827− userUpdateRowUserFaculty :: FacultyUrl
828−} deriving (GHC.Generic, SOP.Generic, SOP.HasDatatypeInfo)
829−
830−
831432data NamedChannelCreation = NamedChannelCreation {
832433 namedChannelCreationOwner :: UserId,
833434 namedChannelCreationName :: Text,
834435 namedChannelCreationTags :: Vector Text,
835436 namedChannelCreationPeopleIds :: Vector UserId
836−} deriving (GHC.Generic, SOP.Generic, SOP.HasDatatypeInfo)
437+} deriving GHC.Generic
438+
439+instance SOP.Generic NamedChannelCreation
440+instance SOP.HasDatatypeInfo NamedChannelCreation
837441
838442addUserToChannelCreate :: UserId -> NamedChannelCreationRequest -> NamedChannelCreation
839443addUserToChannelCreate uId (NamedChannelCreationRequest cName cTags cPeople) = NamedChannelCreation uId cName cTags cPeople
@@ -863,33 +467,11 @@ getChannelForUser uId cId = do
863467 Nothing -> lift $ S.throwError S.err404
864468
865469-- TODO: Reject bad requests. Should be binary.
866−verifyUsers :: [UserId] -> StaticPQ ()
867−verifyUsers uIds =
470+verifiedUsers :: [UserId] -> StaticPQ [UserId]
471+verifiedUsers uIds =
868472 case verifiedUserIdsQ uIds of
869− Just uIdsQ -> do
870− (us :: [UserId]) <- fmap fromOnly <$> (runQuery uIdsQ >>= getRows)
871− when (length us /= length uIds) $ lift $ S.throwError impossibleContent
872− Nothing -> return ()
873−
874−deleteGroupChat :: GroupChatId -> StaticPQ ()
875−deleteGroupChat gId = do
876− _ <- manipulateParams deleteGroupChatQ (Only gId)
877− _ <- manipulateParams deleteGroupChatMessagesQ (Only gId)
878− return ()
879− where
880− deleteGroupChatQ :: Manipulation Schema (TuplePG (Only GroupChatId)) '[]
881− deleteGroupChatQ = deleteFrom_ #groupChats
882− (#groupChatId .== param @1)
883− deleteGroupChatMessagesQ :: Manipulation Schema (TuplePG (Only GroupChatId)) '[]
884− deleteGroupChatMessagesQ = deleteFrom_ #messages
885− (#messageStorageDestinationGroupId .== param @1)
886−
887−-- NOTE: Not transactional
888−createUser :: UserCreation -> StaticPQ UserId
889−createUser creation = do
890− _ <- manipulateParams removeUnverifiedUserQ (Only $ userCreationUserEmail creation)
891− rowThing <- manipulateParams createUserQ creation >>= firstRow >>= fromMaybe500
892− return (fromOnly rowThing)
473+ Just uIdsQ -> fmap fromOnly <$> (runQuery uIdsQ >>= getRows)
474+ Nothing -> return []
893475
894476-- MARK: FrontEnd functions
895477
@@ -921,53 +503,37 @@ getLastPosts pId lim dir = do
921503 postRows <- runQueryParams (getLastPostsQ id lim dir) (Only pId) >>= getRows
922504 rowsToPosts postRows `liftMaybe` wrapIntoResponse
923505
924−getUserWithEmail :: UserEmail -> StaticPQ (Maybe (User FacultyUrl))
925−getUserWithEmail email = runQueryParams getUserWithEmailQ (Only email) >>= firstRow
506+getUser :: UserId -> StaticPQ User
507+getUser uId = runQueryParams getUserQ (Only uId) >>= getRow 0
926508
927−getUsers :: [UserId] -> StaticPQ (Maybe [User Faculty])
928−getUsers ids = getUsersQ ids `liftMaybe` (fmap (Just . map rowToUser) . (runQuery >=> getRows))
509+getUsers :: [UserId] -> StaticPQ (Maybe [User])
510+getUsers ids = getUsersQ ids `liftMaybe` (fmap Just . (runQuery >=> getRows))
929511
930−getAllUsers :: StaticPQ [User Faculty]
931−getAllUsers = fmap rowToUser <$> (runQuery getAllUsersQ >>= getRows)
512+getAllUsers :: StaticPQ [User]
513+getAllUsers = runQuery getAllUsersQ >>= getRows
932514
933515publishPost :: UserId -> PostCreation -> StaticPQ PostId
934−publishPost uId pc = commitedTransactionallyUpdate $ do
516+publishPost uId pc = commitedTransactionallyUpdate $ do
935517 pId' <- manipulateParams createPostRowQ (uId, postCreationTags pc) >>= firstRow
936518 case pId' of
937519 (Just (Only pId)) -> do
938− traversePrepared_ createPostElementRowsQ $ elementsToRows pId (map stripPostElement $ postCreationBody pc)
520+ traversePrepared_ createPostElementRowsQ $ elementsToRows pId (postCreationBody pc)
939521 return pId
940522 _ -> lift $ S.throwError S.err404
941523
942524createNewChannel :: UserId -> NamedChannelCreationRequest -> StaticPQ (Maybe NamedChannelId)
943525createNewChannel uId req = commitedTransactionallyUpdate $ do
944− let uIds = toList $ namedChannelCreationRequestPeopleIds req
945− verifyUsers uIds
526+ uIds <- verifiedUsers . toList $ namedChannelCreationRequestPeopleIds req
946527 cId <- manipulateParams createNamedChannelQ (addUserToChannelCreate uId (req {namedChannelCreationRequestPeopleIds = fromList uIds})) >>= firstRow
947528 return $ fmap fromOnly cId
948529
949−channelCountForUser :: UserId -> StaticPQ Int64
950−channelCountForUser uId = (fromMaybe 0 . fmap fromOnly) <$> (runQueryParams countChannelsQ (Only uId) >>= firstRow)
951−
952530updateChannel :: UserId -> NamedChannel UserId -> StaticPQ ()
953−updateChannel uId req = do
954− let uIds = namedChannelPeopleIds req
955− verifyUsers (toList uIds)
531+updateChannel uId req = commitedTransactionallyUpdate $ do
532+ uIds <- fmap fromList . verifiedUsers . toList $ namedChannelPeopleIds req
956533 _ <- getChannelForUser uId $ namedChannelId req
957− _ <- commitedTransactionallyUpdate $ manipulateParams updateNamedChannelQ $ addUserToChannelUpdate uId (req {namedChannelPeopleIds = uIds})
534+ _ <- manipulateParams updateNamedChannelQ $ addUserToChannelUpdate uId (req {namedChannelPeopleIds = uIds})
958535 return ()
959536
960−getFaculty :: Text -> StaticPQ Faculty
961−getFaculty t = runQueryParams getFacultyQ (Only t) >>= firstRow >>= fromMaybe404
962−
963−isValidFaculty :: FacultyUrl -> StaticPQ Bool
964−isValidFaculty url = do
965− (faculty :: Maybe Faculty) <- runQueryParams getFacultyQ (Only url) >>= firstRow
966− return $ isJust faculty
967−
968−updateFaculty :: Faculty -> StaticPQ ()
969−updateFaculty fac = void $ manipulateParams updateFacultyQ fac
970−
971537getChannelPosts
972538 :: UserId
973539 -> Maybe PostId
@@ -1001,13 +567,13 @@ getAllChannels uId = fmap (map removeUserFromChannel) $
1001567 runQueryParams getAllChannelsForUserQ (Only uId) >>= getRows
1002568
1003569deleteNamedChannel :: UserId -> NamedChannelId -> StaticPQ ()
1004−deleteNamedChannel uId cId = do
570+deleteNamedChannel uId cId = commitedTransactionallyUpdate $ do
1005571 _ <- getChannelForUser uId cId
1006− _ <- commitedTransactionallyUpdate $ manipulateParams deleteNamedChannelQ (Only cId)
572+ _ <- manipulateParams deleteNamedChannelQ (Only cId)
1007573 return ()
1008574
1009−getTags :: Limit -> StaticPQ [String]
1010−getTags lim = map fromOnly <$> (runQuery (getNpopularTagsQ lim) >>= getRows)
575+getTags :: StaticPQ [String]
576+getTags = map fromOnly <$> (runQuery getTagsQ >>= getRows)
1011577
1012578getDialogs
1013579 :: UserId
@@ -1040,48 +606,43 @@ getMessagesInGroupChat
1040606getMessagesInGroupChat selfId gId mId lim dir = do
1041607 _ <- getGroupChatForUser selfId gId
1042608 resp <- runQueryParams (getGroupMessagesQ lim dir) (gId, mId) >>= getRows
1043− return . catMaybes . map restoreMessage $ resp
609+ (return . sequence . map restoreMessage $ resp) >>= fromMaybe500
1044610
1045611sendMessage :: UserId -> MessageCreation -> StaticPQ MessageId
1046−sendMessage selfId (MessageCreation (Just gId) uId@Nothing (Jsonb body)) = do
612+sendMessage selfId (MessageCreation (Just gId) uId@Nothing body) = do
1047613 _ <- getGroupChatForUser selfId gId
1048− (Only mId) <- manipulateParams createMessageQ (selfId, Just gId, uId, Jsonb $ stripPostElement body)
614+ (Only mId) <- manipulateParams createMessageQ (selfId, Just gId, uId, body)
1049615 >>= firstRow >>= fromMaybe500
1050616 return mId
1051−sendMessage selfId (MessageCreation gId@Nothing (Just uId) (Jsonb body)) = do
1052− verifyUsers [uId]
1053− (Only mId) <- manipulateParams createMessageQ (selfId, gId, Just uId, Jsonb $ stripPostElement body)
617+sendMessage selfId (MessageCreation gId@Nothing (Just uId) body) = do
618+ (Only mId) <- manipulateParams createMessageQ (selfId, gId, Just uId, body)
1054619 >>= firstRow >>= fromMaybe500
1055620 return mId
1056621sendMessage _ _ = lift $ S.throwError S.err400
1057622
1058623createGroupChat :: UserId -> GroupChatCreation -> StaticPQ GroupChatId
1059624createGroupChat uId (GroupChatCreation (Jsonb chatUsers) name) = do
1060− verifyUsers . M.keys $ chatUsers
1061− commitedTransactionallyUpdate $ do
1062− let chatUsers' = M.insert uId maxChatPermissions chatUsers
1063− (Only chatId) <- manipulateParams createGroupChatQ (GroupChatCreation (Jsonb chatUsers') name)
1064− >>= firstRow >>= fromMaybe500
1065− _ <- manipulateParams createPhantomMessageQ (Only chatId)
1066− return chatId
625+ let chatUsers' = M.insert uId maxChatPermissions chatUsers
626+ (Only chatId) <- manipulateParams createGroupChatQ (GroupChatCreation (Jsonb chatUsers') name)
627+ >>= firstRow >>= fromMaybe500
628+ -- TODO: Do somehting with this
629+ _ <- sendMessage uId $ MessageCreation (Just chatId) Nothing (Jsonb $ Markdown "henlo")
630+ return chatId
1067631
1068632updateGroupChat :: UserId -> GroupChat -> StaticPQ ()
1069−updateGroupChat uId (GroupChat gId (Jsonb perms) name) = do
1070− verifyUsers . M.keys $ perms
1071− commitedTransactionallyUpdate $ do
1072− (GroupChat _ (Jsonb perms') _) <- getGroupChatForUser uId gId
1073− let userPerm = fromMaybe minChatPermissions $ M.lookup uId perms'
1074− unless (isAdmin userPerm) $ lift $ S.throwError $ S.err401
1075− let newPerms = M.insert uId userPerm perms
1076− void $ manipulateParams updateGroupChatQ (GroupChat gId (Jsonb newPerms) name)
633+updateGroupChat uId (GroupChat gId (Jsonb perms) name) = commitedTransactionallyUpdate $ do
634+ (GroupChat _ (Jsonb perms') _) <- getGroupChatForUser uId gId
635+ let userPerm = fromMaybe minChatPermissions $ M.lookup uId perms'
636+ when (isAdmin userPerm) $
637+ lift $ S.throwError $ S.err401
638+ let newPerms = M.adjust (max userPerm) uId perms
639+ void $ manipulateParams updateGroupChatQ (GroupChat gId (Jsonb newPerms) name)
1077640
1078641leaveGroupChat :: UserId -> GroupChatId -> StaticPQ ()
1079642leaveGroupChat uId gId = commitedTransactionallyUpdate $ do
1080643 gChat@(GroupChat _ (Jsonb perms) _) <- getGroupChatForUser uId gId
1081644 let newPerms = M.delete uId perms
1082− if M.null newPerms
1083− then deleteGroupChat gId
1084− else void $ manipulateParams updateGroupChatQ gChat{groupChatUsers = Jsonb newPerms}
645+ void $ manipulateParams updateGroupChatQ gChat{groupChatUsers = Jsonb newPerms}
1085646
1086647getGroupChatForUser :: UserId -> GroupChatId -> StaticPQ GroupChat
1087648getGroupChatForUser uId gId = do
@@ -1090,152 +651,8 @@ getGroupChatForUser uId gId = do
1090651 (Just c) -> return c
1091652 Nothing -> lift $ S.throwError S.err404
1092653
1093−groupChatIsValidForUser :: UserId -> GroupChatId -> StaticPQ Bool
1094−groupChatIsValidForUser uId gId = do
1095− (chat :: Maybe GroupChat) <- runQueryParams getGroupChatForUserQ (show uId, gId) >>= firstRow
1096− return $ isJust chat
1097−
1098−getFacultyFromQuery :: T.Text -> StaticPQ [Faculty]
1099−getFacultyFromQuery query =
1100− runQueryParams getFacultySearchResultQ (Only . queryFromTextAbbr $ query) >>= getRows
1101−
1102−searchUsers :: T.Text -> StaticPQ [User Faculty]
1103−searchUsers query = do
1104− unless (validateText query) $ lift $ throwError lengthExceeded
1105− if (T.null . T.filter isAlpha $ query)
1106− then return []
1107− else map rowToUser <$> (runQueryParams searchUsersQ (Only . queryFromText $ query) >>= getRows)
1108−
1109−searchTags :: T.Text -> StaticPQ [Text]
1110−searchTags "" = return []
1111−searchTags query = map fromOnly <$>
1112− (runQueryParams searchTagsQ
1113− (Only . (<> ":*") . T.strip $ query) >>= getRows)
1114−
1115−createToken :: Token -> StaticPQ ()
1116−createToken token = void $ manipulateParams createTokenQ token
1117−
1118−killAllTokens :: UserId -> StaticPQ ()
1119−killAllTokens = void . manipulateParams killAllTokensQ . Only
1120−
1121−pruneAuth :: StaticPQ ()
1122−pruneAuth = do
1123− _ <- commitedTransactionallyUpdate . manipulate $ pruneDeadTokensQ
1124− _ <- commitedTransactionallyUpdate . manipulate $ pruneGhostUsersQ
1125− return ()
1126−
1127−searchChannels :: UserId -> Text -> StaticPQ [NamedChannel UserId]
1128−searchChannels uId query = fmap (map removeUserFromChannel) $
1129− runQueryParams searchChannelsForUserQ (uId, queryFromText query) >>= getRows
1130−
1131−-- NOTE: Not transactional
1132−validateToken :: Token -> StaticPQ ()
1133−validateToken token = do
1134− void $ manipulateParams updateTokenQ token{tokenVerificationCode = Nothing, tokenActivationCode = Nothing}
1135− void $ manipulateParams validateUserQ $ Only $ tokenUserId token
1136−
1137−invalidateToken :: UserId -> ByteString -> StaticPQ ()
1138−invalidateToken uId bs = void . manipulateParams deleteTokenQ $ (uId, bs)
1139−
1140−tryVerifyToken :: UserId -> ByteString -> Int -> StaticPQ (Maybe Token)
1141−tryVerifyToken uId tokenData code = commitedTransactionallyUpdate $ do
1142− token' <- runQueryParams getTokenQ (uId, hash tokenData) >>= firstRow
1143− case token' of
1144− Nothing -> return Nothing
1145− Just token -> do
1146− if (tokenVerificationCode token == Just (hashVerificationCode code))
1147− then validateToken token >> (return $ Just token)
1148− else do
1149− let triesLeft = tokenActivationTriesLeft token - 1
1150− if triesLeft <= 0
1151− then invalidateToken (tokenUserId token) (tokenValue token) >> return Nothing
1152− else do
1153− let newToken = token{tokenActivationTriesLeft = triesLeft}
1154− _ <- manipulateParams updateTokenQ newToken
1155− return Nothing
1156−
1157−tryActivateToken :: UserId -> ByteString -> StaticPQ (Maybe Token)
1158−tryActivateToken uId activationData = do
1159− let hData = hash activationData
1160− token' <- runQueryParams getTokenActivationQ (uId, hData) >>= firstRow
1161− case token' of
1162− Nothing -> return Nothing
1163− Just token -> do
1164− if (tokenActivationCode token == Just hData)
1165− then commitedTransactionallyUpdate (validateToken token) >> (return $ Just token)
1166− else return Nothing
1167−
1168−deactivateToken :: UserId -> ByteString -> StaticPQ (Maybe Text)
1169−deactivateToken uId dData = fmap fromOnly <$> (manipulateParams deactivateTokenQ (uId, hash dData) >>= firstRow)
1170−
1171−getVerifiedToken :: UserId -> ByteString -> StaticPQ (Maybe Token)
1172−getVerifiedToken uId tokenData = runQueryParams getVerifiedTokenQ (uId, hash tokenData) >>= firstRow
1173−
1174−updateUser :: (UserUpdateRow) -> StaticPQ (Maybe UserId)
1175−updateUser user = fmap fromOnly <$> (manipulateParams updateUserQ (normalizeUser user) >>= firstRow)
1176−
1177654-- MARK: Utils
1178655
1179−normalizeUser :: UserUpdateRow -> UserUpdateRow
1180−normalizeUser UserUpdateRow{..} = UserUpdateRow {
1181− userUpdateRowId = userUpdateRowId,
1182− userUpdateRowFirstName = T.strip $ userUpdateRowFirstName,
1183− userUpdateRowMiddleName = T.strip $ userUpdateRowMiddleName,
1184− userUpdateRowLastName = T.strip $ userUpdateRowLastName,
1185− userUpdateRowUserFaculty = T.toLower . T.strip $ userUpdateRowUserFaculty
1186−}
1187−
1188−data UserRow = UserRow {
1189− userRowId :: UserId,
1190− userRowFirstName :: Text,
1191− userRowMiddleName :: Text,
1192− userRowLastName :: Text,
1193− userRowFacultyUrl :: FacultyUrl,
1194− userRowEmail :: Text,
1195− userRowIsValidated :: Bool,
1196− userRowFacultyName :: T.Text,
1197− userRowFacultyURL :: FacultyUrl,
1198− userRowFacultyPath :: T.Text,
1199− userRowFacultyCampusName :: T.Text,
1200− userRowFacultyCampusCode :: T.Text,
1201− userRowFacultyTags :: V.Vector T.Text,
1202− userRowFacultyAddress :: T.Text
1203−} deriving (GHC.Generic, SOP.Generic, SOP.HasDatatypeInfo)
1204−
1205−rowToUser :: UserRow -> User Faculty
1206−rowToUser UserRow{..} = User {
1207− userId = userRowId,
1208− firstName = userRowFirstName,
1209− middleName = userRowMiddleName,
1210− lastName = userRowLastName,
1211− userFaculty = Faculty {
1212− facultyName = userRowFacultyName,
1213− facultyUrl = userRowFacultyURL,
1214− facultyPath = userRowFacultyPath,
1215− facultyCampusName = userRowFacultyCampusName,
1216− facultyCampusCode = userRowFacultyCampusCode,
1217− facultyTags = userRowFacultyTags,
1218− facultyAddress = userRowFacultyAddress
1219− },
1220− userEmail = userRowEmail
1221− }
1222−
1223−queryFromTextAbbr :: Text -> Text
1224−queryFromTextAbbr = T.unwords . L.intersperse "&" . map processTerm . filter (not . T.null) . T.split (not . isAlphaNum)
1225− where
1226− processTerm :: Text -> Text
1227− processTerm t = "(" <> t <> ":*" <> (
1228− if T.length t <= 5
1229− then (" | (" <>) . (<> ")") . T.unwords . L.intersperse "<->" . map (<> ":*") . map T.singleton . T.unpack $ t
1230− else mempty
1231− ) <> ")"
1232−
1233−queryFromText :: Text -> Text
1234−queryFromText = T.unwords . L.intersperse "&" . map processTerm . filter (not . T.null) . T.split (not . isPrint)
1235− where
1236− processTerm :: Text -> Text
1237− processTerm t = t <> ":*"
1238−
1239656directionToOperator
1240657 :: P.PaginationDirection
1241658 -> (ConditionOp schema0 from0 grouping0 params0 nullity0 nullity1 (ty :: PGType),
@@ -1266,7 +683,7 @@ unsafeIntToExpr = UnsafeExpression . pack . show
1266683
1267684data ResponseWithUsers r = ResponseWithUsers {
1268685 response :: r,
1269− users :: M.Map UserId (User Faculty)
686+ users :: M.Map UserId User
1270687} deriving (Generic)
1271688
1272689instance (ToJSON r) => ToJSON (ResponseWithUsers r)
@@ -1308,9 +725,5 @@ liftMaybe Nothing _ = pure Nothing
1308725liftMaybe (Just a) f = f a
1309726
1310727fromMaybe500 :: Maybe a -> StaticPQ a
1311−fromMaybe500 Nothing = lift $ S.throwError S.err500
1312−fromMaybe500 (Just a) = return a
1313−
1314−fromMaybe404 :: Maybe a -> StaticPQ a
1315−fromMaybe404 Nothing = lift $ S.throwError S.err404
1316−fromMaybe404 (Just a) = return a
728+fromMaybe500 Nothing = lift $ S.throwError S.err404
729+fromMaybe500 (Just a) = return a