57 lines
1.4 KiB
C++
57 lines
1.4 KiB
C++
#include "crow.h"
|
|
|
|
extern "C" {
|
|
#include "ternarytrie.h"
|
|
}
|
|
|
|
int main() {
|
|
// Read in API key
|
|
char *api_key = getenv("LANDER_API_KEY");
|
|
|
|
if (api_key == NULL) {
|
|
printf("No API key provided.");
|
|
return 1;
|
|
}
|
|
|
|
TernaryTrie *trie = ternarytrie_init();
|
|
|
|
std::string file_path = "lander.data";
|
|
ternarytrie_populate(trie, file_path.c_str());
|
|
|
|
crow::SimpleApp app;
|
|
|
|
CROW_ROUTE(app, "/<string>")
|
|
.methods(crow::HTTPMethod::Post)(
|
|
[api_key, trie](const crow::request &req, std::string s) {
|
|
// Authenticate request
|
|
std::string provided_api_key = req.get_header_value("X-Api-Key");
|
|
|
|
if (strcmp(api_key, provided_api_key.c_str()) != 0) {
|
|
return crow::response(crow::status::UNAUTHORIZED);
|
|
}
|
|
|
|
bool res = ternarytrie_add(trie, s.c_str(), req.body.c_str());
|
|
|
|
if (!res) {
|
|
return crow::response(crow::status::CONFLICT);
|
|
}
|
|
|
|
return crow::response(crow::status::NO_CONTENT);
|
|
});
|
|
CROW_ROUTE(app, "/<string>")
|
|
.methods(crow::HTTPMethod::Get)(
|
|
[trie](crow::response &res, std::string s) {
|
|
char *payload = ternarytrie_search(trie, s.c_str());
|
|
|
|
if (payload != NULL) {
|
|
res.redirect(payload);
|
|
} else {
|
|
res.code = 404;
|
|
}
|
|
|
|
res.end();
|
|
});
|
|
|
|
app.port(18080).multithreaded().run();
|
|
}
|