#include <thread>
#include <iostream>
#include <mutex>
#include <chrono>
#include <deque>
#include <array>
#include <cstdio>

#include "test.h"

#include "libipc/platform/detail.h"
#if defined(IPC_OS_LINUX_)
#include <pthread.h>
#include <time.h>

TEST(PThread, Robust) {
    pthread_mutexattr_t ma;
    pthread_mutexattr_init(&ma);
    pthread_mutexattr_setpshared(&ma, PTHREAD_PROCESS_SHARED);
    pthread_mutexattr_setrobust(&ma, PTHREAD_MUTEX_ROBUST);
    pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
    pthread_mutex_init(&mutex, &ma);

    std::thread{[&mutex] {
        pthread_mutex_lock(&mutex);
        // pthread_mutex_unlock(&mutex);
    }}.join();

    struct timespec tout;
    clock_gettime(CLOCK_REALTIME, &tout);
    int r = pthread_mutex_timedlock(&mutex, &tout);
    EXPECT_EQ(r, EOWNERDEAD);

    pthread_mutex_consistent(&mutex);
    pthread_mutex_unlock(&mutex);
    pthread_mutex_destroy(&mutex);
}
#elif defined(IPC_OS_WINDOWS_)
#include <Windows.h>
#include <tchar.h>

TEST(PThread, Robust) {
    HANDLE lock = CreateMutex(NULL, FALSE, _T("test-robust"));
    std::thread{[] {
        HANDLE lock = CreateMutex(NULL, FALSE, _T("test-robust"));
        WaitForSingleObject(lock, 0);
    }}.join();

    DWORD r = WaitForSingleObject(lock, 0);
    EXPECT_EQ(r, WAIT_ABANDONED);

    CloseHandle(lock);
}
#endif // OS

#include "libipc/mutex.h"

TEST(Sync, Mutex) {
    ipc::sync::mutex lock;
    EXPECT_TRUE(lock.open("test-mutex-robust"));
    std::thread{[] {
        ipc::sync::mutex lock {"test-mutex-robust"};
        EXPECT_TRUE(lock.valid());
        EXPECT_TRUE(lock.lock());
    }}.join();

    EXPECT_THROW(lock.try_lock(), std::system_error);

    // int i = 0;
    // EXPECT_TRUE(lock.lock());
    // i = 100;
    // auto t2 = std::thread{[&i] {
    //     ipc::sync::mutex lock {"test-mutex-robust"};
    //     EXPECT_TRUE(lock.valid());
    //     EXPECT_FALSE(lock.try_lock());
    //     EXPECT_TRUE(lock.lock());
    //     i += i;
    //     EXPECT_TRUE(lock.unlock());
    // }};
    // std::this_thread::sleep_for(std::chrono::seconds(1));
    // EXPECT_EQ(i, 100);
    // EXPECT_TRUE(lock.unlock());
    // t2.join();
    // EXPECT_EQ(i, 200);
}

#include "libipc/semaphore.h"

TEST(Sync, Semaphore) {
    ipc::sync::semaphore sem;
    EXPECT_TRUE(sem.open("test-sem"));
    std::thread{[] {
        ipc::sync::semaphore sem {"test-sem"};
        EXPECT_TRUE(sem.post(1000));
    }}.join();

    for (int i = 0; i < 1000; ++i) {
        EXPECT_TRUE(sem.wait(0));
    }
    EXPECT_FALSE(sem.wait(0));
}

#include "libipc/condition.h"

TEST(Sync, Condition) {
    ipc::sync::condition cond;
    EXPECT_TRUE(cond.open("test-cond"));
    ipc::sync::mutex lock;
    EXPECT_TRUE(lock.open("test-mutex"));
    std::deque<int> que;

    auto job = [&que](int num) {
        ipc::sync::condition cond {"test-cond"};
        ipc::sync::mutex lock {"test-mutex"};
        for (int i = 0; i < 10; ++i) {
            int val = 0;
            {
                std::lock_guard<ipc::sync::mutex> guard {lock};
                while (que.empty()) {
                    ASSERT_TRUE(cond.wait(lock));
                }
                val = que.front();
                que.pop_front();
            }
            EXPECT_NE(val, 0);
            std::printf("test-cond-%d: %d\n", num, val);
        }
        for (;;) {
            int val = 0;
            {
                std::lock_guard<ipc::sync::mutex> guard {lock};
                while (que.empty()) {
                    ASSERT_TRUE(cond.wait(lock, 1000));
                }
                val = que.front();
                que.pop_front();
            }
            if (val == 0) {
                std::printf("test-cond-%d: exit.\n", num);
                return;
            }
            std::printf("test-cond-%d: %d\n", num, val);
        }
    };
    std::array<std::thread, 10> test_conds;
    for (int i = 0; i < (int)test_conds.size(); ++i) {
        test_conds[i] = std::thread{job, i};
    }

    for (int i = 1; i < 100; ++i) {
        {
            std::lock_guard<ipc::sync::mutex> guard {lock};
            que.push_back(i);
            ASSERT_TRUE(cond.notify(lock));
        }
        std::this_thread::sleep_for(std::chrono::milliseconds(20));
    }
    for (int i = 1; i < 100; ++i) {
        {
            std::lock_guard<ipc::sync::mutex> guard {lock};
            que.push_back(i);
            ASSERT_TRUE(cond.broadcast(lock));
        }
        std::this_thread::sleep_for(std::chrono::milliseconds(20));
    }
    {
        std::lock_guard<ipc::sync::mutex> guard {lock};
        for (int i = 0; i < (int)test_conds.size(); ++i) {
            que.push_back(0);
        }
        ASSERT_TRUE(cond.broadcast(lock));
    }

    for (auto &t : test_conds) t.join();
}

/**
 * https://stackoverflow.com/questions/51730660/is-this-a-bug-in-glibc-pthread
*/
TEST(Sync, ConditionRobust) {
    printf("XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX 1\n");
    ipc::sync::condition cond {"test-cond"};
    printf("XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX 2\n");
    ipc::sync::mutex lock {"test-mutex"};
    printf("XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX 3\n");
    ASSERT_TRUE(lock.lock());
    std::thread unlock {[] {
        printf("WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW 1\n");
        ipc::sync::condition cond {"test-cond"};
        printf("WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW 2\n");
        ipc::sync::mutex lock {"test-mutex"};
        printf("WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW 3\n");
        {
            std::lock_guard<ipc::sync::mutex> guard {lock};
        }
        std::this_thread::sleep_for(std::chrono::seconds(1));
        printf("WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW 4\n");
        ASSERT_TRUE(cond.broadcast(lock));
        printf("WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW 5\n");
    }};
    printf("XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX 4\n");
    ASSERT_TRUE(cond.wait(lock));
    printf("XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX 5\n");
    ASSERT_TRUE(lock.unlock());
    printf("XXXXXXXXXXXXXXXXXXXXXXXXXXXXXX 6\n");
    unlock.join();
}