/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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 org.apache.shiro.web.subject.support;
  
  import org.apache.shiro.subject.Subject;
  import org.apache.shiro.subject.support.DefaultSubjectContext;
  import org.apache.shiro.web.subject.WebSubject;
  import org.apache.shiro.web.subject.WebSubjectContext;
  
  import javax.servlet.ServletRequest;
  import javax.servlet.ServletResponse;
  
  /**
   * Default {@code WebSubjectContext} implementation that provides for additional storage and retrieval of
   * a {@link ServletRequest} and {@link ServletResponse}.
   *
   * @since 1.0
   */
  public class DefaultWebSubjectContext extends DefaultSubjectContext implements WebSubjectContext {
  
      private static final long serialVersionUID = 8188555355305827739L;
  
      private static final String SERVLET_REQUEST = DefaultWebSubjectContext.class.getName() + ".SERVLET_REQUEST";
      private static final String SERVLET_RESPONSE = DefaultWebSubjectContext.class.getName() + ".SERVLET_RESPONSE";
  
      public DefaultWebSubjectContext() {
      }
  
      public DefaultWebSubjectContext(WebSubjectContext context) {
          super(context);
      }
  
      @Override
      public String resolveHost() {
          String host = super.resolveHost();
          if (host == null) {
              ServletRequest request = resolveServletRequest();
              if (request != null) {
                  host = request.getRemoteHost();
              }
          }
          return host;
      }
  
      public ServletRequest getServletRequest() {
          return getTypedValue(SERVLET_REQUEST, ServletRequest.class);
      }
  
      public void setServletRequest(ServletRequest request) {
          if (request != null) {
              put(SERVLET_REQUEST, request);
          }
      }
  
      public ServletRequest resolveServletRequest() {
  
          ServletRequest request = getServletRequest();
  
          //fall back on existing subject instance if it exists:
          if (request == null) {
              Subject existing = getSubject();
              if (existing instanceof WebSubject) {
                  request = ((WebSubject) existing).getServletRequest();
              }
          }
  
          return request;
      }
  
      public ServletResponse getServletResponse() {
          return getTypedValue(SERVLET_RESPONSE, ServletResponse.class);
      }
  
      public void setServletResponse(ServletResponse response) {
          if (response != null) {
              put(SERVLET_RESPONSE, response);
          }
      }
  
      public ServletResponse resolveServletResponse() {
  
          ServletResponse response = getServletResponse();
  
         //fall back on existing subject instance if it exists:
         if (response == null) {
             Subject existing = getSubject();
             if (existing instanceof WebSubject) {
                 response = ((WebSubject) existing).getServletResponse();
             }
         }
 
         return response;
     }
 }