/*
	COUNTDOWN LIVE - by Mattias Buelens
	File: TreeView.js
	Desc: Class for working with tree-view controls
	Original from Script Editor by Matty, rewritten by myself
*/

var TV_FIRST = 0x1100;
var TVI_FIRST = -0xFFFF;
var TVI_LAST = -0xFFFE;
var TVI_ROOT = -0x10000;
var TVI_SORT = -0xFFFD;
var TVIF_CHILDREN = 0x40;
var TVIF_HANDLE = 0x10;
var TVIF_IMAGE = 0x2;
var TVIF_INTEGRAL = 0x80;
var TVIF_PARAM = 0x4;
var TVIF_SELECTEDIMAGE = 0x20;
var TVIF_STATE = 0x8;
var TVIF_TEXT = 0x1;
var TVGN_CARET = 0x9;
var TVGN_CHILD = 0x4;
var TVGN_DROPHILITE = 0x8;
var TVGN_FIRSTVISIBLE = 0x5;
var TVGN_LASTVISIBLE = 0xA;
var TVGN_NEXT = 0x1;
var TVGN_NEXTVISIBLE = 0x6;
var TVGN_PARENT = 0x3;
var TVGN_PREVIOUS = 0x2;
var TVGN_PREVIOUSVISIBLE = 0x7;
var TVGN_ROOT = 0x0;
var TVM_DELETEITEM = (TV_FIRST + 1);
var TVM_ENSUREVISIBLE = (TV_FIRST + 20);
var TVM_EXPAND = (TV_FIRST + 2);
var TVM_GETCOUNT = (TV_FIRST + 5);
var TVM_GETIMAGELIST = (TV_FIRST + 8);
var TVM_GETISEARCHSTRINGW = (TV_FIRST + 64);
var TVM_GETITEMHEIGHT = (TV_FIRST + 28);
var TVM_GETITEMRECT = (TV_FIRST + 4);
var TVM_GETITEMSTATE = (TV_FIRST + 39);
var TVM_GETITEMW = (TV_FIRST + 62);
var TVM_GETNEXTITEM = (TV_FIRST + 10);
var TVM_GETVISIBLECOUNT = (TV_FIRST + 16);
var TVM_HITTEST = (TV_FIRST + 17);
var TVM_INSERTITEMW = (TV_FIRST + 50);
var TVM_SELECTITEM = (TV_FIRST + 11);
var TVM_SETIMAGELIST = (TV_FIRST + 9);
var TVM_SETITEMW = (TV_FIRST + 63);
var TVN_FIRST = (-400);
var TVN_LAST = (-499);
var TVN_SELCHANGEDW = (TVN_FIRST-51);
var TVN_SINGLEEXPAND = (TVN_FIRST-15);

