#include "zthread.hpp"

#include "mutex.hpp"

extern "C" {
#include "basic\zlog.h"
}

using namespace iflytop;
using namespace std;

static ZThread g_default_thread;

static void zosthread_default_task(void const *argument) {
  ZThread *thread = (ZThread *)argument;
  ZASSERT(thread);
  thread->threadcb();
};

void ZThread::threadcb() {
  m_status = kidle;

  while (true) {
    if (m_threadisworkingFlagCallSide) {
      m_status = kworking;
      if (m_taskfunction) m_taskfunction();
      if (m_taskfunction_exitcb) m_taskfunction_exitcb();
      m_status = kdead;

      while (m_threadisworkingFlagCallSide) {
        vTaskDelay(10);
      }
      m_status = kidle;
    }
    vTaskDelay(10);
  }
}

void ZThread::init(const char *threadname, int stack_size, osPriority priority) {
  // int r_task_create = 0;
  ZASSERT(threadname);

  m_lock = xSemaphoreCreateMutex();
  ZASSERT(m_lock);

  m_stacksize             = stack_size;
  m_uxPriority            = osPriorityNormal;
  m_taskfunction          = nullptr;
  m_zthreadstartworkevent = xEventGroupCreate();
  m_name                  = threadname;

  osThreadDef_t threadhandl = {(char *)threadname, zosthread_default_task, m_uxPriority, 0, m_stacksize, NULL, NULL};
  m_defaultTaskHandle       = osThreadCreate(&threadhandl, this);
  ZASSERT(m_defaultTaskHandle != NULL);
}
void ZThread::start(zosthread_cb_t cb) { start(cb, nullptr); }

void ZThread::start(zosthread_cb_t cb, zosthread_cb_t exitcb) {
  stop();

  m_taskfunction        = cb;
  m_taskfunction_exitcb = exitcb;

  ZASSERT(m_taskfunction);
  xSemaphoreTake(m_lock, portMAX_DELAY);
  m_threadisworkingFlagCallSide = true;
  // xEventGroupSetBits(m_zthreadstartworkevent, 0x01);
  while (m_status == kidle) {
    xTaskNotifyGive(m_defaultTaskHandle);
    vTaskDelay(1);
  }
  xSemaphoreGive(m_lock);
}

void ZThread::stop() {
  xSemaphoreTake(m_lock, portMAX_DELAY);
  m_threadisworkingFlagCallSide = false;
  // xEventGroupSetBits(m_zthreadstartworkevent, 0x01);
  while (m_status != kidle) {
    xTaskNotifyGive(m_defaultTaskHandle);
    vTaskDelay(1);
  }
  xSemaphoreGive(m_lock);
}

void ZThread::sleep(uint32_t ms) { ulTaskNotifyTake(pdFALSE, pdMS_TO_TICKS(ms)); }
void ZThread::wake() {
  BaseType_t state;
  if (xPortIsInsideInterrupt()) {
    vTaskNotifyGiveFromISR(m_defaultTaskHandle, &state);
  } else {
    xTaskNotifyGive(m_defaultTaskHandle);
  }
}