Reminder functional with User persistance

Neocortexxx
1 parent 60ae7aae
Showing 2 changed files with 34 additions and 1 deletions
services/MainBot/src/main/java/de/bht/beuthbot/drools/DroolsService.java
services/MainBot/src/main/resources/de/bht/beuthbot/drools/Canteen.drl
@@ -10,11 +10,14 @@ import de.bht.beuthbot.attachments.AttachmentStore;
 import de.bht.beuthbot.canteen.Parser;
 import de.bht.beuthbot.canteen.model.CanteenData;
 import de.bht.beuthbot.conf.Application;
+import de.bht.beuthbot.daos.AppUserDAO;
 import de.bht.beuthbot.drools.model.DroolsMessage;
 import de.bht.beuthbot.jms.ProcessQueue;
 import de.bht.beuthbot.jms.ProcessQueueMessageProtocol;
 import de.bht.beuthbot.jms.Target;
 import de.bht.beuthbot.jms.TaskMessage;
+import de.bht.beuthbot.model.Messenger;
+import de.bht.beuthbot.model.entities.AppUser;
 import org.kie.api.KieServices;
 import org.kie.api.runtime.KieContainer;
 import org.kie.api.runtime.KieSession;
@@ -28,6 +31,7 @@ import javax.inject.Inject;
 import javax.jms.JMSException;
 import javax.jms.Message;
 import javax.jms.MessageListener;
+import java.util.List;
 /**
  * Created by sJantzen on 13.06.2017.
@@ -66,6 +70,9 @@ public class DroolsService implements MessageListener {
     @Resource(lookup = "java:global/global/ApplicationBean")
     private Application application;
+    @Resource(lookup = "java:global/global/AppUserDAOImpl")
+    private AppUserDAO userDAO;
+
     @Inject
     private Parser parser;
@@ -114,7 +121,32 @@ public class DroolsService implements MessageListener {
         // org/drools/examples/helloworld/HelloWorld.drl file
         CanteenData canteenData = parser.parse();
+        //TODO userDAO.findByTelegramId & userDAO.findByFacebookId
+        List<AppUser> allUsers = userDAO.findAll();
+        AppUser tempUser = null;
+
+        for (AppUser user : allUsers) {
+            if(botMessage.getMessenger() == Messenger.TELEGRAM) {
+                tempUser = user.getTelegramUserId().equals(botMessage.getSenderID().toString()) ? user : null;
+            } else if(botMessage.getMessenger() == Messenger.FACEBOOK) {
+                tempUser = user.getFacebookUserId().equals(botMessage.getSenderID().toString()) ? user : null;
+            }
+        }
+
+        if(tempUser == null) {
+            tempUser = userDAO.createUser();
+        }
+        if(botMessage.getMessenger() == Messenger.TELEGRAM) {
+            tempUser.setTelegramUserId(botMessage.getSenderID().toString());
+        } else if(botMessage.getMessenger() == Messenger.FACEBOOK) {
+            tempUser.setFacebookUserId(botMessage.getSenderID().toString());
+        }
+        AppUser finishedUser = userDAO.saveOrUpdate(tempUser);
+
+
+
         ksession.setGlobal("canteenData", canteenData);
+        ksession.setGlobal("user", finishedUser);
         // The application can insert facts into the session
         // Map incoming ApiAiMessages and RasaMessages to DroolsMessage
@@ -40,6 +40,7 @@ import de.bht.beuthbot.model.Messenger
 import de.bht.beuthbot.scheduler.model.Reminder;
 global de.bht.beuthbot.canteen.model.CanteenData canteenData;
+global de.bht.beuthbot.model.entities.AppUser user;
 // ---------------------- GREETING ------------------------------------
 rule "Hello Drools"
@@ -125,7 +126,7 @@ rule &quot;Create reminder&quot;
     then
         SchedulerBean s = new SchedulerBean();
         TaskMessage message = new TaskMessage(m);
-        String messageText = s.createReminder(message);
+        String messageText = s.createReminder(message, user);
         m.setText(messageText);
 end