module Main (main) where

import           Data.List        (sort)
import           Ide.Plugin.Notes (Log, descriptor)
import           System.FilePath  ((</>))
import           Test.Hls

plugin :: PluginTestDescriptor Log
plugin = mkPluginTestDescriptor descriptor "notes"

main :: IO ()
main = defaultTestRunner $
  testGroup "Notes"
    [ gotoNoteTests
    , noteReferenceTests
    , hoverNoteTests
    ]

runSessionWithServer' :: FilePath -> (FilePath -> Session a) -> IO a
runSessionWithServer' fp act =
    runSessionWithTestConfig def
        { testLspConfig = def
        , testPluginDescriptor = plugin
        , testDirLocation = Left fp
        } act

noteReferenceTests :: TestTree
noteReferenceTests = testGroup "Note References"
   [
      testCase "multi_file" $ runSessionWithServer' testDataDir $ \dir -> do
        doc <- openDoc "NoteDef.hs" "haskell"
        waitForKickDone
        refs0 <- getReferences doc (Position 21 15) False
        let refs = sort refs0
            fp = dir </> "NoteDef.hs"

        liftIO $ sort refs @?= sort
            [ Location (filePathToUri (dir </> "Other.hs")) (Range (Position 6 13) (Position 6 13))
            , Location (filePathToUri fp) (Range (Position 9 9) (Position 9 9))
            , Location (filePathToUri fp) (Range (Position 5 67) (Position 5 67))
            ]
    ]

gotoNoteTests :: TestTree
gotoNoteTests = testGroup "Goto Note Definition"
    [
      testCase "single_file" $ runSessionWithServer' testDataDir $ \dir -> do
        doc <- openDoc "NoteDef.hs" "haskell"
        waitForKickDone
        defs <- getDefinitions doc (Position 3 41)
        let fp = dir </> "NoteDef.hs"
        liftIO $ defs @?= InL (Definition (InR [Location (filePathToUri fp) (Range (Position 11 9) (Position 11 9))]))
    , testCase "liberal_format" $ runSessionWithServer' testDataDir $ \dir -> do
        doc <- openDoc "NoteDef.hs" "haskell"
        waitForKickDone
        defs <- getDefinitions doc (Position 5 64)
        let fp = dir </> "NoteDef.hs"
        liftIO $ defs @?= InL (Definition (InR [Location (filePathToUri fp) (Range (Position 21 11) (Position 21 11))]))

    , testCase "invalid_note" $ runSessionWithServer' testDataDir $ const $ do
        doc <- openDoc "NoteDef.hs" "haskell"
        waitForKickDone
        defs <- getDefinitions doc (Position 6 54)
        liftIO $ defs @?= InL (Definition (InR []))

    , testCase "no_note" $ runSessionWithServer' testDataDir $ const $ do
        doc <- openDoc "NoteDef.hs" "haskell"
        waitForKickDone
        defs <- getDefinitions doc (Position 1 0)
        liftIO $ defs @?= InL (Definition (InR []))

    , testCase "unopened_file" $ runSessionWithServer' testDataDir $ \dir -> do
        doc <- openDoc "Other.hs" "haskell"
        waitForKickDone
        defs <- getDefinitions doc (Position 5 20)
        let fp = dir </> "NoteDef.hs"
        liftIO $ defs @?= InL (Definition (InR [Location (filePathToUri fp) (Range (Position 15 6) (Position 15 6))]))
    ]

testDataDir :: FilePath
testDataDir = "plugins" </> "hls-notes-plugin" </> "test" </> "testdata"

hoverNoteTests :: TestTree
hoverNoteTests = testGroup "Hover Notes"
  [ testCase "hover normal-notes" $
      runSessionWithServer' testDataDir $ \_dir -> do
        let file = "HoverNote.hs"
            pos  = Position 24 10
        doc <- openDoc file "haskell"
        waitForKickDone
        hover <- getHover doc pos

        let expected =
              Just $ Hover (InL $ MarkupContent MarkupKind_Markdown
              "Normal-Notes\n\nContent of Normal Notes\n\n")
              (Just (Range (Position 24 3) (Position 24 22)))

        liftIO $ hover @?= expected


  , testCase "hover multi-notes-one" $
      runSessionWithServer' testDataDir $ \_dir -> do
        let file = "HoverNote.hs"
            pos  = Position 25 10
        doc <- openDoc file "haskell"
        waitForKickDone
        hover <- getHover doc pos

        let expected =
              Just $ Hover (InL $ MarkupContent MarkupKind_Markdown
              "Multi-Notes-one\n\nContent of Multi-Notes-one\n\n\n")
              (Just (Range (Position 25 3) (Position 25 25)))

        liftIO $ hover @?= expected


  , testCase "hover multi-notes-two" $
      runSessionWithServer' testDataDir $ \_dir -> do
        let file = "HoverNote.hs"
            pos  = Position 26 10
        doc <- openDoc file "haskell"
        waitForKickDone
        hover <- getHover doc pos

        let expected =
              Just $ Hover (InL $ MarkupContent MarkupKind_Markdown
              "Multi-Notes-two\n\nContent of Multi-Notes-two\n\n")
              (Just (Range (Position 26 3) (Position 26 25)))

        liftIO $ hover @?= expected


  , testCase "hover single-comment-declaration" $
      runSessionWithServer' testDataDir $ \_dir -> do
        let file = "HoverNote.hs"
            pos  = Position 27 10
        doc <- openDoc file "haskell"
        waitForKickDone
        hover <- getHover doc pos

        let expected =
              Just $ Hover (InL $ MarkupContent MarkupKind_Markdown
              "Single Comment Declaration\n\nContent of Single Comment Declaration\n\n")
              (Just (Range (Position 27 3) (Position 27 36)))

        liftIO $ hover @?= expected

  , testCase "hover single-comment leak" $ runSessionWithServer' testDataDir $ \_dir -> do
        let file = "NoteDef.hs"
            pos  = Position 5 69
        doc <- openDoc file "haskell"
        waitForKickDone
        hover <- getHover doc pos

        let expected =
              Just $ Hover (InL $ MarkupContent MarkupKind_Markdown
              "Single line comments\n\nGHC's notes script only allows multiline comments to define notes, but in the\nHLS codebase this single line style can be found as well.\n\n")
              (Just (Range (Position 5 61) (Position 5 88)))

        liftIO $ hover @?= expected
  ]
