content: /*
   * Copyright 2013 Agorava
   *
   * 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 org.agorava.jsf;
  
  import org.agorava.api.atinject.BeanResolver;
  import org.agorava.api.storage.UserSessionRepository;
  
  import javax.faces.application.ViewHandler;
  import javax.faces.application.ViewHandlerWrapper;
  import javax.faces.context.FacesContext;
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Map;
  
  /**
   * @author Antoine Sabot-Durand
   */
  public class RepoAwareViewHandler extends ViewHandlerWrapper {
  
  
      private final ViewHandler delegate;
  
      private BeanResolver resolver = BeanResolver.getInstance();
  
  
      public RepoAwareViewHandler(ViewHandler delegate) {
          this.delegate = delegate;
      }
  
  
      @Override
      public ViewHandler getWrapped() {
          return delegate;
      }
  
  
      @Override
      public String getActionURL(FacesContext facesContext, String viewId) {
          String actionUrl = super.getActionURL(facesContext, viewId);
          UserSessionRepository repo = (UserSessionRepository) resolver.resolve("currentRepo");
  
  
          return new FacesUrlTransformer(actionUrl).appendParamIfNecessary("repoid",
                  repo.getId()).getUrl();
      }
  
  
      @Override
      public String getRedirectURL(FacesContext context, String viewId, Map<String, List<String>> parameters, boolean
              includeViewParams) {
          UserSessionRepository repo = (UserSessionRepository) resolver.resolve("currentRepo");
          List<String> values = new ArrayList<String>();
          values.add(repo.getId());
          parameters.put("repoid", values);
          return super.getRedirectURL(context, viewId, parameters, includeViewParams);
      }
  
      @Override
      public String getBookmarkableURL(FacesContext context, String viewId, Map<String, List<String>> parameters,
                                       boolean includeViewParams) {
          UserSessionRepository repo = (UserSessionRepository) resolver.resolve("currentRepo");
          List<String> values = new ArrayList<String>();
          values.add(repo.getId());
          parameters.put("repoid", values);
          return super.getBookmarkableURL(context, viewId, parameters, includeViewParams);
      }
  }
  