BadgeController extends ContainerAware

rogeriopradoj · May 29, 2013 · c2bd71e · c2bd71e
1 parent c1a8a21
commit c2bd71e
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/PUGX/BadgeBundle/Controller/BadgeController.php b/src/PUGX/BadgeBundle/Controller/BadgeController.php
@@ -11,7 +11,7 @@
 
 namespace PUGX\BadgeBundle\Controller;
 
-use Symfony\Bundle\FrameworkBundle\Controller\Controller;
+use Symfony\Component\DependencyInjection\ContainerAware;
 use Symfony\Component\HttpFoundation\Response;
 use Symfony\Component\HttpKernel\Exception\NotFoundHttpException;
 use Symfony\Component\HttpFoundation\StreamedResponse;
@@ -20,17 +20,17 @@
 use PUGX\BadgeBundle\Service\ImageCreator;
 use PUGX\BadgeBundle\Exception\InvalidArgumentException;
 
-class BadgeController extends Controller
+class BadgeController extends ContainerAware
 {
  public function downloadsAction($repository, $type = 'total')
  {
- $imageCreator = $this->get('image_creator');
+ $imageCreator = $this->container->get('image_creator');
  $outputFilename = sprintf('%s.png', $type);
  $httpCode = 500;
 
  // get the statistic from packagist
  try {
- $downloads = $this->get('badger')->getPackageDownloads($repository, $type);
+ $downloads = $this->container->get('badger')->getPackageDownloads($repository, $type);
  $downloadsText = $imageCreator->transformNumberToReadableFormat($downloads);
  $httpCode = 200;
  } catch (InvalidArgumentException $e) {
@@ -45,10 +45,10 @@ public function downloadsAction($repository, $type = 'total')
 
  public function versionAction($repository)
  {
- $imageCreator = $this->get('image_creator');
+ $imageCreator = $this->container->get('image_creator');
  $outputFilename = sprintf('%s.png', 'version');
 
- $version = $this->get('badger')->getLatestStableVersion($repository);
+ $version = $this->container->get('badger')->getLatestStableVersion($repository);
 
  if ($version) {
  $image = $imageCreator->createStableImage($version);
@@ -61,7 +61,7 @@ public function versionAction($repository)
 
  protected function streamImage($image, $outputFilename)
  {
- $imageCreator = $this->get('image_creator');
+ $imageCreator = $this->container->get('image_creator');
 
  //generating the streamed response
  $response = new StreamedResponse(null);