tabctl.c

//for mingw
#ifndef _WIN32_IE
#define _WIN32_IE 0x0500
#endif

#include "const.h"
#include "utils.h"
#include "sistatusbar.h"
#include "tabctl.h"

#define PM_CLOSE_CURRENT	1314
#define PM_CLOSE_OTHER		1315
#define PM_CLOSE_RIGHT		1316
#define PM_CLOSE_LEFT		1317
#define PM_ABOUT			1400

extern HWND hwnd_tab_ctl;
extern GetU8FlagFn GetU8Flag;
static WNDPROC old_tab_ctl_proc = NULL;
static int last_row_count = 1;
static int last_rbutton_item = -1;
int last_active_item = -1;
static COLORREF color_table[10] =
{
	RGB(255,255,255),	//default
	RGB(204,182,238),	//c
	RGB(210,227,178),	//h
	RGB(219,236,249),	//txt
	RGB(249,179,180),	//js
	RGB(255,241,134),	//vb
};

//函数申明
static void SiTabCtl_AddCloseItem(void);

static LRESULT CALLBACK TabCtlSubClass(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	LRESULT lr = CallWindowProc(old_tab_ctl_proc, hWnd, uMsg, wParam, lParam);

	switch(uMsg)
	{
	case WM_RBUTTONDOWN:
		{
			SiTabCtl_OnRButtonClk();
		}
		break;
	case WM_LBUTTONDBLCLK:
		{
			SiTabCtl_OnLButtonDblClk();
		}
		break;
	case WM_COMMAND:
		{
			int pm = (int)wParam;
			if(pm == PM_CLOSE_CURRENT)
			{
				if(last_rbutton_item == -1)
					break;
				SiTabCtl_CloseItem(last_rbutton_item);
				last_rbutton_item = -1;
			}
			else if(pm == PM_CLOSE_OTHER)
			{
				int i,count;
				count = SiTabCtl_GetItemCount();
				if(last_rbutton_item == -1)
					break;
				for(i=count; i>last_rbutton_item; i--)
					SiTabCtl_CloseItem(i);
				for(i=last_rbutton_item-1; i>=0; i--)
					SiTabCtl_CloseItem(i);
				last_rbutton_item = -1;
			}
			else if(pm == PM_CLOSE_LEFT)
			{
				int i,count;
				count = SiTabCtl_GetItemCount();
				if(last_rbutton_item == -1)
					break;
				for(i=count; i>last_rbutton_item; i--)
					SiTabCtl_CloseItem(i);
				last_rbutton_item = -1;
			}
			else if(pm == PM_CLOSE_RIGHT)
			{
				int i;
				if(last_rbutton_item == -1)
					break;
				for(i=last_rbutton_item-1; i>=0; i--)
					SiTabCtl_CloseItem(i);
				last_rbutton_item = -1;
			}
			else if(pm == PM_ABOUT)
			{
				char about[1024];
				sprintf(about, "File: sihook.dll\n"
								"Version: %s\n"
								"Author: RedXu\n"
								"Website: https://github.com/redxu/sihook", PLUGIN_VERSION);
				MessageBox(hwnd_tab_ctl, about, "About sitab", MB_OK);
			}
		}
		break;
	default:
		break;
	}

	return lr;
}

