/*
 * Ouroboros - Copyright (C) 2016 - 2021
 *
 * A simple broadcast application
 *
 *    Dimitri Staessens <dimitri@ouroboros.rocks>
 *    Sander Vrijders   <sander@ouroboros.rocks>
 *
 * 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 the copyright holder 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 HOLDER 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.
 */

#define _POSIX_C_SOURCE 199309L

#include <ouroboros/dev.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define BUF_SIZE 256

static void usage(void)
{
        printf("Usage: obc [OPTION]...\n"
               "Sends a message to a broadcast network\n\n"
               "  -n --name                 Name of the broadcast layer\n"
               "  -m --message              A message to send\n"
               "  [-l, --listen             Listen mode]\n"
               "      --help                Display this help text and exit\n");
}

static int reader_main(const char * dst)
{
        int    fd;
        char   buf[BUF_SIZE];

        printf("Starting a reader.\n");

        fd = flow_join(dst, NULL, NULL);
        if (fd < 0) {
                printf("Failed to join broadcast.\n");
                return -1;
        }

        printf("New flow.\n");

        while (true) {
                ssize_t count = flow_read(fd, &buf, BUF_SIZE);
                if (count < 0) {
                        printf("Failed to read.\n");
                        flow_dealloc(fd);
                        break;
                }

                printf("Message is %.*s.\n", (int) count, buf);
        }

        return 0;
}

static int writer_main(const char * dst,
                       const char * message)
{
        int     fd  = 0;
        size_t  len = strlen(message) + 1;

        fd = flow_join(dst, NULL, NULL);
        if (fd < 0) {
                printf("Failed to join broadcast.\n");
                return -1;
        }

        if (flow_write(fd, message, len) < 0) {
                printf("Failed to write packet.\n");
                flow_dealloc(fd);
                return -1;
        }

        flow_dealloc(fd);

        return 0;
}

int main(int     argc,
         char ** argv)
{
        int    ret    = -1;
        bool   reader = false;
        char * name   = NULL;
        char * msg    = "Ouroboros multicast rocks!";

        argc--;
        argv++;
        while (argc > 0) {
                if (strcmp(*argv, "-l") == 0 ||
                    strcmp(*argv, "--listen") == 0) {
                        reader = true;
                } else if (strcmp(*argv, "-n") == 0 ||
                           strcmp(*argv, "--name") == 0) {
                        name = *(argv + 1);
                        argc--;
                        argv++;
                } else if (strcmp(*argv, "-m") == 0 ||
                           strcmp(*argv, "--message") == 0) {
                        msg = *(argv + 1);
                        argc--;
                        argv++;
                } else {
                        usage();
                        return 0;
                }
                argc--;
                argv++;
        }

        if (name == NULL) {
                printf("Please specify a name.\n\n");
                usage();
                exit(EXIT_FAILURE);
        }

        if (reader)
                ret = reader_main(name);
        else
                ret = writer_main(name, msg);

        return ret;
}