
ammonite.repl.frontend.ReplAPI.scala Maven / Gradle / Ivy
package ammonite.repl.frontend
import java.io.File
import ammonite.ops._
import ammonite.repl._
import ammonite.repl.interp.Frame
import org.apache.ivy.plugins.resolver.RepositoryResolver
import pprint.{PPrinter, PPrint, Config}
import scala.collection.mutable
import scala.reflect.runtime.universe._
import acyclic.file
import scala.util.control.ControlThrowable
class ReplAPIHolder {
var repl0: FullReplAPI = null
implicit lazy val repl = repl0
}
/**
* Thrown to exit the REPL cleanly
*/
case class ReplExit(value: Any) extends ControlThrowable
trait ReplAPI {
/**
* Exit the Ammonite REPL. You can also use Ctrl-D to exit
*/
def exit = throw ReplExit(())
/**
* Exit the Ammonite REPL. You can also use Ctrl-D to exit
*/
def exit(value: Any) = throw ReplExit(value)
/**
* Read/writable prompt for the shell. Use this to change the
* REPL prompt at any time!
*/
val prompt: Ref[String]
/**
* The front-end REPL used to take user input. Modifiable!
*/
val frontEnd: Ref[FrontEnd]
/**
* Display help text if you don't know how to use the REPL
*/
def help: String
/**
* The last exception that was thrown in the REPL; `null` if nothing has
* yet been thrown. Useful if you want additional information from the
* thrown exception than the printed stack trace (e.g. many exceptions have
* additional metadata attached) or if you want to show the stack trace
* on an exception that doesn't normally print it (e.g. seeing the stack
* when a Ctrl-C interrupt happened) via `lastException.printStackTrace`.
*/
def lastException: Throwable
/**
* History of commands that have been entered into the shell, including
* previous sessions
*/
def fullHistory: History
/**
* History of commands that have been entered into the shell during the
* current session
*/
def history: History
/**
* Get the `Type` object of [[T]]. Useful for finding
* what its methods are and what you can do with it
*/
def typeOf[T: WeakTypeTag]: Type
/**
* Get the `Type` object representing the type of `t`. Useful
* for finding what its methods are and what you can do with it
*
*/
def typeOf[T: WeakTypeTag](t: => T): Type
/**
* Tools related to loading external scripts and code into the REPL
*/
def load: Load
/**
* resolvers to use when loading jars
*/
def resolvers: Ref[List[Resolver]]
/**
* The colors that will be used to render the Ammonite REPL in the terminal
*/
val colors: Ref[Colors]
/**
* Throw away the current scala.tools.nsc.Global and get a new one
*/
def newCompiler(): Unit
/**
* Access the compiler to do crazy things if you really want to!
*/
def compiler: scala.tools.nsc.Global
/**
* Show all the imports that are used to execute commands going forward
*/
def imports: String
/**
* Controls how things are pretty-printed in the REPL. Feel free
* to shadow this with your own definition to change how things look
*/
implicit val pprintConfig: Ref[pprint.Config]
implicit def derefPPrint(implicit t: Ref[pprint.Config]): pprint.Config = t()
/**
* Current width of the terminal
*/
def width: Int
/**
* Current height of the terminal
*/
def height: Int
def replArgs: Vector[ammonite.repl.Bind[_]]
/**
* Lets you configure the pretty-printing of a value. By default, it simply
* disables truncation and prints the entire thing, but you can set other
* parameters as well if you want.
*/
def show[T: PPrint](implicit cfg: Config): T => Unit
def show[T: PPrint](t: T,
width: Integer = 0,
height: Integer = null,
indent: Integer = null,
colors: pprint.Colors = null)
(implicit cfg: Config = Config.Defaults.PPrintConfig): Unit
/**
* Functions that can be used to manipulate the current REPL session:
* check-pointing progress, reverting to earlier checkpoints, or deleting
* checkpoints by name.
*
* Frames get pushed on a stack; by default, a saved frame is
* accessible simply by calling `load`. If you provide a name
* when `save`ing a checkpoint, it can later be `load`ed directly
* by providing the same name to `load`
*
* Un-named checkpoints are garbage collected, together with their
* classloader and associated data, when they are no longer accessible
* due to `restore`. Named checkpoints are kept forever; call `delete`
* on them if you really want them to go away.
*/
def sess: Session
}
trait Session{
/**
* The current stack of frames
*/
def frames: List[Frame]
/**
* Checkpoints your current work, placing all future work into its own
* frames. If a name is provided, it can be used to quickly recover
* that checkpoint later.
*/
def save(name: String = ""): Unit
/**
* Discards the last frames, effectively reverting your session to
* the last `save`-ed checkpoint. If a name is provided, it instead reverts
* your session to the checkpoint with that name.
*/
def load(name: String = ""): SessionChanged
/**
* Resets you to the last save point. If you pass in `num`, it resets
* you to that many savepoints since the last one.
*/
def pop(num: Int = 1): SessionChanged
/**
* Deletes a named checkpoint, allowing it to be garbage collected if it
* is no longer accessible.
*/
def delete(name: String): Unit
}
trait LoadJar {
/**
* Load a `.jar` file or directory into your JVM classpath
*/
def cp(jar: Path): Unit
/**
* Load a library from its maven/ivy coordinates
*/
def ivy(coordinates: (String, String, String), verbose: Boolean = true): Unit
}
trait Load extends (String => Unit) with LoadJar{
/**
* Loads a command into the REPL and
* evaluates them one after another
*/
def apply(line: String): Unit
/**
* Loads and executes the scriptfile on the specified path.
* Compilation units separated by `@\n` are evaluated sequentially.
* If an error happens it prints an error message to the console.
*/
def exec(path: Path): Unit
def module(path: Path): Unit
def plugin: LoadJar
}
// End of ReplAPI
/**
* Things that are part of the ReplAPI that aren't really "public"
*/
abstract class FullReplAPI extends ReplAPI{
val Internal: Internal
trait Internal{
def combinePrints(iters: Iterator[String]*): Iterator[String]
/**
* Kind of an odd signature, splitting out [[T]] and [[V]]. This is
* seemingly useless but necessary because when you add both [[pprint.TPrint]]
* and [[PPrint]] context bounds to the same type, Scala's type inference
* gets confused and does the wrong thing
*/
def print[T: pprint.TPrint: WeakTypeTag, V: PPrint]
(value: => T, value2: => V, ident: String, custom: Option[String])
(implicit cfg: Config): Iterator[String]
def printDef(definitionLabel: String, ident: String): Iterator[String]
def printImport(imported: String): Iterator[String]
}
def typeOf[T: WeakTypeTag] = scala.reflect.runtime.universe.weakTypeOf[T]
def typeOf[T: WeakTypeTag](t: => T) = scala.reflect.runtime.universe.weakTypeOf[T]
}
object ReplAPI{
def initReplBridge(holder: Class[ReplAPIHolder], api: ReplAPI) = {
val method = holder
.getDeclaredMethods
.find(_.getName == "repl0_$eq")
.get
method.invoke(null, api)
}
}
trait DefaultReplAPI extends FullReplAPI {
def help =
"""Welcome to the Ammonite Scala REPL! Enter a Scala expression and it will be evaluated.
|All your standard Bash hotkeys should work for navigating around or editing the line
|being entered, as well as some GUI hotkeys like alt-shift-left/right to select words
|to replace. Hit to autocomplete possible names.
|
|For a list of REPL built-ins and configuration, use `repl.`. For a more detailed
|description of how to use the REPL, check out https://lihaoyi.github.io/Ammonite
""".stripMargin.trim
object Internal extends Internal{
def combinePrints(iters: Iterator[String]*) = {
iters.toIterator
.filter(_.nonEmpty)
.flatMap(Iterator("\n") ++ _)
.drop(1)
}
def print[T: pprint.TPrint: WeakTypeTag, V: PPrint](value: => T,
value2: => V,
ident: String,
custom: Option[String])
(implicit cfg: pprint.Config) = {
if (typeOf[T] =:= typeOf[Unit]) Iterator()
else {
val implicitPPrint = implicitly[PPrint[V]]
val rhs = custom match {
case None => implicitPPrint.render(value2, cfg)
case Some(s) => Iterator(cfg.colors.literalColor, s, cfg.colors.endColor)
}
Iterator(
colors().ident(), ident, colors().reset(), ": ",
implicitly[pprint.TPrint[T]].render(cfg), " = "
) ++ rhs
}
}
def printDef(definitionLabel: String, ident: String) = {
Iterator(
"defined ", colors().`type`(), definitionLabel, " ",
colors().ident(), ident, colors().reset()
)
}
def printImport(imported: String) = {
Iterator(colors().`type`(), "import ", colors().ident(), imported, colors().reset())
}
}
}
object ReplBridge extends ammonite.repl.frontend.ReplAPIHolder{}
case class SessionChanged(removedImports: Set[scala.Symbol],
addedImports: Set[scala.Symbol],
removedJars: Set[java.net.URL],
addedJars: Set[java.net.URL])
object SessionChanged{
implicit val pprinter: PPrinter[SessionChanged] = PPrinter[SessionChanged]{
(data, config) =>
val output = mutable.Buffer.empty[String]
def printDelta[T: PPrint](name: String, d: Iterable[T]) = {
if (d.nonEmpty){
Iterator("\n", name, ": ") ++ pprint.tokenize(d)(implicitly, config)
}else Iterator()
}
val res = Iterator(
printDelta("Removed Imports", data.removedImports),
printDelta("Added Imports", data.addedImports),
printDelta("Removed Jars", data.removedJars),
printDelta("Added Jars", data.addedJars)
)
res.flatten
}
def delta(oldFrame: Frame, newFrame: Frame): SessionChanged = {
def frameSymbols(f: Frame) = f.imports.map(_.toName).map(Symbol(_)).toSet
new SessionChanged(
frameSymbols(oldFrame) -- frameSymbols(newFrame),
frameSymbols(newFrame) -- frameSymbols(oldFrame),
oldFrame.classloader.allJars.toSet -- newFrame.classloader.allJars.toSet,
newFrame.classloader.allJars.toSet -- oldFrame.classloader.allJars.toSet
)
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy