From db5294332c9798dc7928d92ebf3378cbf6c3cc11 Mon Sep 17 00:00:00 2001
From: weichien00 <weichien00@gmail.com>
Date: Fri, 9 Jul 2021 00:34:07 +0800
Subject: [PATCH] update: address,conpanyModule,user models .

---
 app/Models/Address.php       | 8 --------
 app/Models/CompanyModule.php | 8 ++++++++
 app/Models/User.php          | 9 +++++++++
 3 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/app/Models/Address.php b/app/Models/Address.php
index 7857756f..15340b3c 100644
--- a/app/Models/Address.php
+++ b/app/Models/Address.php
@@ -35,14 +35,6 @@ class Address extends AbstractModel
         return $this->BelongsTo(Country::class, 'country_id', 'id');
     }
 
-    /**
-     * @return BelongsTo
-     */
-    public function company(): BelongsTo
-    {
-        return $this->BelongsTo(Company::class, 'company_id', 'id');
-    }
-
     /**
      * @return BelongsTo
      */
diff --git a/app/Models/CompanyModule.php b/app/Models/CompanyModule.php
index 9c6026d8..34ea5ec3 100644
--- a/app/Models/CompanyModule.php
+++ b/app/Models/CompanyModule.php
@@ -31,6 +31,14 @@ class CompanyModule extends AbstractModel
         return $this->morphMany(Contact::class, 'owner');
     }
 
+    /**
+     * @return MorphMany
+     */
+    public function addresses(): morphMany
+    {
+        return $this->morphMany(Address::class, 'owner');
+    }
+
     /**
      * @return belongsToMany
      */
diff --git a/app/Models/User.php b/app/Models/User.php
index 5d5cfc2f..5ed99155 100644
--- a/app/Models/User.php
+++ b/app/Models/User.php
@@ -4,6 +4,7 @@ namespace App\Models;
 
 use Illuminate\Database\Eloquent\Relations\BelongsToMany;
 use Illuminate\Database\Eloquent\Relations\HasMany;
+use Illuminate\Database\Eloquent\Relations\MorphMany;
 
 use Spatie\Permission\Traits\HasRoles;
 
@@ -64,4 +65,12 @@ class User extends AbstractModel implements
     {
         return $this->belongsToMany(companyModule::class, (new CompanyEmployee())->getTable(), 'user_id', 'company_module_id');
     }
+
+    /**
+     * @return MorphMany
+     */
+    public function contacts(): morphMany
+    {
+        return $this->morphMany(Contact::class, 'owner');
+    }
 }