Customizing banner text and prompt

Author: mandar2812

build.sbt

@@ -1,4 +1,6 @@
 import sbt._
+import NativePackagerHelper._
+import java.io.File
 
 maintainer := "Mandar Chandorkar <[email protected]>"
 
@@ -9,6 +11,8 @@ packageDescription := "DynaML is a scala library/repl for implementing and worki
   "which can be extended easily to implement advanced models for small and large scale applications.\n\n"+
   "But the library can also be used as an educational/research tool for data analysis."
 
+val dataDirectory = settingKey[File]("The directory holding the data files for running example scripts")
+
 lazy val commonSettings = Seq(
   name := "DynaML",
   organization := "io.github.mandar2812",
@@ -40,7 +44,9 @@ lazy val commonSettings = Seq(
     "org.jzy3d" % "jzy3d-api" % "0.9.1" % "compile",
     "com.lihaoyi" % "ammonite-repl_2.11.7" % "0.5.8"
   ),
-  initialCommands in console := """io.github.mandar2812.dynaml.DynaML.run();"""
+  dataDirectory := new File("data/"),
+  initialCommands in console := """io.github.mandar2812.dynaml.DynaML.run(banner="""" +
+    target.value.getPath + """/universal/stage/conf/banner.txt");"""
 )
 
 lazy val DynaML = (project in file(".")).enablePlugins(JavaAppPackaging, BuildInfoPlugin)
@@ -49,12 +55,15 @@ lazy val DynaML = (project in file(".")).enablePlugins(JavaAppPackaging, BuildIn
     buildInfoKeys := Seq[BuildInfoKey](name, version, scalaVersion, sbtVersion),
     buildInfoPackage := "io.github.mandar2812.dynaml.repl",
     buildInfoUsePackageAsPath := true,
-    mappings in Universal += {
+    mappings in Universal ++= Seq({
       // we are using the reference.conf as default application.conf
       // the user can override settings here
       val init = (resourceDirectory in Compile).value / "DynaMLInit.scala"
       init -> "conf/DynaMLInit.scala"
-    },
+    }, {
+      val banner = (resourceDirectory in Compile).value / "dynamlBanner.txt"
+      banner -> "conf/banner.txt"
+    }),
     javaOptions in Universal ++= Seq(
       // -J params will be added as jvm parameters
       "-J-Xmx2048m",

src/main/resources/dynamlBanner.txt

@@ -0,0 +1,11 @@
+      ___           ___           ___           ___           ___           ___
+     /\  \         |\__\         /\__\         /\  \         /\__\         /\__\
+    /::\  \        |:|  |       /::|  |       /::\  \       /::|  |       /:/  /
+   /:/\:\  \       |:|  |      /:|:|  |      /:/\:\  \     /:|:|  |      /:/  /
+  /:/  \:\__\      |:|__|__   /:/|:|  |__   /::\~\:\  \   /:/|:|__|__   /:/  /
+ /:/__/ \:|__|     /::::\__\ /:/ |:| /\__\ /:/\:\ \:\__\ /:/ |::::\__\ /:/__/
+ \:\  \ /:/  /    /:/~~/~    \/__|:|/:/  / \/__\:\/:/  / \/__/~~/:/  / \:\  \
+  \:\  /:/  /    /:/  /          |:/:/  /       \::/  /        /:/  /   \:\  \
+   \:\/:/  /     \/__/           |::/  /        /:/  /        /:/  /     \:\  \
+    \::/__/                      /:/  /        /:/  /        /:/  /       \:\__\
+     ~~                          \/__/         \/__/         \/__/         \/__/

src/main/scala/io/github/mandar2812/dynaml/DynaML.scala

@@ -129,10 +129,12 @@ object DynaML {
 
     val storage = Storage(defaultAmmoniteHome, None)
     val repl = new DynaMLRepl(
-      System.in, System.out, System.err,
+      input = System.in, output = System.out, error = System.err,
       storage = Ref(storage),
       predef = "",
-      replArgs
+      replArgs = replArgs,
+      promptStr = "DynaML>",
+      bannerText = "conf/banner.txt"
     )
 
     repl.run()
@@ -148,14 +150,17 @@ object DynaML {
           predefFile: Option[Path] = None,
           file: Option[Path] = None,
           args: Seq[String] = Vector.empty,
-          kwargs: Map[String, String] = Map.empty) = {
+          kwargs: Map[String, String] = Map.empty,
+          prompt: String = "DynaML>",
+          banner: String = "conf/banner.txt") = {
 
     Timer("Repl.run Start")
     def storage = Storage(ammoniteHome, predefFile)
     lazy val repl = new DynaMLRepl(
       System.in, System.out, System.err,
       storage = Ref(storage),
-      predef = predef
+      predef = predef, promptStr = prompt,
+      bannerText = banner
     )
     (file, code) match{
       case (None, None) => println("Loading..."); repl.run()

src/main/scala/io/github/mandar2812/dynaml/repl/DynaMLRepl.scala

@@ -4,6 +4,8 @@ import java.io.{InputStream, OutputStream}
 
 import ammonite.repl.{Bind, Ref, Repl, Storage}
 
+import scala.io.Source
+
 /**
   * Created by mandar on 1/6/16.
   */
@@ -12,20 +14,17 @@ class DynaMLRepl(input: InputStream,
                  error: OutputStream,
                  storage: Ref[Storage],
                  predef: String = "",
-                 replArgs: Seq[Bind[_]] = Nil) extends
+                 replArgs: Seq[Bind[_]] = Nil,
+                 promptStr: String,
+                 bannerText: String) extends
   Repl(input, output, error, storage, predef, replArgs) {
 
-  override val prompt = Ref("DynaML>")
+  override val prompt = Ref(promptStr)
+
+  val banner = Source.fromFile(bannerText).getLines.mkString("\n")
 
   override def printBanner(): Unit = {
-    printStream.println("    ___       ___       ___       ___       ___       ___ "+
-      "  \n   /\\  \\     /\\__\\     /\\__\\     /\\  \\     /\\__\\ "+
-      "    /\\__\\  \n  /::\\  \\   |::L__L   /:| _|_   /::\\  \\   /::L_L_ "+
-      "  /:/  /  \n /:/\\:\\__\\  |:::\\__\\ /::|/\\__\\ /::\\:\\__\\ "+
-      "/:/L:\\__\\ /:/__/   \n \\:\\/:/  /  /:;;/__/ \\/|::/  / \\/\\::/ "+
-      " / \\/_/:/  / \\:\\  \\   \n  \\::/  /   \\/__/      |:/  /    /:/ "+
-      " /    /:/  /   \\:\\__\\  \n   \\/__/               \\/__/    "+
-      " \\/__/     \\/__/     \\/__/  ")
+    printStream.println(banner)
     val version = BuildInfo.version
     printStream.println("\nWelcome to DynaML "+version+
       "\nInteractive Scala shell for Machine Learning Research")