root/buddyloo/files/lib/system/event/listener/BuddylooRegisterFormListener.class.php @ 323

<?php
// WCF include
require_once(WCF_DIR.'lib/system/event/EventListener.class.php');

/**
 * checks if the friend is invited
 *
 * @author      Torben Brodt
 * @package     de.easy-coding.wcf.buddyloo
 * @license     GNU General Public License <http://opensource.org/licenses/gpl-3.0.html>
 */
class BuddylooRegisterFormListener implements EventListener {
        protected $eventObj;
        protected $className;
        
        //data
        protected $friends = array();

        /**
         * @see EventListener::execute()
         */
        public function execute($eventObj, $className, $eventName) {    
                $this->eventObj = $eventObj;
                $this->className = $className;

                switch ($eventName) {
                        case 'assignVariables':
                                if(!REGISTER_INVITATIONS) return;
                                $this->assignVariables();
                                break;
                        case 'validate':
                                if(!REGISTER_INVITATIONS) return;
                                $this->fetchInvitations();
                                $this->validate();
                                break;
                        case 'saved':
                                $this->fetchInvitations();
                                $this->saved();
                                break;
                }
        }
        
        /**
         * show info... registration just for invited users
         */
        protected function assignVariables() {
                WCF::getTPL()->append('userMessages', '<p class="info">'.WCF::getLanguage()->get('wcf.buddyloo.invitation.register.description').'</p>');
        }
        
        /**
         * fetchs invitations
         */
        protected function fetchInvitations() {
                $email = $this->eventObj->email;
                $sql = "SELECT          userID
                        FROM            wcf".WCF_N."_buddyloo_invitation
                        WHERE           email = '".escapeString($this->eventObj->email)."'";
                $result = WCF::getDB()->sendQuery($sql);
                while ($row = WCF::getDB()->fetchArray($result)) {
                        $this->friends[] = $row['userID'];
                }
        }
        
        /**
         * check if the user had an invitation and add him to friendlist
         */
        protected function validate() {
                if(count($this->friends) == 0) {
                        throw new UserInputException('email', 'notValid');
                }
        }
        
        /**
         * check if the user had an invitation and add him to friendlist
         */
        protected function saved() {
                $inserts = '';
                foreach($this->friends as $friend)  {
                        if (!empty($inserts)) $inserts .= ',';
                        $inserts .= "(".$friend.", ".$this->eventObj->user->userID.")";
                }
                
                if(!empty($inserts)) {
                        $sql = "INSERT IGNORE INTO      wcf".WCF_N."_buddyloo
                                                        (userID, whiteUserID)
                                VALUES                  ".$inserts;
                        WCF::getDB()->sendQuery($sql);
                
                        $sql = "DELETE FROM             wcf".WCF_N."_buddyloo_invitation
                                WHERE                   email = '".escapeString($this->eventObj->user->email)."'; ";
                        WCF::getDB()->sendQuery($sql);
                        
                        $this->updateCache();
                }
        }
        
                

        /**
         * updates cache
         */
        protected function updateCache() {
                // clear cache
                WCF::getCache()->clear(WBB_DIR.'cache', 'cache.buddyloo.*.php');

                // add resource
                WCF::getCache()->addResource('buddyloo.whitelist', 
                        WCF_DIR.'cache/cache.buddyloo.whitelist.php',
                        WCF_DIR.'lib/system/cache/CacheBuilderBuddylooWhiteList.class.php');
        }
}
?>