Add support for Fingerprint Chromium and a screenshot endpoint

Updated the browser manager to support Fingerprint Chromium configuration. Introduced a new API endpoint for taking screenshots with thread-safe handling. Minor improvements and adjustments in element handling within the mouse interaction method.

Author: luispater

internal/api/handlers.go

@@ -17,6 +17,7 @@ import (
 )
 
 var RequestMutex sync.Mutex
+var ScreenshotMutex sync.Mutex
 
 // APIHandlers contains the handlers for API endpoints
 type APIHandlers struct {
@@ -36,6 +37,29 @@ func NewAPIHandlers(appConfig *config.AppConfig, queue *RequestQueue, pages map[
 	}
 }
 
+func (h *APIHandlers) TakeScreenshot(c *gin.Context) {
+	defer ScreenshotMutex.Unlock()
+	ScreenshotMutex.Lock()
+	instanceName, ok := c.GetQuery("name")
+	if !ok {
+		c.Status(http.StatusNotFound)
+		return
+	}
+	if page, hasKey := h.pages[instanceName]; hasKey {
+		c.Header("Content-Type", "image/png")
+		c.Header("Cache-Control", "no-cache")
+		c.Header("Connection", "keep-alive")
+		c.Header("Access-Control-Allow-Origin", "*")
+		screenshot, err := page.Screenshot()
+		if err != nil {
+			c.JSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("Invalid request: %v", err), "code": 500})
+		}
+		_, _ = c.Writer.Write(screenshot)
+	} else {
+		c.Status(http.StatusNotFound)
+	}
+}
+
 // ChatCompletions handles the /v1/chat/completions endpoint
 func (h *APIHandlers) ChatCompletions(c *gin.Context) {
 	defer RequestMutex.Unlock()

internal/api/server.go

@@ -91,6 +91,8 @@ func (s *Server) setupRoutes() {
 			},
 		})
 	})
+
+	s.engine.GET("/screenshot", s.handlers.TakeScreenshot)
 }
 
 // Start starts the API server

internal/browser/manager.go

@@ -34,17 +34,24 @@ func NewManager(cfg *config.AppConfig, cfgIndex int) (*Manager, error) {
 
 // LaunchBrowserAndContext launches the configured browser and creates a new context with storage state.
 func (m *Manager) LaunchBrowserAndContext() error {
+	var browser playwright.Browser
+	var err error
 	launchOptions := playwright.BrowserTypeLaunchOptions{
 		Headless: playwright.Bool(m.Cfg.Headless),
 	}
-	if m.Cfg.CamoufoxPath != "" {
+	if m.Cfg.FingerprintChromiumPath != "" {
+		launchOptions.ExecutablePath = playwright.String(m.Cfg.FingerprintChromiumPath)
+		launchOptions.Args = []string{"--fingerprint=1000", "--timezone='America/Los_Angeles'", "--ignore-certificate-errors"}
+		log.Debugf("Attempting to launch Fingerprint Chromium from: %s", m.Cfg.CamoufoxPath)
+		browser, err = m.pw.Chromium.Launch(launchOptions)
+	} else if m.Cfg.CamoufoxPath != "" {
 		launchOptions.ExecutablePath = playwright.String(m.Cfg.CamoufoxPath)
 		log.Debugf("Attempting to launch Camoufox from: %s", m.Cfg.CamoufoxPath)
+		browser, err = m.pw.Firefox.Launch(launchOptions)
 	} else {
 		log.Debug("Camoufox path not specified, launching default Playwright Firefox.")
 	}
 
-	browser, err := m.pw.Firefox.Launch(launchOptions)
 	if err != nil {
 		return err
 	}

internal/config/config.go

@@ -7,12 +7,13 @@ import (
 
 // AppConfig holds the application configuration.
 type AppConfig struct {
-	Version      string              `yaml:"version"`
-	Debug        bool                `yaml:"debug"`
-	CamoufoxPath string              `yaml:"camoufox-path"`
-	Headless     bool                `yaml:"headless"`
-	ApiPort      string              `yaml:"api-port"`
-	Instance     []AppConfigInstance `yaml:"instance"`
+	Version                 string              `yaml:"version"`
+	Debug                   bool                `yaml:"debug"`
+	CamoufoxPath            string              `yaml:"camoufox-path"`
+	FingerprintChromiumPath string              `yaml:"fingerprint-chromium-path"`
+	Headless                bool                `yaml:"headless"`
+	ApiPort                 string              `yaml:"api-port"`
+	Instance                []AppConfigInstance `yaml:"instance"`
 }
 type AppConfigRunner struct {
 	Init            string `yaml:"init"`

internal/method/mouse.go

@@ -8,12 +8,13 @@ import (
 
 func (m *Method) Click(elementSelector string, timeout float64) error {
 	log.Debugf("Attempting to find and click element with selector: %s", elementSelector)
-	element := m.page.Locator(elementSelector).First() // Use First() to get the first match if multiple
+	element := m.page.Locator(elementSelector) // Use First() to get the first match if multiple
 	count, err := element.Count()
 	if err != nil {
 		return fmt.Errorf("error counting elements with selector '%s': %v", elementSelector, err)
 	}
 	if count > 0 {
+		element = element.First()
 		isVisible, errIsVisible := element.IsVisible()
 		if errIsVisible != nil {
 			return fmt.Errorf("error checking visibility of element '%s': %v", elementSelector, errIsVisible)