Apigility Add Role to access_tokens table

Module.php

<?php
/**
 * @license   http://opensource.org/licenses/BSD-3-Clause BSD-3-Clause
 * @copyright Copyright (c) 2014 Zend Technologies USA Inc. (http://www.zend.com)
 */

namespace Application;

use Zend\Mvc\ModuleRouteListener;
use Zend\Mvc\MvcEvent;

class Module
{
    public function onBootstrap(MvcEvent $e)
    {
        $eventManager        = $e->getApplication()->getEventManager();
        $moduleRouteListener = new ModuleRouteListener();
        $moduleRouteListener->attach($eventManager);
        
        $eventManager -> attach(
            MvcEvent::EVENT_FINISH,
            array($this, 'onFinish')
        );
    }
    
    public function onFinish(MvcEvent $e)
    {
        $routeMatch = $e -> getRouteMatch();
        $routeParams = $routeMatch -> getParams();
        if ($routeParams['controller'] == 'ZF\OAuth2\Controller\Auth' 
                && $routeParams['action'] == 'token') {
            if ($e->getResult()->getStatusCode() == 200) {
                $data = json_decode($e->getResult()->getContent(), true);
                if (isset($data['access_token']) && $data['access_token'] != '' && !$data['role']) {
                    //find user role from token
                    $token = $entityManager->tokenRepository->findOneByAccessToken($data['access_token']);
                    $user = $entityManager->userRepository->findOneByUsername($token->getUserId());
                    $token->setRole($user->getRole());
                    $entityManager->persist($token);
                    $entityManager->flush();
                    $data['role'] = $user->getRole();
                    $e->getResult()->setContent(json_encode($data));
                }
            }
        }
    }
}