From 049e25f866d62388573997b5238f9cea7fd9cf41 Mon Sep 17 00:00:00 2001
From: miguel angel rasteu <miguelangel.rasteu@pandorafms.com>
Date: Fri, 6 Oct 2023 13:51:18 +0200
Subject: [PATCH] #8686 Add filter in managemnet tree group

---
 pandora_console/godmode/groups/group_list.php | 307 +++++++++++++-----
 pandora_console/include/ajax/tree.ajax.php    |   2 +
 .../include/class/TreeGroupEdition.class.php  | 148 +++++++--
 pandora_console/include/functions_agents.php  |  21 ++
 4 files changed, 373 insertions(+), 105 deletions(-)

diff --git a/pandora_console/godmode/groups/group_list.php b/pandora_console/godmode/groups/group_list.php
index 00bbdddb39..201d3cea4e 100644
--- a/pandora_console/godmode/groups/group_list.php
+++ b/pandora_console/godmode/groups/group_list.php
@@ -757,6 +757,125 @@ if ($tab == 'tree') {
     /*
      * Group tree view.
      */
+
+    $table = new stdClass();
+    $table->width = '100%';
+    $table->class = 'databox filters filter-table-adv';
+    if (is_metaconsole() === true) {
+        $table->cellspacing = 0;
+        $table->cellpadding = 0;
+    }
+
+    $search_group_string = get_parameter('search_group_string', '');
+    $search_agent_string = get_parameter('search_agent_string', '');
+    $agent_status = get_parameter('agent_status', '');
+    $show_not_init_agents = get_parameter('show_not_init_agents', 1);
+    $show_not_init_modules = get_parameter('show_not_init_modules', 1);
+    $show_full_hirearchy = get_parameter('show_full_hirearchy', 1);
+
+
+    $table->data = [];
+    $table->head = [];
+    $table->style = [];
+
+    $table->style[0] = 'width: 50%;';
+    $table->style[1] = 'width: 50%;';
+
+    $table->data[0][0] = html_print_label_input_block(
+        __('Search group'),
+        html_print_input_text(
+            'search_group_string',
+            $search_group_string,
+            '',
+            25,
+            255,
+            true,
+            false,
+            false,
+            '',
+        )
+    );
+
+    $agents_status_list = agents_status_list();
+
+    $table->data[0][1] = html_print_label_input_block(
+        __('Search by agent status').ui_print_help_tip(__('Shows the groups that contain an agent with the status that has been searched'), true),
+        html_print_select(
+            $agents_status_list,
+            'agent_status',
+            $agent_status,
+            '',
+            __('All'),
+            '',
+            true,
+            false,
+            false,
+            'w200p',
+            false,
+            'width: 100%;'
+        )
+    );
+
+    $table->data[1][0] = html_print_label_input_block(
+        __('Search by agent').ui_print_help_tip(__('Shows groups that contain an agent matching the search'), true),
+        html_print_input_text(
+            'search_agent_string',
+            $search_agent_string,
+            '',
+            25,
+            255,
+            true,
+            false,
+            false,
+            '',
+        )
+    );
+
+    $table->data[1][1] = html_print_label_input_block(
+        __('Show full hierarchy'),
+        html_print_checkbox_switch_extended(
+            'show_full_hirearchy',
+            1,
+            $show_full_hirearchy,
+            false,
+            '',
+            '',
+            true
+        )
+    );
+
+    $table->data[3][0] = '&nbsp;';
+
+    $table->data[3][1] = html_print_submit_button(
+        __('Filter'),
+        'filter',
+        false,
+        [
+            'class' => 'float-right',
+            'icon'  => 'search',
+        ],
+        true
+    );
+
+    $form = "<form method='post' action=''>";
+    $form .= html_print_table($table, true);
+    $form .= '</form>';
+
+
+
+    ui_toggle(
+        $form,
+        '<span class="subsection_header_title">'.__('Filters').'</span>',
+        'filter_form',
+        '',
+        true,
+        false,
+        '',
+        'white-box-content',
+        'box-flat white_table_graph fixed_filter_bar'
+    );
+
+
     echo "<div id='tree-controller-recipient'></div>";
 } else {
     /*
@@ -826,6 +945,8 @@ if ($tab == 'tree') {
         );
     $form .= '</form>';
 
+
+
     ui_toggle(
         $form,
         '<span class="subsection_header_title">'.__('Filters').'</span>',
@@ -1058,87 +1179,119 @@ $tab = 'group_edition';
 <?php } ?>
 
 <script type="text/javascript">
-    var treeController = TreeController.getController();
-    treeController.meta = <?php echo (is_metaconsole() === true) ? 1 : 0; ?>;
 
-    if (typeof treeController.recipient != 'undefined' && treeController.recipient.length > 0)
-            treeController.recipient.empty();
+    let show_full_hirearchy = "<?php echo $show_full_hirearchy; ?>";
+    let show_not_init_agents = "<?php echo $show_not_init_agents; ?>";
+    let show_not_init_modules = "<?php echo $show_not_init_modules; ?>";
 
-        showSpinner();
+    $('#checkbox-show_full_hirearchy').on("change", function() { 
+        if (show_full_hirearchy == 1) {
+            show_full_hirearchy = 0;
+        } else {
+            show_full_hirearchy = 1;
+        }
+    });
 
-        var parameters = {};
-        parameters['page'] = "include/ajax/tree.ajax";
-        parameters['getChildren'] = 1;
-        parameters['type'] = "<?php echo $tab; ?>";
-        parameters['filter'] = {};
-        parameters['filter']['searchGroup'] = '';
-        parameters['filter']['searchAgent'] = '';
-        parameters['filter']['statusAgent'] = '';
-        parameters['filter']['searchModule'] = '';
-        parameters['filter']['statusModule'] = '';
-        parameters['filter']['groupID'] = '';
-        parameters['filter']['tagID'] = '';
-        parameters['filter']['searchHirearchy'] = 1;
-        parameters['filter']['show_not_init_agents'] = 1;
-        parameters['filter']['show_not_init_modules'] = 1;
+    $('#button-filter').on('click', function(event) {
+        console.log('here');
+        event.preventDefault();
 
-        $.ajax({
-            type: "POST",
-            url: "<?php echo ui_get_full_url('ajax.php', false, false, false); ?>",
-            data: parameters,
-            success: function(data) {
-                if (data.success) {
-                    hideSpinner();
+        load_tree(show_full_hirearchy, show_not_init_agents, show_not_init_modules);
+    });
 
-                    treeController.init({
-                        recipient: $("div#tree-controller-recipient"),
-                        page: parameters['page'],
-                        emptyMessage: "<?php echo __('No data found'); ?>",
-                        foundMessage: "<?php echo __('Found groups'); ?>",
-                        tree: data.tree,
-                        baseURL: "<?php echo ui_get_full_url(false, false, false, is_metaconsole()); ?>",
-                        ajaxURL: "<?php echo ui_get_full_url('ajax.php', false, false, false); ?>",
-                        filter: parameters['filter'],
-                        counterTitles: {
-                            total: {
-                                agents: "<?php echo __('Total agents'); ?>",
-                                modules: "<?php echo __('Total modules'); ?>",
-                                none: "<?php echo __('Total'); ?>"
-                            },
-                            alerts: {
-                                agents: "<?php echo __('Fired alerts'); ?>",
-                                modules: "<?php echo __('Fired alerts'); ?>",
-                                none: "<?php echo __('Fired alerts'); ?>"
-                            },
-                            critical: {
-                                agents: "<?php echo __('Critical agents'); ?>",
-                                modules: "<?php echo __('Critical modules'); ?>",
-                                none: "<?php echo __('Critical'); ?>"
-                            },
-                            warning: {
-                                agents: "<?php echo __('Warning agents'); ?>",
-                                modules: "<?php echo __('Warning modules'); ?>",
-                                none: "<?php echo __('Warning'); ?>"
-                            },
-                            unknown: {
-                                agents: "<?php echo __('Unknown agents'); ?>",
-                                modules: "<?php echo __('Unknown modules'); ?>",
-                                none: "<?php echo __('Unknown'); ?>"
-                            },
-                            not_init: {
-                                agents: "<?php echo __('Not init agents'); ?>",
-                                modules: "<?php echo __('Not init modules'); ?>",
-                                none: "<?php echo __('Not init'); ?>"
-                            },
-                            ok: {
-                                agents: "<?php echo __('Normal agents'); ?>",
-                                modules: "<?php echo __('Normal modules'); ?>",
-                                none: "<?php echo __('Normal'); ?>"
+    window.addEventListener('load', function() {
+        load_tree(show_full_hirearchy, show_not_init_agents, show_not_init_modules);
+    });
+
+
+    
+
+    function load_tree(show_full_hirearchy, show_not_init_agents, show_not_init_modules) {
+        var treeController = TreeController.getController();
+        treeController.meta = <?php echo (is_metaconsole() === true) ? 1 : 0; ?>;
+
+        if (typeof treeController.recipient != 'undefined' && treeController.recipient.length > 0)
+                treeController.recipient.empty();
+
+            showSpinner();
+
+            var parameters = {};
+            parameters['page'] = "include/ajax/tree.ajax";
+            parameters['getChildren'] = 1;
+            parameters['type'] = "<?php echo $tab; ?>";
+            parameters['filter'] = {};
+            parameters['filter']['searchGroup'] = $('#text-search_group_string').val();
+            parameters['filter']['searchAgent'] = $('#text-search_agent_string').val();
+            parameters['filter']['statusAgent'] = $('#agent_status option:selected').val();
+            parameters['filter']['searchModule'] = '';
+            parameters['filter']['statusModule'] = '';
+            parameters['filter']['groupID'] = '';
+            parameters['filter']['tagID'] = '';
+            parameters['filter']['searchHirearchy'] = 1;
+            parameters['filter']['show_full_hirearchy'] = show_full_hirearchy;
+            parameters['filter']['show_not_init_agents'] = show_not_init_agents;
+            parameters['filter']['show_not_init_modules'] = show_not_init_modules;
+
+            $.ajax({
+                type: "POST",
+                url: "<?php echo ui_get_full_url('ajax.php', false, false, false); ?>",
+                data: parameters,
+                success: function(data) {
+                    if (data.success) {
+                        hideSpinner();
+
+                        treeController.init({
+                            recipient: $("div#tree-controller-recipient"),
+                            page: parameters['page'],
+                            emptyMessage: "<?php echo __('No data found'); ?>",
+                            foundMessage: "<?php echo __('Found groups'); ?>",
+                            tree: data.tree,
+                            baseURL: "<?php echo ui_get_full_url(false, false, false, is_metaconsole()); ?>",
+                            ajaxURL: "<?php echo ui_get_full_url('ajax.php', false, false, false); ?>",
+                            filter: parameters['filter'],
+                            counterTitles: {
+                                total: {
+                                    agents: "<?php echo __('Total agents'); ?>",
+                                    modules: "<?php echo __('Total modules'); ?>",
+                                    none: "<?php echo __('Total'); ?>"
+                                },
+                                alerts: {
+                                    agents: "<?php echo __('Fired alerts'); ?>",
+                                    modules: "<?php echo __('Fired alerts'); ?>",
+                                    none: "<?php echo __('Fired alerts'); ?>"
+                                },
+                                critical: {
+                                    agents: "<?php echo __('Critical agents'); ?>",
+                                    modules: "<?php echo __('Critical modules'); ?>",
+                                    none: "<?php echo __('Critical'); ?>"
+                                },
+                                warning: {
+                                    agents: "<?php echo __('Warning agents'); ?>",
+                                    modules: "<?php echo __('Warning modules'); ?>",
+                                    none: "<?php echo __('Warning'); ?>"
+                                },
+                                unknown: {
+                                    agents: "<?php echo __('Unknown agents'); ?>",
+                                    modules: "<?php echo __('Unknown modules'); ?>",
+                                    none: "<?php echo __('Unknown'); ?>"
+                                },
+                                not_init: {
+                                    agents: "<?php echo __('Not init agents'); ?>",
+                                    modules: "<?php echo __('Not init modules'); ?>",
+                                    none: "<?php echo __('Not init'); ?>"
+                                },
+                                ok: {
+                                    agents: "<?php echo __('Normal agents'); ?>",
+                                    modules: "<?php echo __('Normal modules'); ?>",
+                                    none: "<?php echo __('Normal'); ?>"
+                                }
                             }
-                        }
-                    });
-                }
-            },
-            dataType: "json"
-        });
+                        });
+                    }
+                },
+                dataType: "json"
+            });
+    }
+
+    
 </script>
diff --git a/pandora_console/include/ajax/tree.ajax.php b/pandora_console/include/ajax/tree.ajax.php
index 319b98e168..600b30e74d 100644
--- a/pandora_console/include/ajax/tree.ajax.php
+++ b/pandora_console/include/ajax/tree.ajax.php
@@ -59,6 +59,7 @@ if (is_ajax() === true) {
         $metaID = (int) get_parameter('metaID', 0);
         $childrenMethod = get_parameter('childrenMethod', 'on_demand');
 
+
         $default_filters = [
             'searchAgent'  => '',
             'statusAgent'  => AGENT_STATUS_ALL,
@@ -69,6 +70,7 @@ if (is_ajax() === true) {
         ];
         $filter = get_parameter('filter', $default_filters);
 
+
         $agent_a = check_acl($config['id_user'], 0, 'AR');
         $agent_w = check_acl($config['id_user'], 0, 'AW');
         $access = ($agent_a === true) ? 'AR' : (($agent_w === true) ? 'AW' : 'AR');
diff --git a/pandora_console/include/class/TreeGroupEdition.class.php b/pandora_console/include/class/TreeGroupEdition.class.php
index 4071266a58..48a103c724 100644
--- a/pandora_console/include/class/TreeGroupEdition.class.php
+++ b/pandora_console/include/class/TreeGroupEdition.class.php
@@ -106,20 +106,24 @@ class TreeGroupEdition extends TreeGroup
         }
 
         // Build the group hierarchy.
-        foreach ($groups as $id => $group) {
-            if (isset($groups[$id]['parent']) === true
-                && ($groups[$id]['parent'] != 0)
-            ) {
-                $parent = $groups[$id]['parent'];
-                // Parent exists.
-                if (isset($groups[$parent]['children']) === false) {
-                    $groups[$parent]['children'] = [];
-                }
+        if (isset($this->filter['show_full_hirearchy']) === false
+            || (isset($this->filter['show_full_hirearchy']) === true && (bool) $this->filter['show_full_hirearchy'] === true)
+        ) {
+            foreach ($groups as $id => $group) {
+                if (isset($groups[$id]['parent']) === true
+                    && ($groups[$id]['parent'] != 0)
+                ) {
+                    $parent = $groups[$id]['parent'];
+                    // Parent exists.
+                    if (isset($groups[$parent]['children']) === false) {
+                        $groups[$parent]['children'] = [];
+                    }
 
-                // Store a reference to the group into the parent.
-                $groups[$parent]['children'][] = &$groups[$id];
-                // This group was introduced into a parent.
-                $groups[$id]['have_parent'] = true;
+                    // Store a reference to the group into the parent.
+                    $groups[$parent]['children'][] = &$groups[$id];
+                    // This group was introduced into a parent.
+                    $groups[$id]['have_parent'] = true;
+                }
             }
         }
 
@@ -167,24 +171,112 @@ class TreeGroupEdition extends TreeGroup
         ];
 
         $group_acl = '';
-        if (users_can_manage_group_all('AR') === false) {
-            $user_groups_str = implode(',', $this->userGroupsArray);
-            $group_acl = sprintf(
-                'AND id_grupo IN (%s)',
-                $user_groups_str
+        $search_agent = '';
+        $status_agent = '';
+        $inner_agent = '';
+
+        if ((bool) is_metaconsole() === true) {
+            if (users_can_manage_group_all('AR') === false) {
+                $user_groups_str = implode(',', $this->userGroupsArray);
+                $group_acl = sprintf(
+                    ' AND tgrupo.id_grupo IN (%s) ',
+                    $user_groups_str
+                );
+            }
+
+            if (isset($this->filter['searchAgent']) === true && empty($this->filter['searchAgent']) === false
+                || isset($this->filter['statusAgent']) === true && strlen($this->filter['statusAgent']) > 0
+            ) {
+                $inner_agent = 'INNER JOIN tmetaconsole_agent ON tgrupo.id_grupo = tmetaconsole_agent.id_grupo';
+            }
+
+            if (isset($this->filter['searchAgent']) === true && empty($this->filter['searchAgent']) === false) {
+                $search_agent = ' AND tmetaconsole_agent.alias LIKE "%'.$this->filter['searchAgent'].'%" ';
+            }
+
+            if (isset($this->filter['statusAgent']) === true && strlen($this->filter['statusAgent']) > 0) {
+                switch ($this->filter['statusAgent']) {
+                    case AGENT_STATUS_NORMAL:
+                        $status_agent = ' AND (
+                            tmetaconsole_agent.critical_count = 0
+                            AND tmetaconsole_agent.warning_count = 0
+                            AND tmetaconsole_agent.unknown_count = 0 
+                            AND tmetaconsole_agent.normal_count > 0)';
+                    break;
+
+                    case AGENT_STATUS_WARNING:
+                        $status_agent = ' AND (
+                            tmetaconsole_agent.critical_count = 0 
+                            AND tmetaconsole_agent.warning_count > 0
+                            AND tmetaconsole_agent.total_count > 0)';
+                    break;
+
+                    case AGENT_STATUS_CRITICAL:
+                        $status_agent = ' AND tmetaconsole_agent.critical_count > 0';
+                    break;
+
+                    case AGENT_STATUS_UNKNOWN:
+                        $status_agent = ' AND (
+                            tmetaconsole_agent.critical_count = 0 
+                            AND tmetaconsole_agent.warning_count = 0 
+                            AND tmetaconsole_agent.unknown_count > 0)';
+                    break;
+
+                    case AGENT_STATUS_NOT_NORMAL:
+                        $status_agent = ' AND (
+                            tmetaconsole_agent.normal_count <> total_count
+                            OR tmetaconsole_agent.total_count = notinit_count)';
+                    break;
+
+                    case AGENT_STATUS_NOT_INIT:
+                        $status_agent = ' AND (
+                            tmetaconsole_agent.total_count = 0
+                            OR tmetaconsole_agent.total_count = notinit_count)';
+                    break;
+
+                    default:
+                        // Nothing to do.
+                    break;
+                }
+            }
+
+            $sql = sprintf(
+                'SELECT tgrupo.id_grupo AS gid,
+                tgrupo.nombre as name,
+                tgrupo.parent,
+                tgrupo.icon
+                FROM tgrupo
+                %s
+                WHERE 1=1
+                %s
+                %s
+                %s ',
+                $inner_agent,
+                $search_agent,
+                $status_agent,
+                $group_acl
+            );
+        } else {
+            if (users_can_manage_group_all('AR') === false) {
+                $user_groups_str = implode(',', $this->userGroupsArray);
+                $group_acl = sprintf(
+                    'AND id_grupo IN (%s)',
+                    $user_groups_str
+                );
+            }
+
+            $sql = sprintf(
+                'SELECT id_grupo AS gid,
+                nombre as name,
+                parent,
+                icon
+                FROM tgrupo
+                WHERE 1=1 
+                %s ',
+                $group_acl
             );
         }
 
-        $sql = sprintf(
-            'SELECT id_grupo AS gid,
-			nombre as name,
-            parent,
-            icon
-			FROM tgrupo
-            WHERE 1=1 %s',
-            $group_acl
-        );
-
         $stats = db_get_all_rows_sql($sql);
         $group_stats = [];
         foreach ($stats as $group) {
diff --git a/pandora_console/include/functions_agents.php b/pandora_console/include/functions_agents.php
index 720b0e6e55..f7457b3d3c 100644
--- a/pandora_console/include/functions_agents.php
+++ b/pandora_console/include/functions_agents.php
@@ -4856,4 +4856,25 @@ function get_resume_agent_concat($id_agente, $all_groups, $agent)
     $agent_contact .= html_print_table($table_contact, true);
 
     return $agent_contact;
+}
+
+
+/**
+ * Return an array with a list of status agents
+ *
+ * @return array.
+ */
+
+
+function agents_status_list()
+{
+    $status_list = [];
+    $status_list[AGENT_STATUS_NORMAL] = __('Normal');
+    $status_list[AGENT_STATUS_WARNING] = __('Warning');
+    $status_list[AGENT_STATUS_CRITICAL] = __('Critical');
+    $status_list[AGENT_STATUS_UNKNOWN] = __('Unknown');
+    $status_list[AGENT_STATUS_NOT_NORMAL] = __('Not normal');
+    $status_list[AGENT_STATUS_NOT_INIT] = __('Not init');
+
+    return $status_list;
 }
\ No newline at end of file