void SiTabCtl_Create(HWND parent)
{
	hwnd_tab_ctl = CreateWindowEx(0,WC_TABCONTROL,"",
								WS_CHILD|WS_VISIBLE|TCS_BUTTONS|TCS_FLATBUTTONS|TCS_BOTTOM|TCS_MULTILINE|TCS_RAGGEDRIGHT|TCS_SCROLLOPPOSITE|TCS_FOCUSNEVER|TCS_OWNERDRAWFIXED,
								0,0,0,SI_TAB_HEIGHT,parent,NULL,NULL,NULL);
								
	HFONT font = CreateFont(16,0,0,0,FW_NORMAL,FALSE,FALSE,0,DEFAULT_CHARSET,OUT_DEFAULT_PRECIS,CLIP_DEFAULT_PRECIS,DEFAULT_QUALITY,DEFAULT_PITCH | FF_SWISS,"Courier New");
	SendMessage(hwnd_tab_ctl,WM_SETFONT,(WPARAM)font,MAKELPARAM(1,0));
							
	SiTabCtl_AddCloseItem();
	last_row_count = 1;
	old_tab_ctl_proc = (WNDPROC)GetWindowLong(hwnd_tab_ctl,GWL_WNDPROC);
	if(old_tab_ctl_proc == NULL)
	{
		OutputDebugString("old_tab_ctl_proc null!");
	}
	SetWindowLong(hwnd_tab_ctl,GWL_WNDPROC,(DWORD)TabCtlSubClass);
}

//获取页个数
int SiTabCtl_GetItemCount(void)
{
	//保留关闭按钮
	return TabCtrl_GetItemCount(hwnd_tab_ctl)-1;
}

//获取行数
int SiTabCtl_GetRowCount(void)
{
	int row = TabCtrl_GetRowCount(hwnd_tab_ctl);
	if(row == 0)
		row = 1;
	return row;
}

//获取高度
int SiTabCtl_GetHeight(void)
{
	return (SiTabCtl_GetRowCount()*SI_TAB_HEIGHT);
}

//添加关闭Item
static void SiTabCtl_AddCloseItem(void)
{
	RECT rect;
	TCITEM tci;
	memset(&tci,0,sizeof(TCITEM));
	tci.mask = TCIF_TEXT;
	tci.pszText = "全部关闭";
	TabCtrl_InsertItem(hwnd_tab_ctl,SI_TAB_CLOSE_IDX,&tci);
	//设置固定高度
	SendMessage(hwnd_tab_ctl,TCM_GETITEMRECT,SI_TAB_CLOSE_IDX,(LPARAM)&rect);
	SendMessage(hwnd_tab_ctl,TCM_SETITEMSIZE,SI_TAB_CLOSE_IDX,MAKELPARAM(rect.right-rect.left+1,SI_TAB_HEIGHT));
}

//添加Item
void SiTabCtl_AddItem(char* title,HWND hwnd)
{
	//创建
	int idx = SiTabCtl_GetItemCount();
	RECT rect;
	TCITEM tci;
	memset(&tci,0,sizeof(TCITEM));
	tci.mask = TCIF_TEXT | TCIF_PARAM;
	tci.pszText = (LPSTR)title;
	tci.lParam = (LPARAM)hwnd;
	TabCtrl_InsertItem(hwnd_tab_ctl,idx,&tci);
	//设置固定高度
	SendMessage(hwnd_tab_ctl,TCM_GETITEMRECT,idx,(LPARAM)&rect);
	SendMessage(hwnd_tab_ctl,TCM_SETITEMSIZE,idx,MAKELPARAM(rect.right-rect.left+1,SI_TAB_HEIGHT));
}

//查找item
int SiTabCtl_FindItem(HWND hwnd)
{
	int rtv = -1;
	int i;
	int count = SiTabCtl_GetItemCount();
	TCITEM tci;
	for(i=0;i<count;i++)
	{
		memset(&tci,0,sizeof(TCITEM));
		tci.mask = TCIF_TEXT | TCIF_PARAM;
		TabCtrl_GetItem(hwnd_tab_ctl,i,&tci);
		if((HWND)tci.lParam == hwnd)
		{
			rtv = i;
			break;
		}
	}

	return rtv;
}

//删除Item
void SiTabCtl_DelItem(HWND hwnd)
{
	int idx = SiTabCtl_FindItem(hwnd);
	if(idx == -1)
		return;
	TabCtrl_DeleteItem(hwnd_tab_ctl,idx);
}

