fix print_palette for multi-args

mcpcpc · Jul 14, 2020 · d44ffc3 · d44ffc3
1 parent 4fd3838
commit d44ffc3
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/kfc.c b/kfc.c
@@ -194,8 +194,8 @@ main(int argc, char **argv)
 
 	extern char *optarg;
 	extern int optind, optopt;
-	int eflag, sflag, rflag, lflag;
-	eflag = sflag = rflag = lflag = 0;
+	int eflag, sflag, rflag, lflag, pflag;
+	eflag = sflag = rflag = lflag = pflag = 0;
 	p.mode = "dark";
 
 	if ( (p.conf = getenv("XDG_CONFIG_HOME")) == NULL )
@@ -250,7 +250,7 @@ main(int argc, char **argv)
 				p.sval = optarg;
                 break;
             case 'p':
-				print_palette();
+				pflag++;
 				break;
             case ':':
                 fprintf(stderr, "Option -%c requires an operand\n", optopt);
@@ -288,5 +288,10 @@ main(int argc, char **argv)
 		select_palette();
 	}
 
+	if (pflag)
+	{
+	    print_palette();
+	}
+
     return 0;
 }