From: Alan Knowles <alan@roojs.com>
Date: Fri, 15 Nov 2013 09:01:49 +0000 (+0800)
Subject: DataObjects/Companies.php
X-Git-Url: http://git.roojs.org/?a=commitdiff_plain;h=8a459d23b15cedd9d49840b43ecdf0fac8d125b9;p=Pman.Core

DataObjects/Companies.php
---

diff --git a/DataObjects/Companies.php b/DataObjects/Companies.php
index cffb2416..d6501c54 100644
--- a/DataObjects/Companies.php
+++ b/DataObjects/Companies.php
@@ -322,17 +322,24 @@ class Pman_Core_DataObjects_Companies extends DB_DataObject
     function initCompanies($roo, $opts)
     {
         $companies = DB_DataObject::factory('companies');
-        $enum = DB_DataObject::Factory('core_enum')->lookup('comptype', $opts['type']);
+        $enum = DB_DataObject::Factory('core_enum')->lookup('comptype',
+                empty($opts['add-company-with-type']) ? 'OWNER' : $opts['type);
         
         $companies->setFrom(array(
             'name' => $opts['add-company'],
             'comptype' => $type,
             'comptype_id' => $enum,
+        ));
+        if ($companies->find(true)) {
+            $roo->jerr("company already exists");
+        }
+        $companies->setFrom(array(
             'background_color' => '',
             'created_dt' => $this->sqlValue('NOW()'),
             'updated_dt' => $this->sqlValue('NOW()')
         ));
         
+        
         $companies->insert();
         $companies->onInsert(array(), $roo);
     }