mqtt/main.c

#include <stdint.h>
#include "user_interface.h"
#include "ets_sys.h"
#include "os_type.h"
#include "osapi.h"
#include "mem.h"
#include "espconn.h"
#include "mqtt.h"
#include "onewire.h"
#include "main.h"

os_timer_t tcpTimer;
os_timer_t pingTimer;
os_timer_t pubTimer;

LOCAL void ICACHE_FLASH_ATTR con(void *arg) {
  os_printf("Entered con!\n");
  mqtt_session_t *pSession = (mqtt_session_t *)arg;
  mqttSend(pSession, NULL, 0, MQTT_MSG_TYPE_CONNECT);

  os_timer_disarm(&pingTimer);
  os_timer_setfn(&pingTimer, (os_timer_func_t *)ping, arg);
  os_timer_arm(&pingTimer, 1000, 0);
}

LOCAL void ICACHE_FLASH_ATTR pubuint(void *arg) {
  os_printf("Entered pubuint!\n");
  mqtt_session_t *pSession = (mqtt_session_t *)arg;
  uint8_t *data = (uint8_t *)(pSession->userData);
  char *dataStr = os_zalloc(20 * sizeof(char));
  intToStr(*data, dataStr, 4);
  int32_t dataLen = os_strlen(dataStr);
  mqttSend(pSession, (uint8_t *)dataStr, dataLen, MQTT_MSG_TYPE_PUBLISH);
  os_timer_disarm(&pingTimer);
  os_timer_setfn(&pingTimer, (os_timer_func_t *)ping, arg);
  os_timer_arm(&pingTimer, 1000, 0);
}

LOCAL void ICACHE_FLASH_ATTR pubfloat(void *arg) {
  os_printf("Entered pubfloat!\n");
  mqtt_session_t *pSession = (mqtt_session_t *)arg;
  float *data = (float *)(pSession->userData);
  char *dataStr = os_zalloc(20 * sizeof(char));
  ftoa(*data, dataStr, 2);
  int32_t dataLen = os_strlen(dataStr);
  os_printf("Encoded string: %s\tString length: %d\n", dataStr, dataLen);
  mqttSend(pSession, (uint8_t *)dataStr, dataLen, MQTT_MSG_TYPE_PUBLISH);
  os_timer_disarm(&pingTimer);
  os_timer_setfn(&pingTimer, (os_timer_func_t *)ping, arg);
  os_timer_arm(&pingTimer, 1000, 0);
}

LOCAL void ICACHE_FLASH_ATTR ping(void *arg) {
#ifdef DEBUG
  os_printf("Entered ping!\n");
#endif
  mqtt_session_t *pSession = (mqtt_session_t *)arg;
  mqttSend(pSession, NULL, 0, MQTT_MSG_TYPE_PINGREQ);
  os_timer_disarm(&pingTimer);
  os_timer_setfn(&pingTimer, (os_timer_func_t *)ping, arg);
  os_timer_arm(&pingTimer, 1000, 0);
}

LOCAL void ICACHE_FLASH_ATTR sub(void *arg) {
  os_printf("Entered sub!\n");
  mqtt_session_t *pSession = (mqtt_session_t *)arg;
  mqttSend(pSession, NULL, 0, MQTT_MSG_TYPE_SUBSCRIBE);

}

LOCAL void ICACHE_FLASH_ATTR discon(void *arg) {
  os_printf("Entered discon!\n");
  mqtt_session_t *pSession = (mqtt_session_t *)arg;
  mqttSend(pSession, NULL, 0, MQTT_MSG_TYPE_DISCONNECT);
  os_timer_disarm(&pingTimer);
  os_timer_disarm(&pubTimer);
}

LOCAL void ICACHE_FLASH_ATTR dataLog(void *arg) {
  mqtt_session_t *pSession = (mqtt_session_t *)arg;
  oneWire_t DS18;
  oneWire_t *pDS18 = &DS18;
  uint16_t temp = 0;
  sint32 time = 0;

  pDS18->temperature = 0;
  reset();
  transact(pDS18, SKIP_ROM);
  transact(pDS18, CONVERT_T);
  time = system_get_time() + 750000;
  while(system_get_time() < time);
  reset();
  transact(pDS18, SKIP_ROM);
  transact(pDS18, SCRATCH_READ);
  //os_printf("\nReceived onewire data: %x %x\n", pDS18->scratchpad[0], pDS18->scratchpad[1]);
  temp = ((uint16_t)pDS18->scratchpad[1] << 8) | pDS18->scratchpad[0];
  pDS18->temperature += (temp >> 4) + ((temp & 0x0F) * 0.0625);
  pSession->userData = (void *)&pDS18->temperature;
  os_printf("\nTemperature is: %d.%02d\n", (int)pDS18->temperature, (int)(pDS18->temperature * 100) % 100);
  pubfloat(pSession); // publish the temperature
  //discon(pSession);
  return;
}

