User substitution jmix-projects/jmix-security#30

content/project/single-module-application/src/main/java/${project_rootPath}/screen/main/MainScreen.java

@@ -5,6 +5,7 @@
 import io.jmix.ui.component.Button;
 import io.jmix.ui.component.Window;
 import io.jmix.ui.component.mainwindow.Drawer;
+import io.jmix.ui.component.mainwindow.UserIndicator;
 import io.jmix.ui.icon.JmixIcon;
 import io.jmix.ui.navigation.Route;
 import io.jmix.ui.screen.Screen;
@@ -13,6 +14,7 @@
 import io.jmix.ui.screen.UiControllerUtils;
 import io.jmix.ui.screen.UiDescriptor;
 import org.springframework.beans.factory.annotation.Autowired;
+import io.jmix.securitydata.impl.substitution.UserSubstitutionManager;
 
 @UiController("${normalizedPrefix_underscore}MainScreen")
 @UiDescriptor("main-screen.xml")
@@ -21,13 +23,18 @@ public class MainScreen extends Screen implements Window.HasWorkArea {
 
     @Autowired
     private ScreenTools screenTools;
+    @Autowired
+    private UserSubstitutionManager userSubstitutionManager;
 
     @Autowired
     private AppWorkArea workArea;
     @Autowired
     private Drawer drawer;
     @Autowired
     private Button collapseDrawerButton;
+    @Autowired
+    private UserIndicator userIndicator;
+
 
     @Override
     public AppWorkArea getWorkArea() {
@@ -44,6 +51,15 @@ private void onCollapseDrawerButtonClick(Button.ClickEvent event) {
         }
     }
 
+    @Subscribe
+    public void onAfterInit(AfterInitEvent afterInitEvent) {
+        userIndicator.setAdditionalUsers(userSubstitutionManager.getCurrentSubstitutedUsers());
+        userIndicator.addSubstituteStep(user->{
+            userSubstitutionManager.substituteUser(user.getUsername());
+        });
+        userIndicator.refreshUser();
+    }
+
     @Subscribe
     public void onAfterShow(AfterShowEvent event) {
         screenTools.openDefaultScreen(