From 3794eeb31ff1df0c2c4314417fa42d8aab1b89f3 Mon Sep 17 00:00:00 2001
From: Wang Renxin <hellotony521@qq.com>
Date: Tue, 27 Oct 2015 11:52:48 +0800
Subject: [PATCH] +added an mb_set_import_handler function.

---
 core/my_basic.c | 30 ++++++++++++++++++++++--------
 1 file changed, 22 insertions(+), 8 deletions(-)

diff --git a/core/my_basic.c b/core/my_basic.c
index d8660a6..03728ed 100755
--- a/core/my_basic.c
+++ b/core/my_basic.c
@@ -539,6 +539,7 @@ typedef struct mb_interpreter_t {
 	mb_error_handler_t error_handler;
 	mb_print_func_t printer;
 	mb_input_func_t inputer;
+	mb_import_handler_t import_handler;
 	void* userdata;
 } mb_interpreter_t;
 
@@ -3464,14 +3465,16 @@ _data_e _get_symbol_type(mb_interpreter_t* s, char* sym, _raw_t* value) {
 					safe_free(buf);
 				}
 			} else {
-				_set_current_error(s, SE_PS_FILE_OPEN_FAILED, 0);
-				if(s->error_handler) {
-					(s->error_handler)(s, s->last_error, (char*)mb_get_error_desc(s->last_error),
-						s->last_error_func,
-						s->last_error_pos,
-						s->last_error_row,
-						s->last_error_col,
-						result);
+				if(!s->import_handler || s->import_handler(s, sym + 1) != MB_FUNC_OK) {
+					_set_current_error(s, SE_PS_FILE_OPEN_FAILED, 0);
+					if(s->error_handler) {
+						(s->error_handler)(s, s->last_error, (char*)mb_get_error_desc(s->last_error),
+							s->last_error_func,
+							s->last_error_pos,
+							s->last_error_row,
+							s->last_error_col,
+							result);
+					}
 				}
 			}
 			context->parsing_state = _PS_STRING;
@@ -7933,6 +7936,17 @@ int mb_set_inputer(struct mb_interpreter_t* s, mb_input_func_t p) {
 	return result;
 }
 
+int mb_set_import_handler(struct mb_interpreter_t* s, mb_import_handler_t h) {
+	/* Set an import handler to an interpreter instance */
+	int result = MB_FUNC_OK;
+
+	mb_assert(s);
+
+	s->import_handler = h;
+
+	return result;
+}
+
 int mb_gets(char* buf, int s) {
 	/* Safe stdin reader function */
 	int result = 0;