first

modules/browser/browser.go

@@ -0,0 +1,68 @@
+package browser
+
+import (
+	"errors"
+	"fmt"
+
+	"github.com/tebeka/selenium"
+)
+
+// Browser -
+type Browser struct {
+	SVC  *selenium.Service
+	Opts Options
+}
+
+// Options -
+type Options struct {
+	SeleniumPath string
+	DriverPath   string
+	Port         int
+}
+
+var svc *Browser
+
+// NewService -
+func NewService(opts Options) (err error) {
+	opt := []selenium.ServiceOption{
+		selenium.StartFrameBuffer(),           // Start an X frame buffer for the browser to run in.
+		selenium.GeckoDriver(opts.DriverPath), // Specify the path to GeckoDriver in order to use Firefox.
+	}
+
+	svc = &Browser{}
+	svc.Opts = opts
+	svc.SVC, err = selenium.NewSeleniumService(opts.SeleniumPath, opts.Port, opt...)
+	if err != nil {
+		svc = nil
+		return err
+	}
+
+	return
+}
+
+// StopService -
+func StopService() error {
+	if svc == nil {
+		return nil
+	}
+	if err := svc.SVC.Stop(); err != nil {
+		return err
+	}
+	svc = nil
+	return nil
+}
+
+// NewWD -
+func NewWD() (selenium.WebDriver, error) {
+	if svc == nil {
+		return nil, errors.New("service not init")
+	}
+
+	caps := selenium.Capabilities{"browserName": "firefox"}
+	wd, err := selenium.NewRemote(caps, fmt.Sprintf("http://localhost:%d/wd/hub", svc.Opts.Port))
+	if err != nil {
+		return nil, err
+	}
+
+	return wd, nil
+}