1 files changed, 115 insertions, 0 deletions
diff --git a/extensions/Push/lib/Connectors.pm b/extensions/Push/lib/Connectors.pm
new file mode 100644
index 000000000..e765b4a43
--- /dev/null
+++ b/extensions/Push/lib/Connectors.pm
@@ -0,0 +1,115 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+#
+# This Source Code Form is "Incompatible With Secondary Licenses", as
+# defined by the Mozilla Public License, v. 2.0.
+
+package Bugzilla::Extension::Push::Connectors;
+
+use strict;
+use warnings;
+
+use Bugzilla::Extension::Push::Util;
+use Bugzilla::Constants;
+use Bugzilla::Util qw(trick_taint);
+use File::Basename;
+
+sub new {
+    my ($class) = @_;
+    my $self = {};
+    bless($self, $class);
+
+    $self->{names} = [];
+    $self->{objects} = {};
+    $self->{path} = bz_locations->{'extensionsdir'} . '/Push/lib/Connector';
+
+    my $logger = Bugzilla->push_ext->logger;
+    foreach my $file (glob($self->{path} . '/*.pm')) {
+        my $name = basename($file);
+        $name =~ s/\.pm$//;
+        next if $name eq 'Base';
+        if (length($name) > 32) {
+            $logger->info("Ignoring connector '$name': Name longer than 32 characters");
+        }
+        push @{$self->{names}}, $name;
+        $logger->debug("Found connector '$name'");
+    }
+
+    return $self;
+}
+
+sub _load {
+    my ($self) = @_;
+    return if scalar keys %{$self->{objects}};
+
+    my $logger = Bugzilla->push_ext->logger;
+    foreach my $name (@{$self->{names}}) {
+        next if exists $self->{objects}->{$name};
+        my $file = $self->{path} . "/$name.pm";
+        trick_taint($file);
+        require $file;
+        my $package = "Bugzilla::Extension::Push::Connector::$name";
+
+        $logger->debug("Loading connector '$name'");
+        my $old_error_mode = Bugzilla->error_mode;
+        Bugzilla->error_mode(ERROR_MODE_DIE);
+        eval {
+            my $connector = $package->new();
+            $connector->load_config();
+            $self->{objects}->{$name} = $connector;
+        };
+        if ($@) {
+            $logger->error("Connector '$name' failed to load: " . clean_error($@));
+        }
+        Bugzilla->error_mode($old_error_mode);
+    }
+}
+
+sub stop {
+    my ($self) = @_;
+    my $logger = Bugzilla->push_ext->logger;
+    foreach my $connector ($self->list) {
+        next unless $connector->enabled;
+        $logger->debug("Stopping '" . $connector->name . "'");
+        eval {
+            $connector->stop();
+        };
+        if ($@) {
+            $logger->error("Connector '" . $connector->name . "' failed to stop: " . clean_error($@));
+            $logger->debug("Connector '" . $connector->name . "' failed to stop: $@");
+        }
+    }
+}
+
+sub reload {
+    my ($self) = @_;
+    $self->stop();
+    $self->{objects} = {};
+    $self->_load();
+}
+
+sub names {
+    my ($self) = @_;
+    return @{$self->{names}};
+}
+
+sub list {
+    my ($self) = @_;
+    $self->_load();
+    return sort { $a->name cmp $b->name } values %{$self->{objects}};
+}
+
+sub exists {
+    my ($self, $name) = @_;
+    $self->by_name($name) ? 1 : 0;
+}
+
+sub by_name {
+    my ($self, $name) = @_;
+    return unless exists $self->{objects}->{$name};
+    return $self->{objects}->{$name};
+}
+
+1;
+