/*
 * Copyright (C) 2015 - 2018, IBEROXARXA SERVICIOS INTEGRALES, S.L.
 * Copyright (C) 2015 - 2018, Jaume Olivé Petrus (jolive@whitecatboard.org)
 *
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * 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.
 *     * Neither the name of the <organization> nor the
 *       names of its contributors may be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *     * The WHITECAT logotype cannot be changed, you can remove it, but you
 *       cannot change it in any way. The WHITECAT logotype is:
 *
 *          /\       /\
 *         /  \_____/  \
 *        /_____________\
 *        W H I T E C A T
 *
 *     * Redistributions in binary form must retain all copyright notices printed
 *       to any local or remote output device. This include any reference to
 *       Lua RTOS, whitecatboard.org, Lua, and other copyright notices that may
 *       appear in the future.
 *
 * 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 <COPYRIGHT HOLDER> 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.
 *
 * Lua RTOS, FreeRTOS additions
 *
 */

//#include "luartos.h"

//#include "esp_attr.h"

//#include "lua.h"
#include <FreeRTOS.h>
#include <task.h>
#include "adds.h"

#include <stdint.h>
#include <string.h>

#include <stdlib.h>

#include <malloc.h>


void panic(const char*);

#define THREAD_LOCAL_STORAGE_POINTER_ID 0

//// Reference to lua_thread, which is created in app_main
//extern pthread_t lua_thread;

//// Global state
//static lua_State *gL = NULL;

static int compare(const void *a, const void *b) {
	if (((task_info_t *)a)->task_type < ((task_info_t *)b)->task_type) {
		return 1;
	} else if (((task_info_t*)a)->task_type > ((task_info_t *)b)->task_type) {
		return -1;
	} else {
		if (((task_info_t *)a)->core < ((task_info_t *)b)->core) {
			return -1;
		} else if (((task_info_t *)a)->core > ((task_info_t *)b)->core) {
			return 1;
		} else {
			return strcmp(((task_info_t *)a)->name, ((task_info_t *)b)->name);
		}
	}
}

void uxSetThreadId(UBaseType_t id) {
	lua_rtos_tcb_t *lua_rtos_tcb;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(xTaskGetCurrentTaskHandle(), THREAD_LOCAL_STORAGE_POINTER_ID))) {
		// Store thread id into Lua RTOS specific TCB parts
		lua_rtos_tcb->threadid = id;
	}
}

UBaseType_t uxGetThreadId() {
	lua_rtos_tcb_t *lua_rtos_tcb;
	int threadid = 0;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(xTaskGetCurrentTaskHandle(), THREAD_LOCAL_STORAGE_POINTER_ID))) {
		// Get current thread od from Lua RTOS specific TCB parts
		threadid = lua_rtos_tcb->threadid;
	}

	return threadid;
}

void uxSetThreadStatus(TaskHandle_t h, pthread_status_t status) {
	lua_rtos_tcb_t *lua_rtos_tcb;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(xTaskGetCurrentTaskHandle(), THREAD_LOCAL_STORAGE_POINTER_ID))) {
		// Store thread id into Lua RTOS specific TCB parts
		lua_rtos_tcb->status = status;
	}
}

void uxSetLThread(lthread_t *lthread) {
	lua_rtos_tcb_t *lua_rtos_tcb;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(xTaskGetCurrentTaskHandle(), THREAD_LOCAL_STORAGE_POINTER_ID))) {
		lua_rtos_tcb->lthread = lthread;
	}
}

void uxSetLuaState(lua_State* L) {
	lua_rtos_tcb_t *lua_rtos_tcb;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(xTaskGetCurrentTaskHandle(), THREAD_LOCAL_STORAGE_POINTER_ID))) {
		// Store current lua state into Lua RTOS specific TCB parts
		if (!lua_rtos_tcb->lthread) {
			lua_rtos_tcb->lthread = calloc(1, sizeof(lthread_t));
			if (lua_rtos_tcb->lthread == NULL) {
				panic("No thread TLS!");
			}
			//assert(lua_rtos_tcb->lthread);
		}
		lua_rtos_tcb->lthread->L = L;
	}

//	// If this is the lua thread, store state in global state
//	if ((uxGetThreadId() == lua_thread) && (gL == NULL)) {
//		gL = L;
//	}
}

lua_State* pvGetLuaState() {
	lua_rtos_tcb_t *lua_rtos_tcb;
	lua_State *L = NULL;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(NULL, THREAD_LOCAL_STORAGE_POINTER_ID))) {

		// Get current lua state from Lua RTOS specific TCB parts
		if (lua_rtos_tcb->lthread) {
			L = lua_rtos_tcb->lthread->L;
		}
	}

//	if (L == NULL) {
//		L = gL;
//	}

	return L;
}

