ifhub.club/application/classes/modules/poll/Poll.class.php

<?php
/*
 * LiveStreet CMS
 * Copyright © 2013 OOO "ЛС-СОФТ"
 *
 * ------------------------------------------------------
 *
 * Official site: www.livestreetcms.com
 * Contact e-mail: office@livestreetcms.com
 *
 * GNU General Public License, version 2:
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 *
 * ------------------------------------------------------
 *
 * @link http://www.livestreetcms.com
 * @copyright 2013 OOO "ЛС-СОФТ"
 * @author Maxim Mzhelskiy <rus.engine@gmail.com>
 *
 */

/**
 * Модуль опросов
 *
 * @package application.modules.poll
 * @since 2.0
 */
class ModulePoll extends ModuleORM
{

    /**
     * Объект текущего пользователя
     *
     * @var ModuleUser_EntityUser|null
     */
    protected $oUserCurrent;

    protected $aTargetTypes = array(
        'topic' => array(),
    );

    /**
     * Инициализация
     *
     */
    public function Init()
    {
        parent::Init();
        $this->oUserCurrent = $this->User_GetUserCurrent();
    }

    /**
     * Возвращает список типов объектов
     *
     * @return array
     */
    public function GetTargetTypes()
    {
        return $this->aTargetTypes;
    }

    /**
     * Добавляет в разрешенные новый тип
     *
     * @param string $sTargetType Тип
     * @param array $aParams Параметры
     * @return bool
     */
    public function AddTargetType($sTargetType, $aParams = array())
    {
        if (!array_key_exists($sTargetType, $this->aTargetTypes)) {
            $this->aTargetTypes[$sTargetType] = $aParams;
            return true;
        }
        return false;
    }

    /**
     * Проверяет разрешен ли данный тип
     *
     * @param string $sTargetType Тип
     * @return bool
     */
    public function IsAllowTargetType($sTargetType)
    {
        return in_array($sTargetType, array_keys($this->aTargetTypes));
    }

    /**
     * Возвращает парметры нужного типа
     *
     * @param string $sTargetType
     *
     * @return mixed
     */
    public function GetTargetTypeParams($sTargetType)
    {
        if ($this->IsAllowTargetType($sTargetType)) {
            return $this->aTargetTypes[$sTargetType];
        }
    }

    /**
     * Проверка объекта target - владелец медиа
     *
     * @param string $sTargetType Тип
     * @param int $iTargetId ID владельца
     * @return bool
     */
    public function CheckTarget($sTargetType, $iTargetId)
    {
        if (!$this->IsAllowTargetType($sTargetType)) {
            return false;
        }
        $sMethod = 'CheckTarget' . func_camelize($sTargetType);
        if (method_exists($this, $sMethod)) {
            return $this->$sMethod($iTargetId);
        }
        return false;
    }

    /**
     * Заменяет временный идентификатор на необходимый ID объекта
     *
     * @param string $sTargetType
     * @param string $sTargetId
     * @param null|string $sTargetTmp Если не задан, то берется их куки "poll_target_tmp_{$sTargetType}"
     */
    public function ReplaceTargetTmpById($sTargetType, $sTargetId, $sTargetTmp = null)
    {
        $sCookieKey = 'poll_target_tmp_' . $sTargetType;
        if (is_null($sTargetTmp) and $this->Session_GetCookie($sCookieKey)) {
            $sTargetTmp = $this->Session_GetCookie($sCookieKey);
            $this->Session_DropCookie($sCookieKey);
        }
        if (is_string($sTargetTmp)) {
            $aPollItems = $this->Poll_GetPollItemsByTargetTmpAndTargetType($sTargetTmp, $sTargetType);
            foreach ($aPollItems as $oPoll) {
                $oPoll->setTargetTmp(null);
                $oPoll->setTargetId($sTargetId);
                $oPoll->Update();
            }
        }
    }

