Merge pull request #35 from lloricode/enhance/implementation

Add implementation

config/config.php

@@ -7,6 +7,11 @@
      */
     'table' => 'vouchers',
 
+    /*
+     * Model to use
+     */
+    'model' => BeyondCode\Vouchers\Models\Voucher::class,
+
     /*
      * Database pivot table name for vouchers and users relation
      */
@@ -48,4 +53,4 @@
      * The user model that belongs to vouchers.
      */
     'user_model' => \App\User::class,
-];
+];

src/Facades/Vouchers.php

@@ -5,7 +5,7 @@
 use Illuminate\Support\Facades\Facade;
 
 /**
- * @see \BeyondCode\Vouchers\VouchersClass
+ * @mixin \BeyondCode\Vouchers\Vouchers
  */
 class Vouchers extends Facade
 {

src/Traits/CanRedeemVouchers.php

@@ -2,7 +2,7 @@
 
 namespace BeyondCode\Vouchers\Traits;
 
-use Vouchers;
+use BeyondCode\Vouchers\Facades\Vouchers;
 use BeyondCode\Vouchers\Models\Voucher;
 use BeyondCode\Vouchers\Events\VoucherRedeemed;
 use BeyondCode\Vouchers\Exceptions\VoucherExpired;
@@ -57,4 +57,4 @@ public function vouchers()
     {
         return $this->belongsToMany(Voucher::class)->withPivot('redeemed_at');
     }
-}
+}

src/Traits/HasVouchers.php

@@ -14,7 +14,7 @@ trait HasVouchers
      */
     public function vouchers()
     {
-        return $this->morphMany(Voucher::class, 'model');
+        return $this->morphMany(config('vouchers.model', Voucher::class), 'model');
     }
 
     /**

src/Vouchers.php

@@ -11,10 +11,13 @@ class Vouchers
 {
     /** @var VoucherGenerator */
     private $generator;
+    /** @var \BeyondCode\Vouchers\Models\Voucher  */
+    private $voucherModel;
 
     public function __construct(VoucherGenerator $generator)
     {
         $this->generator = $generator;
+        $this->voucherModel = app(config('vouchers.model', Voucher::class));
     }
 
     /**
@@ -47,7 +50,7 @@ public function create(Model $model, int $amount = 1, array $data = [], $expires
         $vouchers = [];
 
         foreach ($this->generate($amount) as $voucherCode) {
-            $vouchers[] = Voucher::create([
+            $vouchers[] = $this->voucherModel->create([
                 'model_id' => $model->getKey(),
                 'model_type' => $model->getMorphClass(),
                 'code' => $voucherCode,
@@ -67,7 +70,7 @@ public function create(Model $model, int $amount = 1, array $data = [], $expires
      */
     public function check(string $code)
     {
-        $voucher = Voucher::whereCode($code)->first();
+        $voucher = $this->voucherModel->whereCode($code)->first();
 
         if (is_null($voucher)) {
             throw VoucherIsInvalid::withCode($code);
@@ -86,7 +89,7 @@ protected function getUniqueVoucher(): string
     {
         $voucher = $this->generator->generateUnique();
 
-        while (Voucher::whereCode($voucher)->count() > 0) {
+        while ($this->voucherModel->whereCode($voucher)->count() > 0) {
             $voucher = $this->generator->generateUnique();
         }