- Added Head task.

- Added verbose parameter to GenericTask in order to provide task scoped output control.

Author: codejitsu

tasks-dsl/src/main/scala/net/codejitsu/tasks/Download.scala

@@ -17,12 +17,13 @@ import net.codejitsu.tasks.dsl._
  * @param user user.
  */
 final case class Download[S <: Stage](hosts: Hosts, url: URL,
-                                      destinationPath: String, usingSudo: Boolean = false,
+                                      destinationPath: Option[String] = None, usingSudo: Boolean = false,
                                       usingPar: Boolean = false,
                                       exec: String = Download.getExec(),
-                                      params: List[String] = List("-P"))(implicit user: User, stage: S, rights: S Allow Download[S])
-  extends GenericTask("wget", "download url", hosts, exec, params ++ List(url.toString) ++ List(destinationPath),
-    usingSudo, usingPar, taskRepr = s"download url '${url.toString}'") with UsingSudo[Download[S]] with UsingParallelExecution[Download[S]] {
+                                      params: List[String] = List("-P"),
+                                      verbose: VerbosityLevel = NoOutput)(implicit user: User, stage: S, rights: S Allow Download[S])
+  extends GenericTask("wget", "download url", hosts, exec, params ++ List(url.toString) ++ Download.getDestination(destinationPath),
+    usingSudo, usingPar, taskRepr = s"download url '${url.toString}'", verbose = Option(verbose)) with UsingSudo[Download[S]] with UsingParallelExecution[Download[S]] {
 
   override def sudo: Download[S] = copy[S](usingSudo = true)
   override def par: Download[S] = copy[S](usingPar = true)
@@ -34,4 +35,7 @@ object Download {
     case MacOS => "/usr/local/bin/wget"
     case _ => throw new IllegalArgumentException("Not supported OS")
   }
+
+  def getDestination(destinationPath: Option[String]): List[String] =
+    destinationPath.fold(List.empty[String])(d => List(d))
 }

tasks-dsl/src/main/scala/net/codejitsu/tasks/GenericTask.scala

@@ -8,7 +8,7 @@ import net.codejitsu.tasks.dsl.Tasks._
 abstract class GenericTask(name: String, desc: String, hosts: Hosts, exec: String,
                            params: List[String] = Nil, usingSudo: Boolean = false,
                            usingPar: Boolean = false, cmd: Command = Start, taskRepr: String = "",
-                           pipeCmd: Seq[String] = Seq[String]())(implicit user: User) extends TaskM[Boolean] {
+                           pipeCmd: Seq[String] = Seq[String](), verbose: Option[VerbosityLevel] = None)(implicit user: User) extends TaskM[Boolean] {
   override def description: String = desc
 
   protected val procs: Processes = name on hosts ~> {
@@ -28,15 +28,15 @@ abstract class GenericTask(name: String, desc: String, hosts: Hosts, exec: Strin
     procs ! cmd
   }
 
-  override def run(verbose: VerbosityLevel = NoOutput, input: Option[TaskResult[_]] = None): TaskResult[Boolean] =
+  override def run(verbose: VerbosityLevel, input: Option[TaskResult[_]] = None): TaskResult[Boolean] =
     LoggedRun(
-      verbose,
+      this.verbose.fold(verbose)(v => v),
       usingSudo,
       usingPar,
       hosts,
       taskRepr,
       task,
       input
-    )(verbose)
+    )(this.verbose.fold(verbose)(v => v))
 }
 

tasks-dsl/src/main/scala/net/codejitsu/tasks/Grep.scala

@@ -19,11 +19,12 @@ final case class Grep[S <: Stage](hosts: Hosts,
                                  usingPar: Boolean = false,
                                  exec: String = Grep.getExec(),
                                  params: List[String] = Nil,
-                                 pattern: Option[String] = None)(implicit user: User, stage: S, rights: S Allow Grep[S])
+                                 pattern: Option[String] = None,
+                                 verbose: VerbosityLevel = NoOutput)(implicit user: User, stage: S, rights: S Allow Grep[S])
   extends GenericTask("grep", "searches any given input files, selecting lines that match one or more patterns", hosts, exec,
     Grep.makeCommandLine(params, pattern, target),
-    usingSudo, usingPar, taskRepr = s"grep file '${target}' for patterns",
-    pipeCmd = Grep.pipeCmd(pattern)) with UsingSudo[Grep[S]] with UsingParallelExecution[Grep[S]] {
+    usingSudo, usingPar, taskRepr = s"grep file ${Grep.getFileDescription(target)} for patterns",
+    pipeCmd = Grep.pipeCmd(pattern), verbose = Option(verbose)) with UsingSudo[Grep[S]] with UsingParallelExecution[Grep[S]] {
 
   override def sudo: Grep[S] = copy[S](usingSudo = true)
   override def par: Grep[S] = copy[S](usingPar = true)
@@ -44,4 +45,7 @@ object Grep {
   }
 
   def pipeCmd(pattern: Option[String]): Seq[String] = Seq(getExec(), "-o") ++ pattern.fold(Seq[String]())(p => Seq[String](p))
+
+  def getFileDescription(target: Option[String]): String =
+    target.fold("")(f => s"'$f'")
 }

