Refactor auth configuration and add enhanced state saving

Reorganized authentication settings into a nested structure for better clarity and maintainability. Introduced periodic authentication state saving with timestamp checks and directory creation support. Updated logic to include validation for authentication presence via configurable selectors.

Author: luispater

internal/browser/manager.go

@@ -52,14 +52,14 @@ func (m *Manager) LaunchBrowserAndContext() error {
 	log.Debug("Browser launched successfully.")
 
 	contextOptions := playwright.BrowserNewContextOptions{}
-	if m.Cfg.Instance[m.cfgIndex].AuthFile != "" {
-		if _, err = os.Stat(m.Cfg.Instance[m.cfgIndex].AuthFile); err == nil {
-			log.Infof("Loading authentication state from: %s", m.Cfg.Instance[m.cfgIndex].AuthFile)
-			contextOptions.StorageStatePath = playwright.String(m.Cfg.Instance[m.cfgIndex].AuthFile)
+	if m.Cfg.Instance[m.cfgIndex].Auth.File != "" {
+		if _, err = os.Stat(m.Cfg.Instance[m.cfgIndex].Auth.File); err == nil {
+			log.Infof("Loading authentication state from: %s", m.Cfg.Instance[m.cfgIndex].Auth.File)
+			contextOptions.StorageStatePath = playwright.String(m.Cfg.Instance[m.cfgIndex].Auth.File)
 		} else if os.IsNotExist(err) {
-			log.Infof("Authentication state file not found at %s. Proceeding without loading state. It will be created if you save state.", m.Cfg.Instance[m.cfgIndex].AuthFile)
+			log.Infof("Authentication state file not found at %s. Proceeding without loading state. It will be created if you save state.", m.Cfg.Instance[m.cfgIndex].Auth.File)
 		} else {
-			log.Infof("Error checking auth state file %s: %v. Proceeding without loading state.", m.Cfg.Instance[m.cfgIndex].AuthFile, err)
+			log.Infof("Error checking auth state file %s: %v. Proceeding without loading state.", m.Cfg.Instance[m.cfgIndex].Auth.File, err)
 		}
 	}
 	contextOptions.Proxy = &playwright.Proxy{

internal/config/config.go

@@ -20,14 +20,19 @@ type AppConfigRunner struct {
 	ContextCanceled string `yaml:"context_canceled"`
 }
 type AppConfigInstance struct {
-	Name        string          `yaml:"name"`
-	Adapter     string          `yaml:"adapter"`
-	ProxyURL    string          `yaml:"proxy-url"`
-	URL         string          `yaml:"url"`
-	SniffPort   string          `yaml:"sniff-port"`
-	SniffDomain string          `yaml:"sniff-domain"`
-	AuthFile    string          `yaml:"auth-file"`
-	Runner      AppConfigRunner `yaml:"runner"`
+	Name        string                `yaml:"name"`
+	Adapter     string                `yaml:"adapter"`
+	ProxyURL    string                `yaml:"proxy-url"`
+	URL         string                `yaml:"url"`
+	SniffPort   string                `yaml:"sniff-port"`
+	SniffDomain string                `yaml:"sniff-domain"`
+	Auth        AppConfigInstanceAuth `yaml:"auth"`
+	Runner      AppConfigRunner       `yaml:"runner"`
+}
+
+type AppConfigInstanceAuth struct {
+	File  string `yaml:"file"`
+	Check string `yaml:"check"`
 }
 
 // LoadConfig loads configuration from environment variables or defaults.

main.go

@@ -9,6 +9,7 @@ import (
 	"os"
 	"os/signal"
 	"path"
+	"path/filepath"
 	"strings"
 	"syscall"
 	"time"
@@ -213,28 +214,57 @@ func main() {
 			os.Exit(0)
 		case <-time.After(5 * time.Second):
 			for instanceName, p := range pages {
-				if p.URL() == mapCfg[instanceName].URL {
-					if _, err = os.Stat(mapCfg[instanceName].AuthFile); os.IsNotExist(err) {
-						storageState, errStorageState := p.Context().StorageState()
-						if errStorageState != nil {
-							log.Debugf("Error getting storage state: %v", err)
-							continue
-						}
-						jsonData, errMarshalIndent := json.MarshalIndent(storageState, "", "  ")
-						if errMarshalIndent != nil {
-							log.Debugf("Error marshalling storage state to JSON: %v", err)
-							continue
-						}
+				if mapCfg[instanceName].Auth.Check != "" {
+					checkLocator := pages[instanceName].Locator(mapCfg[instanceName].Auth.Check)
+					count, errCount := checkLocator.Count()
+					if errCount != nil || count == 0 {
+						continue
+					}
+				}
+
+				saveState := false
+				if fileInfo, errStat := os.Stat(mapCfg[instanceName].Auth.File); os.IsNotExist(errStat) {
+					saveState = true
+				} else {
+					lastModified := fileInfo.ModTime()
+					now := time.Now()
+					duration := now.Sub(lastModified)
+					if duration > 5*time.Minute {
+						saveState = true
+					}
+				}
+
+				if saveState {
+					storageState, errStorageState := p.Context().StorageState()
+					if errStorageState != nil {
+						log.Debugf("Error getting storage state: %v", err)
+						continue
+					}
+					jsonData, errMarshalIndent := json.MarshalIndent(storageState, "", "  ")
+					if errMarshalIndent != nil {
+						log.Debugf("Error marshalling storage state to JSON: %v", err)
+						continue
+					}
 
-						err = os.WriteFile(mapCfg[instanceName].AuthFile, jsonData, 0644)
+					authAbsPath, errAbs := filepath.Abs(mapCfg[instanceName].Auth.File)
+					if errAbs != nil {
+						continue
+					}
+					authDirName := filepath.Dir(authAbsPath)
+					_, errStat := os.Stat(filepath.Dir(authAbsPath))
+					if os.IsNotExist(errStat) {
+						err = os.MkdirAll(authDirName, 0755)
 						if err != nil {
-							log.Debugf("Error writing storage state to file %s: %v", mapCfg[instanceName].AuthFile, err)
-							continue
-						} else {
-							log.Debugf("Successfully writing storage state to file %s", mapCfg[instanceName].AuthFile)
 							continue
 						}
 					}
+
+					err = os.WriteFile(mapCfg[instanceName].Auth.File, jsonData, 0644)
+					if err != nil {
+						log.Debugf("Error writing storage state to file %s: %v", mapCfg[instanceName].Auth.File, err)
+					} else {
+						log.Debugf("Successfully writing storage state to file %s", mapCfg[instanceName].Auth.File)
+					}
 				}
 			}
 		}