/*
 * 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;

import com.juick.Tag;
import com.juick.server.*;
import com.juick.server.helpers.TagStats;
import net.coobird.thumbnailator.Thumbnails;
import org.apache.commons.lang3.StringEscapeUtils;
import org.apache.commons.lang3.math.NumberUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.core.JdbcTemplate;
import rocks.xmpp.addr.Jid;
import rocks.xmpp.core.session.XmppSession;
import rocks.xmpp.core.stanza.model.Message;
import rocks.xmpp.extensions.nick.model.Nickname;
import rocks.xmpp.extensions.oob.model.x.OobX;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
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.ArrayList;
import java.util.List;

/**
 * @author Ugnich Anton
 */
public class NewMessage {

    private static final Logger logger = LoggerFactory.getLogger(NewMessage.class);

    private String tmpDir, imgDir;

    public NewMessage(String tmpDir, String imgDir) {
        this.tmpDir = tmpDir;
        this.imgDir = imgDir;
    }

    protected void doGetNewMessage(JdbcTemplate sql, HttpServletRequest request, HttpServletResponse response,
                                   com.juick.User visitor) throws ServletException, IOException {
        response.setContentType("text/html; charset=UTF-8");
        try (PrintWriter out = response.getWriter()) {
            PageTemplates.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>");
            PageTemplates.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 = "";
            } 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(sql, out, visitor);
            out.println("</section>");

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

    void printUserTags(JdbcTemplate sql, PrintWriter out, com.juick.User visitor) {
        List<TagStats> tags = TagQueries.getTagsStats(sql, 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 = "";
            try {
                taglink = "<a onclick=\"return addTag('" + StringEscapeUtils.escapeHtml4(tags.get(i).getTag().getName()) + "')\" href=\"/" +
                        visitor.getName() + "/?tag=" + URLEncoder.encode(tags.get(i).getTag().getName(), "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>");
    }

    public void doPostMessage(JdbcTemplate sql, HttpServletRequest request, HttpServletResponse response, XmppSession xmpp, com.juick.User visitor) throws ServletException, IOException {
        String body = request.getParameter("body");
        if (body == null || body.length() < 1 || body.length() > 4096) {
            response.sendError(HttpServletResponse.SC_BAD_REQUEST);
            return;
        }
        body = body.replace("\r", "");

        String tagsStr = request.getParameter("tags");
        List<com.juick.Tag> tags = new ArrayList<Tag>();
        String tagsArr[] = new String[1];
        if (tagsStr != null && !tagsStr.isEmpty()) {
            tagsArr = tagsStr.split("[ \\,]");
            for (int i = 0; i < tagsArr.length; i++) {
                if (tagsArr[i].startsWith("*")) {
                    tagsArr[i] = tagsArr[i].substring(1);
                }
                if (tagsArr[i].length() > 64) {
                    tagsArr[i] = tagsArr[i].substring(0, 64);
                }
            }
            tags = TagQueries.getTags(sql, tagsArr, true);
            while (tags.size() > 5) {
                tags.remove(5);
            }
        }

        String attachmentFName = null;
        try {
            attachmentFName = Utils.receiveMultiPartFile(request, "attach");
        } catch (Exception e) {
            logger.error("MULTIPART ERROR", e);
            response.sendError(HttpServletResponse.SC_BAD_REQUEST);
            return;
        }

        String paramImg = request.getParameter("img");
        if (attachmentFName == null && paramImg != null && paramImg.length() > 10) {
            try {
                URL imgUrl = new URL(paramImg);
                attachmentFName = Utils.downloadImage(imgUrl);
            } catch (Exception e) {
                logger.error("DOWNLOAD ERROR", e);
                response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
                return;
            }
        }

        String attachmentType = attachmentFName != null ? attachmentFName.substring(attachmentFName.length() - 3) : null;
        int mid = MessagesQueries.createMessage(sql, visitor.getUid(), body, attachmentType, tags);
        SubscriptionsQueries.subscribeMessage(sql, 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 = MessagesQueries.getMessage(sql, mid);
        xmsg.addExtension(jmsg);
        xmsg.addExtension(new Nickname("@" + jmsg.getUser().getName()));

        if (attachmentFName != null) {
            String fname = mid + "." + attachmentType;
            String attachmentURL = "http://i.juick.com/photos-1024/" + fname;

            Path origName = Paths.get(imgDir, "p", fname);
            Files.move(Paths.get(tmpDir, attachmentFName), origName);
            Thumbnails.of(origName.toFile()).size(1024, 1024).outputQuality(0.9)
                    .toFile(Paths.get(imgDir, "photos-1024", fname).toFile());
            Thumbnails.of(origName.toFile()).size(512, 512).outputQuality(0.9)
                    .toFile(Paths.get(imgDir, "photos-512", fname).toFile());
            Thumbnails.of(origName.toFile()).size(160, 120).outputQuality(0.9)
                    .toFile(Paths.get(imgDir, "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 (xmpp != null) {

            String tagsStr2 = "";
            for (String tag : tagsArr) {
                tagsStr2 += " *" + tag;
            }
            xmsg.setBody("@" + jmsg.getUser().getName() + ":" + tagsStr2 + "\n" + body + "\n\n#" + mid + " http://juick.com/" + mid);

            xmsg.setTo(Jid.of("juick@s2s.juick.com"));
            xmpp.send(xmsg);

            xmsg.setTo(Jid.of("juick@ws.juick.com"));
            xmpp.send(xmsg);

            xmsg.setTo(Jid.of("juick@push.juick.com"));
            xmpp.send(xmsg);

            xmsg.setTo(Jid.of("twitter@crosspost.juick.com"));
            xmpp.send(xmsg);
            xmsg.setTo(Jid.of("fb@crosspost.juick.com"));
            xmpp.send(xmsg);

            xmsg.setTo(Jid.of("jubo@nologin.ru"));
            xmpp.send(xmsg);
        } else {
            logger.warn("XMPP unavailable");
        }

        //

        response.setContentType("text/html; charset=UTF-8");
        try (PrintWriter out = response.getWriter()) {
            PageTemplates.pageHead(out, visitor, "Сообщение опубликовано", null);
            PageTemplates.pageNavigation(out, visitor, null);

            String hashtags = "";
            String tagscomma = "";
            for (int i = 0; i < tagsArr.length; i++) {
                if (i > 0) {
                    hashtags += " ";
                    tagscomma += ",";
                }
                hashtags += "#" + tagsArr[i];
                tagscomma += tagsArr[i];
            }

            String url = URLEncoder.encode("http://juick.com/" + mid, "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", "utf-8") + url;

            out.println("<section id=\"content\">");
            out.println("<h1>Сообщение опубликовано</h1>");
            out.println("<p>Поделитесь своим новым постом в социальных сетях:</p>");
            if (CrosspostQueries.getTwitterTokens(sql, visitor.getUid()).isPresent()) {
                out.println("<p class=\"social\"><a href=\"https://twitter.com/intent/tweet?text=" + URLEncoder.encode(sharetwi, "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, "utf-8") + "&event=" + sharelj + "&prop_taglist=" + URLEncoder.encode(tagscomma, "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 (CrosspostQueries.getFacebookToken(sql, 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>");

            PageTemplates.pageHomeColumn(out, sql, visitor);

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

    public void doPostComment(JdbcTemplate sql, HttpServletRequest request, HttpServletResponse response, XmppSession xmpp, com.juick.User visitor) throws ServletException, IOException {
        int mid = NumberUtils.toInt(request.getParameter("mid"), 0);
        if (mid == 0) {
            response.sendError(HttpServletResponse.SC_BAD_REQUEST);
            return;
        }
        com.juick.Message msg = MessagesQueries.getMessage(sql, mid);
        if (msg == null) {
            response.sendError(HttpServletResponse.SC_NOT_FOUND);
            return;
        }

        int rid = NumberUtils.toInt(request.getParameter("rid"), 0);
        com.juick.Message reply = null;
        if (rid > 0) {
            reply = MessagesQueries.getReply(sql, mid, rid);
            if (reply == null) {
                response.sendError(HttpServletResponse.SC_NOT_FOUND);
                return;
            }
        }

        String body = request.getParameter("body");
        if (body == null || body.length() < 1 || body.length() > 4096) {
            response.sendError(HttpServletResponse.SC_BAD_REQUEST);
            return;
        }
        body = body.replace("\r", "");

        if ((msg.ReadOnly && msg.getUser().getUid() != visitor.getUid()) || UserQueries.isInBLAny(sql, msg.getUser().getUid(), visitor.getUid()) || (reply != null && UserQueries.isInBLAny(sql, reply.getUser().getUid(), visitor.getUid()))) {
            response.sendError(HttpServletResponse.SC_FORBIDDEN);
            return;
        }

        String attachmentFName = null;
        try {
            attachmentFName = Utils.receiveMultiPartFile(request, "attach");
        } catch (Exception e) {
            logger.error("MULTIPART ERROR", e);
            response.sendError(HttpServletResponse.SC_BAD_REQUEST);
            return;
        }

        String paramImg = request.getParameter("img");
        if (attachmentFName == null && paramImg != null && paramImg.length() > 10) {
            try {
                attachmentFName = Utils.downloadImage(new URL(paramImg));
            } catch (Exception e) {
                logger.error("DOWNLOAD ERROR", e);
                response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
                return;
            }
        }

        String attachmentType = attachmentFName != null ? attachmentFName.substring(attachmentFName.length() - 3) : null;
        int ridnew = MessagesQueries.createReply(sql, mid, rid, visitor.getUid(), body, attachmentType);
        SubscriptionsQueries.subscribeMessage(sql, 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 = MessagesQueries.getReply(sql, 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 (attachmentFName != null) {
            String fname = mid + "-" + ridnew + "." + attachmentType;
            String attachmentURL = "http://i.juick.com/photos-1024/" + fname;

            Path origName = Paths.get(imgDir, "p", fname);
            Files.move(Paths.get(tmpDir, attachmentFName), origName);
            Thumbnails.of(origName.toFile()).size(1024, 1024).outputQuality(0.9)
                    .toFile(Paths.get(imgDir, "photos-1024", fname).toFile());
            Thumbnails.of(origName.toFile()).size(512, 512).outputQuality(0.9)
                    .toFile(Paths.get(imgDir, "photos-512", fname).toFile());
            Thumbnails.of(origName.toFile()).size(160, 120).outputQuality(0.9)
                    .toFile(Paths.get(imgDir, "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 (xmpp != 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"));
            xmpp.send(xmsg);

            xmsg.setTo(Jid.of("juick@ws.juick.com"));
            xmpp.send(xmsg);

            xmsg.setTo(Jid.of("juick@push.juick.com"));
            xmpp.send(xmsg);
        } else {
            logger.warn("XMPP unavailable");
        }

        Utils.sendTemporaryRedirect(response, "/" + msg.getUser().getName() + "/" + mid + "#" + ridnew);
    }

    public void doPostRecomm(JdbcTemplate sql, HttpServletRequest request, HttpServletResponse response, XmppSession xmpp, com.juick.User visitor) throws ServletException, IOException {
        int mid = NumberUtils.toInt(request.getParameter("mid"), 0);
        if (mid == 0) {
            response.sendError(HttpServletResponse.SC_BAD_REQUEST);
            return;
        }
        com.juick.Message msg = MessagesQueries.getMessage(sql, mid);
        if (msg == null) {
            response.sendError(HttpServletResponse.SC_NOT_FOUND);
            return;
        }
        if (msg.getUser().getUid() == visitor.getUid()) {
            response.sendError(HttpServletResponse.SC_FORBIDDEN);
            return;
        }

        boolean res = MessagesQueries.recommendMessage(sql, mid, visitor.getUid());

        if (res) {
            if (xmpp != 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);
                xmpp.send(xmsg);
            } else {
                logger.warn("XMPP unavailable");
            }

            Utils.replyJSON(request, response, "{\"status\":\"ok\"}");
        } else {
            response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
        }
    }
}