From 7a198a0b2a64b111135e43686f441d6373301cc5 Mon Sep 17 00:00:00 2001
From: Brent Yorgey <byorgey@gmail.com>
Date: Tue, 1 Nov 2022 21:56:35 -0500
Subject: [PATCH] copy parent robot context to child when executing `build`

Fixes #394.
---
 data/scenarios/Testing/394-build-drill.yaml | 9 ++++++++-
 src/Swarm/Game/Robot.hs                     | 7 ++++++-
 src/Swarm/Game/Step.hs                      | 3 ++-
 test/integration/Main.hs                    | 3 +--
 4 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/data/scenarios/Testing/394-build-drill.yaml b/data/scenarios/Testing/394-build-drill.yaml
index 1d09d720b..48bc79eb6 100644
--- a/data/scenarios/Testing/394-build-drill.yaml
+++ b/data/scenarios/Testing/394-build-drill.yaml
@@ -18,7 +18,14 @@ solution: |
   r <- build {
     wait 2;
     log "Hi, I am builder";
-    forever (build {log "Hi, I am pusher"; turn forward; forever push}; log "- robot built")
+    forever (
+      build {
+        log "Hi, I am pusher";
+        require "drill";          // #540
+        forever push
+      };
+      log "- robot built"
+    )
   };
   wait 10;
   place "detonator";
diff --git a/src/Swarm/Game/Robot.hs b/src/Swarm/Game/Robot.hs
index 2d52ff457..a13cbc93c 100644
--- a/src/Swarm/Game/Robot.hs
+++ b/src/Swarm/Game/Robot.hs
@@ -60,6 +60,7 @@ module Swarm.Game.Robot (
   inventoryHash,
   robotCapabilities,
   robotContext,
+  trobotContext,
   robotID,
   robotParentID,
   robotHeavy,
@@ -322,9 +323,13 @@ robotOrientation = robotEntity . entityOrientation
 robotInventory :: Lens' Robot Inventory
 robotInventory = robotEntity . entityInventory
 
--- | The robot's context
+-- | The robot's context.
 robotContext :: Lens' Robot RobotContext
 
+-- | The robot's context.
+trobotContext :: Lens' TRobot RobotContext
+trobotContext = lens _robotContext (\r c -> r {_robotContext = c})
+
 -- | The (unique) ID number of the robot.  This is only a Getter since
 --   the robot ID is immutable.
 robotID :: Getter Robot RID
diff --git a/src/Swarm/Game/Step.hs b/src/Swarm/Game/Step.hs
index 37270fdb3..d8cc11d72 100644
--- a/src/Swarm/Game/Step.hs
+++ b/src/Swarm/Game/Step.hs
@@ -1339,8 +1339,9 @@ execConst c vs s k = do
         createdAt <- getNow
 
         -- Construct the new robot and add it to the world.
+        parentCtx <- use robotContext
         newRobot <-
-          addTRobot $
+          addTRobot . (trobotContext .~ parentCtx) $
             mkRobot
               ()
               (Just pid)
diff --git a/test/integration/Main.hs b/test/integration/Main.hs
index 1e6eff02d..be103e103 100644
--- a/test/integration/Main.hs
+++ b/test/integration/Main.hs
@@ -169,8 +169,7 @@ testScenarioSolution _ci _em =
         ]
     , testGroup
         "Regression tests"
-        [ expectFailBecause "Awaiting fix (#394)" $
-            testSolution Default "Testing/394-build-drill"
+        [ testSolution Default "Testing/394-build-drill"
         , testSolution Default "Testing/373-drill"
         , testSolution Default "Testing/428-drowning-destroy"
         , testSolution' Default "Testing/475-wait-one" $ \g -> do