From db0eae9d5df594917b37b083c9039e4f6d806840 Mon Sep 17 00:00:00 2001
From: oldrind <1401019000@qq.com>
Date: Thu, 28 Jan 2016 17:49:15 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E5=8D=87template=E7=B1=BB=E6=B5=8B?=
 =?UTF-8?q?=E8=AF=95=E8=A6=86=E7=9B=96=E7=8E=87=EF=BC=8C=E4=BF=AE=E6=AD=A3?=
 =?UTF-8?q?include=E6=A0=87=E7=AD=BE=E4=BC=A0=E5=8F=98=E9=87=8F=E7=9A=84?=
 =?UTF-8?q?=E9=94=99=E8=AF=AF=20=EF=BC=8C=E4=BB=A5$=E5=BC=80=E5=A4=B4?=
 =?UTF-8?q?=E8=A1=A8=E7=A4=BA=E4=BC=A0=E7=9A=84=E6=98=AF=E5=8F=98=E9=87=8F?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 library/think/Template.php                    |  19 +-
 library/think/template/TagLib.php             |  14 +-
 tests/thinkphp/library/think/extend.html      |  11 ++
 tests/thinkphp/library/think/include.html     |   1 +
 tests/thinkphp/library/think/layout.html      |   2 +
 tests/thinkphp/library/think/templateTest.php | 165 +++++++++---------
 6 files changed, 119 insertions(+), 93 deletions(-)
 create mode 100644 tests/thinkphp/library/think/extend.html
 create mode 100644 tests/thinkphp/library/think/include.html
 create mode 100644 tests/thinkphp/library/think/layout.html

diff --git a/library/think/Template.php b/library/think/Template.php
index ffc263b9..82f5c83b 100644
--- a/library/think/Template.php
+++ b/library/think/Template.php
@@ -383,6 +383,9 @@ class Template
                     $parseStr = $this->parseTemplateName($file);
                     // 替换变量
                     foreach ($array as $k => $v) {
+                        if (0 == strpos($v, '$')) {
+                            $v = ltrim($this->config['tpl_begin'], '\\') . $v . ltrim($this->config['tpl_end'], '\\');
+                        }
                         $parseStr = str_replace('[' . $k . ']', $v, $parseStr);
                     }
                     // 再次对包含文件进行模板分析
