Refactoring and doc

src/DoctrineAuditBundle/AuditReader.php

@@ -22,11 +22,19 @@ public function __construct(AuditConfiguration $configuration, EntityManagerInte
         $this->entityManager = $entityManager;
     }
 
+    /**
+     * @return AuditConfiguration
+     */
     public function getConfiguration(): AuditConfiguration
     {
         return $this->configuration;
     }
 
+    /**
+     * Returns an array of audit table names indexed by entity FQN
+     *
+     * @return array
+     */
     public function getEntities(): array
     {
         $entities = $this->entityManager->getConfiguration()->getMetadataDriverImpl()->getAllClassNames();
@@ -41,12 +49,21 @@ public function getEntities(): array
         return $audited;
     }
 
-    public function getAudits(string $entity, int $id = null, int $page = 1, int $pageSize = 50): array
+    /**
+     * Returns an array of audited entries/operations
+     *
+     * @param string|object $entity
+     * @param int|null $id
+     * @param int $page
+     * @param int $pageSize
+     * @return array
+     */
+    public function getAudits($entity, int $id = null, int $page = 1, int $pageSize = 50): array
     {
         $connection = $this->entityManager->getConnection();
         $auditTable = implode('', [
             $this->configuration->getTablePrefix(),
-            $this->getEntityTableName($entity),
+            $this->getEntityTableName(is_string($entity) ? $entity : get_class($entity)),
             $this->configuration->getTableSuffix(),
         ]);
 
@@ -73,12 +90,17 @@ public function getAudits(string $entity, int $id = null, int $page = 1, int $pa
         ;
     }
 
-    public function getAudit(string $entity, int $id)
+    /**
+     * @param string|object $entity
+     * @param int $id
+     * @return mixed
+     */
+    public function getAudit($entity, int $id)
     {
         $connection = $this->entityManager->getConnection();
         $auditTable = implode('', [
             $this->configuration->getTablePrefix(),
-            $this->getEntityTableName($entity),
+            $this->getEntityTableName(is_string($entity) ? $entity : get_class($entity)),
             $this->configuration->getTableSuffix(),
         ]);
 
@@ -101,9 +123,7 @@ public function getAudit(string $entity, int $id)
     /**
      * Returns the table name of $entity.
      *
-     * @param EntityManagerInterface $em
-     * @param $entity
-     *
+     * @param string|object $entity
      * @return string
      */
     public function getEntityTableName($entity): string

src/DoctrineAuditBundle/Controller/AuditController.php

@@ -32,11 +32,11 @@ public function listAuditsAction()
     public function showEntityHistoryAction(string $entity, int $id = null, int $page = 1, int $pageSize = 50)
     {
         $reader = $this->container->get('dh_doctrine_audit.reader');
-        $results = $reader->getAudits($entity, $id, $page, $pageSize);
+        $entries = $reader->getAudits($entity, $id, $page, $pageSize);
 
         return $this->render('DHDoctrineAuditBundle:Audit:entity_history.html.twig', [
             'entity'     => $entity,
-            'entries'   => $results,
+            'entries'   => $entries,
         ]);
     }
 
@@ -48,11 +48,11 @@ public function showEntityHistoryAction(string $entity, int $id = null, int $pag
     public function showAuditEntryAction(string $entity, int $id)
     {
         $reader = $this->container->get('dh_doctrine_audit.reader');
-        $results = $reader->getAudit($entity, $id);
+        $data = $reader->getAudit($entity, $id);
 
         return $this->render('DHDoctrineAuditBundle:Audit:entity_audit_details.html.twig', [
             'entity' => $entity,
-            'entry' => $results[0],
+            'entry' => $data[0],
         ]);
     }
 }

src/DoctrineAuditBundle/Resources/config/services.yaml

@@ -20,3 +20,9 @@ services:
     arguments: ["@dh_doctrine_audit.configuration"]
     tags:
       - { name: doctrine.event_subscriber, connection: default }
+
+  dh_doctrine_audit.twig_extension:
+    class: DH\DoctrineAuditBundle\Twig\Extension\TwigExtension
+    arguments: ['@doctrine']
+    tags:
+      -  { name: twig.extension }

src/DoctrineAuditBundle/Twig/Extension/TwigExtension.php

@@ -0,0 +1,57 @@
+<?php
+
+namespace DH\DoctrineAuditBundle\Twig\Extension;
+
+use Symfony\Bridge\Doctrine\RegistryInterface;
+
+class TwigExtension extends \Twig_Extension
+{
+    protected $doctrine;
+
+    public function getFunctions()
+    {
+        // Register the function in twig :
+        // In your template you can use it as : {{findUser(123)}}
+        return [
+            new \Twig_SimpleFunction('findUser', [$this, 'findUser']),
+            new \Twig_SimpleFunction('class', [$this, 'getClass']),
+            new \Twig_SimpleFunction('tablename', [$this, 'getTablename']),
+        ];
+    }
+
+    public function __construct(RegistryInterface $doctrine)
+    {
+        $this->doctrine = $doctrine;
+    }
+
+    public function findUser($id, $repository)
+    {
+        $em = $this->doctrine->getManager();
+        $repo = $em->getRepository($repository);
+
+        return $repo->find($id);
+    }
+
+    public function getClass($entity)
+    {
+        return get_class($entity);
+    }
+
+    public function getTablename($entity)
+    {
+        return $this
+            ->doctrine
+            ->getManager()
+            ->getClassMetadata(get_class($entity))
+            ->table['name']
+        ;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getName()
+    {
+        return 'twig_extensions';
+    }
+}