aboutsummaryrefslogtreecommitdiff
path: root/src/dotty/tools/dotc/repl/AmmoniteReader.scala
blob: 0c4dd80ee078f57595f0627dbf2dd790a0a1ce45 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package dotty.tools
package dotc
package repl

import core.Contexts._
import ammonite.terminal._
import LazyList._
import Ansi.Color
import filters._
import BasicFilters._
import GUILikeFilters._
import util.SourceFile

class AmmoniteReader extends InteractiveReader {
  val interactive = true

  val reader = new java.io.InputStreamReader(System.in)
  val writer = new java.io.OutputStreamWriter(System.out)
  val cutPasteFilter = ReadlineFilters.CutPasteFilter()
  var history = List.empty[String]
  val selectionFilter = GUILikeFilters.SelectionFilter(indent = 2)
  val multilineFilter: Filter = Filter("multilineFilter") {
    case TermState(lb ~: rest, b, c, _)
      if (lb == 10 || lb == 13) => // Enter

      BasicFilters.injectNewLine(b, c, rest)
  }
  def readLine(prompt: String)(implicit ctx: Context): String = {
    val historyFilter = new HistoryFilter(
      () => history.toVector,
      Console.BLUE,
      AnsiNav.resetForegroundColor
    )

    val allFilters = Filter.merge(
      UndoFilter(),
      historyFilter,
      selectionFilter,
      GUILikeFilters.altFilter,
      GUILikeFilters.fnFilter,
      ReadlineFilters.navFilter,
      //autocompleteFilter,
      cutPasteFilter,
      //multilineFilter,
      BasicFilters.all
    )

    Terminal.readLine(
      Console.BLUE + prompt + Console.RESET,
      reader,
      writer,
      allFilters,
      displayTransform = (buffer, cursor) => {
        val ansiBuffer = Ansi.Str.parse(SyntaxHighlighting(buffer))
        //val ansiBuffer = Ansi.Str.parse(SyntaxHighlighting(new SourceFile("<console>", buffer)))
        val (newBuffer, cursorOffset) = SelectionFilter.mangleBuffer(
          selectionFilter, ansiBuffer, cursor, Ansi.Reversed.On
        )
        val newNewBuffer = HistoryFilter.mangleBuffer(
          historyFilter, newBuffer, cursor,
          Ansi.Color.Green
        )

        (newNewBuffer, cursorOffset)
      }
    ) match {
      case Some(s) => history = s :: history; s
      case None => ":q"
    }
  }
}