Add Dia browser support on macOS

Author: YNDmitry

browser/all/import.go

@@ -7,6 +7,7 @@ import (
 	_ "github.com/browserutils/kooky/browser/browsh"
 	_ "github.com/browserutils/kooky/browser/chrome"
 	_ "github.com/browserutils/kooky/browser/chromium"
+	_ "github.com/browserutils/kooky/browser/dia"
 	_ "github.com/browserutils/kooky/browser/dillo"
 	_ "github.com/browserutils/kooky/browser/edge"
 	_ "github.com/browserutils/kooky/browser/elinks"

browser/dia/dia.go

@@ -0,0 +1,31 @@
+package dia
+
+import (
+	"context"
+
+	"github.com/browserutils/kooky"
+	"github.com/browserutils/kooky/internal/chrome"
+	"github.com/browserutils/kooky/internal/cookies"
+)
+
+func ReadCookies(ctx context.Context, filename string, filters ...kooky.Filter) ([]*kooky.Cookie, error) {
+	return cookies.SingleRead(cookieStore, filename, filters...).ReadAllCookies(ctx)
+}
+
+func TraverseCookies(filename string, filters ...kooky.Filter) kooky.CookieSeq {
+	return cookies.SingleRead(cookieStore, filename, filters...)
+}
+
+// CookieStore has to be closed with CookieStore.Close() after use.
+func CookieStore(filename string, filters ...kooky.Filter) (kooky.CookieStore, error) {
+	return cookieStore(filename, filters...)
+}
+
+func cookieStore(filename string, filters ...kooky.Filter) (*cookies.CookieJar, error) {
+	s := &chrome.CookieStore{}
+	s.FileNameStr = filename
+	s.BrowserStr = `dia`
+	s.SetSafeStorage(`Dia`, ``, ``)
+
+	return cookies.NewCookieJar(s, filters...), nil
+}

browser/dia/find.go

@@ -0,0 +1,47 @@
+//go:build darwin && !ios
+
+package dia
+
+import (
+	"github.com/browserutils/kooky"
+	"github.com/browserutils/kooky/internal/chrome"
+	chromefind "github.com/browserutils/kooky/internal/chrome/find"
+	"github.com/browserutils/kooky/internal/cookies"
+)
+
+type diaFinder struct{}
+
+var _ kooky.CookieStoreFinder = (*diaFinder)(nil)
+
+func init() {
+	kooky.RegisterFinder(`dia`, &diaFinder{})
+}
+
+func (f *diaFinder) FindCookieStores() kooky.CookieStoreSeq {
+	return func(yield func(kooky.CookieStore, error) bool) {
+		for file, err := range chromefind.FindCookieStoreFiles(diaChromiumRoots, `dia`) {
+			if err != nil {
+				if !yield(nil, err) {
+					return
+				}
+				continue
+			}
+			if file == nil {
+				continue
+			}
+			cookieStore := &chrome.CookieStore{
+				DefaultCookieStore: cookies.DefaultCookieStore{
+					BrowserStr:           file.Browser,
+					ProfileStr:           file.Profile,
+					OSStr:                file.OS,
+					IsDefaultProfileBool: file.IsDefaultProfile,
+					FileNameStr:          file.Path,
+				},
+			}
+			cookieStore.SetSafeStorage(`Dia`, ``, ``)
+			if !yield(&cookies.CookieJar{CookieStore: cookieStore}, nil) {
+				return
+			}
+		}
+	}
+}

browser/dia/find_darwin.go

@@ -0,0 +1,17 @@
+//go:build darwin && !ios
+
+package dia
+
+import (
+	"os"
+	"path/filepath"
+)
+
+func diaChromiumRoots(yield func(string, error) bool) {
+	cfgDir, err := os.UserConfigDir()
+	if err != nil {
+		_ = yield(``, err)
+		return
+	}
+	_ = yield(filepath.Join(cfgDir, `Dia`, `User Data`), nil)
+}

internal/chrome/README.md

@@ -6,6 +6,7 @@
 |----------------|-----------------------------|-------------------|-----------------------|
 | Chrome         | Chrome Safe Storage         | chrome            | com.google.Chrome     |
 | Chromium       | Chromium Safe Storage       | chromium          | org.chromium.Chromium |
+| Dia            | Dia Safe Storage            | dia               |                       |
 | Brave          | Brave Safe Storage          | brave             |                       |
 | Microsoft Edge | Microsoft Edge Safe Storage | chromium (shared) |                       |
 | Opera          | uses Chromium Safe Storage  | chromium (shared) |                       |