#include "Server.hpp"
int a = 0;

bool Server::_signal = false;

Server::Server() {}

Server::~Server() {}

// void Server::setPassword(const std::string& password) {
//     _password = password;
// }

std::string Server::getPassowrd() const {
    return _password;
}

void Server::parseArgs(int ac, char **av) {
    if (ac != 3)
        throw std::runtime_error("Usage: ./ircserv <port> <password>");
    std::string port(av[1]);
    std::string pwd(av[2]);
    if (port.empty() || port.find_first_not_of("0123456789") != std::string::npos)
        throw std::runtime_error("Error: Invalid arguments");

    long _port = atol(av[1]);
    if (!(_port >= 1 && _port <= 65535))
        throw std::runtime_error("Error: Invalid arguments");

    if (pwd.empty())
        throw std::runtime_error("Error: Invalid arguments");

    this->_port = _port;
    this->_password = pwd;
}

void Server::receiveSignal(int signum) {
    _signal = true;
    (void)signum;
}

void Server::init() {
    signal(SIGINT, receiveSignal);
    signal(SIGQUIT, receiveSignal);

    createServerSocket();
    std::cout << ">>> SERVER STARTED <<<" << std::endl;
    std::cout << "Waiting for connections..." << std::endl;
}

void Server::run() {
    while (!_signal) {
        int ret = poll(&_fds[0], _fds.size(), -1);
        if (ret == -1)
            throw std::runtime_error("Error: poll() failed");

        for (size_t i = 0; i < _fds.size(); ++i) {
            if (_fds[i].revents & POLLIN) {
                if (_fds[i].fd == _serverSocketFd)
                    handleClientConnection();
                else
                    handleClientData(_fds[i].fd);
            }
        }
    }
    closeFds();
}

void Server::createServerSocket() {
    _serverSocketFd = socket(AF_INET, SOCK_STREAM, 0);
    if (_serverSocketFd == -1)
        throw std::runtime_error("Error: failed to create socket");

    int optval = 1;
    if (setsockopt(_serverSocketFd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)) == -1)
        throw std::runtime_error("Error: setsockopt() failed");

    if (fcntl(_serverSocketFd, F_SETFL, O_NONBLOCK) == -1)
        throw std::runtime_error("Error: fcntl() failed");

    bindServerSocket();

    if (listen(_serverSocketFd, SOMAXCONN) == -1)
        throw std::runtime_error("Error: listen() failed");

    addPollfd(_serverSocketFd, POLLIN, 0);
}

void Server::bindServerSocket() {
    struct sockaddr_in sa;
    sa.sin_family = AF_INET;
    sa.sin_port = htons(_port);
    sa.sin_addr.s_addr = INADDR_ANY;
    if (bind(_serverSocketFd, (struct sockaddr*)&sa, sizeof(sa)) == -1) {
        throw std::runtime_error("Error: failed to bind socket");
    }
}

void Server::addPollfd(int fd, short events, short revents) {
    struct pollfd newPollfd;
    newPollfd.fd = fd;
    newPollfd.events = events;
    newPollfd.revents = revents;
    _fds.push_back(newPollfd);
}

void Server::handleClientConnection() {
    for (size_t i = 0; i < _fds.size(); ++i) {
        if (_fds[i].fd == _serverSocketFd && (_fds[i].revents & POLLIN)) {
            struct sockaddr_in client_addr;
            socklen_t clientAddrSize = sizeof(sockaddr_in);
            int newFd = accept(_serverSocketFd, (struct sockaddr *)&client_addr, &clientAddrSize);
            if (newFd == -1) {
                throw std::runtime_error("Error: accept() failed");
            }
            std::string passwordRequest = "Please enter the password:\n";
            send(newFd, passwordRequest.c_str(), passwordRequest.length(), 0);

            addPollfd(newFd, POLLIN, 0);
            _clients.push_back(Client(newFd, inet_ntoa((client_addr.sin_addr))));
            

            std::cout << "Client <" << newFd << "> Connected" << std::endl;
        }
    }
}


