move draftRecord HasOne to HasRevisor Trait

src/Concerns/HasPublishing.php

@@ -192,25 +192,6 @@ public function publishedRecord(): HasOne
         );
     }
 
-    /**
-     * Get the draft record for this model
-     *
-     * @throws Exception
-     */
-    public function draftRecord(): HasOne
-    {
-        if ($this->isDraftTableRecord()) {
-            throw new Exception('The draft record HasOne relationship is only available to Published and Version records');
-        }
-
-        $instance = (new static)->withDraftContext();
-        $localKey = $this->isVersionTableRecord() ? 'record_id' : $this->getKeyName();
-
-        return $this->newHasOne(
-            $instance, $this, $instance->getModel()->getTable().'.'.$this->getKeyName(), $localKey
-        );
-    }
-
     /**
      * Get the publisher relationship for this model
      */

src/Concerns/HasRevisor.php

@@ -4,7 +4,9 @@
 
 namespace Indra\Revisor\Concerns;
 
+use Exception;
 use Illuminate\Database\Eloquent\Builder;
+use Illuminate\Database\Eloquent\Relations\HasOne;
 use Illuminate\Support\Facades\DB;
 use Illuminate\Support\Str;
 use Indra\Revisor\Contracts\HasRevisor as HasRevisorContract;
@@ -92,6 +94,25 @@ public function getDraftTable(): string
         return Revisor::getDraftTableFor($this->getBaseTable());
     }
 
+    /**
+     * Get the draft record for this model
+     *
+     * @throws Exception
+     */
+    public function draftRecord(): HasOne
+    {
+        if ($this->isDraftTableRecord()) {
+            throw new Exception('The draft record HasOne relationship is only available to Published and Version records');
+        }
+
+        $instance = (new static)->withDraftContext();
+        $localKey = $this->isVersionTableRecord() ? 'record_id' : $this->getKeyName();
+
+        return $this->newHasOne(
+            $instance, $this, $instance->getModel()->getTable().'.'.$this->getKeyName(), $localKey
+        );
+    }
+
     /**
      * Get a Builder instance for the Draft table
      */