DeepSQLAI
diff --git a/‎.env.example‎
Lines changed: 3 additions & 0 deletions b/‎.env.example‎
Lines changed: 3 additions & 0 deletions
diff --git a/‎AGENTS.md‎
Lines changed: 10 additions & 3 deletions b/‎AGENTS.md‎
Lines changed: 10 additions & 3 deletions
diff --git a/‎CLAUDE.md‎
Lines changed: 5 additions & 2 deletions b/‎CLAUDE.md‎
Lines changed: 5 additions & 2 deletions
diff --git a/‎backend/src/main/java/com/dbaagent/controller/AuthController.java‎
Lines changed: 16 additions & 5 deletions b/‎backend/src/main/java/com/dbaagent/controller/AuthController.java‎
Lines changed: 16 additions & 5 deletions
diff --git a/‎backend/src/main/java/com/dbaagent/controller/ImpersonationController.java‎
Lines changed: 136 additions & 0 deletions b/‎backend/src/main/java/com/dbaagent/controller/ImpersonationController.java‎
Lines changed: 136 additions & 0 deletions
diff --git a/‎backend/src/main/java/com/dbaagent/model/SecurityEventType.java‎
Lines changed: 2 additions & 0 deletions b/‎backend/src/main/java/com/dbaagent/model/SecurityEventType.java‎
Lines changed: 2 additions & 0 deletions
diff --git a/‎backend/src/main/java/com/dbaagent/security/ImpersonationContext.java‎
Lines changed: 50 additions & 0 deletions b/‎backend/src/main/java/com/dbaagent/security/ImpersonationContext.java‎
Lines changed: 50 additions & 0 deletions
diff --git a/‎backend/src/main/java/com/dbaagent/security/JwtAuthenticationFilter.java‎
Lines changed: 16 additions & 2 deletions b/‎backend/src/main/java/com/dbaagent/security/JwtAuthenticationFilter.java‎
Lines changed: 16 additions & 2 deletions
@@ -141,6 +141,9 @@ EMBEDDING_FAIL_OPEN=false
 # AGENT_WEBUI_URL          Agent HTTP API. Compose default: http://deepsql-agent:8787
 # AGENT_PROVISIONER_URL   Per-user profile provisioner. Compose default:
 #                         http://deepsql-agent:8788/provision
+# DEEPSQL_API_BASE_URL    Where the agent container's MCP tools call the backend.
+#                         Compose default: http://backend:8080/api/
+#                         Native Java + Compose agent: http://host.docker.internal:8080/api/
 # AGENT_PROVISION_SECRET  Shared secret between backend and agent (required).
 # DEEPSQL_AGENT_PORT / DEEPSQL_AGENT_PROVISIONER_PORT — host port mappings
 # (compose binds these to 127.0.0.1 only; public path is nginx /agent-api).
 
@@ -264,9 +264,16 @@ only covers cloud-specific, non-obvious caveats.
   non-`public` schemas (`crm`, `sales`, `finance`, `hr`, `inventory`) for Brain /
   MCP cross-schema checks. Prefer schema-qualified SQL (`sales.orders`); bare
   names follow the role’s `search_path` (usually `public`).
-- **`AGENT_WEBUI_URL` for native runs.** Default is `http://deepsql-agent:8787`
-  (Compose DNS). Native local must set `AGENT_WEBUI_URL=http://127.0.0.1:8787` in
-  `.env` or CLI/Slack `AgentChatClient` cannot reach the agent API.
+- **`AGENT_WEBUI_URL` / `AGENT_PROVISIONER_URL` for native runs.** Compose
+  defaults (`http://deepsql-agent:8787` and `…:8788/provision`) do not resolve
+  on the host. Native local must point both at loopback
+  (`http://127.0.0.1:8787` and `http://127.0.0.1:8788/provision`) or the Agent
+  tab returns 503 `Could not provision the DeepSQL Agent for this user`.
+  `scripts/start-backend.sh` remaps those hostnames automatically when they
+  don't resolve. If the agent container is used with a host-side Java backend,
+  set `DEEPSQL_API_BASE_URL=http://host.docker.internal:8080/api/` so MCP
+  tools can reach the native process (compose publishes `host.docker.internal`
+  via `extra_hosts`).
 - **DeepSQL CLI (`deepsql`) for agent testing.** Install from the repo package:
   `cd mcp && DEEPSQL_SKIP_AGENT_SETUP=1 npm install -g .` (prefix
   `~/.npm-global`, keep that on `PATH`). Auth against local backend with an MCP
 
@@ -93,7 +93,7 @@ backend/
     repository/     # Spring Data repositories
     provider/       # Database dialect registry (PostgreSQL, MySQL)
     config/         # Spring configuration
