From ee1c4cd0f615d275d0ed1133d9fcee0e427d07ff Mon Sep 17 00:00:00 2001
From: Joshua Clayton <joshua.clayton@gmail.com>
Date: Sat, 7 May 2016 16:25:54 -0400
Subject: [PATCH] Add flag to filter by likelihood

---
 app/Main.hs                  | 23 +++++++++++++++++++++--
 src/Unused/ResponseFilter.hs |  8 ++++++++
 src/Unused/Types.hs          |  2 +-
 3 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/app/Main.hs b/app/Main.hs
index 0dd4404..82cec2f 100644
--- a/app/Main.hs
+++ b/app/Main.hs
@@ -3,13 +3,14 @@ module Main where
 import Options.Applicative
 import System.IO (hSetBuffering, BufferMode(NoBuffering), stdout)
 import Unused.Parser (parseLines)
-import Unused.Types (ParseResponse)
-import Unused.ResponseFilter (withOneOccurrence, withOneFile)
+import Unused.Types (ParseResponse, RemovalLikelihood(..))
+import Unused.ResponseFilter (withOneOccurrence, withOneFile, withLikelihoods)
 import Unused.CLI (SearchRunner(..), executeSearch, printParseError, printSearchResults, resetScreen)
 
 data Options = Options
     { oSearchRunner :: SearchRunner
     , oAllOccurrencesAndFiles :: Bool
+    , oLikelihoods :: [RemovalLikelihood]
     }
 
 main :: IO ()
@@ -40,6 +41,7 @@ optionFilters o =
   where
     filters =
       [ if oAllOccurrencesAndFiles o then id else withOneOccurrence . withOneFile
+      , withLikelihoods $ oLikelihoods o
       ]
 
 parseOptions :: Parser Options
@@ -47,6 +49,7 @@ parseOptions =
     Options
     <$> parseSearchRunner
     <*> parseDisplayAllMatches
+    <*> parseLikelihoods
 
 parseSearchRunner :: Parser SearchRunner
 parseSearchRunner =
@@ -60,3 +63,19 @@ parseDisplayAllMatches = switch $
     short 'a'
     <> long "all"
     <> help "Display all files and occurrences"
+
+parseLikelihoods :: Parser [RemovalLikelihood]
+parseLikelihoods = many $
+    parseLikelihood <$> parseLikelihoodOption
+
+parseLikelihood :: String -> RemovalLikelihood
+parseLikelihood "high" = High
+parseLikelihood "medium" = Medium
+parseLikelihood "low" = Low
+parseLikelihood _ = Unknown
+
+parseLikelihoodOption :: Parser String
+parseLikelihoodOption = strOption $
+    short 'l'
+    <> long "likelihood"
+    <> help "[Allows multiple] [Allowed values: high, medium, low] Display results based on likelihood"
diff --git a/src/Unused/ResponseFilter.hs b/src/Unused/ResponseFilter.hs
index 170fc6d..ffcd80e 100644
--- a/src/Unused/ResponseFilter.hs
+++ b/src/Unused/ResponseFilter.hs
@@ -1,6 +1,7 @@
 module Unused.ResponseFilter
     ( withOneFile
     , withOneOccurrence
+    , withLikelihoods
     , oneFile
     , oneOccurence
     , isClassOrModule
@@ -22,9 +23,16 @@ withOneOccurrence = applyFilter (const oneOccurence)
 oneOccurence :: TermResults -> Bool
 oneOccurence = (== 1) . trTotalOccurrences
 
+withLikelihoods :: [RemovalLikelihood] -> ParseResponse -> ParseResponse
+withLikelihoods [] = id
+withLikelihoods l = applyFilter (const $ includesLikelihood l)
+
 oneFile :: TermResults -> Bool
 oneFile = (== 1) . trTotalFiles
 
+includesLikelihood :: [RemovalLikelihood] -> TermResults -> Bool
+includesLikelihood l = (`elem` l) . trRemovalLikelihood
+
 isClassOrModule :: TermResults -> Bool
 isClassOrModule = matchRegex "^[A-Z]" . trTerm
 
diff --git a/src/Unused/Types.hs b/src/Unused/Types.hs
index 9816be7..f0f78a7 100644
--- a/src/Unused/Types.hs
+++ b/src/Unused/Types.hs
@@ -23,7 +23,7 @@ data TermResults = TermResults
     , trRemovalLikelihood :: RemovalLikelihood
     } deriving Show
 
-data RemovalLikelihood = High | Medium | Low deriving Show
+data RemovalLikelihood = High | Medium | Low | Unknown deriving (Eq, Show)
 
 type TermMatchSet = Map.Map String TermResults