git.maemo.org Git - monky/blob - src/eve.c

   1 /* Conky, a system monitor, based on torsmo
   2  *
   3  * Copyright (c) 2008 Asbjørn Zweidorff Kjær
   4  * Copyright (c) 2005-2009 Brenden Matthews, Philip Kovacs, et. al.
   5  *      (see AUTHORS)
   6  * All rights reserved.
   7  *
   8  * This program is free software: you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation, either version 3 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  *
  20  */
  21
  22 #include "eve.h"
  23 #include "config.h"
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <unistd.h>
  27 #include <string.h>
  28 #include <sys/stat.h>
  29 #include <sys/types.h>
  30 #include <utime.h>
  31
  32 #include <libxml/parser.h>
  33 #include <libxml/tree.h>
  34 #include <libxml/xmlwriter.h>
  35
  36 #include <curl/curl.h>
  37 #include <curl/types.h>
  38 #include <curl/easy.h>
  39
  40 #include <time.h>
  41
  42 int num_chars = 0;
  43 Character eveCharacters[MAXCHARS];
  44
  45 static size_t write_data(void *ptr, size_t size, size_t nmemb, void *stream)
  46 {
  47         size_t realsize = 0;
  48         struct xmlData *data = 0;
  49         data = (struct xmlData *)stream;
  50         realsize = size * nmemb;
  51
  52         data->data = (char *)realloc(data->data, data->size + realsize + 1);
  53         if (data->data) {
  54                 memcpy(&(data->data[data->size]), ptr, realsize);
  55                 data->size += realsize;
  56                 data->data[data->size] = '\0';
  57         }
  58
  59         return realsize;
  60 }
  61
  62 int parseTrainingXml(char *data, Character * s)
  63 {
  64         char *skill, *level, *ends, *cache;
  65         xmlNodePtr n;
  66         xmlDocPtr doc = 0;
  67         xmlNodePtr root = 0;
  68         struct tm end_tm, cache_tm;
  69
  70         if (!data)
  71                 return 1;
  72
  73         doc = xmlReadMemory(data, strlen(data), "", NULL, 0);
  74         root = xmlDocGetRootElement(doc);
  75         for (n = root->children; n; n = n->next) {
  76                 if (n->type == XML_ELEMENT_NODE) {
  77                         if (!strcasecmp((const char *)n->name, "error")) {
  78                                 return 1;
  79                         } else if (!strcasecmp((const char *)n->name, "result")) {
  80                                 xmlNodePtr c;
  81                                 for (c = n->children; c; c = c->next) {
  82                                         if (!strcasecmp((const char *)c->name, "trainingEndTime")) {
  83                                                 ends = (char *)c->children->content;
  84                                         } else if (!strcasecmp((const char *)c->name, "trainingTypeID")) {
  85                                                 if (c->children->content)
  86                                                         skill = (char *)c->children->content;
  87                                         } else if (!strcasecmp((const char *)c->name, "trainingToLevel")) {
  88                                                 level = (char *)c->children->content;
  89                                         }
  90                                 }
  91                         } else if (!strcasecmp((const char *)n->name, "cachedUntil")) {
  92                                 cache = (char *)n->children->content;
  93                         }
  94                 }
  95         }
  96
  97         strptime(ends, "%Y-%m-%d %H:%M:%S", &end_tm);
  98         strptime(cache, "%Y-%m-%d %H:%M:%S", &cache_tm);
  99         s->skill = atoi(skill);
 100         s->level = atoi(level);
 101         s->ends = end_tm;
 102         s->cache = cache_tm;
 103
 104         xmlFreeDoc(doc);
 105         return 0;
 106 }
 107
 108 char *getXmlFromAPI(const char *userid, const char *apikey, const char *charid, const char *url)
 109 {
 110         struct curl_httppost *post = NULL;
 111         struct curl_httppost *last = NULL;
 112         struct xmlData chr;
 113         char *content;
 114         CURL *curl_handle;
 115         int rc = 0;
 116
 117         chr.data = NULL;
 118         chr.size = 0;
 119
 120
 121         curl_global_init(CURL_GLOBAL_ALL);
 122         curl_handle = curl_easy_init();
 123         curl_easy_setopt(curl_handle, CURLOPT_FAILONERROR, 1);
 124         curl_easy_setopt(curl_handle, CURLOPT_URL, url);
 125         curl_easy_setopt(curl_handle, CURLOPT_NOPROGRESS, 1);
 126         curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, write_data);
 127         curl_easy_setopt(curl_handle, CURLOPT_WRITEDATA, (void *)&chr);
 128
 129         if (userid != NULL && apikey != NULL && charid != NULL) {
 130                 curl_formadd(&post, &last, CURLFORM_COPYNAME, "userID", CURLFORM_COPYCONTENTS, userid, CURLFORM_END);
 131                 curl_formadd(&post, &last, CURLFORM_COPYNAME, "apiKey", CURLFORM_COPYCONTENTS, apikey, CURLFORM_END);
 132                 curl_formadd(&post, &last, CURLFORM_COPYNAME, "characterID", CURLFORM_COPYCONTENTS, charid, CURLFORM_END);
 133
 134                 curl_easy_setopt(curl_handle, CURLOPT_HTTPPOST, post);
 135         }
 136
 137         if ((rc = curl_easy_perform(curl_handle)) != CURLE_OK) {
 138                 return NULL;
 139         }
 140
 141         content = strdup(chr.data);
 142         curl_easy_cleanup(curl_handle);
 143
 144         return content;
 145 }
 146
 147 void init_eve(void)
 148 {
 149         int i;
 150
 151         for (i = 0; i < MAXCHARS; i++) {
 152                 eveCharacters[i].charid = NULL;
 153                 eveCharacters[i].skillname = NULL;
 154                 eveCharacters[i].time = NULL;
 155                 eveCharacters[i].level = 0;
 156                 eveCharacters[i].skill = 0;
 157                 eveCharacters[i].delay = 0;
 158         }
 159 }
 160
 161 char *eve(char *userid, char *apikey, char *charid)
 162 {
 163         Character *chr = NULL;
 164         const char *skillfile = "/tmp/.cesf";
 165         int i = 0;
 166         char *output = 0;
 167         char *timel = 0;
 168         char *skill = 0;
 169         char *content = 0;
 170         time_t now = 0;
 171         char *error = 0;
 172
 173
 174         for (i = 0; i < MAXCHARS; i++) {
 175                 if (eveCharacters[i].charid != NULL) {
 176                         if (strcasecmp(eveCharacters[i].charid, charid) == 0) {
 177                                 chr = &eveCharacters[i];
 178                                 break;
 179                         }
 180                 }
 181         }
 182
 183         if (!chr) {
 184                 if (num_chars == MAXCHARS - 1)
 185                         return NULL;
 186                 chr = &eveCharacters[num_chars];
 187                 chr->charid = strdup(charid);
 188                 num_chars++;
 189         }
 190
 191         if (chr->delay > 0) {
 192                 now = time(NULL);
 193                 if (now < chr->delay) {
 194                         output = strdup("Server error");
 195                         return output;
 196                 } else
 197                         chr->delay = 0;
 198         }
 199
 200         if (isCacheValid(chr->cache)) {
 201                 output = (char *)malloc(200 * sizeof(char));
 202                 timel = strdup(formatTime(&chr->ends));
 203                 sprintf(output, EVE_OUTPUT_FORMAT, chr->skillname, chr->level, timel);
 204                 free(timel);
 205                 return output;
 206         } else {
 207                 content = getXmlFromAPI(userid, apikey, charid, EVEURL_TRAINING);
 208                 if (content == NULL) {
 209                         error = strdup("Server error");
 210                         now = time(NULL);
 211                         now += (time_t) 1800;
 212                         chr->delay = now;
 213                         return error;
 214                 }
 215
 216                 if (parseTrainingXml(content, chr)) {
 217                         output = (char *)malloc(200 * sizeof(char));
 218                         sprintf(output, "API error");
 219                         return output;
 220                 }
 221
 222                 output = (char *)malloc(200 * sizeof(char));
 223                 timel = formatTime(&chr->ends);
 224                 skill = getSkillname(skillfile, chr->skill);
 225
 226                 chr->skillname = strdup(skill);
 227
 228                 sprintf(output, EVE_OUTPUT_FORMAT, chr->skillname, chr->level, timel);
 229                 free(skill);
 230                 return output;
 231         }
 232
 233 }
 234
 235 char *formatTime(struct tm *ends)
 236 {
 237         struct timeval tv;
 238         struct timezone tz;
 239         double offset = 0;
 240         time_t now = 0;
 241         time_t tEnds = 0;
 242         long lin = 0;
 243         long lie = 0;
 244         long diff = 0;
 245
 246         gettimeofday(&tv, &tz);
 247         offset = (double)(tz.tz_minuteswest * 60);
 248         now = time(NULL);
 249         tEnds = mktime(ends);
 250         lin = (long)now;
 251         lin += (long)offset;
 252         lie = (long)tEnds;
 253         diff = (lie - lin);
 254
 255         if (diff > 0) {
 256                 int days = (int)(diff / 60 / 60 / 24);
 257                 int hours = (int)((diff / 60 / 60) - (days * 24));
 258                 int minutes = (int)((diff / 60) - ((hours * 60) + (days * 60 * 24)));
 259                 int seconds = (int)(diff - ((minutes * 60) + (hours * 60 * 60) + (days * 60 * 60 * 24)));
 260                 char *output = malloc(100 * sizeof(char));
 261
 262                 if (days > 0)
 263                         sprintf(output, "%dd, %dh, %02dm and %02ds", days, hours, minutes, seconds);
 264                 else if (hours > 0)
 265                         sprintf(output, "%dh, %02dm and %02ds", hours, minutes, seconds);
 266                 else
 267                         sprintf(output, "%02dm and %02ds", minutes, seconds);
 268
 269                 return output;
 270         } else {
 271                 char *output = strdup("Done");
 272                 return output;
 273         }
 274 }
 275
 276 int isCacheValid(struct tm cached)
 277 {
 278         struct timeval tv;
 279         struct timezone tz;
 280         double offset = 0;
 281         time_t now = 0;
 282         time_t cache = 0;
 283         double diff = 0;
 284
 285         gettimeofday(&tv, &tz);
 286         offset = (double)(tz.tz_minuteswest * 60);
 287         now = time(NULL);
 288         cache = mktime(&cached);
 289         diff = difftime(cache, now);
 290
 291         if (diff < offset)
 292                 return 0;
 293         else
 294                 return 1;
 295 }
 296
 297 int file_exists(const char *filename)
 298 {
 299         struct stat fi;
 300
 301         if ((stat(filename, &fi)) == 0) {
 302                 if (fi.st_size > 0)
 303                         return 1;
 304                 else
 305                         return 0;
 306         } else
 307                 return 0;
 308 }
 309
 310 void writeSkilltree(char *content, const char *filename)
 311 {
 312         FILE *fp = fopen(filename, "w");
 313         fwrite(content, sizeof(char), strlen(content), fp);
 314         fclose(fp);
 315 }
 316
 317 char *getSkillname(const char *file, int skillid)
 318 {
 319         char *skilltree;
 320         char *skill = NULL;
 321         xmlNodePtr n;
 322         xmlDocPtr doc = 0;
 323         xmlNodePtr root = 0;
 324
 325         if (!file_exists(file)) {
 326                 skilltree = getXmlFromAPI(NULL, NULL, NULL, EVEURL_SKILLTREE);
 327                 writeSkilltree(skilltree, file);
 328                 free(skilltree);
 329         }
 330
 331         doc = xmlReadFile(file, NULL, 0);
 332         if (!doc)
 333                 return NULL;
 334
 335         root = xmlDocGetRootElement(doc);
 336
 337         for (n = root->children; n; n = n->next) {
 338                 xmlNodePtr o;
 339                 for (o = n->children; o; o = o->next) {
 340                         xmlNodePtr p;
 341                         for (p = o->children; p; p = p->next) {
 342                                 xmlNodePtr q;
 343                                 for (q = p->children; q; q = q->next) {
 344                                         xmlNodePtr r;
 345                                         for (r = q->children; r; r = r->next) {
 346                                                 xmlElementPtr ele = (xmlElementPtr) r;
 347                                                 xmlAttrPtr attr = (xmlAttrPtr) ele->attributes;
 348                                                 char *mySkill;
 349                                                 int id, assigned = 0;
 350
 351                                                 while (attr != NULL) {
 352                                                         if (!strcasecmp((const char *)attr->name, "typeName")) {
 353                                                                 mySkill = strdup((const char *)attr->children->content);
 354                                                                 assigned = 1;
 355                                                         } else if (!strcasecmp((const char *)attr->name, "typeID")) {
 356                                                                 id = atoi((const char *)attr->children->content);
 357                                                         }
 358                                                         attr = attr->next;
 359                                                 }
 360
 361                                                 if (id == skillid) {
 362                                                         skill = strdup(mySkill);
 363                                                         /* free(mySkill); */
 364                                                         goto END;
 365                                                 }
 366                                                 if (assigned)
 367                                                         free(mySkill);
 368                                         }
 369                                 }
 370                         }
 371                 }
 372         }
 373       END:
 374         xmlFreeDoc(doc);
 375
 376         return skill;
 377 }