From f794f2ff83ccd88cc96463d66a9ce46d90ad0497 Mon Sep 17 00:00:00 2001
From: Mo Langning <133737702+molangning@users.noreply.github.com>
Date: Sat, 25 Nov 2023 00:12:09 +0800
Subject: [PATCH] added new line checker for all push

---
 .bin/new-line-checker.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/.bin/new-line-checker.py b/.bin/new-line-checker.py
index 2258454e..c7716327 100755
--- a/.bin/new-line-checker.py
+++ b/.bin/new-line-checker.py
@@ -3,16 +3,16 @@
 # test string: ./.bin/new-line-checker.py "Fuzzing/file-extensions-all-cases.txt Fuzzing/file-extensions-lower-case.txt Fuzzing/file-extensions-upper-case.txt Fuzzing/file-extensions.txt"
 
 import os
+import sys
 
 print("[+] New line check")
 
-files = []
-dirs = ['sources/','blacklists/','whitelists/']
+files=sys.argv[1].split(" ")
 
-for i in dirs:
-    for root,_,file_list in os.walk(i):
-        for file in file_list:
-            files.append(os.path.join(root,file))
+for i in files:
+    if not os.path.isfile(i):
+        print("[!] %s does not exist!"%(i))
+        exit(2)
 
 for i in files:
     f=open(i,"r")
@@ -24,7 +24,7 @@ for i in files:
     if contents[-1] == '\n':
         print("[!] %s ends with a new line"%(i))
         exit(2)
-    # print("[+] %s passed new line check"%(i))
+    print("[+] %s passed new line check"%(i))
 
 print("[+] All files passed checks")
-exit(0)
\ No newline at end of file
+# exit(0)
\ No newline at end of file