150 lines
4.0 KiB
PHP
150 lines
4.0 KiB
PHP
<?php
|
|
|
|
namespace Gregwar\CaptchaBundle\Generator;
|
|
|
|
use Symfony\Component\Finder\Finder;
|
|
use Symfony\Component\HttpFoundation\Session\SessionInterface;
|
|
use Symfony\Component\Routing\RouterInterface;
|
|
|
|
use Gregwar\Captcha\CaptchaBuilderInterface;
|
|
use Gregwar\Captcha\PhraseBuilderInterface;
|
|
|
|
/**
|
|
* Uses configuration parameters to call the services that generate captcha images
|
|
*
|
|
* @author Gregwar <g.passault@gmail.com>
|
|
* @author Jeremy Livingston <jeremy.j.livingston@gmail.com>
|
|
*/
|
|
class CaptchaGenerator
|
|
{
|
|
/**
|
|
* @var \Symfony\Component\Routing\RouterInterface
|
|
*/
|
|
protected $router;
|
|
|
|
/**
|
|
* @var CaptchaBuilder
|
|
*/
|
|
protected $builder;
|
|
|
|
/**
|
|
* @var PhraseBuilder
|
|
*/
|
|
protected $phraseBuilder;
|
|
|
|
/**
|
|
* @var ImageFileHandler
|
|
*/
|
|
protected $imageFileHandler;
|
|
|
|
/**
|
|
* @param \Symfony\Component\Routing\RouterInterface $router
|
|
* @param CaptchaBuilderInterface $builder
|
|
* @param ImageFileHandlerInterface $imageFileHandler
|
|
*/
|
|
public function __construct(RouterInterface $router, CaptchaBuilderInterface $builder, PhraseBuilderInterface $phraseBuilder, ImageFileHandler $imageFileHandler)
|
|
{
|
|
$this->router = $router;
|
|
$this->builder = $builder;
|
|
$this->phraseBuilder = $phraseBuilder;
|
|
$this->imageFileHandler = $imageFileHandler;
|
|
}
|
|
|
|
/**
|
|
* Get the captcha URL, stream, or filename that will go in the image's src attribute
|
|
*
|
|
* @param $key
|
|
* @param array $options
|
|
*
|
|
* @return array
|
|
*/
|
|
public function getCaptchaCode(array &$options)
|
|
{
|
|
$this->builder->setPhrase($this->getPhrase($options));
|
|
|
|
// Randomly execute garbage collection and returns the image filename
|
|
if ($options['as_file']) {
|
|
$this->imageFileHandler->collectGarbage();
|
|
|
|
return $this->generate($options);
|
|
}
|
|
|
|
// Returns the image generation URL
|
|
if ($options['as_url']) {
|
|
return $this->router->generate('gregwar_captcha.generate_captcha', array('key' => $options['session_key']));
|
|
}
|
|
|
|
return 'data:image/jpeg;base64,' . base64_encode($this->generate($options));
|
|
}
|
|
|
|
/**
|
|
* Sets the phrase to the builder
|
|
*/
|
|
public function setPhrase($phrase)
|
|
{
|
|
$this->builder->setPhrase($phrase);
|
|
}
|
|
|
|
/**
|
|
* @param string $key
|
|
* @param array $options
|
|
*
|
|
* @return string
|
|
*/
|
|
public function generate(array &$options)
|
|
{
|
|
$this->builder->setDistortion($options['distortion']);
|
|
|
|
if ($options['backgroundColor'] != array()) {
|
|
if (count($options['backgroundColor'])!=3) {
|
|
throw new \RuntimeException('backgroundColor should be an array of r, g and b');
|
|
}
|
|
|
|
$color = $options['backgroundColor'];
|
|
$this->builder->setBackgroundColor($color[0], $color[1], $color[2]);
|
|
}
|
|
|
|
$fingerprint = isset($options['fingerprint']) ? $options['fingerprint'] : null;
|
|
|
|
$content = $this->builder->build(
|
|
$options['width'],
|
|
$options['height'],
|
|
$options['font'],
|
|
$fingerprint
|
|
)->getGd();
|
|
|
|
if ($options['keep_value']) {
|
|
$options['fingerprint'] = $this->builder->getFingerprint();
|
|
}
|
|
|
|
if (!$options['as_file']) {
|
|
ob_start();
|
|
imagejpeg($content, null, $options['quality']);
|
|
|
|
return ob_get_clean();
|
|
}
|
|
|
|
return $this->imageFileHandler->saveAsFile($content);
|
|
}
|
|
|
|
/**
|
|
* @param string $key
|
|
* @param array $options
|
|
*
|
|
* @return string
|
|
*/
|
|
public function getPhrase(array &$options)
|
|
{
|
|
// Get the phrase that we'll use for this image
|
|
if ($options['keep_value'] && isset($options['phrase'])) {
|
|
$phrase = $options['phrase'];
|
|
} else {
|
|
$phrase = $this->phraseBuilder->build($options['length'], $options['charset']);
|
|
$options['phrase'] = $phrase;
|
|
}
|
|
|
|
return $phrase;
|
|
}
|
|
}
|
|
|