package org.to.telegramfinalproject.Client;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.to.telegramfinalproject.Models.Message;

import java.sql.Timestamp;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.UUID;

import static org.to.telegramfinalproject.Client.ActionHandler.sendWithResponse;

public class SidebarHandler {
    private final Scanner scanner;
    private final ActionHandler actionHandler;
    private final String userUUID;

    public SidebarHandler(Scanner scanner, ActionHandler actionHandler) {
        this.scanner = scanner;
        this.actionHandler = actionHandler;
        this.userUUID = Session.getUserUUID();

    }

    public void handleSidebarAction(SidebarAction action) {
        switch (action) {
            case MY_PROFILE:
                openUserProfile();
                break;
            case NEW_GROUP:
                createNewGroup();
                break;
            case NEW_CHANNEL:
                createNewChannel();
                break;
            case CONTACTS:
                showContacts();
                break;
            case SAVED_MESSAGES:
                getSavedMessagesData(userUUID);
                break;
            case SETTINGS:
                openSettings();
                break;
            case FEATURES:
                showTelegramFeatures();
                break;
            case Q_AND_A:
                showTelegramQA();
                break;
        }
    }

    private void openUserProfile() {
        // Step 1: Request profile info from the server
        JSONObject request = new JSONObject();
        request.put("action", "get_user_profile");

        JSONObject response = sendWithResponse(request);
        if (response == null || !response.optString("status", "fail").equals("success")) {
            System.out.println("Failed to load profile information.");
            return;
        }

        JSONObject profile;
        try {
            profile = response.getJSONObject("data");
        } catch (JSONException e) {
            System.out.println("Received malformed profile data from server.");
            return;
        }

        String profileName = profile.optString("profile_name", "NOT-AVAILABLE");
        String userId = profile.optString("user_id", "NOT-AVAILABLE");
        String status = profile.optString("status", "NOT-AVAILABLE");
        String bio = profile.optString("bio", "NOT-AVAILABLE");
        String profilePictureUrl = profile.optString("profile_picture_url", "NOT-AVAILABLE");

        // Step 2: Show profile info
        System.out.println("===== My Profile =====");
        System.out.println("Profile Picture URL : " + profilePictureUrl);
        System.out.println("Profile Name        : " + profileName);
        System.out.println("User ID             : @" + userId);
        System.out.println("Status              : " + status);
        System.out.println("Bio                 : " + bio);
        System.out.println("======================");

        // Step 3: Ask if user wants to edit anything
        System.out.println("Do you want to edit any of these? (yes/no)");
        while (true) {
            String choice = scanner.nextLine().trim().toLowerCase();
            if (choice.equals("no")) return;
            if (!choice.equals("yes")) {
                System.out.println("Invalid input. Please enter 'yes' or 'no'.");
                continue;
            }

            // Step 4: Show editable fields
            System.out.println("Which field do you want to edit?");
            System.out.println("1. Profile Name");
            System.out.println("2. User ID");
            System.out.println("3. Bio");
            System.out.println("4. Profile Picture URL");
            System.out.println("0. Cancel");

            String option = scanner.nextLine().trim();
            switch (option) {
                case "1" -> editProfileName();
                case "2" -> editUserId();
                case "3" -> editBio();
                case "4" -> editProfilePictureUrl();
                case "0" -> { return; }
                default -> System.out.println("Invalid choice. Try again.");
            }

            // Re-fetch and re-display profile after editing
            openUserProfile();
            return;
        }
    }

    private void editProfileName() {
        System.out.println("Enter your new profile name:");
        String newProfileName = scanner.nextLine().trim();

        // Step 1: Validate input
        while (newProfileName.trim().isEmpty()) {
            System.out.println("Profile name cannot be empty. Enter another name.");
            newProfileName = scanner.nextLine().trim();
        }

        // Step 2: Create request
        JSONObject request = new JSONObject();
        request.put("action", "edit_profile_name");
        request.put("new_profile_name", newProfileName);

        // Step 3: Send request and receive response
        JSONObject response = sendWithResponse(request);

        // Step 4: Handle response
        if (response == null || !response.optString("status", "fail").equals("success")) {
            System.out.println("Failed to update profile name.");
        } else {
            System.out.println("Profile name updated successfully!");
        }
    }

