ConnectOut-sdp/sdp2023

package com.sdpteam.connectout.chat;

import static com.sdpteam.connectout.profile.Profile.NULL_USER;

import java.util.function.BiConsumer;
import java.util.function.Consumer;
import java.util.function.Function;

import com.firebase.ui.database.FirebaseListOptions;
import com.google.android.material.floatingactionbutton.FloatingActionButton;
import com.sdpteam.connectout.R;
import com.sdpteam.connectout.authentication.AuthenticatedUser;
import com.sdpteam.connectout.authentication.GoogleAuth;
import com.sdpteam.connectout.remoteStorage.FileStorageFirebase;
import com.sdpteam.connectout.remoteStorage.ImageSelectionFragment;
import com.squareup.picasso.Picasso;

import android.content.Context;
import android.content.Intent;
import android.graphics.PorterDuff;
import android.graphics.drawable.Drawable;
import android.net.Uri;
import android.os.Bundle;
import android.text.format.DateFormat;
import android.view.Gravity;
import android.view.MenuItem;
import android.view.View;
import android.widget.AbsListView;
import android.widget.EditText;
import android.widget.ImageView;
import android.widget.ListAdapter;
import android.widget.ListView;
import android.widget.TextView;
import androidx.annotation.NonNull;
import androidx.appcompat.app.AppCompatActivity;
import androidx.core.graphics.drawable.DrawableCompat;
import androidx.fragment.app.FragmentTransaction;

public class ChatActivity extends AppCompatActivity {

    public static final String NULL_CHAT = "null_chat";
    private final String YOU = "You";
    private final ImageSelectionFragment imageSelectionFragment = new ImageSelectionFragment(R.drawable.mountain_image);
    public ChatViewModel viewModel = new ChatViewModel(new ChatFirebaseDataSource());
    public final String userName = viewModel.getProfileUserName();
    public AuthenticatedUser au = new GoogleAuth().loggedUser();
    public String uid = (au == null) ? NULL_USER : au.uid;
    public String chatId;
    private View inputRow2; // when wanting to send an image a view will appear over the text field
    private Uri selectedImage = null;

