main.cpp: added options -file, -buf and -ss to specify the TokenList interface [skip ci]

Author: firewave

main.cpp

@@ -9,8 +9,9 @@
 #include <cstring>
 #include <fstream>
 #include <iostream>
-#include <sys/stat.h>
+#include <sstream>
 #include <string>
+#include <sys/stat.h>
 #include <utility>
 #include <vector>
 
@@ -27,7 +28,12 @@ int main(int argc, char **argv)
 {
     bool error = false;
     const char *filename = nullptr;
-    bool use_istream = false;
+    enum {
+        File,
+        Istream,
+        Sstream,
+        CharBuffer
+    } toklist_inf = File;
     bool fail_on_error = false;
     bool linenrs = false;
 
@@ -86,8 +92,8 @@ int main(int argc, char **argv)
                     }
                     dui.includes.emplace_back(std::move(value));
                 } else if (std::strncmp(arg, "-is",3)==0) {
+                    toklist_inf = Istream;
                     found = true;
-                    use_istream = true;
                 }
                 break;
             case 's':
@@ -101,6 +107,10 @@ int main(int argc, char **argv)
                     }
                     dui.std = std::move(value);
                 }
+                else if (std::strncmp(arg, "-ss",3)==0) {
+                    toklist_inf = Sstream;
+                    found = true;
+                }
                 break;
             case 'q':
                 found = true;
@@ -111,13 +121,25 @@ int main(int argc, char **argv)
                 error_only = true;
                 break;
             case 'f':
-                found = true;
-                fail_on_error = true;
+                if (std::strncmp(arg, "-file",5)==0) {
+                    toklist_inf = File;
+                    found = true;
+                }
+                else {
+                    fail_on_error = true;
+                    found = true;
+                }
                 break;
             case 'l':
                 linenrs = true;
                 found = true;
                 break;
+            case 'b':
+                if (std::strncmp(arg, "-buf",4)==0) {
+                    toklist_inf = CharBuffer;
+                    found = true;
+                }
+                break;
             }
             if (!found) {
                 std::cout << "error: option '" << arg << "' is unknown." << std::endl;
@@ -148,7 +170,10 @@ int main(int argc, char **argv)
         std::cout << "  -UNAME          Undefine NAME." << std::endl;
         std::cout << "  -std=STD        Specify standard." << std::endl;
         std::cout << "  -q              Quiet mode (no output)." << std::endl;
-        std::cout << "  -is             Use std::istream interface." << std::endl;
+        std::cout << "  -file           Use file TokenList interface (default)." << std::endl;
+        std::cout << "  -is             Use std::istream TokenList interface." << std::endl;
+        std::cout << "  -ss             Use std::stringstream interface." << std::endl;
+        std::cout << "  -buf            Use char buffer TokenList interface." << std::endl;
         std::cout << "  -e              Output errors only." << std::endl;
         std::cout << "  -f              Fail when errors were encountered (exitcode 1)." << std::endl;
         std::cout << "  -l              Print lines numbers." << std::endl;
@@ -188,8 +213,21 @@ int main(int argc, char **argv)
     simplecpp::TokenList outputTokens(files);
     {
         simplecpp::TokenList *rawtokens;
-        if (use_istream) {
-            rawtokens = new simplecpp::TokenList(f, files,filename,&outputList);
+        if (toklist_inf == Istream) {
+            rawtokens = new simplecpp::TokenList(f,files,filename,&outputList);
+        }
+        else if (toklist_inf == Sstream || toklist_inf == CharBuffer) {
+            std::ostringstream oss;
+            oss << f.rdbuf();
+            f.close();
+            const std::string s = oss.str();
+            if (toklist_inf == Sstream) {
+                std::istringstream iss(s);
+                rawtokens = new simplecpp::TokenList(iss,files,filename,&outputList);
+            }
+            else {
+                rawtokens = new simplecpp::TokenList(s.data(),s.size(),files,filename,&outputList);
+            }
         } else {
             f.close();
             rawtokens = new simplecpp::TokenList(filename,files,&outputList);