2 files changed, 18 insertions, 9 deletions
diff --git a/book/src/main/scala/book/BookData.scala b/book/src/main/scala/book/BookData.scala
index af8d2cb..ca6efff 100644
--- a/book/src/main/scala/book/BookData.scala
+++ b/book/src/main/scala/book/BookData.scala
@@ -1,9 +1,14 @@
 package book
 
+import java.io.File
+
 import acyclic.file
 import scalatags.Text.TypedTag
 import scalatags.Text.all._
 object BookData {
+  val cloneRoot = System.getProperty("clone.root") + "/"
+
+
   lazy val javaAPIs = {
     import java.io.File
     def recursiveListFiles(f: File): Array[File] = {
@@ -12,12 +17,12 @@ object BookData {
     }
 
     val roots = Seq(
-      "examples/scala-js/javalanglib/src/main/scala",
-      "examples/scala-js/javalib/src/main/scala"
+      "scala-js/javalanglib/src/main/scala",
+      "scala-js/javalib/src/main/scala"
     )
     for{
       root <- roots
-      file <- recursiveListFiles(new File(root))
+      file <- recursiveListFiles(new File(cloneRoot + root))
       if file != null
       if file.isFile
     } yield{
@@ -42,4 +47,13 @@ object BookData {
   def more = div(cls:="pure-u-1 pure-u-md-13-24")
   def less = div(cls:="pure-u-1 pure-u-md-11-24")
   def half = div(cls:="pure-u-1 pure-u-md-1-2")
+
+
+  val hl = new Highlighter(
+    Seq(
+      s"$cloneRoot/scala-js" -> "https://github.com/scala-js/scala-js",
+      s"$cloneRoot/workbench-example-app" -> "https://github.com/lihaoyi/workbench-example-app",
+      "" -> "https://github.com/lihaoyi/hands-on-scala-js"
+    )
+  )
 }
diff --git a/book/src/main/scala/book/Utils.scala b/book/src/main/scala/book/Utils.scala
index b59fb59..2c861d0 100644
--- a/book/src/main/scala/book/Utils.scala
+++ b/book/src/main/scala/book/Utils.scala
@@ -123,7 +123,7 @@ object lnk{
   }
 }
 
-object hl{
+class Highlighter(mappings: Seq[(String, String)]){
   def highlight(snippet: Seq[String], lang: String) = {
     val string = snippet.mkString
     val lines = string.split("\n", -1)
@@ -147,11 +147,6 @@ object hl{
   def diff(code: String*) = highlight(code, "diff")
   def html(code: String*) = highlight(code, "xml")
 
-  val mappings = Seq(
-    "examples/scala-js" -> "https://github.com/scala-js/scala-js",
-    "examples/workbench-example-app" -> "https://github.com/lihaoyi/workbench-example-app",
-    "" -> "https://github.com/lihaoyi/hands-on-scala-js"
-  )
   def ref(filepath: String, start: String = "", end: String = "\n") = {
 
     val lang = filepath.split('.').last match {