// FUNCTIONS OF TOP GGGGGG.. START FROM HEEEREEE 



//triiiiiiiiiiiimmmmmm
std::string trim(const std::string& str) {
    // Find the first non-whitespace character
    size_t first = str.find_first_not_of(" \t\n\r");

    // If the string is all whitespace, return an empty string
    if (std::string::npos == first) {
        return "";
    }

    // Find the last non-whitespace character
    size_t last = str.find_last_not_of(" \t\n\r");

    // Return the trimmed substring
    return str.substr(first, last - first + 1);
}


// here i create the channel and add the nicks of the users with some checks
void Server::createChannel(const std::string& channelName, const std::string& nickname, int fd) {
    // Check if the channel already exists
    std::map<std::string, Channel>::iterator it = channels.find(channelName);
    if (it == channels.end()) {
        // Channel doesn't exist, so create it and add the user
        Channel newChannel(channelName);
        newChannel.addClient(nickname, fd);
        newChannel.addOperator(nickname);
        std::cout << nickname << " : im the operator of the channel guys. " << std::endl;
        channels.insert(std::make_pair(channelName, newChannel)); // Insert the new channel into the map
        std::cout << "Channel '" << channelName << "' created by '" << nickname << "'" << std::endl;
    } else {
        // Channel already exists, just add the user to it
        it->second.addClient(nickname, fd);
        std::cout << "User '" << nickname << "' joined channel '" << channelName << "'" << std::endl;
    }
}

bool startsWith(const std::string& str, const std::string& prefix) {
    return str.substr(0, prefix.length()) == prefix;
}

 void Server::setNickname(int fd, const std::string& nickname) {
        nicknames[fd] = nickname; // Associate the nickname with the client's socket descriptor
    }

 void Server::setUsernameoperators(int fd, const std::string& username) {
        usernamesoperators[fd] = username; // Associate the nickname with the client's socket descriptor
    }
 void Server::setUsernames(int fd, const std::string& username) {
        usernames[fd] = username; // Associate the nickname with the client's socket descriptor
    }
 void Server::setUsernameregular(int fd, const std::string& username) {
        usernamesregulars[fd] = username; // Associate the nickname with the client's socket descriptor
    }

//it's a just a response for my client 
void sendResponse(int fd, const std::string& message) {
        // Convert the message string to a C-style string
        const char* msg = message.c_str();
        
        // Get the length of the message
        size_t len = strlen(msg);                 
        
        // Send the message to the client
        ssize_t bytesSent = send(fd, msg, len, 0);
        
        // Check if the send operation was successful
        if (bytesSent == -1) {
            // Handle send error
            // You can log an error message or take appropriate action
            std::cerr << "Error sending response to client <" << fd << ">" << std::endl;
        } else {
            // Output the sent message to the console
            std::cout << "Response sent to client <" << fd << ">: " << message << std::endl;
        }
    }



//here finding the users for the handling the privet msg 
int findUserFd(const std::string& nickname, const std::map<int, std::string>& usernames) {
    // Iterate over each entry in the usernames map
    for (std::map<int, std::string>::const_iterator it = usernames.begin(); it != usernames.end(); ++it) {
        // Check if the nickname matches
        if (it->second == nickname) {
            // Return the file descriptor associated with the nickname
            return it->first;
        }
    }
    // Return -1 if the nickname is not found
    return -1;
}


///iterate to find the username with the fd

// std::string Server::findUsernameforsending(int fd) {
//     std::map<int, std::string>::iterator it;
//     for (it = usernames.begin(); it != usernames.end(); ++it) {
//         if (it->first == fd) {
//             return it->second; // Return the file descriptor if the nickname matches
//         }
//     }
// }
//forget it

