package io.github.fentonmartin.network.firebase;

import android.support.annotation.NonNull;
import android.support.annotation.Nullable;

import com.google.android.gms.tasks.OnCompleteListener;
import com.google.android.gms.tasks.OnFailureListener;
import com.google.android.gms.tasks.OnSuccessListener;
import com.google.android.gms.tasks.Task;
import com.google.firebase.firestore.DocumentReference;
import com.google.firebase.firestore.DocumentSnapshot;
import com.google.firebase.firestore.EventListener;
import com.google.firebase.firestore.FirebaseFirestore;
import com.google.firebase.firestore.FirebaseFirestoreException;
import com.google.firebase.firestore.ListenerRegistration;
import com.google.firebase.firestore.QuerySnapshot;
import com.google.firebase.firestore.SetOptions;

import java.util.Map;

public class FirebaseUtils {

    public static void sendData(String collection, String document, Map<String, Object> data, FirebaseCallback callback) {
        FirebaseFirestore db = FirebaseFirestore.getInstance();
        db.collection(collection)
                .document(document)
                .set(data, SetOptions.merge())
                .addOnSuccessListener(new OnSuccessListener<Void>() {
                    @Override
                    public void onSuccess(Void aVoid) {
                        if (callback != null)
                            callback.onSuccess(null);
                    }
                })
                .addOnFailureListener(new OnFailureListener() {
                    @Override
                    public void onFailure(@NonNull Exception e) {
                        if (callback != null)
                            callback.onFailure(e);
                    }
                });
    }

    public static void readData(String collection, String id, FirebaseCallback callback) {
        FirebaseFirestore db = FirebaseFirestore.getInstance();
        db.collection(collection)
                .document(id)
                .get()
                .addOnSuccessListener(new OnSuccessListener<DocumentSnapshot>() {
                    @Override
                    public void onSuccess(DocumentSnapshot documentSnapshot) {
                        if (callback != null)
                            callback.onSuccess(documentSnapshot);
                    }
                })
                .addOnFailureListener(new OnFailureListener() {
                    @Override
                    public void onFailure(@NonNull Exception e) {
                        if (callback != null)
                            callback.onFailure(e);
                    }
                });
    }

    public static void readCollection(String collection, FirebaseCollectionCallback callback) {
        FirebaseFirestore db = FirebaseFirestore.getInstance();
        db.collection(collection)
                .get()
                .addOnCompleteListener(new OnCompleteListener<QuerySnapshot>() {
                    @Override
                    public void onComplete(@NonNull Task<QuerySnapshot> task) {
                        callback.onComplete(task);
                    }
                });
    }

    private static ListenerRegistration registration;

    public static void realtimeCollection(String collection, String id, FirebaseCallback callback) {
        FirebaseFirestore db = FirebaseFirestore.getInstance();
        DocumentReference docRef = db.collection(collection).document(id);
        if (registration != null) registration.remove();
        registration = docRef.addSnapshotListener(new EventListener<DocumentSnapshot>() {
            @Override
            public void onEvent(@Nullable DocumentSnapshot snapshot,
                                @Nullable FirebaseFirestoreException e) {
                if (callback != null) {
                    callback.onSuccess(snapshot);
                    callback.onFailure(e);
                }
            }
        });
    }

    public static void realtimeCollectionRemove() {
        if (registration != null) registration.remove();
    }
}