tasks-dsl/src/main/scala/net/codejitsu/tasks/Head.scala

@@ -0,0 +1,39 @@
+// Copyright (C) 2015, codejitsu.
+
+package net.codejitsu.tasks
+
+import net.codejitsu.tasks.dsl.Allow
+import net.codejitsu.tasks.dsl.Hosts
+import net.codejitsu.tasks.dsl.Stage
+import net.codejitsu.tasks.dsl.User
+import net.codejitsu.tasks.dsl.UsingParallelExecution
+import net.codejitsu.tasks.dsl.UsingSudo
+
+/**
+ * Head task.
+ *
+ * @param hosts target hosts
+ * @param target target file
+ * @param usingSudo true, if task have to be started with sudo
+ * @param usingPar true, if parallel execution required.
+ * @param user user
+ */
+final case class Head[S <: Stage](hosts: Hosts,
+                                 target: Option[String] = None,
+                                 usingSudo: Boolean = false,
+                                 usingPar: Boolean = false,
+                                 exec: String = "/usr/bin/head")(implicit user: User, stage: S, rights: S Allow Head[S])
+  extends GenericTask("head", "the first part of file", hosts, exec, Nil,
+    usingSudo, usingPar, taskRepr = s"display file ${Head.getFileDescription(target)}",
+    pipeCmd = Head.pipeCmd()) with UsingSudo[Head[S]] with UsingParallelExecution[Head[S]] {
+
+  override def sudo: Head[S] = copy[S](usingSudo = true)
+  override def par: Head[S] = copy[S](usingPar = true)
+}
+
+object Head {
+  def getFileDescription(target: Option[String]): String =
+    target.fold("")(f => s"'$f'")
+
+  def pipeCmd(): Seq[String] = Seq("/usr/bin/head")
+}

tasks-dsl/src/main/scala/net/codejitsu/tasks/dsl/Allow.scala

@@ -35,4 +35,5 @@ object Allow {
   implicit val devAllowTail: Dev Allow Tail[Dev] = new Allow[Dev, Tail[Dev]]
   implicit val devAllowClear: Dev Allow Clear[Dev] = new Allow[Dev, Clear[Dev]]
   implicit val devAllowGrep: Dev Allow Grep[Dev] = new Allow[Dev, Grep[Dev]]
+  implicit val devAllowHead: Dev Allow Head[Dev] = new Allow[Dev, Head[Dev]]
 }

tasks-dsl/src/test/scala/GrepTest.scala

@@ -72,30 +72,26 @@ class GrepTest extends FlatSpec with Matchers {
     taskResult.err should be (empty)
     taskResult.out.filter(_.startsWith("<title")).size should be (0)
   }
-/*
+
   it should "support complex piping" in {
     val getRawFeed =
-      Download(Localhost, new URL("http://stackoverflow.com"), "/feeds/tag?tagnames=scala&sort=newest"/*,
-      params = List("-q", "-O-")*/)
+      Download(Localhost, new URL("http://stackoverflow.com/feeds/tag?tagnames=scala&sort=newest"),
+        params = List("-q", "-O-"))
 
     val getRawTitles = Grep(Localhost, params = List("-o"), pattern = Option("<title type=\"text\">[^<]*"))
 
     val getTitles = Grep(Localhost, params = List("-o"), pattern = Option("[^>]*$"))
 
     val task =
-      getRawFeed //pipeTo
-//      getRawTitles //pipeTo
-//      getTitles
+      getRawFeed pipeTo
+      getRawTitles pipeTo
+      getTitles pipeTo
+      Head(Localhost)
 
     val taskResult = task.run(FullOutput)
 
-    println(taskResult)
-
-    //taskResult.out.foreach(println)
-
     taskResult.res.isSuccess should be (true)
     taskResult.err should be (empty)
     taskResult.out.filter(_.startsWith("<title")).size should be (0)
   }
-  */
 }