Migration: Moved reference method to trait

1 files changed, 36 insertions, 0 deletions

diff --git a/db/migrations/Reference.php b/db/migrations/Reference.php
new file mode 100644
index 00000000..4c35b59b
--- /dev/null
+++ b/db/migrations/Reference.php
@@ -0,0 +1,36 @@
+<?php
+
+namespace Engelsystem\Migrations;
+
+use Illuminate\Database\Schema\Blueprint;
+
+trait Reference
+{
+    /**
+     * @param Blueprint $table
+     * @param bool      $setPrimary
+     */
+    protected function referencesUser(Blueprint $table, $setPrimary = true)
+    {
+        $this->references($table, 'users', 'user_id', $setPrimary);
+    }
+
+    /**
+     * @param Blueprint $table
+     * @param string    $targetTable
+     * @param string    $fromColumn
+     * @param bool      $setPrimary
+     */
+    protected function references(Blueprint $table, $targetTable, $fromColumn, $setPrimary = false)
+    {
+        $table->unsignedInteger($fromColumn);
+
+        if ($setPrimary) {
+            $table->primary($fromColumn);
+        }
+
+        $table->foreign($fromColumn)
+            ->references('id')->on($targetTable)
+            ->onDelete('cascade');
+    }
+}