Added support query for sqlite

yourjhay · May 18, 2020 · c4b018e · c4b018e
1 parent f816e27
commit c4b018e
Showing 1 changed file with 31 additions and 3 deletions.
diff --git a/src/Simple/Engine/Console.php b/src/Simple/Engine/Console.php
@@ -312,6 +312,7 @@ private function migrate($file, $com)
             try {
             $table = 'users';
             $db = new PDO("sqlite:"."./database/database.db");
+            $db->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);
             $db->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);  
                 if($file=='-c'){
                     $sql = $com;
@@ -323,10 +324,37 @@ private function migrate($file, $com)
                         password_hash TEXT NOT NULL,
                         reset_token TEXT NULL)";
                 }           
-                    echo($sql);
+
                     echo PHP_EOL;
-                    $db->exec($sql);
-                print("Command successfull\n");          
+                    $command_ = explode(' ',$sql);
+                    if(strtoupper($command_[0]) === "SELECT"){
+                        $res = $db->query($sql);   
+                        $i=0;          
+                        $col=[]; 
+                        $rows=[]; 
+                        $v=[];
+                        $table = new \LucidFrame\Console\ConsoleTable(); 
+                        foreach($res as $row)
+                        {
+                            $i++;
+                            foreach($row as $key => $val){
+                               if($i==1){
+                                    $col[] .= $key;
+
+                                } $v[]=$val;
+                            }
+                            $rows[$i] = $v;
+                            unset($v);
+                        }                       
+                        $table->setHeaders($col);                      
+                        foreach($rows as $r){
+                            $table->addRow($r);
+                        }
+                        $table->display();
+                    }else{
+                        $c = $db->exec($sql);
+                        print("Command successfull. $c affected rows.\n");  
+                    }                  
             } catch (Exception $e) {
                 echo "Unable to connect".PHP_EOL;
                 echo $e->getMessage();