You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

629 lines
20 KiB

/*
* Copyright (C) 2003 Ulf Erikson <ulferikson@fastmail.fm>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* 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.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
#define WINVER 0x0500
#define _WIN32_WINNT 0x0500
#ifdef __MINGW32__
# define _WIN32_IE 0x0500
# include <windows.h>
# include <commctrl.h>
# include "../missing.h"
#endif
#define KMELEON_PLUGIN_EXPORTS
#include "ie_favorites.h"
#include "kmeleon_plugin.h"
#include "../rebar_menu/hot_tracking.h"
#include "../../app/KMeleonConst.h"
#include "LocalesUtils.h"
extern Locale* gLoc;
#include "Utils.h"
#include <sys/stat.h>
#include <errno.h>
#include <wininet.h>
extern kmeleonPlugin kPlugin;
extern void * KMeleonWndProc;
#include <string.h>
#define MAX_BUTTONS 32
UINT GetSiteIcon(CBookmarkNode* node)
{
if (node->iconurl.length()) {
CT_to_UTF8 url(node->iconurl.c_str());
UINT idx = kPlugin.kFuncs->GetIconIdx(url);
if (idx>0) return idx;
}
CT_to_UTF8 _url(node->url.c_str());
char* url = strdup(_url);
char* begin = strstr(url, "://");
if (!begin) {free(url);return 0;}
char* end = strchr(begin+3, '/');
if (end) *end = 0;
UINT i = kPlugin.kFuncs->GetIconIdx(url);
free(url);
return i;
}
TB *create_TB(HWND hWnd) {
TB *ptr = (TB*)calloc(1, sizeof(struct favoritesTB));
if (ptr) {
ptr->hWnd = hWnd;
ptr->next = root;
root = ptr;
}
return ptr;
}
TB *find_TB(HWND hWnd) {
TB *ptr = root;
while (ptr && ptr->hWnd != hWnd && ptr->hWndTB != hWnd)
ptr = ptr->next;
return ptr;
}
void remove_TB(HWND hWnd) {
TB *prev = NULL, *ptr = root;
while (ptr && (ptr->hWnd != hWnd)) {
prev = ptr;
ptr = ptr->next;
}
if (ptr) {
if (prev)
prev->next = ptr->next;
else
root = ptr->next;
if (ptr->hWndTB) {
TBBUTTON btn = {0};
while (SendMessage(ptr->hWndTB, TB_GETBUTTON, 0 /*index*/, (LPARAM)&btn)) {
if (IsMenu((HMENU)(btn.idCommand-SUBMENU_OFFSET))){
HMENU tmpMenu = (HMENU)(btn.idCommand-SUBMENU_OFFSET);
DestroyMenu(tmpMenu);
}
SendMessage(ptr->hWndTB, TB_DELETEBUTTON, 0 /*index*/, 0);
}
DestroyWindow(ptr->hWndTB);
}
free(ptr);
}
}
// Subclass procedure for the toolbar
LRESULT APIENTRY WndTBSubclassProc(
HWND hwnd,
UINT uMsg,
WPARAM wParam,
LPARAM lParam)
{
switch (uMsg) {
case WM_SIZE:
{
if (pNewTB == NULL &&
(wParam == SIZE_MAXIMIZED || wParam == SIZE_RESTORED)) {
TB *tmpTB = find_TB(hwnd);
if (tmpTB && tmpTB->hWnd)
SendMessage(tmpTB->hWnd, WM_COMMAND, nUpdateTB, 0);
}
}
break;
default:
break;
}
return CallWindowProc(wpOrigTBWndProc, hwnd, uMsg, wParam, lParam);
}
void findNick(char *nick, char **url)
{
HKEY hKey;
DWORD dwSize;
TCHAR regkey[128] = _T("Software\\Microsoft\\Internet Explorer\\SearchUrl\\");
_tcsncat(regkey, CUTF8_to_T(nick), 80);
if (RegOpenKey(HKEY_CURRENT_USER, (LPCTSTR)regkey, &hKey) == ERROR_SUCCESS) {
if (*url)
free(*url);
*url = (char *) calloc(INTERNET_MAX_URL_LENGTH+1, 1);
TCHAR wurl[INTERNET_MAX_URL_LENGTH+1];
dwSize = INTERNET_MAX_URL_LENGTH;
RegQueryValueEx(hKey, NULL, NULL, NULL, (LPBYTE)wurl, &dwSize);
RegCloseKey(hKey);
strncpy(*url, CT_to_UTF8(wurl), INTERNET_MAX_URL_LENGTH);
}
}
// Build Menu
void BuildMenu(HMENU menu, CBookmarkNode *node, BOOL isContinuation)
{
// screen height
int cy = GetSystemMetrics(SM_CYSCREEN);
// menu line height
// int cmenu = GetSystemMetrics(SM_CYMENU);
// if bmp_menu is enabled, the menu items will actually be at least 18 pixels... but this system call won't reflect that
// in any case, SM_CYMENU gets the height of the menu bar, not a menu item
// for now we'll just assume bmp_menu is enabled and they're 18 pixels...
#define cmenu 18
// space to allow above menu for title bar, menu bar (assuming maximized window), and extra frame junk
#define MENUPADDING 50
int maxLength = (gMenuAutoDetect) ? (int)((cy-MENUPADDING)/cmenu) : gMaxMenuLength;
CBookmarkNode *child;
int count = GetMenuItemCount(menu);
for (child = (isContinuation) ? node : node->child ; child ; child = child->next , count++) {
bEmpty = false;
if (count == maxLength) {
HMENU childMenu = CreatePopupMenu();
AppendMenu(menu, MF_STRING|MF_POPUP, (UINT)childMenu, gLoc->GetString(IDS_MORE));
BuildMenu(childMenu, child, true);
break;
}
else if (child->type == BOOKMARK_SEPARATOR) {
AppendMenu(menu, MF_SEPARATOR, 0, _T(""));
}
else if (child->type == BOOKMARK_FOLDER) {
HMENU childMenu = CreatePopupMenu();
child->id = (UINT)childMenu; // we have to save off the HMENU for the rebar
// condense the title and escape ampersands
TCHAR *pszTemp = fixString(child->text.c_str(), 40);
AppendMenu(menu, MF_STRING|MF_POPUP, (UINT)childMenu, pszTemp);
int pos;
for (pos=0;pos<GetMenuItemCount(menu);pos++)
if (GetSubMenu(menu,pos) == childMenu)
break;
MENUITEMINFO mi = {0};
mi.cbSize = sizeof(mi);
mi.fMask = MIIM_DATA | MIIM_FTYPE;
mi.fType = MF_OWNERDRAW;
mi.dwItemData = (ULONG_PTR)child->text.c_str();
int xx = SetMenuItemInfo(menu, pos, TRUE, &mi);
free(pszTemp);
BuildMenu(childMenu, child, false);
}
else if (child->type == BOOKMARK_BOOKMARK) {
if (!child->text.empty())
{
// condense the title and escape ampersands
TCHAR *pszTemp = fixString(child->text.c_str(), 40);
if (pszTemp) {
AppendMenu(menu, MF_STRING, child->id, pszTemp);
MENUITEMINFO mi = {0};
mi.cbSize = sizeof(mi);
mi.fMask = MIIM_DATA | MIIM_FTYPE;
mi.fType = MF_STRING | MF_OWNERDRAW;
mi.dwItemData = (ULONG_PTR)child->text.c_str();
SetMenuItemInfo(menu, child->id, FALSE, &mi);
delete pszTemp;
}
}
}
}
kPlugin.kFuncs->SetMenuDrawProc(menu, DrawBitmap);
}
void RebuildMenu() {
// delete the old bookmarks from the menu (FIXME - needs to be more robust than "delete everything after the first bookmark position" - there may be normal menu items there (if the user is weird))
while (DeleteMenu(gMenuFavorites, nFirstFavoritesPosition, MF_BYPOSITION));
// and rebuild
if (gMenuSortOrder)
gFavoritesRoot.sort(gMenuSortOrder);
BuildMenu(gMenuFavorites, gFavoritesRoot.FindSpecialNode(BOOKMARK_FLAG_BM), false);
// need to rebuild the rebar, too, in case it had submenus (whose ids are now invalid)
TB *ptr = root;
while (ptr) {
RebuildRebarMenu( ptr->hWndTB );
SendMessage(ptr->hWnd, WM_COMMAND, nUpdateTB, 0);
ptr = ptr->next;
}
}
int setTBButtonWidth(HWND hWndTB)
{
if (!bRebarEnabled || !hWndTB)
return 0;
int nMinWidth = nButtonMinWidth > 0 ? nButtonMinWidth * nHRes / nHSize : nButtonMinWidth;
int nMaxWidth = nButtonMaxWidth > 0 ? nButtonMaxWidth * nHRes / nHSize : nButtonMaxWidth;
CBookmarkNode *toolbarNode = gFavoritesRoot.FindSpecialNode(BOOKMARK_FLAG_TB);
int width = 0;
if (nMaxWidth > 0) {
RECT rect = {0};
if (GetWindowRect(hWndTB,&rect) == 0)
return 0;
width = rect.right - rect.left;
int i = 0;
CBookmarkNode *tmpNode = toolbarNode->child;
while (tmpNode && i<MAX_BUTTONS) {
i++;
tmpNode = tmpNode->next;
}
if (i) {
if (width > 0)
width = width / i;
else
width = nMaxWidth;
}
if (width > nMaxWidth) width = nMaxWidth;
if (width < nMinWidth) width = nMinWidth;
}
SendMessage(hWndTB, TB_SETBUTTONWIDTH, 0,
MAKELONG(nMaxWidth >= 0 ? width : 0,
nMaxWidth >= 0 ? width : 0));
return nMaxWidth;
}
// Build Rebar
void BuildRebar(HWND hWndTB)
{
if (!bRebarEnabled || !hWndTB)
return;
int nMaxWidth = nButtonMaxWidth > 0 ? nButtonMaxWidth * nHRes / nHSize : nButtonMaxWidth;
SendMessage(hWndTB, TB_SETBUTTONWIDTH, 0, MAKELONG(nButtonMaxWidth >= 0 ? nButtonMaxWidth : 0, nButtonMaxWidth >= 0 ? nButtonMaxWidth : 0));
CBookmarkNode *toolbarNode = gFavoritesRoot.FindSpecialNode(BOOKMARK_FLAG_TB);
int stringID, nButtons;
CBookmarkNode *child;
for (nButtons=0,child=toolbarNode->child; child && nButtons<MAX_BUTTONS; child=child->next, nButtons++) {
if (child->type == BOOKMARK_SEPARATOR) {
nButtons--;
continue;
}
// condense the title and escape ampersands
TCHAR *buttonString = fixString(child->text.c_str(), nMaxWidth > 0 ? 0 : -nMaxWidth);
TCHAR *buttonString2 = new TCHAR[_tcslen(buttonString)+2];
_tcscpy(buttonString2, buttonString);
buttonString2[_tcslen(buttonString)+1] = 0;
stringID = SendMessage(hWndTB, TB_ADDSTRING, (WPARAM)NULL, (LPARAM)buttonString2);
delete buttonString2;
delete buttonString;
TBBUTTON button = {0};
button.iString = stringID;
button.fsState = TBSTATE_ENABLED;
button.fsStyle = TBSTYLE_BUTTON | ((nButtonMaxWidth < 0) ? TBSTYLE_AUTOSIZE : 0) |
TBSTYLE_ALTDRAG | TBSTYLE_WRAPABLE;
if (child->type == BOOKMARK_FOLDER){
// toolbar may not be contained in bookmark menu, so we can't just use its submenus
HMENU childMenu = CreatePopupMenu();
BuildMenu(childMenu, child, false);
button.idCommand = MENU_TO_COMMAND((UINT)childMenu);
if (bButtonIcons)
button.iBitmap = MAKELONG(IMAGE_FOLDER_CLOSED, 0);
else
button.iBitmap = 0;
button.fsStyle |= TBSTYLE_DROPDOWN;
}
else {
button.idCommand = child->id;
if (bButtonIcons)
button.iBitmap = MAKELONG(IMAGE_BOOKMARK, 0);
else
button.iBitmap = 0;
}
SendMessage(hWndTB, TB_INSERTBUTTON, (WPARAM)-1, (LPARAM)&button);
}
if (bChevronEnabled) {
TBBUTTON button = {0};
button.fsState = TBSTATE_ENABLED;
button.fsStyle = TBSTYLE_SEP;
SendMessage(hWndTB, TB_INSERTBUTTON, (WPARAM)0, (LPARAM)&button);
button.iBitmap = IMAGE_CHEVRON;
button.idCommand = nDropdownCommand;
button.fsState = TBSTATE_ENABLED;
button.fsStyle = TBSTYLE_BUTTON | TBSTYLE_AUTOSIZE | TBSTYLE_DROPDOWN;
button.iString = -1;
SendMessage(hWndTB, TB_INSERTBUTTON, (WPARAM)0, (LPARAM)&button);
}
}
void UpdateRebarMenu(HWND hWndTB)
{
if (hWndTB && bRebarEnabled) {
setTBButtonWidth(hWndTB);
}
}
// BUG! Memory is leaking. Need to free TB_ADDSTRING strings too?
void RebuildRebarMenu(HWND hWndTB)
{
TBBUTTON btn = {0};
while (SendMessage(hWndTB, TB_GETBUTTON, 0 /*index*/, (LPARAM)&btn)) {
if (IsMenu((HMENU)(btn.idCommand-SUBMENU_OFFSET))){
HMENU tmpMenu = (HMENU)(btn.idCommand-SUBMENU_OFFSET);
DestroyMenu(tmpMenu);
}
SendMessage(hWndTB, TB_DELETEBUTTON, 0 /*index*/, 0);
}
BuildRebar(hWndTB);
}
int addLink(char *url, char *title, char *iconurl)
{
if (!url || !title)
return false;
if (!*title) title = url;
CBookmarkNode *newNode = new
CBookmarkNode(kPlugin.kFuncs->GetCommandIDs(1),
CUTF8_to_T(title), _T(""),
BOOKMARK_BOOKMARK, time(NULL));
if (!newNode)
return false;
newNode->url = CUTF8_to_T(url);
if (iconurl) newNode->iconurl = CUTF8_to_T(iconurl);
if (CreateFavorite(newNode) == 0) {
delete gFavoritesRoot.child;
delete gFavoritesRoot.next;
gFavoritesRoot.child = NULL;
gFavoritesRoot.next = NULL;
ReadFavorites(gFavoritesPath, _T(""), gFavoritesRoot);
}
RebuildMenu();
#if 0
TB *ptr = root;
while (ptr) {
RebuildRebarMenu( ptr->hWndTB );
ptr = ptr->next;
}
#endif
return true;
}
void OpenURL(TCHAR *aUrl)
{
char szOpenURLcmd[80];
CT_to_UTF8 url(aUrl);
kPlugin.kFuncs->GetPreference(PREF_STRING, PREFERENCE_FAVORITES_OPENURL, szOpenURLcmd, _T(""));
if (*szOpenURLcmd) {
char *plugin = szOpenURLcmd;
char *parameter = strchr(szOpenURLcmd, '(');
if (parameter) {
*parameter++ = 0;
char *close = strchr(parameter, ')');
if (close) {
*close = 0;
if (kPlugin.kFuncs->SendMessage(plugin, PLUGIN_NAME, parameter, (WPARAM)(const char*)url, 0))
return;
}
}
int idOpen = kPlugin.kFuncs->GetID(szOpenURLcmd);
if (idOpen == kPlugin.kFuncs->GetID("ID_OPEN_LINK")) {
kPlugin.kFuncs->NavigateTo(url, OPEN_NORMAL, NULL);
return;
}else if (idOpen == kPlugin.kFuncs->GetID("ID_OPEN_LINK_IN_BACKGROUND")) {
kPlugin.kFuncs->NavigateTo(url, OPEN_BACKGROUND, NULL);
return;
}else if (idOpen == kPlugin.kFuncs->GetID("ID_OPEN_LINK_IN_NEW_WINDOW")) {
kPlugin.kFuncs->NavigateTo(url, OPEN_NEW, NULL);
return;
}else if (idOpen == kPlugin.kFuncs->GetID("ID_OPEN_LINK_IN_NEW_TAB")) {
kPlugin.kFuncs->NavigateTo(url, OPEN_NEWTAB, NULL);
return;
}else if (idOpen == kPlugin.kFuncs->GetID("ID_OPEN_LINK_IN_BACKGROUNDTAB")) {
kPlugin.kFuncs->NavigateTo(url, OPEN_BACKGROUNDTAB, NULL);
return;
}
}
kPlugin.kFuncs->NavigateTo(url, OPEN_NORMAL, NULL);
}
void RefreshFavorites()
{
delete gFavoritesRoot.child;
delete gFavoritesRoot.next;
gFavoritesRoot.child = NULL;
gFavoritesRoot.next = NULL;
ReadFavorites(gFavoritesPath, _T(""), gFavoritesRoot);
RebuildMenu();
}
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) {
// store these in static vars so that the BeginHotTrack call can access them
static NMTOOLBAR tbhdr;
static NMHDR hdr;
if (message == WM_FAVORITE_NOTIFICATION) {
RefreshFavorites();
}
else if (message == WM_COMMAND) {
WORD command = LOWORD(wParam);
if (command == nConfigCommand){
Config(hWnd);
return 1;
}
else if (command == nUpdateTB){
#if 0
TB *tmpTB = find_TB(hWnd);
if (tmpTB && tmpTB->hWndTB) {
WINDOWPLACEMENT wp = {0};
wp.length = sizeof (WINDOWPLACEMENT);
if (GetWindowPlacement(tmpTB->hWndTB, &wp) != 0)
if (wp.showCmd == SW_SHOWNORMAL ||
wp.showCmd == SW_SHOWMAXIMIZED)
UpdateRebarMenu( tmpTB->hWndTB );
}
#endif
return 1;
}
else if (command == nAddCommand ||
command == nAddLinkCommand) {
char *title = 0;
char *url = 0;
char *iconurl =0;
if (command == nAddCommand) {
kmeleonDocInfo *dInfo = kPlugin.kFuncs->GetDocInfo(hWnd);
if (dInfo) {
url = dInfo->url;
title = dInfo->title;
iconurl = dInfo->iconurl;
}
addLink(url, title, iconurl);
}
else {
int retLen = kPlugin.kFuncs->GetGlobalVar(PREF_STRING, "LinkURL", NULL);
if (retLen) {
char *retVal = new char[retLen+1];
kPlugin.kFuncs->GetGlobalVar(PREF_STRING, "LinkURL", retVal);
url = title = retVal;
addLink(url, title);
delete retVal;
}
}
return true;
}
else if (command == nEditCommand) {
ShellExecute(hWnd, _T("explore"), gFavoritesPath, NULL, gFavoritesPath, SW_SHOWNORMAL);
return true;
}
else if (command == wm_deferhottrack) {
BeginHotTrack(&tbhdr, kPlugin.hDllInstance, hWnd);
return true;
}
else if (CBookmarkNode *node = gFavoritesRoot.FindNode(command)) {
node->lastVisit = time(NULL);
// a .URL file is formatted just like an .INI file, so we can
// use GetPrivateProfileString() to get the information we want
TCHAR *ptr = (TCHAR *)node->url.c_str();
if (ptr && *ptr == 0) {
TCHAR url[INTERNET_MAX_URL_LENGTH];
TCHAR path[INTERNET_MAX_URL_LENGTH];
_tcscpy(path, gFavoritesPath);
_tcscat(path, node->path.c_str());
GetPrivateProfileString(_T("InternetShortcut"), _T("URL"), _T(""), url, INTERNET_MAX_URL_LENGTH, path);
node->url = url;
}
OpenURL((TCHAR *)node->url.c_str());
return true;
}
}
else if (message == WM_NOTIFY) {
hdr = *((LPNMHDR)lParam);
if ((long)hdr.code == TBN_DROPDOWN) {
tbhdr = *((LPNMTOOLBAR)lParam);
// this is the little down arrow thing
if (tbhdr.iItem == nDropdownCommand){
RECT rc;
WPARAM index = 0;
SendMessage(tbhdr.hdr.hwndFrom, TB_GETITEMRECT, index, (LPARAM) &rc);
POINT pt = { rc.left, rc.bottom };
ClientToScreen(tbhdr.hdr.hwndFrom, &pt);
TrackPopupMenu((HMENU)gMenuFavorites, TPM_LEFTALIGN, pt.x, pt.y, 0, hWnd, NULL);
}
else if (IsMenu((HMENU)(tbhdr.iItem-SUBMENU_OFFSET))){
TCHAR toolbarName[11];
GetWindowText(tbhdr.hdr.hwndFrom, toolbarName, 10);
if (_tcscmp(toolbarName, _T(TOOLBAND_NAME)) != 0) {
// oops, this isn't our toolbar
return CallWindowProc((WNDPROC)KMeleonWndProc, hWnd, message, wParam, lParam);
}
// post a message to defer exceution of BeginHotTrack
PostMessage(hWnd, WM_COMMAND, wm_deferhottrack, (LPARAM)NULL);
return DefWindowProc(hWnd, message, wParam, lParam);
}
}
}
else if (message == WM_MENUSELECT) {
UINT id = LOWORD(wParam);
if (id >= nConfigCommand && id < nFirstFavoriteCommand) {
if (id == nConfigCommand)
kPlugin.kFuncs->SetStatusBarText(CT_to_UTF8(gLoc->GetString(IDS_CONFIGURE)));
else if (id == nAddCommand)
kPlugin.kFuncs->SetStatusBarText(CT_to_UTF8(gLoc->GetString(IDS_ADD)));
else if (id == nAddLinkCommand)
kPlugin.kFuncs->SetStatusBarText(CT_to_UTF8(gLoc->GetString(IDS_ADDLINK)));
else if (id == nEditCommand)
kPlugin.kFuncs->SetStatusBarText(CT_to_UTF8(gLoc->GetString(IDS_EDIT)));
return true;
}
else if (CBookmarkNode *node = gFavoritesRoot.FindNode(LOWORD(id))) {
TCHAR *ptr = (TCHAR *)node->url.c_str();
if (ptr && *ptr == 0) {
TCHAR url[INTERNET_MAX_URL_LENGTH];
TCHAR path[INTERNET_MAX_URL_LENGTH];
_tcscpy(path, gFavoritesPath);
_tcscat(path, node->path.c_str());
GetPrivateProfileString(_T("InternetShortcut"), _T("URL"), _T(""), url, INTERNET_MAX_URL_LENGTH, path);
node->url = url;
}
kPlugin.kFuncs->SetStatusBarText(CT_to_UTF8(node->url.c_str()));
return true;
}
else {
kPlugin.kFuncs->SetStatusBarText("");
}
}
return CallWindowProc((WNDPROC)KMeleonWndProc, hWnd, message, wParam, lParam);
}