From 16508000a77d767c190dcd113195eed8c5070e75 Mon Sep 17 00:00:00 2001
From: NorbiPeti <szatmari.norbert.peter@gmail.com>
Date: Mon, 4 Jun 2018 21:20:40 +0200
Subject: [PATCH] Saving it to CSV just like that

---
 src/main/java/MainHeh.java | 66 +++++++++++++++++++++-----------------
 1 file changed, 37 insertions(+), 29 deletions(-)

diff --git a/src/main/java/MainHeh.java b/src/main/java/MainHeh.java
index 0ead1ad..199a971 100644
--- a/src/main/java/MainHeh.java
+++ b/src/main/java/MainHeh.java
@@ -3,52 +3,60 @@ import sx.blah.discord.api.IDiscordClient;
 import sx.blah.discord.api.events.IListener;
 import sx.blah.discord.handle.impl.events.ReadyEvent;
 import sx.blah.discord.handle.obj.IChannel;
-import sx.blah.discord.handle.obj.IMessage;
 import sx.blah.discord.util.MessageHistory;
+import sx.blah.discord.util.RequestBuffer;
 
 import java.io.File;
 import java.io.IOException;
 import java.nio.file.Files;
+import java.nio.file.StandardOpenOption;
 import java.time.Instant;
-import java.util.concurrent.atomic.AtomicInteger;
-import java.util.function.Consumer;
+import java.util.function.BiConsumer;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 
 public class MainHeh {
     public static void main(String... args) throws IOException {
         IDiscordClient idc = new ClientBuilder().withToken(Files.readAllLines(new File("Token.txt").toPath()).get(0))
                 .registerListener((IListener<ReadyEvent>) event -> {
                     IChannel chan = event.getClient().getChannelByID(249663564057411596L);
-                    final AtomicInteger x = new AtomicInteger();
-                    /*while(true) {
-                        chan.getMessageHistory(100).stream()
-                                .filter(msg -> msg.getContent().matches(
-                                        "(\\S+( \\(formerly.+\\))* joined the game)" +
-                                        "|(\\S+ left the game)")
-                                || (msg.getEmbeds().size()>0&&msg.getEmbeds().get(0).getTitle()!=null
-                                        &&msg.getEmbeds().get(0).getTitle().matches(
-                                        "(Server recovered from a crash - chat connected\\.)"
-                                )))
-                                .map(IMessage::getContent).forEach(System.out::println);
-                        //break;
-                        chan.getMessageHistoryFrom()
-                    }*/
-                    Consumer<MessageHistory> cmh = mh -> {
-                        mh.stream()
-                                .filter(msg -> msg.getContent().matches(
-                                        "(\\S+( \\(formerly.+\\))* joined the game)" +
-                                                "|(\\S+ left the game)")
-                                        || (msg.getEmbeds().size() > 0 && msg.getEmbeds().get(0).getTitle() != null
-                                        && msg.getEmbeds().get(0).getTitle().matches(
-                                        "(Server recovered from a crash - chat connected\\.)"
-                                )))
-                                .map(IMessage::getContent).forEach(System.out::println);
+                    Pattern patplayer = Pattern.compile("(\\S+).*");
+                    File f = new File("data.csv");
+                    f.delete();
+                    long mc = 0;
+                    BiConsumer<MessageHistory, Boolean> cmh = (mh, skip) -> {
+                        RequestBuffer.request(() -> { //? eh
+                            try {
+                                Files.write(f.toPath(),
+                                        (Iterable<String>) mh.stream().skip(skip ? 1 : 0) //Skip 'from' message to avoid double processing
+                                                .filter(msg -> msg.getContent().matches(
+                                                        "(\\S+( \\(formerly.+\\))* joined the game)" +
+                                                                "|(\\S+ left the game)")
+                                                        || (msg.getEmbeds().size() > 0 && msg.getEmbeds().get(0).getTitle() != null
+                                                        && msg.getEmbeds().get(0).getTitle().matches(
+                                                        "(Server recovered from a crash - chat connected\\.)"
+                                                )))
+                                                .map(msg -> {
+                                                    Matcher m = patplayer.matcher(msg.getContent());
+                                                    return (m.matches()
+                                                            ? m.group(1) + (msg.getContent().contains("joined") ? "\tjoin" : "\tleave")
+                                                            : "EVERYBODY\tLEFT")
+                                                            + "\t" + msg.getTimestamp().toString();
+                                                })::iterator, StandardOpenOption.CREATE, StandardOpenOption.APPEND);
+                            } catch (IOException e) {
+                                e.printStackTrace();
+                            }
+                        }).get(); //Wait for completion
                     };
                     MessageHistory mh = chan.getMessageHistoryFrom(Instant.now(), 100);
-                    cmh.accept(mh);
+                    cmh.accept(mh, false);
+                    System.out.println("Processed " + (mc += mh.size()) + " messages.");
                     while (mh.size() > 0) {
+                        cmh.accept(mh, true);
                         mh = chan.getMessageHistoryFrom(mh.get(mh.size() - 1).getLongID(), 100);
-                        cmh.accept(mh);
+                        System.out.println("Processed " + (mc += mh.size()) + " messages.");
                     }
+                    System.out.println("DONE!");
                 }).login();
     }
 }