lthread_t *pvGetLThread() {
	lua_rtos_tcb_t *lua_rtos_tcb;
	lthread_t *lthread = NULL;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(NULL, THREAD_LOCAL_STORAGE_POINTER_ID))) {
		lthread = lua_rtos_tcb->lthread;
	}

	return lthread;
}

uint32_t uxGetSignaled(TaskHandle_t h) {
	lua_rtos_tcb_t *lua_rtos_tcb;
	int signaled = 0;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(h, THREAD_LOCAL_STORAGE_POINTER_ID))) {
		// Get current signeled mask from Lua RTOS specific TCB parts
		signaled = lua_rtos_tcb->signaled;
	}

	return signaled;
}

void uxSetSignaled(TaskHandle_t h, int s) {
	lua_rtos_tcb_t *lua_rtos_tcb;

	// Get Lua RTOS specific TCB parts for current task
	if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(h, THREAD_LOCAL_STORAGE_POINTER_ID))) {
		// Store current signaled mask into Lua RTOS specific TCB parts
		lua_rtos_tcb->signaled = s;
	}
}

uint8_t ucGetCoreID(TaskHandle_t h) {
	tskTCB_t *task = (tskTCB_t *)h;

	return task->xCoreID;
}

int uxGetStack(TaskHandle_t h) {
	tskTCB_t *task = (tskTCB_t *)h;

	return task->pxEndOfStack - task->pxStack + 4;
}

task_info_t *GetTaskInfo() {
	tskTCB_t *ctask;
	task_info_t *info;
	uint8_t task_type;
	lua_rtos_tcb_t *lua_rtos_tcb;
	TaskStatus_t *status_array;
	UBaseType_t task_num = 0;
	UBaseType_t start_task_num = 0;
	uint32_t total_runtime = 0;

	//Allocate status_array
	start_task_num = uxTaskGetNumberOfTasks();

	status_array = (TaskStatus_t *)calloc(start_task_num, sizeof(TaskStatus_t));
	if (!status_array) {
		return NULL;
	}

//#ifndef CONFIG_FREERTOS_USE_TRACE_FACILITY
//#warning Please enable CONFIG_FREERTOS_USE_TRACE_FACILITY to support thread.list
//	free(status_array);
//	return NULL;
#ifndef configUSE_TRACE_FACILITY
#warning Please enable configUSE_TRACE_FACILITY to support thread.list
	free(status_array);
	return NULL;
#else
	task_num = uxTaskGetSystemState(status_array, (start_task_num), &total_runtime);
	// For percentage calculations.
	total_runtime /= 100UL;

	info = (task_info_t *)calloc(task_num + 1, sizeof(task_info_t));
	if (!info) {
		free(status_array);
		return NULL;
	}
#endif

	for(int i = 0; i <task_num; i++){
	    if (status_array[i].eCurrentState == eDeleted) {
	        continue;
	    }

		// Get the task TCB
		ctask = (tskTCB_t *)status_array[i].xHandle;

		// Get the task type
		// 0: freertos task
		// 1: pthread task
		// 2: lua thread task
		task_type = 0;

		info[i].thid = 0;
		info[i].status = 0;

		// Get Lua RTOS specific TCB parts for current task
		if ((lua_rtos_tcb = pvTaskGetThreadLocalStoragePointer(status_array[i].xHandle, THREAD_LOCAL_STORAGE_POINTER_ID))) {
			// Task has Lua RTOS specific TCB parts
			if (lua_rtos_tcb->lthread) {
				// Lua thread
				task_type = 2;
			} else {
				// pthread
				task_type = 1;
			}

			info[i].thid = lua_rtos_tcb->threadid;
			info[i].lthread = lua_rtos_tcb->lthread;
			info[i].status = lua_rtos_tcb->status;
		}

		// Populate info item
		info[i].prio = status_array[i].uxCurrentPriority;
		info[i].task_type = task_type;
		info[i].core = ctask->xCoreID;

		// Some system tasks shows 255!!
		if (info[i].core > 1) {
			info[i].core = 0;
		}

		info[i].free_stack = uxTaskGetStackHighWaterMark(status_array[i].xHandle);
		info[i].stack_size = ctask->pxEndOfStack - ctask->pxStack + 4;
		memcpy(info[i].name, status_array[i].pcTaskName, configMAX_TASK_NAME_LEN);

		info[i].cpu_usage = 0;
#ifdef CONFIG_FREERTOS_GENERATE_RUN_TIME_STATS
		if( total_runtime > 0 ) {
			// only gives valid values if CONFIG_FREERTOS_GENERATE_RUN_TIME_STATS is defined
			info[i].cpu_usage = status_array[i].ulRunTimeCounter / total_runtime;
		}
#endif

	}

	free(status_array);

	qsort (info, task_num, sizeof (task_info_t), compare);

	return info;
}