From: Lucie Scarlet <lucie@scarlet.moe>
Date: Fri, 12 Jul 2024 10:27:03 +0000 (+0200)
Subject: Added feedback for what project it generates for
X-Git-Url: https://git.chaotic.ninja/gitweb/lucie/?a=commitdiff_plain;h=7b8e412326c14551c1145f9eeff2bd327fec37aa;p=cmakegen.git

Added feedback for what project it generates for
---

diff --git a/cmakegen b/cmakegen
index 738dc02..9cd750d 100755
--- a/cmakegen
+++ b/cmakegen
@@ -10,16 +10,17 @@ CMAKE_FILE = getcwd() + "/CMakeLists.txt"
 def main() -> int:
     cmake_contents: str
     project_name : str = getcwd().split("/")[-1]
-    is_c: bool = True if argv[1] in ("-c",) else False
+    is_c: bool = True if len(argv) > 1 and argv[1] in ("-c",) else False
     template = CMAKE_C_TEMPLATE if is_c else CMAKE_TEMPLATE
     with open(template, "r") as f:
         cmake_contents = f.read()
     cmake_contents = cmake_contents.replace("PROJECT_TEMPLATE_NAME", project_name)
     with open(CMAKE_FILE, "w") as f:
         f.write(cmake_contents)
-    print("-" * len(CMAKE_FILE))
-    print(CMAKE_FILE)
-    print("-" * len(CMAKE_FILE))
+    output = CMAKE_FILE + f" - {'C' if is_c else 'C++'} project"
+    print("-" * len(output))
+    print(output)
+    print("-" * len(output))
     return 0
 
 if __name__ == "__main__":