#include "link.h"
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

static inline Args parse_args(int argc, char** argv);

int main(int argc, char** argv)
{
    Args args = parse_args(argc, argv);
    FileIter input_files = { 1, argv, args.input_files };

    switch (args.profile) {
        case Profile_Linked:
        case Profile_Shared:
            fprintf(stderr,
                "profile '%s' not implemented\n",
                profile_strs[args.profile]);
            break;
        case Profile_Kern:
            return link_kern(&args, input_files);
    }
}

static inline Args parse_args(int argc, char** argv)
{
    const char* output_file = NULL;
    int input_files = 0;
    Profile profile = Profile_Linked;
    for (int i = 1; i < argc; ++i) {
        if (strcmp(argv[i], "-o") == 0) {
            i += 1;
            if (i >= argc) {
                fprintf(stderr, "error: no filename given to -o\n");
                exit(1);
            }
            output_file = argv[i];
        } else if (strcmp(argv[i], "-p") == 0) {
            i += 1;
            if (i >= argc) {
                fprintf(stderr, "error: no profile given to -p\n");
                exit(1);
            }

            size_t strs_size = sizeof(profile_strs) / sizeof(profile_strs[0]);
            bool found = false;
            for (size_t si = 0; si < strs_size; ++si) {
                if (strcmp(profile_strs[si], argv[i]) == 0) {
                    profile = (Profile)si;
                    found = true;
                    break;
                }
            }
            if (!found) {
                fprintf(stderr, "error: invalid profile '%s'\n", argv[i]);
                fprintf(stderr, "info: valid profiles: linked, shared, kern\n");
                exit(1);
            }
        } else {
            input_files += 1;
        }
    }
    if (input_files == 0) {
        fprintf(stderr, "error: no input files specified\n");
        exit(1);
    }
    if (output_file == NULL) {
        output_file = "a.out";
    }
    return (Args) {
        output_file,
        input_files,
        profile,
    };
}