    private FloatingActionButton selectImageBtn;
    private FloatingActionButton deselectImageBtn;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_chat);

        chatId = getIntent().getStringExtra("chatId");
        chatId = (chatId == null) ? NULL_CHAT : chatId;

        //we enable the return button in the ActionBar
        getSupportActionBar().setDisplayHomeAsUpEnabled(true);

        EditText input = findViewById(R.id.chat_input);
        FloatingActionButton chatFab = findViewById(R.id.chat_fab);

        chatFab.setOnClickListener((view) -> {
            if (selectedImage == null) {
                viewModel.saveMessage(new ChatMessage(userName, uid, input.getText().toString(), chatId));
                input.setText("");
            } else {
                FileStorageFirebase storageFirebase = new FileStorageFirebase();
                storageFirebase.uploadFile(selectedImage, "jpg").thenAccept(uri -> {
                    viewModel.saveMessage(new ChatMessage(userName, uid, input.getText().toString(), chatId, uri.toString()));
                    imageSelectionFragment.reset();
                    input.setText("");
                    hideSelectImage();
                });
            }
        });

        //set up the ListView
        setUpListAdapter();

        setUpImageSelectionFragment();

        setUpImageSelectionButtons();

        hideSelectImage(); // by default hide
    }

    private void setUpImageSelectionButtons() {
        inputRow2 = findViewById(R.id.messageInputRow2);

        selectImageBtn = findViewById(R.id.select_image_button);
        selectImageBtn.setOnClickListener(e -> selectImage());

        deselectImageBtn = findViewById(R.id.deselect_image_button);
        deselectImageBtn.setOnClickListener(v -> hideSelectImage());
    }

    private void selectImage() {
        imageSelectionFragment.performOpenSelection();
        inputRow2.setVisibility(View.VISIBLE);
        selectImageBtn.setVisibility(View.GONE);
    }

    private void hideSelectImage() {
        inputRow2.setVisibility(View.GONE);
        selectImageBtn.setVisibility(View.VISIBLE);
        selectedImage = null;
    }

    private void setUpImageSelectionFragment() {
        imageSelectionFragment.setOnImageSelectedListener(imageUri -> {
            this.selectedImage = imageUri;
        });
        FragmentTransaction transaction = getSupportFragmentManager().beginTransaction();
        transaction.replace(R.id.chat_image_upload_container, imageSelectionFragment);
        transaction.commit();
    }

    /**
     * Returns to the previous activity if the return button in the Actionbar is pressed
     */
    @Override
    public boolean onOptionsItemSelected(@NonNull MenuItem item) {
        if (item.getItemId() == android.R.id.home) {
            this.finish();
            return true;
        }

        return super.onOptionsItemSelected(item);
    }

    /**
     * The view must not interact directly with Firebase, as such the FirebaseListAdapter is created
     * by the model.
     * However the adapter needs indirect access to the view elements as such we pass lambdas so that
     * the Model has no direct access.
     */
    private void setUpListAdapter() {
        ListView listOfMessages = findViewById(R.id.list_of_messages);
        listOfMessages.setStackFromBottom(true);
        listOfMessages.setTranscriptMode(AbsListView.TRANSCRIPT_MODE_ALWAYS_SCROLL);

        Function<FirebaseListOptions.Builder<ChatMessage>, FirebaseListOptions.Builder<ChatMessage>> setLayout = a -> a.setLayout(R.layout.message);
        Function<FirebaseListOptions.Builder<ChatMessage>, FirebaseListOptions.Builder<ChatMessage>> setLifecycleOwner = a -> a.setLifecycleOwner(this);
        BiConsumer<View, ChatMessage> populateView = populateViewBiConsumer();
        Consumer<ListAdapter> setAdapter = adapter -> listOfMessages.setAdapter(adapter);

        viewModel.setUpListAdapter(setLayout, setLifecycleOwner, populateView, setAdapter, chatId);
    }

    /**
     * Creates the view for a single message
     * <p>
     * Your messages are to the right and others' are to the left
     */
    private BiConsumer<View, ChatMessage> populateViewBiConsumer() {
        return (v, chatMessage) -> {
            TextView messageText = v.findViewById(R.id.message_text);
            TextView messageUser = v.findViewById(R.id.message_user);
            TextView messageTime = v.findViewById(R.id.message_time);
            // Set the text of the message's view
            messageText.setText(chatMessage.getMessageText());
            messageUser.setText(chatMessage.getUserName());
            // Format the date before showing it
            messageTime.setText(DateFormat.format("dd-MM-yyyy (HH:mm:ss)",
                    chatMessage.getMessageTime()));
            messageTime.setGravity(Gravity.END);
            if (chatMessage.getUserId().equals(uid)) {//your messages are to the right of the screen
                messageText.setGravity(Gravity.END);
                messageUser.setText(YOU);
                messageUser.setGravity(Gravity.CENTER);
                Drawable backgroundDrawable = DrawableCompat.wrap(v.getBackground()).mutate();
                backgroundDrawable.setColorFilter(0x58F8F800, PorterDuff.Mode.ADD);
            }

            String imageUrl = chatMessage.getImageUrl();
            ImageView messageImage = v.findViewById(R.id.message_image);
            if (imageUrl.isEmpty()) {
                messageImage.setVisibility(View.GONE);
            } else {
                messageImage.setVisibility(View.VISIBLE);
                Picasso.get().load(imageUrl).into(messageImage);
            }
        };
    }

    public static void openChat(Context context, String eventID) {
        final Intent intent = new Intent(context, ChatActivity.class);
        intent.putExtra("chatId", eventID);
        context.startActivity(intent);
    }
}