DataObjects/Core_company.php

[Pman.Core] / DataObjects / Core_company.php

diff --git a/DataObjects/Core_company.php b/DataObjects/Core_company.php
index 17c657c..13ddb6b 100644 (file)
--- a/DataObjects/Core_company.php
+++ b/DataObjects/Core_company.php
@@ -2,7 +2,7 @@
 /**
  * Table Definition for Companies
  */
-require_once 'DB/DataObject.php';
+class_exists('DB_DataObject') ? '' : require_once 'DB/DataObject.php';
 
 class Pman_Core_DataObjects_Core_Company extends DB_DataObject 
 {
@@ -240,13 +240,29 @@ class Pman_Core_DataObjects_Core_Company extends DB_DataObject
             $this->update();
         }
         
-        
-        
-        
+    }
+    
+    function beforeInsert($q, $roo)
+    {
+        if(!empty($q['_check_name'])){
+            if($this->checkName()){
+                $roo->jok('OK');
+            }
+            
+            $roo->jerr('EXIST');
+        }
     }
     
     function beforeUpdate($old, $q,$roo)
     {
+        if(!empty($q['_check_name'])){
+            if($this->checkName()){
+                $roo->jok('OK');
+            }
+            
+            $roo->jerr('EXIST');
+        }
+        
         if(!empty($q['_merge_id'])){
             $this->merge($q['_merge_id'], $roo);
         }
@@ -466,4 +482,22 @@ class Pman_Core_DataObjects_Core_Company extends DB_DataObject
         $roo->jok('Merged');
         
     }
+    
+    function checkName()
+    {
+        $company = DB_DataObject::factory('core_company');
+        $company->setFrom(array(
+            'name' => $this->name
+        ));
+        
+        if(!empty($this->id)){
+            $company->whereAdd("id != {$this->id}");
+        }
+        
+        if(!$company->find(true)){
+            return true;
+        }
+        
+        return false;
+    }
 }