package com.juick.api.controllers;

import com.juick.User;
import com.juick.api.ApiServer;
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.MessagesService;
import com.juick.service.SubscriptionService;
import com.juick.service.UserService;
import com.juick.util.UserUtils;
import org.apache.commons.io.FilenameUtils;
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.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.*;
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 java.awt.image.BufferedImage;
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;

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

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

    @RequestMapping(value = "/post", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    @ResponseStatus(value = HttpStatus.OK)
    public void doPostMessage(
            @RequestParam String body,
            @RequestParam(required = false) String img,
            @RequestParam(required = false) MultipartFile attach) throws IOException {
        User visitor = UserUtils.getCurrentUser();

        if (visitor.isAnonymous())
            throw new HttpForbiddenException();

        if (body == null || body.length() < 1 || body.length() > 4096) {
            throw new HttpBadRequestException();
        }
        body = body.replace("\r", StringUtils.EMPTY);

        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();
            }
        }

        if (apiServer.getXmpp() != null) {
            Message xmsg = new Message();
            xmsg.setFrom(Jid.of(String.valueOf(visitor.getUid()), "uid.juick.com", "perl"));
            xmsg.setTo(Jid.of("juick@juick.com/Juick"));
            xmsg.setBody(body);
            try {
                if (StringUtils.isNotEmpty(attachmentFName)) {
                    String attachmentUrl = String.format("juick://%s", attachmentFName);
                    xmsg.addExtension(new OobX(new URI(attachmentUrl), "!!!!Juick!!"));
                }
                apiServer.getXmpp().sendMessage(xmsg);
            } catch (URISyntaxException e1) {
                logger.warn("attachment error", e1);
            }
        }
    }

    @RequestMapping(value = "/comment", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public com.juick.Message doPostComment(
            @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 {
        User visitor = UserUtils.getCurrentUser();
        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", StringUtils.EMPTY);

        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 = StringUtils.isNotEmpty(attachmentFName) ? 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 (StringUtils.isNotEmpty(attachmentFName)) {
                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);
                BufferedImage originalImage = ImageIO.read(origName.toFile());
                ImageIO.write(Scalr.resize(originalImage, 1024),
                        FilenameUtils.getExtension(origName.toString()),
                        Paths.get(apiServer.imgDir, "photos-1024", fname).toFile());
                ImageIO.write(Scalr.resize(originalImage, 512),
                        FilenameUtils.getExtension(origName.toString()),
                        Paths.get(apiServer.imgDir, "photos-512", fname).toFile());
                ImageIO.write(Scalr.resize(originalImage, 160),
                        FilenameUtils.getExtension(origName.toString()),
                        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;
    }
}