//handling privet msg between users only 
void Server::handlePrivateMessage(int senderFd, const std::string& recipient, const std::string& message) {
    // Find the recipient's connection (socket file descriptor)
    int recipientFd = findUserFd1(recipient);
    // std::string  sendernameuser = findUsernameforsending(senderFd);

    if (recipientFd != -1) {
        // Forward the private message to the recipient's client
        std::string privateMessage = ":" + usernames[senderFd] + " PRIVMSG " + recipient + " :" + message + "\r\n";
        send(recipientFd, privateMessage.c_str(), privateMessage.length(), 0);
    } else {
        // Handle case where recipient is not found (e.g., user not online)
        std::string errorMessage = ":server.host NOTICE " + usernames[senderFd] + " :Error: User '" + recipient + "' not found or offline\r\n";
        send(senderFd, errorMessage.c_str(), errorMessage.length(), 0);
    }
}

//this find is for finding nickname of the users i need to brodcasting to 
int Server::findUserFd1(const std::string& username) {
    std::map<int, std::string>::iterator it;
    for (it = usernames.begin(); it != usernames.end(); ++it) {
        if (it->second == username) {
            return it->first; // Return the file descriptor if the nickname matches
        }
    }
    return -1; // Return -1 if the nickname is not found
}

int Server::findUserFdforkickregulars(const std::string& username) {
    std::map<int, std::string>::iterator it;
    for (it = usernamesregulars.begin(); it != usernamesregulars.end(); ++it) {
        if (it->second == username) {
            return it->first; // Return the file descriptor if the nickname matches
        }
    }
    return -1; // Return -1 if the nickname is not found
}
// brodcasting msg to all nicks in the  channel 
void Server::broadcastMessage(const std::string& channel, const std::string& senderNickname, const std::string& msg) {
    // Check if the channel exists
    debugPrintChannels();
    std::map<std::string, Channel>::iterator it = channels.find(channel);
    if (it == channels.end()) {
        std::cerr << "Channel " << channel << " does not exist" << std::endl;
        return;
    }

    // Construct the IRC message with the correct format for broadcasting
    std::string message = ":" + senderNickname + " PRIVMSG #" + channel + " :" + msg + "\r\n";

    // Get a reference to the vector of clients in the channel
    const std::vector<std::string>& clients = it->second.getClients();

    // Iterate over the vector of clients and send the message to each one
    for (size_t i = 0; i < clients.size(); ++i) {
        // Get the current client nickname
        const std::string& client = clients[i];

        // Skip sending the message to the sender
        std::cout << "this is the client name : "<< client <<  std::endl;
        std::cout << "this is the nickname name : " << senderNickname << std::endl;

        if (client == senderNickname) {
            continue;
        }

        // Find the file descriptor associated with the client nickname
        int recipientFd = it->second.getUserFd(client);

        // If the file descriptor is found, send the message to the client
        if (recipientFd != -1) {
            std::cout << message << std::endl;
            send(recipientFd, message.c_str(), message.size(), 0);
        } else {
            // If the file descriptor is not found, print an error message
            std::cerr << "Client " << client << " not found" << std::endl;
        }
    }
}

//check if ope or not
bool Server::isOperator(int fd) {
    // Check if the file descriptor exists in the map of operators
    return usernamesoperators.find(fd) != usernamesoperators.end();
}


void Server::kickUser(int fd) {
    // Close the socket connection for the given file descriptor
    close(fd);

    // Remove the user from any data structures tracking active connections
    // For example, if you have a map of file descriptors to usernames:
    std::map<int, std::string>::iterator it = usernamesregulars.find(fd);
    if (it != usernamesregulars.end()) {
        // Remove the user from the map
        usernamesregulars.erase(it);
    }
}



// ........AND FINISH HERE ......//////


