added user id scope

akaunting · Jan 29, 2021 · 42af602 · 42af602
1 parent 26451e0
commit 42af602
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/app/Models/Common/Company.php b/app/Models/Common/Company.php
@@ -325,6 +325,20 @@ public function scopeEnabled($query, $value = 1)
         return $query->where('enabled', $value);
     }
 
+    /**
+     * Scope to only include companies of a given user id.
+     *
+     * @param \Illuminate\Database\Eloquent\Builder $query
+     * @param int $user_id
+     * @return \Illuminate\Database\Eloquent\Builder
+     */
+    public function scopeUserId($query, $user_id)
+    {
+        return $query->whereHas('users', function ($query) use ($user_id) {
+            $query->where('user_id', $user_id);
+        });
+    }
+
     /**
      * Sort by company name
      *

diff --git a/app/Models/Common/Dashboard.php b/app/Models/Common/Dashboard.php
@@ -45,6 +45,20 @@ public function widgets()
         return $this->hasMany('App\Models\Common\Widget')->orderBy('sort', 'asc');
     }
 
+    /**
+     * Scope to only include dashboards of a given user id.
+     *
+     * @param \Illuminate\Database\Eloquent\Builder $query
+     * @param int $user_id
+     * @return \Illuminate\Database\Eloquent\Builder
+     */
+    public function scopeUserId($query, $user_id)
+    {
+        return $query->whereHas('users', function ($query) use ($user_id) {
+            $query->where('user_id', $user_id);
+        });
+    }
+
     /**
      * Create a new factory instance for the model.
      *