Added an overridden definition of getModel() in the record & collection templates to return the exact type of model for the collection and record

Author: rotimi

src/OrmClassesGenerator.php

@@ -324,6 +324,9 @@ protected function generateCollectionClassFile(string $tableName, string $collec
             '{{{NAME_SPACE}}}'                              => ($this->config['namespace'] === null) ? '' : "namespace {$this->config['namespace']}\\{$collectionOrModelNamePrefix};",
             '{{{MODEL_OR_COLLECTION_CLASS_NAME_PREFIX}}}'   => $collectionOrModelNamePrefix,
             '{{{COLLECTION_EXTENDED}}}'                     => $this->config['collection_class_to_extend'],
+            '{{{MODEL_EXTENDED}}}'                          => $this->config['model_class_to_extend'],
+            '{{{RECORD_CLASS_NAME_PREFIX}}}'                => $recordNamePrefix,
+            '{{{TABLE_NAME}}}'                              => $tableName,
         ];
 
         return strtr($this->loadedCollectionTemplateFile, $translations);
@@ -355,7 +358,9 @@ protected function generateModelClassFile(string $tableName, string $collectionO
         $translations = [
             '{{{NAME_SPACE}}}'                              => ($this->config['namespace'] === null) ? '' : "namespace {$this->config['namespace']}\\{$collectionOrModelNamePrefix};",
             '{{{MODEL_OR_COLLECTION_CLASS_NAME_PREFIX}}}'   => $collectionOrModelNamePrefix,
+            '{{{COLLECTION_EXTENDED}}}'                     => $this->config['collection_class_to_extend'],
             '{{{MODEL_EXTENDED}}}'                          => $this->config['model_class_to_extend'],
+            '{{{RECORD_EXTENDED}}}'                         => $this->config['record_class_to_extend'],
             '{{{RECORD_CLASS_NAME_PREFIX}}}'                => $recordNamePrefix,
             '{{{CREATED_TIMESTAMP_COLUMN_NAME}}}'           => ($this->config['created_timestamp_column_name'] === null || !$createdColExists) ? 'null' : "'{$this->config['created_timestamp_column_name']}'",
             '{{{UPDATED_TIMESTAMP_COLUMN_NAME}}}'           => ($this->config['updated_timestamp_column_name'] === null || !$updatedColExists) ? 'null' : "'{$this->config['updated_timestamp_column_name']}'",
@@ -373,8 +378,12 @@ protected function generateRecordClassFile(string $tableName, string $collection
         $translations = [
             '{{{NAME_SPACE}}}'                              => ($this->config['namespace'] === null) ? '' : "namespace {$this->config['namespace']}\\{$collectionOrModelNamePrefix};",
             '{{{RECORD_CLASS_NAME_PREFIX}}}'                => $recordNamePrefix,
+            '{{{COLLECTION_EXTENDED}}}'                     => $this->config['collection_class_to_extend'],
+            '{{{MODEL_EXTENDED}}}'                          => $this->config['model_class_to_extend'],
             '{{{RECORD_EXTENDED}}}'                         => $this->config['record_class_to_extend'],
             '{{{DB_COLS_AS_PHP_CLASS_PROPERTIES}}}'         => $phpDocDbColNamesAsPhpClassProperties,
+            '{{{MODEL_OR_COLLECTION_CLASS_NAME_PREFIX}}}'   => $collectionOrModelNamePrefix,
+            '{{{TABLE_NAME}}}'                              => $tableName,
         ];
 
         return strtr($this->loadedRecordTemplateFile, $translations);

templates/TypeRecord.php.tpl

@@ -9,4 +9,8 @@ declare(strict_types=1);
 class {{{RECORD_CLASS_NAME_PREFIX}}}Record extends {{{RECORD_EXTENDED}}} {
 
     //put your code here
+    public function getModel(): {{{MODEL_OR_COLLECTION_CLASS_NAME_PREFIX}}}Model {
+        
+        return parent::getModel();
+    }
 }

templates/TypesCollection.php.tpl

@@ -6,4 +6,8 @@ declare(strict_types=1);
 class {{{MODEL_OR_COLLECTION_CLASS_NAME_PREFIX}}}Collection extends {{{COLLECTION_EXTENDED}}} {
 
     //put your code here
+    public function getModel(): {{{MODEL_OR_COLLECTION_CLASS_NAME_PREFIX}}}Model {
+        
+        return parent::getModel();
+    }
 }