Scala Ammonite Script to move and symlink directories

move-and-link

#!/usr/bin/env amm

import java.nio.file.Path
import java.nio.file.Paths
import $ivy.`org.jline:jline:3.19.0`
import org.jline.terminal._
import org.jline.reader._
import scala.sys.process._

import $ivy.`com.github.pathikrit::better-files:3.9.1`
import better.files._
import File._
import java.io.{File => JFile}

import scala.language.postfixOps

@main
def main(toMoveF: JFile, targetDirF: JFile = new JFile("/data/ran-out-of-space")) {
  val terminal = TerminalBuilder.terminal()
  val lineReader = LineReaderBuilder.builder()
                        .terminal(terminal)
                        .build()

  val reader = LineReaderBuilder.builder().build()

  val toMove = toMoveF.toPath
  val targetDir = targetDirF.toPath
  val destF = new JFile(targetDirF, toMove.toString)
  val dest = destF.toPath

  var cmds = Seq[Seq[String]]()
  cmds = cmds :+ Seq("set", "-e")
  cmds = cmds :+ Seq("set", "-x")
  cmds = cmds :+ Seq("mkdir", "-p", targetDir.toString)
  cmds = cmds :+ Seq("cd", toMove.toAbsolutePath.getParent.toString)
  cmds = cmds :+ Seq("cp", "-a", toMove.toString, dest.toString)
  cmds = cmds :+ Seq("rm", "-rf", dest.toString)
  cmds = cmds :+ Seq("mv", toMove.toString, dest.toString)
  cmds = cmds :+ Seq("ln", "-s", dest.toString, toMove.toString)
  
  def quote(s: String): String = s"'${s.replaceAll("'", "\\'")}'"
  
  println("About to execute the following commands:")
  val script = cmds.map(_.map(quote).mkString(" ")).mkString("\n")
  println(script)
  if (confirm("Proceed?")) {
    File.usingTemporaryFile() { tempFile =>
      tempFile.overwrite(script)
      Seq("sh") #< tempFile.toJava !
    }
  }

  def confirm(question: String): Boolean = {
    ask(s"$question [Y/n]").toLowerCase match {
      case "" | "y" | "yes" => true
      case "n" | "no" => false
    }
  }

  def ask(what: String): String = {
    reader.readLine(s"$what ")
  }

}