void Server::handleClientData(int fd) {
    std::string command;
    char buffer[BUFFER_SIZE];
    memset(buffer, 0, sizeof(buffer));

    ssize_t bytesRead;
    while ((bytesRead = recv(fd, buffer, BUFFER_SIZE - 1, 0)) > 0) {
        bool foundEof = false;
        for (ssize_t i = 0; i < bytesRead; ++i) {
            if (buffer[i] == '\n') {
                foundEof = true;
                break;
            }
        }

        if (!foundEof) {
            buffer[bytesRead] = '\0';
            command += buffer;
        } else {
            command.append(buffer, bytesRead - 1);
            std::cout << "Received data from client " << fd << ": " << command << std::endl;

//******************* FROM THERE IM STARTING TOP GGG ************  .
            if (startsWith(command, "pass"))
            {
                std::string cmd, password;
                std::istringstream iss(command);
                iss >> cmd >> password;
                password = trim(password); // Remove leading/trailing whitespace
                std::string  passwordoftheserver = getPassowrd();
                if (passwordoftheserver != password)
                {
                    std::string errorMessage = "Error: Incorrect password\n";
                    send(fd, errorMessage.c_str(), errorMessage.length(), 0);
                }
                else {
                    std::string confirmation = "Welcome sir\n";
                    send(fd, confirmation.c_str(), confirmation.length(), 0);
                    a = 1;
                }

                // for (size_t i = 0; i < _clients.size(); ++i) {
                //     if (_clients[i].getFd() == fd) {
                //         _clients[i].setPassword(password);
                //         std::cout << "Password set for client " << fd << ": " << password << std::endl;
                //         break;
                //     }
                // }
            }
            if (startsWith(command, "nick")) {
                std::string cmd, nick;
                std::istringstream iss(command);
                iss >> cmd >> nick;
                nick = trim(nick);
                for (size_t i = 0; i < _clients.size(); ++i) {
                    if (_clients[i].getFd() == fd) {
                        _clients[i].setNick(nick);
                        std::cout << "Password set for client " << fd << ": " << nick << std::endl;
                        break;
                    }
                }
                // Extract the nickname from the command
                // std::string nickname = command.substr(8); // Assuming "/setnick " is 9 characters long

                // Handle setting the nickname for the client's connection


                // Send a response back to the client confirming the action
                sendResponse(fd, "Nickname set to: " + nick + '\n');
                a = 2;
            } else if (startsWith(command, "user")) {

                std::istringstream iss(command);
                std::string cmd, username, dontworry, dontworry1, realname, nickname;
                iss >> cmd >> username >> dontworry >> dontworry1 >> realname;
                // Remove leading and trailing whitespace from parameters
                username = trim(username);
                dontworry = trim(dontworry);
                dontworry1 = trim(dontworry1);
                realname = trim(realname);

                setUsernames(fd, username);

                for (size_t i = 0; i < _clients.size(); ++i) {
                    if (_clients[i].getFd() == fd) {
                        _clients[i].setUser(username);
                        _clients[i].setName(realname);
                        nickname = _clients[i].getNick();
                        break;
                    }
                }

                sendResponse(fd, "username set to: " + username + '\n');
                sendResponse(fd, "realname set to: " + realname + '\n');

                std::string one = ":irc.topg 001 " + nickname + " :Welcome to the topg Network, " + nickname + '\n';
                std::string two = ":irc.topg 002 " + nickname + " :Your host is topg, running version 3030" + '\n';
                std::string tre = ":irc.topg 003 " + nickname + " :This server was created Tue Nov 30 2011 at 11:11:25 EET" + '\n';
                std::string foor = ":irc.topg 004 " + nickname + " topg tella(enterprise)-2.3(12)-netty(5.4c)-proxy(0.9) oOiwscrknfbghexzSjFI bhijklmMnoOstvcdSuU bkohv" + '\n';
                send(fd, one.c_str(), one.length(), 0);
                send(fd, two.c_str(), two.length(), 0);
                send(fd, tre.c_str(), tre.length(), 0);
                send(fd, foor.c_str(), foor.length(), 0);



                




                // if (privilege_level == "operator" )
                // {
                //     //si moskir hna atbda lkhdma dyalk 
                //     // std::cout << "we need to handle this " << std::endl;
                //     setUsernameoperators(fd, username);
                //     sendResponse(fd, "Username set to: " + username + " with privilege_level : " + privilege_level + '\n');
                // }
                // else if (privilege_level == "regular") {
                //     setUsernameregular(fd, username);
                //     sendResponse(fd, "Username set to: " + username + " with privilege_level : " + privilege_level + '\n');

                // }
                

                // Process other commands or messages
                // processCommand(fd, command);
            } else if (startsWith(command, "JOIN ")) {
                std::string nick;
                for (size_t i = 0; i < _clients.size(); ++i) {
                    if (_clients[i].getFd() == fd) {
                        nick = _clients[i].getNick();
                        std::cout << "this is the nick for checkig : " << nick << std::endl;
                        break;
                    }
                }
                std::string chanelname = command.substr(6);
                std::cout << chanelname << std::endl;
                chanelname = trim(chanelname);
                std::cout << chanelname << std::endl;
                createChannel(chanelname, nick, fd);



            } else if (startsWith(command, "PRIVMSG ")) {
                    // Extract the recipient and the message from the command
                    std::istringstream iss(command);
                    std::string cmd, recipient, message;
                    std::string niiick;
                    iss >> cmd >> recipient;
                    recipient = trim(recipient);
                    std::getline(iss, message);
                    message = trim(message);
                    message = message.substr(1);
                    if (recipient[0] == '#')
                    {

                        recipient = recipient.substr(1);
                        std::cout << "this the chanel name and was good : " << recipient << std::endl;
                        for (size_t i = 0; i < _clients.size(); ++i) {
                            if (_clients[i].getFd() == fd) {
                                niiick =  _clients[i].getNick();
                                break;
                        }
                    }
                        broadcastMessage(recipient, niiick, message);
                    }
                    else
                    {
                        handlePrivateMessage(fd, recipient, message);
                    }
                    std::cout << "Recipient: " << recipient << std::endl;
                    std::cout << "Message: " << message << std::endl;
            }
            else if(startsWith(command, "KICK ") && isOperator(fd)) {
                    std::string userkicked = command.substr(5);
                    userkicked = trim(userkicked);

                    int finduserfd = findUserFdforkickregulars(userkicked);
                    if (finduserfd != -1)
                    {
                        kickUser(finduserfd);
                        sendResponse(fd, "User '" + userkicked + "' has been kicked from the server." + '\n');
                    }
                    else
                        sendResponse(fd, "Error: User '" + userkicked + "' not found or offline.\n");

            }
            else if (startsWith(command, "KICK "))
                sendResponse(fd, "Error: You don't have permission to use this command.\n");

//**************** STOOOOOOP HERE TOP G ... 
            break;
        }
    }

    if (bytesRead == 0) {
        std::cout << "Client <" << fd << "> Disconnected" << std::endl;
        clientCleanup(fd);
    } else if (bytesRead == -1) {
        std::cerr << "Error reading data from client <" << fd << ">" << std::endl;
        clientCleanup(fd);
    }
}


void Server::clientCleanup(int fd) {
    for (std::vector<pollfd>::iterator it = _fds.begin(); it != _fds.end(); ++it) {
        if (it->fd == fd) {
            _fds.erase(it);
            close(fd);
            break;
        }
    }

    for (std::vector<Client>::iterator it = _clients.begin(); it != _clients.end(); ++it) {
        if (it->getFd() == fd) {
            _clients.erase(it);
            break;
        }
    }
}

void Server::closeFds() {
    for (size_t i = 0; i < _clients.size(); i++){
        int fd = _clients[i].getFd();
        std::cout << "Client <" << fd << "> Disconnected" << std::endl;
        close(fd);
    }

    if (_serverSocketFd != -1)
        close(_serverSocketFd);

    _fds.clear();
}