#include <stdio.h>

#include <string>

#include "exceptions.h"

using std::string;

static const char *kCaughtNothing               = "nothing";
static const char *kCaughtGeneric               = "generic";
static const char *kCaughtBase                  = "ExceptionBase";
static const char *kCaughtA                             = "ExceptionA";
static const char *kCaughtB                             = "ExceptionB";
static const char *kCaughtVirtualBase   = "VirtualExceptionBase";
static const char *kCaughtVirtualA              = "VirtualExceptionA";
static const char *kCaughtVirtualB              = "VirtualExceptionB";
static const char *kCaughtInt                   = "VirtualInt";


static string
catchBase(void (*function)())
{
        try {
                (*function)();
        } catch (ExceptionBase exception) {
                return kCaughtBase;
        } catch (...) {
                return kCaughtGeneric;
        }
        return kCaughtNothing;
}


static string
catchA(void (*function)())
{
        try {
                (*function)();
        } catch (ExceptionA exception) {
                return kCaughtA;
        } catch (...) {
                return kCaughtGeneric;
        }
        return kCaughtNothing;
}


static string
catchB(void (*function)())
{
        try {
                (*function)();
        } catch (ExceptionB exception) {
                return kCaughtB;
        } catch (...) {
                return kCaughtGeneric;
        }
        return kCaughtNothing;
}


static string
catchVirtualBase(void (*function)())
{
        try {
                (*function)();
        } catch (VirtualExceptionBase exception) {
                return kCaughtVirtualBase;
        } catch (...) {
                return kCaughtGeneric;
        }
        return kCaughtNothing;
}


static string
catchVirtualA(void (*function)())
{
        try {
                (*function)();
        } catch (VirtualExceptionA exception) {
                return kCaughtVirtualA;
        } catch (...) {
                return kCaughtGeneric;
        }
        return kCaughtNothing;
}


static string
catchVirtualB(void (*function)())
{
        try {
                (*function)();
        } catch (VirtualExceptionB exception) {
                return kCaughtVirtualB;
        } catch (...) {
                return kCaughtGeneric;
        }
        return kCaughtNothing;
}


static string
catchInt(void (*function)())
{
        try {
                (*function)();
        } catch (int exception) {
                return kCaughtInt;
        } catch (...) {
                return kCaughtGeneric;
        }
        return kCaughtNothing;
}

static string
catchAny(void (*function)())
{
        try {
                (*function)();
        } catch (int exception) {
                return kCaughtInt;
        } catch (VirtualExceptionA exception) {
                return kCaughtVirtualA;
        } catch (VirtualExceptionB exception) {
                return kCaughtVirtualB;
        } catch (VirtualExceptionBase exception) {
                return kCaughtVirtualBase;
        } catch (ExceptionA exception) {
                return kCaughtA;
        } catch (ExceptionB exception) {
                return kCaughtB;
        } catch (ExceptionBase exception) {
                return kCaughtBase;
        } catch (...) {
                return kCaughtGeneric;
        }
        return kCaughtNothing;
}

static void
test(string (*catcher)(void (*)()), void (*thrower)(), const char *expected)
{
        string caught((*catcher)(thrower));
        if (caught != expected) {
                printf("ERROR: expected exception: %s, but caught: %s\n", expected,
                        caught.c_str());
        }
}

int
main()
{
        test(catchBase, throwBase, kCaughtBase);
        test(catchBase, throwA, kCaughtBase);
        test(catchBase, throwB, kCaughtBase);
        test(catchBase, throwVirtualBase, kCaughtBase);
        test(catchBase, throwVirtualA, kCaughtBase);
        test(catchBase, throwVirtualB, kCaughtBase);
        test(catchBase, throwInt, kCaughtGeneric);

        test(catchA, throwBase, kCaughtGeneric);
        test(catchA, throwA, kCaughtA);
        test(catchA, throwB, kCaughtGeneric);
        test(catchA, throwVirtualBase, kCaughtGeneric);
        test(catchA, throwVirtualA, kCaughtGeneric);
        test(catchA, throwVirtualB, kCaughtGeneric);
        test(catchA, throwInt, kCaughtGeneric);

        test(catchVirtualBase, throwBase, kCaughtGeneric);
        test(catchVirtualBase, throwA, kCaughtGeneric);
        test(catchVirtualBase, throwB, kCaughtGeneric);
        test(catchVirtualBase, throwVirtualBase, kCaughtVirtualBase);
        test(catchVirtualBase, throwVirtualA, kCaughtVirtualBase);
        test(catchVirtualBase, throwVirtualB, kCaughtVirtualBase);
        test(catchVirtualBase, throwInt, kCaughtGeneric);

        test(catchVirtualA, throwBase, kCaughtGeneric);
        test(catchVirtualA, throwA, kCaughtGeneric);
        test(catchVirtualA, throwB, kCaughtGeneric);
        test(catchVirtualA, throwVirtualBase, kCaughtGeneric);
        test(catchVirtualA, throwVirtualA, kCaughtVirtualA);
        test(catchVirtualA, throwVirtualB, kCaughtGeneric);
        test(catchVirtualA, throwInt, kCaughtGeneric);

        test(catchInt, throwBase, kCaughtGeneric);
        test(catchInt, throwA, kCaughtGeneric);
        test(catchInt, throwB, kCaughtGeneric);
        test(catchInt, throwVirtualBase, kCaughtGeneric);
        test(catchInt, throwVirtualA, kCaughtGeneric);
        test(catchInt, throwVirtualB, kCaughtGeneric);
        test(catchInt, throwInt, kCaughtInt);

        test(catchAny, throwBase, kCaughtBase);
        test(catchAny, throwA, kCaughtA);
        test(catchAny, throwB, kCaughtB);
        test(catchAny, throwVirtualBase, kCaughtVirtualBase);
        test(catchAny, throwVirtualA, kCaughtVirtualA);
        test(catchAny, throwVirtualB, kCaughtVirtualB);
        test(catchAny, throwInt, kCaughtInt);

        return 0;
}