shibdib / Dramiel

src/plugins/onMessage/auth.php

<?php
/**
 * The MIT License (MIT).
 *
 * Copyright (c) 2016 Robert Sardinia
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
use Discord\Discord;
use Discord\Parts\Channel\Message;

/**
 * Class auth.
 *
 * @property  message
 */
class auth
{
    /*
     * @var
     */
    public $config;
    /*
     * @var
     */
    public $discord;
    /*
     * @var
     */
    public $logger;
    public $solarSystems;
    public $triggers = [];
    public $guildID;
    public $roleName;
    public $corpID;
    public $db;
    public $dbUser;
    public $dbPass;
    public $dbName;
    public $forceName;
    public $ssoUrl;
    public $nameEnforce;
    public $allianceID;
    public $allyroleName;

    /**
     * @param $config
     * @param $discord
     * @param $logger
     */
    public function init($config, $discord, $logger)
    {
        $this->config = $config;
        $this->discord = $discord;
        $this->logger = $logger;
        $this->db = $config['database']['host'];
        $this->dbUser = $config['database']['user'];
        $this->dbPass = $config['database']['pass'];
        $this->dbName = $config['database']['database'];
        $this->corpID = (int) $config['plugins']['auth']['corpID'];
        $this->allianceID = (int) $config['plugins']['auth']['allianceID'];
        $this->guildID = $config['plugins']['auth']['guildID'];
        $this->roleName = $config['plugins']['auth']['corpMemberRole'];
        $this->allyroleName = $config['plugins']['auth']['allyMemberRole'];
        $this->nameEnforce = $config['plugins']['auth']['nameEnforce'];
        $this->ssoUrl = $config['plugins']['auth']['url'];
    }


    public function tick()
    {
    }

    /**
     * @param $msgData
     * @param $message
     *
     * @return null
     */
    public function onMessage($msgData, $message)
    {
        $this->message = $message;
        $userID = $msgData['message']['fromID'];
        $userName = $msgData['message']['from'];
        $message = $msgData['message']['message'];
        $data = command($message, $this->information()['trigger'], $this->config['bot']['trigger']);
        if (isset($data['trigger'])) {
            if (isset($this->config['bot']['primary'])) {
                $userID = $msgData['message']['fromID'];
                $channelInfo = $this->message->getFullChannelAttribute();
                $guildID = $channelInfo[@guild_id];
                if ($guildID != $this->config['bot']['primary']) {
                    $this->message->reply('**Failure:** The auth code your attempting to use is for another discord server');

                    return;
                }
            }
            $code = $data['messageString'];
            $result = selectPending($this->db, $this->dbUser, $this->dbPass, $this->dbName, $code);

            if (strlen($code) < 12) {
                $this->message->reply('Invalid Code, check '.$this->config['bot']['trigger'].'help auth for more info.');

                return;
            }

            while ($rows = $result->fetch_assoc()) {
                $charid = (int) $rows['characterID'];
                $corpid = (int) $rows['corporationID'];
                $allianceid = (int) $rows['allianceID'];
                $url = "https://api.eveonline.com/eve/CharacterName.xml.aspx?ids=$charid";
                $xml = makeApiRequest($url);



                // We have an error, show it it
                if ($xml->error) {
                    $this->message->reply('**Failure:** Eve API error, please try again in a little while.');

                    return;
                }

                if (!isset($xml->result->rowset->row)) {
                    $this->message->reply('**Failure:** Eve API error, please try again in a little while.');

                    return;
                } elseif ($this->nameEnforce == 'true') {
                    foreach ($xml->result->rowset->row as $character) {
                        if ($character->attributes()->name != $userName) {
                            $this->message->reply('**Failure:** Your discord name must match your character name.');
                            $this->logger->addInfo('User was denied due to not having the correct name '.$character->attributes()->name);

                            return;
                        }
                    }
                }
                foreach ($xml->result->rowset->row as $character) {
                    $eveName = $character->attributes()->name;
                    if ($corpid === $this->corpID) {

This code seems to be duplicated across your project.

                        $roles = $this->message->getFullChannelAttribute()->getGuildAttribute()->getRolesAttribute();
                        $member = $this->message->getFullChannelAttribute()->getGuildAttribute()->getMembersAttribute()->get('id', $userID);
                        foreach ($roles as $role) {
                            $roleName = $role->name;
                            if ($roleName == $this->roleName) {
                                $member->addRole($role);
                                $member->save();
                                insertUser($this->db, $this->dbUser, $this->dbPass, $this->dbName, $userID, $charid, $eveName, 'corp');
                                disableReg($this->db, $this->dbUser, $this->dbPass, $this->dbName, $code);
                                $this->message->reply('**Success:** You have now been added to the '.$this->roleName.' group. To get more roles, talk to the CEO / Directors');
                                $this->logger->addInfo('User authed and added to corp group '.$eveName);

                                return;
                            }
                        }
                    }
                    if ($allianceid === $this->allianceID) {

This code seems to be duplicated across your project.

                        $roles = $this->message->getFullChannelAttribute()->getGuildAttribute()->getRolesAttribute();
                        $member = $this->message->getFullChannelAttribute()->getGuildAttribute()->getMembersAttribute()->get('id', $userID);
                        foreach ($roles as $role) {
                            $roleName = $role->name;
                            if ($roleName == $this->allyroleName) {
                                $member->addRole($role);
                                $member->save();
                                insertUser($this->db, $this->dbUser, $this->dbPass, $this->dbName, $userID, $charid, $eveName, 'ally');
                                disableReg($this->db, $this->dbUser, $this->dbPass, $this->dbName, $code);
                                $this->message->reply('**Success:** You have now been added to the '.$this->allyroleName.' group. To get more roles, talk to the CEO / Directors');
                                $this->logger->addInfo('User authed and added to the alliance group '.$eveName);

                                return;
                            }
                        }
                    }
                    $this->message->reply('**Failure:** There are no roles available for your corp/alliance.');
                    $this->logger->addInfo('User was denied due to not being in the correct corp or alliance '.$eveName);

                    return;
                }
            }
            $this->message->reply('**Failure:** There was an issue with your code.');
            $this->logger->addInfo('User was denied due to not being in the correct corp or alliance '.$userName);

            return;
        }
    }

    /**
     * @return array
     */
    public function information()
    {
        return [
            'name'        => 'auth',
            'trigger'     => [$this->config['bot']['trigger'].'auth'],
            'information' => 'SSO based auth system. '.$this->ssoUrl.' Visit the link and login with your main EVE account, select the correct character, and put the !auth <string> you receive in chat.',
        ];
    }

    /**
     * @param $msgData
     */
    public function onMessageAdmin($msgData)
    {
    }
}