From 5c912a2abeb947bbd98f66ba815e233f7a031462 Mon Sep 17 00:00:00 2001
From: CnTeng <rxsnakepi@gmail.com>
Date: Sat, 31 Aug 2024 15:49:43 +0800
Subject: [PATCH] fix(parser): add whitespace infinitely to headers

---
 internal/parser/parser.go | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/internal/parser/parser.go b/internal/parser/parser.go
index 93e2e9e..f4d79e0 100644
--- a/internal/parser/parser.go
+++ b/internal/parser/parser.go
@@ -37,8 +37,10 @@ type Document struct {
 	Valid     bool
 }
 
-var caser = cases.Title(language.Und)
-var metaDataRegex = regexp.MustCompile("^# @")
+var (
+	caser         = cases.Title(language.Und)
+	metaDataRegex = regexp.MustCompile("^# @")
+)
 
 func parseHTTPLine(line string) (method string, url string, version string) {
 	method = ""
@@ -120,10 +122,10 @@ func parseSection(section string, document *Document) Section {
 		} else if in_header {
 			if strings.Contains(line, ":") {
 				httpVersion := parsedSection.Version
-				line = strings.Trim(line, " ")
+				line = strings.TrimSpace(line)
 				splits := strings.Split(line, ":")
 				headerName := strings.ToLower(splits[0])
-				headerValue := strings.Join(splits[1:], ":")
+				headerValue := strings.TrimSpace(strings.Join(splits[1:], ":"))
 				if httpVersion != "HTTP/2" && httpVersion != "HTTP/3" {
 					headerName = caser.String(headerName)
 				}
@@ -232,7 +234,7 @@ func parser(filepath string, flags config.ConfigFlags) {
 			if flags.Verbose {
 				log.Info("Writing", filepath, documentString)
 			}
-			err := os.WriteFile(filepath, []byte(documentString), 0644)
+			err := os.WriteFile(filepath, []byte(documentString), 0o644)
 			if err != nil {
 				log.Fatal("Error writing file", "error", err)
 			}
@@ -245,5 +247,4 @@ func parser(filepath string, flags config.ConfigFlags) {
 			}
 		}
 	}
-
-}
\ No newline at end of file
+}