0.5 support downstream pipelines

.gitlab-ci.yml

@@ -4,7 +4,7 @@ stages:
   - deploy
 
 variables:
-  PROJECT_VERSION: "0.4"
+  PROJECT_VERSION: "0.5"
 
 build docker:
   stage: build

HISTORY.md

@@ -1,3 +1,6 @@
+# 0.5
+- support downstream pipelines
+
 # 0.4
 - global command "pipeline-mermaid"
 - custom message ability

pipeline_mermaid/gitlab_helper.py

@@ -33,8 +33,8 @@ def __show_pipeline_in_mr(self, merge_request, pipeline_id, type, message):
             return
         text = ''
         if message:
-            text += message
-        text += '\n' + self.pipeline_as_mermaid(pipeline_id, type)
+            text += message + '\n'
+        text += self.pipeline_as_mermaid(pipeline_id, type)
         merge_request.notes.create({"body": text})
 
     def __get_mr(self):
@@ -52,11 +52,20 @@ def __get_mr_by_branch(self, branch_name):
     def pipeline_as_mermaid(self, pipeline_id, type):
         project = self.get_project()
         pipeline = project.pipelines.get(id=pipeline_id)
-        jobs = pipeline.jobs.list()
+        jobs = get_jobs_with_child(pipeline)
         generator = get_mermaid_generator(jobs, type)
         return generator.to_mermaid()
 
 
+def get_jobs_with_child(pl):
+    jobs = pl.jobs.list()
+    for bridge in pl.bridges.list():
+        if bridge.downstream_pipeline:
+            child_pl = pl.manager.get(bridge.downstream_pipeline['id'])
+            jobs += get_jobs_with_child(child_pl)
+    return jobs
+
+
 def get_mermaid_generator(jobs, type):
     if type == 'gantt':
         return GanttGenerator(jobs)
@@ -67,7 +76,3 @@ def get_mermaid_generator(jobs, type):
     else:
         raise Exception("Unknown type of diagram")
 
-
-if __name__ == "__main__":
-    gl = GitlabHelper()
-    result = eval("GitlabHelper." + sys.argv[1])(gl, *sys.argv[2:])