diff options
author | Vitaly Takmazov | 2017-10-24 17:18:29 +0300 |
---|---|---|
committer | Vitaly Takmazov | 2017-10-24 22:34:41 +0300 |
commit | de2cc2db73a5de42d9bfaeb92604f28abe2a328a (patch) | |
tree | 352a3d76fcbc987f068527ff043174308a666654 /juick-crosspost/src/main/java/com/juick/service | |
parent | 881a546bf7a56550eec1e38ce839f47a1cb1a750 (diff) |
crosspost: refactoring
* spring-social
* get tokens via service api
Diffstat (limited to 'juick-crosspost/src/main/java/com/juick/service')
-rw-r--r-- | juick-crosspost/src/main/java/com/juick/service/Crosspost.java | 164 | ||||
-rw-r--r-- | juick-crosspost/src/main/java/com/juick/service/rest/CrosspostRestService.java | 143 |
2 files changed, 307 insertions, 0 deletions
diff --git a/juick-crosspost/src/main/java/com/juick/service/Crosspost.java b/juick-crosspost/src/main/java/com/juick/service/Crosspost.java new file mode 100644 index 00000000..a8458439 --- /dev/null +++ b/juick-crosspost/src/main/java/com/juick/service/Crosspost.java @@ -0,0 +1,164 @@ +/* + * Copyright (C) 2008-2017, Juick + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as + * published by the Free Software Foundation, either version 3 of the + * License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ +package com.juick.service; + +import com.fasterxml.jackson.databind.ObjectMapper; +import com.juick.Message; +import com.juick.util.MessageUtils; +import org.apache.commons.codec.CharEncoding; +import org.apache.commons.io.IOUtils; +import org.apache.commons.lang3.StringUtils; +import org.apache.commons.lang3.tuple.Pair; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.social.twitter.api.Twitter; +import org.springframework.social.twitter.api.impl.TwitterTemplate; +import org.springframework.web.socket.TextMessage; +import org.springframework.web.socket.WebSocketSession; +import org.springframework.web.socket.handler.TextWebSocketHandler; + +import javax.inject.Inject; +import javax.net.ssl.HttpsURLConnection; +import java.io.OutputStreamWriter; +import java.net.URL; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; + +/** + * @author Ugnich Anton + */ +public class Crosspost extends TextWebSocketHandler { + final static String FBURL = "https://graph.facebook.com/me/feed"; + final static String VKURL = "https://api.vk.com/method/wall.post"; + + private static Logger logger = LoggerFactory.getLogger(Crosspost.class); + + @Inject + private CrosspostService crosspostService; + + @Value("${twitter_consumer_key:}") + private String twitter_consumer_key; + @Value("${twitter_consumer_secret:}") + private String twitter_consumer_secret; + @Inject + private ObjectMapper jsonMapper; + + @Override + protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { + Message msg = jsonMapper.readValue(message.asBytes(), Message.class); + if (msg.getMid() > 0 && msg.getRid() == 0) { + if (StringUtils.isNotEmpty(crosspostService.getTwitterName(msg.getUser().getUid()))) { + if (msg.getTags().stream().noneMatch(t -> t.getName().equals("notwitter"))) { + twitterPost(msg); + } + } + // TODO: approve application for facebook crosspost + } + } + + public boolean facebookPost(final com.juick.Message jmsg) { + String token = crosspostService.getFacebookToken(jmsg.getUser().getUid()).orElse(StringUtils.EMPTY); + if (token.isEmpty()) { + return false; + } + + logger.info("FB: #{}", jmsg.getMid()); + + String status = MessageUtils.getMessageHashTags(jmsg) + "\n" + jmsg.getText(); + + boolean ret = false; + try { + String body = "access_token=" + + URLEncoder.encode(token, CharEncoding.UTF_8) + + "&message=" + + URLEncoder.encode(status, CharEncoding.UTF_8) + + "&link=http%3A%2F%2Fjuick.com%2F" + + jmsg.getMid(); + + HttpsURLConnection conn = (HttpsURLConnection) new URL(FBURL).openConnection(); + conn.setRequestProperty("Content-Type", "application/x-www-form-urlencoded"); + conn.setRequestProperty("User-Agent", "Juick"); + conn.setRequestProperty("Content-Length", Integer.toString(body.length())); + conn.setUseCaches(false); + conn.setDoInput(true); + conn.setDoOutput(true); + conn.setRequestMethod("POST"); + conn.connect(); + + OutputStreamWriter wr = new OutputStreamWriter(conn.getOutputStream()); + wr.write(body); + wr.close(); + + ret = StringUtils.isNotEmpty(IOUtils.toString(conn.getInputStream(), StandardCharsets.UTF_8)); + + conn.disconnect(); + } catch (Exception e) { + logger.error("fbPost exception", e); + } + return ret; + } + + public boolean vkontaktePost(final com.juick.Message jmsg) { + Pair<String, String> tokens = crosspostService.getVkTokens(jmsg.getUser().getUid()).orElse(Pair.of(StringUtils.EMPTY, StringUtils.EMPTY)); + if (tokens.getLeft().isEmpty() || tokens.getRight().isEmpty()) { + return false; + } + + logger.info("VK: #", jmsg.getMid()); + + String status = MessageUtils.getMessageHashTags(jmsg) + "\n" + jmsg.getText() + "\nhttp://juick.com/" + jmsg.getMid(); + + boolean ret = false; + try { + String body = "owner_id=" + tokens.getLeft() + "&access_token=" + URLEncoder.encode(tokens.getRight(), CharEncoding.UTF_8) + "&from_group=1&message=" + URLEncoder.encode(status, CharEncoding.UTF_8); + + HttpsURLConnection conn = (HttpsURLConnection) new URL(VKURL).openConnection(); + conn.setRequestProperty("Content-Type", "application/x-www-form-urlencoded"); + conn.setRequestProperty("User-Agent", "Juick"); + conn.setRequestProperty("Content-Length", Integer.toString(body.length())); + conn.setUseCaches(false); + conn.setDoInput(true); + conn.setDoOutput(true); + conn.setRequestMethod("POST"); + conn.connect(); + + OutputStreamWriter wr = new OutputStreamWriter(conn.getOutputStream()); + wr.write(body); + wr.close(); + + ret = StringUtils.isNotEmpty(IOUtils.toString(conn.getInputStream(), StandardCharsets.UTF_8)); + + conn.disconnect(); + } catch (Exception e) { + logger.error("vkPost exception", e); + } + return ret; + } + + public void twitterPost(final com.juick.Message jmsg) { + crosspostService.getTwitterToken(jmsg.getUser().getUid()).ifPresent(t -> { + String status = MessageUtils.getMessageHashTags(jmsg) + jmsg.getText(); + if (status.length() > 115) { + status = status.substring(0, 114) + "…"; + } + status += " http://juick.com/" + jmsg.getMid(); + Twitter twitter = new TwitterTemplate(twitter_consumer_key, twitter_consumer_secret, t.getToken(), t.getSecret()); + twitter.timelineOperations().updateStatus(status); + }); + } +} diff --git a/juick-crosspost/src/main/java/com/juick/service/rest/CrosspostRestService.java b/juick-crosspost/src/main/java/com/juick/service/rest/CrosspostRestService.java new file mode 100644 index 00000000..52b72c3f --- /dev/null +++ b/juick-crosspost/src/main/java/com/juick/service/rest/CrosspostRestService.java @@ -0,0 +1,143 @@ +package com.juick.service.rest; + +import com.juick.ExternalToken; +import com.juick.User; +import com.juick.server.helpers.ApplicationStatus; +import com.juick.service.CrosspostService; +import org.apache.commons.lang3.StringUtils; +import org.apache.commons.lang3.tuple.Pair; +import org.springframework.stereotype.Component; +import org.springframework.web.client.RestTemplate; + +import javax.inject.Inject; +import java.util.Optional; + +@Component +public class CrosspostRestService implements CrosspostService { + @Inject + private RestTemplate rest; + + @Override + public Optional<ExternalToken> getTwitterToken(int uid) { + User user = rest.getForObject("https://api.juick.com/tokens?uid={uid}", User.class, uid); + if (user != null) { + return user.getTokens().stream() + .filter(t -> t.getType().equals("twitter")).findFirst(); + } + return Optional.empty(); + } + + @Override + public boolean deleteTwitterToken(Integer uid) { + return false; + } + + @Override + public Optional<String> getFacebookToken(int uid) { + return null; + } + + @Override + public ApplicationStatus getFbCrossPostStatus(int uid) { + return null; + } + + @Override + public boolean enableFBCrosspost(Integer uid) { + return false; + } + + @Override + public void disableFBCrosspost(Integer uid) { + + } + + @Override + public String getTwitterName(int uid) { + ExternalToken token = getTwitterToken(uid).orElse(null); + if (token != null) { + return token.getName(); + } + return StringUtils.EMPTY; + } + + @Override + public String getTelegramName(int uid) { + return null; + } + + @Override + public Optional<Pair<String, String>> getVkTokens(int uid) { + return null; + } + + @Override + public void deleteVKUser(Integer uid) { + + } + + @Override + public int getUIDbyFBID(long fbID) { + return 0; + } + + @Override + public boolean createFacebookUser(long fbID, String loginhash, String token, String fbName, String fbLink) { + return false; + } + + @Override + public boolean updateFacebookUser(long fbID, String token, String fbName, String fbLink) { + return false; + } + + @Override + public int getUIDbyVKID(long vkID) { + return 0; + } + + @Override + public boolean createVKUser(long vkID, String loginhash, String token, String vkName, String vkLink) { + return false; + } + + @Override + public String getFacebookNameByHash(String hash) { + return null; + } + + @Override + public String getTelegramNameByHash(String hash) { + return null; + } + + @Override + public boolean setFacebookUser(String hash, int uid) { + return false; + } + + @Override + public String getVKNameByHash(String hash) { + return null; + } + + @Override + public boolean setVKUser(String hash, int uid) { + return false; + } + + @Override + public boolean setTelegramUser(String hash, int uid) { + return false; + } + + @Override + public String getJIDByHash(String hash) { + return null; + } + + @Override + public boolean setJIDUser(String hash, int uid) { + return false; + } +} |