var TreeView = function(hWnd, sCtrlId) {
	//Get the control handle
	switch(arguments.length) {
		case 0:	return false;
		//PlusWnd and CtrlId given
		case 2:	hWnd = hWnd.GetControlHandle(sCtrlId);
	}

	//Array which stores the elements in the root
	hRootItems = [];

	return {
		fromObject : function(obj, hParent, fCallback) {
			if(typeof obj !== "object") return false;
			hParent = (typeof hParent === "number") ? hParent : TVI_ROOT;
			fCallback = (typeof fCallback === "function") ? fCallback : function() { };

			//Is the object an array?
			if('splice' in obj && 'join' in obj) {
				//Add the different trees to the root
				for(var i=0; i<obj.length; i++) this.fromObject(obj[i], hParent);
				return;
			}

			//Add the item to the root
			var hItem = this.addItem(obj.name, hParent, (obj.icon || false));
			fCallback(obj, hItem);

			//Does the object has any childs?
			if('childs' in obj) {
				//Add the childs to the created item
				for(var key in obj.childs) this.fromObject(obj.childs[key], hItem, fCallback);
			}
		},

		addItem : function(sLabel, hParent, iImage) {
			//Create a pointer to a null-terminated string for the item label
			var pszText = Interop.Allocate((sLabel.length + 1) * 2);
			pszText.WriteString(0, sLabel);
			//Create a TVINSERTSTRUCT structure
			var TVINSERTSTRUCT = Interop.Allocate(40);
			with(TVINSERTSTRUCT){
				WriteDWORD(0, hParent);
				WriteDWORD(4, TVI_LAST);
				WriteDWORD(24, pszText.DataPtr);
				if(typeof iImage === "number") {
					WriteDWORD(8, TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE);
					WriteDWORD(32, iImage);
					WriteDWORD(36, iImage);
				} else {
					WriteDWORD(8, TVIF_TEXT);
				}
			}

			//Insert the item
			var hItem = Interop.Call('user32', 'SendMessageW', hWnd, TVM_INSERTITEMW, 0, TVINSERTSTRUCT);
			//Clear the structures
			pszText.Size = 0;
			TVINSERTSTRUCT.Size = 0;
			//If the item was inserted to the root of the tree-view, save its handle
			if(hParent === 0 || hParent === TVI_ROOT){
				hRootItems.push(hItem);
			}

			return hItem;
		},

		getSelectedItem : function() {
			return Interop.Call('user32', 'SendMessageW', hWnd, TVM_GETNEXTITEM, TVGN_CARET, 0);
		},

		getItemText : function(hItem){
			hItem = (typeof hItem === 'number') ? hItem : this.getSelectedItem();

			//Create a TVITEM structure
			var TVITEM = Interop.Allocate(24);
			var pszText = Interop.Allocate((1024+1)*2);
			with(TVITEM) {
				WriteDWORD(0, TVIF_TEXT | TVIF_HANDLE);
				WriteDWORD(4, hItem);
				WriteDWORD(16, pszText.DataPtr);
				WriteDWORD(20, pszText.Size);
			}

			//Fill the structure with interesting data
			Interop.Call('user32', 'SendMessageW', hWnd, TVM_GETITEMW, 0, TVITEM);
			//Get the item text
			var returns = (pszText.ReadString(0, false) !== '') ? pszText.ReadString(0) : "";
			//Clear the structures
			pszText.Size = 0;
			TVITEM.Size = 0;

			return returns;
		},

		getParents : function(hItem, bAsString) {
			hItem = (typeof hItem === "number") ? hItem : this.getSelectedItem();
			bAsString = (bAsString == true);

			//Create an array to store the parents in
			var aParents = [];
			//Get the parent of the item
			var hParent = Interop.Call('user32', 'SendMessageW', hWnd, TVM_GETNEXTITEM, TVGN_PARENT, hItem);
			//Continue as long as we find a parent node
			while(hParent !== 0) {
				aParents.push(bAsString ? this.getItemText(hParent) : hParent);
				var hParent = Interop.Call('user32', 'SendMessageW', hWnd, TVM_GETNEXTITEM, TVGN_PARENT, hParent)
			}

			//Return the array of parents
			return aParents;
		},

		setItemImage : function (hItem, iImage) {
			if(!(typeof hItem === "number" && typeof iImage === "number")) return false;
			var TVITEM = Interop.Allocate(32);
			with(TVITEM){
				WriteDWORD(0, TVIF_HANDLE | TVIF_IMAGE | TVIF_SELECTEDIMAGE);
				WriteDWORD(4, hItem);
				WriteDWORD(24, iImage);
				WriteDWORD(28, iImage);
			}
			var Result = Interop.Call('user32', 'SendMessageW', hWnd, TVM_SETITEMW, 0, TVITEM);
			TVITEM.Size = 0;
			return Result;
		},

		selectItem : function(hItem) {
			return Interop.Call('user32', 'SendMessageW', hWnd, TVM_SELECTITEM, TVGN_CARET, hItem);
		},

		deleteItem : function(hItem) {
			return Interop.Call('user32', 'SendMessageW', hWnd, TVM_DELETEITEM, 0, hItem);
		},

		deleteAll : function() {
			return this.deleteItem(0);
		},

		getItemCount : function() {
			return Interop.Call('user32', 'SendMessageW', hWnd, TVM_GETCOUNT, 0, 0);
		}
	};
}