From 13ce6243394f7708a00f995c170e6e8d9223a566 Mon Sep 17 00:00:00 2001
From: Alejandro Gallardo Escobar <alex@artica.es>
Date: Fri, 2 Jan 2015 12:37:19 +0100
Subject: [PATCH] Added the os icons to the os tree

---
 pandora_console/include/class/Tree.class.php     | 16 ++++++++++------
 .../include/javascript/tree/TreeController.js    |  9 ++++++++-
 2 files changed, 18 insertions(+), 7 deletions(-)

diff --git a/pandora_console/include/class/Tree.class.php b/pandora_console/include/class/Tree.class.php
index 9d73fa1e39..8f8b5adb60 100644
--- a/pandora_console/include/class/Tree.class.php
+++ b/pandora_console/include/class/Tree.class.php
@@ -81,6 +81,7 @@ class Tree {
 
 		$list_os = os_get_os();
 
+		// Transform the os array to use the item id as key
 		if (!empty($list_os)) {
 			$list_os_aux = array();
 			foreach ($list_os as $os) {
@@ -125,7 +126,7 @@ class Tree {
 
 			$this->processModule($module);
 
-			// Module group
+			// OS item
 			if ($actual_os_root['id'] === (int)$value['id_os']) {
 				// Agent
 				if (empty($actual_agent) || $actual_agent['id'] !== (int)$value['id_agente']) {
@@ -151,6 +152,7 @@ class Tree {
 						$actual_os_root['counters'][$actual_agent['status']]++;
 				}
 				else {
+					// Add the module to the agent
 					$actual_agent['children'][] = $module;
 				}
 			}
@@ -163,13 +165,14 @@ class Tree {
 					$nodes[] = $actual_os_root;
 				}
 
-				// Create new module group
+				// Create new os item
 				$actual_os_root = array();
 				$actual_os_root['id'] = (int) $value['id_os'];
+				$actual_os_root['type'] = $this->type;
 
 				if (isset($list_os[$actual_os_root['id']])) {
 					$actual_os_root['name'] = $list_os[$actual_os_root['id']]['name'];
-					$actual_os_root['icon'] = $list_os[$actual_os_root['id']]['icon'];
+					$actual_os_root['icon'] = $list_os[$actual_os_root['id']]['icon_name'];
 				}
 				else {
 					$actual_os_root['name'] = __('Other');
@@ -204,9 +207,9 @@ class Tree {
 					$actual_os_root['counters'][$actual_agent['status']]++;
 			}
 		}
-		// If there is an agent and a module group open and not saved
+		// If there is an agent and an os item opened and not saved
 		if ($actual_os_root['id'] !== null) {
-			// Add the last agent to the module group
+			// Add the last agent to the os item
 			$actual_os_root['children'][] = $actual_agent;
 			// Add the last os to the branch
 			$nodes[] = $actual_os_root;
@@ -709,6 +712,7 @@ class Tree {
 				// Create new module group
 				$actual_module_group_root = array();
 				$actual_module_group_root['id'] = (int) $module['id_module_group'];
+				$actual_module_group_root['type'] = $this->type;
 
 				if (isset($module_groups[$module['id_module_group']])) {
 					$actual_module_group_root['name'] = $module_groups[$module['id_module_group']];
@@ -734,7 +738,7 @@ class Tree {
 					$actual_module_group_root['counters'][$actual_agent['status']]++;
 			}
 		}
-		// If there is an agent and a module group open and not saved
+		// If there is an agent and a module group opened and not saved
 		if ($actual_module_group_root['id'] !== null) {
 			// Add the last agent to the module group
 			$actual_module_group_root['children'][] = $actual_agent;
diff --git a/pandora_console/include/javascript/tree/TreeController.js b/pandora_console/include/javascript/tree/TreeController.js
index 70a805a109..c595ad9ae8 100644
--- a/pandora_console/include/javascript/tree/TreeController.js
+++ b/pandora_console/include/javascript/tree/TreeController.js
@@ -301,7 +301,7 @@ TreeController = {
 						case 'group':
 							if (typeof element.icon != 'undefined' && element.icon.length > 0) {
 								$content.append('<img src="'+(controller.baseURL.length > 0 ? controller.baseURL : '')
-									+'images/groups_small/'+element.icon+'" />')
+									+'images/groups_small/'+element.icon+'" /> ')
 							}
 							$content.append(element.name);
 							break;
@@ -311,6 +311,13 @@ TreeController = {
 						case 'module':
 							$content.append(element.name);
 							break;
+						case 'os':
+							if (typeof element.icon != 'undefined' && element.icon.length > 0) {
+								$content.append('<img src="'+(controller.baseURL.length > 0 ? controller.baseURL : '')
+									+'images/os_icons/'+element.icon+'" /> ')
+							}
+							$content.append(element.name);
+							break;
 						default:
 							$content.append(element.name);
 							break;