import dgram, { RemoteInfo } from 'dgram';
import { UdpSocketCommunicator } from "../socket_communicator/udp_socket_communicator";
import { OperationHandler } from "../operations_base/operation_handler";
import { GeneralOperations} from "../operations_custom/general_operations";

export class UdpServer {
    private readonly udpServer: dgram.Socket;
    private readonly operationHandler: OperationHandler;
    private readonly port: number;
    private readonly host: string;

    constructor(host: string, port: number) {
        this.udpServer = dgram.createSocket('udp4');
        this.operationHandler = OperationHandler.getInstance();
        this.host = host;
        this.port = port;

        // Load only the GeneralOperations into the operation handler
        this.operationHandler.loadPlugin(new GeneralOperations());
    }

    // Start the UDP server
    public start(): void {
        this.udpServer.on('message', this.handleUdpMessages.bind(this));
        this.udpServer.on('error', this.handleError);
        this.udpServer.on('listening', this.handleListening.bind(this));

        // Bind the server to the UDP port and host
        this.udpServer.bind(this.port, this.host);
    }

    // Handle incoming UDP messages
    private async handleUdpMessages(msg: Buffer, rinfo: RemoteInfo): Promise<void> {
        const ip = rinfo.address;
        const port = rinfo.port;

        console.log(`Received message from ${ip}:${port}`);

        // Create a temporary communicator for the incoming message
        const communicator = new UdpSocketCommunicator(this.udpServer, ip, port, this.operationHandler);

        // Process the incoming message using the communicator
        communicator.handleIncomingMessage(msg.toString());

        const communicatorResult = communicator.getHandlerResult();
        if (communicatorResult) {
            // Send response back to the client using the temporary communicator
            await communicator.sendMessage(communicatorResult.operationCode, communicatorResult.metaInfo);
        } else {
            console.error(`No handler result for ${ip}:${port}`);
        }
    }

    // Handle UDP server errors
    private handleError(err: Error): void {
        console.error(`UDP server error:\n${err.stack}`);
    }

    // Handle when the UDP server starts listening
    private handleListening(): void {
        const address = this.udpServer.address();
        console.log(`UDP server listening on ${address.address}:${address.port}`);
    }
}