✨ display message mutators for given message

Author: joaomlneto

simulator/src/main/java/byzzbench/simulator/controller/SimulatorController.java

@@ -3,14 +3,17 @@
 import byzzbench.simulator.Replica;
 import byzzbench.simulator.service.SimulatorService;
 import byzzbench.simulator.transport.Event;
+import byzzbench.simulator.transport.MessageEvent;
 import byzzbench.simulator.transport.MessageMutator;
 import lombok.RequiredArgsConstructor;
 import org.springframework.web.bind.annotation.*;
 
 import java.io.Serializable;
 import java.util.List;
+import java.util.Map;
 import java.util.Optional;
 import java.util.Set;
+import java.util.stream.Collectors;
 
 @RestController
 @RequiredArgsConstructor
@@ -94,6 +97,29 @@ public Event getMessage(@PathVariable Long eventId) {
         return simulatorService.getScenarioExecutor().getTransport().getEvents().get(eventId);
     }
 
+    @GetMapping("/event/{eventId}/mutators")
+    public Set<Long> getMessageMutators(@PathVariable Long eventId) {
+        Event e = simulatorService.getScenarioExecutor().getTransport().getEvents().get(eventId);
+
+        // if it is not a message, return an empty set
+        if (!(e instanceof MessageEvent)) {
+            return Set.of();
+        }
+
+        Set<Map.Entry<Long, MessageMutator>> messageMutators = simulatorService.getScenarioExecutor().getTransport()
+                .getMutators().entrySet()
+                .stream()
+                // filter mutators that can be applied to the message
+                .filter(entry -> entry.getValue().getInputClasses().contains(((MessageEvent) e).getPayload().getClass()))
+                //.map(entry -> entry.getKey())
+                .collect(Collectors.toSet());
+
+        System.out.println("Message Mutators for " + e + ": " + messageMutators);
+
+        // return their keys
+        return messageMutators.stream().map(Map.Entry::getKey).collect(Collectors.toSet());
+    }
+
     @PostMapping("/event/{eventId}/deliver")
     public void deliverMessage(@PathVariable Long eventId) throws Exception {
         simulatorService.getScenarioExecutor().getTransport().deliverEvent(eventId);

webui/components/messages/MutateMessageMenu/MutateMessageMenu.tsx

@@ -0,0 +1,94 @@
+"use client";
+
+import { MutateMessageMenuEntry } from "@/components/messages/MutateMessageMenu/MutateMessageMenuEntry";
+import {
+  useDropMessage,
+  useGetMessageMutators,
+} from "@/lib/byzzbench-client/generated";
+import { ActionIcon, Burger, Menu, rem } from "@mantine/core";
+import { showNotification } from "@mantine/notifications";
+import { IconSend, IconTrash } from "@tabler/icons-react";
+import { useQueryClient } from "@tanstack/react-query";
+import React from "react";
+
+type MutateMessageMenuProps = {
+  messageId: number;
+};
+
+export const MutateMessageMenu = ({ messageId }: MutateMessageMenuProps) => {
+  const queryClient = useQueryClient();
+  const { data } = useGetMessageMutators(messageId);
+  const { mutate: dropMessage } = useDropMessage(messageId);
+
+  return (
+    <Menu>
+      <Menu.Target>
+        <Burger />
+      </Menu.Target>
+      <Menu.Dropdown>
+        <Menu.Label>Mutate and Deliver</Menu.Label>
+        {data?.data.map((mutator) => (
+          <MutateMessageMenuEntry key={mutator} mutatorId={mutator} />
+        ))}
+
+        <Menu.Divider />
+
+        <Menu.Label>Danger zone</Menu.Label>
+        <Menu.Item
+          color="red"
+          leftSection={
+            <IconTrash style={{ width: rem(14), height: rem(14) }} />
+          }
+          onClick={(e) => {
+            e.preventDefault();
+            dropMessage(null as never, {
+              onSuccess: () => {
+                showNotification({
+                  message: `Message ${messageId} dropped`,
+                });
+              },
+              onError: () => {
+                showNotification({
+                  message: "Failed to drop message",
+                  color: "red",
+                });
+              },
+              onSettled: async () => {
+                await queryClient.invalidateQueries();
+              },
+            });
+          }}
+        >
+          Drop message
+        </Menu.Item>
+      </Menu.Dropdown>
+    </Menu>
+  );
+
+  return (
+    <ActionIcon
+      onClick={(e) => {
+        e.preventDefault();
+        const mutate = (a: any, b: any) => {};
+        mutate(null as never, {
+          onSuccess: () => {
+            showNotification({
+              message: `Message ${messageId} delivered`,
+            });
+          },
+          onError: () => {
+            showNotification({
+              message: "Failed to deliver message",
+              color: "red",
+            });
+          },
+          onSettled: async () => {
+            await queryClient.invalidateQueries();
+          },
+        });
+      }}
+    >
+      <IconSend />
+    </ActionIcon>
+  );
+};

webui/components/messages/MutateMessageMenu/MutateMessageMenuEntry.tsx

@@ -0,0 +1,46 @@
+"use client";
+
+import { useGetMutator } from "@/lib/byzzbench-client/generated";
+import { Menu, rem } from "@mantine/core";
+import { IconBug } from "@tabler/icons-react";
+import { useQueryClient } from "@tanstack/react-query";
+import React from "react";
+
+type MutateMessageMenuEntryProps = {
+  mutatorId: number;
+};
+
+export const MutateMessageMenuEntry = ({
+  mutatorId,
+}: MutateMessageMenuEntryProps) => {
+  const queryClient = useQueryClient();
+  const { data } = useGetMutator(mutatorId);
+
+  return (
+    <Menu.Item
+      leftSection={<IconBug style={{ width: rem(14), height: rem(14) }} />}
+      key={data?.data.name}
+      /*
+      onClick={() => {
+        mutator.mutate(null as never, {
+          onSuccess: () => {
+            showNotification({
+              message: `Message ${messageId} mutated`,
+            });
+          },
+          onError: () => {
+            showNotification({
+              message: "Failed to mutate message",
+              color: "red",
+            });
+          },
+          onSettled: async () => {
+            await queryClient.invalidateQueries();
+          },
+        });
+      }}*/
+    >
+      {data?.data.name}
+    </Menu.Item>
+  );
+};

webui/components/messages/MutateMessageMenu/index.ts

@@ -0,0 +1,2 @@
+export * from "./MutateMessageMenu";
+export * from "./MutateMessageMenuEntry";

webui/components/messages/NodeMailbox/NodeMailboxEntry.tsx

@@ -2,6 +2,7 @@ import {
   DeliverMessageActionIcon,
   DropMessageActionIcon,
 } from "@/components/messages";
+import { MutateMessageMenu } from "@/components/messages/MutateMessageMenu";
 import { NodeStateNavLink } from "@/components/NodeStateNavLink";
 import { useGetMessage } from "@/lib/byzzbench-client/generated";
 import { Badge, Collapse, Container, Group, Text } from "@mantine/core";
@@ -54,6 +55,7 @@ export const NodeMailboxEntry = ({ messageId }: { messageId: number }) => {
           <Group gap="xs" wrap="nowrap">
             <DeliverMessageActionIcon messageId={messageId} />
             <DropMessageActionIcon messageId={messageId} />
+            <MutateMessageMenu messageId={messageId} />
           </Group>
         )}
       </Group>