1: <?php
  2: 
  3: namespace Budabot\Core\Modules;
  4: 
  5: use Budabot\Core\SQLException;
  6: 
  7: /**
  8:  * Authors: 
  9:  *  - Tyrence (RK2)
 10:  *
 11:  * @Instance
 12:  *
 13:  * Commands this controller contains:
 14:  *  @DefineCommand(
 15:  *      command       = 'loadsql',
 16:  *      accessLevel   = 'admin',
 17:  *      description   = 'Manually reload an SQL file',
 18:  *      help          = 'loadsql.txt'
 19:  *  )
 20:  *  @DefineCommand(
 21:  *      command       = 'querysql',
 22:  *      accessLevel   = 'superadmin',
 23:  *      description   = 'Run an SQL query and see the results'
 24:  *  )
 25:  *  @DefineCommand(
 26:  *      command       = 'executesql',
 27:  *      accessLevel   = 'superadmin',
 28:  *      description   = 'Execute an SQL statement'
 29:  *  )
 30:  */
 31: class SQLController {
 32: 
 33:     /**
 34:      * Name of the module.
 35:      * Set automatically by module loader.
 36:      */
 37:     public $moduleName;
 38: 
 39:     /** @Inject */
 40:     public $accessManager;
 41:     
 42:     /** @Inject */
 43:     public $db;
 44: 
 45:     /** @Inject */
 46:     public $commandManager;
 47: 
 48:     /** @Inject */
 49:     public $text;
 50: 
 51:     /**
 52:      * @Setup
 53:      * This handler is called on bot startup.
 54:      */
 55:     public function setup() {
 56: 
 57:     }
 58:     
 59:     /**
 60:      * @HandlesCommand("executesql")
 61:      * @Matches("/^executesql (.*)$/i")
 62:      */
 63:     public function executesqlCommand($message, $channel, $sender, $sendto, $args) {
 64:         $sql = htmlspecialchars_decode($args[1]);
 65: 
 66:         try {
 67:             $num_rows = $this->db->exec($sql);
 68:             $msg = "$num_rows rows affected.";
 69:         } catch (SQLException $e) {
 70:             $msg = $this->text->makeBlob("SQL Error", $e->getMessage());
 71:         }
 72:         $sendto->reply($msg);
 73:     }
 74: 
 75:     /**
 76:      * @HandlesCommand("querysql")
 77:      * @Matches("/^querysql (.*)$/si")
 78:      */
 79:     public function querysqlCommand($message, $channel, $sender, $sendto, $args) {
 80:         $sql = htmlspecialchars_decode($args[1]);
 81: 
 82:         try {
 83:             $data = $this->db->query($sql);
 84:             $count = count($data);
 85: 
 86:             $msg = $this->text->makeBlob("Results ($count)", print_r($data, true));
 87:         } catch (SQLException $e) {
 88:             $msg = $this->text->makeBlob("SQL Error", $e->getMessage());
 89:         }
 90:         $sendto->reply($msg);
 91:     }
 92: 
 93:     /**
 94:      * @HandlesCommand("loadsql")
 95:      * @Matches("/^loadsql (.*) (.*)$/i")
 96:      */
 97:     public function loadsqlCommand($message, $channel, $sender, $sendto, $args) {
 98:         $module = strtoupper($args[1]);
 99:         $name = strtolower($args[2]);
100:     
101:         $this->db->beginTransaction();
102:     
103:         $msg = $this->db->loadSQLFile($module, $name, true);
104:     
105:         $this->db->commit();
106:     
107:         $sendto->reply($msg);
108:     }
109: }
110: