Bex\Behat\ScreenshotExtension\Service\ScreenshotTaker

Complexity

Total Complexity

6

Size/Duplication

Total Lines	63
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Importance

Changes	5
Bugs	1	Features	1

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	takeScreenshot()	0	8	2
A	getCombinedImage()	0	17	2
A	reset()	0	4	1

<?php

namespace Bex\Behat\ScreenshotExtension\Service;

use Behat\Mink\Mink;
use Bex\Behat\ScreenshotExtension\Driver\ImageDriverInterface;
use Symfony\Component\Console\Output\OutputInterface;

/**
 * This class is responsible for taking screenshot by using the Mink session
 *
 * @license http://opensource.org/licenses/MIT The MIT License
 */
class ScreenshotTaker
{
    /** @var Mink $mink */
    private $mink;

    /** @var OutputInterface $output */
    private $output;

    /**
     * @var array $screenshots
     */
    private $screenshots;

    /**
     * Constructor
     *
     * @param Mink $mink
     * @param OutputInterface $output
     */
    public function __construct(Mink $mink, OutputInterface $output)
    {
        $this->mink = $mink;
        $this->output = $output;
    }

    /**
     * Save the screenshot as the given filename
     */
    public function takeScreenshot()
    {
        try {
            $this->screenshots[] = $this->mink->getSession()->getScreenshot();
        } catch (\Exception $e) {
            $this->output->writeln($e->getMessage());
        }        
    }

    public function getCombinedImage()
    {
        $im = new \Imagick();

        foreach ($this->screenshots as $screenshot) {
            $im->readImageBlob($screenshot);
        }

        /* Append the images into one */
        $im->resetIterator();
        $combined = $im->appendImages(true);

        /* Output the image */
        $combined->setImageFormat("png");

        return (string)$combined;
    }

    /**
     * Remove previous images
     */
    public function reset()
    {
        $this->screenshots = [];
    }
}