package com.appspot.plucial.controller.account;
import java.util.List;
import org.slim3.controller.Controller;
import org.slim3.controller.Navigation;
import com.appspot.plucial.Constants;
import com.appspot.plucial.exception.UserMaxLimitException;
import com.appspot.plucial.model.UserModel;
import com.appspot.plucial.service.UserService;
import com.appspot.plucial.service.UserUrlsService;
import com.google.api.client.googleapis.auth.oauth2.GoogleAuthorizationCodeTokenRequest;
import com.google.api.client.googleapis.auth.oauth2.GoogleCredential;
import com.google.api.client.googleapis.auth.oauth2.GoogleTokenResponse;
import com.google.api.client.http.HttpTransport;
import com.google.api.client.http.javanet.NetHttpTransport;
import com.google.api.client.json.jackson2.JacksonFactory;
import com.google.api.server.spi.response.UnauthorizedException;
import com.google.api.services.oauth2.Oauth2;
import com.google.api.services.oauth2.model.Tokeninfo;
import com.google.api.services.plus.Plus;
import com.google.api.services.plus.model.Person;
import com.google.api.services.plus.model.Person.Urls;
/**
* ユーザー追加コントローラー
* @author takahara
*
*/
public class UserLoginController extends Controller {
private static final HttpTransport TRANSPORT = new NetHttpTransport();
private static final JacksonFactory JSON_FACTORY = new JacksonFactory();
@Override
public Navigation run() throws Exception {
String error = asString("error");
// 承認をキャンセルした場合などの承認エラーが発生した場合
if(error != null && error.length() > 0) {
return redirect("/account/login");
}
try {
// ユーザーログイン
login();
}catch(UserMaxLimitException ex) {
return forward("/responsive/account/user_max_limit.jsp");
}catch(Exception e) {
e.printStackTrace();
throw new UnauthorizedException("No Login:login");
}
// return null; (ポップアップ用)
return forward("/account/loggedIn");
}
/**
* ユーザー登録
* @return
* @throws Exception
*/
private void login() throws Exception {
String code = asString("code");
// ---------------------------------------------------------
// トークン情報の取得
// ---------------------------------------------------------
// 承認コードをアクセス・更新トークンにアップグレードします。(ポップアップ用)
// GoogleTokenResponse tokenResponse =
// new GoogleAuthorizationCodeTokenRequest(
// TRANSPORT,
// JSON_FACTORY,
// Constants.GOOGLE_PROJECT_CLIENT_ID,
// Constants.GOOGLE_PROJECT_CLIENT_SECRET,
// code,
// "postmessage").execute();
// 承認コードをアクセス・更新トークンにアップグレードします。
GoogleTokenResponse tokenResponse =
new GoogleAuthorizationCodeTokenRequest(
TRANSPORT,
JSON_FACTORY,
Constants.GOOGLE_PROJECT_CLIENT_ID,
Constants.GOOGLE_PROJECT_CLIENT_SECRET,
code,
request.getRequestURL().toString()).execute();
// トークン情報の取得（アクセストークン、リフレッシュトークン・・・）
GoogleCredential credential = new GoogleCredential.Builder()
.setJsonFactory(JSON_FACTORY)
.setTransport(TRANSPORT)
.setClientSecrets(Constants.GOOGLE_PROJECT_CLIENT_ID, Constants.GOOGLE_PROJECT_CLIENT_SECRET).build()
.setFromTokenResponse(tokenResponse);
// ---------------------------------------------------------
// トークン情報の有効チェック
// ---------------------------------------------------------
// トークンの有効チェック
Oauth2 oauth2 = new Oauth2.Builder(
TRANSPORT, JSON_FACTORY, credential)
.setApplicationName(Constants.GOOGLE_APPLICATION_NAME)
.build();
Tokeninfo tokenInfo = oauth2.tokeninfo()
.setAccessToken(credential.getAccessToken()).execute();
// トークン情報にエラーがあれば、中断すしま。
if (tokenInfo.containsKey("error")) {
throw new Exception();
}
// 受け取ったトークンが自分のアプリのものであることを確認します。
if (!tokenInfo.getIssuedTo().equals(Constants.GOOGLE_PROJECT_CLIENT_ID)) {
throw new Exception();
}
// ユーザー情報の取得
UserModel userModel = UserService.getOrNull(tokenInfo.getUserId());
if(userModel == null) {
// ユーザー制限チェック
List<UserModel> userList = UserService.getAllUserList();
if(userList.size() > 1000) {
throw new UserMaxLimitException();
}
// ---------------------------------------------------------
// ユーザー登録
// ---------------------------------------------------------
// Google Plus APIを使ってユーザー情報を取得する
Plus plus = new Plus.Builder(TRANSPORT, JSON_FACTORY, credential)
.setApplicationName(Constants.GOOGLE_APPLICATION_NAME)
.build();
Person person = plus.people().get("me").execute();
userModel = UserService.put(
tokenInfo.getUserId(),
tokenInfo.getEmail(),
person.getUrl(),
person.getDisplayName(),
person.getImage(),
person.getTagline(),
person.getBraggingRights(),
person.getAboutMe(),
person.getCover(),
credential.getAccessToken(),
credential.getRefreshToken()
);
// URLS登録
if(person.getUrls() != null && person.getUrls().size() > 0) {
List<Urls> urlsList = person.getUrls();
for(Urls urls: urlsList) {
UserUrlsService.put(userModel, urls.getValue(), urls.getType(), urls.getLabel());
}
}
// ユーザー数のキャッシュをクリア
UserService.clearUserCountAndListMemcache();
}else {
// ---------------------------------------------------------
// ユーザーログイン
// ---------------------------------------------------------
// ログインユーザーのアクセストークンとリフレッシュトークンを更新
userModel.setAccessToken(credential.getAccessToken());
if(credential.getRefreshToken() != null) {
userModel.setRefreshToken(credential.getRefreshToken());
}
UserService.put(userModel);
}
// ユーザー情報をセッションに入れる
sessionScope("userModel", userModel);
}
}