scripts.js

// Import Firebase modules
import { initializeApp } from "https://www.gstatic.com/firebasejs/9.6.1/firebase-app.js";
import { 
    getAuth, 
    createUserWithEmailAndPassword, 
    signInWithEmailAndPassword, 
    signOut, 
    onAuthStateChanged 
} from "https://www.gstatic.com/firebasejs/9.6.1/firebase-auth.js";
import { 
    getFirestore, 
    collection, 
    addDoc, 
    doc, 
    setDoc, 
    onSnapshot, 
    query, 
    where, 
    serverTimestamp,
    getDoc,
    getDocs // Add this import
} from "https://www.gstatic.com/firebasejs/9.6.1/firebase-firestore.js";

// Firebase Configuration
const firebaseConfig = {
    apiKey: "AIzaSyAn_zNl4kzmMIXPjG0EE9IJTUwPEXK_mL8",
    authDomain: "valohq.firebaseapp.com",
    databaseURL: "https://valohq-default-rtdb.asia-southeast1.firebasedatabase.app",
    projectId: "valohq",
    storageBucket: "valohq.firebasestorage.app",
    messagingSenderId: "368885437583",
    appId: "1:368885437583:web:7124c203a9466cfc25ccfa",
    measurementId: "G-FQRMHJ06YQ"
};

// Initialize Firebase
const app = initializeApp(firebaseConfig);
const auth = getAuth(app);
const db = getFirestore(app);

// Authentication Functions
export function signUp(email, password, displayName, riotId) {
    createUserWithEmailAndPassword(auth, email, password)
        .then((userCredential) => {
            console.log('User registered:', userCredential.user.email);
            saveUserData(userCredential.user, displayName, riotId); // Save user data
        })
        .catch((error) => {
            console.error('Error signing up:', error.message);
        });
}

export function signIn(email, password) {
    signInWithEmailAndPassword(auth, email, password)
        .then((userCredential) => {
            console.log('User logged in:', userCredential.user.email);
            window.location.href = '/'; // Redirect to homepage
        })
        .catch((error) => {
            console.error('Error signing in:', error.message);
        });
}

export function signOutUser() {
    signOut(auth)
        .then(() => {
            console.log('User signed out');
            window.location.href = '/login.html'; // Redirect to login page after sign out
        })
        .catch((error) => {
            console.error('Error signing out:', error.message);
        });
}

// Save User Data to Firestore
export function saveUserData(user, displayName, riotId) {
    setDoc(doc(db, 'users', user.uid), {
        email: user.email,
        displayName: displayName,
        riotId: riotId
    }).then(() => {
        console.log('User data saved to Firestore');
        window.location.href = '/'; // Redirect to homepage
    }).catch((error) => {
        console.error('Error saving user data:', error);
    });
}

// Fetch User Data from Firestore
export function fetchUserData(userId) {
    getDoc(doc(db, 'users', userId))
        .then((doc) => {
            if (doc.exists()) {
                console.log('User data:', doc.data());
            } else {
                console.error('User data not found');
            }
        })
        .catch((error) => {
            console.error('Error fetching user data:', error);
        });
}

// Send Invite
export function sendInvite(toUserId) {
    const fromUserId = auth.currentUser?.uid;

    if (!fromUserId) {
        console.error('User not logged in');
        return;
    }

    // Fetch the sender's display name
    getDoc(doc(db, 'users', fromUserId))
        .then((doc) => {
            if (doc.exists()) {
                const senderData = doc.data();
                const senderDisplayName = senderData.displayName;

                // Send the invite with the sender's display name
                addDoc(collection(db, 'invites'), {
                    from: fromUserId,
                    fromDisplayName: senderDisplayName, // Add sender's display name
                    to: toUserId,
                    timestamp: serverTimestamp()
                }).then(() => {
                    console.log('Invite sent');
                }).catch((error) => {
                    console.error('Error sending invite:', error);
                });
            } else {
                console.error('Sender data not found');
            }
        })
        .catch((error) => {
            console.error('Error fetching sender data:', error);
        });
}

// Listen for Invites
let unsubscribeInvites = null; // Store the unsubscribe function for the Firestore listener

export function listenForInvites() {
    const currentUserId = auth.currentUser?.uid;

    if (!currentUserId) {
        console.error('User not logged in');
        return;
    }

    const q = query(collection(db, 'invites'), where('to', '==', currentUserId));
    unsubscribeInvites = onSnapshot(q, (snapshot) => {
        console.log('Received snapshot:', snapshot); // Log the snapshot
        snapshot.docChanges().forEach((change) => {
            if (change.type === 'added') {
                const invite = change.doc.data();
                console.log('New invite from:', invite.fromDisplayName);
                showInvitePrompt(invite); // Pass the entire invite object
            }
        });
    }, (error) => {
        console.error('Error listening for invites:', error);
    });
}

// Show Invite Prompt
export function showInvitePrompt(invite) {
    const prompt = document.getElementById('invitePrompt');
    const inviteFrom = document.getElementById('inviteFrom');

    if (!prompt || !inviteFrom) {
        console.error('Invite prompt elements not found');
        return;
    }

    // Display the sender's display name
    inviteFrom.textContent = invite.fromDisplayName;
    prompt.style.display = 'block';

    document.getElementById('acceptInvite').addEventListener('click', () => {
        alert('You accepted the invite from ' + invite.fromDisplayName);
        prompt.style.display = 'none';
    });

    document.getElementById('rejectInvite').addEventListener('click', () => {
        alert('You rejected the invite from ' + invite.fromDisplayName);
        prompt.style.display = 'none';
    });
}

// Fetch recipient's UID
export function getUserIdByEmail(email) {
    const q = query(collection(db, 'users'), where('email', '==', email));
    return getDocs(q)
        .then((querySnapshot) => {
            if (!querySnapshot.empty) {
                return querySnapshot.docs[0].id; // Return the first matching user's UID
            } else {
                console.error('User not found');
                return null;
            }
        })
        .catch((error) => {
            console.error('Error fetching user ID:', error);
            return null;
        });
}

// Initialize App
export function initApp() {
    onAuthStateChanged(auth, (user) => {
        const currentPage = window.location.pathname.split('/').pop();

        if (user) {
            console.log('User is signed in:', user.email);
            fetchUserData(user.uid); // Fetch user data
            listenForInvites(); // Listen for invites

            // Update UI
            const userEmailElement = document.getElementById('userEmail');
            const loginButton = document.getElementById('loginButton');

            if (userEmailElement && loginButton) {
                userEmailElement.textContent = user.email;
                loginButton.textContent = 'Logout';
                loginButton.href = '#'; // Prevent default behavior
                loginButton.addEventListener('click', () => {
                    signOutUser();
                });
            }

            // Example: Send an invite to a recipient
            const recipientEmail = 'recipient@example.com'; // Replace with the recipient's email
            getUserIdByEmail(recipientEmail)
                .then((toUserId) => {
                    if (toUserId) {
                        sendInvite(toUserId); // Send the invite
                    } else {
                        console.error('Recipient not found');
                    }
                });
        } else {
            console.log('User is signed out');
            // Redirect to login page only if not already on login or register page
            if (currentPage !== 'login.html' && currentPage !== 'register.html') {
                window.location.href = '/login.html';
            }
        }
    });
}

// Cleanup Firestore listeners when the page is unloaded
window.addEventListener('beforeunload', () => {
    if (unsubscribeInvites) {
        unsubscribeInvites(); // Unsubscribe from the Firestore listener
    }
});