implement option handling

1 files changed, 26 insertions, 2 deletions

diff --git a/src/srcut.cpp b/src/srcut.cpp
index 8c59440..44131a0 100644
--- a/src/srcut.cpp
+++ b/src/srcut.cpp
@@ -3,6 +3,7 @@
 #include <map>
 #include <sstream>
 #include <string>
+#include <unistd.h>
 #include <vector>
 
 using namespace std;
@@ -103,11 +104,34 @@ print(ostream& os, const string& key, const map<string, vector<string>>& contain
 int
 main(int argc, char* argv[])
 {
+        int show_tags = 0;
+        int ch;
+
+        while ( (ch = getopt(argc, argv, "t")) != -1 ) {
+                switch ( ch ) {
+                case 't':
+                        show_tags = 1;
+                        break;
+                default:
+                        cerr << "unbekannte option" << endl;
+                        return EXIT_FAILURE;
+                }
+        }
+        argc -= optind;
+        argv += optind;
+
         map<string, vector<string>> container;
 
         cin >> container;
 
-        for ( int i = 1; i < argc; ++i ) {
-                print(cout, argv[i], container);
+        if ( show_tags ) {
+                for ( const auto& it: container ) {
+                        cout << it.first << endl;
+                }
+        }
+        else {
+                for ( int i = 0; i < argc; ++i ) {
+                        print(cout, argv[i], container);
+                }
         }
 }