-    security/       # JWT auth, RBAC
+    security/       # JWT auth, RBAC, admin profile switch (`ImpersonationService`)
     llm/            # LLM provider registry, config resolver, OpenAI-compatible provider
     util/           # Shared utilities
   src/test/         # JUnit 5 tests
@@ -110,7 +110,7 @@ src/                # Frontend (React)
   components/       # UI components
     tabs/           # 40+ specialized tabs
     sections/       # Top-level sidebar destinations (Agent, Dashboards, Brain,
-                    # Performance = Slow Queries + Workload, Editor, Docs)
+                    # Performance = Slow Queries + Workload, Editor)
   lib/
     api/client.js   # Centralized API layer (axios, 25+ modules)
     stores/         # Zustand stores (dashboard, connection, chat, UI)
@@ -209,6 +209,9 @@ returns a number).
 4. **Tooltips**: Always use `HelpTooltip` component, never plain `title` attributes.
 5. **Design**: Minimal black/white/grey palette, Inter font, subtle transitions. See UX guidelines in full CLAUDE.md.
 
+### Admin profile switch
+Admins can **View as** a sub-user from the top-right of the home layout (`ProfileSwitch`) to verify connection ACLs, chat/editor policies, and role-gated nav. The admin JWT stays on the session; `ImpersonationService` sets an httpOnly `impersonate_user` cookie and `JwtAuthenticationFilter` overlays the target principal. `POST|DELETE|GET /api/admin/impersonate` are excluded from the overlay so stop/list still run as the real admin. Cannot target another ADMIN, self, or a non-ACTIVE account. `/auth/me` returns the **effective** user plus `impersonating` / `impersonatorUsername`.
+
 ### Git Rules
 - Do NOT commit automatically — wait for explicit user instruction.
 - Conventional commits: `feat:`, `fix:`, `refactor:`, `docs:`, `test:`, `chore:`, `perf:`, `ci:`
 
@@ -9,6 +9,8 @@
 import com.dbaagent.service.AuthSessionService;
 import com.dbaagent.service.PasswordlessAuthService;
 import com.dbaagent.service.PermissionService;
+import com.dbaagent.service.ImpersonationService;
+import com.dbaagent.security.ImpersonationContext;
 import com.dbaagent.service.SystemConfigService;
 import com.dbaagent.service.UserInviteService;
 import jakarta.servlet.http.Cookie;
@@ -47,6 +49,7 @@ public class AuthController {
     private final PrivateBetaRequestRepository privateBetaRequestRepository;
     private final SystemConfigService systemConfigService;
     private final AgentBridgeService agentBridgeService;
+    private final ImpersonationService impersonationService;
 
     @Value("${security.cookie.refresh-name:refresh_token}")
     private String refreshCookieName;
@@ -201,12 +204,21 @@ public ResponseEntity<?> refreshSession(HttpServletRequest httpRequest, HttpServ
             return ResponseEntity.status(401).body(Map.of("message", "Session expired"));
         }
         authSessionService.writeSessionCookies(httpResponse, refreshed.get());
+        User effectiveUser = impersonationService.resolveFromCookie(httpRequest, user)
+            .map(ImpersonationContext.State::target)
+            .orElse(user);
         // Keep the user's agent token alive for as long as the UI session lives.
         // The SPA refreshes on access-token expiry (~every 15 min of activity), so
         // this slides the agent token forward on each active interval — a logged-in
         // UI never ends up with a dead agent.
         agentBridgeService.extendAgentTokens(user.getUsername());
-        return ResponseEntity.ok(toAuthPayload(user, user.getRoleEnum(), permissionService.getEffectivePermissionCodes(user.getRoleEnum())));
+        Map<String, Object> payload = toAuthPayload(
+            effectiveUser,
+            effectiveUser.getRoleEnum(),
+            permissionService.getEffectivePermissionCodes(effectiveUser.getRoleEnum())
+        );
+        impersonationService.decorateAuthPayload(httpRequest, user, payload);
+        return ResponseEntity.ok(payload);
     }
 
     @PostMapping("/logout")
@@ -304,7 +316,7 @@ public ResponseEntity<?> acceptInvite(
     }
 
     @GetMapping("/me")