    private void editUserId() {
        while (true) {
            System.out.println("Enter your new user ID: ");
            String newUserId = scanner.nextLine().trim();

            if (newUserId.isEmpty()) {
                System.out.println("User ID cannot be empty.");
                continue;
            }

            if (newUserId.contains(" ")) {
                System.out.println("User ID cannot contain spaces.");
                continue;
            }

            if (!newUserId.matches("^[a-zA-Z0-9_]+$")) {
                System.out.println("User ID can only contain letters, digits, and underscores.");
                continue;
            }

            // Send to server
            JSONObject request = new JSONObject();
            request.put("action", "edit_user_id");
            request.put("new_user_id", newUserId);
            JSONObject response = sendWithResponse(request);

            if (response.getString("status").equals("success")) {
                System.out.println("User ID updated successfully.");
                break;
            } else {
                // Server-side error message (e.g., ID already exists)
                System.out.println(response.getString("message"));
            }
        }
    }

    private void editBio() {
        System.out.println("Enter your new bio:");
        String newBio = scanner.nextLine().trim();

        // Limit the bio length
        if (newBio.length() > 70) {
            System.out.println("Bio cannot be more than 70 characters.");
            return;
        }

        JSONObject request = new JSONObject();
        request.put("action", "edit_bio");
        request.put("new_bio", newBio);

        JSONObject response = sendWithResponse(request);

        if (response.getString("status").equals("success")) {
            System.out.println("Bio updated successfully.");
        } else {
            System.out.println(response.getString("message"));
        }
    }

    private void editProfilePictureUrl() {
        while (true) {
            System.out.println("Enter new profile picture URL (or leave empty to remove):");
            String newImageUrl = scanner.nextLine().trim();

            if (newImageUrl.contains(" ")) {
                System.out.println("URL cannot contain spaces.");
                continue;
            }

            if (!newImageUrl.isEmpty() && !newImageUrl.matches("^(http|https)://.*$")) {
                System.out.println("Invalid URL format. Please enter a valid HTTP/HTTPS link.");
                continue;
            }

            JSONObject request = new JSONObject();
            request.put("action", "edit_profile_picture");
            request.put("new_image_url", newImageUrl);
            JSONObject response = sendWithResponse(request);

            if (response.getString("status").equals("success")) {
                System.out.println("Bio updated successfully.");
            } else {
                System.out.println(response.getString("message"));
            }
            break;
        }
    }

    private void createNewGroup() {
        actionHandler.createGroup();
    }

    private void createNewChannel() {
        actionHandler.createChannel();
    }

    private void showContacts() {
        System.out.println("📇 Showing contacts...");
    }

