Bug 909843: trackingflags do not support "Can be set on bug creation"

1 files changed, 17 insertions, 0 deletions

diff --git a/extensions/TrackingFlags/Extension.pm b/extensions/TrackingFlags/Extension.pm
index e3b8bb4e7..42b3b9f03 100644
--- a/extensions/TrackingFlags/Extension.pm
+++ b/extensions/TrackingFlags/Extension.pm
@@ -59,6 +59,7 @@ sub template_before_process {
     {
         $vars->{'tracking_flags'} = Bugzilla::Extension::TrackingFlags::Flag->match({
             product   => $vars->{'product'}->name,
+            enter_bug => 1,
             is_active => 1,
         });
 
@@ -169,6 +170,11 @@ sub db_schema_abstract_schema {
                 NOTNULL => 1,
                 DEFAULT => '0',
             },
+            enter_bug => {
+                TYPE    => 'BOOLEAN',
+                NOTNULL => 1,
+                DEFAULT => 'TRUE',
+            },
             is_active => {
                 TYPE    => 'BOOLEAN',
                 NOTNULL => 1,
@@ -265,11 +271,22 @@ sub db_schema_abstract_schema {
 
 sub install_update_db {
     my $dbh = Bugzilla->dbh;
+
     my $fk = $dbh->bz_fk_info('tracking_flags', 'field_id');
     if ($fk and !defined $fk->{DELETE}) {
         $fk->{DELETE} = 'CASCADE';
         $dbh->bz_alter_fk('tracking_flags', 'field_id', $fk);
     }
+
+    $dbh->bz_add_column(
+        'tracking_flags',
+        'enter_bug',
+        {
+            TYPE    => 'BOOLEAN',
+            NOTNULL => 1,
+            DEFAULT => 'TRUE',
+        }
+    );
 }
 
 sub active_custom_fields {