void ICACHE_FLASH_ATTR user_init() {
  LOCAL mqtt_session_t globalSession;
  LOCAL mqtt_session_t *pGlobalSession = &globalSession;
  char ssid[32] = "yourwifissid";
  char passkey[64] = "yourwifipass";
  struct station_config stationConf;

  gpio_init(); // init gpio so we can use the LED & onewire bus
  uart_div_modify(0, UART_CLK_FREQ / 921600); // set UART to 921600
  wifi_status_led_install(0, PERIPHS_IO_MUX_GPIO0_U, FUNC_GPIO0); // set GPIO0 as status LED
  stationConf.bssid_set = 0;
  os_memcpy(&stationConf.ssid, ssid, 32); // copy the ssid and passkey into the station_config struct
  os_memcpy(&stationConf.password, passkey, 64);
  wifi_set_opmode_current(0x01); //station mode
  wifi_station_set_config_current(&stationConf); // tell it about our config, this auto-connects us as well

  // prepare the TCP/MQTT connection stuff
  // Adafruit IO is at 52.5.238.97
  // use strtoarr.py to generate these
  static const char ioUser[11] = { 0x4d, 0x72, 0x41, 0x75, 0x72, 0x65, 0x6c, 0x69, 0x75, 0x73, 0x52 }; // MrAureliusR
  static const uint8_t ioUser_len = 11;
  static const char ioKey[32] = { }; // use strtoarr.py to generate these
  static const uint8_t ioKey_len = 32;
  static const char ioTopic[27] = { 0x4d, 0x72, 0x41, 0x75, 0x72, 0x65, 0x6c, 0x69, 0x75, 0x73, 0x52, 0x2f, 0x66, 0x65, 0x65, 0x64, 0x73, 0x2f, 0x74, 0x65, 0x73, 0x74, 0x74, 0x6f, 0x70, 0x69, 0x63 }; // MrAureliusR/feeds/testtopic
  static const uint8_t ioTopic_len = 27;
  static const char clientID[5] = { 0x46, 0x52, 0x5a, 0x4e, 0x30 }; // FRZN0
  static const uint8_t clientID_len = 5;
  pGlobalSession->port = 1883; // mqtt port
  static const char adafruitIO_ip[4] = {52, 5, 238, 97};

  // copy all the above info into our mqtt_session_t
  os_memcpy(pGlobalSession->ip, adafruitIO_ip, 4);
  pGlobalSession->username_len = ioUser_len;
  pGlobalSession->username = os_zalloc(sizeof(uint8_t) * pGlobalSession->username_len);
  os_memcpy(pGlobalSession->username, ioUser, pGlobalSession->username_len);
  pGlobalSession->password_len = ioKey_len;
  pGlobalSession->password = os_zalloc(sizeof(uint8_t) * pGlobalSession->password_len);
  os_memcpy(pGlobalSession->password, ioKey, pGlobalSession->password_len);
  pGlobalSession->topic_name_len = ioTopic_len;
  pGlobalSession->topic_name = os_zalloc(sizeof(uint8_t) * pGlobalSession->topic_name_len);
  os_memcpy(pGlobalSession->topic_name, ioTopic, pGlobalSession->topic_name_len);
  pGlobalSession->client_id_len = clientID_len;
  pGlobalSession->client_id = os_zalloc(sizeof(uint8_t) * pGlobalSession->client_id_len);
  os_memcpy(pGlobalSession->client_id, clientID, pGlobalSession->client_id_len);

  // set up all the timers to connect and log data
  os_timer_setfn(&tcpTimer, (os_timer_func_t *)tcpConnect, pGlobalSession);
  os_timer_arm(&tcpTimer, 12000, 0);
  os_timer_setfn(&pingTimer, (os_timer_func_t *)con, pGlobalSession);
  os_timer_arm(&pingTimer, 16000, 0);
  os_timer_setfn(&pubTimer, (os_timer_func_t *)dataLog, pGlobalSession);
  os_timer_arm(&pubTimer, 20000, 1);
}