UserQueryBuilder

Author: mooxbot

src/Models/WpUser.php

@@ -11,6 +11,7 @@
 use Illuminate\Notifications\Notifiable;
 use Illuminate\Support\Str;
 use Moox\Press\Database\Factories\WpUserFactory;
+use Moox\Press\QueryBuilder\UserQueryBuilder;
 
 /**
  * @property int $ID
@@ -98,6 +99,15 @@ protected static function boot()
         });
     }
 
+    protected function newBaseQueryBuilder()
+    {
+        $connection = $this->getConnection();
+        $grammar = $connection->getQueryGrammar();
+        $processor = $connection->getPostProcessor();
+
+        return new UserQueryBuilder($connection, $grammar, $processor);
+    }
+
     protected $casts = [
         'user_registered' => 'datetime',
         'spam' => 'boolean',

src/QueryBuilder/UserQueryBuilder.php

@@ -0,0 +1,24 @@
+<?php
+
+namespace Moox\Press\QueryBuilder;
+
+use Illuminate\Database\Query\Builder;
+
+class UserQueryBuilder extends Builder
+{
+    protected $aliasMap = [
+        'email' => 'user_email',
+        'name' => 'user_login',
+        'password' => 'user_pass',
+        'id' => 'ID',
+    ];
+
+    public function where($column, $operator = null, $value = null, $boolean = 'and')
+    {
+        if (is_string($column) && array_key_exists($column, $this->aliasMap)) {
+            $column = $this->aliasMap[$column];
+        }
+
+        return parent::where($column, $operator, $value, $boolean);
+    }
+}