Joel Grunbaum
2022-01-13 d2cfd3eeeb8b6af3b7ccda01e6c1ac581a2df398
bom.cpp
@@ -1,60 +1,57 @@
#include "bom.hpp"
#include "book.hpp"
#include "ftplibpp/ftplib.h"
#include "rapidxml/rapidxml.hpp"
#include <cstddef>
#include <cstdio>
#include <curl/curl.h>
#include <fstream>
#include <iterator>
#include <stdio.h>
#include <string>
#include <unordered_map>
#include <vector>
namespace bom
{
CURL* curl{NULL};
std::string result;
static size_t my_write(void* buffer, std::size_t size, std::size_t nmemb,
                       void* stream)
{
   std::string& text = *static_cast<std::string*>(stream);
   std::size_t totalsize = size * nmemb;
   text.append(static_cast<char*>(buffer), totalsize);
   return totalsize;
}
ftplib* ftp;
void initialise()
{
   curl_global_init(CURL_GLOBAL_DEFAULT);
   curl = curl_easy_init();
   if (curl) {
      curl_easy_setopt(curl, CURLOPT_URL,
                       "ftp://ftp.bom.gov.au/anon/gen/fwo/IDN60920.xml");
      curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, my_write);
      curl_easy_setopt(curl, CURLOPT_WRITEDATA, &result);
      curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);
   }
   ftp = new ftplib();
   ftp->Connect("ftp.bom.gov.au:21");
   ftp->Login("anonymous", "");
}
void destroy() { curl_global_cleanup(); }
void destroy()
{
   ftp->Quit();
   delete ftp;
}
void updateBom(std::unordered_map<std::string, book::Book>& bs)
{
   CURLcode res;
   res = curl_easy_perform(curl);
   curl_easy_cleanup(curl);
   if (CURLE_OK != res) {
      fprintf(stderr, "Curl failed: %d\n", res);
   }
   ftp->Get("bom_data.xml", "/anon/gen/fwo/IDN60920.xml",
            ftplib::transfermode::ascii);
   std::ifstream fs("bom_data.xml");
   std::vector<char> buffer{std::istreambuf_iterator<char>(fs),
                            istreambuf_iterator<char>()};
   buffer.push_back('\0');
   rapidxml::xml_document<> d;
   d.parse<0>(&result[0]);
   rapidxml::xml_node<>* n = d.first_node();
   for (rapidxml::xml_attribute<>* a = n->first_attribute(); a;
        a = a->next_attribute()) {
      std::cout << a->name() << ": " << a->value() << std::endl;
   d.parse<0>(&buffer[0]);
   // Walk stations
   for (rapidxml::xml_node<>* n = d.first_node()->last_node()->first_node(); n;
        n = n->next_sibling()) {
      int bom_id = std::stoi(n->first_attribute()->next_attribute()->value());
      for (auto& i : bs) {
         if (i.second.stationId == bom_id) {
            // Should be apparent temp
            i.second.bomPrice = std::stod(n->first_node()
                                              ->first_node()
                                              ->first_node()
                                              ->first_node()
                                              ->value());
         }
      }
   }
}
} // namespace bom