sbt/main/State.scala

88 lines
2.8 KiB
Scala

/* sbt -- Simple Build Tool
* Copyright 2008, 2009, 2010 Mark Harrah
*/
package sbt
import java.io.File
import CommandSupport.FailureWall
final case class State(
configuration: xsbti.AppConfiguration,
processors: Seq[Command],
exitHooks: Set[ExitHook],
onFailure: Option[String],
commands: Seq[String],
attributes: AttributeMap,
next: Next.Value
) extends Identity {
lazy val combinedParser = Command.combine(processors)(this)
}
trait Identity {
override final def hashCode = super.hashCode
override final def equals(a: Any) = super.equals(a)
override final def toString = super.toString
}
object Next extends Enumeration {
val Reload, Fail, Done, Continue = Value
}
trait StateOps {
def process(f: (String, State) => State): State
def ::: (commands: Seq[String]): State
def :: (command: String): State
def continue: State
def reboot(full: Boolean): State
def reload: State
def exit(ok: Boolean): State
def fail: State
def ++ (newCommands: Seq[Command]): State
def + (newCommand: Command): State
def get[T](key: AttributeKey[T]): Option[T]
def put[T](key: AttributeKey[T], value: T): State
def baseDir: File
def runExitHooks(): State
}
object State
{
implicit def stateOps(s: State): StateOps = new StateOps {
def process(f: (String, State) => State): State =
s.commands match {
case Seq(x, xs @ _*) => f(x, s.copy(commands = xs))
case Seq() => exit(true)
}
s.copy(commands = s.commands.drop(1))
def ::: (newCommands: Seq[String]): State = s.copy(commands = newCommands ++ s.commands)
def :: (command: String): State = (command :: Nil) ::: this
def ++ (newCommands: Seq[Command]): State = s.copy(processors = (s.processors ++ newCommands).distinct)
def + (newCommand: Command): State = this ++ (newCommand :: Nil)
def baseDir: File = s.configuration.baseDirectory
def setNext(n: Next.Value) = s.copy(next = n)
def continue = setNext(Next.Continue)
def reboot(full: Boolean) = throw new xsbti.FullReload(s.commands.toArray, full)
def reload = setNext(Next.Reload)
def exit(ok: Boolean) = setNext(if(ok) Next.Done else Next.Fail)
def get[T](key: AttributeKey[T]) = s.attributes.get(key)
def put[T](key: AttributeKey[T], value: T) = s.copy(attributes = s.attributes.put(key, value))
def fail =
{
val remaining = s.commands.dropWhile(_ != FailureWall)
if(remaining.isEmpty)
applyOnFailure(s, Nil, exit(ok = false))
else
applyOnFailure(s, remaining, s.copy(commands = remaining))
}
private[this] def applyOnFailure(s: State, remaining: Seq[String], noHandler: => State): State =
s.onFailure match
{
case Some(c) => s.copy(commands = c +: remaining, onFailure = None)
case None => noHandler
}
def runExitHooks(): State = {
ExitHooks.runExitHooks(s.exitHooks.toSeq)
s.copy(exitHooks = Set.empty)
}
}
}