Shorthand for adding foreign key references to a table

Credit to http://github.com/maintainable/framework/commit/1b2f572c510ebfac71e9681ed9e453e7e86f2314

diff --git a/framework/Db/lib/Horde/Db/Adapter/Base/TableDefinition.php b/framework/Db/lib/Horde/Db/Adapter/Base/TableDefinition.php
index 4f26e45..ea2b918 100644 (file)
--- a/framework/Db/lib/Horde/Db/Adapter/Base/TableDefinition.php
+++ b/framework/Db/lib/Horde/Db/Adapter/Base/TableDefinition.php
@@ -149,6 +149,31 @@ class Horde_Db_Adapter_Base_TableDefinition implements ArrayAccess, IteratorAggr
     }
 
     /**
+     * Add one or several references to foreign keys
+     *
+     * This method returns <tt>self</tt>.
+     */
+    public function belongsTo($columns)
+    {
+        if (!is_array($columns)) { $columns = array($columns); }
+        foreach ($columns as $col) {
+            $this->column($col . '_id', 'integer');
+        }
+
+        return $this;
+    }
+
+    /**
+     * Alias for the belongsTo() method
+     *
+     * This method returns <tt>self</tt>.
+     */
+    public function references($columns)
+    {
+        return $this->belongsTo($columns);
+    }
+
+    /**
      * Use __call to provide shorthand column creation ($this->integer(), etc.)
      */
     public function __call($method, $arguments)