From a31c9a91c96e4ee6837ca73ee70cee73fac93d1b Mon Sep 17 00:00:00 2001
From: Kotaro Inoue <k.musaino@gmail.com>
Date: Mon, 24 Jul 2023 00:40:59 +0900
Subject: [PATCH] Fix to enable console log only for chrome driver

Signed-off-by: Kotaro Inoue <k.musaino@gmail.com>
---
 librarejob/rarejob.go | 30 +++++++++++++++++++++++++-----
 1 file changed, 25 insertions(+), 5 deletions(-)

diff --git a/librarejob/rarejob.go b/librarejob/rarejob.go
index 8e12c9a..4bee04d 100644
--- a/librarejob/rarejob.go
+++ b/librarejob/rarejob.go
@@ -49,10 +49,18 @@ type Client interface {
 	Teardown() error
 }
 
+type browserType string
+
+const (
+	browserTypeFirefox browserType = "firefox"
+	browserTypeChrome  browserType = "chrome"
+)
+
 type client struct {
-	s     *selenium.Service
-	wd    selenium.WebDriver
-	debug bool
+	s       *selenium.Service
+	wd      selenium.WebDriver
+	browser browserType
+	debug   bool
 }
 
 type ClientOpts struct {
@@ -70,7 +78,7 @@ func NewClient(opts ClientOpts) (Client, error) {
 	var err error
 	url := "127.0.0.1"
 	port := 4444
-	browserName := "firefox"
+	browserName := browserTypeFirefox
 	if opts.SeleniumHost == "" {
 		if opts.SeleniumPort != nil {
 			port = *opts.SeleniumPort
@@ -88,7 +96,14 @@ func NewClient(opts ClientOpts) (Client, error) {
 	}
 
 	if opts.SeleniumBrowserName != "" {
-		browserName = opts.SeleniumBrowserName
+		switch opts.SeleniumBrowserName {
+		case string(browserTypeFirefox):
+			browserName = browserTypeFirefox
+		case string(browserTypeChrome):
+			browserName = browserTypeChrome
+		default:
+			return nil, fmt.Errorf("invalid browser name: %s", opts.SeleniumBrowserName)
+		}
 	}
 
 	urlPrefix := fmt.Sprintf("http://%s:%d/wd/hub", url, port)
@@ -329,6 +344,11 @@ func (c *client) Teardown() error {
 func (c *client) flushConsoleLogs() {
 	defer zap.L().Sync()
 
+	if c.browser != browserTypeChrome {
+		zap.L().Warn("console log is only available for chrome browser")
+		return
+	}
+
 	// output console log
 	clog, err := c.wd.Log(log.Browser)
 	if err != nil {