mirror of
https://github.com/vim/vim.git
synced 2025-09-23 03:43:49 -04:00
updated for version 7.0002
This commit is contained in:
979
src/GvimExt/gvimext.cpp
Normal file
979
src/GvimExt/gvimext.cpp
Normal file
@@ -0,0 +1,979 @@
|
||||
/* vi:set ts=8 sts=4 sw=4:
|
||||
*
|
||||
* VIM - Vi IMproved gvimext by Tianmiao Hu
|
||||
*
|
||||
* Do ":help uganda" in Vim to read copying and usage conditions.
|
||||
* Do ":help credits" in Vim to see a list of people who contributed.
|
||||
*/
|
||||
|
||||
/*
|
||||
* gvimext is a DLL which is used for the "Edit with Vim" context menu
|
||||
* extension. It implements a MS defined interface with the Shell.
|
||||
*
|
||||
* If you have any questions or any suggestions concerning gvimext, please
|
||||
* contact Tianmiao Hu: tianmiao@acm.org.
|
||||
*/
|
||||
|
||||
#include "gvimext.h"
|
||||
|
||||
#ifdef __BORLANDC__
|
||||
# include <dir.h>
|
||||
# ifndef _strnicmp
|
||||
# define _strnicmp(a, b, c) strnicmp((a), (b), (c))
|
||||
# endif
|
||||
#else
|
||||
static char *searchpath(char *name);
|
||||
#endif
|
||||
|
||||
// Always get an error while putting the following stuff to the
|
||||
// gvimext.h file as class protected variables, give up and
|
||||
// declare them as global stuff
|
||||
FORMATETC fmte = {CF_HDROP,
|
||||
(DVTARGETDEVICE FAR *)NULL,
|
||||
DVASPECT_CONTENT,
|
||||
-1,
|
||||
TYMED_HGLOBAL
|
||||
};
|
||||
STGMEDIUM medium;
|
||||
HRESULT hres = 0;
|
||||
UINT cbFiles = 0;
|
||||
|
||||
//
|
||||
// Get the name of the Gvim executable to use, with the path.
|
||||
// When "runtime" is non-zero, we were called to find the runtime directory.
|
||||
// Returns the path in name[MAX_PATH]. It's empty when it fails.
|
||||
//
|
||||
static void
|
||||
getGvimName(char *name, int runtime)
|
||||
{
|
||||
HKEY keyhandle;
|
||||
DWORD hlen;
|
||||
|
||||
// Get the location of gvim from the registry.
|
||||
name[0] = 0;
|
||||
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, "Software\\Vim\\Gvim", 0,
|
||||
KEY_READ, &keyhandle) == ERROR_SUCCESS)
|
||||
{
|
||||
hlen = MAX_PATH;
|
||||
if (RegQueryValueEx(keyhandle, "path", 0, NULL, (BYTE *)name, &hlen)
|
||||
!= ERROR_SUCCESS)
|
||||
name[0] = 0;
|
||||
else
|
||||
name[hlen] = 0;
|
||||
RegCloseKey(keyhandle);
|
||||
}
|
||||
|
||||
// Registry didn't work, use the search path.
|
||||
if (name[0] == 0)
|
||||
strcpy(name, searchpath("gvim.exe"));
|
||||
|
||||
if (!runtime)
|
||||
{
|
||||
// Only when looking for the executable, not the runtime dir, we can
|
||||
// search for the batch file or a name without a path.
|
||||
if (name[0] == 0)
|
||||
strcpy(name, searchpath("gvim.bat"));
|
||||
if (name[0] == 0)
|
||||
strcpy(name, "gvim"); // finds gvim.bat or gvim.exe
|
||||
|
||||
// avoid that Vim tries to expand wildcards in the file names
|
||||
strcat(name, " --literal");
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
// Get the Vim runtime directory into buf[MAX_PATH].
|
||||
// The result is empty when it failed.
|
||||
// When it works, the path ends in a slash or backslash.
|
||||
//
|
||||
static void
|
||||
getRuntimeDir(char *buf)
|
||||
{
|
||||
int idx;
|
||||
|
||||
getGvimName(buf, 1);
|
||||
if (buf[0] != 0)
|
||||
{
|
||||
// When no path found, use the search path to expand it.
|
||||
if (strchr(buf, '/') == NULL && strchr(buf, '\\') == NULL)
|
||||
strcpy(buf, searchpath(buf));
|
||||
|
||||
// remove "gvim.exe" from the end
|
||||
for (idx = strlen(buf) - 1; idx >= 0; idx--)
|
||||
if (buf[idx] == '\\' || buf[idx] == '/')
|
||||
{
|
||||
buf[idx + 1] = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
// GETTEXT: translated messages and menu entries
|
||||
//
|
||||
#ifndef FEAT_GETTEXT
|
||||
# define _(x) x
|
||||
#else
|
||||
# define _(x) (*dyn_libintl_gettext)(x)
|
||||
# define VIMPACKAGE "vim"
|
||||
# ifndef GETTEXT_DLL
|
||||
# define GETTEXT_DLL "libintl.dll"
|
||||
# endif
|
||||
|
||||
// Dummy functions
|
||||
static char *null_libintl_gettext(const char *);
|
||||
static char *null_libintl_textdomain(const char *);
|
||||
static char *null_libintl_bindtextdomain(const char *, const char *);
|
||||
static int dyn_libintl_init(char *dir);
|
||||
static void dyn_libintl_end(void);
|
||||
|
||||
static HINSTANCE hLibintlDLL = 0;
|
||||
static char *(*dyn_libintl_gettext)(const char *) = null_libintl_gettext;
|
||||
static char *(*dyn_libintl_textdomain)(const char *) = null_libintl_textdomain;
|
||||
static char *(*dyn_libintl_bindtextdomain)(const char *, const char *)
|
||||
= null_libintl_bindtextdomain;
|
||||
|
||||
//
|
||||
// Attempt to load libintl.dll. If it doesn't work, use dummy functions.
|
||||
// "dir" is the directory where the libintl.dll might be.
|
||||
// Return 1 for success, 0 for failure.
|
||||
//
|
||||
static int
|
||||
dyn_libintl_init(char *dir)
|
||||
{
|
||||
int i;
|
||||
static struct
|
||||
{
|
||||
char *name;
|
||||
FARPROC *ptr;
|
||||
} libintl_entry[] =
|
||||
{
|
||||
{"gettext", (FARPROC*)&dyn_libintl_gettext},
|
||||
{"textdomain", (FARPROC*)&dyn_libintl_textdomain},
|
||||
{"bindtextdomain", (FARPROC*)&dyn_libintl_bindtextdomain},
|
||||
{NULL, NULL}
|
||||
};
|
||||
|
||||
// No need to initialize twice.
|
||||
if (hLibintlDLL)
|
||||
return 1;
|
||||
|
||||
// Load gettext library, first try the Vim runtime directory, then search
|
||||
// the path.
|
||||
strcat(dir, GETTEXT_DLL);
|
||||
hLibintlDLL = LoadLibrary(dir);
|
||||
if (!hLibintlDLL)
|
||||
{
|
||||
hLibintlDLL = LoadLibrary(GETTEXT_DLL);
|
||||
if (!hLibintlDLL)
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Get the addresses of the functions we need.
|
||||
for (i = 0; libintl_entry[i].name != NULL
|
||||
&& libintl_entry[i].ptr != NULL; ++i)
|
||||
{
|
||||
if ((*libintl_entry[i].ptr = GetProcAddress(hLibintlDLL,
|
||||
libintl_entry[i].name)) == NULL)
|
||||
{
|
||||
dyn_libintl_end();
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
static void
|
||||
dyn_libintl_end(void)
|
||||
{
|
||||
if (hLibintlDLL)
|
||||
FreeLibrary(hLibintlDLL);
|
||||
hLibintlDLL = NULL;
|
||||
dyn_libintl_gettext = null_libintl_gettext;
|
||||
dyn_libintl_textdomain = null_libintl_textdomain;
|
||||
dyn_libintl_bindtextdomain = null_libintl_bindtextdomain;
|
||||
}
|
||||
|
||||
static char *
|
||||
null_libintl_gettext(const char *msgid)
|
||||
{
|
||||
return (char *)msgid;
|
||||
}
|
||||
|
||||
static char *
|
||||
null_libintl_bindtextdomain(const char *domainname, const char *dirname)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static char *
|
||||
null_libintl_textdomain(const char* domainname)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
//
|
||||
// Setup for translating strings.
|
||||
//
|
||||
static void
|
||||
dyn_gettext_load(void)
|
||||
{
|
||||
char szBuff[MAX_PATH];
|
||||
char szLang[MAX_PATH];
|
||||
DWORD len;
|
||||
HKEY keyhandle;
|
||||
int gotlang = 0;
|
||||
|
||||
strcpy(szLang, "LANG=");
|
||||
|
||||
// First try getting the language from the registry, this can be
|
||||
// used to overrule the system language.
|
||||
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, "Software\\Vim\\Gvim", 0,
|
||||
KEY_READ, &keyhandle) == ERROR_SUCCESS)
|
||||
{
|
||||
len = MAX_PATH;
|
||||
if (RegQueryValueEx(keyhandle, "lang", 0, NULL, (BYTE*)szBuff, &len)
|
||||
== ERROR_SUCCESS)
|
||||
{
|
||||
szBuff[len] = 0;
|
||||
strcat(szLang, szBuff);
|
||||
gotlang = 1;
|
||||
}
|
||||
RegCloseKey(keyhandle);
|
||||
}
|
||||
|
||||
if (!gotlang && getenv("LANG") == NULL)
|
||||
{
|
||||
// Get the language from the system.
|
||||
// Could use LOCALE_SISO639LANGNAME, but it's not in Win95.
|
||||
// LOCALE_SABBREVLANGNAME gives us three letters, like "enu", we use
|
||||
// only the first two.
|
||||
len = GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_SABBREVLANGNAME,
|
||||
(LPTSTR)szBuff, MAX_PATH);
|
||||
if (len >= 2 && _strnicmp(szBuff, "en", 2) != 0)
|
||||
{
|
||||
// There are a few exceptions (probably more)
|
||||
if (_strnicmp(szBuff, "cht", 3) == 0
|
||||
|| _strnicmp(szBuff, "zht", 3) == 0)
|
||||
strcpy(szBuff, "zh_TW");
|
||||
else if (_strnicmp(szBuff, "chs", 3) == 0
|
||||
|| _strnicmp(szBuff, "zhc", 3) == 0)
|
||||
strcpy(szBuff, "zh_CN");
|
||||
else if (_strnicmp(szBuff, "jp", 2) == 0)
|
||||
strcpy(szBuff, "ja");
|
||||
else
|
||||
szBuff[2] = 0; // truncate to two-letter code
|
||||
strcat(szLang, szBuff);
|
||||
gotlang = 1;
|
||||
}
|
||||
}
|
||||
if (gotlang)
|
||||
putenv(szLang);
|
||||
|
||||
// Try to locate the runtime files. The path is used to find libintl.dll
|
||||
// and the vim.mo files.
|
||||
getRuntimeDir(szBuff);
|
||||
if (szBuff[0] != 0)
|
||||
{
|
||||
len = strlen(szBuff);
|
||||
if (dyn_libintl_init(szBuff))
|
||||
{
|
||||
strcpy(szBuff + len, "lang");
|
||||
|
||||
(*dyn_libintl_bindtextdomain)(VIMPACKAGE, szBuff);
|
||||
(*dyn_libintl_textdomain)(VIMPACKAGE);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
dyn_gettext_free(void)
|
||||
{
|
||||
dyn_libintl_end();
|
||||
}
|
||||
#endif // FEAT_GETTEXT
|
||||
|
||||
//
|
||||
// Global variables
|
||||
//
|
||||
UINT g_cRefThisDll = 0; // Reference count of this DLL.
|
||||
HINSTANCE g_hmodThisDll = NULL; // Handle to this DLL itself.
|
||||
|
||||
|
||||
//---------------------------------------------------------------------------
|
||||
// DllMain
|
||||
//---------------------------------------------------------------------------
|
||||
extern "C" int APIENTRY
|
||||
DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
|
||||
{
|
||||
switch (dwReason)
|
||||
{
|
||||
case DLL_PROCESS_ATTACH:
|
||||
// Extension DLL one-time initialization
|
||||
g_hmodThisDll = hInstance;
|
||||
break;
|
||||
|
||||
case DLL_PROCESS_DETACH:
|
||||
break;
|
||||
}
|
||||
|
||||
return 1; // ok
|
||||
}
|
||||
|
||||
static void
|
||||
inc_cRefThisDLL()
|
||||
{
|
||||
#ifdef FEAT_GETTEXT
|
||||
if (g_cRefThisDll == 0)
|
||||
dyn_gettext_load();
|
||||
#endif
|
||||
InterlockedIncrement((LPLONG)&g_cRefThisDll);
|
||||
}
|
||||
|
||||
static void
|
||||
dec_cRefThisDLL()
|
||||
{
|
||||
#ifdef FEAT_GETTEXT
|
||||
if (InterlockedDecrement((LPLONG)&g_cRefThisDll) == 0)
|
||||
dyn_gettext_free();
|
||||
#else
|
||||
InterlockedDecrement((LPLONG)&g_cRefThisDll);
|
||||
#endif
|
||||
}
|
||||
|
||||
//---------------------------------------------------------------------------
|
||||
// DllCanUnloadNow
|
||||
//---------------------------------------------------------------------------
|
||||
|
||||
STDAPI DllCanUnloadNow(void)
|
||||
{
|
||||
return (g_cRefThisDll == 0 ? S_OK : S_FALSE);
|
||||
}
|
||||
|
||||
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppvOut)
|
||||
{
|
||||
*ppvOut = NULL;
|
||||
|
||||
if (IsEqualIID(rclsid, CLSID_ShellExtension))
|
||||
{
|
||||
CShellExtClassFactory *pcf = new CShellExtClassFactory;
|
||||
|
||||
return pcf->QueryInterface(riid, ppvOut);
|
||||
}
|
||||
|
||||
return CLASS_E_CLASSNOTAVAILABLE;
|
||||
}
|
||||
|
||||
CShellExtClassFactory::CShellExtClassFactory()
|
||||
{
|
||||
m_cRef = 0L;
|
||||
|
||||
inc_cRefThisDLL();
|
||||
}
|
||||
|
||||
CShellExtClassFactory::~CShellExtClassFactory()
|
||||
{
|
||||
dec_cRefThisDLL();
|
||||
}
|
||||
|
||||
STDMETHODIMP CShellExtClassFactory::QueryInterface(REFIID riid,
|
||||
LPVOID FAR *ppv)
|
||||
{
|
||||
*ppv = NULL;
|
||||
|
||||
// Any interface on this object is the object pointer
|
||||
|
||||
if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IClassFactory))
|
||||
{
|
||||
*ppv = (LPCLASSFACTORY)this;
|
||||
|
||||
AddRef();
|
||||
|
||||
return NOERROR;
|
||||
}
|
||||
|
||||
return E_NOINTERFACE;
|
||||
}
|
||||
|
||||
STDMETHODIMP_(ULONG) CShellExtClassFactory::AddRef()
|
||||
{
|
||||
return InterlockedIncrement((LPLONG)&m_cRef);
|
||||
}
|
||||
|
||||
STDMETHODIMP_(ULONG) CShellExtClassFactory::Release()
|
||||
{
|
||||
if (InterlockedDecrement((LPLONG)&m_cRef))
|
||||
return m_cRef;
|
||||
|
||||
delete this;
|
||||
|
||||
return 0L;
|
||||
}
|
||||
|
||||
STDMETHODIMP CShellExtClassFactory::CreateInstance(LPUNKNOWN pUnkOuter,
|
||||
REFIID riid,
|
||||
LPVOID *ppvObj)
|
||||
{
|
||||
*ppvObj = NULL;
|
||||
|
||||
// Shell extensions typically don't support aggregation (inheritance)
|
||||
|
||||
if (pUnkOuter)
|
||||
return CLASS_E_NOAGGREGATION;
|
||||
|
||||
// Create the main shell extension object. The shell will then call
|
||||
// QueryInterface with IID_IShellExtInit--this is how shell extensions are
|
||||
// initialized.
|
||||
|
||||
LPCSHELLEXT pShellExt = new CShellExt(); //Create the CShellExt object
|
||||
|
||||
if (NULL == pShellExt)
|
||||
return E_OUTOFMEMORY;
|
||||
|
||||
return pShellExt->QueryInterface(riid, ppvObj);
|
||||
}
|
||||
|
||||
|
||||
STDMETHODIMP CShellExtClassFactory::LockServer(BOOL fLock)
|
||||
{
|
||||
return NOERROR;
|
||||
}
|
||||
|
||||
// *********************** CShellExt *************************
|
||||
CShellExt::CShellExt()
|
||||
{
|
||||
m_cRef = 0L;
|
||||
m_pDataObj = NULL;
|
||||
|
||||
inc_cRefThisDLL();
|
||||
}
|
||||
|
||||
CShellExt::~CShellExt()
|
||||
{
|
||||
if (m_pDataObj)
|
||||
m_pDataObj->Release();
|
||||
|
||||
dec_cRefThisDLL();
|
||||
}
|
||||
|
||||
STDMETHODIMP CShellExt::QueryInterface(REFIID riid, LPVOID FAR *ppv)
|
||||
{
|
||||
*ppv = NULL;
|
||||
|
||||
if (IsEqualIID(riid, IID_IShellExtInit) || IsEqualIID(riid, IID_IUnknown))
|
||||
{
|
||||
*ppv = (LPSHELLEXTINIT)this;
|
||||
}
|
||||
else if (IsEqualIID(riid, IID_IContextMenu))
|
||||
{
|
||||
*ppv = (LPCONTEXTMENU)this;
|
||||
}
|
||||
|
||||
if (*ppv)
|
||||
{
|
||||
AddRef();
|
||||
|
||||
return NOERROR;
|
||||
}
|
||||
|
||||
return E_NOINTERFACE;
|
||||
}
|
||||
|
||||
STDMETHODIMP_(ULONG) CShellExt::AddRef()
|
||||
{
|
||||
return InterlockedIncrement((LPLONG)&m_cRef);
|
||||
}
|
||||
|
||||
STDMETHODIMP_(ULONG) CShellExt::Release()
|
||||
{
|
||||
|
||||
if (InterlockedDecrement((LPLONG)&m_cRef))
|
||||
return m_cRef;
|
||||
|
||||
delete this;
|
||||
|
||||
return 0L;
|
||||
}
|
||||
|
||||
|
||||
//
|
||||
// FUNCTION: CShellExt::Initialize(LPCITEMIDLIST, LPDATAOBJECT, HKEY)
|
||||
//
|
||||
// PURPOSE: Called by the shell when initializing a context menu or property
|
||||
// sheet extension.
|
||||
//
|
||||
// PARAMETERS:
|
||||
// pIDFolder - Specifies the parent folder
|
||||
// pDataObj - Spefifies the set of items selected in that folder.
|
||||
// hRegKey - Specifies the type of the focused item in the selection.
|
||||
//
|
||||
// RETURN VALUE:
|
||||
//
|
||||
// NOERROR in all cases.
|
||||
//
|
||||
// COMMENTS: Note that at the time this function is called, we don't know
|
||||
// (or care) what type of shell extension is being initialized.
|
||||
// It could be a context menu or a property sheet.
|
||||
//
|
||||
|
||||
STDMETHODIMP CShellExt::Initialize(LPCITEMIDLIST pIDFolder,
|
||||
LPDATAOBJECT pDataObj,
|
||||
HKEY hRegKey)
|
||||
{
|
||||
// Initialize can be called more than once
|
||||
if (m_pDataObj)
|
||||
m_pDataObj->Release();
|
||||
|
||||
// duplicate the object pointer and registry handle
|
||||
|
||||
if (pDataObj)
|
||||
{
|
||||
m_pDataObj = pDataObj;
|
||||
pDataObj->AddRef();
|
||||
}
|
||||
|
||||
return NOERROR;
|
||||
}
|
||||
|
||||
|
||||
//
|
||||
// FUNCTION: CShellExt::QueryContextMenu(HMENU, UINT, UINT, UINT, UINT)
|
||||
//
|
||||
// PURPOSE: Called by the shell just before the context menu is displayed.
|
||||
// This is where you add your specific menu items.
|
||||
//
|
||||
// PARAMETERS:
|
||||
// hMenu - Handle to the context menu
|
||||
// indexMenu - Index of where to begin inserting menu items
|
||||
// idCmdFirst - Lowest value for new menu ID's
|
||||
// idCmtLast - Highest value for new menu ID's
|
||||
// uFlags - Specifies the context of the menu event
|
||||
//
|
||||
// RETURN VALUE:
|
||||
//
|
||||
//
|
||||
// COMMENTS:
|
||||
//
|
||||
|
||||
STDMETHODIMP CShellExt::QueryContextMenu(HMENU hMenu,
|
||||
UINT indexMenu,
|
||||
UINT idCmdFirst,
|
||||
UINT idCmdLast,
|
||||
UINT uFlags)
|
||||
{
|
||||
UINT idCmd = idCmdFirst;
|
||||
|
||||
hres = m_pDataObj->GetData(&fmte, &medium);
|
||||
if (medium.hGlobal)
|
||||
cbFiles = DragQueryFile((HDROP)medium.hGlobal, (UINT)-1, 0, 0);
|
||||
|
||||
// InsertMenu(hMenu, indexMenu++, MF_SEPARATOR|MF_BYPOSITION, 0, NULL);
|
||||
|
||||
// Initialize m_cntOfHWnd to 0
|
||||
m_cntOfHWnd = 0;
|
||||
// Retieve all the vim instances
|
||||
EnumWindows(EnumWindowsProc, (LPARAM)this);
|
||||
|
||||
if (cbFiles > 1)
|
||||
{
|
||||
InsertMenu(hMenu,
|
||||
indexMenu++,
|
||||
MF_STRING|MF_BYPOSITION,
|
||||
idCmd++,
|
||||
_("Edit with &multiple Vims"));
|
||||
|
||||
InsertMenu(hMenu,
|
||||
indexMenu++,
|
||||
MF_STRING|MF_BYPOSITION,
|
||||
idCmd++,
|
||||
_("Edit with single &Vim"));
|
||||
|
||||
if (cbFiles <= 4)
|
||||
{
|
||||
// Can edit up to 4 files in diff mode
|
||||
InsertMenu(hMenu,
|
||||
indexMenu++,
|
||||
MF_STRING|MF_BYPOSITION,
|
||||
idCmd++,
|
||||
_("&Diff with Vim"));
|
||||
m_edit_existing_off = 3;
|
||||
}
|
||||
else
|
||||
m_edit_existing_off = 2;
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
InsertMenu(hMenu,
|
||||
indexMenu++,
|
||||
MF_STRING|MF_BYPOSITION,
|
||||
idCmd++,
|
||||
_("Edit with &Vim"));
|
||||
m_edit_existing_off = 1;
|
||||
}
|
||||
|
||||
// Now display all the vim instances
|
||||
for (int i = 0; i < m_cntOfHWnd; i++)
|
||||
{
|
||||
char title[MAX_PATH];
|
||||
char temp[MAX_PATH];
|
||||
|
||||
// Obtain window title, continue if can not
|
||||
if (GetWindowText(m_hWnd[i], title, MAX_PATH - 1) == 0)
|
||||
continue;
|
||||
// Truncate the title before the path, keep the file name
|
||||
char *pos = strchr(title, '(');
|
||||
if (pos != NULL)
|
||||
{
|
||||
if (pos > title && pos[-1] == ' ')
|
||||
--pos;
|
||||
*pos = 0;
|
||||
}
|
||||
// Now concatenate
|
||||
strncpy(temp, _("Edit with existing Vim - &"), MAX_PATH - 1);
|
||||
strncat(temp, title, MAX_PATH - 1);
|
||||
InsertMenu(hMenu,
|
||||
indexMenu++,
|
||||
MF_STRING|MF_BYPOSITION,
|
||||
idCmd++,
|
||||
temp);
|
||||
}
|
||||
// InsertMenu(hMenu, indexMenu++, MF_SEPARATOR|MF_BYPOSITION, 0, NULL);
|
||||
|
||||
// Must return number of menu items we added.
|
||||
return ResultFromShort(idCmd-idCmdFirst);
|
||||
}
|
||||
|
||||
//
|
||||
// FUNCTION: CShellExt::InvokeCommand(LPCMINVOKECOMMANDINFO)
|
||||
//
|
||||
// PURPOSE: Called by the shell after the user has selected on of the
|
||||
// menu items that was added in QueryContextMenu().
|
||||
//
|
||||
// PARAMETERS:
|
||||
// lpcmi - Pointer to an CMINVOKECOMMANDINFO structure
|
||||
//
|
||||
// RETURN VALUE:
|
||||
//
|
||||
//
|
||||
// COMMENTS:
|
||||
//
|
||||
|
||||
STDMETHODIMP CShellExt::InvokeCommand(LPCMINVOKECOMMANDINFO lpcmi)
|
||||
{
|
||||
HRESULT hr = E_INVALIDARG;
|
||||
|
||||
// If HIWORD(lpcmi->lpVerb) then we have been called programmatically
|
||||
// and lpVerb is a command that should be invoked. Otherwise, the shell
|
||||
// has called us, and LOWORD(lpcmi->lpVerb) is the menu ID the user has
|
||||
// selected. Actually, it's (menu ID - idCmdFirst) from QueryContextMenu().
|
||||
if (!HIWORD(lpcmi->lpVerb))
|
||||
{
|
||||
UINT idCmd = LOWORD(lpcmi->lpVerb);
|
||||
|
||||
if (idCmd >= m_edit_existing_off)
|
||||
{
|
||||
// Existing with vim instance
|
||||
hr = PushToWindow(lpcmi->hwnd,
|
||||
lpcmi->lpDirectory,
|
||||
lpcmi->lpVerb,
|
||||
lpcmi->lpParameters,
|
||||
lpcmi->nShow,
|
||||
idCmd - m_edit_existing_off);
|
||||
}
|
||||
else
|
||||
{
|
||||
switch (idCmd)
|
||||
{
|
||||
case 0:
|
||||
hr = InvokeGvim(lpcmi->hwnd,
|
||||
lpcmi->lpDirectory,
|
||||
lpcmi->lpVerb,
|
||||
lpcmi->lpParameters,
|
||||
lpcmi->nShow);
|
||||
break;
|
||||
case 1:
|
||||
hr = InvokeSingleGvim(lpcmi->hwnd,
|
||||
lpcmi->lpDirectory,
|
||||
lpcmi->lpVerb,
|
||||
lpcmi->lpParameters,
|
||||
lpcmi->nShow,
|
||||
0);
|
||||
break;
|
||||
case 2:
|
||||
hr = InvokeSingleGvim(lpcmi->hwnd,
|
||||
lpcmi->lpDirectory,
|
||||
lpcmi->lpVerb,
|
||||
lpcmi->lpParameters,
|
||||
lpcmi->nShow,
|
||||
1);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
return hr;
|
||||
}
|
||||
|
||||
STDMETHODIMP CShellExt::PushToWindow(HWND hParent,
|
||||
LPCSTR pszWorkingDir,
|
||||
LPCSTR pszCmd,
|
||||
LPCSTR pszParam,
|
||||
int iShowCmd,
|
||||
int idHWnd)
|
||||
{
|
||||
HWND hWnd = m_hWnd[idHWnd];
|
||||
|
||||
// Show and bring vim instance to foreground
|
||||
if (IsIconic(hWnd) != 0)
|
||||
ShowWindow(hWnd, SW_RESTORE);
|
||||
else
|
||||
ShowWindow(hWnd, SW_SHOW);
|
||||
SetForegroundWindow(hWnd);
|
||||
|
||||
// Post the selected files to the vim instance
|
||||
PostMessage(hWnd, WM_DROPFILES, (WPARAM)medium.hGlobal, 0);
|
||||
|
||||
return NOERROR;
|
||||
}
|
||||
|
||||
STDMETHODIMP CShellExt::GetCommandString(UINT idCmd,
|
||||
UINT uFlags,
|
||||
UINT FAR *reserved,
|
||||
LPSTR pszName,
|
||||
UINT cchMax)
|
||||
{
|
||||
if (uFlags == GCS_HELPTEXT && cchMax > 35)
|
||||
lstrcpy(pszName, _("Edits the selected file(s) with Vim"));
|
||||
|
||||
return NOERROR;
|
||||
}
|
||||
|
||||
BOOL CALLBACK CShellExt::EnumWindowsProc(HWND hWnd, LPARAM lParam)
|
||||
{
|
||||
char temp[MAX_PATH];
|
||||
|
||||
// First do a bunch of check
|
||||
// No invisible window
|
||||
if (!IsWindowVisible(hWnd)) return TRUE;
|
||||
// No child window ???
|
||||
// if (GetParent(hWnd)) return TRUE;
|
||||
// Class name should be Vim, if failed to get class name, return
|
||||
if (GetClassName(hWnd, temp, sizeof(temp)) == 0)
|
||||
return TRUE;
|
||||
// Compare class name to that of vim, if not, return
|
||||
if (_strnicmp(temp, "vim", sizeof("vim")) != 0)
|
||||
return TRUE;
|
||||
// First check if the number of vim instance exceeds MAX_HWND
|
||||
CShellExt *cs = (CShellExt*) lParam;
|
||||
if (cs->m_cntOfHWnd >= MAX_HWND) return TRUE;
|
||||
// Now we get the vim window, put it into some kind of array
|
||||
cs->m_hWnd[cs->m_cntOfHWnd] = hWnd;
|
||||
cs->m_cntOfHWnd ++;
|
||||
|
||||
return TRUE; // continue enumeration (otherwise this would be false)
|
||||
}
|
||||
|
||||
#ifdef WIN32
|
||||
// This symbol is not defined in older versions of the SDK or Visual C++.
|
||||
|
||||
#ifndef VER_PLATFORM_WIN32_WINDOWS
|
||||
# define VER_PLATFORM_WIN32_WINDOWS 1
|
||||
#endif
|
||||
|
||||
static DWORD g_PlatformId;
|
||||
|
||||
//
|
||||
// Set g_PlatformId to VER_PLATFORM_WIN32_NT (NT) or
|
||||
// VER_PLATFORM_WIN32_WINDOWS (Win95).
|
||||
//
|
||||
static void
|
||||
PlatformId(void)
|
||||
{
|
||||
static int done = FALSE;
|
||||
|
||||
if (!done)
|
||||
{
|
||||
OSVERSIONINFO ovi;
|
||||
|
||||
ovi.dwOSVersionInfoSize = sizeof(ovi);
|
||||
GetVersionEx(&ovi);
|
||||
|
||||
g_PlatformId = ovi.dwPlatformId;
|
||||
done = TRUE;
|
||||
}
|
||||
}
|
||||
|
||||
# ifndef __BORLANDC__
|
||||
static char *
|
||||
searchpath(char *name)
|
||||
{
|
||||
static char widename[2 * MAX_PATH];
|
||||
static char location[2 * MAX_PATH + 2];
|
||||
|
||||
// There appears to be a bug in FindExecutableA() on Windows NT.
|
||||
// Use FindExecutableW() instead...
|
||||
PlatformId();
|
||||
if (g_PlatformId == VER_PLATFORM_WIN32_NT)
|
||||
{
|
||||
MultiByteToWideChar(CP_ACP, 0, (LPCTSTR)name, -1,
|
||||
(LPWSTR)widename, MAX_PATH);
|
||||
if (FindExecutableW((LPCWSTR)widename, (LPCWSTR)"",
|
||||
(LPWSTR)location) > (HINSTANCE)32)
|
||||
{
|
||||
WideCharToMultiByte(CP_ACP, 0, (LPWSTR)location, -1,
|
||||
(LPSTR)widename, 2 * MAX_PATH, NULL, NULL);
|
||||
return widename;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (FindExecutableA((LPCTSTR)name, (LPCTSTR)"",
|
||||
(LPTSTR)location) > (HINSTANCE)32)
|
||||
return location;
|
||||
}
|
||||
return "";
|
||||
}
|
||||
# endif
|
||||
#endif
|
||||
|
||||
STDMETHODIMP CShellExt::InvokeGvim(HWND hParent,
|
||||
LPCSTR pszWorkingDir,
|
||||
LPCSTR pszCmd,
|
||||
LPCSTR pszParam,
|
||||
int iShowCmd)
|
||||
{
|
||||
char m_szFileUserClickedOn[MAX_PATH];
|
||||
char cmdStr[MAX_PATH];
|
||||
UINT i;
|
||||
|
||||
for (i = 0; i < cbFiles; i++)
|
||||
{
|
||||
DragQueryFile((HDROP)medium.hGlobal,
|
||||
i,
|
||||
m_szFileUserClickedOn,
|
||||
sizeof(m_szFileUserClickedOn));
|
||||
|
||||
getGvimName(cmdStr, 0);
|
||||
strcat(cmdStr, " \"");
|
||||
|
||||
if ((strlen(cmdStr) + strlen(m_szFileUserClickedOn) + 2) < MAX_PATH)
|
||||
{
|
||||
strcat(cmdStr, m_szFileUserClickedOn);
|
||||
strcat(cmdStr, "\"");
|
||||
|
||||
STARTUPINFO si;
|
||||
PROCESS_INFORMATION pi;
|
||||
|
||||
ZeroMemory(&si, sizeof(si));
|
||||
si.cb = sizeof(si);
|
||||
|
||||
// Start the child process.
|
||||
if (!CreateProcess(NULL, // No module name (use command line).
|
||||
cmdStr, // Command line.
|
||||
NULL, // Process handle not inheritable.
|
||||
NULL, // Thread handle not inheritable.
|
||||
FALSE, // Set handle inheritance to FALSE.
|
||||
0, // No creation flags.
|
||||
NULL, // Use parent's environment block.
|
||||
NULL, // Use parent's starting directory.
|
||||
&si, // Pointer to STARTUPINFO structure.
|
||||
&pi) // Pointer to PROCESS_INFORMATION structure.
|
||||
)
|
||||
{
|
||||
MessageBox(
|
||||
hParent,
|
||||
_("Error creating process: Check if gvim is in your path!"),
|
||||
_("gvimext.dll error"),
|
||||
MB_OK);
|
||||
}
|
||||
else
|
||||
{
|
||||
CloseHandle( pi.hProcess );
|
||||
CloseHandle( pi.hThread );
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
MessageBox(
|
||||
hParent,
|
||||
_("Path length too long!"),
|
||||
_("gvimext.dll error"),
|
||||
MB_OK);
|
||||
}
|
||||
}
|
||||
|
||||
return NOERROR;
|
||||
}
|
||||
|
||||
|
||||
STDMETHODIMP CShellExt::InvokeSingleGvim(HWND hParent,
|
||||
LPCSTR pszWorkingDir,
|
||||
LPCSTR pszCmd,
|
||||
LPCSTR pszParam,
|
||||
int iShowCmd,
|
||||
int useDiff)
|
||||
{
|
||||
char m_szFileUserClickedOn[MAX_PATH];
|
||||
char *cmdStr;
|
||||
size_t cmdlen;
|
||||
size_t len;
|
||||
UINT i;
|
||||
|
||||
cmdlen = MAX_PATH;
|
||||
cmdStr = (char *)malloc(cmdlen);
|
||||
getGvimName(cmdStr, 0);
|
||||
if (useDiff)
|
||||
strcat(cmdStr, " -d");
|
||||
for (i = 0; i < cbFiles; i++)
|
||||
{
|
||||
DragQueryFile((HDROP)medium.hGlobal,
|
||||
i,
|
||||
m_szFileUserClickedOn,
|
||||
sizeof(m_szFileUserClickedOn));
|
||||
|
||||
len = strlen(cmdStr) + strlen(m_szFileUserClickedOn) + 4;
|
||||
if (len > cmdlen)
|
||||
{
|
||||
cmdlen = len + MAX_PATH;
|
||||
cmdStr = (char *)realloc(cmdStr, cmdlen);
|
||||
}
|
||||
strcat(cmdStr, " \"");
|
||||
strcat(cmdStr, m_szFileUserClickedOn);
|
||||
strcat(cmdStr, "\"");
|
||||
}
|
||||
|
||||
STARTUPINFO si;
|
||||
PROCESS_INFORMATION pi;
|
||||
|
||||
ZeroMemory(&si, sizeof(si));
|
||||
si.cb = sizeof(si);
|
||||
|
||||
// Start the child process.
|
||||
if (!CreateProcess(NULL, // No module name (use command line).
|
||||
cmdStr, // Command line.
|
||||
NULL, // Process handle not inheritable.
|
||||
NULL, // Thread handle not inheritable.
|
||||
FALSE, // Set handle inheritance to FALSE.
|
||||
0, // No creation flags.
|
||||
NULL, // Use parent's environment block.
|
||||
NULL, // Use parent's starting directory.
|
||||
&si, // Pointer to STARTUPINFO structure.
|
||||
&pi) // Pointer to PROCESS_INFORMATION structure.
|
||||
)
|
||||
{
|
||||
MessageBox(
|
||||
hParent,
|
||||
_("Error creating process: Check if gvim is in your path!"),
|
||||
_("gvimext.dll error"),
|
||||
MB_OK);
|
||||
}
|
||||
else
|
||||
{
|
||||
CloseHandle(pi.hProcess);
|
||||
CloseHandle(pi.hThread);
|
||||
}
|
||||
|
||||
free(cmdStr);
|
||||
|
||||
return NOERROR;
|
||||
}
|
Reference in New Issue
Block a user