//------------------------------------------------------
// xwintool
//
//  simple tool for manipulating X-windows
//
//  written by: David Lawrence  5/14/2014
//------------------------------------------------------


#include <stdlib.h>

#include <iostream>
#include <iomanip>
#include <map>
#include <string>
#include <vector>
using namespace std;

#include <X11/Xlib.h>

#define _DBG__ cout<<__FILE__<<":"<<__LINE__<<endl;
#define _DBG_  cout<<__FILE__<<":"<<__LINE__<<" ";


bool SHOW_TREE = false;
bool KILL_WINDOWS = false;
vector<string> WINDOWS_TO_FIND;

void Usage(void);
void ParseCommandLineArguments(int narg, char *argv[]);
void FindWindows(Display *d, Window w, map<string, vector<Window> > &windows);
string GetWindowName(Display *d, Window w);

//-----------------
// main
//-----------------
int main(int narg, char *argv[])
{
	ParseCommandLineArguments(narg, argv);

	Display *display = XOpenDisplay(0);
	if(display==NULL){
		cout << "Unable to open X display!" << endl;
		exit(-1);
	}
	
	Window winRoot = XDefaultRootWindow(display);

	map<string, vector<Window> > windows;
	FindWindows(display, winRoot, windows);
	
	vector<Window> windows_killed;
	for(unsigned int i=0; i<WINDOWS_TO_FIND.size(); i++){
		string name = WINDOWS_TO_FIND[i];
		vector<Window> &wins = windows[name];
		cout << name << ": ";
		if(wins.size() == 0){
			cout << "No windows found" << endl;
		}else{
			for(unsigned int j=0; j<wins.size(); j++){
				cout << hex << wins[j] << " ";
				if(KILL_WINDOWS){
					XDestroyWindow(display, wins[j]);
					windows_killed.push_back(wins[j]);
				}
			}
			cout << endl;
		}
	}
	
	if(windows_killed.size()>0){
		cout << "windows killed: ";
		for(unsigned int i=0; i<windows_killed.size(); i++){
			cout << hex << windows_killed[i] << dec << " ";
		}
		cout << endl;
	}
	
	XCloseDisplay(display);

	return 0;
}

//-----------------
// Usage
//-----------------
void Usage(void)
{
	cout << endl;
	cout << "Usage:" << endl;
	cout << "    xwintool  [options] window_name [window_name] ... " << endl;
	cout << endl;
	cout << "  This is a utility that can be used to close a window" << endl;
	cout << "(or set of windows) identified by name. If just the" << endl;
	cout << "window name is given, then the Window resource ids of" << endl;
	cout << "all windows with that name are given. If the \"-kill\"" << endl;
	cout << "option is given, then all windows of all the specified" << endl;
	cout << "names will be closed. Note that the window will be" << endl;
	cout << "destroyed a'la XDestroyWindow which may not be the" << endl;
	cout << "cleanest way to shut the program down, but it should" << endl;
	cout << "be effective." << endl;
	cout << endl;
	cout << "options:" <<endl;
	cout << "   -tree      print the complete window hierarchy" << endl;
	cout << "   -kill      kill all windows with the specified name(s)" << endl;
	cout << endl;

	exit(0);
}

//-----------------
// ParseCommandLineArguments
//-----------------
void ParseCommandLineArguments(int narg, char *argv[])
{
	if(narg==1) Usage();

	for(int i=1; i<narg; i++){
		string arg(argv[i]);
		if(arg == "-tree") SHOW_TREE=1;
		if(arg == "-kill") KILL_WINDOWS=1;
		if(arg.find("-")!=0) WINDOWS_TO_FIND.push_back(argv[i]);
	}
}

//-----------------
// FindWindows
//-----------------
void FindWindows(Display *d, Window w, map<string, vector<Window> > &windows)
{
	static int depth = 0;

	// Get window name
	string name = GetWindowName(d, w);
	if(depth == 0) name = "root";
	
	// Add this window to the map
	windows[name].push_back(w);

	// Get List of children
	Window root_return, parent_return;
	Window *children_return;
	unsigned int nchildren=0;
	XQueryTree(d, w, &root_return, &parent_return, &children_return, &nchildren);

	// Optionally print window name	
	if(SHOW_TREE)cout << string(depth*2, ' ') << name << "  (children: " << nchildren << ")" << endl;
	
	// Loop over children
	for(unsigned int i=0; i<nchildren; i++){
		depth++;
		FindWindows(d, children_return[i], windows);
		depth--;
	}

	if(nchildren > 0) XFree(children_return);
}

//-----------------
// GetWindowName
//-----------------
string GetWindowName(Display *d, Window w)
{
	string name = "<unknown>";

	Atom nameAtom = XInternAtom(d,"_NET_WM_NAME",false);
	Atom utf8Atom = XInternAtom(d,"UTF8_STRING",false);
	Atom type;
	int format;
	unsigned long nitems, after;
	unsigned char *data = 0;
	
	int status = XGetWindowProperty(d, w, nameAtom, 0, 65536,
                                    false, utf8Atom, &type, &format,
                                    &nitems, &after, &data);

	if(status == Success) {
		if (data) {
			name = string((char*)data);
			XFree(data);
		}
	}
	
	return name;
}