2015-10-18 02:59:43 +00:00
|
|
|
-- eidolon -- A simple gallery in Haskell and Yesod
|
|
|
|
-- Copyright (C) 2015 Amedeo Molnár
|
|
|
|
--
|
|
|
|
-- This program is free software: you can redistribute it and/or modify
|
|
|
|
-- it under the terms of the GNU Affero General Public License as published
|
|
|
|
-- by the Free Software Foundation, either version 3 of the License, or
|
|
|
|
-- (at your option) any later version.
|
|
|
|
--
|
|
|
|
-- This program is distributed in the hope that it will be useful,
|
|
|
|
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
-- GNU Affero General Public License for more details.
|
|
|
|
--
|
|
|
|
-- You should have received a copy of the GNU Affero General Public License
|
|
|
|
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
2015-10-16 13:32:47 +00:00
|
|
|
module Handler.Search where
|
|
|
|
|
|
|
|
import Import
|
2015-10-17 23:28:25 +00:00
|
|
|
import Handler.Commons
|
2015-10-16 19:57:18 +00:00
|
|
|
import Data.Time.Clock
|
2015-10-16 13:32:47 +00:00
|
|
|
import Data.Aeson
|
2015-10-16 19:57:18 +00:00
|
|
|
import Data.Maybe
|
|
|
|
import qualified Data.Text as T
|
2015-10-16 13:32:47 +00:00
|
|
|
import Database.Bloodhound
|
2015-10-17 01:09:04 +00:00
|
|
|
import Network.HTTP.Client (responseBody)
|
2015-10-17 18:03:29 +00:00
|
|
|
import System.FilePath.Posix
|
2015-10-16 13:32:47 +00:00
|
|
|
|
|
|
|
getSearchR :: Handler Html
|
|
|
|
getSearchR = do
|
|
|
|
((res, widget), _) <- runFormGet searchForm
|
2015-10-17 19:52:01 +00:00
|
|
|
case res of
|
|
|
|
FormSuccess query -> do
|
|
|
|
(ru, ra, rm, rc) <- getResults query
|
2015-10-25 21:23:00 +00:00
|
|
|
let a = decode (responseBody ru) :: Maybe (SearchResult SearchUser)
|
|
|
|
let b = decode (responseBody ra) :: Maybe (SearchResult SearchAlbum)
|
|
|
|
let c = decode (responseBody rm) :: Maybe (SearchResult SearchMedium)
|
|
|
|
let d = decode (responseBody rc) :: Maybe (SearchResult SearchComment)
|
|
|
|
let hitListA = case a of {
|
|
|
|
Just as -> hits $ searchHits as;
|
|
|
|
Nothing -> []}
|
|
|
|
let hitListB = case b of {
|
|
|
|
Just bs -> hits $ searchHits bs;
|
|
|
|
Nothing -> []}
|
|
|
|
let hitListC = case c of {
|
|
|
|
Just cs -> hits $ searchHits cs;
|
|
|
|
Nothing -> []}
|
|
|
|
let hitListD = case d of {
|
|
|
|
Just ds -> hits $ searchHits ds;
|
|
|
|
Nothing -> []}
|
|
|
|
userIdList <- return $ mapMaybe (\h -> do
|
2015-10-16 19:57:18 +00:00
|
|
|
if
|
|
|
|
hitIndex h == IndexName "user"
|
|
|
|
then do
|
|
|
|
DocId theId <- return $ hitDocId h
|
|
|
|
Just $ (packKey theId :: UserId)
|
|
|
|
else
|
|
|
|
Nothing
|
|
|
|
) hitListA
|
2015-10-25 21:23:00 +00:00
|
|
|
albumIdList <- return $ mapMaybe (\h -> do
|
2015-10-16 19:57:18 +00:00
|
|
|
if
|
|
|
|
hitIndex h == IndexName "album"
|
|
|
|
then do
|
|
|
|
DocId theId <- return $ hitDocId h
|
|
|
|
Just $ (packKey theId :: AlbumId)
|
|
|
|
else
|
|
|
|
Nothing
|
|
|
|
) hitListB
|
2015-10-25 21:23:00 +00:00
|
|
|
mediumIdList <- return $ mapMaybe (\h -> do
|
2015-10-16 19:57:18 +00:00
|
|
|
if
|
|
|
|
hitIndex h == IndexName "medium"
|
|
|
|
then do
|
|
|
|
DocId theId <- return $ hitDocId h
|
|
|
|
Just $ (packKey theId :: MediumId)
|
|
|
|
else
|
|
|
|
Nothing
|
|
|
|
) hitListC
|
2015-10-25 21:23:00 +00:00
|
|
|
commentIdList <- return $ mapMaybe (\h -> do
|
2015-10-16 19:57:18 +00:00
|
|
|
if
|
|
|
|
hitIndex h == IndexName "comment"
|
|
|
|
then do
|
|
|
|
DocId theId <- return $ hitDocId h
|
|
|
|
Just $ (packKey theId :: CommentId)
|
|
|
|
else
|
|
|
|
Nothing
|
|
|
|
) hitListD
|
2015-10-17 01:09:04 +00:00
|
|
|
userList <- return . catMaybes =<< mapM (\i -> runDB $ selectFirst [UserId ==. i] []) userIdList
|
|
|
|
albumList <- return . catMaybes =<< mapM (\i -> runDB $ selectFirst [AlbumId ==. i] []) albumIdList
|
|
|
|
mediumList <- return . catMaybes =<< mapM (\i -> runDB $ selectFirst [MediumId ==. i] []) mediumIdList
|
|
|
|
commentList <- return . catMaybes =<< mapM (\i -> runDB $ selectFirst [CommentId ==. i] []) commentIdList
|
|
|
|
let allEmpty = (null userList) && (null albumList) && (null mediumList) && (null commentList)
|
2015-10-17 19:52:01 +00:00
|
|
|
defaultLayout $ do
|
|
|
|
setTitle $ toHtml $ "Eidolon :: Search results for " ++ (T.unpack query)
|
2015-10-17 01:09:04 +00:00
|
|
|
$(widgetFile "result")
|
2015-10-17 19:52:01 +00:00
|
|
|
_ ->
|
|
|
|
defaultLayout $ do
|
|
|
|
setTitle "Eidolon :: Search"
|
2015-10-16 19:57:18 +00:00
|
|
|
$(widgetFile "search")
|
2015-10-16 13:32:47 +00:00
|
|
|
|
2015-10-16 19:57:18 +00:00
|
|
|
searchForm :: Form T.Text
|
2015-10-16 13:32:47 +00:00
|
|
|
searchForm = renderDivs $ areq (searchField True) "Search" Nothing
|
|
|
|
|
2015-10-17 18:03:29 +00:00
|
|
|
getResults :: Text -> Handler (Reply, Reply, Reply, Reply)
|
2015-10-16 13:32:47 +00:00
|
|
|
getResults query = do
|
2016-08-27 11:10:18 +00:00
|
|
|
-- esQuery <- return $ QueryFuzzyLikeThisQuery $ FuzzyLikeThisQuery
|
|
|
|
-- { fuzzyLikeFields = [FieldName "_all"]
|
|
|
|
-- , fuzzyLikeText = query
|
|
|
|
-- , fuzzyLikeMaxQueryTerms = MaxQueryTerms 25
|
|
|
|
-- , fuzzyLikeIgnoreTermFrequency = IgnoreTermFrequency False
|
|
|
|
-- , fuzzyLikeFuzziness = Fuzziness 0.6
|
|
|
|
-- , fuzzyLikePrefixLength = PrefixLength 0
|
|
|
|
-- , fuzzyLikeBoost = Boost 1.0
|
|
|
|
-- , fuzzyLikeAnalyzer = Nothing
|
|
|
|
-- }
|
|
|
|
esQuery <- return $ QueryFuzzyQuery $ FuzzyQuery
|
|
|
|
{ fuzzyQueryField = FieldName "_all"
|
|
|
|
, fuzzyQueryValue = query
|
|
|
|
, fuzzyQueryPrefixLength = PrefixLength 0
|
|
|
|
, fuzzyQueryMaxExpansions = MaxExpansions 50
|
|
|
|
, fuzzyQueryFuzziness = Fuzziness 0.6
|
|
|
|
, fuzzyQueryBoost = Boost 1.0
|
2015-10-18 00:51:51 +00:00
|
|
|
}
|
2015-10-17 23:28:25 +00:00
|
|
|
su <- runBH' $ searchByIndex (IndexName "user") $ mkSearch (Just esQuery) Nothing
|
|
|
|
sa <- runBH' $ searchByIndex (IndexName "album") $ mkSearch (Just esQuery) Nothing
|
|
|
|
sm <- runBH' $ searchByIndex (IndexName "medium") $ mkSearch (Just esQuery) Nothing
|
|
|
|
sc <- runBH' $ searchByIndex (IndexName "comment") $ mkSearch (Just esQuery) Nothing
|
2015-10-17 18:03:29 +00:00
|
|
|
return (su, sa, sm, sc)
|
2015-10-16 19:57:18 +00:00
|
|
|
|
|
|
|
data SearchUser = SearchUser
|
|
|
|
{ suName :: T.Text
|
|
|
|
, suSlug :: T.Text
|
|
|
|
}
|
|
|
|
|
|
|
|
instance FromJSON SearchUser where
|
|
|
|
parseJSON (Object o) = SearchUser
|
|
|
|
<$> o .: "name"
|
|
|
|
<*> o .: "slug"
|
|
|
|
parseJSON _ = mempty
|
|
|
|
|
|
|
|
data SearchAlbum = SearchAlbum
|
|
|
|
{ saName :: T.Text }
|
|
|
|
|
|
|
|
instance FromJSON SearchAlbum where
|
|
|
|
parseJSON (Object o) = SearchAlbum <$> o .: "name"
|
|
|
|
parseJSON _ = mempty
|
|
|
|
|
|
|
|
data SearchMedium = SearchMedium
|
|
|
|
{ smName :: Text
|
|
|
|
, smTime :: UTCTime
|
|
|
|
, smDescription :: Textarea
|
|
|
|
, smTags :: [T.Text]
|
|
|
|
}
|
|
|
|
|
|
|
|
instance FromJSON SearchMedium where
|
|
|
|
parseJSON (Object o) = SearchMedium
|
|
|
|
<$> o .: "name"
|
|
|
|
<*> o .: "time"
|
|
|
|
<*> o .: "description"
|
|
|
|
<*> o .: "tags"
|
|
|
|
parseJSON _ = mempty
|
|
|
|
|
|
|
|
data SearchComment = SearchComment
|
|
|
|
{ scAuthor :: Text
|
|
|
|
, scTime :: UTCTime
|
|
|
|
, scContent :: Text
|
|
|
|
}
|
|
|
|
|
|
|
|
instance FromJSON SearchComment where
|
|
|
|
parseJSON (Object o) = SearchComment
|
|
|
|
<$> o .: "author"
|
|
|
|
<*> o .: "time"
|
|
|
|
<*> o .: "content"
|
|
|
|
parseJSON _ = mempty
|