예제 #1
0
파일: dao.php 프로젝트: pixxid/xengine
 /**
  * Génère les dao pour l'ensemble des tables (ou seulement moduleName)
  * @param string $option1
  * @param string $option2
  * @param string $option3
  * @param string $option4
  * @param string $option5
  *
  * @return bool
  */
 public function generate($option1 = null, $option2 = null, $option3 = null, $option4 = null, $option5 = null)
 {
     // On gère les différentes options passées
     $generateAllModels = false;
     $generateDao = false;
     $generateDaoCust = false;
     $overWriteDaoCust = false;
     $generateBusiness = false;
     $verbose = false;
     $model = null;
     // On génère tous les modèles ?
     if ($option1 === '--all' || $option2 === '--all' || $option3 === '--all' || $option4 === '--all' || $option5 === '--all') {
         $generateAllModels = true;
         // On regarde maintenant si un modèle particulier a été demandé
     } elseif ($option1 != null && !in_array($option1, array('--all', '--business', '--dao', '--daocust', '--verbose'))) {
         $model = $option1;
     } elseif ($option2 != null && !in_array($option2, array('--all', '--business', '--dao', '--daocust', '--verbose'))) {
         $model = $option2;
     } elseif ($option3 != null && !in_array($option3, array('--all', '--business', '--dao', '--daocust', '--verbose'))) {
         $model = $option3;
     } elseif ($option4 != null && !in_array($option4, array('--all', '--business', '--dao', '--daocust', '--verbose'))) {
         $model = $option4;
     } elseif ($option5 != null && !in_array($option5, array('--all', '--business', '--dao', '--daocust', '--verbose'))) {
         $model = $option5;
     }
     // On génère les dao ?
     if ($option1 === '--dao' || $option2 === '--dao' || $option3 === '--dao' || $option4 === '--dao' || $option5 === '--dao') {
         $generateDao = true;
     }
     // On génère les daoCust ?
     if ($option1 === '--daocust' || $option2 === '--daocust' || $option3 === '--daocust' || $option4 === '--daocust' || $option5 === '--daocust') {
         $generateDaoCust = true;
         $overWriteDaoCust = true;
     }
     // On génère les business ?
     if ($option1 === '--business' || $option2 === '--business' || $option3 === '--business' || $option4 === '--business' || $option5 === '--business') {
         $generateBusiness = true;
     }
     // On affiche le détail ?
     if ($option1 === '--verbose' || $option2 === '--verbose' || $option3 === '--verbose' || $option4 === '--verbose' || $option5 === '--verbose') {
         $verbose = true;
     }
     // Si aucune option n'a été passée, on génère tous les dao
     if (!$generateBusiness && !$generateDao && !$generateDaoCust) {
         $generateBusiness = true;
         $generateDao = true;
         $generateDaoCust = true;
     }
     $models = array();
     // Récupère l'ensemble des tables de la base
     $tables = $this->listTables();
     // On liste les tables afin que l'utilisateur valide celles qu'il veut, si l'option -a n'a pas été passée
     if (!$generateAllModels && $model === null) {
         echo helper::warning("Veuillez sélectionner les modèles à générer. Tappez [ENTER]/[o] pour valider le modèle, [n] pour le rejeter.\r\n");
         foreach ($tables as $table) {
             echo helper::success($table[0] . "\r\n");
             echo helper::info(">> ");
             $input = trim(fgets(STDIN));
             if (!in_array($input, array('n', 'N', 'no', 'NO', 'NON', 'non'))) {
                 $models[] = $table[0];
             }
         }
         // Si c'est un modèle particulier demandé
     } elseif ($model !== null) {
         // On vérifie que le modèle demandé est présent dans la liste des tables
         foreach ($tables as $table) {
             if (strtolower($model) === strtolower($table[0])) {
                 $models[] = $table[0];
                 break;
             }
         }
         // Sinon on les prend tous automatiquement
     } else {
         if ($verbose) {
             echo helper::info("Génération de tous les modèles\r\n");
         }
         foreach ($tables as $table) {
             $models[] = $table[0];
         }
     }
     // On va créer chaque modèle
     foreach ($models as $model) {
         try {
             $columns = $this->listColumns($model);
             $fullColumns = $this->prepareColumns($model, $columns);
             // Création du fichier business
             if ($generateBusiness) {
                 writeBusiness::write($model, $fullColumns, $verbose);
             }
             // Création du fichier dao
             if ($generateDao) {
                 writeDao::write($model, $fullColumns, $verbose);
             }
             // Création du fichier daoCust
             if ($generateDaoCust) {
                 writeDaoCust::write($model, $fullColumns, $overWriteDaoCust, $verbose);
             }
         } catch (\Exception $e) {
             echo helper::error("Génération pour {$model} : {$e->getMessage()}\r\n");
             return false;
         }
     }
     // Si des modèles ont été générés, on l'indique
     if (sizeof($models)) {
         echo helper::success("Génération terminée.\r\n");
     } else {
         echo helper::warning("Aucun modèle généré.\r\n");
     }
     return true;
 }