//------------------------------------------------------------------------------
// <copyright file="LqrSMACProcessTree.h" company="LiquidRobotics">
//
// Copyright (c) 2014 Liquid Robotics, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//
// This software is associated with a release.  Liquid Robotics makes no
// representation that the design, interfaces or component distribution will
// remain unchanged in future releases.
//
// </copyright>
//
// <summary>
//   Declaration of the ProcessTree class, which enumerates information about
//   running processes.
// </summary>
//
// <owner>Mike Cookson</owner>
//------------------------------------------------------------------------------

#include "LqrSMACSharedLib.h"
#include <list>
#include <map>
#include <ostream>
#include <set>
#include <string>
#include <boost/shared_ptr.hpp>
#include <sys/types.h>
#include <unistd.h>

namespace LiquidRobotics {
namespace SMACFramework {

class SHARED_LIB_PUBLIC ProcessTree
{
public:

	typedef std::list<pid_t> PIDList;
	typedef std::set<pid_t> PIDSet;
	typedef std::string string;

	struct Node
	{
		typedef boost::shared_ptr<Node> Ptr;

		pid_t pid;
		string procName;
		string commandLine;
		pid_t parentPID;
		PIDSet children;
	};

	ProcessTree(bool autoLoad = true, bool includeCmdLines = true);
	~ProcessTree();

	void AddChildrenOfPID(PIDSet& resultSet, pid_t pid, bool recursive = false) const;
	void GetChildrenOfPID(PIDSet& resultSet, pid_t pid, bool recursive = false) const;
	const Node& GetNode(pid_t pid) const;
	pid_t GetParentOfChild(pid_t childPID) const;
	void GetPidListBelow(pid_t pid, PIDList& result, bool reversed = false) const;
	bool HaveNode(pid_t pid) const;
	void Refresh(bool includeCmdLines = true);
	void Stream(std::ostream& out) const;

private:

	typedef std::map<pid_t,Node::Ptr> NodeMap;

	void AddPidListBelow(pid_t pid, PIDList& result, PIDSet& tracking) const;
	void AddPidListBelowReversed(pid_t pid, PIDList& result, PIDSet& tracking) const;

	NodeMap nodeMap;
};

std::ostream& operator<<(std::ostream& out, const ProcessTree::Node& node);
std::ostream& operator<<(std::ostream& out, const ProcessTree& tree);

}
}
