workspace/TrainDelayBot/src/com/yuji/tdb/TrainDelayBotServlet.java

   1 package com.yuji.tdb;\r
   2 \r
   3 import java.io.IOException;\r
   4 \r
   5 import javax.servlet.http.HttpServlet;\r
   6 import javax.servlet.http.HttpServletRequest;\r
   7 import javax.servlet.http.HttpServletResponse;\r
   8 \r
   9 import twitter4j.Twitter;\r
  10 import twitter4j.TwitterException;\r
  11 import twitter4j.TwitterFactory;\r
  12 import twitter4j.auth.AccessToken;\r
  13 import twitter4j.auth.OAuthAuthorization;\r
  14 import twitter4j.auth.RequestToken;\r
  15 import twitter4j.conf.Configuration;\r
  16 import twitter4j.conf.ConfigurationBuilder;\r
  17 \r
  18 import com.yuji.tdb.common.CommonUtil;\r
  19 import com.yuji.tdb.twitter.TwitterUtil;\r
  20 \r
  21 @SuppressWarnings("serial")\r
  22 public class TrainDelayBotServlet extends HttpServlet {\r
  23         private static RequestToken requestToken = null;\r
  24         private static AccessToken accessToken = null;\r
  25         private static String consumerKey = null;\r
  26         private static String consumerSecret = null;\r
  27 \r
  28         public void doGet(HttpServletRequest req, HttpServletResponse resp)\r
  29                         throws IOException {\r
  30                 resp.setContentType("text/plain");\r
  31                 resp.getWriter().println(" traindelaybot Hello, world");\r
  32 \r
  33                 String consumerKey = req.getParameter("key");\r
  34                 String consumerSecret = req.getParameter("secret");\r
  35                 String result = req.getParameter("result");\r
  36                 if (!CommonUtil.isNull(consumerKey)\r
  37                                 && !CommonUtil.isNull(consumerSecret)) {\r
  38                         setOAuthAuthorization(req, resp, consumerKey, consumerSecret);\r
  39                         return;\r
  40                 } else if (!CommonUtil.isNull(result)) {\r
  41                         resultOAuthAuthorization();\r
  42                         return;\r
  43                 }\r
  44 \r
  45                 // TODO null check\r
  46                 TwitterUtil util = TwitterUtil.getInstance();\r
  47                 util.main(TrainDelayBotServlet.consumerKey,\r
  48                                 TrainDelayBotServlet.consumerSecret, accessToken);\r
  49         }\r
  50 \r
  51         private void setOAuthAuthorization(HttpServletRequest req,\r
  52                         HttpServletResponse resp, String consumerKey, String consumerSecret) {\r
  53                 try {\r
  54                         ConfigurationBuilder confBuilder = new ConfigurationBuilder();\r
  55 \r
  56                         confBuilder.setDebugEnabled(true);\r
  57                         confBuilder.setOAuthConsumerKey(consumerKey);\r
  58                         confBuilder.setOAuthConsumerSecret(consumerSecret);\r
  59                         Configuration conf = confBuilder.build();\r
  60 \r
  61                         String url = req.getRequestURL().toString();\r
  62 \r
  63                         OAuthAuthorization oauth = new OAuthAuthorization(conf);\r
  64                         // Twitter\82Ì\94F\8fØ\89æ\96Ê\82©\82ç\82Ì\96ß\82è\90æ\82ð\8ew\92è\82µ\82Ü\82·\81B\r
  65                         // AccessToken\83R\83\93\83g\83\8d\81[\83\89\82ð\8ew\92è\82µ\82Ü\82·\81B\r
  66                         String callbackURL = url + "?result=yes";\r
  67                         // RequestToken requestToken =\r
  68                         // oauth.getOAuthRequestToken(callbackURL);\r
  69                         requestToken = oauth.getOAuthRequestToken(callbackURL);\r
  70                         // RequestToken\82ð\83Z\83b\83V\83\87\83\93\82É\95Û\91¶\82µ\82Ä\82¨\82«\82Ü\82·\81B\r
  71                         // sessionScope("RequestToken", requestToken);\r
  72                         // Twitter\82Ì\94F\8fØ\89æ\96Ê\82Ö\83\8a\83_\83C\83\8c\83N\83g\82µ\82Ü\82·\81B\r
  73 \r
  74                         TrainDelayBotServlet.consumerKey = consumerKey;\r
  75                         TrainDelayBotServlet.consumerSecret = consumerSecret;\r
  76 \r
  77                         resp.sendRedirect(requestToken.getAuthenticationURL());\r
  78                 } catch (TwitterException e) {\r
  79                         // TODO Auto-generated catch block\r
  80                         e.printStackTrace();\r
  81                 } catch (IOException e) {\r
  82                         // TODO Auto-generated catch block\r
  83                         e.printStackTrace();\r
  84                 }\r
  85         }\r
  86 \r
  87         private void resultOAuthAuthorization() {\r
  88                 try {\r
  89                         // Twitter twitter = new TwitterFactory().getInstance();\r
  90                         if (requestToken == null || CommonUtil.isNull(consumerKey)\r
  91                                         || CommonUtil.isNull(consumerSecret)) {\r
  92                                 return;\r
  93                         }\r
  94                         // String tokenKey = requestToken.getToken();\r
  95                         // String tokenSecret = requestToken.getTokenSecret();\r
  96 \r
  97                         ConfigurationBuilder confBuilder = new ConfigurationBuilder();\r
  98                         confBuilder.setDebugEnabled(true).setOAuthConsumerKey(consumerKey)\r
  99                                         .setOAuthConsumerSecret(consumerSecret);\r
 100                         TwitterFactory factory = new TwitterFactory(confBuilder.build());\r
 101                         Twitter twitter = factory.getInstance();\r
 102 \r
 103                         accessToken = twitter.getOAuthAccessToken(requestToken);\r
 104                 } catch (TwitterException e) {\r
 105                         // TODO Auto-generated catch block\r
 106                         e.printStackTrace();\r
 107                 }\r
 108         }\r
 109 }\r