xref: /vim-8.2.3635/src/GvimExt/gvimext.cpp (revision cf839732)
1f4b8e57fSBram Moolenaar /* vi:set ts=8 sts=4 sw=4:
2f4b8e57fSBram Moolenaar  *
3f4b8e57fSBram Moolenaar  * VIM - Vi IMproved	gvimext by Tianmiao Hu
4f4b8e57fSBram Moolenaar  *
5f4b8e57fSBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
6f4b8e57fSBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
7f4b8e57fSBram Moolenaar  */
8f4b8e57fSBram Moolenaar 
9f4b8e57fSBram Moolenaar /*
10f4b8e57fSBram Moolenaar  * gvimext is a DLL which is used for the "Edit with Vim" context menu
11f4b8e57fSBram Moolenaar  * extension.  It implements a MS defined interface with the Shell.
12f4b8e57fSBram Moolenaar  *
13f4b8e57fSBram Moolenaar  * If you have any questions or any suggestions concerning gvimext, please
14f4b8e57fSBram Moolenaar  * contact Tianmiao Hu: [email protected].
15f4b8e57fSBram Moolenaar  */
16f4b8e57fSBram Moolenaar 
17f4b8e57fSBram Moolenaar #include "gvimext.h"
18f4b8e57fSBram Moolenaar 
19f4b8e57fSBram Moolenaar #ifdef __BORLANDC__
20f4b8e57fSBram Moolenaar # include <dir.h>
21f4b8e57fSBram Moolenaar # ifndef _strnicmp
22f4b8e57fSBram Moolenaar #  define _strnicmp(a, b, c) strnicmp((a), (b), (c))
23f4b8e57fSBram Moolenaar # endif
24f4b8e57fSBram Moolenaar #else
25f4b8e57fSBram Moolenaar static char *searchpath(char *name);
26f4b8e57fSBram Moolenaar #endif
27f4b8e57fSBram Moolenaar 
28f4b8e57fSBram Moolenaar // Always get an error while putting the following stuff to the
29f4b8e57fSBram Moolenaar // gvimext.h file as class protected variables, give up and
30f4b8e57fSBram Moolenaar // declare them as global stuff
31f4b8e57fSBram Moolenaar FORMATETC fmte = {CF_HDROP,
32f4b8e57fSBram Moolenaar 		  (DVTARGETDEVICE FAR *)NULL,
33f4b8e57fSBram Moolenaar 		  DVASPECT_CONTENT,
34f4b8e57fSBram Moolenaar 		  -1,
35f4b8e57fSBram Moolenaar 		  TYMED_HGLOBAL
36f4b8e57fSBram Moolenaar 		 };
37f4b8e57fSBram Moolenaar STGMEDIUM medium;
38f4b8e57fSBram Moolenaar HRESULT hres = 0;
39f4b8e57fSBram Moolenaar UINT cbFiles = 0;
40f4b8e57fSBram Moolenaar 
41e759a7aaSBram Moolenaar /* The buffers size used to be MAX_PATH (256 bytes), but that's not always
42e759a7aaSBram Moolenaar  * enough */
43e759a7aaSBram Moolenaar #define BUFSIZE 1100
44e759a7aaSBram Moolenaar 
45f4b8e57fSBram Moolenaar //
46f4b8e57fSBram Moolenaar // Get the name of the Gvim executable to use, with the path.
47f4b8e57fSBram Moolenaar // When "runtime" is non-zero, we were called to find the runtime directory.
48e759a7aaSBram Moolenaar // Returns the path in name[BUFSIZE].  It's empty when it fails.
49f4b8e57fSBram Moolenaar //
50f4b8e57fSBram Moolenaar     static void
51f4b8e57fSBram Moolenaar getGvimName(char *name, int runtime)
52f4b8e57fSBram Moolenaar {
53f4b8e57fSBram Moolenaar     HKEY	keyhandle;
54f4b8e57fSBram Moolenaar     DWORD	hlen;
55f4b8e57fSBram Moolenaar 
56f4b8e57fSBram Moolenaar     // Get the location of gvim from the registry.
57f4b8e57fSBram Moolenaar     name[0] = 0;
58f4b8e57fSBram Moolenaar     if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, "Software\\Vim\\Gvim", 0,
59f4b8e57fSBram Moolenaar 				       KEY_READ, &keyhandle) == ERROR_SUCCESS)
60f4b8e57fSBram Moolenaar     {
61e759a7aaSBram Moolenaar 	hlen = BUFSIZE;
62f4b8e57fSBram Moolenaar 	if (RegQueryValueEx(keyhandle, "path", 0, NULL, (BYTE *)name, &hlen)
63f4b8e57fSBram Moolenaar 							     != ERROR_SUCCESS)
64f4b8e57fSBram Moolenaar 	    name[0] = 0;
65f4b8e57fSBram Moolenaar 	else
66f4b8e57fSBram Moolenaar 	    name[hlen] = 0;
67f4b8e57fSBram Moolenaar 	RegCloseKey(keyhandle);
68f4b8e57fSBram Moolenaar     }
69f4b8e57fSBram Moolenaar 
70f4b8e57fSBram Moolenaar     // Registry didn't work, use the search path.
71f4b8e57fSBram Moolenaar     if (name[0] == 0)
727baa45dbSBram Moolenaar 	strcpy(name, searchpath((char *)"gvim.exe"));
73f4b8e57fSBram Moolenaar 
74f4b8e57fSBram Moolenaar     if (!runtime)
75f4b8e57fSBram Moolenaar     {
76f4b8e57fSBram Moolenaar 	// Only when looking for the executable, not the runtime dir, we can
77f4b8e57fSBram Moolenaar 	// search for the batch file or a name without a path.
78f4b8e57fSBram Moolenaar 	if (name[0] == 0)
797baa45dbSBram Moolenaar 	    strcpy(name, searchpath((char *)"gvim.bat"));
80f4b8e57fSBram Moolenaar 	if (name[0] == 0)
81f4b8e57fSBram Moolenaar 	    strcpy(name, "gvim");	// finds gvim.bat or gvim.exe
82f4b8e57fSBram Moolenaar 
83f4b8e57fSBram Moolenaar 	// avoid that Vim tries to expand wildcards in the file names
84f4b8e57fSBram Moolenaar 	strcat(name, " --literal");
85f4b8e57fSBram Moolenaar     }
86f4b8e57fSBram Moolenaar }
87f4b8e57fSBram Moolenaar 
887e6d3bd3SBram Moolenaar     static void
897e6d3bd3SBram Moolenaar getGvimNameW(wchar_t *nameW)
907e6d3bd3SBram Moolenaar {
917e6d3bd3SBram Moolenaar     char *name;
927e6d3bd3SBram Moolenaar 
937e6d3bd3SBram Moolenaar     name = (char *)malloc(BUFSIZE);
947e6d3bd3SBram Moolenaar     getGvimName(name, 0);
957e6d3bd3SBram Moolenaar     mbstowcs(nameW, name, BUFSIZE);
967e6d3bd3SBram Moolenaar     free(name);
977e6d3bd3SBram Moolenaar }
987e6d3bd3SBram Moolenaar 
99f4b8e57fSBram Moolenaar //
100e759a7aaSBram Moolenaar // Get the Vim runtime directory into buf[BUFSIZE].
101f4b8e57fSBram Moolenaar // The result is empty when it failed.
102f4b8e57fSBram Moolenaar // When it works, the path ends in a slash or backslash.
103f4b8e57fSBram Moolenaar //
104f4b8e57fSBram Moolenaar     static void
105f4b8e57fSBram Moolenaar getRuntimeDir(char *buf)
106f4b8e57fSBram Moolenaar {
107f4b8e57fSBram Moolenaar     int		idx;
108f4b8e57fSBram Moolenaar 
109f4b8e57fSBram Moolenaar     getGvimName(buf, 1);
110f4b8e57fSBram Moolenaar     if (buf[0] != 0)
111f4b8e57fSBram Moolenaar     {
112f4b8e57fSBram Moolenaar 	// When no path found, use the search path to expand it.
113f4b8e57fSBram Moolenaar 	if (strchr(buf, '/') == NULL && strchr(buf, '\\') == NULL)
114f4b8e57fSBram Moolenaar 	    strcpy(buf, searchpath(buf));
115f4b8e57fSBram Moolenaar 
116f4b8e57fSBram Moolenaar 	// remove "gvim.exe" from the end
117a93fa7eeSBram Moolenaar 	for (idx = (int)strlen(buf) - 1; idx >= 0; idx--)
118f4b8e57fSBram Moolenaar 	    if (buf[idx] == '\\' || buf[idx] == '/')
119f4b8e57fSBram Moolenaar 	    {
120f4b8e57fSBram Moolenaar 		buf[idx + 1] = 0;
121f4b8e57fSBram Moolenaar 		break;
122f4b8e57fSBram Moolenaar 	    }
123f4b8e57fSBram Moolenaar     }
124f4b8e57fSBram Moolenaar }
125f4b8e57fSBram Moolenaar 
126f4b8e57fSBram Moolenaar //
127f4b8e57fSBram Moolenaar // GETTEXT: translated messages and menu entries
128f4b8e57fSBram Moolenaar //
129f4b8e57fSBram Moolenaar #ifndef FEAT_GETTEXT
130f4b8e57fSBram Moolenaar # define _(x)  x
131f4b8e57fSBram Moolenaar #else
132f4b8e57fSBram Moolenaar # define _(x)  (*dyn_libintl_gettext)(x)
133f4b8e57fSBram Moolenaar # define VIMPACKAGE "vim"
134f4b8e57fSBram Moolenaar # ifndef GETTEXT_DLL
135f4b8e57fSBram Moolenaar #  define GETTEXT_DLL "libintl.dll"
136f4b8e57fSBram Moolenaar # endif
137f4b8e57fSBram Moolenaar 
138f4b8e57fSBram Moolenaar // Dummy functions
139f4b8e57fSBram Moolenaar static char *null_libintl_gettext(const char *);
140f4b8e57fSBram Moolenaar static char *null_libintl_textdomain(const char *);
141f4b8e57fSBram Moolenaar static char *null_libintl_bindtextdomain(const char *, const char *);
142f4b8e57fSBram Moolenaar static int dyn_libintl_init(char *dir);
143f4b8e57fSBram Moolenaar static void dyn_libintl_end(void);
144f4b8e57fSBram Moolenaar 
145*cf839732SBram Moolenaar static wchar_t *oldenv = NULL;
146f4b8e57fSBram Moolenaar static HINSTANCE hLibintlDLL = 0;
147f4b8e57fSBram Moolenaar static char *(*dyn_libintl_gettext)(const char *) = null_libintl_gettext;
148f4b8e57fSBram Moolenaar static char *(*dyn_libintl_textdomain)(const char *) = null_libintl_textdomain;
149f4b8e57fSBram Moolenaar static char *(*dyn_libintl_bindtextdomain)(const char *, const char *)
150f4b8e57fSBram Moolenaar 						= null_libintl_bindtextdomain;
151f4b8e57fSBram Moolenaar 
152f4b8e57fSBram Moolenaar //
153f4b8e57fSBram Moolenaar // Attempt to load libintl.dll.  If it doesn't work, use dummy functions.
154f4b8e57fSBram Moolenaar // "dir" is the directory where the libintl.dll might be.
155f4b8e57fSBram Moolenaar // Return 1 for success, 0 for failure.
156f4b8e57fSBram Moolenaar //
157f4b8e57fSBram Moolenaar     static int
158f4b8e57fSBram Moolenaar dyn_libintl_init(char *dir)
159f4b8e57fSBram Moolenaar {
160f4b8e57fSBram Moolenaar     int		i;
161f4b8e57fSBram Moolenaar     static struct
162f4b8e57fSBram Moolenaar     {
163f4b8e57fSBram Moolenaar 	char	    *name;
164f4b8e57fSBram Moolenaar 	FARPROC	    *ptr;
165f4b8e57fSBram Moolenaar     } libintl_entry[] =
166f4b8e57fSBram Moolenaar     {
1677baa45dbSBram Moolenaar 	{(char *)"gettext",		(FARPROC*)&dyn_libintl_gettext},
1687baa45dbSBram Moolenaar 	{(char *)"textdomain",		(FARPROC*)&dyn_libintl_textdomain},
1697baa45dbSBram Moolenaar 	{(char *)"bindtextdomain",	(FARPROC*)&dyn_libintl_bindtextdomain},
170f4b8e57fSBram Moolenaar 	{NULL, NULL}
171f4b8e57fSBram Moolenaar     };
172f4b8e57fSBram Moolenaar 
173f4b8e57fSBram Moolenaar     // No need to initialize twice.
174f4b8e57fSBram Moolenaar     if (hLibintlDLL)
175f4b8e57fSBram Moolenaar 	return 1;
176f4b8e57fSBram Moolenaar 
177f4b8e57fSBram Moolenaar     // Load gettext library, first try the Vim runtime directory, then search
178f4b8e57fSBram Moolenaar     // the path.
179f4b8e57fSBram Moolenaar     strcat(dir, GETTEXT_DLL);
180f4b8e57fSBram Moolenaar     hLibintlDLL = LoadLibrary(dir);
181f4b8e57fSBram Moolenaar     if (!hLibintlDLL)
182f4b8e57fSBram Moolenaar     {
183f4b8e57fSBram Moolenaar 	hLibintlDLL = LoadLibrary(GETTEXT_DLL);
184f4b8e57fSBram Moolenaar 	if (!hLibintlDLL)
185f4b8e57fSBram Moolenaar 	    return 0;
186f4b8e57fSBram Moolenaar     }
187f4b8e57fSBram Moolenaar 
188f4b8e57fSBram Moolenaar     // Get the addresses of the functions we need.
189f4b8e57fSBram Moolenaar     for (i = 0; libintl_entry[i].name != NULL
190f4b8e57fSBram Moolenaar 					 && libintl_entry[i].ptr != NULL; ++i)
191f4b8e57fSBram Moolenaar     {
192f4b8e57fSBram Moolenaar 	if ((*libintl_entry[i].ptr = GetProcAddress(hLibintlDLL,
193f4b8e57fSBram Moolenaar 					      libintl_entry[i].name)) == NULL)
194f4b8e57fSBram Moolenaar 	{
195f4b8e57fSBram Moolenaar 	    dyn_libintl_end();
196f4b8e57fSBram Moolenaar 	    return 0;
197f4b8e57fSBram Moolenaar 	}
198f4b8e57fSBram Moolenaar     }
199f4b8e57fSBram Moolenaar     return 1;
200f4b8e57fSBram Moolenaar }
201f4b8e57fSBram Moolenaar 
202f4b8e57fSBram Moolenaar     static void
203f4b8e57fSBram Moolenaar dyn_libintl_end(void)
204f4b8e57fSBram Moolenaar {
205f4b8e57fSBram Moolenaar     if (hLibintlDLL)
206f4b8e57fSBram Moolenaar 	FreeLibrary(hLibintlDLL);
207f4b8e57fSBram Moolenaar     hLibintlDLL			= NULL;
208f4b8e57fSBram Moolenaar     dyn_libintl_gettext		= null_libintl_gettext;
209f4b8e57fSBram Moolenaar     dyn_libintl_textdomain	= null_libintl_textdomain;
210f4b8e57fSBram Moolenaar     dyn_libintl_bindtextdomain	= null_libintl_bindtextdomain;
211f4b8e57fSBram Moolenaar }
212f4b8e57fSBram Moolenaar 
213f4b8e57fSBram Moolenaar     static char *
214f4b8e57fSBram Moolenaar null_libintl_gettext(const char *msgid)
215f4b8e57fSBram Moolenaar {
216f4b8e57fSBram Moolenaar     return (char *)msgid;
217f4b8e57fSBram Moolenaar }
218f4b8e57fSBram Moolenaar 
219f4b8e57fSBram Moolenaar     static char *
220e6a91fd9SBram Moolenaar null_libintl_bindtextdomain(const char * /* domainname */, const char * /* dirname */)
221f4b8e57fSBram Moolenaar {
222f4b8e57fSBram Moolenaar     return NULL;
223f4b8e57fSBram Moolenaar }
224f4b8e57fSBram Moolenaar 
225f4b8e57fSBram Moolenaar     static char *
226e6a91fd9SBram Moolenaar null_libintl_textdomain(const char*  /* domainname */)
227f4b8e57fSBram Moolenaar {
228f4b8e57fSBram Moolenaar     return NULL;
229f4b8e57fSBram Moolenaar }
230f4b8e57fSBram Moolenaar 
231f4b8e57fSBram Moolenaar //
232f4b8e57fSBram Moolenaar // Setup for translating strings.
233f4b8e57fSBram Moolenaar //
234f4b8e57fSBram Moolenaar     static void
235f4b8e57fSBram Moolenaar dyn_gettext_load(void)
236f4b8e57fSBram Moolenaar {
237e759a7aaSBram Moolenaar     char    szBuff[BUFSIZE];
238e759a7aaSBram Moolenaar     char    szLang[BUFSIZE];
239f4b8e57fSBram Moolenaar     DWORD   len;
240f4b8e57fSBram Moolenaar     HKEY    keyhandle;
241f4b8e57fSBram Moolenaar     int	    gotlang = 0;
242f4b8e57fSBram Moolenaar 
243f4b8e57fSBram Moolenaar     strcpy(szLang, "LANG=");
244f4b8e57fSBram Moolenaar 
245f4b8e57fSBram Moolenaar     // First try getting the language from the registry, this can be
246f4b8e57fSBram Moolenaar     // used to overrule the system language.
247f4b8e57fSBram Moolenaar     if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, "Software\\Vim\\Gvim", 0,
248f4b8e57fSBram Moolenaar 				       KEY_READ, &keyhandle) == ERROR_SUCCESS)
249f4b8e57fSBram Moolenaar     {
250e759a7aaSBram Moolenaar 	len = BUFSIZE;
251f4b8e57fSBram Moolenaar 	if (RegQueryValueEx(keyhandle, "lang", 0, NULL, (BYTE*)szBuff, &len)
252f4b8e57fSBram Moolenaar 							     == ERROR_SUCCESS)
253f4b8e57fSBram Moolenaar 	{
254f4b8e57fSBram Moolenaar 	    szBuff[len] = 0;
255f4b8e57fSBram Moolenaar 	    strcat(szLang, szBuff);
256f4b8e57fSBram Moolenaar 	    gotlang = 1;
257f4b8e57fSBram Moolenaar 	}
258f4b8e57fSBram Moolenaar 	RegCloseKey(keyhandle);
259f4b8e57fSBram Moolenaar     }
260f4b8e57fSBram Moolenaar 
261f4b8e57fSBram Moolenaar     if (!gotlang && getenv("LANG") == NULL)
262f4b8e57fSBram Moolenaar     {
263f4b8e57fSBram Moolenaar 	// Get the language from the system.
264f4b8e57fSBram Moolenaar 	// Could use LOCALE_SISO639LANGNAME, but it's not in Win95.
265f4b8e57fSBram Moolenaar 	// LOCALE_SABBREVLANGNAME gives us three letters, like "enu", we use
266f4b8e57fSBram Moolenaar 	// only the first two.
267f4b8e57fSBram Moolenaar 	len = GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_SABBREVLANGNAME,
268e759a7aaSBram Moolenaar 						    (LPTSTR)szBuff, BUFSIZE);
269f4b8e57fSBram Moolenaar 	if (len >= 2 && _strnicmp(szBuff, "en", 2) != 0)
270f4b8e57fSBram Moolenaar 	{
271f4b8e57fSBram Moolenaar 	    // There are a few exceptions (probably more)
272f4b8e57fSBram Moolenaar 	    if (_strnicmp(szBuff, "cht", 3) == 0
273f4b8e57fSBram Moolenaar 					  || _strnicmp(szBuff, "zht", 3) == 0)
274f4b8e57fSBram Moolenaar 		strcpy(szBuff, "zh_TW");
275f4b8e57fSBram Moolenaar 	    else if (_strnicmp(szBuff, "chs", 3) == 0
276f4b8e57fSBram Moolenaar 					  || _strnicmp(szBuff, "zhc", 3) == 0)
277f4b8e57fSBram Moolenaar 		strcpy(szBuff, "zh_CN");
278f4b8e57fSBram Moolenaar 	    else if (_strnicmp(szBuff, "jp", 2) == 0)
279f4b8e57fSBram Moolenaar 		strcpy(szBuff, "ja");
280f4b8e57fSBram Moolenaar 	    else
281f4b8e57fSBram Moolenaar 		szBuff[2] = 0;	// truncate to two-letter code
282f4b8e57fSBram Moolenaar 	    strcat(szLang, szBuff);
283f4b8e57fSBram Moolenaar 	    gotlang = 1;
284f4b8e57fSBram Moolenaar 	}
285f4b8e57fSBram Moolenaar     }
286f4b8e57fSBram Moolenaar     if (gotlang)
287f4b8e57fSBram Moolenaar 	putenv(szLang);
288f4b8e57fSBram Moolenaar 
289f4b8e57fSBram Moolenaar     // Try to locate the runtime files.  The path is used to find libintl.dll
290f4b8e57fSBram Moolenaar     // and the vim.mo files.
291f4b8e57fSBram Moolenaar     getRuntimeDir(szBuff);
292f4b8e57fSBram Moolenaar     if (szBuff[0] != 0)
293f4b8e57fSBram Moolenaar     {
294a93fa7eeSBram Moolenaar 	len = (DWORD)strlen(szBuff);
295f4b8e57fSBram Moolenaar 	if (dyn_libintl_init(szBuff))
296f4b8e57fSBram Moolenaar 	{
297f4b8e57fSBram Moolenaar 	    strcpy(szBuff + len, "lang");
298f4b8e57fSBram Moolenaar 
299f4b8e57fSBram Moolenaar 	    (*dyn_libintl_bindtextdomain)(VIMPACKAGE, szBuff);
300f4b8e57fSBram Moolenaar 	    (*dyn_libintl_textdomain)(VIMPACKAGE);
301f4b8e57fSBram Moolenaar 	}
302f4b8e57fSBram Moolenaar     }
303f4b8e57fSBram Moolenaar }
304f4b8e57fSBram Moolenaar 
305f4b8e57fSBram Moolenaar     static void
306f4b8e57fSBram Moolenaar dyn_gettext_free(void)
307f4b8e57fSBram Moolenaar {
308f4b8e57fSBram Moolenaar     dyn_libintl_end();
309f4b8e57fSBram Moolenaar }
310f4b8e57fSBram Moolenaar #endif // FEAT_GETTEXT
311f4b8e57fSBram Moolenaar 
312f4b8e57fSBram Moolenaar //
313f4b8e57fSBram Moolenaar // Global variables
314f4b8e57fSBram Moolenaar //
315f4b8e57fSBram Moolenaar UINT      g_cRefThisDll = 0;    // Reference count of this DLL.
316f4b8e57fSBram Moolenaar HINSTANCE g_hmodThisDll = NULL;	// Handle to this DLL itself.
317f4b8e57fSBram Moolenaar 
318f4b8e57fSBram Moolenaar 
319f4b8e57fSBram Moolenaar //---------------------------------------------------------------------------
320f4b8e57fSBram Moolenaar // DllMain
321f4b8e57fSBram Moolenaar //---------------------------------------------------------------------------
322f4b8e57fSBram Moolenaar extern "C" int APIENTRY
323e6a91fd9SBram Moolenaar DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID  /* lpReserved */)
324f4b8e57fSBram Moolenaar {
325f4b8e57fSBram Moolenaar     switch (dwReason)
326f4b8e57fSBram Moolenaar     {
327f4b8e57fSBram Moolenaar     case DLL_PROCESS_ATTACH:
328f4b8e57fSBram Moolenaar 	// Extension DLL one-time initialization
329f4b8e57fSBram Moolenaar 	g_hmodThisDll = hInstance;
330f4b8e57fSBram Moolenaar 	break;
331f4b8e57fSBram Moolenaar 
332f4b8e57fSBram Moolenaar     case DLL_PROCESS_DETACH:
333f4b8e57fSBram Moolenaar 	break;
334f4b8e57fSBram Moolenaar     }
335f4b8e57fSBram Moolenaar 
336f4b8e57fSBram Moolenaar     return 1;   // ok
337f4b8e57fSBram Moolenaar }
338f4b8e57fSBram Moolenaar 
339f4b8e57fSBram Moolenaar     static void
340f4b8e57fSBram Moolenaar inc_cRefThisDLL()
341f4b8e57fSBram Moolenaar {
342f4b8e57fSBram Moolenaar #ifdef FEAT_GETTEXT
343*cf839732SBram Moolenaar     if (g_cRefThisDll == 0) {
344f4b8e57fSBram Moolenaar 	dyn_gettext_load();
345*cf839732SBram Moolenaar 	oldenv = GetEnvironmentStringsW();
346*cf839732SBram Moolenaar     }
347f4b8e57fSBram Moolenaar #endif
348f4b8e57fSBram Moolenaar     InterlockedIncrement((LPLONG)&g_cRefThisDll);
349f4b8e57fSBram Moolenaar }
350f4b8e57fSBram Moolenaar 
351f4b8e57fSBram Moolenaar     static void
352f4b8e57fSBram Moolenaar dec_cRefThisDLL()
353f4b8e57fSBram Moolenaar {
354f4b8e57fSBram Moolenaar #ifdef FEAT_GETTEXT
355*cf839732SBram Moolenaar     if (InterlockedDecrement((LPLONG)&g_cRefThisDll) == 0) {
356f4b8e57fSBram Moolenaar 	dyn_gettext_free();
357*cf839732SBram Moolenaar 	if (oldenv != NULL) {
358*cf839732SBram Moolenaar 	    FreeEnvironmentStringsW(oldenv);
359*cf839732SBram Moolenaar 	    oldenv = NULL;
360*cf839732SBram Moolenaar 	}
361*cf839732SBram Moolenaar     }
362f4b8e57fSBram Moolenaar #else
363f4b8e57fSBram Moolenaar     InterlockedDecrement((LPLONG)&g_cRefThisDll);
364f4b8e57fSBram Moolenaar #endif
365f4b8e57fSBram Moolenaar }
366f4b8e57fSBram Moolenaar 
367f4b8e57fSBram Moolenaar //---------------------------------------------------------------------------
368f4b8e57fSBram Moolenaar // DllCanUnloadNow
369f4b8e57fSBram Moolenaar //---------------------------------------------------------------------------
370f4b8e57fSBram Moolenaar 
371f4b8e57fSBram Moolenaar STDAPI DllCanUnloadNow(void)
372f4b8e57fSBram Moolenaar {
373f4b8e57fSBram Moolenaar     return (g_cRefThisDll == 0 ? S_OK : S_FALSE);
374f4b8e57fSBram Moolenaar }
375f4b8e57fSBram Moolenaar 
376f4b8e57fSBram Moolenaar STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppvOut)
377f4b8e57fSBram Moolenaar {
378f4b8e57fSBram Moolenaar     *ppvOut = NULL;
379f4b8e57fSBram Moolenaar 
380f4b8e57fSBram Moolenaar     if (IsEqualIID(rclsid, CLSID_ShellExtension))
381f4b8e57fSBram Moolenaar     {
382f4b8e57fSBram Moolenaar 	CShellExtClassFactory *pcf = new CShellExtClassFactory;
383f4b8e57fSBram Moolenaar 
384f4b8e57fSBram Moolenaar 	return pcf->QueryInterface(riid, ppvOut);
385f4b8e57fSBram Moolenaar     }
386f4b8e57fSBram Moolenaar 
387f4b8e57fSBram Moolenaar     return CLASS_E_CLASSNOTAVAILABLE;
388f4b8e57fSBram Moolenaar }
389f4b8e57fSBram Moolenaar 
390f4b8e57fSBram Moolenaar CShellExtClassFactory::CShellExtClassFactory()
391f4b8e57fSBram Moolenaar {
392f4b8e57fSBram Moolenaar     m_cRef = 0L;
393f4b8e57fSBram Moolenaar 
394f4b8e57fSBram Moolenaar     inc_cRefThisDLL();
395f4b8e57fSBram Moolenaar }
396f4b8e57fSBram Moolenaar 
397f4b8e57fSBram Moolenaar CShellExtClassFactory::~CShellExtClassFactory()
398f4b8e57fSBram Moolenaar {
399f4b8e57fSBram Moolenaar     dec_cRefThisDLL();
400f4b8e57fSBram Moolenaar }
401f4b8e57fSBram Moolenaar 
402f4b8e57fSBram Moolenaar STDMETHODIMP CShellExtClassFactory::QueryInterface(REFIID riid,
403f4b8e57fSBram Moolenaar 						   LPVOID FAR *ppv)
404f4b8e57fSBram Moolenaar {
405f4b8e57fSBram Moolenaar     *ppv = NULL;
406f4b8e57fSBram Moolenaar 
407f4b8e57fSBram Moolenaar     // Any interface on this object is the object pointer
408f4b8e57fSBram Moolenaar 
409f4b8e57fSBram Moolenaar     if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IClassFactory))
410f4b8e57fSBram Moolenaar     {
411f4b8e57fSBram Moolenaar 	*ppv = (LPCLASSFACTORY)this;
412f4b8e57fSBram Moolenaar 
413f4b8e57fSBram Moolenaar 	AddRef();
414f4b8e57fSBram Moolenaar 
415f4b8e57fSBram Moolenaar 	return NOERROR;
416f4b8e57fSBram Moolenaar     }
417f4b8e57fSBram Moolenaar 
418f4b8e57fSBram Moolenaar     return E_NOINTERFACE;
419f4b8e57fSBram Moolenaar }
420f4b8e57fSBram Moolenaar 
421f4b8e57fSBram Moolenaar STDMETHODIMP_(ULONG) CShellExtClassFactory::AddRef()
422f4b8e57fSBram Moolenaar {
423f4b8e57fSBram Moolenaar     return InterlockedIncrement((LPLONG)&m_cRef);
424f4b8e57fSBram Moolenaar }
425f4b8e57fSBram Moolenaar 
426f4b8e57fSBram Moolenaar STDMETHODIMP_(ULONG) CShellExtClassFactory::Release()
427f4b8e57fSBram Moolenaar {
428f4b8e57fSBram Moolenaar     if (InterlockedDecrement((LPLONG)&m_cRef))
429f4b8e57fSBram Moolenaar 	return m_cRef;
430f4b8e57fSBram Moolenaar 
431f4b8e57fSBram Moolenaar     delete this;
432f4b8e57fSBram Moolenaar 
433f4b8e57fSBram Moolenaar     return 0L;
434f4b8e57fSBram Moolenaar }
435f4b8e57fSBram Moolenaar 
436f4b8e57fSBram Moolenaar STDMETHODIMP CShellExtClassFactory::CreateInstance(LPUNKNOWN pUnkOuter,
437f4b8e57fSBram Moolenaar 						      REFIID riid,
438f4b8e57fSBram Moolenaar 						      LPVOID *ppvObj)
439f4b8e57fSBram Moolenaar {
440f4b8e57fSBram Moolenaar     *ppvObj = NULL;
441f4b8e57fSBram Moolenaar 
442f4b8e57fSBram Moolenaar     // Shell extensions typically don't support aggregation (inheritance)
443f4b8e57fSBram Moolenaar 
444f4b8e57fSBram Moolenaar     if (pUnkOuter)
445f4b8e57fSBram Moolenaar 	return CLASS_E_NOAGGREGATION;
446f4b8e57fSBram Moolenaar 
447f4b8e57fSBram Moolenaar     // Create the main shell extension object.  The shell will then call
448f4b8e57fSBram Moolenaar     // QueryInterface with IID_IShellExtInit--this is how shell extensions are
449f4b8e57fSBram Moolenaar     // initialized.
450f4b8e57fSBram Moolenaar 
451f4b8e57fSBram Moolenaar     LPCSHELLEXT pShellExt = new CShellExt();  //Create the CShellExt object
452f4b8e57fSBram Moolenaar 
453f4b8e57fSBram Moolenaar     if (NULL == pShellExt)
454f4b8e57fSBram Moolenaar 	return E_OUTOFMEMORY;
455f4b8e57fSBram Moolenaar 
456f4b8e57fSBram Moolenaar     return pShellExt->QueryInterface(riid, ppvObj);
457f4b8e57fSBram Moolenaar }
458f4b8e57fSBram Moolenaar 
459f4b8e57fSBram Moolenaar 
460e6a91fd9SBram Moolenaar STDMETHODIMP CShellExtClassFactory::LockServer(BOOL  /* fLock */)
461f4b8e57fSBram Moolenaar {
462f4b8e57fSBram Moolenaar     return NOERROR;
463f4b8e57fSBram Moolenaar }
464f4b8e57fSBram Moolenaar 
465f4b8e57fSBram Moolenaar // *********************** CShellExt *************************
466f4b8e57fSBram Moolenaar CShellExt::CShellExt()
467f4b8e57fSBram Moolenaar {
468f4b8e57fSBram Moolenaar     m_cRef = 0L;
469f4b8e57fSBram Moolenaar     m_pDataObj = NULL;
470f4b8e57fSBram Moolenaar 
471f4b8e57fSBram Moolenaar     inc_cRefThisDLL();
472f4b8e57fSBram Moolenaar }
473f4b8e57fSBram Moolenaar 
474f4b8e57fSBram Moolenaar CShellExt::~CShellExt()
475f4b8e57fSBram Moolenaar {
476f4b8e57fSBram Moolenaar     if (m_pDataObj)
477f4b8e57fSBram Moolenaar 	m_pDataObj->Release();
478f4b8e57fSBram Moolenaar 
479f4b8e57fSBram Moolenaar     dec_cRefThisDLL();
480f4b8e57fSBram Moolenaar }
481f4b8e57fSBram Moolenaar 
482f4b8e57fSBram Moolenaar STDMETHODIMP CShellExt::QueryInterface(REFIID riid, LPVOID FAR *ppv)
483f4b8e57fSBram Moolenaar {
484f4b8e57fSBram Moolenaar     *ppv = NULL;
485f4b8e57fSBram Moolenaar 
486f4b8e57fSBram Moolenaar     if (IsEqualIID(riid, IID_IShellExtInit) || IsEqualIID(riid, IID_IUnknown))
487f4b8e57fSBram Moolenaar     {
488f4b8e57fSBram Moolenaar 	*ppv = (LPSHELLEXTINIT)this;
489f4b8e57fSBram Moolenaar     }
490f4b8e57fSBram Moolenaar     else if (IsEqualIID(riid, IID_IContextMenu))
491f4b8e57fSBram Moolenaar     {
492f4b8e57fSBram Moolenaar 	*ppv = (LPCONTEXTMENU)this;
493f4b8e57fSBram Moolenaar     }
494f4b8e57fSBram Moolenaar 
495f4b8e57fSBram Moolenaar     if (*ppv)
496f4b8e57fSBram Moolenaar     {
497f4b8e57fSBram Moolenaar 	AddRef();
498f4b8e57fSBram Moolenaar 
499f4b8e57fSBram Moolenaar 	return NOERROR;
500f4b8e57fSBram Moolenaar     }
501f4b8e57fSBram Moolenaar 
502f4b8e57fSBram Moolenaar     return E_NOINTERFACE;
503f4b8e57fSBram Moolenaar }
504f4b8e57fSBram Moolenaar 
505f4b8e57fSBram Moolenaar STDMETHODIMP_(ULONG) CShellExt::AddRef()
506f4b8e57fSBram Moolenaar {
507f4b8e57fSBram Moolenaar     return InterlockedIncrement((LPLONG)&m_cRef);
508f4b8e57fSBram Moolenaar }
509f4b8e57fSBram Moolenaar 
510f4b8e57fSBram Moolenaar STDMETHODIMP_(ULONG) CShellExt::Release()
511f4b8e57fSBram Moolenaar {
512f4b8e57fSBram Moolenaar 
513f4b8e57fSBram Moolenaar     if (InterlockedDecrement((LPLONG)&m_cRef))
514f4b8e57fSBram Moolenaar 	return m_cRef;
515f4b8e57fSBram Moolenaar 
516f4b8e57fSBram Moolenaar     delete this;
517f4b8e57fSBram Moolenaar 
518f4b8e57fSBram Moolenaar     return 0L;
519f4b8e57fSBram Moolenaar }
520f4b8e57fSBram Moolenaar 
521f4b8e57fSBram Moolenaar 
522f4b8e57fSBram Moolenaar //
523f4b8e57fSBram Moolenaar //  FUNCTION: CShellExt::Initialize(LPCITEMIDLIST, LPDATAOBJECT, HKEY)
524f4b8e57fSBram Moolenaar //
525f4b8e57fSBram Moolenaar //  PURPOSE: Called by the shell when initializing a context menu or property
526f4b8e57fSBram Moolenaar //	     sheet extension.
527f4b8e57fSBram Moolenaar //
528f4b8e57fSBram Moolenaar //  PARAMETERS:
529f4b8e57fSBram Moolenaar //    pIDFolder - Specifies the parent folder
530f4b8e57fSBram Moolenaar //    pDataObj  - Spefifies the set of items selected in that folder.
531f4b8e57fSBram Moolenaar //    hRegKey   - Specifies the type of the focused item in the selection.
532f4b8e57fSBram Moolenaar //
533f4b8e57fSBram Moolenaar //  RETURN VALUE:
534f4b8e57fSBram Moolenaar //
535f4b8e57fSBram Moolenaar //    NOERROR in all cases.
536f4b8e57fSBram Moolenaar //
537f4b8e57fSBram Moolenaar //  COMMENTS:   Note that at the time this function is called, we don't know
538f4b8e57fSBram Moolenaar //		(or care) what type of shell extension is being initialized.
539f4b8e57fSBram Moolenaar //		It could be a context menu or a property sheet.
540f4b8e57fSBram Moolenaar //
541f4b8e57fSBram Moolenaar 
542e6a91fd9SBram Moolenaar STDMETHODIMP CShellExt::Initialize(LPCITEMIDLIST  /* pIDFolder */,
543f4b8e57fSBram Moolenaar 				   LPDATAOBJECT pDataObj,
544e6a91fd9SBram Moolenaar 				   HKEY  /* hRegKey */)
545f4b8e57fSBram Moolenaar {
546f4b8e57fSBram Moolenaar     // Initialize can be called more than once
547f4b8e57fSBram Moolenaar     if (m_pDataObj)
548f4b8e57fSBram Moolenaar 	m_pDataObj->Release();
549f4b8e57fSBram Moolenaar 
550f4b8e57fSBram Moolenaar     // duplicate the object pointer and registry handle
551f4b8e57fSBram Moolenaar 
552f4b8e57fSBram Moolenaar     if (pDataObj)
553f4b8e57fSBram Moolenaar     {
554f4b8e57fSBram Moolenaar 	m_pDataObj = pDataObj;
555f4b8e57fSBram Moolenaar 	pDataObj->AddRef();
556f4b8e57fSBram Moolenaar     }
557f4b8e57fSBram Moolenaar 
558f4b8e57fSBram Moolenaar     return NOERROR;
559f4b8e57fSBram Moolenaar }
560f4b8e57fSBram Moolenaar 
561f4b8e57fSBram Moolenaar 
562f4b8e57fSBram Moolenaar //
563f4b8e57fSBram Moolenaar //  FUNCTION: CShellExt::QueryContextMenu(HMENU, UINT, UINT, UINT, UINT)
564f4b8e57fSBram Moolenaar //
565f4b8e57fSBram Moolenaar //  PURPOSE: Called by the shell just before the context menu is displayed.
566f4b8e57fSBram Moolenaar //	     This is where you add your specific menu items.
567f4b8e57fSBram Moolenaar //
568f4b8e57fSBram Moolenaar //  PARAMETERS:
569f4b8e57fSBram Moolenaar //    hMenu      - Handle to the context menu
570f4b8e57fSBram Moolenaar //    indexMenu  - Index of where to begin inserting menu items
571f4b8e57fSBram Moolenaar //    idCmdFirst - Lowest value for new menu ID's
572f4b8e57fSBram Moolenaar //    idCmtLast  - Highest value for new menu ID's
573f4b8e57fSBram Moolenaar //    uFlags     - Specifies the context of the menu event
574f4b8e57fSBram Moolenaar //
575f4b8e57fSBram Moolenaar //  RETURN VALUE:
576f4b8e57fSBram Moolenaar //
577f4b8e57fSBram Moolenaar //
578f4b8e57fSBram Moolenaar //  COMMENTS:
579f4b8e57fSBram Moolenaar //
580f4b8e57fSBram Moolenaar 
581f4b8e57fSBram Moolenaar STDMETHODIMP CShellExt::QueryContextMenu(HMENU hMenu,
582f4b8e57fSBram Moolenaar 					 UINT indexMenu,
583f4b8e57fSBram Moolenaar 					 UINT idCmdFirst,
584e6a91fd9SBram Moolenaar 					 UINT  /* idCmdLast */,
585e6a91fd9SBram Moolenaar 					 UINT  /* uFlags */)
586f4b8e57fSBram Moolenaar {
587f4b8e57fSBram Moolenaar     UINT idCmd = idCmdFirst;
588f4b8e57fSBram Moolenaar 
589f4b8e57fSBram Moolenaar     hres = m_pDataObj->GetData(&fmte, &medium);
590f4b8e57fSBram Moolenaar     if (medium.hGlobal)
591f4b8e57fSBram Moolenaar 	cbFiles = DragQueryFile((HDROP)medium.hGlobal, (UINT)-1, 0, 0);
592f4b8e57fSBram Moolenaar 
593f4b8e57fSBram Moolenaar     // InsertMenu(hMenu, indexMenu++, MF_SEPARATOR|MF_BYPOSITION, 0, NULL);
594f4b8e57fSBram Moolenaar 
595f4b8e57fSBram Moolenaar     // Initialize m_cntOfHWnd to 0
596f4b8e57fSBram Moolenaar     m_cntOfHWnd = 0;
597ce35c88aSBram Moolenaar 
598ce35c88aSBram Moolenaar     HKEY keyhandle;
599ce35c88aSBram Moolenaar     bool showExisting = true;
600ce35c88aSBram Moolenaar 
601ce35c88aSBram Moolenaar     // Check whether "Edit with existing Vim" entries are disabled.
602ce35c88aSBram Moolenaar     if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, "Software\\Vim\\Gvim", 0,
603ce35c88aSBram Moolenaar 				       KEY_READ, &keyhandle) == ERROR_SUCCESS)
604ce35c88aSBram Moolenaar     {
605ce35c88aSBram Moolenaar 	if (RegQueryValueEx(keyhandle, "DisableEditWithExisting", 0, NULL,
606ce35c88aSBram Moolenaar 						 NULL, NULL) == ERROR_SUCCESS)
607ce35c88aSBram Moolenaar 	    showExisting = false;
608ce35c88aSBram Moolenaar 	RegCloseKey(keyhandle);
609ce35c88aSBram Moolenaar     }
610ce35c88aSBram Moolenaar 
611ce35c88aSBram Moolenaar     // Retrieve all the vim instances, unless disabled.
612ce35c88aSBram Moolenaar     if (showExisting)
613f4b8e57fSBram Moolenaar 	EnumWindows(EnumWindowsProc, (LPARAM)this);
614f4b8e57fSBram Moolenaar 
615f4b8e57fSBram Moolenaar     if (cbFiles > 1)
616f4b8e57fSBram Moolenaar     {
617f4b8e57fSBram Moolenaar 	InsertMenu(hMenu,
618f4b8e57fSBram Moolenaar 		indexMenu++,
619f4b8e57fSBram Moolenaar 		MF_STRING|MF_BYPOSITION,
620f4b8e57fSBram Moolenaar 		idCmd++,
621f4b8e57fSBram Moolenaar 		_("Edit with &multiple Vims"));
622f4b8e57fSBram Moolenaar 
623f4b8e57fSBram Moolenaar 	InsertMenu(hMenu,
624f4b8e57fSBram Moolenaar 		indexMenu++,
625f4b8e57fSBram Moolenaar 		MF_STRING|MF_BYPOSITION,
626f4b8e57fSBram Moolenaar 		idCmd++,
627f4b8e57fSBram Moolenaar 		_("Edit with single &Vim"));
628f4b8e57fSBram Moolenaar 
629f4b8e57fSBram Moolenaar 	if (cbFiles <= 4)
630f4b8e57fSBram Moolenaar 	{
631f4b8e57fSBram Moolenaar 	    // Can edit up to 4 files in diff mode
632f4b8e57fSBram Moolenaar 	    InsertMenu(hMenu,
633f4b8e57fSBram Moolenaar 		    indexMenu++,
634f4b8e57fSBram Moolenaar 		    MF_STRING|MF_BYPOSITION,
635f4b8e57fSBram Moolenaar 		    idCmd++,
63615d0a8c7SBram Moolenaar 		    _("Diff with Vim"));
637f4b8e57fSBram Moolenaar 	    m_edit_existing_off = 3;
638f4b8e57fSBram Moolenaar 	}
639f4b8e57fSBram Moolenaar 	else
640f4b8e57fSBram Moolenaar 	    m_edit_existing_off = 2;
641f4b8e57fSBram Moolenaar 
642f4b8e57fSBram Moolenaar     }
643f4b8e57fSBram Moolenaar     else
644f4b8e57fSBram Moolenaar     {
645f4b8e57fSBram Moolenaar 	InsertMenu(hMenu,
646f4b8e57fSBram Moolenaar 		indexMenu++,
647f4b8e57fSBram Moolenaar 		MF_STRING|MF_BYPOSITION,
648f4b8e57fSBram Moolenaar 		idCmd++,
649f4b8e57fSBram Moolenaar 		_("Edit with &Vim"));
650f4b8e57fSBram Moolenaar 	m_edit_existing_off = 1;
651f4b8e57fSBram Moolenaar     }
652f4b8e57fSBram Moolenaar 
653f4b8e57fSBram Moolenaar     // Now display all the vim instances
654f4b8e57fSBram Moolenaar     for (int i = 0; i < m_cntOfHWnd; i++)
655f4b8e57fSBram Moolenaar     {
656e759a7aaSBram Moolenaar 	char title[BUFSIZE];
657e759a7aaSBram Moolenaar 	char temp[BUFSIZE];
658f4b8e57fSBram Moolenaar 
659f4b8e57fSBram Moolenaar 	// Obtain window title, continue if can not
660e759a7aaSBram Moolenaar 	if (GetWindowText(m_hWnd[i], title, BUFSIZE - 1) == 0)
661f4b8e57fSBram Moolenaar 	    continue;
662f4b8e57fSBram Moolenaar 	// Truncate the title before the path, keep the file name
663f4b8e57fSBram Moolenaar 	char *pos = strchr(title, '(');
664f4b8e57fSBram Moolenaar 	if (pos != NULL)
665f4b8e57fSBram Moolenaar 	{
666f4b8e57fSBram Moolenaar 	    if (pos > title && pos[-1] == ' ')
667f4b8e57fSBram Moolenaar 		--pos;
668f4b8e57fSBram Moolenaar 	    *pos = 0;
669f4b8e57fSBram Moolenaar 	}
670f4b8e57fSBram Moolenaar 	// Now concatenate
671e759a7aaSBram Moolenaar 	strncpy(temp, _("Edit with existing Vim - "), BUFSIZE - 1);
672dc7e00e8SBram Moolenaar 	temp[BUFSIZE - 1] = '\0';
673dc7e00e8SBram Moolenaar 	strncat(temp, title, BUFSIZE - 1 - strlen(temp));
674dc7e00e8SBram Moolenaar 	temp[BUFSIZE - 1] = '\0';
675f4b8e57fSBram Moolenaar 	InsertMenu(hMenu,
676f4b8e57fSBram Moolenaar 		indexMenu++,
677f4b8e57fSBram Moolenaar 		MF_STRING|MF_BYPOSITION,
678f4b8e57fSBram Moolenaar 		idCmd++,
679f4b8e57fSBram Moolenaar 		temp);
680f4b8e57fSBram Moolenaar     }
681f4b8e57fSBram Moolenaar     // InsertMenu(hMenu, indexMenu++, MF_SEPARATOR|MF_BYPOSITION, 0, NULL);
682f4b8e57fSBram Moolenaar 
683f4b8e57fSBram Moolenaar     // Must return number of menu items we added.
684f4b8e57fSBram Moolenaar     return ResultFromShort(idCmd-idCmdFirst);
685f4b8e57fSBram Moolenaar }
686f4b8e57fSBram Moolenaar 
687f4b8e57fSBram Moolenaar //
688f4b8e57fSBram Moolenaar //  FUNCTION: CShellExt::InvokeCommand(LPCMINVOKECOMMANDINFO)
689f4b8e57fSBram Moolenaar //
690f4b8e57fSBram Moolenaar //  PURPOSE: Called by the shell after the user has selected on of the
691f4b8e57fSBram Moolenaar //	     menu items that was added in QueryContextMenu().
692f4b8e57fSBram Moolenaar //
693f4b8e57fSBram Moolenaar //  PARAMETERS:
694f4b8e57fSBram Moolenaar //    lpcmi - Pointer to an CMINVOKECOMMANDINFO structure
695f4b8e57fSBram Moolenaar //
696f4b8e57fSBram Moolenaar //  RETURN VALUE:
697f4b8e57fSBram Moolenaar //
698f4b8e57fSBram Moolenaar //
699f4b8e57fSBram Moolenaar //  COMMENTS:
700f4b8e57fSBram Moolenaar //
701f4b8e57fSBram Moolenaar 
702f4b8e57fSBram Moolenaar STDMETHODIMP CShellExt::InvokeCommand(LPCMINVOKECOMMANDINFO lpcmi)
703f4b8e57fSBram Moolenaar {
704f4b8e57fSBram Moolenaar     HRESULT hr = E_INVALIDARG;
705f4b8e57fSBram Moolenaar 
706f4b8e57fSBram Moolenaar     // If HIWORD(lpcmi->lpVerb) then we have been called programmatically
707f4b8e57fSBram Moolenaar     // and lpVerb is a command that should be invoked.  Otherwise, the shell
708f4b8e57fSBram Moolenaar     // has called us, and LOWORD(lpcmi->lpVerb) is the menu ID the user has
709f4b8e57fSBram Moolenaar     // selected.  Actually, it's (menu ID - idCmdFirst) from QueryContextMenu().
710f4b8e57fSBram Moolenaar     if (!HIWORD(lpcmi->lpVerb))
711f4b8e57fSBram Moolenaar     {
712f4b8e57fSBram Moolenaar 	UINT idCmd = LOWORD(lpcmi->lpVerb);
713f4b8e57fSBram Moolenaar 
714f4b8e57fSBram Moolenaar 	if (idCmd >= m_edit_existing_off)
715f4b8e57fSBram Moolenaar 	{
716f4b8e57fSBram Moolenaar 	    // Existing with vim instance
717f4b8e57fSBram Moolenaar 	    hr = PushToWindow(lpcmi->hwnd,
718f4b8e57fSBram Moolenaar 		    lpcmi->lpDirectory,
719f4b8e57fSBram Moolenaar 		    lpcmi->lpVerb,
720f4b8e57fSBram Moolenaar 		    lpcmi->lpParameters,
721f4b8e57fSBram Moolenaar 		    lpcmi->nShow,
722f4b8e57fSBram Moolenaar 		    idCmd - m_edit_existing_off);
723f4b8e57fSBram Moolenaar 	}
724f4b8e57fSBram Moolenaar 	else
725f4b8e57fSBram Moolenaar 	{
726f4b8e57fSBram Moolenaar 	    switch (idCmd)
727f4b8e57fSBram Moolenaar 	    {
728f4b8e57fSBram Moolenaar 		case 0:
729f4b8e57fSBram Moolenaar 		    hr = InvokeGvim(lpcmi->hwnd,
730f4b8e57fSBram Moolenaar 			    lpcmi->lpDirectory,
731f4b8e57fSBram Moolenaar 			    lpcmi->lpVerb,
732f4b8e57fSBram Moolenaar 			    lpcmi->lpParameters,
733f4b8e57fSBram Moolenaar 			    lpcmi->nShow);
734f4b8e57fSBram Moolenaar 		    break;
735f4b8e57fSBram Moolenaar 		case 1:
736f4b8e57fSBram Moolenaar 		    hr = InvokeSingleGvim(lpcmi->hwnd,
737f4b8e57fSBram Moolenaar 			    lpcmi->lpDirectory,
738f4b8e57fSBram Moolenaar 			    lpcmi->lpVerb,
739f4b8e57fSBram Moolenaar 			    lpcmi->lpParameters,
740f4b8e57fSBram Moolenaar 			    lpcmi->nShow,
741f4b8e57fSBram Moolenaar 			    0);
742f4b8e57fSBram Moolenaar 		    break;
743f4b8e57fSBram Moolenaar 		case 2:
744f4b8e57fSBram Moolenaar 		    hr = InvokeSingleGvim(lpcmi->hwnd,
745f4b8e57fSBram Moolenaar 			    lpcmi->lpDirectory,
746f4b8e57fSBram Moolenaar 			    lpcmi->lpVerb,
747f4b8e57fSBram Moolenaar 			    lpcmi->lpParameters,
748f4b8e57fSBram Moolenaar 			    lpcmi->nShow,
749f4b8e57fSBram Moolenaar 			    1);
750f4b8e57fSBram Moolenaar 		    break;
751f4b8e57fSBram Moolenaar 	    }
752f4b8e57fSBram Moolenaar 	}
753f4b8e57fSBram Moolenaar     }
754f4b8e57fSBram Moolenaar     return hr;
755f4b8e57fSBram Moolenaar }
756f4b8e57fSBram Moolenaar 
757e6a91fd9SBram Moolenaar STDMETHODIMP CShellExt::PushToWindow(HWND  /* hParent */,
758e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszWorkingDir */,
759e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszCmd */,
760e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszParam */,
761e6a91fd9SBram Moolenaar 				   int  /* iShowCmd */,
762f4b8e57fSBram Moolenaar 				   int idHWnd)
763f4b8e57fSBram Moolenaar {
764f4b8e57fSBram Moolenaar     HWND hWnd = m_hWnd[idHWnd];
765f4b8e57fSBram Moolenaar 
766f4b8e57fSBram Moolenaar     // Show and bring vim instance to foreground
767f4b8e57fSBram Moolenaar     if (IsIconic(hWnd) != 0)
768f4b8e57fSBram Moolenaar 	ShowWindow(hWnd, SW_RESTORE);
769f4b8e57fSBram Moolenaar     else
770f4b8e57fSBram Moolenaar 	ShowWindow(hWnd, SW_SHOW);
771f4b8e57fSBram Moolenaar     SetForegroundWindow(hWnd);
772f4b8e57fSBram Moolenaar 
773f4b8e57fSBram Moolenaar     // Post the selected files to the vim instance
774f4b8e57fSBram Moolenaar     PostMessage(hWnd, WM_DROPFILES, (WPARAM)medium.hGlobal, 0);
775f4b8e57fSBram Moolenaar 
776f4b8e57fSBram Moolenaar     return NOERROR;
777f4b8e57fSBram Moolenaar }
778f4b8e57fSBram Moolenaar 
779e6a91fd9SBram Moolenaar STDMETHODIMP CShellExt::GetCommandString(UINT_PTR  /* idCmd */,
780f4b8e57fSBram Moolenaar 					 UINT uFlags,
781e6a91fd9SBram Moolenaar 					 UINT FAR * /* reserved */,
782f4b8e57fSBram Moolenaar 					 LPSTR pszName,
783f4b8e57fSBram Moolenaar 					 UINT cchMax)
784f4b8e57fSBram Moolenaar {
785f4b8e57fSBram Moolenaar     if (uFlags == GCS_HELPTEXT && cchMax > 35)
786f4b8e57fSBram Moolenaar 	lstrcpy(pszName, _("Edits the selected file(s) with Vim"));
787f4b8e57fSBram Moolenaar 
788f4b8e57fSBram Moolenaar     return NOERROR;
789f4b8e57fSBram Moolenaar }
790f4b8e57fSBram Moolenaar 
791f4b8e57fSBram Moolenaar BOOL CALLBACK CShellExt::EnumWindowsProc(HWND hWnd, LPARAM lParam)
792f4b8e57fSBram Moolenaar {
793e759a7aaSBram Moolenaar     char temp[BUFSIZE];
794f4b8e57fSBram Moolenaar 
795f4b8e57fSBram Moolenaar     // First do a bunch of check
796f4b8e57fSBram Moolenaar     // No invisible window
797f4b8e57fSBram Moolenaar     if (!IsWindowVisible(hWnd)) return TRUE;
798f4b8e57fSBram Moolenaar     // No child window ???
799f4b8e57fSBram Moolenaar     // if (GetParent(hWnd)) return TRUE;
800f4b8e57fSBram Moolenaar     // Class name should be Vim, if failed to get class name, return
801f4b8e57fSBram Moolenaar     if (GetClassName(hWnd, temp, sizeof(temp)) == 0)
802f4b8e57fSBram Moolenaar 	return TRUE;
803f4b8e57fSBram Moolenaar     // Compare class name to that of vim, if not, return
804f4b8e57fSBram Moolenaar     if (_strnicmp(temp, "vim", sizeof("vim")) != 0)
805f4b8e57fSBram Moolenaar 	return TRUE;
806f4b8e57fSBram Moolenaar     // First check if the number of vim instance exceeds MAX_HWND
807f4b8e57fSBram Moolenaar     CShellExt *cs = (CShellExt*) lParam;
808f4b8e57fSBram Moolenaar     if (cs->m_cntOfHWnd >= MAX_HWND) return TRUE;
809f4b8e57fSBram Moolenaar     // Now we get the vim window, put it into some kind of array
810f4b8e57fSBram Moolenaar     cs->m_hWnd[cs->m_cntOfHWnd] = hWnd;
811f4b8e57fSBram Moolenaar     cs->m_cntOfHWnd ++;
812f4b8e57fSBram Moolenaar 
813f4b8e57fSBram Moolenaar     return TRUE; // continue enumeration (otherwise this would be false)
814f4b8e57fSBram Moolenaar }
815f4b8e57fSBram Moolenaar 
816f4b8e57fSBram Moolenaar #ifdef WIN32
817f4b8e57fSBram Moolenaar // This symbol is not defined in older versions of the SDK or Visual C++.
818f4b8e57fSBram Moolenaar 
819f4b8e57fSBram Moolenaar #ifndef VER_PLATFORM_WIN32_WINDOWS
820f4b8e57fSBram Moolenaar # define VER_PLATFORM_WIN32_WINDOWS 1
821f4b8e57fSBram Moolenaar #endif
822f4b8e57fSBram Moolenaar 
823f4b8e57fSBram Moolenaar static DWORD g_PlatformId;
824f4b8e57fSBram Moolenaar 
825f4b8e57fSBram Moolenaar //
826f4b8e57fSBram Moolenaar // Set g_PlatformId to VER_PLATFORM_WIN32_NT (NT) or
827f4b8e57fSBram Moolenaar // VER_PLATFORM_WIN32_WINDOWS (Win95).
828f4b8e57fSBram Moolenaar //
829f4b8e57fSBram Moolenaar     static void
830f4b8e57fSBram Moolenaar PlatformId(void)
831f4b8e57fSBram Moolenaar {
832f4b8e57fSBram Moolenaar     static int done = FALSE;
833f4b8e57fSBram Moolenaar 
834f4b8e57fSBram Moolenaar     if (!done)
835f4b8e57fSBram Moolenaar     {
836f4b8e57fSBram Moolenaar 	OSVERSIONINFO ovi;
837f4b8e57fSBram Moolenaar 
838f4b8e57fSBram Moolenaar 	ovi.dwOSVersionInfoSize = sizeof(ovi);
839f4b8e57fSBram Moolenaar 	GetVersionEx(&ovi);
840f4b8e57fSBram Moolenaar 
841f4b8e57fSBram Moolenaar 	g_PlatformId = ovi.dwPlatformId;
842f4b8e57fSBram Moolenaar 	done = TRUE;
843f4b8e57fSBram Moolenaar     }
844f4b8e57fSBram Moolenaar }
845f4b8e57fSBram Moolenaar 
846f4b8e57fSBram Moolenaar # ifndef __BORLANDC__
847f4b8e57fSBram Moolenaar     static char *
848f4b8e57fSBram Moolenaar searchpath(char *name)
849f4b8e57fSBram Moolenaar {
850e759a7aaSBram Moolenaar     static char widename[2 * BUFSIZE];
851e759a7aaSBram Moolenaar     static char location[2 * BUFSIZE + 2];
852f4b8e57fSBram Moolenaar 
853f4b8e57fSBram Moolenaar     // There appears to be a bug in FindExecutableA() on Windows NT.
854f4b8e57fSBram Moolenaar     // Use FindExecutableW() instead...
855f4b8e57fSBram Moolenaar     PlatformId();
856f4b8e57fSBram Moolenaar     if (g_PlatformId == VER_PLATFORM_WIN32_NT)
857f4b8e57fSBram Moolenaar     {
858f4b8e57fSBram Moolenaar 	MultiByteToWideChar(CP_ACP, 0, (LPCTSTR)name, -1,
859e759a7aaSBram Moolenaar 		(LPWSTR)widename, BUFSIZE);
860f4b8e57fSBram Moolenaar 	if (FindExecutableW((LPCWSTR)widename, (LPCWSTR)"",
861f4b8e57fSBram Moolenaar 		    (LPWSTR)location) > (HINSTANCE)32)
862f4b8e57fSBram Moolenaar 	{
863f4b8e57fSBram Moolenaar 	    WideCharToMultiByte(CP_ACP, 0, (LPWSTR)location, -1,
864e759a7aaSBram Moolenaar 		    (LPSTR)widename, 2 * BUFSIZE, NULL, NULL);
865f4b8e57fSBram Moolenaar 	    return widename;
866f4b8e57fSBram Moolenaar 	}
867f4b8e57fSBram Moolenaar     }
868f4b8e57fSBram Moolenaar     else
869f4b8e57fSBram Moolenaar     {
870f4b8e57fSBram Moolenaar 	if (FindExecutableA((LPCTSTR)name, (LPCTSTR)"",
871f4b8e57fSBram Moolenaar 		    (LPTSTR)location) > (HINSTANCE)32)
872f4b8e57fSBram Moolenaar 	    return location;
873f4b8e57fSBram Moolenaar     }
8747baa45dbSBram Moolenaar     return (char *)"";
875f4b8e57fSBram Moolenaar }
876f4b8e57fSBram Moolenaar # endif
877f4b8e57fSBram Moolenaar #endif
878f4b8e57fSBram Moolenaar 
879f4b8e57fSBram Moolenaar STDMETHODIMP CShellExt::InvokeGvim(HWND hParent,
880e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszWorkingDir */,
881e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszCmd */,
882e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszParam */,
883e6a91fd9SBram Moolenaar 				   int  /* iShowCmd */)
884f4b8e57fSBram Moolenaar {
8857e6d3bd3SBram Moolenaar     wchar_t m_szFileUserClickedOn[BUFSIZE];
8867e6d3bd3SBram Moolenaar     wchar_t cmdStrW[BUFSIZE];
887f4b8e57fSBram Moolenaar     UINT i;
888f4b8e57fSBram Moolenaar 
889f4b8e57fSBram Moolenaar     for (i = 0; i < cbFiles; i++)
890f4b8e57fSBram Moolenaar     {
8917e6d3bd3SBram Moolenaar 	DragQueryFileW((HDROP)medium.hGlobal,
892f4b8e57fSBram Moolenaar 		i,
893f4b8e57fSBram Moolenaar 		m_szFileUserClickedOn,
894f4b8e57fSBram Moolenaar 		sizeof(m_szFileUserClickedOn));
895f4b8e57fSBram Moolenaar 
8967e6d3bd3SBram Moolenaar 	getGvimNameW(cmdStrW);
8977e6d3bd3SBram Moolenaar 	wcscat(cmdStrW, L" \"");
898f4b8e57fSBram Moolenaar 
8997e6d3bd3SBram Moolenaar 	if ((wcslen(cmdStrW) + wcslen(m_szFileUserClickedOn) + 2) < BUFSIZE)
900f4b8e57fSBram Moolenaar 	{
9017e6d3bd3SBram Moolenaar 	    wcscat(cmdStrW, m_szFileUserClickedOn);
9027e6d3bd3SBram Moolenaar 	    wcscat(cmdStrW, L"\"");
903f4b8e57fSBram Moolenaar 
9047e6d3bd3SBram Moolenaar 	    STARTUPINFOW si;
905f4b8e57fSBram Moolenaar 	    PROCESS_INFORMATION pi;
906f4b8e57fSBram Moolenaar 
907f4b8e57fSBram Moolenaar 	    ZeroMemory(&si, sizeof(si));
908f4b8e57fSBram Moolenaar 	    si.cb = sizeof(si);
909f4b8e57fSBram Moolenaar 
910f4b8e57fSBram Moolenaar 	    // Start the child process.
9117e6d3bd3SBram Moolenaar 	    if (!CreateProcessW(NULL,	// No module name (use command line).
9127e6d3bd3SBram Moolenaar 			cmdStrW,	// Command line.
913f4b8e57fSBram Moolenaar 			NULL,		// Process handle not inheritable.
914f4b8e57fSBram Moolenaar 			NULL,		// Thread handle not inheritable.
915f4b8e57fSBram Moolenaar 			FALSE,		// Set handle inheritance to FALSE.
916*cf839732SBram Moolenaar 			oldenv == NULL ? 0 : CREATE_UNICODE_ENVIRONMENT,
917*cf839732SBram Moolenaar 			oldenv,		// Use unmodified environment block.
918f4b8e57fSBram Moolenaar 			NULL,		// Use parent's starting directory.
919f4b8e57fSBram Moolenaar 			&si,		// Pointer to STARTUPINFO structure.
920f4b8e57fSBram Moolenaar 			&pi)		// Pointer to PROCESS_INFORMATION structure.
921f4b8e57fSBram Moolenaar 	       )
922f4b8e57fSBram Moolenaar 	    {
923f4b8e57fSBram Moolenaar 		MessageBox(
924f4b8e57fSBram Moolenaar 		    hParent,
925f4b8e57fSBram Moolenaar 		    _("Error creating process: Check if gvim is in your path!"),
926f4b8e57fSBram Moolenaar 		    _("gvimext.dll error"),
927f4b8e57fSBram Moolenaar 		    MB_OK);
928f4b8e57fSBram Moolenaar 	    }
929f4b8e57fSBram Moolenaar 	    else
930f4b8e57fSBram Moolenaar 	    {
931f4b8e57fSBram Moolenaar 		CloseHandle( pi.hProcess );
932f4b8e57fSBram Moolenaar 		CloseHandle( pi.hThread );
933f4b8e57fSBram Moolenaar 	    }
934f4b8e57fSBram Moolenaar 	}
935f4b8e57fSBram Moolenaar 	else
936f4b8e57fSBram Moolenaar 	{
937f4b8e57fSBram Moolenaar 	    MessageBox(
938f4b8e57fSBram Moolenaar 		hParent,
939f4b8e57fSBram Moolenaar 		_("Path length too long!"),
940f4b8e57fSBram Moolenaar 		_("gvimext.dll error"),
941f4b8e57fSBram Moolenaar 		MB_OK);
942f4b8e57fSBram Moolenaar 	}
943f4b8e57fSBram Moolenaar     }
944f4b8e57fSBram Moolenaar 
945f4b8e57fSBram Moolenaar     return NOERROR;
946f4b8e57fSBram Moolenaar }
947f4b8e57fSBram Moolenaar 
948f4b8e57fSBram Moolenaar 
949f4b8e57fSBram Moolenaar STDMETHODIMP CShellExt::InvokeSingleGvim(HWND hParent,
950e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszWorkingDir */,
951e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszCmd */,
952e6a91fd9SBram Moolenaar 				   LPCSTR  /* pszParam */,
953e6a91fd9SBram Moolenaar 				   int  /* iShowCmd */,
954f4b8e57fSBram Moolenaar 				   int useDiff)
955f4b8e57fSBram Moolenaar {
9567e6d3bd3SBram Moolenaar     wchar_t	m_szFileUserClickedOn[BUFSIZE];
9577e6d3bd3SBram Moolenaar     wchar_t	*cmdStrW;
958f4b8e57fSBram Moolenaar     size_t	cmdlen;
959f4b8e57fSBram Moolenaar     size_t	len;
960f4b8e57fSBram Moolenaar     UINT i;
961f4b8e57fSBram Moolenaar 
962e759a7aaSBram Moolenaar     cmdlen = BUFSIZE;
9637e6d3bd3SBram Moolenaar     cmdStrW  = (wchar_t *) malloc(cmdlen * sizeof(wchar_t));
9647e6d3bd3SBram Moolenaar     getGvimNameW(cmdStrW);
9657e6d3bd3SBram Moolenaar 
966f4b8e57fSBram Moolenaar     if (useDiff)
9677e6d3bd3SBram Moolenaar 	wcscat(cmdStrW, L" -d");
968f4b8e57fSBram Moolenaar     for (i = 0; i < cbFiles; i++)
969f4b8e57fSBram Moolenaar     {
9707e6d3bd3SBram Moolenaar 	DragQueryFileW((HDROP)medium.hGlobal,
971f4b8e57fSBram Moolenaar 		i,
972f4b8e57fSBram Moolenaar 		m_szFileUserClickedOn,
973f4b8e57fSBram Moolenaar 		sizeof(m_szFileUserClickedOn));
974f4b8e57fSBram Moolenaar 
9757e6d3bd3SBram Moolenaar 	len = wcslen(cmdStrW) + wcslen(m_szFileUserClickedOn) + 4;
976f4b8e57fSBram Moolenaar 	if (len > cmdlen)
977f4b8e57fSBram Moolenaar 	{
978e759a7aaSBram Moolenaar 	    cmdlen = len + BUFSIZE;
9797e6d3bd3SBram Moolenaar 	    cmdStrW = (wchar_t *)realloc(cmdStrW, cmdlen * sizeof(wchar_t));
980f4b8e57fSBram Moolenaar 	}
9817e6d3bd3SBram Moolenaar 	wcscat(cmdStrW, L" \"");
9827e6d3bd3SBram Moolenaar 	wcscat(cmdStrW, m_szFileUserClickedOn);
9837e6d3bd3SBram Moolenaar 	wcscat(cmdStrW, L"\"");
984f4b8e57fSBram Moolenaar     }
985f4b8e57fSBram Moolenaar 
9867e6d3bd3SBram Moolenaar     STARTUPINFOW si;
987f4b8e57fSBram Moolenaar     PROCESS_INFORMATION pi;
988f4b8e57fSBram Moolenaar 
989f4b8e57fSBram Moolenaar     ZeroMemory(&si, sizeof(si));
990f4b8e57fSBram Moolenaar     si.cb = sizeof(si);
991f4b8e57fSBram Moolenaar 
992f4b8e57fSBram Moolenaar     // Start the child process.
9937e6d3bd3SBram Moolenaar     if (!CreateProcessW(NULL,	// No module name (use command line).
9947e6d3bd3SBram Moolenaar 		cmdStrW,	// Command line.
995f4b8e57fSBram Moolenaar 		NULL,		// Process handle not inheritable.
996f4b8e57fSBram Moolenaar 		NULL,		// Thread handle not inheritable.
997f4b8e57fSBram Moolenaar 		FALSE,		// Set handle inheritance to FALSE.
998*cf839732SBram Moolenaar 		oldenv == NULL ? 0 : CREATE_UNICODE_ENVIRONMENT,
999*cf839732SBram Moolenaar 		oldenv,		// Use unmodified environment block.
1000f4b8e57fSBram Moolenaar 		NULL,		// Use parent's starting directory.
1001f4b8e57fSBram Moolenaar 		&si,		// Pointer to STARTUPINFO structure.
1002f4b8e57fSBram Moolenaar 		&pi)		// Pointer to PROCESS_INFORMATION structure.
1003f4b8e57fSBram Moolenaar        )
1004f4b8e57fSBram Moolenaar     {
1005f4b8e57fSBram Moolenaar 	MessageBox(
1006f4b8e57fSBram Moolenaar 	    hParent,
1007f4b8e57fSBram Moolenaar 	    _("Error creating process: Check if gvim is in your path!"),
1008f4b8e57fSBram Moolenaar 	    _("gvimext.dll error"),
1009f4b8e57fSBram Moolenaar 	    MB_OK);
1010f4b8e57fSBram Moolenaar     }
1011f4b8e57fSBram Moolenaar     else
1012f4b8e57fSBram Moolenaar     {
1013f4b8e57fSBram Moolenaar 	CloseHandle(pi.hProcess);
1014f4b8e57fSBram Moolenaar 	CloseHandle(pi.hThread);
1015f4b8e57fSBram Moolenaar     }
1016f4b8e57fSBram Moolenaar 
10177e6d3bd3SBram Moolenaar     free(cmdStrW);
1018f4b8e57fSBram Moolenaar 
1019f4b8e57fSBram Moolenaar     return NOERROR;
1020f4b8e57fSBram Moolenaar }
1021