//关闭标签页
void SiTabCtl_CloseItem(int idx)
{
	HWND hwnd;
	if(idx < 0)
		return;

	TCITEM tci;
	memset(&tci,0,sizeof(TCITEM));
	tci.mask = TCIF_TEXT | TCIF_PARAM;
	TabCtrl_GetItem(hwnd_tab_ctl,idx,&tci);
	hwnd = (HWND)tci.lParam;

	SendMessage(hwnd,WM_SYSCOMMAND,SC_CLOSE,0);	
}

//设置Item文字
void SiTabCtl_SetItemText(HWND hwnd,char* text)
{
	int idx = SiTabCtl_FindItem(hwnd);
	if(idx == -1)
		return;
	TCITEM tci;
	memset(&tci,0,sizeof(TCITEM));
	tci.mask = TCIF_TEXT | TCIF_PARAM;
	TabCtrl_GetItem(hwnd_tab_ctl,idx,&tci);

	tci.pszText = (LPSTR)text;
	TabCtrl_SetItem(hwnd_tab_ctl,idx,&tci);
}

//选中Item
void SiTabCtl_SetCurItem(HWND hwnd)
{
	int idx = SiTabCtl_FindItem(hwnd);
	if(idx == -1)
		return;
	TabCtrl_SetCurSel(hwnd_tab_ctl,idx);

	//notify si statusbar
	if(CheckUtf8Loaded())
	{
		char title[256];
		TCITEM tci;
		memset(&tci,0,sizeof(TCITEM));
		tci.mask = TCIF_TEXT;
		tci.pszText = (LPSTR)title;
		tci.cchTextMax = SI_BUF_SIZE;
		TabCtrl_GetItem(hwnd_tab_ctl,idx,&tci);

		int u8 = GetU8Flag(tci.pszText);
		if(u8 != -1)
		{
			SiStatusBar_ShowCodec(u8);
		}
	}
}

//获取当前Item
int SiTabCtl_GetCurItem(void)
{
	return TabCtrl_GetCurSel(hwnd_tab_ctl);
}

//获取鼠标所在Item
int SiTabCtl_GetCursorItem(void)
{
	POINT point;
	GetCursorPos(&point);
	ScreenToClient(hwnd_tab_ctl, &point);
	TCHITTESTINFO info;
    info.pt.x = point.x;
    info.pt.y = point.y;
    info.flags = TCHT_ONITEM;

    return TabCtrl_HitTest(hwnd_tab_ctl, &info);
}

//判断是否行数改变
BOOL SiTabCtl_IsRowChanged(void)
{
	int rtv = FALSE;
	int row = SiTabCtl_GetRowCount();
	if(last_row_count != row)
	{
		rtv = TRUE;
		last_row_count = row;
	}

	return rtv;
}

//位置改变
void SiTabCtl_OnPosChanging(WINDOWPOS* pos)
{
	if(!(pos->flags & SWP_NOSIZE))
	{
		int row;
		SetWindowPos(hwnd_tab_ctl,NULL,pos->x,pos->y,pos->cx,SI_TAB_HEIGHT,pos->flags);
		row = SiTabCtl_GetRowCount();
		SetWindowPos(hwnd_tab_ctl,NULL,0,0,pos->cx,SI_TAB_HEIGHT*row,SWP_NOMOVE);
	}
}

//标签选中改变
void SiTabCtl_OnSelChange(void)
{
	HWND hwnd;
	int idx = SiTabCtl_GetCurItem();
	TCITEM tci;
	memset(&tci,0,sizeof(TCITEM));
	tci.mask = TCIF_TEXT | TCIF_PARAM;
	TabCtrl_GetItem(hwnd_tab_ctl,idx,&tci);
	hwnd = (HWND)tci.lParam;
	SendMessage(GetParent(hwnd),WM_MDIACTIVATE,(WPARAM)hwnd,0);
}

