cpp-shapeways/swApiClient.cpp at master · Shapeways/cpp-shapeways · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include "swApiClient.h"

using namespace std;
using json = nlohmann::json;

/* Helper Functions */
struct curl_slist * swApiClient::SetAuthHeader() {
  char *headerStr = (char*)malloc(250);
  strcpy(headerStr, "Authorization: Bearer ");
  strcat(headerStr, this->accessToken.c_str());
  struct curl_slist *headers = NULL;
  headers = curl_slist_append(headers, headerStr);
  return headers;
}

static size_t WriteCallback(void *contents, size_t size, size_t numElements, void *userp)
{
    ((string*)userp)->append((char*)contents, size * numElements);
    return size * numElements;
}

/*
====================
swApiClient Constructor

  Constructs swApiClient
====================
*/
swApiClient::swApiClient( string cKey, string cSecret) {
    consumerKey = cKey;
    consumerSecret = cSecret;
    accessToken = "";
}

/*
====================
swApiClient Authorize

  Connect to the Shapeways API using oauth2
====================
*/
void swApiClient::Authorize() {

  CURL *curl;
  CURLcode res;
  string readBuffer;

  // prepare auth payload
  char *postData = (char*)malloc(250);
  strcpy(postData, "grant_type=client_credentials&client_id=");
  strcat(postData, this->consumerKey.c_str());
  strcat(postData, "&client_secret=");
  strcat(postData, this->consumerSecret.c_str());

  curl = curl_easy_init();
  if(curl) {
    curl_easy_setopt(curl, CURLOPT_URL, "https://api.shapeways.com/oauth2/token");
    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
    curl_easy_setopt(curl, CURLOPT_POSTFIELDS, postData);
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, &readBuffer);
    res = curl_easy_perform(curl);
    free(postData);

    // If non-200 response, log error
    if (res != 0) {
      throw runtime_error(curl_easy_strerror(res));
    }
    curl_easy_cleanup(curl);
    json j = json::parse(readBuffer);
    try {
      this->accessToken = j["access_token"];
    }
    catch (const std::exception& e) {
      throw runtime_error(readBuffer);
    }

  } else {
    throw runtime_error("Could not initialize curl object");
  }
}

/*
====================
swApiClient GetMaterials

  Connect to the Shapeways API using oauth2
====================
*/
string swApiClient::GetMaterials() {
  CURL *curl;
  CURLcode res;
  string readBuffer;

  // Ensure bearer token present
  if (this->accessToken == "") {
    throw runtime_error("No access token set: did you run .Authorize()?");
  }

  curl = curl_easy_init();
  struct curl_slist *headers = NULL;
  if(curl) {

    headers = this->SetAuthHeader();
    curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
    curl_easy_setopt(curl, CURLOPT_URL, "https://api.shapeways.com/materials/v1");
    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
    curl_easy_setopt(curl, CURLOPT_HTTPGET, 1L);
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, &readBuffer);
    res = curl_easy_perform(curl);
    if (res != 0) {
      string error = curl_easy_strerror(res);
      cout << error << endl;
      cout << readBuffer << endl;
      throw runtime_error(error);
    }
    curl_slist_free_all(headers);
    curl_easy_cleanup(curl);
  }
  else {
    throw runtime_error("Could not initialize curl object");
  }
  json j = json::parse(readBuffer);
  return j.dump();
}