From 4ddb74e4e0596abc54d43d746cbcaf2025f8ebe9 Mon Sep 17 00:00:00 2001
From: Valk Richard Li <48872266+ValKmjolnir@users.noreply.github.com>
Date: Thu, 21 Nov 2019 12:21:33 +0800
Subject: [PATCH] Fixed a bug.

---
 balloon/abstract_syntax_tree.h |  1 +
 balloon/balloon_lexer.h        |  6 ++++++
 balloon/balloon_parse.h        | 10 +++++++++-
 balloon/main.cpp               |  3 +++
 4 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/balloon/abstract_syntax_tree.h b/balloon/abstract_syntax_tree.h
index 7f1a681..1eee1aa 100644
--- a/balloon/abstract_syntax_tree.h
+++ b/balloon/abstract_syntax_tree.h
@@ -71,6 +71,7 @@ class abstract_syntax_tree
 				case __number:std::cout<<": "<<number;break;
 				case __string:std::cout<<": "<<str;break;
 				case __id:
+				case __dynamic_id:
 				case __call_array:
 				case __call_hash:
 				case __call_function:std::cout<<": "<<name;break;
diff --git a/balloon/balloon_lexer.h b/balloon/balloon_lexer.h
index 7c1731a..9702705 100644
--- a/balloon/balloon_lexer.h
+++ b/balloon/balloon_lexer.h
@@ -92,8 +92,14 @@ class resource_file
 			totalsource.clear();
 			return;
 		}
+		void clear_lib_code()
+		{
+			libsource.clear();
+			return;
+		}
 		void add_lib()
 		{
+			libsource.clear();
 			std::string lib_name;
 			lib_name="lib/math.nas";
 			if(access("lib/math.nas",0))
diff --git a/balloon/balloon_parse.h b/balloon/balloon_parse.h
index c5dd2de..a0c7074 100644
--- a/balloon/balloon_parse.h
+++ b/balloon/balloon_parse.h
@@ -678,7 +678,15 @@ abstract_syntax_tree balloon_parse::func_generate()
 			temp.set_clear();
 			temp.set_line(this_token.line);
 			temp.set_type(this_token.type);
-			temp.set_name(this_token.str);
+			if(this_token.type==__dynamic_id)
+			{
+				std::string tempstr="";
+				for(int i=0;i<this_token.str.length()-3;++i)
+					tempstr+=this_token.str[i];
+				temp.set_name(tempstr);
+			}
+			else
+				temp.set_name(this_token.str);
 			para.add_child(temp);
 			if(this_token.type==__dynamic_id)
 			{
diff --git a/balloon/main.cpp b/balloon/main.cpp
index 1d06e83..360d376 100644
--- a/balloon/main.cpp
+++ b/balloon/main.cpp
@@ -40,6 +40,7 @@ int main()
 			break;
 		else if(command=="lexer")
 		{
+			prog.clear_lib_code(); // avoid print lib code
 			lex.scanner(prog.get_resource());
 			lex.print_token_list();
 		}
@@ -51,6 +52,7 @@ int main()
 		}
 		else if(command=="parser")
 		{
+			prog.clear_lib_code(); // avoid print lib code
 			lex.scanner(prog.get_resource());
 			lex.generate_detail_token();
 			if(!lex.get_error())
@@ -67,6 +69,7 @@ int main()
 		}
 		else if(command=="ast")
 		{
+			prog.clear_lib_code(); // avoid print lib code
 			lex.scanner(prog.get_resource());
 			lex.generate_detail_token();
 			if(!lex.get_error())