From efe494e10b6c38836854f375a2382006a40a2b1f Mon Sep 17 00:00:00 2001
From: Dylan Hardison <dylan@mozilla.com>
Date: Mon, 4 Jan 2016 16:11:38 -0500
Subject: Bug 1233878 - tracking flags don't show up in the view of the bug
 right after filing

---
 extensions/BugModal/Extension.pm | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'extensions/BugModal/Extension.pm')

diff --git a/extensions/BugModal/Extension.pm b/extensions/BugModal/Extension.pm
index e929f3fcb..b1356bc34 100644
--- a/extensions/BugModal/Extension.pm
+++ b/extensions/BugModal/Extension.pm
@@ -165,9 +165,17 @@ sub template_before_process {
         return;
     }
 
-    return unless $file =~ m#^bug/show-([^\.]+)\.html\.tmpl$#;
-    my $format = $1;
-    return unless _alternative_show_bug_format() eq $format;
+    if ($file =~ m#^bug/show-([^\.]+)\.html\.tmpl$#) {
+        my $format = $1;
+        return unless _alternative_show_bug_format() eq $format;
+    }
+    elsif ($file ne 'bug_modal/edit.html.tmpl') {
+        return;
+    }
+
+    if ($vars->{bug} && !$vars->{bugs}) {
+        $vars->{bugs} = [$vars->{bug}];
+    }
 
     return unless
         $vars->{bugs}
-- 
cgit v1.2.3-24-g4f1b