From 7f24374ce0f2859bf4eb9f9c92b5ab2283132ade Mon Sep 17 00:00:00 2001
From: firewave <firewave@users.noreply.github.com>
Date: Thu, 3 Jul 2025 12:37:26 +0200
Subject: [PATCH] fixed #462 - added CLI option `-l` to print line numbers

---
 main.cpp      |  8 +++++++-
 simplecpp.cpp | 15 ++++++++++++---
 simplecpp.h   |  4 ++--
 3 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/main.cpp b/main.cpp
index 2397e847..366e412f 100644
--- a/main.cpp
+++ b/main.cpp
@@ -18,6 +18,7 @@ int main(int argc, char **argv)
     const char *filename = nullptr;
     bool use_istream = false;
     bool fail_on_error = false;
+    bool linenrs = false;
 
     // Settings..
     simplecpp::DUI dui;
@@ -74,6 +75,10 @@ int main(int argc, char **argv)
                 fail_on_error = true;
                 found = true;
                 break;
+            case 'l':
+                linenrs = true;
+                found = true;
+                break;
             }
             if (!found) {
                 std::cout << "error: option '" << arg << "' is unknown." << std::endl;
@@ -107,6 +112,7 @@ int main(int argc, char **argv)
         std::cout << "  -is             Use std::istream 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;
         std::exit(0);
     }
 
@@ -137,7 +143,7 @@ int main(int argc, char **argv)
     // Output
     if (!quiet) {
         if (!error_only)
-            std::cout << outputTokens.stringify() << std::endl;
+            std::cout << outputTokens.stringify(linenrs) << std::endl;
 
         for (const simplecpp::Output &output : outputList) {
             std::cerr << output.location.file() << ':' << output.location.line << ": ";
diff --git a/simplecpp.cpp b/simplecpp.cpp
index b1505cb6..8da158d1 100644
--- a/simplecpp.cpp
+++ b/simplecpp.cpp
@@ -546,24 +546,33 @@ void simplecpp::TokenList::push_back(Token *tok)
     backToken = tok;
 }
 
-void simplecpp::TokenList::dump() const
+void simplecpp::TokenList::dump(bool linenrs) const
 {
-    std::cout << stringify() << std::endl;
+    std::cout << stringify(linenrs) << std::endl;
 }
 
-std::string simplecpp::TokenList::stringify() const
+std::string simplecpp::TokenList::stringify(bool linenrs) const
 {
     std::ostringstream ret;
     Location loc(files);
+    bool filechg = true;
     for (const Token *tok = cfront(); tok; tok = tok->next) {
         if (tok->location.line < loc.line || tok->location.fileIndex != loc.fileIndex) {
             ret << "\n#line " << tok->location.line << " \"" << tok->location.file() << "\"\n";
             loc = tok->location;
+            filechg = true;
+        }
+
+        if (linenrs && filechg) {
+            ret << loc.line << ": ";
+            filechg = false;
         }
 
         while (tok->location.line > loc.line) {
             ret << '\n';
             loc.line++;
+            if (linenrs)
+                ret << loc.line << ": ";
         }
 
         if (sameline(tok->previous, tok))
diff --git a/simplecpp.h b/simplecpp.h
index 43a03730..2285e76c 100644
--- a/simplecpp.h
+++ b/simplecpp.h
@@ -288,8 +288,8 @@ namespace simplecpp {
         }
         void push_back(Token *tok);
 
-        void dump() const;
-        std::string stringify() const;
+        void dump(bool linenrs = false) const;
+        std::string stringify(bool linenrs = false) const;
 
         void readfile(Stream &stream, const std::string &filename=std::string(), OutputList *outputList = nullptr);
         void constFold();