/* * Juick * Copyright (C) 2008-2011, Ugnich Anton * * 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 . */ package com.juick.www.controllers; import com.juick.Status; import com.juick.Tag; import com.juick.server.helpers.TagStats; import com.juick.server.util.HttpBadRequestException; import com.juick.server.util.HttpForbiddenException; import com.juick.server.util.HttpNotFoundException; import com.juick.server.util.HttpUtils; import com.juick.service.*; import com.juick.util.UserUtils; import com.juick.www.Utils; import com.juick.www.WebApp; import org.apache.commons.io.FilenameUtils; import org.apache.commons.lang3.CharEncoding; import org.apache.commons.lang3.StringEscapeUtils; import org.apache.commons.lang3.StringUtils; import org.imgscalr.Scalr; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.core.env.Environment; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.multipart.MultipartFile; import rocks.xmpp.addr.Jid; import rocks.xmpp.core.stanza.model.Message; import rocks.xmpp.extensions.nick.model.Nickname; import rocks.xmpp.extensions.oob.model.x.OobX; import javax.imageio.ImageIO; import javax.inject.Inject; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.awt.image.BufferedImage; import java.io.IOException; import java.io.PrintWriter; import java.io.UnsupportedEncodingException; import java.net.URI; import java.net.URISyntaxException; import java.net.URL; import java.net.URLEncoder; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.util.List; /** * @author Ugnich Anton */ @Controller public class NewMessage { @Inject Environment env; @Inject TagService tagService; @Inject MessagesService messagesService; @Inject UserService userService; @Inject SubscriptionService subscriptionService; @Inject CrosspostService crosspostService; @Inject WebApp webApp; @Inject PageTemplates templates; private static final Logger logger = LoggerFactory.getLogger(NewMessage.class); @GetMapping("/post") protected void doGetNewMessage(HttpServletRequest request, HttpServletResponse response) throws IOException { com.juick.User visitor = UserUtils.getCurrentUser(); if (visitor.getUid() == 0) { Utils.sendTemporaryRedirect(response, "/login"); return; } response.setContentType("text/html; charset=UTF-8"); try (PrintWriter out = response.getWriter()) { templates.pageHead(out, visitor, "Написать", "" + "" + "" + ""); templates.pageNavigation(out, visitor, null); out.println("
"); if (messagesService.isReadonly()) { out.println("

Временно недоступно

"); } else { out.println("
"); out.println("

Место: Отменить

"); out.println("

Фото: (JPG, PNG, до 10Мб)

"); String body = request.getParameter("body"); if (body == null) { body = StringUtils.EMPTY; } else { if (body.length() > 4096) { body = body.substring(0, 4096); } body = StringEscapeUtils.escapeHtml4(body); } out.println("


"); out.println("" + "" + "

"); out.println("
"); out.println("
"); out.println("

Теги:

"); printUserTags(out, visitor); } out.println("
"); templates.pageFooter(request, out, visitor, false); templates.pageEnd(out); } } void printUserTags(PrintWriter out, com.juick.User visitor) { List tags = tagService.getUserTagStats(visitor.getUid()); if (tags.isEmpty()) { return; } int min = tags.get(0).getUsageCount(); int max = tags.get(0).getUsageCount(); for (int i = 1; i < tags.size(); i++) { int usagecnt = tags.get(i).getUsageCount(); if (usagecnt < min) { min = usagecnt; } if (usagecnt > max) { max = usagecnt; } } max -= min; out.print("

"); for (int i = 0; i < tags.size(); i++) { if (i > 0) { out.print(" "); } String taglink = StringUtils.EMPTY; try { taglink = "" + StringEscapeUtils.escapeHtml4(tags.get(i).getTag().getName()) + ""; } catch (UnsupportedEncodingException e) { } int usagecnt = tags.get(i).getUsageCount(); if (usagecnt <= max / 5 + min) { out.print("" + taglink + ""); } else if (usagecnt <= max / 5 * 2 + min) { out.print(taglink); } else if (usagecnt <= max / 5 * 3 + min) { out.print("" + taglink + ""); } else if (usagecnt <= max / 5 * 4 + min) { out.print("" + taglink + ""); } else { out.print("" + taglink + ""); } } out.println("

"); } @PostMapping("/post") public void doPostMessage(HttpServletRequest request, HttpServletResponse response, @RequestParam(required = false) String img, @RequestParam String body, @RequestParam(required = false) MultipartFile attach) throws IOException { com.juick.User visitor = UserUtils.getCurrentUser(); if (visitor.getUid() == 0) { response.sendError(HttpServletResponse.SC_FORBIDDEN); return; } if (body == null || body.length() < 1 || body.length() > 4096) { response.sendError(HttpServletResponse.SC_BAD_REQUEST); return; } body = body.replace("\r", StringUtils.EMPTY); List tags = webApp.parseTags(request.getParameter("tags")); String attachmentFName = HttpUtils.receiveMultiPartFile(attach, webApp.getTmpDir()); if (StringUtils.isBlank(attachmentFName) && img != null && img.length() > 10) { try { URL imgUrl = new URL(img); attachmentFName = HttpUtils.downloadImage(imgUrl); } catch (Exception e) { logger.error("DOWNLOAD ERROR", e); throw new HttpBadRequestException(); } } String attachmentType = StringUtils.isNotEmpty(attachmentFName) ? attachmentFName.substring(attachmentFName.length() - 3) : null; int mid = messagesService.createMessage(visitor.getUid(), body, attachmentType, tags); subscriptionService.subscribeMessage(mid, visitor.getUid()); Message xmsg = new Message(); xmsg.setFrom(Jid.of("juick@juick.com")); xmsg.setType(Message.Type.CHAT); xmsg.setThread("juick-" + mid); com.juick.Message jmsg = messagesService.getMessage(mid); xmsg.addExtension(jmsg); xmsg.addExtension(new Nickname("@" + jmsg.getUser().getName())); if (StringUtils.isNotEmpty(attachmentFName)) { String fname = mid + "." + attachmentType; String attachmentURL = "http://i.juick.com/photos-1024/" + fname; Path origName = Paths.get(webApp.getImgDir(), "p", fname); Files.move(Paths.get(webApp.getTmpDir(), attachmentFName), origName); BufferedImage originalImage = ImageIO.read(origName.toFile()); ImageIO.write(Scalr.resize(originalImage, 1024), FilenameUtils.getExtension(origName.toString()), Paths.get(webApp.getImgDir(), "photos-1024", fname).toFile()); ImageIO.write(Scalr.resize(originalImage, 512), FilenameUtils.getExtension(origName.toString()), Paths.get(webApp.getImgDir(), "photos-512", fname).toFile()); ImageIO.write(Scalr.resize(originalImage, 160), FilenameUtils.getExtension(origName.toString()), Paths.get(webApp.getImgDir(), "ps", fname).toFile()); body = attachmentURL + "\n" + body; try { xmsg.addExtension(new OobX(new URI(attachmentURL))); } catch (URISyntaxException e) { logger.warn("invalid uri: {} exception {}", attachmentURL, e); } } if (webApp.getXmpp() != null) { xmsg.setBody("@" + jmsg.getUser().getName() + ":" + jmsg.getTagsString() + "\n" + body + "\n\n#" + mid + " http://juick.com/" + mid); xmsg.setTo(Jid.of("juick@s2s.juick.com")); webApp.getXmpp().send(xmsg); xmsg.setTo(Jid.of("juick@ws.juick.com")); webApp.getXmpp().send(xmsg); xmsg.setTo(Jid.of("juick@push.juick.com")); webApp.getXmpp().send(xmsg); xmsg.setTo(Jid.of("jubo@nologin.ru")); webApp.getXmpp().send(xmsg); } else { logger.warn("XMPP unavailable"); } // response.setContentType("text/html; charset=UTF-8"); try (PrintWriter out = response.getWriter()) { templates.pageHead(out, visitor, "Сообщение опубликовано", null); templates.pageNavigation(out, visitor, null); String hashtags = StringUtils.EMPTY; String tagscomma = StringUtils.EMPTY; for (int i = 0; i < jmsg.getTags().size(); i++) { if (i > 0) { hashtags += " "; tagscomma += ","; } hashtags += "#" + jmsg.getTags().get(i); tagscomma += jmsg.getTags().get(i); } String url = URLEncoder.encode("http://juick.com/" + mid, CharEncoding.UTF_8); String sharetwi = hashtags + " " + body; if (sharetwi.length() > 115) { sharetwi = sharetwi.substring(0, 114) + "…"; } sharetwi += " http://juick.com/" + mid; String sharelj = URLEncoder.encode(body + "\n", CharEncoding.UTF_8) + url; out.println("
"); out.println("

Сообщение опубликовано

"); out.println("

Поделитесь своим новым постом в социальных сетях:

"); if (crosspostService.getTwitterTokens(visitor.getUid()).isPresent()) { out.println("

Отправить в Twitter

"); } out.println("

Отправить в LiveJournal

"); out.println("

Отправить в ВКонтакте

"); if (crosspostService.getFacebookToken(visitor.getUid()).isPresent()) { out.println("

Отправить в Facebook

"); } out.println("

Отправить в Google+

"); out.println("

Ссылка на сообщение: http://juick.com/" + mid + "

"); out.println("
"); templates.pageHomeColumn(out, visitor); templates.pageFooter(request, out, visitor, false); templates.pageEnd(out); } } @PostMapping("/comment") @ResponseBody public String doPostComment( @RequestParam(required = false, defaultValue = "0") Integer mid, @RequestParam(required = false, defaultValue = "0") Integer rid, @RequestParam String body, @RequestParam(required = false) String img, @RequestParam(required = false) MultipartFile attach) throws IOException { com.juick.User visitor = UserUtils.getCurrentUser(); if (visitor.getUid() == 0) { throw new HttpForbiddenException(); } if (mid == 0) { throw new HttpBadRequestException(); } com.juick.Message msg = messagesService.getMessage(mid); if (msg == null) { throw new HttpNotFoundException(); } com.juick.Message reply = null; if (rid > 0) { reply = messagesService.getReply(mid, rid); if (reply == null) { throw new HttpNotFoundException(); } } if (body.length() < 1 || body.length() > 4096) { throw new HttpBadRequestException(); } body = body.replace("\r", StringUtils.EMPTY); if ((msg.ReadOnly && msg.getUser().getUid() != visitor.getUid()) || userService.isInBLAny(msg.getUser().getUid(), visitor.getUid()) || (reply != null && userService.isInBLAny(reply.getUser().getUid(), visitor.getUid()))) { throw new HttpForbiddenException(); } String attachmentFName = HttpUtils.receiveMultiPartFile(attach, webApp.getTmpDir()); if (StringUtils.isBlank(attachmentFName) && img != null && img.length() > 10) { try { URL imgUrl = new URL(img); attachmentFName = HttpUtils.downloadImage(imgUrl); } catch (Exception e) { logger.error("DOWNLOAD ERROR", e); throw new HttpBadRequestException(); } } String attachmentType = StringUtils.isNotEmpty(attachmentFName) ? attachmentFName.substring(attachmentFName.length() - 3) : null; int ridnew = messagesService.createReply(mid, rid, visitor.getUid(), body, attachmentType); subscriptionService.subscribeMessage(mid, visitor.getUid()); Message xmsg = new Message(); xmsg.setFrom(Jid.of("juick@juick.com")); xmsg.setType(Message.Type.CHAT); xmsg.setThread("juick-" + mid); com.juick.Message jmsg = messagesService.getReply(mid, ridnew); xmsg.addExtension(jmsg); String quote = reply != null ? reply.getText() : msg.getText(); if (quote.length() >= 50) { quote = quote.substring(0, 47) + "..."; } xmsg.addExtension(new Nickname("@" + jmsg.getUser().getName())); if (StringUtils.isNotEmpty(attachmentFName)) { String fname = mid + "-" + ridnew + "." + attachmentType; String attachmentURL = "http://i.juick.com/photos-1024/" + fname; Path origName = Paths.get(webApp.getImgDir(), "p", fname); Files.move(Paths.get(webApp.getTmpDir(), attachmentFName), origName); BufferedImage originalImage = ImageIO.read(origName.toFile()); ImageIO.write(Scalr.resize(originalImage, 1024), FilenameUtils.getExtension(origName.toString()), Paths.get(webApp.getImgDir(), "photos-1024", fname).toFile()); ImageIO.write(Scalr.resize(originalImage, 512), FilenameUtils.getExtension(origName.toString()), Paths.get(webApp.getImgDir(), "photos-512", fname).toFile()); ImageIO.write(Scalr.resize(originalImage, 160), FilenameUtils.getExtension(origName.toString()), Paths.get(webApp.getImgDir(), "ps", fname).toFile()); body = attachmentURL + "\n" + body; try { xmsg.addExtension(new OobX(new URI(attachmentURL))); } catch (URISyntaxException e) { logger.warn("invalid uri: {}, exception {}", attachmentURL, e); } } if (webApp.getXmpp() != null) { xmsg.setBody("Reply by @" + jmsg.getUser().getName() + ":\n>" + quote + "\n" + body + "\n\n#" + mid + "/" + ridnew + " http://juick.com/" + mid + "#" + ridnew); xmsg.setTo(Jid.of("juick@s2s.juick.com")); webApp.getXmpp().send(xmsg); xmsg.setTo(Jid.of("juick@ws.juick.com")); webApp.getXmpp().send(xmsg); xmsg.setTo(Jid.of("juick@push.juick.com")); webApp.getXmpp().send(xmsg); } else { logger.warn("XMPP unavailable"); } return "redirect:/" + msg.getUser().getName() + "/" + mid + "#" + ridnew; } @PostMapping("/like") @ResponseBody public Status doPostRecomm(@RequestParam Integer mid) throws IOException { com.juick.User visitor = UserUtils.getCurrentUser(); if (visitor.getUid() == 0) { throw new HttpForbiddenException(); } com.juick.Message msg = messagesService.getMessage(mid); if (msg == null) { throw new HttpNotFoundException(); } if (msg.getUser().getUid() == visitor.getUid()) { throw new HttpForbiddenException(); } boolean res = messagesService.recommendMessage(mid, visitor.getUid()); if (res) { if (webApp.getXmpp() != null) { Message xmsg = new Message(); xmsg.setFrom(Jid.of("juick@juick.com")); xmsg.setTo(Jid.of("recomm@s2s.juick.com")); com.juick.Message jmsg = new com.juick.Message(); jmsg.setMid(mid); jmsg.setUser(visitor); xmsg.addExtension(jmsg); webApp.getXmpp().send(xmsg); } else { logger.warn("XMPP unavailable"); } return Status.OK; } else { throw new HttpBadRequestException(); } } }