From 2b43b9dfef5327d5b501957ed2fc9413c89404e8 Mon Sep 17 00:00:00 2001
From: Vladislav Stoitsov <vlados.01@gmail.com>
Date: Sat, 4 May 2019 11:55:18 +0300
Subject: [PATCH] Added posibility to have your own stubs

---
 README.md                          | 6 ++++++
 src/Scaffold/ControllerCreator.php | 3 ++-
 src/Scaffold/MigrationCreator.php  | 2 +-
 src/Scaffold/ModelCreator.php      | 2 +-
 4 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index faf6ebb..43371c2 100644
--- a/README.md
+++ b/README.md
@@ -24,6 +24,12 @@ $ php artisan admin:import helpers
 
 See [wiki](http://laravel-admin.org/docs/#/en/extension-helpers?id=helpers)
 
+## Scaffold templates (stubs)   
+Copy `src/Scaffold/stubs` to your project `app/stubs` and add to your `.env` file:
+```
+ADMIN_STUBS_DIR=/app/stubs/
+```
+
 ## Donate
 
 > Help keeping the project development going, by donating a little. Thanks in advance.
diff --git a/src/Scaffold/ControllerCreator.php b/src/Scaffold/ControllerCreator.php
index 038f7f7..0408a32 100644
--- a/src/Scaffold/ControllerCreator.php
+++ b/src/Scaffold/ControllerCreator.php
@@ -123,6 +123,7 @@ public function getPath($name)
      */
     public function getStub()
     {
-        return __DIR__.'/stubs/controller.stub';
+        return (getenv("ADMIN_STUBS_DIR") || __DIR__.'/stubs/'). 'controller.stub';
+
     }
 }
diff --git a/src/Scaffold/MigrationCreator.php b/src/Scaffold/MigrationCreator.php
index 30a50db..36383f6 100644
--- a/src/Scaffold/MigrationCreator.php
+++ b/src/Scaffold/MigrationCreator.php
@@ -27,7 +27,7 @@ public function create($name, $path, $table = null, $create = true)
 
         $path = $this->getPath($name, $path);
 
-        $stub = $this->files->get(__DIR__.'/stubs/create.stub');
+        $stub = $this->files->get((getenv("ADMIN_STUBS_DIR") || __DIR__.'/stubs/'). 'create.stub');
 
         $this->files->put($path, $this->populateStub($name, $stub, $table));
 
diff --git a/src/Scaffold/ModelCreator.php b/src/Scaffold/ModelCreator.php
index 7f4dfc2..35cd1d0 100644
--- a/src/Scaffold/ModelCreator.php
+++ b/src/Scaffold/ModelCreator.php
@@ -233,6 +233,6 @@ public function replaceSpace($stub)
      */
     public function getStub()
     {
-        return __DIR__.'/stubs/model.stub';
+        return (getenv("ADMIN_STUBS_DIR") || __DIR__.'/stubs/'). 'model.stub';
     }
 }