Allow site endpoint to be run locally

Author: abeverley

lib/Brass/API.pm

@@ -141,36 +141,14 @@ get 'api/site/' => sub {
     my $user = var 'api_user'
         or error __"Authentication required";
 
-    my $action = query_parameters->get('action')
-        or error __"Need required action";
-
-    my $schema = schema;
-
-    my $config = $schema->resultset('Config')->next
-        or error __x"Configuration information not defined. Please define in Brass in the Config => Site menu";
-
-    my $output;
-
-    if ($action eq 'smtp')
-    {
-        $output = [$config->smtp_relayhost];
-    }
-    elsif ($action eq 'internal_networks')
-    {
-        $output = [$config->internal_networks_all];
-    }
-    elsif ($action eq 'wazuh_manager')
-    {
-        $output = [$config->wazuh_manager];
-    }
-    else {
-        error __x"Unknown action {action}", action => $action;
-    }
+    my $return = $cdb->run_site(
+        action => query_parameters->get('action'),
+    );
 
     content_type 'application/json';
     encode_json({
         "is_error" => 0,
-        "result"   => encode_json($output),
+        "result"   => encode_json($return),
     });
 };
 

lib/Brass/ConfigDB.pm

@@ -53,7 +53,6 @@ sub _run_local
     my $type = delete $params{type}
         or error __"Please provide type of request with --type";
 
-    # XXX More types to be migrated
     if ($type eq 'pwd')
     {
         $self->run_pwd(%params);
@@ -66,6 +65,13 @@ sub _run_local
     {
         $self->run_server(%params);
     }
+    elsif ($type eq 'site')
+    {
+        $self->run_site(%params);
+    }
+    else {
+        error __x"Unknown type: {type}", type => $type;
+    }
 }
 
 sub _run_remote
@@ -486,6 +492,33 @@ sub run_server
     }
 }
 
+sub run_site
+{   my ($self, %params) = @_;
+
+    my $schema = $self->schema;
+    my $action = $params{action}
+        or error __"Need required action";
+
+    my $config = $schema->resultset('Config')->next
+        or error __x"Configuration information not defined. Please define in Brass in the Config => Site menu";
+
+    if ($action eq 'smtp')
+    {
+        return [$config->smtp_relayhost];
+    }
+    elsif ($action eq 'internal_networks')
+    {
+        return [$config->internal_networks_all];
+    }
+    elsif ($action eq 'wazuh_manager')
+    {
+        return [$config->wazuh_manager];
+    }
+    else {
+        error __x"Unknown action {action}", action => $action;
+    }
+}
+
 sub randompw()
 {   my $pwgen = CtrlO::Crypt::XkcdPassword->new;
     $pwgen->xkcd( words => 3, digits => 2 );