@@ -455,10 +458,12 @@ class Template
         $regex = $this->getRegex($restore ? 'restoreliteral' : 'literal');
         if (preg_match_all($regex, $content, $matches, PREG_SET_ORDER)) {
             if (!$restore) {
+                $count = count($this->literal);
                 // 替换literal标签
                 foreach ($matches as $i => $match) {
                     $this->literal[] = substr($match[0], strlen($match[1]), -strlen($match[2]));
-                    $content         = str_replace($match[0], "<!--###literal{$i}###-->", $content);
+                    $content         = str_replace($match[0], "<!--###literal{$count}###-->", $content);
+                    $count++;
                 }
             } else {
                 // 还原literal标签
@@ -821,14 +826,14 @@ class Template
                     if (isset($vars[2])) {
                         $parseStr = '$_COOKIE[\'' . $vars[1] . '\'][\'' . $vars[2] . '\']';
                     } else {
-                        $parseStr = '\\think\\cookie::get(\'' . $vars[1] . '\')';
+                        $parseStr = '\\think\\Cookie::get(\'' . $vars[1] . '\')';
                     }
                     break;
                 case 'SESSION':
                     if (isset($vars[2])) {
                         $parseStr = '$_SESSION[\'' . $vars[1] . '\'][\'' . $vars[2] . '\']';
                     } else {
-                        $parseStr = '\\think\\session::get(\'' . $vars[1] . '\')';
+                        $parseStr = '\\think\\Session::get(\'' . $vars[1] . '\')';
                     }
                     break;
                 case 'ENV':
@@ -841,13 +846,13 @@ class Template
                     $parseStr = strtoupper($vars[1]);
                     break;
                 case 'LANG':
-                    $parseStr = '\\think\\lang::get(\'' . $vars[1] . '\')';
+                    $parseStr = '\\think\\Lang::get(\'' . $vars[1] . '\')';
                     break;
                 case 'CONFIG':
                     if (isset($vars[2])) {
                         $vars[1] .= '.' . $vars[2];
                     }
-                    $parseStr = '\\think\\config::get(\'' . $vars[1] . '\')';
+                    $parseStr = '\\think\\Config::get(\'' . $vars[1] . '\')';
                     break;
                 default:
                     break;
@@ -862,10 +867,10 @@ class Template
                         $parseStr = 'THINK_VERSION';
                         break;
                     case 'LDELIM':
-                        $parseStr = $this->config['tpl_begin'];
+                        $parseStr = '\'' . ltrim($this->config['tpl_begin'], '\\') . '\'';
                         break;
                     case 'RDELIM':
-                        $parseStr = $this->config['tpl_end'];
+                        $parseStr = '\'' . ltrim($this->config['tpl_end'], '\\') . '\'';
                         break;
                     default:
                         if (defined($vars[0])) {
diff --git a/library/think/template/TagLib.php b/library/think/template/TagLib.php
index be02d6a0..ad410fb4 100644
--- a/library/think/template/TagLib.php
+++ b/library/think/template/TagLib.php
@@ -136,12 +136,12 @@ class TagLib
                 // 标签替换 从后向前
                 foreach ($nodes as $pos => $node) {
                     // 对应的标签名
-                    $name = $tags[1][$node['name']];
+                    $name = $node['name'];
                     // 解析标签属性
                     $attrs  = $this->parseAttr($node['begin'][0], $name);
                     $method = '_' . $name;
                     // 读取标签库中对应的标签内容 replace[0]用来替换标签头，replace[1]用来替换标签尾
-                    $replace = explode($break, $this->$method($attrs, $break, $node['name']));
+                    $replace = explode($break, $this->$method($attrs, $break));
                     if (count($replace) > 1) {
                         while ($beginArray) {
                             $begin = end($beginArray);
@@ -171,13 +171,13 @@ class TagLib
         // 自闭合标签
         if (!empty($tags[0])) {
             $regex   = $this->getRegex(array_keys($tags[0]), 0);
-            $self    = &$this;
-            $content = preg_replace_callback($regex, function ($matches) use (&$tags, &$self) {
-                $name = $tags[0][$matches[1]];
+            $content = preg_replace_callback($regex, function ($matches) use (&$tags) {
+                // 对应的标签名
+                $name = $matches[1];
                 // 解析标签属性
-                $attrs = $self->parseAttr($matches[0], $name);
+                $attrs = $this->parseAttr($matches[0], $name);
                 $method = '_' . $name;
-                return $self->$method($attrs, '', $matches[1]);
+                return $this->$method($attrs, '');
             }, $content);
         }
         return;
diff --git a/tests/thinkphp/library/think/extend.html b/tests/thinkphp/library/think/extend.html
new file mode 100644
index 00000000..194fbbf9
--- /dev/null
+++ b/tests/thinkphp/library/think/extend.html
@@ -0,0 +1,11 @@
+{layout name="layout" replace="[__REPLACE__]" /}
+<div>
+    {include file="include" name="$info.name" value="$info.value" /}
+{block name="main"}
+        main
+{/block}
+{literal}
+    {$name}
+{/literal}
+    <?php echo 'php code'; ?>
+</div>
\ No newline at end of file
diff --git a/tests/thinkphp/library/think/include.html b/tests/thinkphp/library/think/include.html
new file mode 100644
index 00000000..ac622e75
--- /dev/null
+++ b/tests/thinkphp/library/think/include.html
@@ -0,0 +1 @@
+<input name="[name]" value="[value]">
\ No newline at end of file
diff --git a/tests/thinkphp/library/think/layout.html b/tests/thinkphp/library/think/layout.html
new file mode 100644
index 00000000..a5b75729
--- /dev/null
+++ b/tests/thinkphp/library/think/layout.html
@@ -0,0 +1,2 @@
+<nav>[__REPLACE__]
+</nav>
\ No newline at end of file
diff --git a/tests/thinkphp/library/think/templateTest.php b/tests/thinkphp/library/think/templateTest.php
index 86240afc..53b2b629 100644
--- a/tests/thinkphp/library/think/templateTest.php
+++ b/tests/thinkphp/library/think/templateTest.php
@@ -32,7 +32,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {\$name.a??'test'}
@@ -42,7 +42,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {\$name.a?='test'}
@@ -52,7 +52,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {\$name.a?:'test'}
@@ -62,7 +62,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {\$name.a?\$name.b:'no'}
@@ -72,7 +72,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {\$name.a==\$name.b?='test'}
@@ -82,7 +82,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {\$name.a==\$name.b?'a':'b'}
@@ -92,7 +92,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {\$name.a|default='test'==\$name.b?'a':'b'}
@@ -102,7 +102,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {:ltrim(rtrim(\$name.a))}
@@ -112,7 +112,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {~echo(trim(\$name.a))}
@@ -122,7 +122,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {+\$name.a}
@@ -132,7 +132,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 {/*\$name*/}
@@ -140,7 +140,7 @@ EOF;
         $data = '';
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
     }
 
@@ -159,7 +159,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 <#\$info.a==\$info.b?='test'#>
@@ -169,7 +169,7 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
 
         $content = <<<EOF
 <#\$info.a|default='test'?'yes':'no'#>
@@ -179,86 +179,93 @@ EOF;
 EOF;
 
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
     }
 
     public function testTag()
     {
-        $template = new Template();
+        $config['tpl_path'] = dirname(__FILE__) . '/';
+        $config['tpl_suffix'] = '.html';
+        $template = new Template($config);
 
         $content = <<<EOF
-{if \$var.a==\$var.b}
-one
-{elseif !empty(\$var.a) /}
-two
-{else /}
-default
-{/if}
+{extend name="extend" /}
+{block name="main"}
+    {include file="include" name="\$user.name" value="\$user.account" /}
+    {\$message}{literal}{\$message}{/literal}
+{/block}
 EOF;
         $data = <<<EOF
-<?php if(\$var['a']==\$var['b']): ?>
-one
-<?php elseif(!empty(\$var['a'])): ?>
-two
-<?php else: ?>
-default
-<?php endif; ?>
+<nav>
+<div>
+    <input name="<?php echo \$info['name']; ?>" value="<?php echo \$info['value']; ?>">
+
+    <input name="<?php echo \$user['name']; ?>" value="<?php echo \$user['account']; ?>">
+    <?php echo \$message; ?>{\$message}
+
+
+    {\$name}
+
+    <?php echo 'php code'; ?>
+</div>
+</nav>
 EOF;
         $template->parse($content);
-        $this->assertEquals($content, $data);
+        $this->assertEquals($data, $content);
+    }
+
+    public function testThinkVar()
+    {
+        $config['tpl_begin'] = '{';
+        $config['tpl_end'] = '}';
+        $template = new Template($config);
+
+        $_SERVER['SERVER_NAME'] = 'server_name';
+        $_GET['action'] = 'action';
+        $_POST['action'] = 'action';
+        \think\Cookie::set('action', ['name' => 'name']);
+        \think\Session::set('action', ['name' => 'name']);
+        define('SITE_NAME', 'site_name');
 
         $content = <<<EOF
-{switch \$var}
-{case \$a /}
-a
-{/case}
-{case b}
-b
-{/case}
-{default /}
-default
-{/switch}
+{\$Think.SERVER.SERVER_NAME}<br/>
+{\$Think.GET.action}<br/>
+{\$Think.POST.action}<br/>
+{\$Think.COOKIE.action}<br/>
+{\$Think.COOKIE.action.name}<br/>
+{\$Think.SESSION.action}<br/>
+{\$Think.SESSION.action.name}<br/>
+{\$Think.ENV.OS}<br/>
+{\$Think.REQUEST.action}<br/>
+{\$Think.CONST.SITE_NAME}<br/>
+{\$Think.LANG.action}<br/>
+{\$Think.CONFIG.action.name}<br/>
+{\$Think.NOW}<br/>
+{\$Think.VERSION}<br/>
+{\$Think.LDELIM}<br/>
+{\$Think.RDELIM}<br/>
+{\$Think.SITE_NAME}
 EOF;
         $data = <<<EOF
-<?php switch(\$var): ?>
-<?php case \$a: ?>
-a
-<?php break; ?>
-<?php case "b": ?>
-b
-<?php break; ?>
-<?php default: ?>
-default
-<?php endswitch; ?>
+<?php echo \$_SERVER['SERVER_NAME']; ?><br/>
+<?php echo \$_GET['action']; ?><br/>
+<?php echo \$_POST['action']; ?><br/>
+<?php echo \\think\\Cookie::get('action'); ?><br/>
+<?php echo \$_COOKIE['action']['name']; ?><br/>
+<?php echo \\think\\Session::get('action'); ?><br/>
+<?php echo \$_SESSION['action']['name']; ?><br/>
+<?php echo \$_ENV['OS']; ?><br/>
+<?php echo \$_REQUEST['action']; ?><br/>
+<?php echo SITE_NAME; ?><br/>
+<?php echo \\think\\Lang::get('action'); ?><br/>
+<?php echo \\think\\Config::get('action.name'); ?><br/>
+<?php echo date('Y-m-d g:i a',time()); ?><br/>
+<?php echo THINK_VERSION; ?><br/>
+<?php echo '{'; ?><br/>
+<?php echo '}'; ?><br/>
+<?php echo SITE_NAME; ?>
 EOF;
         $template->parse($content);
-        $this->assertEquals($content, $data);
-
-        $content = <<<EOF
-{foreach \$list as \$key=>\$val}
-
-{/foreach}
-EOF;
-        $data = <<<EOF
-<?php foreach(\$list as \$key=>\$val): ?>
-
-<?php endforeach; ?>
-EOF;
-        $template->parse($content);
-        $this->assertEquals($content, $data);
-
-        $content = <<<EOF
-{foreach name="list" id="val" key="key"}
-
-{/foreach}
-EOF;
-        $data = <<<EOF
-<?php if(is_array(\$list)): foreach(\$list as \$key=>\$val): ?>
-
-<?php endforeach; endif; ?>
-EOF;
-        $template->parse($content);
-        $this->assertEquals($content, $data);
-
+        $this->assertEquals($data, $content);
     }
 }