Fix GetUser in daml script to populate the party participant map (#19787

)
digital-asset · Aug 15, 2024 · 691e4cf · 691e4cf
1 parent 9e08ea5
commit 691e4cf
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/...daml-script/runner/src/main/scala/com/digitalasset/daml/lf/engine/script/v1/ScriptF.scala b/...daml-script/runner/src/main/scala/com/digitalasset/daml/lf/engine/script/v1/ScriptF.scala
@@ -545,10 +545,17 @@ object ScriptF {
       for {
         client <- Converter.toFuture(env.clients.getParticipant(participant))
         user <- client.getUser(userId)
-        user <- Converter.toFuture(
+        userValue <- Converter.toFuture(
           Converter.fromOptional(user, Converter.fromUser(env.scriptIds, _))
         )
-      } yield SEAppAtomic(SEValue(continue), Array(SEValue(user)))
+      } yield {
+        (participant, user.flatMap(_.primaryParty)) match {
+          case (Some(participant), Some(party)) =>
+            env.addPartyParticipantMapping(party, participant)
+          case _ =>
+        }
+        SEAppAtomic(SEValue(continue), Array(SEValue(userValue)))
+      }
   }
 
   final case class DeleteUser(

diff --git a/...daml-script/runner/src/main/scala/com/digitalasset/daml/lf/engine/script/v2/ScriptF.scala b/...daml-script/runner/src/main/scala/com/digitalasset/daml/lf/engine/script/v2/ScriptF.scala
@@ -626,10 +626,17 @@ object ScriptF {
       for {
         client <- Converter.toFuture(env.clients.getParticipant(participant))
         user <- client.getUser(userId)
-        user <- Converter.toFuture(
+        userValue <- Converter.toFuture(
           Converter.fromOptional(user, Converter.fromUser(env.scriptIds, _))
         )
-      } yield SEValue(user)
+      } yield {
+        (participant, user.flatMap(_.primaryParty)) match {
+          case (Some(participant), Some(party)) =>
+            env.addPartyParticipantMapping(party, participant)
+          case _ =>
+        }
+        SEValue(userValue)
+      }
   }
 
   final case class DeleteUser(