From 6d78665c9bf1b6d7b3445bd7ea09e7c0b2fc09bb Mon Sep 17 00:00:00 2001 From: "Rafael G. Martins" Date: Wed, 27 May 2015 19:42:53 -0300 Subject: loader: implemented tag filter --- src/loader.c | 44 +++++++++++++++++++++++++++++++++++--------- tests/check_loader.c | 47 +++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 82 insertions(+), 9 deletions(-) diff --git a/src/loader.c b/src/loader.c index 5a521e3..a48c9dc 100644 --- a/src/loader.c +++ b/src/loader.c @@ -99,9 +99,10 @@ blogc_source_parse_from_files(b_trie_t *conf, b_slist_t *l, blogc_error_t **err) { blogc_error_t *tmp_err = NULL; b_slist_t *rv = NULL; - bool first = true; unsigned int with_date = 0; + const char *filter_tag = b_trie_lookup(conf, "FILTER_TAG"); + for (b_slist_t *tmp = l; tmp != NULL; tmp = tmp->next) { char *f = tmp->data; b_trie_t *s = blogc_source_parse_from_file(f, &tmp_err); @@ -115,8 +116,39 @@ blogc_source_parse_from_files(b_trie_t *conf, b_slist_t *l, blogc_error_t **err) rv = NULL; break; } - if (b_trie_lookup(s, "DATE")) + if (filter_tag != NULL) { + const char *tags_str = b_trie_lookup(s, "TAGS"); + // if user wants to filter by tag and no tag is provided, skip it + if (tags_str == NULL) { + b_trie_free(s); + continue; + } + char **tags = b_str_split(tags_str, ',', 0); + bool found = false; + for (unsigned int i = 0; tags[i] != NULL; i++) + if (0 == strcmp(b_str_strip(tags[i]), filter_tag)) + found = true; + b_strv_free(tags); + if (!found) { + b_trie_free(s); + continue; + } + } + if (b_trie_lookup(s, "DATE") != NULL) with_date++; + rv = b_slist_append(rv, s); + } + + if (with_date > 0 && with_date < b_slist_length(rv)) + // fatal error, maybe? + blogc_fprintf(stderr, + "blogc: warning: 'DATE' variable provided for at least one source " + "file, but not for all source files. This means that you may get " + "wrong values for 'DATE_FIRST' and 'DATE_LAST' variables.\n"); + + bool first = true; + for (b_slist_t *tmp = rv; tmp != NULL; tmp = tmp->next) { + b_trie_t *s = tmp->data; if (first) { const char *val = b_trie_lookup(s, "DATE"); if (val != NULL) @@ -134,13 +166,7 @@ blogc_source_parse_from_files(b_trie_t *conf, b_slist_t *l, blogc_error_t **err) if (val != NULL) b_trie_insert(conf, "FILENAME_LAST", b_strdup(val)); } - rv = b_slist_append(rv, s); } - if (with_date > 0 && with_date < b_slist_length(l)) - // fatal error, maybe? - blogc_fprintf(stderr, - "blogc: warning: 'DATE' variable provided for at least one source " - "file, but not for all source files. This means that you may get " - "wrong values for 'DATE_FIRST' and 'DATE_LAST' variables.\n"); + return rv; } diff --git a/tests/check_loader.c b/tests/check_loader.c index 506bf4f..f570297 100644 --- a/tests/check_loader.c +++ b/tests/check_loader.c @@ -176,6 +176,52 @@ test_source_parse_from_files(void **state) } +static void +test_source_parse_from_files_filter_by_tag(void **state) +{ + will_return(__wrap_blogc_file_get_contents, "bola1.txt"); + will_return(__wrap_blogc_file_get_contents, b_strdup( + "ASD: 123\n" + "DATE: 2001-02-03 04:05:06\n" + "TAGS: chunda\n" + "--------\n" + "bola")); + will_return(__wrap_blogc_file_get_contents, "bola2.txt"); + will_return(__wrap_blogc_file_get_contents, b_strdup( + "ASD: 456\n" + "DATE: 2002-02-03 04:05:06\n" + "TAGS: bola, chunda\n" + "--------\n" + "bola")); + will_return(__wrap_blogc_file_get_contents, "bola3.txt"); + will_return(__wrap_blogc_file_get_contents, b_strdup( + "ASD: 789\n" + "DATE: 2003-02-03 04:05:06\n" + "--------\n" + "bola")); + blogc_error_t *err = NULL; + b_slist_t *s = NULL; + s = b_slist_append(s, b_strdup("bola1.txt")); + s = b_slist_append(s, b_strdup("bola2.txt")); + s = b_slist_append(s, b_strdup("bola3.txt")); + b_trie_t *c = b_trie_new(free); + b_trie_insert(c, "FILTER_TAG", b_strdup("chunda")); + b_slist_t *t = blogc_source_parse_from_files(c, s, &err); + assert_null(err); + assert_non_null(t); + assert_int_equal(b_slist_length(t), 2); // it is enough, no need to look at the items + assert_int_equal(b_trie_size(c), 5); + assert_string_equal(b_trie_lookup(c, "FILENAME_FIRST"), "bola1"); + assert_string_equal(b_trie_lookup(c, "FILENAME_LAST"), "bola2"); + assert_string_equal(b_trie_lookup(c, "DATE_FIRST"), "2001-02-03 04:05:06"); + assert_string_equal(b_trie_lookup(c, "DATE_LAST"), "2002-02-03 04:05:06"); + assert_string_equal(b_trie_lookup(c, "FILTER_TAG"), "chunda"); + b_trie_free(c); + b_slist_free_full(s, free); + b_slist_free_full(t, (b_free_func_t) b_trie_free); +} + + static void test_source_parse_from_files_without_all_dates(void **state) { @@ -248,6 +294,7 @@ main(void) unit_test(test_source_parse_from_file), unit_test(test_source_parse_from_file_null), unit_test(test_source_parse_from_files), + unit_test(test_source_parse_from_files_filter_by_tag), unit_test(test_source_parse_from_files_without_all_dates), unit_test(test_source_parse_from_files_null), }; -- cgit v1.2.3-18-g5258