//弹出菜单
static void SiTabCtl_PopupMenu(int index, POINT* point, int type)
{
	HMENU hMenu = CreatePopupMenu();
	if(type == 0)
	{
		AppendMenu(hMenu, MF_STRING, PM_CLOSE_CURRENT, "关闭标签");
		AppendMenu(hMenu, MF_STRING, PM_CLOSE_OTHER, "关闭其他");
		AppendMenu(hMenu, MF_STRING, PM_CLOSE_RIGHT, "关闭右侧");
		AppendMenu(hMenu, MF_STRING, PM_CLOSE_LEFT, "关闭左侧");
	}
	else
	{
		AppendMenu(hMenu, MF_STRING, PM_ABOUT, "关于sitab");
	}
	TrackPopupMenu(hMenu, TPM_LEFTALIGN, point->x, point->y, 0, hwnd_tab_ctl, NULL);
}

//双击选项卡
void SiTabCtl_OnLButtonDblClk(void)
{
	int idx = SiTabCtl_GetCurItem();
	SiTabCtl_CloseItem(idx);
}

//右键单击选项卡
void SiTabCtl_OnRButtonClk(void)
{
	int index, count;
	POINT point;
	count = SiTabCtl_GetItemCount();
	index = SiTabCtl_GetCursorItem();
	if(index < 0)
	{
		last_rbutton_item = -1;
		return;
	}
	last_rbutton_item = index;
	//TabCtrl_SetCurSel(hwnd_tab_ctl, index);
	GetCursorPos(&point);
	SiTabCtl_PopupMenu(index, &point, index == count);
}

//左键单击选项卡
void SiTabCtl_OnLButtonClk(void)
{
	int rtv;
	int idx = SiTabCtl_GetCurItem();
	int count = SiTabCtl_GetItemCount();
	if(idx != count)
		return;
	if(count == 0)
		return;

	rtv = MessageBox(hwnd_tab_ctl,"是否关闭所有标签?","sitab plugin by RedXu",MB_OKCANCEL|MB_ICONQUESTION|MB_DEFBUTTON2);
	if(rtv == IDCANCEL)
	{
		TabCtrl_SetCurSel(hwnd_tab_ctl, last_active_item);
		return;
	}

	while(count > 0)
	{
		HWND hwnd;
		TCITEM tci;
		memset(&tci,0,sizeof(TCITEM));
		tci.mask = TCIF_TEXT | TCIF_PARAM;
		TabCtrl_GetItem(hwnd_tab_ctl,count-1,&tci);
		hwnd = (HWND)tci.lParam;

		SendMessage(hwnd,WM_SYSCOMMAND,SC_CLOSE,0);
		count--;
	}
}

//重绘选项卡
void SiTabCtl_OnDrawItem(DRAWITEMSTRUCT* item)
{
	HBRUSH hBrush;
	TCITEM tci;
	char text[SI_BUF_SIZE];
	int type;

	memset(&tci,0,sizeof(TCITEM));
	tci.mask = TCIF_TEXT | TCIF_STATE;
	tci.pszText = (LPSTR)text;
	tci.cchTextMax = SI_BUF_SIZE;
	TabCtrl_GetItem(hwnd_tab_ctl,item->itemID,&tci);

	type = GetColorIndex(text);
	//创建颜色为hdc（窗口矩形）背景色的实画刷
	hBrush = CreateSolidBrush(color_table[type]);
	if(SiTabCtl_GetCurItem() == item->itemID)
	{
		hBrush = CreateSolidBrush(RGB(255,0,0));
	}
    //SetTextColor(lpDrawItem->hDC, RGB(0, 0, 255));
    //FrameRect(item->hDC,&item->rcItem,hBrush);
    FillRect(item->hDC,&item->rcItem,hBrush);
    DeleteObject(hBrush);
    SetBkMode(item->hDC,TRANSPARENT);
    DrawText(item->hDC,text,strlen(text),&item->rcItem,DT_CENTER|DT_LEFT|DT_VCENTER|DT_SINGLELINE);
}