From 2f9aeb762294fe96606c76c2fefa17a6540639b5 Mon Sep 17 00:00:00 2001
From: BiSaXa <1669855+BiSaXa@users.noreply.github.com>
Date: Sat, 27 Aug 2022 23:26:28 +0300
Subject: some cleaning

---
 main.go | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index 417aeeb..7978414 100644
--- a/main.go
+++ b/main.go
@@ -17,12 +17,12 @@ func main() {
 	if err != nil { // If it's not a directory
 		file, err := os.Open(os.Args[1])
 		utils.CheckError(err)
-		utils.HeaderOut(file)
+		messages.ParseHeader(file)
 		for {
-			code := messages.MessageTypeCheck(file)
+			code := messages.ParseMessage(file)
 			if code == 7 {
-				messages.MessageTypeCheck(file)
-				break // TODO: Check last CustomData
+				messages.ParseMessage(file)
+				break
 			}
 		}
 		defer file.Close()
@@ -30,16 +30,15 @@ func main() {
 	for _, fileinfo := range files { // If it is a directory
 		file, err := os.Open(os.Args[1] + fileinfo.Name())
 		utils.CheckError(err)
-		utils.HeaderOut(file)
+		messages.ParseHeader(file)
 		for {
-			code := messages.MessageTypeCheck(file)
+			code := messages.ParseMessage(file)
 			if code == 7 {
-				messages.MessageTypeCheck(file)
-				break // TODO: Check last CustomData
+				messages.ParseMessage(file)
+				break
 			}
 		}
 		defer file.Close()
 	}
-
 	fmt.Scanln()
 }
-- 
cgit v1.2.3