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