From d16ea7d8fce0229018e0ac98c1aa67be40bf17c2 Mon Sep 17 00:00:00 2001
From: zYne <zYne@625475ce-881a-0410-a577-b389adb331d8>
Date: Thu, 6 Sep 2007 20:52:36 +0000
Subject: [PATCH]

---
 lib/Doctrine/Template/I18n.php | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/lib/Doctrine/Template/I18n.php b/lib/Doctrine/Template/I18n.php
index d3c45fc06..932c2cc9e 100644
--- a/lib/Doctrine/Template/I18n.php
+++ b/lib/Doctrine/Template/I18n.php
@@ -31,19 +31,29 @@
  */
 class Doctrine_Template_I18n extends Doctrine_Template
 {
-    protected $_i18n;
-
     public function __construct(array $options)
     {
-        $this->_i18n = new Doctrine_I18n($options);
+        $this->_plugin = new Doctrine_I18n($options);
     }
     public function setUp()
     {
-        $this->_i18n->setOption('table', $this->_table);
-        $this->_i18n->buildDefinition($this->_table);
+        $this->_plugin->setOption('table', $this->_table);
+        $name = $this->_table->getComponentName();
+        $className = $this->_plugin->getOption('className');
+
+        if (strpos($className, '%CLASS%') !== false) {
+            $this->_plugin->setOption('className', str_replace('%CLASS%', $name, $className));
+            $className = $this->_plugin->getOption('className');
+        }
+
+        $this->_plugin->buildDefinition($this->_table);
+        
+        $id = $this->_table->getIdentifier();
+
+        $this->hasMany($className . ' as Translation', array('local' => $id, 'foreign' => $id));
     }
     public function getI18n()
     {
-        return $this->_i18n;
+        return $this->_plugin;
     }
 }