server.py

import socket
import sys
import threading

from Server.logging_config import server_logger
from Server.server_command import command_parser


class Server:
    def __init__(self, ip, port):
        self.port = int(port)
        self.ip = str(ip)
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    def start(self):
        self.bind_socket()
        self.sock.listen(5)  # Allow up to 5 connections
        self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        server_logger.info(f"Server running on {self.ip}:{self.port}")

    def bind_socket(self):
        while True:
            try:
                self.sock.bind((self.ip, self.port))
                break
            except OSError:
                self.port += 1

    def accept_connections(self):
        while True:
            try:
                conn, addr = self.sock.accept()
                threading.Thread(target=self.handle_connection, args=(conn, addr)).start()
            except KeyboardInterrupt:
                server_logger.info("Server stopped.")
                self.shutdown()
                break
            except Exception as e:
                server_logger.info(f"Error accepting connection: {e}")

    def handle_connection(self, conn: socket.socket, addr):
        server_logger.info(f"[+] User connected: {addr}")
        with conn:
            while True:
                try:
                    data = conn.recv(4096)
                    if not data:
                        break

                    command_parser(data, conn, addr)
                except ConnectionResetError:
                    server_logger.info(f"[-] Connection reset by {addr}")
                    break
                except Exception as e:
                    server_logger.info(f"Error handling connection from {addr}: {e}")
                    break

    def shutdown(self):
        self.sock.close()


def main(ip="127.0.0.1", port=8021):
    server = Server(ip, port)
    server.start()
    server.accept_connections()


if __name__ == "__main__":
    if len(sys.argv) == 3:
        main(sys.argv[1], int(sys.argv[2]))
    else:
        main()