{-# LANGUAGE RecordWildCards #-}
{-# LANGUAGE RecursiveDo     #-}

module Potato.Flow.Controller.Manipulator.LineSpec (
  spec
) where

import           Relude                                         hiding (empty,
                                                                 fromList)

import           Test.Hspec

import           Potato.Flow.GoatTester

import           Potato.Flow
import           Potato.Flow.Controller.Manipulator.TestHelpers
import           Potato.Flow.Methods.LineDrawer
import           Potato.Flow.RenderCache

import qualified Data.List                                      as L
import qualified Data.Text                                      as T
import qualified Data.IntMap as IM


verifyMostRecentlyCreatedLinesLatestLineLabelHasText :: Text -> GoatTester ()
verifyMostRecentlyCreatedLinesLatestLineLabelHasText text = verifyStateObjectHasProperty "verifyMostRecentlyCreatedLinesLatestLineLabelHasText" fetchfn checkfn where
  fetchfn = composeObjectFetcher fetchLatestLine fetchLineLabel_from_latestLine
  checkfn llabel = if _sAutoLineLabel_text llabel == text
    then Nothing
    else Just $ "found line label with text: " <> _sAutoLineLabel_text llabel <> " expected: " <> text



fetchLatestLine :: OwlPFState -> Either Text SAutoLine
fetchLatestLine pfs = do
  sowl <- case maybeGetMostRecentlyCreatedOwl' pfs of
    Nothing -> Left "failed, no 🦉s"
    Just x  -> Right x
  case _owlItem_subItem (_superOwl_elt sowl) of
    OwlSubItemLine x -> Right x
    x                  -> Left $ "expected SAutoLine got: " <> show x

fetchLineLabel_from_latestLine :: SAutoLine -> Either Text SAutoLineLabel
fetchLineLabel_from_latestLine sline = case _sAutoLine_labels sline of
  []    -> Left "most recently created line has no line labels"
  (x:_) -> Right x

verifyMostRecentlyCreatedLinesLatestLineLabelHasPosition :: (Int, Int) -> GoatTester ()
verifyMostRecentlyCreatedLinesLatestLineLabelHasPosition (px, py) = verifyState "verifyMostRecentlyCreatedLinesLatestLineLabelHasPosition" checkfn where
  checkfn gs = r where
    pfs = goatState_pFState gs
    r' = do
      sowl <- case maybeGetMostRecentlyCreatedOwl' (goatState_pFState gs) of
        Nothing -> Left "failed, no 🦉s"
        Just x  -> Right x
      sline <- case _owlItem_subItem (_superOwl_elt sowl) of
        OwlSubItemLine x -> Right x
        x                  -> Left $ "expected SAutoLine got: " <> show x
      llabel <- case _sAutoLine_labels sline of
        []    -> Left "most recently created line has no line labels"
        (x:_) -> Right x
      return $ getSAutoLineLabelPosition pfs sline llabel
    r = case r' of
      Left e -> Just e
      Right (V2 x y) -> if px == x && py == y
        then Nothing
        else Just $ "expected line label position: " <> show (px, py) <> " got " <> show (x, y)

verifyMostRecentlyCreateLineIsAttached :: (Maybe AttachmentLocation, Maybe AttachmentLocation) -> GoatTester ()
verifyMostRecentlyCreateLineIsAttached (mstartalexp, mendalexp) = verifyStateObjectHasProperty "verifyMostRecentlyCreateLineIsAttached" fetchLatestLine checkfn where
  checkfn sline = r where
    mstartal = fmap _attachment_location (_sAutoLine_attachStart sline)
    mendal = fmap _attachment_location (_sAutoLine_attachEnd sline)
    r = if mstartal == mstartalexp && mendal == mendalexp
      then Nothing
      else Just $ "expected attachments (start, end): (" <> show mstartalexp <> ", " <> show mendalexp <> ") got (" <> show mstartal <> ", " <> show mendal <> ")"


blankOwlPFState :: OwlPFState
blankOwlPFState = OwlPFState emptyOwlTree (SCanvas (LBox 0 200))

expectMidpointCount :: Int -> GoatTester ()
expectMidpointCount n = verifyMostRecentlyCreatedLine $ \sline -> toMaybe (L.length (_sAutoLine_midpoints sline) /= n) ("expected " <> show n <> " midpoint, got: " <> show (_sAutoLine_midpoints sline))

expectLabelCount :: Int -> GoatTester ()
expectLabelCount n = verifyMostRecentlyCreatedLine $ \sline -> toMaybe (L.length (_sAutoLine_labels sline) /= n) ("expected 1 label, got: " <> show (_sAutoLine_labels sline))


initSimpleLine :: GoatTester ()
initSimpleLine = drawCanvasLine (0, 0) (100, 0)

basic_test :: Spec
basic_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  initSimpleLine

  setMarker "add a text label"
  canvasMouseDown (40, 0)
  canvasMouseUp (40, 0)
  expectLabelCount 0 -- no label is created yet beacues we haven't entered any text
  pressKeys "meow meow meow meow"
  expectLabelCount 1
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "meow meow meow meow"


  setMarker "add a midpoint"
  canvasMouseDown (50, 0)
  canvasMouseDown (50, 50)
  canvasMouseUp (50, 50)
  expectMidpointCount 1

basic_cancel_test :: Spec
basic_cancel_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  setMarker "mouse down and cancel and ensure no line is created"
  setTool Tool_Line
  canvasMouseDown (0, 0)
  pressEscape
  canvasMouseDown (100, 0)
  canvasMouseUp (110, 0)
  verifyOwlCount 0

  setMarker "draw a line and cancel after moving it"
  setTool Tool_Line
  canvasMouseDown (0, 0)
  canvasMouseDown (100, 0)
  verifyOwlCount 1
  pressEscape
  verifyOwlCount 0
  canvasMouseDown (110, 0)
  canvasMouseUp (110, 0)
  verifyOwlCount 0

  initSimpleLine

  setMarker "create a midpoint and cancel"
  canvasMouseDown (50, 0)
  canvasMouseDown (50, 1)
  expectMidpointCount 1
  pressEscape
  expectMidpointCount 0
  canvasMouseDown (50, 2)
  canvasMouseUp (50, 2)
  expectMidpointCount 0

  setMarker "add a midpoint"
  canvasMouseDown (50, 0)
  canvasMouseDown (50, 50)
  canvasMouseUp (50, 50)
  expectMidpointCount 1



midpoint_modify_basic_test :: Spec
midpoint_modify_basic_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  initSimpleLine

  setMarker "add a midpoint"
  canvasMouseDown (50, 0)
  canvasMouseDown (50, 1)
  canvasMouseUp (50, 1)
  expectMidpointCount 1

  setMarker "move it"
  canvasMouseDown (50, 1)
  canvasMouseDown (50, 10)
  canvasMouseUp (50, 10)
  -- TODO verify position

  setMarker "destroy it by dragging it over the start anchor"
  canvasMouseDown (50, 10)
  canvasMouseDown (0, 0)
  canvasMouseUp (0, 0)
  expectMidpointCount 0

  setMarker "undo its destruction"
  pressUndo
  expectMidpointCount 1

  setMarker "destroy it by dragging it over the end anchor"
  canvasMouseDown (50, 10)
  canvasMouseDown (100, 0)
  canvasMouseUp (100, 0)
  expectMidpointCount 0




midpoint_double_adjacent_delete_test :: Spec
midpoint_double_adjacent_delete_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  initSimpleLine

  setMarker "add a midpoint"
  canvasMouseDown (50, 0)
  canvasMouseDown (50, 1)
  canvasMouseUp (50, 1)
  expectMidpointCount 1

  setMarker "add another midpoint right next to the first"
  canvasMouseDown (20, 0)
  canvasMouseDown (50, 2)
  canvasMouseUp (50, 2)
  expectMidpointCount 2

  setMarker "add a third midpoint"
  canvasMouseDown (90, 0)
  canvasMouseDown (50, 0)
  expectMidpointCount 3

  setMarker "delete the third midpoint by dragging it over the first"
  canvasMouseDown (50, 1)
  expectMidpointCount 2

  setMarker "you should be allowed to drag it over the second midpoint since it's not adjacent (this is the main edge case we are trying to test)"
  canvasMouseDown (50, 2)
  expectMidpointCount 3


initUnitBox :: (Int, Int) -> GoatTester ()
initUnitBox (x, y) = do
  setMarker "draw a 1x1 box"
  setTool Tool_Box
  canvasMouseDown (x, y)
  canvasMouseDown (x+1, y+1)
  canvasMouseUp (x+1, y+1)
  -- TODO verify box is selected

attaching_delete_test :: Spec
attaching_delete_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do
  initUnitBox (0,0)
  verifyOwlCount 1

  setMarker "draw line attached to box"
  setTool Tool_Line
  canvasMouseDown (-1, 0)
  canvasMouseDown (-10, 1)
  canvasMouseUp (-10, 1)
  verifyOwlCount 2

  -- TODO verify line is attached to box

  setMarker "delete the box"
  canvasMouseDown (0, 0)
  canvasMouseUp (0, 0)
  pressDelete
  verifyOwlCount 1

  -- TODO verify the line that's left is in a sensible place

  pressUndo
  verifyOwlCount 2
  -- TODO verify again that the line is in the expected place

attaching_fully_attached_wont_move_test :: Spec
attaching_fully_attached_wont_move_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do
  initUnitBox (0,0)
  initUnitBox (10,0)
  verifyOwlCount 2

  setMarker "draw line attached to box"
  setTool Tool_Line
  canvasMouseDown (1, 0)
  canvasMouseDown (9, 0)
  canvasMouseUp (9, 0)
  verifyOwlCount 3

  -- TODO verify line is attached to box
  -- TODO verify line is selected

  --
  setMarker "draw another line so that we can select both lines and move them together (otherwise you create a midpoint)"
  setTool Tool_Line
  canvasMouseDown (0, -1)
  canvasMouseDown (10, -1)
  canvasMouseUp (10, -1)
  verifyOwlCount 4

  s1 <- getOwlPFState

  setMarker "select both lines"
  canvasMouseDown (5, -5)
  canvasMouseDown (5, 5)
  canvasMouseUp (5, 5)
  verifySelectionCount 2

  setMarker "try and move the lines"
  canvasMouseDown (5, 0)
  canvasMouseDown (5, 10)
  canvasMouseUp (5, 10)

  s2 <- getOwlPFState
  verify "state did not change after attempting to move line" $ if s1 == s2 then Nothing else Just "it changed!!"

  setMarker "select everything"
  canvasMouseDown (-10, -10)
  canvasMouseDown (20, 20)
  canvasMouseUp (20, 20)
  verifySelectionCount 4

  setMarker "move everything"
  canvasMouseDown (5, 0)
  canvasMouseDown (5, 10)
  canvasMouseUp (5, 10)

  s3 <- getOwlPFState
  verify "state did not change after attempting to move line" $ if s2 == s3 then Just "it didn't change!" else Nothing


endpoint_attach_offset_basic_test :: Spec
endpoint_attach_offset_basic_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do
  initUnitBox (0,0)
  initUnitBox (10,0)
  verifyOwlCount 2

  setMarker "draw line attached to box"
  setTool Tool_Line
  canvasMouseDown (1, 0)
  canvasMouseDown (9, 0)
  canvasMouseUp (9, 0)
  verifyOwlCount 3
  verifyMostRecentlyCreateLineIsAttached (Just AL_Right, Just AL_Left)

  setMarker "move the start endpoint on the box we attached to"
  canvasMouseDown (1, 0)
  canvasMouseDown (-1, 0)
  verifyMostRecentlyCreateLineIsAttached (Just AL_Left, Just AL_Left)



endpoint_attach_offset_detach_and_reattach_test :: Spec
endpoint_attach_offset_detach_and_reattach_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do
  drawCanvasBox (0,0,4,4)
  verifyOwlCount 1

  setMarker "draw line attached to box"
  setTool Tool_Line
  canvasMouseDown (2, -1)
  canvasMouseDown (2, -10)
  canvasMouseUp (2, -10)
  verifyOwlCount 2
  verifyMostRecentlyCreateLineIsAttached (Just AL_Top, Nothing)

  setMarker "move the start endpoint on the box we attached to away from the box"
  canvasMouseDown (2, -1)
  canvasMouseDown (2, -100)
  verifyMostRecentlyCreateLineIsAttached (Nothing, Nothing)

  -- TODO this is suppose to work if we drag back to anywhere that projects onto the box, but it's not for some reason and I don't know why
  setMarker "move the start endpoint back onto the box"
  canvasMouseDown (2, -1)
  canvasMouseUp (2, -1)
  verifyMostRecentlyCreateLineIsAttached (Just AL_Top, Nothing)

label_undo_test :: Spec
label_undo_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  initSimpleLine
  verifySelectionCount 1

  setMarker "add a label"
  canvasMouseDown (50, 0)
  canvasMouseUp (50, 0)
  pressKey '1'
  pressKey '2'
  pressKey '3'
  pressKey '4'
  pressKey '5'
  pressReturn
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "12345"
  pressUndo
  -- since this is a newly created label, undo deletes the label
  expectLabelCount 0
  pressRedo
  expectLabelCount 1

label_move_test :: Spec
label_move_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  let screams = "😱😱😱"

  initSimpleLine

  setMarker "add a label"
  canvasMouseDown (50, 0)
  canvasMouseUp (50, 0)
  pressKeys screams
  pressEscape
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText (T.pack screams)
  verifyMostRecentlyCreatedLinesLatestLineLabelHasPosition (50,0)

  setMarker "move the label "
  canvasMouseDown (50, 0)
  canvasMouseDown (40, 0)
  canvasMouseUp (40, 0)
  verifyMostRecentlyCreatedLinesLatestLineLabelHasPosition (40,0)

  setMarker "add a midpoint"
  canvasMouseDown (75, 0)
  canvasMouseDown (76, 0)
  canvasMouseUp (76, 0)
  expectMidpointCount 1

  setMarker "move the label past the midpoint"
  -- TODO this is broken becaues label position got changed after a new midpoint was added
  canvasMouseDown (40, 0)
  canvasMouseDown (90, 0)
  canvasMouseUp (90, 0)
  verifyMostRecentlyCreatedLinesLatestLineLabelHasPosition (90,0)


label_cursor_test :: Spec
label_cursor_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  initSimpleLine
  verifySelectionCount 1

  setMarker "add a label"
  canvasMouseDown (50, 0)
  canvasMouseUp (50, 0)
  pressKey '1'
  pressKey '2'
  pressKey '3'
  pressKey '4'
  pressKey '5'
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "12345"

  setMarker "move the cursor to 3"
  canvasMouseDown (50, 0)
  canvasMouseUp (50, 0)
  pressKey 'A'
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "12A345"

  setMarker "attempt to move the cursor to beyond the line"
  canvasMouseDown (150, 0)
  canvasMouseUp (150, 0)
  verifySelectionCount 0

  setMarker "select the line and type at eol"
  canvasMouseDown (1, 0)
  canvasMouseUp (1, 0)
  verifySelectionCount 1
  canvasMouseDown (47, 0)
  canvasMouseUp (47, 0)
  pressKey 'B'
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "12A345B"

  setMarker "then select directly move the cursor to 1 on the label"
  canvasMouseDown (47, 0)
  canvasMouseUp (47, 0)
  pressKey 'C'
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "C12A345B"



label_delete_test :: Spec
label_delete_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  initSimpleLine
  verifySelectionCount 1

  setMarker "add a label"
  canvasMouseDown (50, 0)
  canvasMouseUp (50, 0)
  setMarker "123"
  pressKey '1'
  pressKey '2'
  pressKey '3'
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "123"

  setMarker "erase it and ensure there is no label"
  pressBackspace
  pressBackspace
  pressBackspace
  expectLabelCount 0
  pressBackspace
  expectLabelCount 0

label_delete_midpoint_test :: Spec
label_delete_midpoint_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  initSimpleLine
  verifySelectionCount 1

  setMarker "add a midpoint"
  canvasMouseDown (50, 0)
  canvasMouseDown (51, 0)
  canvasMouseUp (51, 0)
  expectMidpointCount 1

  -- BROKEN why??
  setMarker "add a label after last midpoint"
  canvasMouseDown (90, 0)
  canvasMouseUp (90, 0)
  pressKey 'D'
  pressReturn
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "D"

  setMarker "delete the midpoint"
  canvasMouseDown (51, 0)
  canvasMouseDown (100, 0)
  canvasMouseUp (100, 0)
  expectMidpointCount 0
  -- ensure the line label has not been affected
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "D"


label_delete_after_back_and_forth_test :: Spec
label_delete_after_back_and_forth_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do
  initSimpleLine
  verifySelectionCount 1

  setMarker "add a label"
  canvasMouseDown (90, 0)
  canvasMouseUp (90, 0)
  pressKeys "chicken"
  pressReturn
  expectLabelCount 1
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "chicken"

  setMarker "add another label"
  canvasMouseDown (10, 0)
  canvasMouseUp (10, 0)
  pressKeys "cat"
  pressReturn
  expectLabelCount 2
  verifyMostRecentlyCreatedLinesLatestLineLabelHasText "cat"


  setMarker "go back to the other label and delete it"
  canvasMouseDown (90, 0)
  canvasMouseUp (90, 0)
  pressBackspace
  pressBackspace
  pressBackspace
  pressBackspace
  pressBackspace
  pressBackspace
  pressBackspace
  pressReturn
  expectLabelCount 1



cache_basic_test :: Spec
cache_basic_test = hSpecGoatTesterWithOwlPFState blankOwlPFState $ do

  initSimpleLine

  sowl <- mustGetMostRecentlyCreatedOwl
  let
    f gs = case IM.lookup (_superOwl_id sowl) (unRenderCache (_goatState_renderCache gs)) of
      Just (OwlItemCache_Line _ _) -> Nothing
      _ -> Just "expected to find cache"
  verifyState "verify cache got created for the line we just created" f



spec :: Spec
spec = do
  describe "Line" $ do
    describe "basic" $ basic_test
    describe "basic_cancel" $ basic_cancel_test
    describe "midpoint_modify_basic" $  midpoint_modify_basic_test
    describe "midpoint_double_adjacent_delete" $  midpoint_double_adjacent_delete_test
    describe "attaching_delete_test" $ attaching_delete_test
    describe "attaching_fully_attached_wont_move_test" $ attaching_fully_attached_wont_move_test
    describe "label_undo_test" $ label_undo_test

    describe "label_cursor_test" $ label_cursor_test
    describe "label_delete_midpoint_test" $ label_delete_midpoint_test
    describe "label_delete_after_back_and_forth_test" $ label_delete_after_back_and_forth_test
    describe "label_delete_test" $ label_delete_test

    describe "endpoint_attach_offset_basic_test" $ endpoint_attach_offset_basic_test
    describe "endpoint_attach_offset_detach_and_reattach_test" $ endpoint_attach_offset_detach_and_reattach_test



    -- TODO enable once you fix line label position after adding/removing midpoints issue
    --describe "label_move_test" $ label_move_test

    -- TODO enable once you fix the "-- TODO DELETE THIS YOU SHOULDN'T HAVE TO DO THIS, this is breaking caching" comment in Goat.hs
    --describe "cache_basic_test" $ cache_basic_test