    /**
     * Возвращает список опросов для объекта
     *
     * @param string $sTargetType
     * @param string $sTargetId
     *
     * @return mixed
     */
    public function GetPollItemsByTarget($sTargetType, $sTargetId)
    {
        $aFilter = array(
            'target_type' => $sTargetType,
            'target_id'   => $sTargetId,
            '#with'       => array('answers')
        );
        if ($this->oUserCurrent) {
            $aFilter['#with']['vote_current'] = array(
                'user_id'        => $this->oUserCurrent->getId(),
                '#value-default' => false
            );
        } else {
            $_this = $this;
            $aFilter['#with']['vote_current'] = array(
                '#value-default'   => false,
                '#callback-filter' => function ($aPollItems, &$aRelationFilter) use ($_this) {
                    $aWhere = array();
                    $aWhereBind = array();
                    foreach ($aPollItems as $oPoll) {
                        /**
                         * Смотрим по IP
                         */
                        if($oPoll->getIsGuestCheckIp()) {
                            $aWhere[] = ' ( t.poll_id = ?d and t.ip = ? ) ';
                            $aWhereBind[] = $oPoll->getId();
                            $aWhereBind[] = func_getIp();
                        }
                        /**
                         * Смотрим в куках
                         */
                        if ($sKey = $_this->Session_GetCookie($_this->GetCookieVoteName($oPoll->getId()))) {
                            $aWhere[] = ' ( t.poll_id = ?d and t.guest_key = ? ) ';
                            $aWhereBind[] = $oPoll->getId();
                            $aWhereBind[] = $sKey;
                        }
                    }
                    if ($aWhere) {
                        $aRelationFilter['#where'] = array(
                            ' ( ' . join(' or ', $aWhere) . ' ) ' => $aWhereBind
                        );
                    } else {
                        $aRelationFilter['#value-set'] = false;
                    }
                }
            );
        }
        $aPollItems = $this->Poll_GetPollItemsByFilter($aFilter);
        return $aPollItems;
    }


    /**
     * Проверка владельца с типом "topic"
     * Название метода формируется автоматически
     *
     * @param int $iTargetId ID владельца
     * @return bool
     */
    public function CheckTargetTopic($iTargetId)
    {
        if ($oTopic = $this->Topic_GetTopicById($iTargetId)) {
            if (!$oTopicType = $this->Topic_GetTopicType($oTopic->getType()) or !$oTopicType->getParam('allow_poll')) {
                return false;
            }
            /**
             * Проверяем права на редактирование топика
             */
            if ($this->ACL_IsAllowEditTopic($oTopic, $this->oUserCurrent)) {
                return true;
            }
        }
        return false;
    }

    /**
     * Голосовал ли пользователь в опросе
     *
     * @param ModulePoll_EntityPoll $oPoll
     * @param int|null $iUserId Если null, то проверяется для гостя
     * @return bool
     */
    public function CheckUserAlreadyVote($oPoll, $iUserId)
    {
        return $this->GetVoteByUser($oPoll, $iUserId) ? true : false;
    }

    /**
     * Возвращает объект голосования текущего пользователя за конкретный опрос
     *
     * @param ModulePoll_EntityPoll $oPoll
     * @param int|null $iUserId Если null, то проверяется для гостя
     * @return ModulePoll_EntityVote
     */
    public function GetVoteByUser($oPoll, $iUserId)
    {
        $iUserId = is_object($iUserId) ? $iUserId->getId() : $iUserId;
        if (is_null($iUserId)) {
            /**
             * Для гостя
             * Два варианта - проверка по IP и по кукам
             */
            if ($oPoll->getIsGuestCheckIp()) {
                if ($oVote = $this->Poll_GetVoteByIpAndPollId(func_getIp(), $oPoll->getId())) {
                    return $oVote;
                }
            }
            /**
             * По кукам
             */
            if ($sKey = $this->Session_GetCookie($this->GetCookieVoteName($oPoll))) {
                return $this->Poll_GetVoteByGuestKeyAndPollId($sKey, $oPoll->getId());
            }
            return false;
        } else {
            /**
             * Для авторизованного
             */
            return $this->Poll_GetVoteByUserIdAndPollId($iUserId, $oPoll->getId());
        }
    }

    /**
     * Возвращает название куки для хранения факта голосования
     *
     * @param $oPoll
     * @return string
     */
    public function GetCookieVoteName($oPoll)
    {
        $iPollId = is_object($oPoll) ? $oPoll->getId() : $oPoll;
        return "poll-vote-{$iPollId}";
    }
}