<?php
//Pandora FMS- http://pandorafms.com
// ==================================================
// Copyright (c) 2005-2018 Artica Soluciones Tecnologicas
// Please see http://pandorafms.org for full contribution list

// This program is free software; you can redistribute it and/or
// modify it under the terms of the  GNU Lesser General Public License
// as published by the Free Software Foundation; version 2

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

global $config;

require_once($config['homedir']."/include/class/Tree.class.php");

class TreeGroupEdition extends TreeGroup {
	public function  __construct($type, $rootType = '', $id = -1, $rootID = -1, $serverID = false, $childrenMethod = "on_demand", $access = 'AR') {
		global $config;

		parent::__construct($type, $rootType, $id, $rootID, $serverID, $childrenMethod, $access);
	}
	protected function getData() {
		if ($this->id == -1) {
			$this->getFirstLevel();
		}
	}

	protected function getProcessedGroups () {
		$processed_groups = array();
		// Index and process the groups
		$groups = $this->getGroupCounters();

		// If user have not permissions in parent, set parent node to 0 (all)
		// Avoid to do foreach for admins
		if (!users_can_manage_group_all("AR")) {
			foreach ($groups as $id => $group) {
				if (!isset($this->userGroups[$groups[$id]['parent']])) {
					$groups[$id]['parent'] = 0;
				}
			}
		}
		// Build the group hierarchy
		foreach ($groups as $id => $group) {
			if (isset($groups[$id]['parent']) && ($groups[$id]['parent'] != 0)) {
				$parent = $groups[$id]['parent'];
				// Parent exists
				if (!isset($groups[$parent]['children'])) {
					$groups[$parent]['children'] = array();
				}
				// 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;
			}
		}

		// Sort the children groups
		foreach ($groups as $id => $group) {
			if (isset($groups[$id]['children'])) {
				usort($groups[$id]['children'], array("Tree", "cmpSortNames"));
			}
		}
		//Filter groups and eliminates the reference to children groups out of her parent
		$groups = array_filter($groups, function ($group) {
			return !$group['have_parent'];
		});

		usort($groups, array("Tree", "cmpSortNames"));
		return $groups;
	}


	protected function getGroupCounters() {
		$messages = array(
			'confirm' => __('Confirm'),
			'cancel' => __('Cancel'),
			'messg' => __('Are you sure?')
		);
		$sql = "SELECT id_grupo AS gid,
			nombre as name, parent, icon
			FROM tgrupo
		";

		$stats = db_get_all_rows_sql($sql);
		$group_stats = array();
		foreach ($stats as $group) {
			$group_stats[$group['gid']]['name']   = $group['name'];
			$group_stats[$group['gid']]['parent'] = $group['parent'];
			$group_stats[$group['gid']]['icon']   = $group['icon'];
			$group_stats[$group['gid']]['id']     = $group['gid'];
			$group_stats[$group['gid']]['type']   = 'group';

			$group_stats[$group['gid']] = $this->getProcessedItem($group_stats[$group['gid']]);
			$group_stats[$group['gid']]['delete']['messages'] = $messages;
			$group_stats[$group['gid']]['edit']   = 1;
			$group_stats[$group['gid']]['alerts'] = '';
		}

		return $group_stats;
	}
}

?>