/*
 * Copyright (c) 2023-2024 Ian Marco Moffett and the Osmora Team.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of Hyra nor the names of its
 *    contributors may be used to endorse or promote products derived from
 *    this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <ostp/session.h>
#include <ostp/server.h>
#include <stdio.h>
#include <string.h>

#define OMOC_PORT 5352

static struct ostp_listener l;

static int
send_to(void *buf, size_t len, struct ostp_client *c)
{
    struct ostp_session *s;

    s = &c->session;
    return send_frame(s->sockfd, buf, len, s->session_key);
}

static int
relay_others(struct ostp_client *sender, const void *buf, size_t len)
{
    struct ostp_client *clients[MAX_CLIENTS];
    char send_buf[4096];
    int client_count;

    client_count = listener_clients(&l, clients);
    if (client_count < 0) {
        return client_count;
    }

    memcpy(send_buf, buf, len);
    printf("MESSAGE FROM SOCKFD: %d\n", sender->sockfd);
    printf("SERVER SOCKFD: %d\n", l.serv_sockfd);

    /* Relay to all *other* clients */
    for (int i = 0; i < client_count; ++i) {
        /* Do not relay to server or self */
        if (clients[i]->sockfd == l.serv_sockfd) {
            printf("SKIPPED SERVER SOCKET (%d)\n", i);
            continue;
        } else if (clients[i]->sockfd == sender->sockfd) {
            printf("SKIPPED SENDER SOCKET (%d)\n", i);
            continue;
        }

        printf("SEND TO CLIENT: %d\n", i);
        session_send(send_buf, len, &clients[i]->session);
        printf("SEND TO CLIENT DONE\n");
    }

    return 0;
}

static int
recv_msg(struct ostp_client *c, const void *buf, size_t len)
{
    return relay_others(c, buf, len);
}

static int
on_conn(struct ostp_client *c)
{
    struct ostp_session *s;
    char buf[4096];

    s = &c->session;
    snprintf(buf, sizeof(buf), "[system]: %s has joined", s->username);
    return relay_others(c, buf, sizeof(buf));
}

int
main(void)
{
    int error;

    listener_init(&l);
    l.port = OMOC_PORT;
    l.on_recv = recv_msg;
    l.on_connect = on_conn;

    /* Bind and poll the listener */
    if ((error = listener_bind(&l)) < 0) {
        return error;
    }
    if ((error = listener_poll(&l)) < 0) {
        return error;
    }

    return 0;
}