dbresult.php

<?php
namespace Bitrix\Vote;
 
use \Bitrix\Main\Localization\Loc;
Loc::loadMessages(__FILE__);
 
class DBResult extends \CDBResult
{
    function fetch()
    {
        if ($res = parent::fetch())
        {
            $prefix = null;
            foreach ($res as $k => $v)
            {
                if (mb_strpos($k, "LAMP") !== false)
                {
                    $prefix = mb_substr($k, 0, mb_strpos($k, "LAMP"));
                    break;
                }
            }
            if ($prefix !== null && $res[$prefix."LAMP"] == "yellow" && !empty($res[$prefix."CHANNEL_ID"]))
            {
                $res[$prefix."LAMP"] = ($res[$prefix."ID"] == \CVote::getActiveVoteId($res[$prefix."CHANNEL_ID"]) ? "green" : "red");
            }
        }
        return $res;
    }
}