add group class

Author: toddtreece

src/AdafruitIO_Group.cpp

@@ -0,0 +1,359 @@
+//
+// Adafruit invests time and resources providing this open source code.
+// Please support Adafruit and open source hardware by purchasing
+// products from Adafruit!
+//
+// Copyright (c) 2015-2016 Adafruit Industries
+// Author: Todd Treece
+// Licensed under the MIT license.
+//
+// All text above must be included in any redistribution.
+//
+#include "AdafruitIO_Group.h"
+#include "AdafruitIO.h"
+
+AdafruitIO_Group::AdafruitIO_Group(AdafruitIO *io, const char *n):AdafruitIO_MQTT()
+{
+  _io = io;
+  name = n;
+
+  _init();
+}
+
+AdafruitIO_Group::~AdafruitIO_Group()
+{
+  if(_sub)
+    delete _sub;
+
+  if(_pub)
+    delete _pub;
+
+  if(data)
+    delete data;
+
+  if(_topic)
+    free(_topic);
+
+  if(_group_url)
+    free(_group_url);
+
+  if(_create_url)
+    free(_create_url);
+}
+
+void AdafruitIO_Group::set(const char *feed, char *value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+void AdafruitIO_Group::set(const char *feed, bool value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+void AdafruitIO_Group::set(const char *feed, String value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+void AdafruitIO_Group::set(const char *feed, int value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+void AdafruitIO_Group::set(const char *feed, unsigned int value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+void AdafruitIO_Group::set(const char *feed, long value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+void AdafruitIO_Group::set(const char *feed, unsigned long value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+void AdafruitIO_Group::set(const char *feed, float value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+void AdafruitIO_Group::set(const char *feed, double value)
+{
+  AdafruitIO_Data *f = getFeed(feed);
+  f->setValue(value);
+}
+
+bool AdafruitIO_Group::save()
+{
+
+  if(data == NULL)
+    return false;
+
+  char csv[150];
+  AdafruitIO_Data *cur_data = data;
+
+  strcpy(csv, "");
+
+  while(cur_data != NULL) {
+
+    strcat(csv, cur_data->feedName());
+    strcat(csv, ",");
+    strcat(csv, cur_data->toChar());
+    strcat(csv, "\n");
+
+    cur_data = cur_data->next_data;
+
+  }
+
+  return _pub->publish(csv);
+
+}
+
+AdafruitIO_Data* AdafruitIO_Group::getFeed(const char *feed)
+{
+  uint8_t i;
+
+  if(data == NULL) {
+    data = new AdafruitIO_Data(feed);
+    return data;
+  }
+
+  AdafruitIO_Data *cur_data = data;
+
+  while(cur_data != NULL) {
+
+    if(strcmp(cur_data->feedName(), feed) == 0) {
+      return cur_data;
+    }
+
+    if(! cur_data->next_data) {
+      cur_data->next_data = new AdafruitIO_Data(feed);
+      return cur_data->next_data;
+    }
+
+    cur_data = cur_data->next_data;
+
+  }
+
+}
+
+void AdafruitIO_Group::onMessage(AdafruitIODataCallbackType cb)
+{
+  uint8_t i;
+
+  if(_groupCallback == NULL) {
+    _groupCallback = new AdafruitIOGroupCallback(cb);
+    return;
+  }
+
+  AdafruitIOGroupCallback *cur_cb = _groupCallback;
+
+  while(cur_cb != NULL) {
+
+    if(! cur_cb->next_cb) {
+      cur_cb->next_cb = new AdafruitIOGroupCallback(cb);
+      return;
+    }
+
+    cur_cb = cur_cb->next_cb;
+
+  }
+
+}
+
+void AdafruitIO_Group::onMessage(const char *feed, AdafruitIODataCallbackType cb)
+{
+  uint8_t i;
+
+  if(_groupCallback == NULL) {
+    _groupCallback = new AdafruitIOGroupCallback(feed, cb);
+    return;
+  }
+
+  AdafruitIOGroupCallback *cur_cb = _groupCallback;
+
+  while(cur_cb != NULL) {
+
+    if(strcmp(cur_cb->feed, feed) == 0) {
+      return;
+    }
+
+    if(! cur_cb->next_cb) {
+      cur_cb->next_cb = new AdafruitIOGroupCallback(feed, cb);
+      return;
+    }
+
+    cur_cb = cur_cb->next_cb;
+
+  }
+
+}
+
+void AdafruitIO_Group::call(AdafruitIO_Data *d)
+{
+  uint8_t i;
+
+  if(_groupCallback == NULL) {
+    return;
+  }
+
+  AdafruitIOGroupCallback *cur_cb = _groupCallback;
+
+  while(cur_cb) {
+
+    if(strcmp(cur_cb->feed, d->feedName()) == 0 || cur_cb->feed == NULL) {
+      cur_cb->dataCallback(d);
+    }
+
+    cur_cb = cur_cb->next_cb;
+
+  }
+
+}
+
+void AdafruitIO_Group::subCallback(char *val, uint16_t len)
+{
+
+  char *line;
+  char *name;
+  char *value;
+
+  if(_groupCallback == NULL)
+    return;
+
+  while((line = strtok_r(val, "\n", &val)) != NULL) {
+
+    name = strtok_r(line, ",", &line);
+
+    // couldn't grab name from line, move on
+    if(! name)
+      continue;
+
+    // don't handle location for now
+    if(strcmp(name, "location") == 0)
+      continue;
+
+    value = strtok_r(line, ",", &line);
+
+    // no value? move on
+    if(! value)
+      continue;
+
+    AdafruitIO_Data *feed = getFeed(name);
+
+    // we couldn't get the data, move on
+    if(! feed)
+      continue;
+
+    feed->setValue(value);
+    call(feed);
+
+  }
+
+}
+
+void AdafruitIO_Group::setLocation(double lat, double lon, double ele)
+{
+  uint8_t i;
+
+  if(data == NULL) {
+    return;
+  }
+
+  AdafruitIO_Data *cur_data = data;
+
+  while(cur_data) {
+    cur_data->setLocation(lat, lon, ele);
+    cur_data = cur_data->next_data;
+  }
+
+}
+
+bool AdafruitIO_Group::exists()
+{
+  _io->_http->startRequest(_group_url, HTTP_METHOD_GET);
+  _io->_http->sendHeader("X-AIO-Key", _io->_key);
+  _io->_http->endRequest();
+  int status = _io->_http->responseStatusCode();
+  _io->_http->responseBody(); // needs to be read even if not used
+  return status == 200;
+}
+
+bool AdafruitIO_Group::create()
+{
+  String body = "name=";
+  body += name;
+
+  _io->_http->startRequest(_create_url, HTTP_METHOD_POST);
+  _io->_http->sendHeader(HTTP_HEADER_CONTENT_TYPE, "application/x-www-form-urlencoded");
+  _io->_http->sendHeader(HTTP_HEADER_CONTENT_LENGTH, body.length());
+  _io->_http->sendHeader("X-AIO-Key", _io->_key);
+  _io->_http->endRequest();
+  _io->_http->write((const byte*)body.c_str(), body.length());
+
+  int status = _io->_http->responseStatusCode();
+  _io->_http->responseBody(); // needs to be read even if not used
+  return status == 201;
+}
+
+
+void AdafruitIO_Group::_init()
+{
+
+  // dynamically allocate memory for mqtt topic and REST URLs
+  _topic = (char *) malloc(sizeof(char) * (strlen(_io->_username) + strlen(name) + 8)); // 8 extra chars for /g/, /csv & null termination
+  _group_url = (char *) malloc(sizeof(char) * (strlen(_io->_username) + strlen(name) + 16)); // 16 extra for api path & null term
+  _create_url = (char *) malloc(sizeof(char) * (strlen(_io->_username) + 15)); // 15 extra for api path & null term
+
+  data = 0;
+
+  if(_topic && _create_url && _group_url) {
+
+    // build topic string
+    strcpy(_topic, _io->_username);
+    strcat(_topic, "/g/");
+    strcat(_topic, name);
+    strcat(_topic, "/csv");
+
+    // build feed url string
+    strcpy(_group_url, "/api/v2/");
+    strcat(_group_url, _io->_username);
+    strcat(_group_url, "/groups/");
+    strcat(_group_url, name);
+
+    // build create url string
+    strcpy(_create_url, "/api/v2/");
+    strcat(_create_url, _io->_username);
+    strcat(_create_url, "/groups");
+
+    // setup subscription
+    _sub = new Adafruit_MQTT_Subscribe(_io->_mqtt, _topic);
+    _pub = new Adafruit_MQTT_Publish(_io->_mqtt, _topic);
+    _io->_mqtt->subscribe(_sub);
+
+    _sub->setCallback(this, &AdafruitIO_MQTT::subCallback);
+
+  } else {
+
+    // malloc failed
+    _topic = 0;
+    _create_url = 0;
+    _group_url = 0;
+    _sub = 0;
+    _pub = 0;
+
+  }
+
+}