#include "CO2_Sense.h"
#include <string.h>
#include "freertos/FreeRTOS.h"
#include "freertos/portmacro.h"
#include "freertos/task.h"
#include <stdio.h>
#include "hal/gpio_types.h"
#include "nvs_flash.h"

#include "lwip/err.h"
#include "lwip/sys.h"

#include "esp_log.h"
#include "esp_sleep.h"
#include "driver/gpio.h"
#include "driver/i2c.h"

#include "ledController.h"
#include "scd4x_i2c.h"
#include "sensirion_i2c_hal.h"
#include "scd4x_data.h"
#include "BTlib_nimble.h"
#include "MMC56x3.h"

void app_main(void){
	init_nvs();
	
	nvs_handle_t nvs_handle;
  ESP_ERROR_CHECK(nvs_open("main", NVS_READWRITE, &nvs_handle));
	
	// setup GPIOs
	gpio_config_t VBAT_OK_c = {
		GPIO_NUM_3,
		GPIO_MODE_INPUT,
		GPIO_PULLUP_DISABLE,
		GPIO_PULLDOWN_DISABLE,
		GPIO_INTR_DISABLE,
	};
	ESP_ERROR_CHECK(gpio_config(&VBAT_OK_c));

	configuration_data_t* conf = malloc(sizeof(configuration_data_t));
	init_conf_from_nvs(conf, nvs_handle);
	
	uint8_t led_nb=0;
	ESP_ERROR_CHECK(nvs_get_u8(nvs_handle, "led_nb", &led_nb));
	// init LEDs driver	
	init_led_driver(conf->leds, conf->led_nb);
	
	// init BLE
	initBle(conf);
	
	// start sensor
	init_i2c();
	init_scd4x();
	
	uint8_t id=0;
	MMC56x3_get_product_ID(&id);
	ESP_LOGI("MAIN", "MMC5603 product id %d", id);
	
	switch(conf->sensor->mode){
		case SCD4X_NORMAL_MODE:
			ESP_ERROR_CHECK(scd4x_start_periodic_measurement());
			ESP_LOGI("scd4x init","started in normal mode");
		break;
		case SCD4X_LP_MODE:
			ESP_ERROR_CHECK(scd4x_start_low_power_periodic_measurement());
			ESP_LOGI("scd4x init","started in low power mode");
		break;
	}
	if(conf->sensor->mode & SCD4X_CAPTURE_ENABLED){
		TaskHandle_t sensor_fetch_handle;
		xTaskCreate(fetch_sensor_task, "FETCH_SENSOR", 4096, conf, tskIDLE_PRIORITY, &sensor_fetch_handle);
	}

	while(1){
		vTaskDelay(1000 / portTICK_PERIOD_MS);
	}
}

void fetch_sensor_task(void* pvParameters){
	configuration_data_t* mainConf = (configuration_data_t*) pvParameters;
	scd4x_config_t* conf = mainConf->sensor;
	scd4x_data_t* measures = conf->measure;
	while(1){
		if(conf->mode == SCD4X_SS_MODE) scd4x_measure_single_shot(); 
		vTaskDelay(conf->delay * 1000 / portTICK_PERIOD_MS);
		uint16_t dataReady;
		scd4x_get_data_ready_status(&dataReady);
		if(!(dataReady & 0x07FF)) continue;
		int16_t error = scd4x_read_measurement(&measures->co2, &measures->temperature, &measures->humidity);
                if (error)
                        ESP_LOGE("sensor fetch", "Error executing scd4x_read_measurement(): %i\n", error);
		update_led_status(mainConf->leds, mainConf->led_nb, conf->measure->co2);
		ble_sensor_notify();
		ESP_LOGI("MAIN", "co2 : %u ppm, temp : %ld m°C, hum : %ld mRH", conf->measure->co2, conf->measure->temperature, conf->measure->humidity);
	}
}

void init_nvs(void){
	esp_err_t ret = nvs_flash_init();
	if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {
		ESP_ERROR_CHECK(nvs_flash_erase());
		ret = nvs_flash_init();
	}
	ESP_ERROR_CHECK(ret);
}

void init_i2c(){
	int i2c_master_port = I2C_NUM_0;
	i2c_config_t conf = {
		.mode = I2C_MODE_MASTER,
    .sda_io_num = CONFIG_SDA_PIN,
    .sda_pullup_en = GPIO_PULLUP_DISABLE,
    .scl_io_num = CONFIG_SCL_PIN,
    .scl_pullup_en = GPIO_PULLUP_DISABLE,
    .master.clk_speed = 10000,
    // .clk_flags = 0,          /*!< Optional, you can use I2C_SCLK_SRC_FLAG_* flags to choose i2c source clock here. */
  };
  i2c_param_config(i2c_master_port, &conf);
  i2c_driver_install(i2c_master_port, I2C_MODE_MASTER, 0, 0, 0);
}

void init_scd4x(){
  scd4x_wake_up();
  scd4x_stop_periodic_measurement();
  scd4x_reinit();
}