added set_telnet_port

This commit is contained in:
wt committed 2024-07-02 10:57:40 +07:00
1 parent df32a84f7e
commit 0ee71e5509
7 files changed
+76 -5

No files matched your search

+1 -1
View File
@@ -88,7 +88,7 @@ void set_telnet(const httplib::Request &request, httplib::Response &response) {
}
telnetfileout << stoi(new_telnet_enabled);
telnetfileout.close();
std::ifstream telnetfile("/etc/openrouter/telnet");
std::ifstream telnetfile("/etc/openrouter/telnet/enabled");
int telnet_enabled;
telnetfile >> telnet_enabled;
telnetfile.close();
+49 -1
View File
@@ -8,6 +8,8 @@
#include <fstream>
#include <string>
#define TELNET_PORT_REQUIRED_STRING R"({"required":"[token,port]"})"
void get_telnet_port(const httplib::Request &request, httplib::Response &response) {
std::cout << GREEN << request.path << RESET << " " << request.method << std::endl;
response.set_header("Access-Control-Allow-Origin", "*");
@@ -42,4 +44,50 @@ void get_telnet_port(const httplib::Request &request, httplib::Response &respons
responsedata << R"({"port":")" << telnet_port << R"("})";
response.set_content(responsedata.str(), JSON_TYPE);
}
}
}
void set_telnet_port(const httplib::Request &request, httplib::Response &response) {
std::cout << GREEN << request.path << RESET << " " << request.method << std::endl;
response.set_header("Access-Control-Allow-Origin", "*");
if (request.body.empty()) {
response.set_content(ACCESS_REQUIRED_STRING, JSON_TYPE);
return;
}
nlohmann::json json_body = nlohmann::json::parse(request.body);
if (json_body["token"] == nullptr) {
response.set_content(ACCESS_REQUIRED_STRING, JSON_TYPE);
return;
}
if (json_body["port"] == nullptr) {
response.set_content(TELNET_PORT_REQUIRED_STRING, JSON_TYPE);
return;
}
httplib::Client cli(AUTH_SERVICE);
std::stringstream body;
body << R"({"token":)" << json_body["token"] << R"(})";
httplib::Result res = cli.Post("/api/access", body.str().c_str(), JSON_TYPE);
if (res->status == httplib::OK_200) {
nlohmann::json resjson = nlohmann::json::parse(res->body);
if (resjson["access"] == "reject") {
response.set_content(resjson.dump().c_str(), JSON_TYPE);
return;
}
std::ofstream telnetportfileout("/etc/openrouter/telnet/port");
std::string new_telnet_port = json_body["port"];
telnetportfileout << stoi(new_telnet_port);
telnetportfileout.close();
system("/etc/init.d/telnet restart");
std::ifstream telnetportfile("/etc/openrouter/telnet/port");
int telnet_port;
telnetportfile >> telnet_port;
telnetportfile.close();
std::stringstream responsedata;
responsedata << R"({"port":")" << telnet_port << R"("})";
response.set_content(responsedata.str(), JSON_TYPE);
}
}
+1
View File
@@ -7,5 +7,6 @@ void get_telnet(const httplib::Request &request, httplib::Response &response);
void set_telnet(const httplib::Request &request, httplib::Response &response);
void get_telnet_port(const httplib::Request &request, httplib::Response &response);
void set_telnet_port(const httplib::Request &request, httplib::Response &response);
#endif // SETTINGS_TELNET_HPP