src/Search.hs

   1 {-
   2     Search.hs
   3
   4     Copyright 2013 Louis-Guillaume Gagnon <louis.guillaume.gagnon@gmail.com>
   5
   6     This program is free software: you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation, either version 3 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 -}
  19
  20 module Search (
  21     search,
  22     usageSearch
  23 ) where
  24
  25 import Data.Char (isDigit)
  26 import Data.List (intersperse)
  27
  28 import Tools.Constants
  29 import Tools.Filter
  30 import Tools.Operation (isHelp)
  31 import Tools.SQL (getAllEntries,getEntry,searchEntries)
  32
  33 search :: [String] -> IO ()
  34
  35 search [] = do
  36     entries <- getAllEntries
  37     putStr $ concat $ intersperse "\n" $ map rowToString entries
  38
  39 search (x:[]) | and $ map isDigit x = do
  40                     entry <- getEntry $ read x
  41                     case entry of
  42                         Left msg  -> error $ "search: " ++ msg
  43                         Right row -> putStr $ rowToString row
  44               | isHelp x  = putStrLn usageSearch
  45               | otherwise = error $ "search: invalid id: " ++ x
  46
  47 search xs = do
  48     filters <- tryGetFilters xs
  49     case filters of
  50         Left msg -> error $ "search: " ++ msg
  51         Right fs -> do
  52             let (single,multi) = separateMulti fs
  53             let as = getValues "Authors"  multi
  54             let ks = getValues "Keywords" multi
  55             let ps = getValues "Pages"    multi
  56             entries <- searchEntries single
  57             let entries' = filtA as $ filtK ks $ filtP ps entries
  58             putStr $ concat $ intersperse "\n" $ map rowToString entries'
  59
  60 usageSearch :: String
  61 usageSearch = "usage: " ++ progName ++ " search <filters>\n\
  62               \or:\n\
  63               \       " ++ progName ++ " search [id]\n" ++ usageFilters