shibdib / Dramiel

src/plugins/onTick/authCheck.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.
 */


/**
 * Class fileAuthCheck
 * @property int nextCheck
 */
class authCheck
{
    /**
     * @var
     */
    var $config;
    /**
     * @var
     */
    var $db;
    var $dbUser;
    var $dbPass;
    var $dbName;
    var $id;
    var $allyID;
    var $corpID;
    var $exempt;
    var $alertChannel;
    /**
     * @var
     */
    var $discord;
    /**
     * @var
     */
    var $channelConfig;
    /**
     * @var int
     */
    var $lastCheck = 0;
    /**
     * @var
     */
    var $logger;

    /**
     * @param $config
     * @param $discord
     * @param $logger
     */
    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->id = $config["bot"]["guild"];
        $this->allyID = $config["plugins"]["auth"]["allianceID"];
        $this->corpID = $config["plugins"]["auth"]["corpID"];
        $this->exempt = $config["plugins"]["auth"]["exempt"];
        $this->alertChannel = $config["plugins"]["auth"]["alertChannel"];
        $this->nextCheck = 0;

        //check if cache has been set
        $permsChecked = getPermCache("permsLastChecked");

        //if not set set for now (30 minutes from now for role removal)
        if ($permsChecked == NULL) {
            setPermCache("permsLastChecked", time() - 5);
            setPermCache("authStateLastChecked", time() + 7200);
        }
    }

    /**
     * @return array
     */
    function information()
    {
        return array(
            "name" => "",
            "trigger" => array(),
            "information" => ""
        );
    }

    function tick()
    {
        $permsChecked = getPermCache("permsLastChecked");
        $stateChecked = getPermCache("authStateLastChecked");

        if ($permsChecked <= time()) {
            $this->logger->addInfo("AuthCheck: Checking for users who have left corp/alliance....");
            $this->checkPermissions();
            $this->logger->addInfo("AuthCheck: Corp/alliance check complete.");
        }

        if ($stateChecked <= time()) {
            $this->logger->addInfo("AuthCheck: Checking for users who have been wrongly given roles....");
            $this->checkAuthState();
            $this->logger->addInfo("AuthCheck: Role check complete.");
        }
    }

    /**
     * @return null
     */

    //Remove members who have roles but never authed
    function checkPermissions()
    {
        //Get guild object
        $guild = $this->discord->guilds->get('id', $this->id);

        //Establish connection to mysql
        $conn = new mysqli($this->db, $this->dbUser, $this->dbPass, $this->dbName);

        $sql = "SELECT characterID, discordID, eveName FROM authUsers WHERE active='yes'";

        $result = $conn->query($sql);

        if ($result->num_rows >= 1) {
            while ($rows = $result->fetch_assoc()) {
                $charID = $rows['characterID'];
                $discordID = $rows['discordID'];
                $member = $guild->members->get("id", $discordID);
                $eveName = $rows['eveName'];
                if (is_null($member->roles)) {
                    continue;
                }

                if ($this->config["plugins"]["auth"]["nameEnforce"] == "true" && !is_null($member)) {
                    $nick = $eveName;
                    $member->setNickname($nick);
                }

                $url = "https://api.eveonline.com/eve/CharacterAffiliation.xml.aspx?ids=$charID";
                $xml = makeApiRequest($url);
                // Stop the process if the api is throwing an error
                if (is_null($xml)) {
                    $this->logger->addInfo("{$eveName} cannot be authed, API issues detected.");
                    return null;
                }
                if ($xml->result->rowset->row[0]) {
                    foreach ($xml->result->rowset->row as $character) {
                        if ($character->attributes()->allianceID != $this->allyID && $character->attributes()->corporationID != $this->corpID) {
                            // Deactivate user in database
                            $sql = "UPDATE authUsers SET active='no' WHERE discordID='$discordID'";
                            $this->logger->addInfo("AuthCheck: {$eveName} account has been deactivated as they are no longer in the correct corp/alliance.");
                            $conn->query($sql);
                        }
                    }
                }
            }
            $nextCheck = time() + 10800;
            setPermCache("permsLastChecked", $nextCheck);
            return null;
        }
        $nextCheck = time() + 10800;
        setPermCache("permsLastChecked", $nextCheck);
        return null;
    }

    //Check user corp/alliance affiliation


    function checkAuthState()
    {

        //Check if exempt roles are set
        if (is_null($this->exempt)) {
            $this->exempt = "0";
        }

        //Establish connection to mysql
        $conn = new mysqli($this->db, $this->dbUser, $this->dbPass, $this->dbName);

        //get bot ID so we don't remove out own roles
        $botID = $this->discord->id;

        //Get guild object
        $guild = $this->discord->guilds->get('id', $this->id);

        //Check to make sure guildID is set correctly
        if (is_null($guild)) {
            $this->logger->addError("Config Error: Ensure the guild entry in the config is the guildID (aka serverID) for the main server that the bot is in.");
            $nextCheck = time() + 7200;
            setPermCache("authLastChecked", $nextCheck);
            return null;
        }

        //Perform check if roles were added without permission
        foreach ($guild->members as $member) {
            $id = $member->id;
            $username = $member->username;
            $roles = $member->roles;

            //Skip to next member if this user has no roles
            if (is_null($roles)) {
                continue;
            }
            $sql = "SELECT * FROM authUsers WHERE discordID='$id' AND active='yes'";
            $result = $conn->query($sql);

            //If they are NOT active in the db, check for roles to remove
            if ($result->num_rows == 0) {
                foreach ($roles as $role) {
                    if (!isset($role->name)) {
                        if ($id != $botID && !in_array($role->name, $this->exempt, true)) {
                            $member->removeRole($role);
                            $guild->members->save($member);
                            // Send the info to the channel
                            $msg = "{$username} has been removed from the {$role->name} role.";

$msg is not used, you could remove the assignment.

                            //queueMessage($msg, $this->alertChannel, $this->guild);

67% of this comment could be valid code. Did you maybe forget this after debugging?

                            $this->logger->addInfo("AuthCheck: {$username} has been removed from the {$role->name} role.");
                        }
                    }
                }
            }
        }
        $nextCheck = time() + 1800;
        setPermCache("authStateLastChecked", $nextCheck);
        return null;
    }
}