Allow behaviors to load classes without being the main model of a capsule

src/Models/Behaviors/HasRevisions.php

@@ -64,12 +64,6 @@ public function deleteSpecificRevisions(int $maxRevisions): void
 
     protected function getRevisionModel(): string
     {
-        $revision = config('twill.namespace') . "\Models\Revisions\\" . class_basename($this) . "Revision";
-
-        if (@class_exists($revision)) {
-            return $revision;
-        }
-
-        return TwillCapsules::getCapsuleForModel(class_basename($this))->getRevisionModel();
+        return TwillCapsules::guessRelatedModelClass('Revision', $this);
     }
 }

src/Models/Behaviors/HasSlug.php

@@ -37,32 +37,17 @@
      *
      * @return object
      */
-    public function getSlugClass(): string
+    public function getSlugClass(): Model
     {
-        return new $this->getSlugModelClass();
+        return new ($this->getSlugModelClass());
     }
 
     /**
      * Returns the fully qualified slug class name for this model.
      */
     public function getSlugModelClass(): string
     {
-        // First load it from the base directory.
-        $slug = config('twill.namespace') . "\\Models\\Slugs\\" . $this->getSlugClassName();
-
-        if (@class_exists($slug)) {
-            return $slug;
-        }
-
-        // Alternatively try to get it from the same directory as the model resides in (nested directory models).
-        $slug = $this->getNamespace() . "\Slugs\\" . $this->getSlugClassName();
-
-        if (@class_exists($slug)) {
-            return $slug;
-        }
-
-        // Finally try to get it from capsules.
-        return TwillCapsules::getCapsuleForModel(class_basename($this))->getSlugModel();
+        return TwillCapsules::guessRelatedModelClass('Slug', $this);
     }
 
     protected function getSlugClassName(): string

src/Models/Behaviors/HasTranslation.php

@@ -15,18 +15,10 @@ trait HasTranslation
 
     /**
      * Returns the fully qualified translation class name for this model.
-     *
-     * @return string|null
      */
-    public function getTranslationModelNameDefault()
+    public function getTranslationModelNameDefault(): string
     {
-        $repository = config('twill.namespace') . "\Models\Translations\\" . class_basename($this) . 'Translation';
-
-        if (@class_exists($repository)) {
-            return $repository;
-        }
-
-        return TwillCapsules::getCapsuleForModel(class_basename($this))->getTranslationModel();
+        return TwillCapsules::guessRelatedModelClass('Translation', $this);
     }
 
     public function scopeWithActiveTranslations(Builder $query, ?string $locale = null): Builder

src/TwillCapsules.php

@@ -4,6 +4,7 @@
 
 use A17\Twill\Exceptions\NoCapsuleFoundException;
 use A17\Twill\Helpers\Capsule;
+use Illuminate\Database\Eloquent\Model;
 use Illuminate\Support\Collection;
 use Illuminate\Support\Str;
 use A17\Twill\Models\Contracts\TwillModelContract;
@@ -161,4 +162,29 @@
             ? app('autoloader')
             : require base_path('vendor/autoload.php');
     }
+
+    /** @return class-string<Model> */
+    public function guessRelatedModelClass(string $related, Model $model): string
+    {
+        $rc = new \ReflectionClass($model);
+        $namespace = $rc->getNamespaceName();
+        $capsule = $rc->getShortName();
+        $relatedClass = $capsule . $related;
+        foreach (
+            [
+            // First load it from the base directory.
+            config('twill.namespace') . "\\Models\\{$related}s\\" . $relatedClass,
+            // Alternatively try to get it from the same directory as the model resides
+            $rc->getName() . $related,
+            // Or in nested directory models.
+            $namespace . "\\{$related}s\\" . $relatedClass,
+            ] as $possibleClass
+        ) {
+            if (@class_exists($possibleClass)) {
+                return $possibleClass;
+            }
+        }
+
+        return call_user_func([$this->getCapsuleForModel($capsule), 'get' . $related . 'Model']);
+    }
 }