path: root/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java

/*
 * 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 <http://www.gnu.org/licenses/>.
 */
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, "Написать", "<script src=\"//maps.google.com/maps?file=api&amp;v=2&amp;sensor=false&amp;key=ABQIAAAAVVtPtxkw4soCEHg44FsNChRB4OFYjAXt73He16Zkp6a_0tPs2RTU6i6UlcMs4QvPBYvIY8rWvcxqOg\" type=\"text/javascript\"></script>"
                    + "<script src=\"//static.juick.com/mc.js\" type=\"text/javascript\" defer=\"defer\"></script>"
                    + "<script src=\"//static.juick.com/maps.js?2010111500\" type=\"text/javascript\" defer=\"defer\"></script>"
                    + "<script src=\"//static.juick.com/post3.js\" type=\"text/javascript\" defer=\"defer\"></script>");
            templates.pageNavigation(out, visitor, null);

            out.println("<section id=\"content\" class=\"pagetext\">");

            out.println("<form action=\"/post2\" method=\"post\" id=\"postmsg\" enctype=\"multipart/form-data\">");
            out.println("<p style=\"text-align: left\"><b>Место: <span id=\"location\"></span></b> <span id=\"locationclear\">&mdash; <a href=\"#\" onclick=\"clearLocation()\">Отменить</a></span></p>");
            out.println("<p style=\"text-align: left\"><b>Фото:</b> <span id=\"attachmentfile\"><input type=\"file\" name=\"attach\"/> <i>(JPG, PNG, до 10Мб)</i></span></p>");

            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("<p><textarea name=\"body\" class=\"newmessage\" rows=\"7\" cols=\"10\">" + body + "</textarea><br/>");

            out.println("<input type=\"hidden\" name=\"place_id\"/>" + "" + "<input type=\"submit\" class=\"subm\" value=\"   Отправить   \"/></p>");
            out.println("</form>");
            out.println("<div id=\"geomap\"></div>");
            out.println("<p style=\"text-align: left\"><b>Теги:</b></p>");
            printUserTags(out, visitor);
            out.println("</section>");

            templates.pageFooter(request, out, visitor, false);
            templates.pageEnd(out);
        }
    }

    void printUserTags(PrintWriter out, com.juick.User visitor) {
        List<TagStats> 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("<p style=\"text-align: justify\">");
        for (int i = 0; i < tags.size(); i++) {
            if (i > 0) {
                out.print(" ");
            }
            String taglink = StringUtils.EMPTY;
            try {
                taglink = "<a onclick=\"return addTag('" + StringEscapeUtils.escapeHtml4(tags.get(i).getTag().getName()) + "')\" href=\"/" +
                        visitor.getName() + "/?tag=" + URLEncoder.encode(tags.get(i).getTag().getName(), CharEncoding.UTF_8) +
                        "\" title=\"" + tags.get(i).getUsageCount() + "\">" + StringEscapeUtils.escapeHtml4(tags.get(i).getTag().getName()) + "</a>";
            } catch (UnsupportedEncodingException e) {
            }
            int usagecnt = tags.get(i).getUsageCount();
            if (usagecnt <= max / 5 + min) {
                out.print("<span style=\"font-size: small\">" + taglink + "</span>");
            } else if (usagecnt <= max / 5 * 2 + min) {
                out.print(taglink);
            } else if (usagecnt <= max / 5 * 3 + min) {
                out.print("<span style=\"font-size: large\">" + taglink + "</span>");
            } else if (usagecnt <= max / 5 * 4 + min) {
                out.print("<span style=\"font-size: x-large\">" + taglink + "</span>");
            } else {
                out.print("<span style=\"font-size: xx-large\">" + taglink + "</span>");
            }
        }
        out.println("</p>");
    }

    @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<Tag> 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("<section id=\"content\">");
            out.println("<h1>Сообщение опубликовано</h1>");
            out.println("<p>Поделитесь своим новым постом в социальных сетях:</p>");
            if (crosspostService.getTwitterTokens(visitor.getUid()).isPresent()) {
                out.println("<p class=\"social\"><a href=\"https://twitter.com/intent/tweet?text=" + URLEncoder.encode(sharetwi, CharEncoding.UTF_8) + "\" class=\"ico32-twi sharenew\">Отправить в Twitter</a></p>");
            }
            out.println("<p class=\"social\"><a href=\"http://www.livejournal.com/update.bml?subject=" + URLEncoder.encode(hashtags, CharEncoding.UTF_8) + "&event=" + sharelj + "&prop_taglist=" + URLEncoder.encode(tagscomma, CharEncoding.UTF_8) + "\" target=\"_blank\" class=\"ico32-lj sharenew\">Отправить в LiveJournal</a></p>");
            out.println("<p class=\"social\"><a href=\"https://vk.com/share.php?url=" + url + "\" class=\"ico32-vk sharenew\">Отправить в ВКонтакте</a></p>");
            if (crosspostService.getFacebookToken(visitor.getUid()).isPresent()) {
                out.println("<p class=\"social\"><a href=\"https://www.facebook.com/sharer/sharer.php?u=" + url + "\" class=\"ico32-fb sharenew\">Отправить в Facebook</a></p>");
            }
            out.println("<p class=\"social\"><a href=\"https://plus.google.com/share?url=" + url + "\" class=\"ico32-gp sharenew\">Отправить в Google+</a></p>");
            out.println("<p>Ссылка на сообщение: <a href=\"http://juick.com/" + mid + "\">http://juick.com/" + mid + "</a></p>");
            out.println("</section>");

            templates.pageHomeColumn(out, visitor);

            templates.pageFooter(request, out, visitor, false);
            templates.pageEnd(out);
        }
    }

    @PostMapping("/comment")
    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();
        }
    }
}