diff --git a/src/main/scala/higherkindness/rules_scala/common/worker/WorkTask.scala b/src/main/scala/higherkindness/rules_scala/common/worker/WorkTask.scala index add4a0ee..e6a9f102 100644 --- a/src/main/scala/higherkindness/rules_scala/common/worker/WorkTask.scala +++ b/src/main/scala/higherkindness/rules_scala/common/worker/WorkTask.scala @@ -1,5 +1,6 @@ package higherkindness.rules_scala.common.worker +import com.google.devtools.build.lib.worker.WorkerProtocol import java.io.PrintStream import java.nio.file.Path @@ -10,4 +11,5 @@ case class WorkTask[S]( workDir: Path, verbosity: Int, isCancelled: Function0[Boolean], + inputs: Option[List[WorkerProtocol.Input]], ) diff --git a/src/main/scala/higherkindness/rules_scala/common/worker/WorkerMain.scala b/src/main/scala/higherkindness/rules_scala/common/worker/WorkerMain.scala index 740a0904..88539e16 100644 --- a/src/main/scala/higherkindness/rules_scala/common/worker/WorkerMain.scala +++ b/src/main/scala/higherkindness/rules_scala/common/worker/WorkerMain.scala @@ -8,6 +8,7 @@ import java.nio.file.Path import java.util.concurrent.{CancellationException, ConcurrentHashMap, ForkJoinPool} import scala.annotation.tailrec import scala.concurrent.{ExecutionContext, ExecutionException, Future} +import scala.jdk.CollectionConverters.* import scala.util.{Failure, Success, Using} import java.time.{ZoneId, ZonedDateTime} @@ -164,7 +165,17 @@ abstract class WorkerMain[S](stdin: InputStream = System.in, stdout: PrintStream maybeOutStream = Some(outStream) maybeOut = Some(out) try { - work(WorkTask(ctx, args, out, sandboxDir, verbosity, isCancelled)) + work( + WorkTask( + ctx, + args, + out, + sandboxDir, + verbosity, + isCancelled, + Some(request.getInputsList.asScala.toList), + ), + ) 0 } catch { case e @ AnnexWorkerError(code, _, _) => @@ -277,6 +288,7 @@ abstract class WorkerMain[S](stdin: InputStream = System.in, stdout: PrintStream workDir = Path.of(""), verbosity = 0, isCancelled = () => false, + inputs = None, ), )