Implement get_github_release_notes_auto function that bases release note data off test contexts

3 files changed, 53 insertions, 4 deletions

diff --git a/src/stasis_main.c b/src/stasis_main.c
index 13576a1..2abcbc4 100644
--- a/src/stasis_main.c
+++ b/src/stasis_main.c
@@ -262,7 +262,8 @@ int main(int argc, char *argv[]) {
 
     // Expose function(s) to the template engine
     // Prototypes can be found in template_func_proto.h
-    tpl_register_func("get_github_release_notes", &get_github_release_notes_tplfunc_entrypoint, 3);
+    tpl_register_func("get_github_release_notes", &get_github_release_notes_tplfunc_entrypoint, 3, NULL);
+    tpl_register_func("get_github_release_notes_auto", &get_github_release_notes_auto_tplfunc_entrypoint, 1, &ctx);
 
     // Set up PREFIX/etc directory information
     // The user may manipulate the base directory path with STASIS_SYSCONFDIR
diff --git a/src/template.c b/src/template.c
index 9fce8a2..8e1357c 100644
--- a/src/template.c
+++ b/src/template.c
@@ -25,11 +25,12 @@ extern void tpl_reset() {
     tpl_pool_func_used = 0;
 }
 
-void tpl_register_func(char *key, void *tplfunc_ptr, int argc) {
+void tpl_register_func(char *key, void *tplfunc_ptr, int argc, void *data_in) {
     struct tplfunc_frame *frame = calloc(1, sizeof(*frame));
     frame->key = strdup(key);
     frame->argc = argc;
     frame->func = tplfunc_ptr;
+    frame->data_in = data_in;
 
     tpl_pool_func[tpl_pool_func_used] = frame;
     tpl_pool_func_used++;
diff --git a/src/template_func_proto.c b/src/template_func_proto.c
index 459a3a6..8618a2e 100644
--- a/src/template_func_proto.c
+++ b/src/template_func_proto.c
@@ -1,8 +1,8 @@
 #include "template_func_proto.h"
 
-int get_github_release_notes_tplfunc_entrypoint(void *frame, void *ptr) {
+int get_github_release_notes_tplfunc_entrypoint(void *frame, void *data_out) {
     int result;
-    char **output = (char **) ptr;
+    char **output = (char **) data_out;
     struct tplfunc_frame *f = (struct tplfunc_frame *) frame;
     char *api_token = getenv("STASIS_GITHUB_TOKEN");
     if (!api_token) {
@@ -16,3 +16,50 @@ int get_github_release_notes_tplfunc_entrypoint(void *frame, void *ptr) {
     return result;
 }
 
+int get_github_release_notes_auto_tplfunc_entrypoint(void *frame, void *data_out) {
+    int result = 0;
+    char **output = (char **) data_out;
+    struct tplfunc_frame *f = (struct tplfunc_frame *) frame;
+    char *api_token = getenv("STASIS_GITHUB_TOKEN");
+    if (!api_token) {
+        api_token = getenv("GITHUB_TOKEN");
+    }
+
+    const struct Delivery *ctx = (struct Delivery *) f->data_in;
+    struct StrList *notes_list = strlist_init();
+    for (size_t i = 0; i < sizeof(ctx->tests) / sizeof(*ctx->tests); i++) {
+        // Get test context
+        const struct Test *test = &ctx->tests[i];
+        if (test->name && test->version && test->repository) {
+            char *repository = strdup(test->repository);
+            char *match = strstr(repository, "spacetelescope/");
+            // Cull repository URL
+            if (match) {
+                replace_text(repository, "https://github.com/", "", 0);
+                if (endswith(repository, ".git")) {
+                    replace_text(repository, ".git", "", 0);
+                }
+                // Record release notes for version relative to HEAD
+                // Using HEAD, GitHub returns the previous tag
+                char *note = NULL;
+                char h1_title[NAME_MAX] = {0};
+                sprintf(h1_title, "# %s", test->name);
+                strlist_append(&notes_list, h1_title);
+                result += get_github_release_notes(api_token ? api_token : "anonymous",
+                                                   repository,
+                                                   test->version,
+                                                   "HEAD",
+                                                   &note);
+                if (note) {
+                    strlist_append(&notes_list, note);
+                    guard_free(note);
+                }
+            }
+        }
+    }
+    // Return all notes as a single string
+    if (strlist_count(notes_list)) {
+        *output = join(notes_list->data, "\n\n");
+    }
+    return result;
+}