/**
    *       Copyright 2010 Newcastle University
    *
    *          http://research.ncl.ac.uk/smart/
    *
    * 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.amber.oauth2.client.demo.controller;
  
  import java.io.IOException;
  
  import javax.servlet.http.HttpServletRequest;
  
  import org.apache.amber.oauth2.client.OAuthClient;
  import org.apache.amber.oauth2.client.URLConnectionClient;
  import org.apache.amber.oauth2.client.demo.Utils;
  import org.apache.amber.oauth2.client.demo.exception.ApplicationException;
  import org.apache.amber.oauth2.client.demo.model.OAuthParams;
  import org.apache.amber.oauth2.client.request.OAuthClientRequest;
  import org.apache.amber.oauth2.client.response.GitHubTokenResponse;
  import org.apache.amber.oauth2.client.response.OAuthAccessTokenResponse;
  import org.apache.amber.oauth2.client.response.OAuthJSONAccessTokenResponse;
  import org.apache.amber.oauth2.common.exception.OAuthProblemException;
  import org.apache.amber.oauth2.common.exception.OAuthSystemException;
  import org.apache.amber.oauth2.common.message.types.GrantType;
  import org.springframework.stereotype.Controller;
  import org.springframework.web.bind.annotation.ModelAttribute;
  import org.springframework.web.bind.annotation.RequestMapping;
  import org.springframework.web.servlet.ModelAndView;
  
  
  /**
   *
   *
   *
   */
  @Controller
  @RequestMapping("/get_token")
  public class TokenController {
  
      @RequestMapping
      public ModelAndView authorize(@ModelAttribute("oauthParams") OAuthParams oauthParams,
                                    HttpServletRequest req) throws OAuthSystemException, IOException {
  
          try {
   
              Utils.validateTokenParams(oauthParams);
  
              OAuthClientRequest request = OAuthClientRequest
                  .tokenLocation(oauthParams.getTokenEndpoint())
                  .setClientId(oauthParams.getClientId())
                  .setClientSecret(oauthParams.getClientSecret())
                  .setRedirectURI(oauthParams.getRedirectUri())
                  .setCode(oauthParams.getAuthzCode())
                  .setGrantType(GrantType.AUTHORIZATION_CODE)
                  .buildBodyMessage();
  
              OAuthClient client = new OAuthClient(new URLConnectionClient());
              String app = Utils.findCookieValue(req, "app");
    
              OAuthAccessTokenResponse oauthResponse = null;
              Class<? extends OAuthAccessTokenResponse> cl = OAuthJSONAccessTokenResponse.class;
  
              if (Utils.FACEBOOK.equals(app)) {
                  cl = GitHubTokenResponse.class;
              } else if (Utils.GITHUB.equals(app)) {
                  cl = GitHubTokenResponse.class;
              }
  
              oauthResponse = client.accessToken(request, cl);
  
              oauthParams.setAccessToken(oauthResponse.getAccessToken());
              oauthParams.setExpiresIn(oauthResponse.getExpiresIn());
              oauthParams.setRefreshToken(Utils.isIssued(oauthResponse.getRefreshToken()));
  
              return new ModelAndView("get_resource");
  
          } catch (ApplicationException e) {
              oauthParams.setErrorMessage(e.getMessage());
              return new ModelAndView("request_token");
          } catch (OAuthProblemException e) {
              StringBuffer sb = new StringBuffer();
              sb.append("</br>");
              sb.append("Error code: ").append(e.getError()).append("</br>");
              sb.append("Error description: ").append(e.getDescription()).append("</br>");
             sb.append("Error uri: ").append(e.getUri()).append("</br>");
             sb.append("State: ").append(e.getState()).append("</br>");
             oauthParams.setErrorMessage(sb.toString());
             return new ModelAndView("get_authz");
         }
     }
 }