1 files changed, 14 insertions, 8 deletions

diff --git a/framework/Wsat/pages/TWsatGenerateAR.php b/framework/Wsat/pages/TWsatGenerateAR.php
index 3d4291e2..d5934686 100644
--- a/framework/Wsat/pages/TWsatGenerateAR.php
+++ b/framework/Wsat/pages/TWsatGenerateAR.php
@@ -1,9 +1,13 @@
 <?php
 
 /**
- * Description of Inicio
- *
- * @author daniels
+ * @author Daniel Sampedro Bello <darthdaniel85@gmail.com>
+ * @link http://www.pradosoft.com/
+ * @copyright Copyright &copy; 2005-2013 PradoSoft
+ * @license http://www.pradosoft.com/license/
+ * @version $Id$
+ * @since 3.3
+ * @package Wsat.pages
  */
 Prado::using("System.Wsat.TWsatARGenerator");
 
@@ -12,13 +16,15 @@ class TWsatGenerateAR extends TPage {
     public function generate($sender) {
         if ($this->IsValid) {
             $table_name = $this->table_name->Text;
-            $class_prefix = $this->class_prefix->Text;
             $output_folder_ns = $this->output_folder->Text;
+            $class_prefix = $this->class_prefix->Text;
+            $class_sufix = $this->class_sufix->Text;
 
             try {
                 $ar_generator = new TWsatARGenerator();
                 $ar_generator->setOpFile($output_folder_ns);
                 $ar_generator->setClasPrefix($class_prefix);
+                $ar_generator->setClassSufix($class_sufix);
 
                 if ($this->build_rel->Checked) {
                     $ar_generator->buildRelations();
@@ -28,20 +34,20 @@ class TWsatGenerateAR extends TPage {
                 } else {
                     $ar_generator->generateAll();
                 }
-                $this->success_panel->CssClass = "success_panel";
+                $this->feedback_panel->CssClass = "green_panel";
                 $this->generation_msg->Text = "The code has been generated successfully.";
             } catch (Exception $ex) {
-                $this->success_panel->CssClass = "exception_panel";
+                $this->feedback_panel->CssClass = "red_panel";
                 $this->generation_msg->Text = $ex->getMessage();
             }
-            $this->success_panel->Visible = true;
+            $this->feedback_panel->Visible = true;
         }
     }
 
     public function preview($sender) {
 //        $ar_generator = new TWsatARGenerator();
 //        $ar_generator->renderAllTablesInformation();
-         throw new THttpException(500, "Not implemented yet.");
+        throw new THttpException(500, "Not implemented yet.");
     }
 
 }