mail wip

1 files changed, 82 insertions, 0 deletions

diff --git a/juick-api/src/main/java/com/juick/api/controllers/Post.java b/juick-api/src/main/java/com/juick/api/controllers/Post.java
index 45eb31e7..cbd5ac0c 100644
--- a/juick-api/src/main/java/com/juick/api/controllers/Post.java
+++ b/juick-api/src/main/java/com/juick/api/controllers/Post.java
@@ -11,7 +11,10 @@ import com.juick.service.MessagesService;
 import com.juick.service.SubscriptionService;
 import com.juick.service.UserService;
 import com.juick.util.UserUtils;
+import org.apache.commons.codec.digest.DigestUtils;
+import org.apache.commons.io.IOUtils;
 import org.apache.commons.lang3.StringUtils;
+import org.apache.commons.mail.util.MimeMessageParser;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.core.env.Environment;
@@ -25,10 +28,21 @@ import rocks.xmpp.extensions.nick.model.Nickname;
 import rocks.xmpp.extensions.oob.model.x.OobX;
 
 import javax.inject.Inject;
+import javax.mail.MessagingException;
+import javax.mail.Session;
+import javax.mail.internet.InternetAddress;
+import javax.mail.internet.MimeMessage;
+import javax.servlet.http.HttpServletRequest;
+import java.io.BufferedInputStream;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
 import java.io.IOException;
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.net.URL;
+import java.nio.charset.StandardCharsets;
+import java.util.Properties;
+import java.util.UUID;
 
 /**
  * Created by vt on 24/11/2016.
@@ -193,4 +207,72 @@ public class Post {
         }
         return jmsg;
     }
+
+    Session session = Session.getDefaultInstance(new Properties());
+
+    @PostMapping("/mail")
+    @ResponseStatus(value = HttpStatus.OK)
+    public void processMail(HttpServletRequest req) throws Exception {
+        String data = IOUtils.toString(req.getInputStream(), StandardCharsets.UTF_8);
+        logger.info("got data: {}", data);
+        MimeMessage msg = new MimeMessage(session, IOUtils.toInputStream(data, StandardCharsets.UTF_8));
+        String from = msg.getFrom().length > 1 ? ((InternetAddress) msg.getSender()).getAddress()
+                : ((InternetAddress) msg.getFrom()[0]).getAddress();
+        logger.info("got msg from {}", from);
+        User visitor = userService.getUserByEmail(from);
+        if (!visitor.isAnonymous()) {
+            MimeMessageParser parser = new MimeMessageParser(msg);
+            parser.parse();
+            final String[] body = {parser.getPlainContent()};
+            if (body[0] == null) {
+                parser.getAttachmentList().stream()
+                        .filter(a -> a.getContentType().equals("text/plain")).findFirst()
+                        .ifPresent(a -> {
+                            try {
+                                body[0] = IOUtils.toString(a.getInputStream(), StandardCharsets.UTF_8);
+                                logger.info("got text: {}", body[0]);
+                            } catch (IOException e) {
+                                logger.info("attachment error: {}", e);
+                            }
+                        });
+            }
+            final String[] attachmentFName = new String[1];
+            parser.getAttachmentList().stream().filter(a ->
+                    a.getContentType().equals("image/jpeg") || a.getContentType().equals("image/png"))
+                    .findFirst().ifPresent(a -> {
+                logger.info("got attachment: {}", a.getContentType());
+                String attachmentType;
+                if (a.getContentType().equals("image/jpeg")) {
+                    attachmentType = "jpg";
+                } else {
+                    attachmentType = "png";
+                }
+                attachmentFName[0] = DigestUtils.md5Hex(UUID.randomUUID().toString()) + "." + attachmentType;
+                try {
+                    logger.info("got inputstream: {}", a.getInputStream());
+                    FileOutputStream fos = new FileOutputStream("/var/www/juick.com/i/tmp/" + attachmentFName[0]);
+                    IOUtils.copy(a.getInputStream(), fos);
+                    fos.close();
+                } catch (IOException e) {
+                    logger.info("attachment error: {}", e);
+                }
+            });
+            rocks.xmpp.core.stanza.model.Message xmsg = new rocks.xmpp.core.stanza.model.Message();
+            xmsg.setType(rocks.xmpp.core.stanza.model.Message.Type.CHAT);
+            xmsg.setFrom(Jid.of(String.valueOf(visitor.getUid()), "uid.juick.com", "mail"));
+            xmsg.setTo(Jid.of("juick@juick.com/Juick"));
+            xmsg.setBody(body[0]);
+            try {
+                if (StringUtils.isNotEmpty(attachmentFName[0])) {
+                    String attachmentUrl = String.format("juick://%s", attachmentFName[0]);
+                    xmsg.addExtension(new OobX(new URI(attachmentUrl), "!!!!Juick!!"));
+                }
+                apiServer.getXmpp().sendMessage(xmsg);
+            } catch (URISyntaxException e1) {
+                logger.warn("attachment error", e1);
+            }
+        } else {
+            logger.info("not registered: {}", from);
+        }
+    }
 }