add komens attachments downloading functionality

src/komens.cpp

@@ -25,7 +25,7 @@ using nlohmann::json;
 
 std::vector<allocation> komens_allocated;
 
-void insert_content(WINDOW *content_win, WINDOW *attachment_win, size_t i,
+void insert_content(WINDOW *content_win, WINDOW *attachment_win,
                     json &resp_from_api);
 
 void komens_page(koment_type type) {
@@ -145,8 +145,8 @@ void komens_page(koment_type type) {
   WINDOW *attachment_win = newwin(1, 1, LINES, COLS);
 
   insert_content(content_win, attachment_win,
-                 item_index(current_item(komens_choise_menu.menu)),
-                 resp_from_api);
+                 resp_from_api["Messages"][item_index(
+                     current_item(komens_choise_menu.menu))]);
 
   attron(COLOR_PAIR(COLOR_BLUE));
   mvprintw(LINES - 2, 0,
@@ -169,22 +169,61 @@ void komens_page(koment_type type) {
     case 'k':
       menu_driver(komens_choise_menu.menu, REQ_UP_ITEM);
       break;
+    default:
+      if (c >= '1' && c <= '9') {
+        size_t index = c - '0' - 1;
+        if (index < resp_from_api["Messages"][item_index(
+                        current_item(komens_choise_menu.menu))]["Attachments"]
+                        .size()) {
+
+          std::string default_path =
+              "~/Downloads/" +
+              resp_from_api["Messages"][item_index(current_item(
+                  komens_choise_menu.menu))]["Attachments"][index]["Name"]
+                  .get<std::string>();
+          char path[256];
+
+          // Create input prompt at bottom of screen
+          move(LINES - 1, 0);
+          clrtoeol();
+          printw("Save path [%s]: ", default_path.c_str());
+          echo();
+          curs_set(1);
+          getnstr(path, sizeof(path) - 1);
+          if (strlen(path) == 0)
+            strcpy(path, default_path.c_str());
+          noecho();
+          curs_set(0);
+          move(LINES - 1, 0);
+          clrtoeol();
+          refresh();
+
+          // Download the attachment
+          bakaapi::download_attachment(
+              resp_from_api["Messages"][item_index(current_item(
+                  komens_choise_menu.menu))]["Attachments"][index]["Id"]
+                  .get<std::string>(),
+              path);
+        }
+      }
+      break;
     }
+
     insert_content(content_win, attachment_win,
-                   item_index(current_item(komens_choise_menu.menu)),
-                   resp_from_api);
+                   resp_from_api["Messages"][item_index(
+                       current_item(komens_choise_menu.menu))]);
     wrefresh(komens_choise_menu.win);
   }
   delete_all(&komens_allocated);
 }
 
