package com.juick.api.controllers;

import com.juick.Tag;
import com.juick.User;
import com.juick.api.ApiServer;
import com.juick.api.util.HttpBadRequestException;
import com.juick.api.util.HttpForbiddenException;
import com.juick.api.util.HttpNotFoundException;
import com.juick.api.util.HttpUtils;
import com.juick.service.MessagesService;
import com.juick.service.SubscriptionService;
import com.juick.service.TagService;
import com.juick.service.UserService;
import com.juick.util.UserUtils;
import net.coobird.thumbnailator.Thumbnails;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.env.Environment;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
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.inject.Inject;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.security.Principal;
import java.util.ArrayList;
import java.util.List;

/**
 * Created by vt on 24/11/2016.
 */
@Controller
@ResponseBody
public class Post {
    private static Logger logger = LoggerFactory.getLogger(ApiServer.class);

    @Inject
    UserService userService;
    @Inject
    ApiServer apiServer;
    @Inject
    TagService tagService;
    @Inject
    MessagesService messagesService;
    @Inject
    SubscriptionService subscriptionService;
    @Inject
    Environment env;

    @RequestMapping(value = "/post", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public com.juick.Message doPostMessage(
            Principal principal,
            @RequestParam String body,
            @RequestParam String tags,
            @RequestParam(required = false) String img,
            @RequestParam(required = false) MultipartFile attach) throws IOException {
        String name = UserUtils.getUsername(principal, null);
        User visitor = userService.getUserByName(name);
        int vuid = visitor.getUid();
        if (vuid == 0) {
            throw new HttpForbiddenException();
        }
        if (body == null || body.length() < 1 || body.length() > 4096) {
            throw new HttpBadRequestException();
        }
        body = body.replace("\r", "");

        List<Tag> tagsList = new ArrayList<>();
        String tagsArr[] = new String[1];
        if (tags != null && !tags.isEmpty()) {
            tagsArr = tags.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);
                }
            }
            tagsList = tagService.getTags(tagsArr, true);
            while (tagsList.size() > 5) {
                tagsList.remove(5);
            }
        }

        String attachmentFName = HttpUtils.receiveMultiPartFile(attach, env.getProperty("upload_tmp_dir",
                "/var/www/juick.com/i/tmp/"));

        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 = attachmentFName != null ? attachmentFName.substring(attachmentFName.length() - 3) : null;
        int mid = messagesService.createMessage(vuid, body, attachmentType, tagsList);
        subscriptionService.subscribeMessage(mid, vuid);
        com.juick.Message jmsg = messagesService.getMessage(mid);
        if (apiServer.getXmpp() != null) {
            Message xmsg = new Message();
            xmsg.setFrom(Jid.of("juick@juick.com"));
            xmsg.setType(Message.Type.CHAT);
            xmsg.setThread("juick-" + 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(apiServer.imgDir, "p", fname);
                Files.move(Paths.get(apiServer.tmpDir, attachmentFName), origName);
                Thumbnails.of(origName.toFile()).size(1024, 1024).outputQuality(0.9)
                        .toFile(Paths.get(apiServer.imgDir, "photos-1024", fname).toFile());
                Thumbnails.of(origName.toFile()).size(512, 512).outputQuality(0.9)
                        .toFile(Paths.get(apiServer.imgDir, "photos-512", fname).toFile());
                Thumbnails.of(origName.toFile()).size(160, 120).outputQuality(0.9)
                        .toFile(Paths.get(apiServer.imgDir, "ps", fname).toFile());

                body = attachmentURL + "\n" + body;
                try {
                    OobX xoob = new OobX(new URI(attachmentURL));
                    xmsg.addExtension(xoob);
                } catch (URISyntaxException e) {
                    logger.error("invalid uri: {}, exception {}", attachmentURL, e);
                }
            }

            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"));
            apiServer.getXmpp().send(xmsg);

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

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

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

            xmsg.setTo(Jid.of("jubo@nologin.ru"));
            apiServer.getXmpp().send(xmsg);
        } else {
            logger.error("XMPP unavailable");
        }
        return jmsg;
    }

    @RequestMapping(value = "/comment", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public com.juick.Message doPostComment(
            Principal principal,
            @RequestParam(defaultValue = "0") int mid,
            @RequestParam(defaultValue = "0") int rid,
            @RequestParam String body,
            @RequestParam(required = false) String img,
            @RequestParam(required = false) MultipartFile attach)
            throws IOException {
        String name = UserUtils.getUsername(principal, null);
        User visitor = userService.getUserByName(name);
        int vuid = visitor.getUid();
        if (vuid == 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 == null || body.length() < 1 || body.length() > 4096) {
            throw new HttpBadRequestException();
        }
        body = body.replace("\r", "");

        if ((msg.ReadOnly && msg.getUser().getUid() != vuid) || userService.isInBLAny(msg.getUser().getUid(), vuid)
                || (reply != null && userService.isInBLAny(reply.getUser().getUid(), vuid))) {
            throw new HttpForbiddenException();
        }

        String attachmentFName = HttpUtils.receiveMultiPartFile(attach, env.getProperty("upload_tmp_dir",
                "/var/www/juick.com/i/tmp/"));

        if (StringUtils.isBlank(attachmentFName) && img != null && img.length() > 10) {
            try {
                attachmentFName = HttpUtils.downloadImage(new URL(img));
            } catch (Exception e) {
                logger.error("DOWNLOAD ERROR", e);
                throw new HttpBadRequestException();
            }
        }

        String attachmentType = attachmentFName != null ? attachmentFName.substring(attachmentFName.length() - 3) : null;
        int ridnew = messagesService.createReply(mid, rid, vuid, body, attachmentType);
        subscriptionService.subscribeMessage(mid, vuid);

        com.juick.Message jmsg = messagesService.getReply(mid, ridnew);

        if (apiServer.getXmpp() != null) {
            Message xmsg = new Message();
            xmsg.setFrom(Jid.of("juick@juick.com"));
            xmsg.setType(Message.Type.CHAT);
            xmsg.setThread("juick-" + mid);
            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(apiServer.imgDir, "p", fname);
                Files.move(Paths.get(apiServer.tmpDir, attachmentFName), origName);
                Thumbnails.of(origName.toFile()).size(1024, 1024).outputQuality(0.9)
                        .toFile(Paths.get(apiServer.imgDir, "photos-1024", fname).toFile());
                Thumbnails.of(origName.toFile()).size(512, 512).outputQuality(0.9)
                        .toFile(Paths.get(apiServer.imgDir, "photos-512", fname).toFile());
                Thumbnails.of(origName.toFile()).size(160, 120).outputQuality(0.9)
                        .toFile(Paths.get(apiServer.imgDir, "ps", fname).toFile());

                body = attachmentURL + "\n" + body;
                try {
                    xmsg.addExtension(new OobX(new URI(attachmentURL)));
                } catch (URISyntaxException e) {
                    logger.error("invalid uri: {}, exception {}", attachmentURL, e);
                }
            }

            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"));
            apiServer.getXmpp().send(xmsg);

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

            xmsg.setTo(Jid.of("juick@push.juick.com"));
            apiServer.getXmpp().send(xmsg);
        } else {
            logger.error("XMPP unavailable");
        }
        return jmsg;
    }
}