Added distro flag

Author: TanmayPatil105

src/fetch.cpp

@@ -436,9 +436,14 @@ void print_process(string art,string color_name)
 /**
  * Utility to print ascii art of Distro
  */
-void print(string color_name)
+void print(string color_name,string distro_name)
 {
-    string os = getOS("/etc/os-release");
+    string os = distro_name;
+
+    if (distro_name == "def")
+    {
+        os = getOS("/etc/os-release");
+    }
 
     map<string, string> ascii_arts = {{"Ubuntu", "ubuntu.ascii"},
                                       {"Debian", "debian.ascii"},
@@ -481,4 +486,8 @@ void print(string color_name)
             return;
         }
     }
+
+    print_process("linux.ascii",color_name);
+    
+    return;
 }

src/fetch.h

@@ -55,7 +55,7 @@ vector<string> getGPU();
 
 string getPackages();
 
-void print(string color);
+void print(string color,string distro_name);
 
 // util.cpp
 

src/main.cpp

@@ -15,9 +15,10 @@ int main(int argc, char *argv[])
 {
     bool test_mode = false;
     string color_name = "def"s;
+    string distro_name = "def"s;
 
     int opt;
-    while((opt = getopt(argc, argv, "ta:")) != -1) 
+    while((opt = getopt(argc, argv, "ta:d:")) != -1) 
     { 
         switch(opt) 
         { 
@@ -27,6 +28,9 @@ int main(int argc, char *argv[])
             case 'a':
                 color_name = string(optarg);
                 break;
+            case 'd':
+                distro_name = string(optarg);
+                break;
             default:
                 return 1;
         }
@@ -45,7 +49,7 @@ int main(int argc, char *argv[])
         return 1;
     }
 
-    print(color_name);
+    print(color_name,distro_name);
     DisplayInfo();
 
     return 0;