service.go

package webshot

import (
	"fmt"
	"os"

	"github.com/tebeka/selenium"
)

// NewWebshot sets up a new browser with the config provided
func NewWebshot(c NewConfig) (*Webshot, error) {
	// var webdriver selenium.WebDriver

	opts := []selenium.ServiceOption{
		selenium.StartFrameBuffer(),
		selenium.GeckoDriver(c.DriverPath),
		selenium.Output(os.Stderr),
	}

	if c.DebugMode {
		selenium.SetDebug(true)
	}

	service, err := selenium.NewGeckoDriverService(c.DriverPath, c.Port, opts[0])
	if err != nil {
		return nil, err
	}

	caps := selenium.Capabilities{"browserName": c.BrowserName}
	wd, err := selenium.NewRemote(caps, fmt.Sprintf("%s:%d", c.Address, c.Port))
	if err != nil {
		return nil, err
	}

	fmt.Printf("The service ran successfully\n")
	return &Webshot{
		Webdriver: wd,
		Service:   service,
	}, nil
}

// Extend allow you to use use all of the functionalities provided by selenium
func (w *Webshot) Extend() selenium.WebDriver {
	return w.Webdriver
}