OSDN Git Service

5141fc12882238e36f4985d972604501f9b95e1c
[traindelaybot/source.git] / workspace / .metadata / .plugins / org.eclipse.core.resources / .history / b5 / b0c6e469f16500111e5bcf098ae37dc9
1 package com.yuji.tdb;\r
2 \r
3 import java.io.IOException;\r
4 import java.util.logging.Logger;\r
5 \r
6 import javax.servlet.http.HttpServlet;\r
7 import javax.servlet.http.HttpServletRequest;\r
8 import javax.servlet.http.HttpServletResponse;\r
9 \r
10 import twitter4j.Twitter;\r
11 import twitter4j.TwitterException;\r
12 import twitter4j.TwitterFactory;\r
13 import twitter4j.auth.AccessToken;\r
14 import twitter4j.auth.OAuthAuthorization;\r
15 import twitter4j.auth.RequestToken;\r
16 import twitter4j.conf.Configuration;\r
17 import twitter4j.conf.ConfigurationBuilder;\r
18 \r
19 import com.yuji.tdb.common.CommonUtil;\r
20 import com.yuji.tdb.db.KeyValueDao;\r
21 import com.yuji.tdb.twitter.TwitterUtil;\r
22 \r
23 @SuppressWarnings("serial")\r
24 public class TrainDelayBotServlet extends HttpServlet {\r
25         private static final Logger log = Logger\r
26                         .getLogger(TrainDelayBotServlet.class.getName());\r
27         // private static RequestToken requestToken = null;\r
28         //private static AccessToken accessToken = null;\r
29         // private static String consumerKey = null;\r
30         // private static String consumerSecret = null;\r
31 \r
32         private KeyValueDao dao = KeyValueDao.getInstance();\r
33 \r
34         public void doGet(HttpServletRequest req, HttpServletResponse resp)\r
35                         throws IOException {\r
36                 try {\r
37                         log.info("start");\r
38 \r
39                         resp.setContentType("text/plain");\r
40                         resp.getWriter().println("<h3>traindelaybot Hello, world</h3>");\r
41 \r
42                         String parmConsumerKey = req.getParameter("key");\r
43                         String parmConsumerSecret = req.getParameter("secret");\r
44                         String parmResult = req.getParameter("result");\r
45                         if (!CommonUtil.isNull(parmConsumerKey)\r
46                                         && !CommonUtil.isNull(parmConsumerSecret)) {\r
47                                 setOAuthAuthorization(req, resp, parmConsumerKey,\r
48                                                 parmConsumerSecret);\r
49                                 return;\r
50                         }\r
51 \r
52                         String consumerKey = dao.get(KeyValueDao.KEY_CONSUMER_KEY);\r
53                         String consumerSecret = dao.get(KeyValueDao.KEY_CONSUMER_SECRET);\r
54                         if (CommonUtil.isNull(consumerKey)\r
55                                         || CommonUtil.isNull(consumerSecret)) {\r
56                                 log.warning("001");\r
57                                 return;\r
58                         }\r
59 \r
60                         if (!CommonUtil.isNull(parmResult)) {\r
61                                 String requestTokenKey = dao.get(KeyValueDao.KEY_REQUEST_TOKEN);\r
62                                 String requestTokenSecret = dao\r
63                                                 .get(KeyValueDao.KEY_REQUEST_TOKEN_SECRET);\r
64 \r
65                                 if (CommonUtil.isNull(requestTokenKey)\r
66                                                 || CommonUtil.isNull(requestTokenSecret)) {\r
67                                         log.warning("003");\r
68                                         return;\r
69                                 }\r
70                                 RequestToken requestToken = new RequestToken(requestTokenKey,\r
71                                                 requestTokenSecret);\r
72                                 resultOAuthAuthorization(consumerKey, consumerSecret,\r
73                                                 requestToken);\r
74                                 return;\r
75                         }\r
76 \r
77                         String accessTokenKey = dao.get(KeyValueDao.KEY_ACCESS_TOKEN);\r
78                         String accessTokenSecret = dao.get(KeyValueDao.KEY_ACCESS_TOKEN_SECRET);\r
79                         if (CommonUtil.isNull(accessTokenKey)\r
80                                         || CommonUtil.isNull(accessTokenSecret)) {\r
81                                 log.warning("003");\r
82                                 return;\r
83                         }\r
84 \r
85                         AccessToken accessToken = new AccessToken(requestTokenKey,\r
86                                         accessTokenSecret);\r
87                         TwitterUtil util = TwitterUtil.getInstance();\r
88                         util.main(consumerKey, consumerSecret, accessToken);\r
89                 } finally {\r
90                         log.info("end");\r
91                 }\r
92 \r
93         }\r
94 \r
95         private void setOAuthAuthorization(HttpServletRequest req,\r
96                         HttpServletResponse resp, String parmConsumerKey,\r
97                         String parmConsumerSecret) {\r
98                 try {\r
99                         ConfigurationBuilder confBuilder = new ConfigurationBuilder();\r
100 \r
101                         confBuilder.setDebugEnabled(true);\r
102                         confBuilder.setOAuthConsumerKey(parmConsumerKey);\r
103                         confBuilder.setOAuthConsumerSecret(parmConsumerSecret);\r
104                         Configuration conf = confBuilder.build();\r
105 \r
106                         String url = req.getRequestURL().toString();\r
107 \r
108                         OAuthAuthorization oauth = new OAuthAuthorization(conf);\r
109                         // Twitter\82Ì\94F\8fØ\89æ\96Ê\82©\82ç\82Ì\96ß\82è\90æ\82ð\8ew\92è\82µ\82Ü\82·\81B\r
110                         // AccessToken\83R\83\93\83g\83\8d\81[\83\89\82ð\8ew\92è\82µ\82Ü\82·\81B\r
111                         String callbackURL = url + "?result=yes";\r
112                         // RequestToken requestToken =\r
113                         // oauth.getOAuthRequestToken(callbackURL);\r
114                         RequestToken requestToken = oauth.getOAuthRequestToken(callbackURL);\r
115                         // RequestToken\82ð\83Z\83b\83V\83\87\83\93\82É\95Û\91\82µ\82Ä\82¨\82«\82Ü\82·\81B\r
116                         // sessionScope("RequestToken", requestToken);\r
117                         // Twitter\82Ì\94F\8fØ\89æ\96Ê\82Ö\83\8a\83_\83C\83\8c\83N\83g\82µ\82Ü\82·\81B\r
118 \r
119                         // consumerKey = parmConsumerKey;\r
120                         // consumerSecret = parmConsumerSecret;\r
121                         dao.put(KeyValueDao.KEY_CONSUMER_KEY, parmConsumerKey);\r
122                         dao.put(KeyValueDao.KEY_CONSUMER_SECRET, parmConsumerSecret);\r
123                         dao.put(KeyValueDao.KEY_REQUEST_TOKEN, requestToken.getToken());\r
124                         dao.put(KeyValueDao.KEY_REQUEST_TOKEN_SECRET,\r
125                                         requestToken.getTokenSecret());\r
126 \r
127                         resp.sendRedirect(requestToken.getAuthenticationURL());\r
128                 } catch (TwitterException e) {\r
129                         // TODO Auto-generated catch block\r
130                         e.printStackTrace();\r
131                         log.severe(e.toString());\r
132                 } catch (IOException e) {\r
133                         // TODO Auto-generated catch block\r
134                         e.printStackTrace();\r
135                         log.severe(e.toString());\r
136                 }\r
137         }\r
138 \r
139         private void resultOAuthAuthorization(String consumerKey,\r
140                         String consumerSecret, RequestToken requestToken) {\r
141                 try {\r
142                         // Twitter twitter = new TwitterFactory().getInstance();\r
143                         // String tokenKey = requestToken.getToken();\r
144                         // String tokenSecret = requestToken.getTokenSecret();\r
145 \r
146                         ConfigurationBuilder confBuilder = new ConfigurationBuilder();\r
147                         confBuilder.setDebugEnabled(true).setOAuthConsumerKey(consumerKey)\r
148                                         .setOAuthConsumerSecret(consumerSecret);\r
149                         TwitterFactory factory = new TwitterFactory(confBuilder.build());\r
150                         Twitter twitter = factory.getInstance();\r
151 \r
152                         AccessToken accessToken = twitter.getOAuthAccessToken(requestToken);\r
153                         dao.put(KeyValueDao.KEY_ACCESS_TOKEN, accessToken.getToken());\r
154                         dao.put(KeyValueDao.KEY_ACCESS_TOKEN_SECRET, accessToken.getTokenSecret());\r
155                 } catch (TwitterException e) {\r
156                         // TODO Auto-generated catch block\r
157                         e.printStackTrace();\r
158                         log.severe(e.toString());\r
159                 }\r
160         }\r
161 }\r