Attaching custom screenshot to gauge report

CustomScreenshotGrabber.java

public class CustomScreenshotGrabber implements CustomScreenshotWriter {
    @Override
    public synchronized String takeScreenshot() {
        if (WebDriverManager.getInstance().getWebDriver() != null) {
            TakesScreenshot driver = (TakesScreenshot) WebDriverManager.getInstance().getWebDriver();
            String screenshotFileName = String.format("screenshot-%s.png", UUID.randomUUID());
            try {
                Files.write(Paths.get((".gauge/screenshots"), screenshotFileName),
                        driver.getScreenshotAs(OutputType.BYTES));
            } catch (IOException e) {
                e.printStackTrace();
            }
            return screenshotFileName;
        }
        else return super.toString();
    }
}


/*
The above code captures the screenshot in the browser page.

Gauge will automatically find this grabber and use it. For use just call Gauge.captureScreenshot();

Also, Gauge captures a screenshot on such a failure.

If multiple custom ScreenGrabber implementations are found in classpath then gauge will pick one randomly to capture 
the screen. This is because gauge selects the first ScreenGrabber it finds, which in turn depends on the order of 
scanning of the libraries.

*/