-void insert_content(WINDOW *content_win, WINDOW *attachment_win, size_t i,
-                    json &resp_from_api) {
+void insert_content(WINDOW *content_win, WINDOW *attachment_win,
+                    json &message) {
   wclear(content_win);
   mvwprintw(content_win, 0, 0, "%s",
-            html_to_string(resp_from_api.at("Messages")[i]["Text"]).c_str());
+            html_to_string(message.at("Text")).c_str());
   wrefresh(content_win);
-  if (!resp_from_api.at("Messages")[i]["Attachments"].empty()) {
+  if (!message.at("Attachments").empty()) {
 
     size_t max_item_lenght = 0;
     {
@@ -192,19 +231,15 @@ void insert_content(WINDOW *content_win, WINDOW *attachment_win, size_t i,
       size_t max_size_lenght = 0;
       size_t tmp_lenght;
 
-      for (size_t j = 0;
-           j < resp_from_api.at("Messages")[i]["Attachments"].size(); j++) {
-        tmp_lenght = resp_from_api.at("Messages")[i]["Attachments"][j]["Name"]
-                         .get<std::string>()
-                         .length();
+      for (size_t j = 0; j < message.at("Attachments").size(); j++) {
+        tmp_lenght =
+            message.at("Attachments")[j]["Name"].get<std::string>().length();
         if (tmp_lenght > max_name_lenght) {
           max_name_lenght = tmp_lenght;
         }
 
         tmp_lenght =
-            std::to_string(
-                resp_from_api.at("Messages")[i]["Attachments"][j]["Size"]
-                    .get<size_t>())
+            std::to_string(message.at("Attachments")[j]["Size"].get<size_t>())
                 .length();
         if (tmp_lenght > max_size_lenght) {
           max_size_lenght = tmp_lenght;
@@ -221,18 +256,14 @@ void insert_content(WINDOW *content_win, WINDOW *attachment_win, size_t i,
 
     wborder(attachment_win, 0, ' ', 0, 0, 0, ACS_HLINE, 0, ACS_HLINE);
     print_in_middle(attachment_win, 0, 0, max_item_lenght + 2, "Attachments",
-                    COLOR_RED);
-    for (size_t j = 0;
-         j < resp_from_api.at("Messages")[i]["Attachments"].size(); j++) {
+                    COLOR_PAIR(COLOR_RED));
+    for (size_t j = 0; j < message.at("Attachments").size(); j++) {
 
-      mvwprintw(attachment_win, j + 1, 2, "%s %s",
-                resp_from_api.at("Messages")[i]["Attachments"][j]["Name"]
-                    .get<std::string>()
-                    .c_str(),
-                std::to_string(
-                    resp_from_api.at("Messages")[i]["Attachments"][j]["Size"]
-                        .get<size_t>())
-                    .c_str());
+      mvwprintw(
+          attachment_win, j + 1, 2, "%s %s",
+          message.at("Attachments")[j]["Name"].get<std::string>().c_str(),
+          std::to_string(message.at("Attachments")[j]["Size"].get<size_t>())
+              .c_str());
 
       wattron(attachment_win, COLOR_PAIR(COLOR_MAGENTA));
       mvwprintw(attachment_win, j + 1, 0, "%zu>", j + 1);

src/main.cpp

@@ -1,8 +1,10 @@
 #include "main.h"
 #include "color.h"
+#include "flags.h"
 #include "helper_funcs.h"
 #include "main_menu.h"
 #include "net.h"
+#include "types.h"
 #include <csignal>
 #include <cstdlib>
 #include <curl/curl.h>
@@ -12,8 +14,6 @@
 #include <regex>
 #include <string>
 #include <unistd.h>
-#include "flags.h"
-#include "types.h"
 
 std::string baka_api_url;
 
@@ -36,14 +36,27 @@ int main(int argc, char **argv) {
 
     int opt;
     while ((opt = getopt(argc, argv, "hVvLS:")) != -1) {
-        switch (opt) {
-            case 'h': PrintHelp(); break;
-            case 'V': PrintVersion(); break;
-            case 'v': config.verbose = true; break;
-            case 'L': DeleteLogin(savedir_path); break;
-            case 'S': config.ignoressl = true; break;
-            default: std::cerr << RED"[ERROR]" << RESET" invalid option: " << (char)optopt << "\ntry: -h\n"; safe_exit(EINVAL);
-        }
+      switch (opt) {
+      case 'h':
+        PrintHelp();
+        break;
+      case 'V':
+        PrintVersion();
+        break;
+      case 'v':
+        config.verbose = true;
+        break;
+      case 'L':
+        DeleteLogin(savedir_path);
+        break;
+      case 'S':
+        config.ignoressl = true;
+        break;
+      default:
+        std::cerr << RED "[ERROR]" << RESET " invalid option: " << (char)optopt
+                  << "\ntry: -h\n";
+        safe_exit(EINVAL);
+      }
     }
 
     std::string urlfile_path = std::string(savedir_path) + "/url";
@@ -79,6 +92,6 @@ int main(int argc, char **argv) {
     get_input_and_login();
   }
   main_menu();
-  
+
   return 0;
 }

src/net.cpp

@@ -11,12 +11,17 @@
 #include <curl/curl.h>
 #include <curses.h>
 #include <dirent.h>
+#include <fcntl.h>
 #include <format>
 #include <fstream>
 #include <iostream>
 #include <nlohmann/json.hpp>
 #include <nlohmann/json_fwd.hpp>
+#include <stdlib.h>
 #include <string>
+#include <unistd.h>
+
+#define ROUND_UP(x, y) ((((x) + (y) - 1)) & ~((y) - 1))
 
 using nlohmann::json;
 
@@ -25,15 +30,44 @@ std::string access_token;
 CURL *curl = curl_easy_init();
 
 // Callback function to write data into a std::string
-size_t WriteCallback(void *contents, size_t size, size_t nmemb,
-                     std::string *userp) {
+size_t WriteCallback_to_string(void *contents, size_t size, size_t nmemb,
+                               void *userp) {
   size_t totalSize = size * nmemb;
-  userp->append((char *)contents, totalSize);
+  static_cast<std::string *>(userp)->append((char *)contents, totalSize);
   return totalSize;
 }
 
-std::tuple<std::string, int> send_curl_request(std::string endpoint, metod type,
-                                               std::string req_data) {
+size_t WriteCallback_to_file(void *contents, size_t size, size_t nmemb,
+                             void *userp) {
+  const size_t sector_size = 4096;
+  size_t total = size * nmemb;
+  void *aligned_buf = aligned_alloc(sector_size, ROUND_UP(total, sector_size));
+
+  // Zero out the buffer before copying
+  memset(aligned_buf, 0, ROUND_UP(total, sector_size));
+  memcpy(aligned_buf, contents, total);
+
+  // Cast userp to fstream*
+  std::fstream *fileStream = static_cast<std::fstream *>(userp);
+
+  // Write to the file using fstream
+  fileStream->write(static_cast<char *>(aligned_buf), total);
+
+  // Check if write was successful
+  if (fileStream->fail()) {
+    free(aligned_buf);
+    return 0;
+  }
+
+  free(aligned_buf);
+  return total;
+}
+
+std::tuple<std::string, int> send_curl_request(
+    std::string endpoint, metod type, std::string req_data,
+    size_t (*WriteCallback_function)(void *, size_t, size_t,
+                                     void *) = WriteCallback_to_string,
+    std::fstream *fileStream = nullptr) {
   std::string response;
   std::string url = baka_api_url + endpoint;
   if (type == GET) {
@@ -42,15 +76,19 @@ std::tuple<std::string, int> send_curl_request(std::string endpoint, metod type,
 
   if (curl) {
     curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
-    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
-    curl_easy_setopt(curl, CURLOPT_WRITEDATA, &response);
-
+    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback_function);
+    if (fileStream) {
+      curl_easy_setopt(curl, CURLOPT_WRITEDATA, fileStream);
+    } else {
+      curl_easy_setopt(curl, CURLOPT_WRITEDATA, &response);
+    }
     if (config.ignoressl) {
       curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L);
       curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0L);
     }
 
     curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
+    curl_easy_setopt(curl, CURLOPT_ACCEPT_ENCODING, "gzip, deflate");
 
     struct curl_slist *headers = NULL;
     headers = curl_slist_append(
@@ -77,7 +115,15 @@ std::tuple<std::string, int> send_curl_request(std::string endpoint, metod type,
     std::cerr << RED "[ERROR] " << RESET "curl not initialised\n";
     safe_exit(20);
   }
-  curl_easy_perform(curl); // Perform the request
+  CURLcode curl_return_code = curl_easy_perform(curl);
+  if (curl_return_code != CURLE_OK) {
+    std::cerr << RED "[ERROR] " << RESET << "curl_easy_perform() failed: "
+              << curl_easy_strerror(curl_return_code) << "\n";
+    safe_exit(21);
+  }
+  if (fileStream) {
+    fileStream->close();
+  }
 
   int http_code = 0;
   curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &http_code);
@@ -172,4 +218,47 @@ access_token_refreshed:
 
   return json::parse(response);
 }
+
+int download_attachment(std::string id, std::string path) {
+  if (config.verbose) {
+    std::clog << "downloading attachment please wait...\n";
+  }
+  if (path[0] == '~') {
+    path = std::string(std::getenv("HOME")) + path.substr(1);
+  }
+
+  std::string savedir_path = std::filesystem::path(path).parent_path().string();
+  if (!std::filesystem::exists(savedir_path)) {
+    if (!std::filesystem::create_directories(savedir_path)) {
+      std::cerr << RED "[ERROR] " RESET
+                << "Failed to create directory: " << savedir_path << "\n";
+      safe_exit(EXIT_FAILURE);
+    }
+  }
+
+  std::fstream fileStream(path, std::ios::out | std::ios::binary);
+  if (!fileStream.is_open()) {
+    std::cerr << RED "[ERROR] " RESET << "Failed to open file for writing\n";
+    safe_exit(EXIT_FAILURE);
+  }
+
+  is_access_token_empty();
+access_token_refreshed:
+  std::string req_data =
+      std::format("Authorization=Bearer&access_token={}", access_token);
+
+  auto [response, http_code] =
+      send_curl_request(std::format("/api/3/komens/attachment/{}", id), GET,
+                        req_data, WriteCallback_to_file, &fileStream);
+  if (http_code != 200) {
+    if (config.verbose) {
+      std::clog << BLUE "[LOG] " RESET << "download failed " << http_code
+                << " is non 200 response\n";
+    }
+    refresh_access_token();
+    goto access_token_refreshed;
+  }
+  return 0;
+}
+
 } // namespace bakaapi

src/net.h

@@ -19,5 +19,6 @@ void login(std::string username, std::string password);
 void refresh_access_token();
 json get_data_from_endpoint(const std::string &endpoint, metod metod,
                             std::string additional_data = "");
+int download_attachment(std::string id, std::string path);
 } // namespace bakaapi
 #endif