-    public ResponseEntity<?> getCurrentUser() {
+    public ResponseEntity<?> getCurrentUser(HttpServletRequest httpRequest) {
         Authentication auth = SecurityContextHolder.getContext().getAuthentication();
         if (auth == null || !auth.isAuthenticated() || "anonymousUser".equals(auth.getPrincipal())) {
             return ResponseEntity.status(401).body(Map.of("message", "Not authenticated"));
@@ -313,9 +325,7 @@ public ResponseEntity<?> getCurrentUser() {
         Role role = user.getRoleEnum();
         Set<String> permissions = permissionService.getEffectivePermissionCodes(role);
         Map<String, Object> response = toAuthPayload(user, role, permissions);
-        response.put("emailVerified", user.isEmailVerified());
-        response.put("accountStatus", user.getAccountStatus());
-        response.put("emailTwoFactorEnabled", systemConfigService.getBoolean("security.workspace.email2fa.enabled"));
+        impersonationService.decorateAuthPayload(httpRequest, user, response);
         return ResponseEntity.ok(response);
     }
 
@@ -360,6 +370,7 @@ private ResponseEntity<?> authResponse(PasswordlessAuthService.AuthFlowResult re
         }
         if (result.sessionAuthentication() != null && result.user() != null && result.role() != null) {
             authSessionService.writeSessionCookies(httpResponse, result.sessionAuthentication());
+            authSessionService.clearImpersonationCookie(httpResponse);
             Set<String> permissionNames = result.permissions() == null ? Set.of() : result.permissions().stream()
                 .map(Enum::name)
                 .collect(Collectors.toSet());
 
@@ -0,0 +1,136 @@
+package com.dbaagent.controller;
+
+import com.dbaagent.model.Role;
+import com.dbaagent.model.User;
+import com.dbaagent.repository.UserRepository;
+import com.dbaagent.security.ImpersonationContext;
+import com.dbaagent.service.ImpersonationService;
+import com.dbaagent.service.PermissionService;
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.ResponseEntity;
+import org.springframework.security.access.prepost.PreAuthorize;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.server.ResponseStatusException;
+
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.Set;
+
+/**
+ * Admin-only profile switch. These paths are excluded from the impersonation
+ * overlay so the caller stays the real administrator while starting, listing,
+ * or stopping a switch.
+ */
+@RestController
+@RequestMapping("/admin/impersonate")
+@PreAuthorize("hasRole('ADMIN')")
+@RequiredArgsConstructor
+public class ImpersonationController {
+
+    private final ImpersonationService impersonationService;
+    private final UserRepository userRepository;
+    private final PermissionService permissionService;
+
+    @GetMapping
+    public ResponseEntity<Map<String, Object>> status(HttpServletRequest request) {
+        User actor = currentAdmin();
+        ImpersonationContext.State state = impersonationService.resolveFromCookie(request, actor).orElse(null);
+        Map<String, Object> body = new LinkedHashMap<>();
+        body.put("impersonating", state != null);
+        body.put("impersonator", Map.of(
+            "id", actor.getId(),
+            "username", actor.getUsername(),
+            "email", actor.getEmail()
+        ));
+        body.put("target", state == null ? null : candidateView(state.target()));
+        body.put("candidates", impersonationService.listCandidates(actor));
+        return ResponseEntity.ok(body);
+    }
+
+    @PostMapping
+    public ResponseEntity<Map<String, Object>> start(
+        @RequestBody Map<String, Object> requestBody,
+        HttpServletRequest request,
+        HttpServletResponse response
+    ) {
+        User actor = currentAdmin();
+        Long userId = readUserId(requestBody);
+        ImpersonationContext.State state = impersonationService.start(actor, userId, request, response);
+        return ResponseEntity.ok(toAuthPayload(state.target(), actor));
+    }
+
+    @DeleteMapping
+    public ResponseEntity<Map<String, Object>> stop(
+        HttpServletRequest request,
+        HttpServletResponse response
+    ) {
+        User actor = currentAdmin();
+        User restored = impersonationService.stop(actor, request, response);
+        Map<String, Object> payload = toAuthPayload(restored, null);
+        payload.put("impersonating", false);
+        return ResponseEntity.ok(payload);
+    }
+
+    private Map<String, Object> toAuthPayload(User user, User impersonator) {
+        Role role = user.getRoleEnum();
+        Set<String> permissions = permissionService.getEffectivePermissionCodes(role);
+        Map<String, Object> payload = new LinkedHashMap<>();
+        payload.put("username", user.getUsername());
+        payload.put("email", user.getEmail());
+        payload.put("role", role.name());
+        payload.put("permissions", permissions);
+        payload.put("emailVerified", user.isEmailVerified());
+        payload.put("accountStatus", user.getAccountStatus());
+        if (impersonator != null) {
+            payload.put("impersonating", true);
+            payload.put("impersonatorUsername", impersonator.getUsername());
+            payload.put("impersonatorEmail", impersonator.getEmail());
+        } else {
+            payload.put("impersonating", false);
+        }
+        return payload;
+    }
+
+    private Map<String, Object> candidateView(User user) {
+        Map<String, Object> dto = new LinkedHashMap<>();
+        dto.put("id", user.getId());
+        dto.put("username", user.getUsername());
+        dto.put("email", user.getEmail());
+        dto.put("role", user.getRole());
+        dto.put("accountStatus", user.getAccountStatus());
+        return dto;
+    }
+
+    private Long readUserId(Map<String, Object> requestBody) {
+        if (requestBody == null || requestBody.get("userId") == null) {
+            return null;
+        }
+        Object raw = requestBody.get("userId");
+        if (raw instanceof Number number) {
+            return number.longValue();
+        }
+        try {
+            return Long.parseLong(String.valueOf(raw).trim());
+        } catch (NumberFormatException e) {
+            return null;
+        }
+    }
+
+    private User currentAdmin() {
+        Authentication auth = SecurityContextHolder.getContext().getAuthentication();
+        if (auth == null || !auth.isAuthenticated() || "anonymousUser".equals(auth.getPrincipal())) {
+            throw new ResponseStatusException(org.springframework.http.HttpStatus.UNAUTHORIZED, "Not authenticated");
+        }
+        return userRepository.findByUsername(auth.getName())
+            .orElseThrow(() -> new ResponseStatusException(org.springframework.http.HttpStatus.UNAUTHORIZED, "User not found"));
+    }
+}
@@ -25,6 +25,8 @@ public enum SecurityEventType {
     SESSION_REFRESHED,
     SESSION_REVOKED,
     SESSION_EXPIRED,
+    IMPERSONATION_STARTED,
+    IMPERSONATION_STOPPED,
     LOGOUT,
     LOGOUT_ALL,
     ACCOUNT_LOCKED,
 
@@ -0,0 +1,50 @@
+package com.dbaagent.security;
+
+import com.dbaagent.model.User;
+
+import java.util.Optional;
+
+/**
+ * Request-scoped impersonation overlay. The admin JWT stays on the session;
+ * {@link JwtAuthenticationFilter} swaps the SecurityContext principal to the
+ * target user and records both identities here so {@code /auth/me} can show a
+ * banner and {@code AccessControlService} can honour the target even when
+ * {@code security.auth.enabled} is false.
+ */
+public final class ImpersonationContext {
+
+    public record State(User impersonator, User target) {
+        public String impersonatorUsername() {
+            return impersonator != null ? impersonator.getUsername() : null;
+        }
+
+        public String impersonatorEmail() {
+            return impersonator != null ? impersonator.getEmail() : null;
+        }
+
+        public String targetUsername() {
+            return target != null ? target.getUsername() : null;
+        }
+    }
+
+    private static final ThreadLocal<State> CURRENT = new ThreadLocal<>();
+
+    private ImpersonationContext() {
+    }
+
+    public static void enter(State state) {
+        CURRENT.set(state);
+    }
+
+    public static void clear() {
+        CURRENT.remove();
+    }
+
+    public static Optional<State> current() {
+        return Optional.ofNullable(CURRENT.get());
+    }
+
+    public static boolean isActive() {
+        return CURRENT.get() != null;
+    }
+}
@@ -16,6 +16,7 @@
 import org.springframework.web.filter.OncePerRequestFilter;
 
 import com.dbaagent.service.AuthSessionService;
+import com.dbaagent.service.ImpersonationService;
 
 import java.io.IOException;
 import java.util.ArrayList;
@@ -40,6 +41,9 @@ public class JwtAuthenticationFilter extends OncePerRequestFilter {
     @Autowired
     private AuthSessionService authSessionService;
 
+    @Autowired
+    private ImpersonationService impersonationService;
+
     @Override
     protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain)
             throws ServletException, IOException {
@@ -69,7 +73,7 @@ protected void doFilterInternal(HttpServletRequest request, HttpServletResponse
             UsernamePasswordAuthenticationToken auth = new UsernamePasswordAuthenticationToken(
                     "admin", null, devAuthorities);
             SecurityContextHolder.getContext().setAuthentication(auth);
-            chain.doFilter(request, response);
+            applyImpersonation(request, response, chain);
             return;
         }
 
@@ -132,7 +136,17 @@ protected void doFilterInternal(HttpServletRequest request, HttpServletResponse
         } else if (isBrainRequest) {
             log.warn("Brain auth missing user: path={}, username={}", requestPath, username);
         }
-        chain.doFilter(request, response);
+        applyImpersonation(request, response, chain);
+    }
+
+    private void applyImpersonation(HttpServletRequest request, HttpServletResponse response, FilterChain chain)
+            throws ServletException, IOException {
+        try {
+            impersonationService.applyToRequest(request);
+            chain.doFilter(request, response);
+        } finally {
+            ImpersonationContext.clear();
+        }
     }
 
     private String extractUsernameSafely(String token) {