sml-projects/fcore/normal-mode/normal-move.sml

structure NormalMove =
struct
  open AppType

  fun moveToStart (app: app_type) : AppType.app_type =
    let
      val
        { buffer
        , windowWidth
        , windowHeight
        , searchList
        , bufferModifyTime
        , visualScrollColumn
        , ...
        } = app

      val cursorIdx = 0
      val startLine = 0
      val buffer = LineGap.goToStart buffer

      val drawMsg = NormalModeTextBuilder.build
        ( startLine
        , cursorIdx
        , buffer
        , windowWidth
        , windowHeight
        , searchList
        , visualScrollColumn
        )
      val drawMsg = Vector.concat drawMsg
      val drawMsg = DrawMsg.DRAW_TEXT drawMsg
      val drawMsg = [MailboxType.DRAW drawMsg]

      val mode = NORMAL_MODE ""
    in
      NormalModeWith.bufferAndCursorIdx
        ( app
        , buffer
        , cursorIdx
        , mode
        , startLine
        , searchList
        , drawMsg
        , bufferModifyTime
        , 0
        )
    end

  fun moveToEnd (app: app_type) =
    let
      val
        { buffer
        , windowWidth
        , windowHeight
        , searchList
        , bufferModifyTime
        , visualScrollColumn = prevScrollColumn
        , startLine = prevLineNumber
        , ...
        } = app

      val buffer = LineGap.goToEnd buffer
      val {line = bufferLine, textLength, ...} = buffer

      val bufferIdx = Int.max (0, textLength - 1)
      val bufferLine = bufferLine - 1

      val buffer = LineGap.goToIdx (bufferIdx, buffer)
      val bufferIdx =
        if Cursor.isOnNewlineAfterChr (buffer, bufferIdx) then
          Int.max (0, bufferIdx - 1)
        else
          bufferIdx

      val buffer = LineGap.goToIdx (bufferIdx, buffer)
      val visualScrollColumn =
        TextScroll.getScrollColumn
          (buffer, bufferIdx, windowWidth, prevScrollColumn)

      val bufferLine =
        TextScroll.getStartLine
          (prevLineNumber, bufferLine, windowHeight, #lineLength buffer)
      val buffer = LineGap.goToLine (bufferLine, buffer)

      val drawMsg = NormalModeTextBuilder.build
        ( bufferLine
        , bufferIdx
        , buffer
        , windowWidth
        , windowHeight
        , searchList
        , visualScrollColumn
        )
      val drawMsg = Vector.concat drawMsg
      val drawMsg = DrawMsg.DRAW_TEXT drawMsg
      val drawMsg = [MailboxType.DRAW drawMsg]

      val mode = NORMAL_MODE ""
    in
      NormalModeWith.bufferAndCursorIdx
        ( app
        , buffer
        , bufferIdx
        , mode
        , bufferLine
        , searchList
        , drawMsg
        , bufferModifyTime
        , visualScrollColumn
        )
    end

  fun finishMoveCursorUpDown
    (app: app_type, newCursorLineNumber, buffer, column, lineIdx) =
    let
      val
        { windowWidth
        , windowHeight
        , visualScrollColumn = prevScrollColumn
        , startLine = prevLineNumber
        , searchList
        , bufferModifyTime
        , ...
        } = app

      val buffer = LineGap.goToIdx (lineIdx, buffer)
      val endOfLineIdx = Cursor.viDlr (buffer, lineIdx, 1)
      val endOfLineIdx =
        if endOfLineIdx >= #textLength buffer - 1 then
          Int.max (0, #textLength buffer - 1)
        else
          endOfLineIdx

      val cursorIdx = Int.min (endOfLineIdx, lineIdx + column)

      val buffer = LineGap.goToIdx (cursorIdx, buffer)

      (* create draw message *)
      val visualScrollColumn =
        TextScroll.getScrollColumn
          (buffer, cursorIdx, windowWidth, prevScrollColumn)

      val startLine =
        TextScroll.getStartLine
          ( prevLineNumber
          , newCursorLineNumber
          , windowHeight
          , #lineLength buffer
          )

      val buffer = LineGap.goToLine (startLine, buffer)

      val drawMsg = NormalModeTextBuilder.build
        ( startLine
        , cursorIdx
        , buffer
        , windowWidth
        , windowHeight
        , searchList
        , visualScrollColumn
        )
      val drawMsg = Vector.concat drawMsg
      val drawMsg = DrawMsg.DRAW_TEXT drawMsg
      val drawMsg = [MailboxType.DRAW drawMsg]

      val mode = NORMAL_MODE ""
    in
      NormalModeWith.bufferAndCursorIdx
        ( app
        , buffer
        , cursorIdx
        , mode
        , startLine
        , searchList
        , drawMsg
        , bufferModifyTime
        , visualScrollColumn
        )
    end

  fun moveCursorUp (app: app_type, count) =
    let
      val {cursorIdx, buffer, ...} = app

      val buffer = LineGap.goToIdx (cursorIdx, buffer)
      val startOfLine = Cursor.vi0 (buffer, cursorIdx)
    in
      if Cursor.isCursorAtStartOfLine (buffer, cursorIdx) then
        let
          val cursorLineNumber = LineGap.idxToLineNumber (cursorIdx + 1, buffer)
          val newCursorLineNumber = Int.max (0, cursorLineNumber - count)
          val buffer = LineGap.goToLine (newCursorLineNumber, buffer)
          val lineIdx = LineGap.lineNumberToIdx (newCursorLineNumber, buffer)

          val lineIdx =
            if Cursor.isPrevChrStartOfLine (buffer, lineIdx) then lineIdx
            else lineIdx - 1

          val buffer = LineGap.goToIdx (lineIdx, buffer)
          val lineIdx = Cursor.vi0 (buffer, lineIdx)

          val lineIdx = Int.max (0, lineIdx)
        in
          finishMoveCursorUpDown (app, newCursorLineNumber, buffer, 0, lineIdx)
        end
      else
        let
          val cursorLineNumber = LineGap.idxToLineNumber (cursorIdx, buffer)
          val newCursorLineNumber = Int.max (cursorLineNumber - count, 0)

          val buffer = LineGap.goToLine (newCursorLineNumber, buffer)
          val lineIdx =
            LineGap.lineNumberToIdx (newCursorLineNumber, buffer) + 1

          val column = cursorIdx - startOfLine
          val column = if newCursorLineNumber = 0 then column - 1 else column
        in
          finishMoveCursorUpDown
            (app, newCursorLineNumber, buffer, column, lineIdx)
        end
    end

  fun moveCursorDown (app: app_type, count) =
    let
      val
        { windowWidth
        , windowHeight
        , cursorIdx
        , buffer
        , startLine = prevLineNumber
        , searchList
        , bufferModifyTime
        , visualScrollColumn = prevScrollColumn
        , ...
        } = app

      (* calculate new idx to move to *)
      val buffer = LineGap.goToIdx (cursorIdx, buffer)
      val startOfLine = Cursor.vi0 (buffer, cursorIdx)
      val column = cursorIdx - startOfLine
    in
      if Cursor.isCursorAtStartOfLine (buffer, cursorIdx) then
        let
          val cursorLineNumber = LineGap.idxToLineNumber (cursorIdx + 1, buffer)
          val newCursorLineNumber = cursorLineNumber + count

          val buffer = LineGap.goToLine (newCursorLineNumber, buffer)
          val lineIdx = LineGap.lineNumberToIdx (newCursorLineNumber, buffer)

          val lineIdx =
            if Cursor.isPrevChrStartOfLine (buffer, lineIdx) then lineIdx
            else lineIdx - 1
          val buffer = LineGap.goToIdx (lineIdx, buffer)
          val lineIdx = Cursor.vi0 (buffer, lineIdx)

          val lineIdx =
            if lineIdx >= #textLength buffer - 1 then
              Int.max (0, #textLength buffer - 1)
            else
              lineIdx
        in
          finishMoveCursorUpDown (app, newCursorLineNumber, buffer, 0, lineIdx)
        end
      else
        let
          val cursorLineNumber = LineGap.idxToLineNumber (cursorIdx, buffer)
          val newCursorLineNumber = cursorLineNumber + count

          val buffer = LineGap.goToLine (newCursorLineNumber, buffer)
          val lineIdx = LineGap.lineNumberToIdx (newCursorLineNumber, buffer)
          val buffer = LineGap.goToIdx (lineIdx, buffer)
        in
          if lineIdx >= #textLength buffer - 1 then
            (* we reached last line *)
            let
              val lineIdx = Int.max (#textLength buffer - 1, 0)
              val buffer = LineGap.goToIdx (lineIdx, buffer)

              val lineIdx =
                if Cursor.isOnNewlineAfterChr (buffer, lineIdx) then lineIdx - 1
                else lineIdx

              val startOfLine = Cursor.vi0 (buffer, lineIdx)
            in
              if cursorIdx >= startOfLine then
                (* we are already on last line so don't move *)
                NormalFinish.buildTextAndClear
                  (app, buffer, cursorIdx, searchList, [], bufferModifyTime)
              else
                finishMoveCursorUpDown
                  (app, newCursorLineNumber, buffer, column, startOfLine)
            end
          else
            let
              val lineIdx =
                if lineIdx >= #textLength buffer - 2 then
                  Int.max (0, #textLength buffer - 2)
                else
                  lineIdx

              val buffer = LineGap.goToIdx (lineIdx, buffer)
              val lineIdx =
                if Cursor.isOnNewlineAfterChr (buffer, lineIdx) then lineIdx + 1
                else lineIdx
            in
              finishMoveCursorUpDown
                (app, newCursorLineNumber, buffer, column, lineIdx)
            end
        end
    end

  fun moveToLine (app: app_type, reqLine) =
    let
      val reqLine = reqLine - 1
    in
      if reqLine = 0 then
        moveToStart app
      else
        let
          val
            { windowWidth
            , windowHeight
            , buffer
            , startLine = prevLineNumber
            , searchList
            , bufferModifyTime
            , visualScrollColumn = prevScrollColumn
            , ...
            } = app
          val buffer = LineGap.goToLine (reqLine, buffer)

          (* get idx of first chr after linebreak *)
          val cursorIdx = LineGap.lineNumberToIdx (reqLine, buffer)
          val buffer = LineGap.goToIdx (cursorIdx, buffer)

          (* we got the line start idx, but we want to move to the index 
           * after it, where the first character of the line is.
           * Unless the next character is a line break,
           * in which case we want to stay at the current idx. *)
          val cursorIdx =
            if Cursor.isNextChrEndOfLine (buffer, cursorIdx) then cursorIdx
            else cursorIdx + 1
          val buffer = LineGap.goToIdx (cursorIdx, buffer)

          val visualScrollColumn =
            TextScroll.getScrollColumn
              (buffer, cursorIdx, windowWidth, prevScrollColumn)

          val cursorLine = LineGap.idxToLineNumber (cursorIdx, buffer)
          val startLine =
            TextScroll.getStartLine
              (prevLineNumber, cursorLine, windowHeight, #lineLength buffer)

          val buffer = LineGap.goToLine (startLine, buffer)

          val drawMsg = NormalModeTextBuilder.build
            ( startLine
            , cursorIdx
            , buffer
            , windowWidth
            , windowHeight
            , searchList
            , visualScrollColumn
            )
          val drawMsg = Vector.concat drawMsg
          val drawMsg = DrawMsg.DRAW_TEXT drawMsg
          val drawMsg = [MailboxType.DRAW drawMsg]

          val mode = NORMAL_MODE ""
        in
          NormalModeWith.bufferAndCursorIdx
            ( app
            , buffer
            , cursorIdx
            , mode
            , startLine
            , searchList
            , drawMsg
            , bufferModifyTime
            , visualScrollColumn
            )
        end
    end

  fun moveToMatchingPair (app: app_type) =
    let
      val
        { buffer
        , cursorIdx
        , windowWidth
        , windowHeight
        , startLine = prevLineNumber
        , searchList
        , bufferModifyTime
        , visualScrollColumn = prevScrollColumn
        , ...
        } = app

      (* move LineGap and buffer to start of line *)
      val buffer = LineGap.goToIdx (cursorIdx, buffer)
      val cursorIdx = Cursor.matchPair (buffer, cursorIdx)
    in
      let
        val buffer = LineGap.goToIdx (cursorIdx, buffer)
        val visualScrollColumn =
          TextScroll.getScrollColumn
            (buffer, cursorIdx, windowWidth, prevScrollColumn)

        val cursorLine = LineGap.idxToLineNumber (cursorIdx, buffer)
        val startLine =
          TextScroll.getStartLine
            (prevLineNumber, cursorLine, windowHeight, #lineLength buffer)

        val buffer = LineGap.goToLine (startLine, buffer)

        val drawMsg = NormalModeTextBuilder.build
          ( startLine
          , cursorIdx
          , buffer
          , windowWidth
          , windowHeight
          , searchList
          , visualScrollColumn
          )
        val drawMsg = Vector.concat drawMsg
        val drawMsg = DrawMsg.DRAW_TEXT drawMsg
        val drawMsg = [MailboxType.DRAW drawMsg]
      in
        NormalModeWith.bufferAndCursorIdx
          ( app
          , buffer
          , cursorIdx
          , NORMAL_MODE ""
          , startLine
          , searchList
          , drawMsg
          , bufferModifyTime
          , visualScrollColumn
          )
      end
    end

  fun firstNonSpaceChr (app: app_type) =
    let
      val
        { buffer
        , cursorIdx
        , windowWidth
        , windowHeight
        , startLine
        , searchList
        , bufferModifyTime
        , ...
        } = app

      (* move LineGap and buffer to start of line *)
      val buffer = LineGap.goToIdx (cursorIdx, buffer)
      val cursorIdx = Cursor.vi0 (buffer, cursorIdx)

      (* move cursorIdx to first character on line *)
      val buffer = LineGap.goToIdx (cursorIdx, buffer)
      val cursorIdx = Cursor.firstNonSpaceChr (buffer, cursorIdx)
    in
      NormalFinish.buildTextAndClear
        (app, buffer, cursorIdx, searchList, [], bufferModifyTime)
    end

  fun helpMoveToChr (app: app_type, buffer, cursorIdx, count, fMove, chr) =
    if count = 0 then
      NormalFinish.buildTextAndClear
        (app, buffer, cursorIdx, #searchList app, [], #bufferModifyTime app)
    else
      let
        (* move LineGap to cursorIdx, which is necessary for finding newCursorIdx *)
        val buffer = LineGap.goToIdx (cursorIdx, buffer)
        val newCursorIdx = fMove (buffer, cursorIdx, chr)
        val newCount = if cursorIdx = newCursorIdx then 0 else count - 1
      in
        helpMoveToChr (app, buffer, newCursorIdx, newCount, fMove, chr)
      end

  fun moveToChr (app: app_type, count, fMove, chr) =
    let val {cursorIdx, buffer, ...} = app
    in helpMoveToChr (app, buffer, cursorIdx, count, fMove, chr)
    end

  fun moveToNextMatch (app: app_type, count) =
    let
      val
        { cursorIdx
        , searchList
        , buffer
        , bufferModifyTime
        , visualScrollColumn
        , ...
        } = app
      val newCursorIdx =
        PersistentVector.nextMatch (cursorIdx, searchList, count)
    in
      if newCursorIdx = ~1 then
        NormalFinish.clearMode app
      else
        NormalFinish.buildTextAndClear
          (app, buffer, newCursorIdx, searchList, [], bufferModifyTime)
    end

  fun moveToPrevMatch (app: app_type, count) =
    let
      val {cursorIdx, searchList, buffer, bufferModifyTime, ...} = app
      val newCursorIdx =
        PersistentVector.prevMatch (cursorIdx, searchList, count)
    in
      if newCursorIdx = ~1 then
        NormalFinish.clearMode app
      else
        NormalFinish.buildTextAndClear
          (app, buffer, newCursorIdx, searchList, [], bufferModifyTime)
    end

  fun toNextChr (app: app_type, count, chr) =
    let
      val {cursorIdx, buffer, searchList, bufferModifyTime, ...} = app
      val buffer = LineGap.goToIdx (cursorIdx, buffer)
      val newCursorIdx =
        Cursor.toNextChr (buffer, cursorIdx, {findChr = chr, count = count})
    in
      if newCursorIdx = ~1 then
        NormalFinish.clearMode app
      else
        NormalFinish.buildTextAndClear
          (app, buffer, newCursorIdx, searchList, [], bufferModifyTime)
    end

  fun tillNextChr (app: app_type, count, chr) =
    let
      val {cursorIdx, buffer, searchList, bufferModifyTime, ...} = app
      val buffer = LineGap.goToIdx (cursorIdx, buffer)
      val newCursorIdx =
        Cursor.toNextChr (buffer, cursorIdx, {findChr = chr, count = count})
    in
      if newCursorIdx = ~1 then
        NormalFinish.clearMode app
      else
        NormalFinish.buildTextAndClear
          (app, buffer, newCursorIdx - 1, searchList, [], bufferModifyTime)
    end
end