From 70b4ee7e82482bfbc4c7cdb88c1f8e1ba99467e5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Magnus=20=C3=85hall?= <magnus.ahall@gibon.se>
Date: Thu, 5 Oct 2023 10:56:10 +0200
Subject: [PATCH] Initial commit

---
 go.mod |  3 +++
 pkg.go | 79 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 82 insertions(+)
 create mode 100644 go.mod
 create mode 100644 pkg.go

diff --git a/go.mod b/go.mod
new file mode 100644
index 0000000..4aa12be
--- /dev/null
+++ b/go.mod
@@ -0,0 +1,3 @@
+module git.gibonuddevalla.se/go/httplog
+
+go 1.21.0
diff --git a/pkg.go b/pkg.go
new file mode 100644
index 0000000..77a3b98
--- /dev/null
+++ b/pkg.go
@@ -0,0 +1,79 @@
+package httplog
+
+import (
+	// Standard
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"time"
+)
+
+type T struct {
+	logdir string
+}
+
+type HttpMessage struct {
+	Sequence  string
+	Timestamp int64
+	Type      string
+	Method    string
+	URL       string
+	Headers   map[string][]string
+	Body      string
+}
+
+func NewHttpLog(logdir string) (logger T, err error) {
+	logger.logdir = logdir
+	err = os.MkdirAll(logdir, 0755)
+	return
+}
+
+func (logger T) logMessage(sequence string, messageType, method, url string, headers http.Header, body []byte) (err error) {
+	var message HttpMessage
+
+	message.Sequence = sequence
+	message.Type = messageType
+	message.Timestamp = time.Now().UnixMilli()
+	message.Method = method
+	message.URL = url
+	message.Body = string(body)
+	message.Headers = make(map[string][]string)
+	for key, val := range headers {
+		message.Headers[key] = val
+	}
+
+	reqJSON, _ := json.Marshal(message)
+	year, month, day := time.Now().Date()
+	fname := fmt.Sprintf("%s/%s_%04d-%02d-%02d.json", logger.logdir, messageType, year, int(month), day)
+	var logfile *os.File
+	logfile, err = os.OpenFile(fname, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0600)
+	if err != nil {
+		return
+	}
+	defer logfile.Close()
+
+	_, err = logfile.Write(reqJSON)
+	if err != nil {
+		return
+	}
+	_, err = logfile.Write([]byte("\n"))
+	return
+}
+
+func (logger T) Request(sequence string, req *http.Request) (err error) {
+	bodyReader, _ := req.GetBody()
+	body, _ := io.ReadAll(bodyReader)
+	err = logger.logMessage(sequence, "request", req.Method, req.URL.String(), req.Header, body)
+	return
+}
+
+func (logger T) Response(sequence string, res *http.Response) (err error) {
+	body, _ := io.ReadAll(res.Body)
+	res.Body.Close()
+	res.Body = io.NopCloser(bytes.NewBuffer(body))
+	err = logger.logMessage(sequence, "response", res.Request.Method, res.Request.URL.String(), res.Header, body)
+	return
+}