    public void getSavedMessagesData(String userId) {
        try {
            // Step 1: Create the request
            JSONObject request = new JSONObject();
            request.put("action", "get_saved_messages");
            request.put("user_id", userId);

            // Step 2: Send request and wait for response
            JSONObject response = ActionHandler.sendWithResponse(request);

            // Step 3: Check the response
            if (!response.optString("status", "fail").equals("success")) {
                System.out.println("Failed to open Saved Messages chat: " + response.optString("message", "Unknown error"));
                return;
            }

            // Step 4: Extract "data" object
            JSONObject data = response.getJSONObject("data");
            UUID chatId = UUID.fromString(data.getString("chat_id"));
            JSONArray messagesArray = data.getJSONArray("messages");

            // Step 5: Parse messages
            List<Message> messages = new ArrayList<>();
            if (!messagesArray.isEmpty()) {
                for (int i = 0; i < messagesArray.length(); i++) {
                    JSONObject msgJson = messagesArray.getJSONObject(i);

                    // Safely extract optional UUIDs
                    UUID replyToId = null;
                    String replyToIdStr = msgJson.optString("reply_to_id", null);
                    if (replyToIdStr != null && !replyToIdStr.equals("null")) {
                        replyToId = UUID.fromString(replyToIdStr);
                    }

                    UUID originalMessageId = null;
                    String originalMessageIdStr = msgJson.optString("original_message_id", null);
                    if (originalMessageIdStr != null && !originalMessageIdStr.equals("null")) {
                        originalMessageId = UUID.fromString(originalMessageIdStr);
                    }

                    UUID forwardedBy = null;
                    String forwardedByStr = msgJson.optString("forwarded_by", null);
                    if (forwardedByStr != null && !forwardedByStr.equals("null")) {
                        forwardedBy = UUID.fromString(forwardedByStr);
                    }

                    UUID forwardedFrom = null;
                    String forwardedFromStr = msgJson.optString("forwarded_from", null);
                    if (forwardedFromStr != null && !forwardedFromStr.equals("null")) {
                        forwardedFrom = UUID.fromString(forwardedFromStr);
                    }

                    Message msg = new Message(
                            UUID.fromString(msgJson.getString("message_id")),
                            UUID.fromString(msgJson.getString("sender_id")),
                            msgJson.getString("receiver_type"),
                            UUID.fromString(msgJson.getString("receiver_id")),
                            msgJson.getString("content"),
                            msgJson.getString("message_type"),
                            msgJson.optString("file_url", null),
                            LocalDateTime.parse(msgJson.getString("send_at").replace(" ", "T")),
                            msgJson.getString("status"),
                            replyToId,
                            msgJson.getBoolean("is_edited"),
                            originalMessageId,
                            forwardedBy,
                            forwardedFrom
                    );

                    messages.add(msg);
                }
            }

            // Step 6: Show chat
            showSavedMessages(chatId, messages);

        } catch (Exception e) {
            System.out.println("An error occurred while retrieving Saved Messages.");
            e.printStackTrace();
        }
    }

    private void showSavedMessages(UUID chatId, List<Message> messages) {
        Scanner scanner = new Scanner(System.in);

        System.out.println("==== Saved Messages ====");

        // Show previous messages
        if (messages.isEmpty()) {
            System.out.println("No messages yet.");
        } else {
            for (Message msg : messages) {
                System.out.println("[" + msg.getSend_at() + "] " + msg.getContent());
            }
        }

        System.out.println("\n(Type your message below, or type 0 to exit)");

        while (true) {
            System.out.print("You: ");
            String content = scanner.nextLine().trim();

            if (content.equals("0")) {
                System.out.println("Exiting Saved Messages.");
                break;
            }

            // Prepare the request JSON
            JSONObject request = new JSONObject();
            request.put("action", "send_message");
            request.put("message_id", UUID.randomUUID().toString());
            request.put("sender_id", userUUID);
            request.put("receiver_type", "private");
            request.put("receiver_id", userUUID); // saved messages = to yourself
            request.put("content", content);
            request.put("message_type", "TEXT");
            request.put("file_url", JSONObject.NULL);
            request.put("status", "READ");
            request.put("reply_to_id", JSONObject.NULL);
            request.put("is_edited", false);
            request.put("original_message_id", JSONObject.NULL);
            request.put("forwarded_by", JSONObject.NULL);
            request.put("forwarded_from", JSONObject.NULL);

            // Send the message and wait for response
            JSONObject response = ActionHandler.sendWithResponse(request);

            if (!response.optString("status", "fail").equals("success")) {
                System.out.println("Failed to send message: " + response.optString("message", "Unknown error"));
            } else {
                System.out.println("Message sent.");
            }
        }
    }

    private void openSettings() {
        System.out.println("⚙️ Opening settings...");
    }

    private void showTelegramFeatures() {
        System.out.println("🌟 Showing Telegram features...");
    }

    private void showTelegramQA() {
        System.out.println("❓ Showing Q&A...");
    }
}