Telegram-Final-Project/src/main/java/org/to/telegramfinalproject/Database/ContactDatabase.java

package org.to.telegramfinalproject.Database;

import org.to.telegramfinalproject.Models.Contact;
import org.to.telegramfinalproject.Models.User;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

public class ContactDatabase {
    public ContactDatabase(){}


    private static Connection getConnection() throws SQLException {
        return ConnectionDb.connect();
    }

    public static boolean addContact(UUID userId, UUID contactId) {
        String sql = """
        INSERT INTO contacts (user_id, contact_id)
        VALUES (?, ?)
        ON CONFLICT DO NOTHING
    """;

        try (Connection conn = ConnectionDb.connect();
             PreparedStatement stmt = conn.prepareStatement(sql)) {
            stmt.setObject(1, userId);
            stmt.setObject(2, contactId);
            return stmt.executeUpdate() > 0;
        } catch (SQLException e) {
            e.printStackTrace();
            return false;
        }
    }



    public boolean removeContact(UUID user_id, UUID contact_id)  {
        String sql = "DELETE FROM contacts WHERE user_id = ? AND contact_id = ?";
        try (Connection connection = getConnection()) {
            PreparedStatement stmt = connection.prepareStatement(sql);
            stmt.setObject(1, user_id);
            stmt.setObject(2, contact_id);
            return stmt.executeUpdate() > 0;
        } catch (SQLException e) {
            e.printStackTrace();
            return false;
        }
    }

    public boolean blockContact(UUID user_id, UUID contact_id) {
        String sql = "UPDATE contacts SET is_blocked = TRUE WHERE user_id = ? AND contact_id = ?";
        try (Connection connection = getConnection()) {
            PreparedStatement stmt = connection.prepareStatement(sql);
            stmt.setObject(1, user_id);
            stmt.setObject(2, contact_id);
            return stmt.executeUpdate() > 0;
        } catch (SQLException e) {
            e.printStackTrace();
            return false;
        }
    }

    public boolean unblockContact(UUID user_id, UUID contact_id) {
        String sql = "UPDATE contacts SET is_blocked = FALSE WHERE user_id = ? AND contact_id = ?";
        try (Connection connection = getConnection()) {
            PreparedStatement stmt = connection.prepareStatement(sql);
            stmt.setObject(1, user_id);
            stmt.setObject(2, contact_id);
            return stmt.executeUpdate() > 0;
        } catch (SQLException e) {
            e.printStackTrace();
            return false;
        }
    }

    public static List<Contact> getContacts(UUID user_id) {
        List<Contact> contacts = new ArrayList<>();
        String sql = "SELECT * FROM contacts WHERE user_id = ?";
        try (Connection connection = getConnection()) {
            PreparedStatement stmt = connection.prepareStatement(sql);
            stmt.setObject(1, user_id);
            ResultSet rs = stmt.executeQuery();
            while (rs.next()) {
                UUID contactId = (UUID) rs.getObject("contact_id");
                boolean isBlocked = rs.getBoolean("is_blocked");
                Timestamp addedAt = rs.getTimestamp("added_at");

                Contact contact = new Contact(user_id, contactId);
                contact.setIs_blocked(isBlocked);
                contact.setAdd_at(addedAt.toLocalDateTime());
                contacts.add(contact);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return contacts;
    }


    public static boolean existsContact(UUID user_id, UUID contact_id) {
        String sql = "SELECT 1 FROM contacts WHERE user_id = ? AND contact_id = ? LIMIT 1"; // stop searching when find the first item in DB(LIMIT 1)
        try (Connection connection = getConnection()) {
            PreparedStatement stmt = connection.prepareStatement(sql);
            stmt.setObject(1, user_id);
            stmt.setObject(2, contact_id);
            ResultSet rs = stmt.executeQuery();
            return rs.next();
        } catch (SQLException e) {
            e.printStackTrace();
            return false;
        }
    }


    public boolean isBlocked(UUID user_id, UUID contact_id) {
        String sql = "SELECT is_blocked FROM contacts WHERE user_id = ? AND contact_id = ?";
        try (Connection connection = getConnection()) {
            PreparedStatement stmt = connection.prepareStatement(sql);
            stmt.setObject(1, user_id);
            stmt.setObject(2, contact_id);
            ResultSet rs = stmt.executeQuery();
            if (rs.next()) {
                return rs.getBoolean("is_blocked");
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return false;
    }

    public List<Contact> searchContacts(UUID user_id, String searchTerm) {
        List<Contact> results = new ArrayList<>(); //ILIKE case-insensitive
        String sql = """
        SELECT c.contact_id, c.added_at, c.is_blocked
        FROM contacts c
        JOIN users u ON c.contact_id = u.internal_uuid
        WHERE c.user_id = ? AND (u.user_id ILIKE ? OR u.profile_name ILIKE ?)
        """;
        try (Connection connection = getConnection()) {
            PreparedStatement stmt = connection.prepareStatement(sql);
            stmt.setObject(1, user_id);
            stmt.setString(2, "%" + searchTerm + "%");
            stmt.setString(3, "%" + searchTerm + "%");
            ResultSet rs = stmt.executeQuery();
            while (rs.next()) {
                UUID contactId = (UUID) rs.getObject("contact_id");
                boolean isBlocked = rs.getBoolean("is_blocked");
                Timestamp addedAt = rs.getTimestamp("added_at");
                Contact contact = new Contact(user_id, contactId);
                contact.setIs_blocked(isBlocked);
                contact.setAdd_at(addedAt.toLocalDateTime());
                results.add(contact);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return results;
    }




}