1 files changed, 89 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/httpcheck/cookiejar.go b/src/go/collectors/go.d.plugin/modules/httpcheck/cookiejar.go
new file mode 100644
index 000000000..628867caa
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/httpcheck/cookiejar.go
@@ -0,0 +1,89 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package httpcheck
+
+import (
+	"bufio"
+	"fmt"
+	"net/http"
+	"net/http/cookiejar"
+	"net/url"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+
+	"golang.org/x/net/publicsuffix"
+)
+
+// TODO: implement proper cookie auth support
+// relevant forum topic: https://community.netdata.cloud/t/howto-http-endpoint-collector-with-cookie-and-user-pass/3981/5?u=ilyam8
+
+// cookie file format: https://everything.curl.dev/http/cookies/fileformat
+func loadCookieJar(path string) (http.CookieJar, error) {
+	file, err := os.Open(path)
+	if err != nil {
+		return nil, err
+	}
+	defer func() { _ = file.Close() }()
+
+	jar, err := cookiejar.New(&cookiejar.Options{PublicSuffixList: publicsuffix.List})
+	if err != nil {
+		return nil, err
+	}
+
+	sc := bufio.NewScanner(file)
+
+	for sc.Scan() {
+		line, httpOnly := strings.CutPrefix(strings.TrimSpace(sc.Text()), "#HttpOnly_")
+
+		if strings.HasPrefix(line, "#") || line == "" {
+			continue
+		}
+
+		parts := strings.Fields(line)
+		if len(parts) != 6 && len(parts) != 7 {
+			return nil, fmt.Errorf("got %d fields in line '%s', want 6 or 7", len(parts), line)
+		}
+
+		for i, v := range parts {
+			parts[i] = strings.TrimSpace(v)
+		}
+
+		cookie := &http.Cookie{
+			Domain:   parts[0],
+			Path:     parts[2],
+			Name:     parts[5],
+			HttpOnly: httpOnly,
+		}
+		cookie.Secure, err = strconv.ParseBool(parts[3])
+		if err != nil {
+			return nil, err
+		}
+		expires, err := strconv.ParseInt(parts[4], 10, 64)
+		if err != nil {
+			return nil, err
+		}
+		if expires > 0 {
+			cookie.Expires = time.Unix(expires, 0)
+		}
+		if len(parts) == 7 {
+			cookie.Value = parts[6]
+		}
+
+		scheme := "http"
+		if cookie.Secure {
+			scheme = "https"
+		}
+		cookieURL := &url.URL{
+			Scheme: scheme,
+			Host:   cookie.Domain,
+		}
+
+		cookies := jar.Cookies(cookieURL)
+		cookies = append(cookies, cookie)
+		jar.SetCookies(cookieURL, cookies)
+	}
+
+	return jar, nil
+}