From 6cea9f20929d8284babb9a666460492c6c7838b0 Mon Sep 17 00:00:00 2001
From: Thulinma <jaron@vietors.com>
Date: Mon, 9 Mar 2020 20:04:45 +0100
Subject: [PATCH] URIReader

---
 CMakeLists.txt     |  3 ++
 test/urireader.cpp | 99 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 102 insertions(+)
 create mode 100644 test/urireader.cpp

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 1be47bc2..eca106d8 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -707,6 +707,9 @@ add_test(LOGTest COMMAND urltest)
 add_executable(downloadertest test/downloader.cpp ${BINARY_DIR}/mist/.headers)
 target_link_libraries(downloadertest mist)
 add_test(DownloaderTest COMMAND downloadertest)
+add_executable(urireadertest test/urireader.cpp ${BINARY_DIR}/mist/.headers)
+target_link_libraries(urireadertest mist)
+add_test(URIReaderTest COMMAND urireadertest)
 add_executable(jsontest test/json.cpp ${BINARY_DIR}/mist/.headers)
 target_link_libraries(jsontest mist)
 add_test(JSONTest COMMAND jsontest)
diff --git a/test/urireader.cpp b/test/urireader.cpp
new file mode 100644
index 00000000..3781268c
--- /dev/null
+++ b/test/urireader.cpp
@@ -0,0 +1,99 @@
+#include "../lib/urireader.cpp"
+#include <iostream>
+
+class URITest : public Util::DataCallback{
+public:
+  int goStdin(bool useCallback = true);
+  int goHTTP(char *, bool useCallback = true);
+  void dump(const char *ptr, size_t size);
+  size_t wanted = 100000000;
+  void dataCallback(const char *ptr, size_t size);
+};
+
+void URITest::dataCallback(const char *ptr, size_t size){
+  dump(ptr, size);
+}
+
+void URITest::dump(const char *ptr, size_t size){
+  if (fwrite(ptr, sizeof(char), size, stdout) != size){INFO_MSG("error: %s", strerror(errno));}
+}
+
+int URITest::goStdin(bool useCallback){
+  HTTP::URIReader U;
+  HTTP::URL uri;
+  uri.path = "-";
+  U.open(uri);
+
+  if (useCallback){
+    MEDIUM_MSG("read from STDIN with callbacks");
+    U.readAll(*this);
+  }else{
+    MEDIUM_MSG("read from STDIN without callbacks");
+    char *dPtr = 0;
+    size_t dLen = 0;
+    U.readAll(dPtr, dLen);
+    dump(dPtr, dLen);
+    // INFO_MSG("length: %d", dLen);
+  }
+
+  return 0;
+}
+
+int URITest::goHTTP(char *uri, bool useCallback){
+  HTTP::URIReader d;
+  d.open(uri);
+
+  if (useCallback){
+    MEDIUM_MSG("read file or url with callbacks");
+    // d.readAll(*this);
+
+    while (!d.isEOF()){d.readSome(10486, *this);}
+
+  }else{
+    MEDIUM_MSG("read file or url without callbacks");
+    char *dPtr = 0;
+    size_t dLen = 0;
+    d.readAll(dPtr, dLen);
+    dump(dPtr, dLen);
+  }
+
+  return 0;
+}
+
+int main(int argc, char **argv){
+  //  Util::Config::printDebugLevel = 10;
+  Util::Config cfg;
+  cfg.activate();
+
+  URITest t;
+
+  if (!isatty(fileno(stdin))){
+    if (argv[1]){
+      t.goStdin(false);
+    }else{
+      t.goStdin();
+    }
+  }else{
+
+    if (argc == 1){
+      std::cout << "no arguments applied!" << std::endl;
+      std::cout << "usage: " << std::endl;
+      std::cout << "STDIN:\t urireader < filename " << std::endl;
+      std::cout << "URL:\t urireader http://url " << std::endl;
+      std::cout << "FILE:\t urireader path_to_file " << std::endl << std::endl;
+      std::cout << "Outputs content to stdout, use ' > outputfile' after the command to write "
+                   "contents to disk"
+                << std::endl
+                << std::endl;
+      return 0;
+    }else{
+      if (argv[2]){
+        t.goHTTP(argv[1], false);
+      }else{
+        t.goHTTP(argv[1]);
+      }
+    }
+  }
+
+  return 0;
+}