Saving user and bot message to database

joelcuy · joelcuy · commit fec2e9bc1767 · 2021-12-29T21:18:04.000+08:00
diff --git a/app/src/main/java/com/example/nobulijava/activity/UserChatBotActivity.java b/app/src/main/java/com/example/nobulijava/activity/UserChatBotActivity.java
@@ -1,7 +1,6 @@
 package com.example.nobulijava.activity;
 
-import static android.view.View.GONE;
-
+import androidx.annotation.NonNull;
 import androidx.appcompat.app.AppCompatActivity;
 import androidx.recyclerview.widget.RecyclerView;
 
@@ -18,6 +17,8 @@
 import com.example.nobulijava.interfaces.BotReply;
 import com.example.nobulijava.model.MessageObj;
 import com.example.nobulijava.utils.SendMessageInBg;
+import com.google.android.gms.tasks.OnCompleteListener;
+import com.google.android.gms.tasks.Task;
 import com.google.api.gax.core.FixedCredentialsProvider;
 import com.google.auth.oauth2.GoogleCredentials;
 import com.google.auth.oauth2.ServiceAccountCredentials;
@@ -28,6 +29,9 @@
 import com.google.cloud.dialogflow.v2.SessionsSettings;
 import com.google.cloud.dialogflow.v2.TextInput;
 import com.google.common.collect.Lists;
+import com.google.firebase.database.DataSnapshot;
+import com.google.firebase.database.DatabaseReference;
+import com.google.firebase.database.FirebaseDatabase;
 
 import java.io.InputStream;
 import java.util.ArrayList;
@@ -43,7 +47,9 @@ public class UserChatBotActivity extends AppCompatActivity implements BotReply {
     TextView textViewDisclaimer;
 
     MessageAdapter messageAdapter;
-    List<MessageObj> messageList = new ArrayList<>();
+    List<MessageObj> messageObjArrayList = new ArrayList<>();
+
+    private DatabaseReference mDatabase;
 
     //dialogFlow
     private SessionsClient sessionsClient;
@@ -58,27 +64,41 @@ protected void onCreate(Bundle savedInstanceState) {
 
         getSupportActionBar().setTitle("Nobuli Chatbot");
 
+        mDatabase = FirebaseDatabase.getInstance().getReference();
+
         recyclerViewChat = findViewById(R.id.chatRecyclerView);
         editTextMessageBox = findViewById(R.id.messageBoxEditText);
         btnSend = findViewById(R.id.sendButton);
         textViewDisclaimer = findViewById(R.id.textView_userChatBot_textDisclaimer);
 
-        messageAdapter = new MessageAdapter(messageList, this);
+        messageAdapter = new MessageAdapter(messageObjArrayList, this);
         recyclerViewChat.setAdapter(messageAdapter);
 
+        mDatabase.child("Message").get().addOnCompleteListener(new OnCompleteListener<DataSnapshot>() {
+            @Override
+            public void onComplete(@NonNull Task<DataSnapshot> task) {
+                for (DataSnapshot msgSnapshot: task.getResult().getChildren()){
+                    MessageObj msgObj = msgSnapshot.getValue(MessageObj.class);
+                    messageObjArrayList.add(msgObj);
+                }
+                Objects.requireNonNull(recyclerViewChat.getAdapter()).notifyDataSetChanged();
+            }
+        });
+
         btnSend.setOnClickListener(new View.OnClickListener() {
             @Override
             public void onClick(View view) {
-                textViewDisclaimer.setVisibility(GONE);
-                String message = editTextMessageBox.getText().toString();
+                textViewDisclaimer.setVisibility(View.GONE);
+                String message = editTextMessageBox.getText().toString().trim();
                 if (!message.isEmpty()) {
-
-                    messageList.add(new MessageObj(message, false));
+                    MessageObj newMessage = new MessageObj(message, false);
+                    messageObjArrayList.add(newMessage);
+                    mDatabase.child("Message").push().setValue(newMessage);
                     editTextMessageBox.setText("");
                     sendMessageToBot(message);
                     Objects.requireNonNull(recyclerViewChat.getAdapter()).notifyDataSetChanged();
                     Objects.requireNonNull(recyclerViewChat.getLayoutManager())
-                            .scrollToPosition(messageList.size() - 1);
+                            .scrollToPosition(messageObjArrayList.size() - 1);
                 } else {
                     Toast.makeText(UserChatBotActivity.this, "Please enter text!", Toast.LENGTH_SHORT).show();
                 }
@@ -118,13 +138,15 @@ public void callback(DetectIntentResponse returnResponse) {
         if (returnResponse != null) {
             String botReply = returnResponse.getQueryResult().getFulfillmentText();
             if (!botReply.isEmpty()) {
-                messageList.add(new MessageObj(botReply, true, returnResponse.getQueryResult().getIntent().getDisplayName()));
+                MessageObj newBotMsg = new MessageObj(botReply, true, returnResponse.getQueryResult().getIntent().getDisplayName());
+                mDatabase.child("Message").push().setValue(newBotMsg);
+                messageObjArrayList.add(newBotMsg);
                 messageAdapter.notifyDataSetChanged();
                 if (returnResponse.getQueryResult().getIntent().getDisplayName().equals("Identify Cyberbully - report")){
-                    messageList.add(new MessageObj("Is there anything I can help you with?", true, "null"));
+                    messageObjArrayList.add(new MessageObj("Is there anything I can help you with?", true, "null"));
                 }
                 messageAdapter.notifyDataSetChanged();
-                Objects.requireNonNull(recyclerViewChat.getLayoutManager()).scrollToPosition(messageList.size() - 1);
+                Objects.requireNonNull(recyclerViewChat.getLayoutManager()).scrollToPosition(messageObjArrayList.size() - 1);
             } else {
                 Toast.makeText(this, "Something went wrong", Toast.LENGTH_SHORT).show();
             }
diff --git a/app/src/main/java/com/example/nobulijava/model/MessageObj.java b/app/src/main/java/com/example/nobulijava/model/MessageObj.java
@@ -11,6 +11,9 @@ public MessageObj(String text, boolean isBot) {
         this.isBot = isBot;
     }
 
+    public MessageObj() {
+    }
+
     public MessageObj(String text, boolean isBot, String intent) {
         this.text = text;
         this.isBot = isBot;

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,9 @@ public MessageObj(String text, boolean isBot) {`
`11`	`11`	`this.isBot = isBot;`
`12`	`12`	`}`
`13`	`13`
	`14`	`+ public MessageObj() {`
	`15`	`+ }`
	`16`	`+`
`14`	`17`	`public MessageObj(String text, boolean isBot, String intent) {`
`15`	`18`	`this.text = text;`
`16`	`19`	`this.isBot = isBot;`