content: /*
   * Copyright 2016 Red Hat, Inc. and/or its affiliates
   * and other contributors as indicated by the @author tags.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   * http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package de.ahus1.keycloak.jetty.core;
  
  import jakarta.servlet.http.HttpSession;
  import org.eclipse.jetty.server.Request;
  import org.jboss.logging.Logger;
  import org.keycloak.KeycloakPrincipal;
  import org.keycloak.KeycloakSecurityContext;
  import org.keycloak.adapters.*;
  import org.keycloak.adapters.spi.HttpFacade;
  
  import java.security.Principal;
  import java.util.Set;
  
  /**
   * @author <a href="mailto:bill@burkecentral.com">Bill Burke</a>
   * @version $Revision: 1 $
   */
  public class JettyRequestAuthenticator extends RequestAuthenticator {
      protected static final Logger log = Logger.getLogger(JettyRequestAuthenticator.class);
      protected Request request;
      protected KeycloakPrincipal<RefreshableKeycloakSecurityContext> principal;
  
      public JettyRequestAuthenticator(HttpFacade facade, KeycloakDeployment deployment, AdapterTokenStore tokenStore, int sslRedirectPort, Request request) {
          super(facade, deployment, tokenStore, sslRedirectPort);
          this.request = request;
      }
  
      @Override
      protected OAuthRequestAuthenticator createOAuthAuthenticator() {
          return new OAuthRequestAuthenticator(this, facade, deployment, sslRedirectPort, tokenStore);
      }
  
      @Override
      protected void completeOAuthAuthentication(final KeycloakPrincipal<RefreshableKeycloakSecurityContext> skp) {
          principal = skp;
          final RefreshableKeycloakSecurityContext securityContext = skp.getKeycloakSecurityContext();
          final Set<String> roles = AdapterUtils.getRolesFromSecurityContext(securityContext);
          OidcKeycloakAccount account = new OidcKeycloakAccount() {
  
              @Override
              public Principal getPrincipal() {
                  return skp;
              }
  
              @Override
              public Set<String> getRoles() {
                  return roles;
              }
  
              @Override
              public KeycloakSecurityContext getKeycloakSecurityContext() {
                  return securityContext;
              }
  
          };
          request.setAttribute(KeycloakSecurityContext.class.getName(), securityContext);
          this.tokenStore.saveAccountInfo(account);
      }
  
      @Override
      protected void completeBearerAuthentication(KeycloakPrincipal<RefreshableKeycloakSecurityContext> principal, String method) {
          this.principal = principal;
          RefreshableKeycloakSecurityContext securityContext = principal.getKeycloakSecurityContext();
          Set<String> roles = AdapterUtils.getRolesFromSecurityContext(securityContext);
          if (log.isDebugEnabled()) {
              log.debug("Completing bearer authentication. Bearer roles: " + roles);
          }
          request.setAttribute(KeycloakSecurityContext.class.getName(), securityContext);
      }
  
  
      @Override
      protected String changeHttpSessionId(boolean create) {
          HttpSession session = request.getSession(create);
          return session != null ? session.getId() : null;
      }
  
  
  }
  