72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"log/slog"
|
|
"strings"
|
|
)
|
|
|
|
type Headers struct {
|
|
Method string
|
|
Uri string
|
|
Proto string
|
|
|
|
KV map[string]string
|
|
}
|
|
|
|
func readHeaders(reader *bufio.Reader) *Headers {
|
|
headers := Headers{}
|
|
kvHeaders := make(map[string]string)
|
|
versionLine, err := reader.ReadString('\n')
|
|
if err != nil {
|
|
slog.Error("Could not read version line", "err", err.Error())
|
|
return nil
|
|
}
|
|
|
|
method, uri, proto := parseRequestLine(versionLine)
|
|
if method == "" || uri == "" || proto == "" {
|
|
slog.Error("Invalid request line", "line", versionLine)
|
|
return nil
|
|
}
|
|
|
|
headers.Method = method
|
|
headers.Uri = uri
|
|
headers.Proto = proto
|
|
|
|
headerStr, err := reader.ReadString('\n')
|
|
if err != nil {
|
|
slog.Error("Could not read headers", "err", err.Error())
|
|
return nil
|
|
}
|
|
|
|
for strings.TrimSpace(headerStr) != "" {
|
|
header := strings.SplitN(headerStr, ":", 2)
|
|
if len(header) == 2 {
|
|
kvHeaders[strings.TrimSpace(header[0])] = strings.TrimSpace(header[1])
|
|
}
|
|
headerStr, err = reader.ReadString('\n')
|
|
if err != nil {
|
|
slog.Error("Could not read headers", "err", err.Error())
|
|
return nil
|
|
}
|
|
}
|
|
headers.KV = kvHeaders
|
|
return &headers
|
|
}
|
|
|
|
func (m *Headers) debugPrintHeaders() {
|
|
for key, value := range m.KV {
|
|
fmt.Printf("%s: %s\n", key, value)
|
|
}
|
|
}
|
|
|
|
func parseRequestLine(versionLine string) (string, string, string) {
|
|
cleanLine := strings.TrimSpace(versionLine)
|
|
parts := strings.Split(cleanLine, " ")
|
|
if len(parts) != 3 {
|
|
return "", "", ""
|
|
}
|
|
return parts[0], parts[1], parts[2]
|
|
}
|