From b81e0c2b1c70badf4131cf8a587a06e1c31df1f5 Mon Sep 17 00:00:00 2001
From: "Rafael G. Martins" <rafael@rafaelmartins.eng.br>
Date: Sun, 19 Apr 2015 14:36:19 -0300
Subject: loader: added compiler-defined variables

---
 src/loader.c | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

(limited to 'src/loader.c')

diff --git a/src/loader.c b/src/loader.c
index 7eefa77..485fb0d 100644
--- a/src/loader.c
+++ b/src/loader.c
@@ -49,6 +49,42 @@ blogc_file_get_contents(const char *path, size_t *len, blogc_error_t **err)
 }
 
 
+char*
+blogc_get_filename(const char *f)
+{
+    if (f == NULL)
+        return NULL;
+
+    if (strlen(f) == 0)
+        return NULL;
+
+    // keep a pointer to original string
+    char *filename = b_strdup(f);
+    char *tmp = filename;
+
+    bool removed_dot = false;
+    for (int i = strlen(tmp); i >= 0 ; i--) {
+
+        // remove last extension
+        if (!removed_dot && tmp[i] == '.') {
+            tmp[i] = '\0';
+            removed_dot = true;
+            continue;
+        }
+
+        if (tmp[i] == '/' || tmp[i] == '\\') {
+            tmp += i + 1;
+            break;
+        }
+    }
+
+    char *final_filename = b_strdup(tmp);
+    free(filename);
+
+    return final_filename;
+}
+
+
 b_slist_t*
 blogc_template_parse_from_file(const char *f, blogc_error_t **err)
 {
@@ -74,6 +110,14 @@ blogc_source_parse_from_file(const char *f, blogc_error_t **err)
     if (s == NULL)
         return NULL;
     b_trie_t *rv = blogc_source_parse(s, len, err);
+
+    // set FILENAME variable
+    if (rv != NULL) {
+        char *filename = blogc_get_filename(f);
+        if (filename != NULL)
+            b_trie_insert(rv, "FILENAME", filename);
+    }
+
     free(s);
     return rv;
 }
-- 
cgit v1.2.3-18-g5258