Colour the first line of a heredoc, and colour output only plans

dflook · Dec 11, 2023 · a9ad47b · a9ad47b
1 parent 949bc76
commit a9ad47b
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/image/src/github_pr_comment/__main__.py b/image/src/github_pr_comment/__main__.py
@@ -511,7 +511,7 @@ def main() -> int:
         changes = os.environ.get('TF_CHANGES') == 'true'
 
         highlighting = ''
-        if 'Plan:' in body:
+        if os.environ.get('TF_CHANGES', 'true') == 'true':
             highlighting = 'diff' if format_type.startswith('diff') else 'hcl'
 
         comment = update_comment(

diff --git a/image/src/github_pr_comment/plan_formatting.py b/image/src/github_pr_comment/plan_formatting.py
@@ -15,8 +15,6 @@ def format_diff(plan_text: str) -> str:
             continue
         elif line.endswith('EOT'):
             heredoc = True
-            lines.append(line)
-            continue
 
         replaced = (re.sub(
             r'^(?P<leading_space>\s+)(?P<operation>[+-/~]+)(?P<trailing>.*)',

diff --git a/tests/test_diff_formatting.py b/tests/test_diff_formatting.py
@@ -236,7 +236,7 @@ def test_heredoc():
 
   # random_string.add will be created
 +   resource "random_string" "add" {
-      + id          = <<EOT
++       id          = <<EOT
   + create
   - destroy
 -/+ destroy and then create replacement