xref: /vim-8.2.3635/src/if_python.c (revision 9dc93ae4)
1071d4279SBram Moolenaar /* vi:set ts=8 sts=4 sw=4:
2071d4279SBram Moolenaar  *
3071d4279SBram Moolenaar  * VIM - Vi IMproved	by Bram Moolenaar
4071d4279SBram Moolenaar  *
5071d4279SBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
6071d4279SBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
7071d4279SBram Moolenaar  * See README.txt for an overview of the Vim source code.
8071d4279SBram Moolenaar  */
9071d4279SBram Moolenaar /*
10071d4279SBram Moolenaar  * Python extensions by Paul Moore.
11071d4279SBram Moolenaar  * Changes for Unix by David Leonard.
12071d4279SBram Moolenaar  *
13071d4279SBram Moolenaar  * This consists of four parts:
14071d4279SBram Moolenaar  * 1. Python interpreter main program
15071d4279SBram Moolenaar  * 2. Python output stream: writes output via [e]msg().
16071d4279SBram Moolenaar  * 3. Implementation of the Vim module for Python
17071d4279SBram Moolenaar  * 4. Utility functions for handling the interface between Vim and Python.
18071d4279SBram Moolenaar  */
19071d4279SBram Moolenaar 
20071d4279SBram Moolenaar #include "vim.h"
21071d4279SBram Moolenaar 
22071d4279SBram Moolenaar #include <limits.h>
23071d4279SBram Moolenaar 
24071d4279SBram Moolenaar /* Python.h defines _POSIX_THREADS itself (if needed) */
25071d4279SBram Moolenaar #ifdef _POSIX_THREADS
26071d4279SBram Moolenaar # undef _POSIX_THREADS
27071d4279SBram Moolenaar #endif
28071d4279SBram Moolenaar 
29071d4279SBram Moolenaar #if defined(_WIN32) && defined(HAVE_FCNTL_H)
30071d4279SBram Moolenaar # undef HAVE_FCNTL_H
31071d4279SBram Moolenaar #endif
32071d4279SBram Moolenaar 
33071d4279SBram Moolenaar #ifdef _DEBUG
34071d4279SBram Moolenaar # undef _DEBUG
35071d4279SBram Moolenaar #endif
36071d4279SBram Moolenaar 
37071d4279SBram Moolenaar #ifdef HAVE_STDARG_H
38071d4279SBram Moolenaar # undef HAVE_STDARG_H	/* Python's config.h defines it as well. */
39071d4279SBram Moolenaar #endif
40be2c9ae9SBram Moolenaar #ifdef _POSIX_C_SOURCE
41be2c9ae9SBram Moolenaar # undef _POSIX_C_SOURCE	/* pyconfig.h defines it as well. */
42be2c9ae9SBram Moolenaar #endif
43be2c9ae9SBram Moolenaar #ifdef _XOPEN_SOURCE
44be2c9ae9SBram Moolenaar # undef _XOPEN_SOURCE	/* pyconfig.h defines it as well. */
45be2c9ae9SBram Moolenaar #endif
46071d4279SBram Moolenaar 
472c45e945SBram Moolenaar #define PY_SSIZE_T_CLEAN
482c45e945SBram Moolenaar 
49071d4279SBram Moolenaar #include <Python.h>
50071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
51071d4279SBram Moolenaar # include "macglue.h"
52071d4279SBram Moolenaar # include <CodeFragments.h>
53071d4279SBram Moolenaar #endif
54071d4279SBram Moolenaar #undef main /* Defined in python.h - aargh */
55071d4279SBram Moolenaar #undef HAVE_FCNTL_H /* Clash with os_win32.h */
56071d4279SBram Moolenaar 
57170bf1aeSBram Moolenaar static void init_structs(void);
58170bf1aeSBram Moolenaar 
5919e60943SBram Moolenaar /* No-op conversion functions, use with care! */
6019e60943SBram Moolenaar #define PyString_AsBytes(obj) (obj)
6119e60943SBram Moolenaar #define PyString_FreeBytes(obj)
6219e60943SBram Moolenaar 
63071d4279SBram Moolenaar #if !defined(FEAT_PYTHON) && defined(PROTO)
64071d4279SBram Moolenaar /* Use this to be able to generate prototypes without python being used. */
65e7cb9cf6SBram Moolenaar # define PyObject Py_ssize_t
66e7cb9cf6SBram Moolenaar # define PyThreadState Py_ssize_t
67e7cb9cf6SBram Moolenaar # define PyTypeObject Py_ssize_t
68e7cb9cf6SBram Moolenaar struct PyMethodDef { Py_ssize_t a; };
69e7cb9cf6SBram Moolenaar # define PySequenceMethods Py_ssize_t
70071d4279SBram Moolenaar #endif
71071d4279SBram Moolenaar 
722c45e945SBram Moolenaar #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000
732c45e945SBram Moolenaar # define PyInt Py_ssize_t
742c45e945SBram Moolenaar # define PyInquiry lenfunc
752c45e945SBram Moolenaar # define PyIntArgFunc ssizeargfunc
762c45e945SBram Moolenaar # define PyIntIntArgFunc ssizessizeargfunc
772c45e945SBram Moolenaar # define PyIntObjArgProc ssizeobjargproc
782c45e945SBram Moolenaar # define PyIntIntObjArgProc ssizessizeobjargproc
79e7cb9cf6SBram Moolenaar # define Py_ssize_t_fmt "n"
802c45e945SBram Moolenaar #else
812c45e945SBram Moolenaar # define PyInt int
822c45e945SBram Moolenaar # define PyInquiry inquiry
832c45e945SBram Moolenaar # define PyIntArgFunc intargfunc
842c45e945SBram Moolenaar # define PyIntIntArgFunc intintargfunc
852c45e945SBram Moolenaar # define PyIntObjArgProc intobjargproc
862c45e945SBram Moolenaar # define PyIntIntObjArgProc intintobjargproc
87e7cb9cf6SBram Moolenaar # define Py_ssize_t_fmt "i"
882c45e945SBram Moolenaar #endif
892c45e945SBram Moolenaar 
90071d4279SBram Moolenaar /* Parser flags */
91071d4279SBram Moolenaar #define single_input	256
92071d4279SBram Moolenaar #define file_input	257
93071d4279SBram Moolenaar #define eval_input	258
94071d4279SBram Moolenaar 
95071d4279SBram Moolenaar #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x020300F0
96071d4279SBram Moolenaar   /* Python 2.3: can invoke ":python" recursively. */
97071d4279SBram Moolenaar # define PY_CAN_RECURSE
98071d4279SBram Moolenaar #endif
99071d4279SBram Moolenaar 
100071d4279SBram Moolenaar # if defined(DYNAMIC_PYTHON) || defined(PROTO)
101071d4279SBram Moolenaar #  ifndef DYNAMIC_PYTHON
102e7cb9cf6SBram Moolenaar #   define HINSTANCE long_u		/* for generating prototypes */
103071d4279SBram Moolenaar #  endif
104071d4279SBram Moolenaar 
105fa5d1e63SBram Moolenaar # ifndef WIN3264
106bd5e15fdSBram Moolenaar #  include <dlfcn.h>
107bd5e15fdSBram Moolenaar #  define FARPROC void*
108bd5e15fdSBram Moolenaar #  define HINSTANCE void*
109644d37b8SBram Moolenaar #  if defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)
110b61f95c3SBram Moolenaar #   define load_dll(n) dlopen((n), RTLD_LAZY)
111b61f95c3SBram Moolenaar #  else
112fa5d1e63SBram Moolenaar #   define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL)
113b61f95c3SBram Moolenaar #  endif
114bd5e15fdSBram Moolenaar #  define close_dll dlclose
115bd5e15fdSBram Moolenaar #  define symbol_from_dll dlsym
116bd5e15fdSBram Moolenaar # else
117ebbcb824SBram Moolenaar #  define load_dll vimLoadLib
118bd5e15fdSBram Moolenaar #  define close_dll FreeLibrary
119bd5e15fdSBram Moolenaar #  define symbol_from_dll GetProcAddress
120bd5e15fdSBram Moolenaar # endif
121bd5e15fdSBram Moolenaar 
122e7cb9cf6SBram Moolenaar /* This makes if_python.c compile without warnings against Python 2.5
123e7cb9cf6SBram Moolenaar  * on Win32 and Win64. */
124e7cb9cf6SBram Moolenaar # undef PyRun_SimpleString
125e7cb9cf6SBram Moolenaar # undef PyArg_Parse
126e7cb9cf6SBram Moolenaar # undef PyArg_ParseTuple
127e7cb9cf6SBram Moolenaar # undef Py_BuildValue
128e7cb9cf6SBram Moolenaar # undef Py_InitModule4
129e7cb9cf6SBram Moolenaar # undef Py_InitModule4_64
130e7cb9cf6SBram Moolenaar 
131071d4279SBram Moolenaar /*
132071d4279SBram Moolenaar  * Wrapper defines
133071d4279SBram Moolenaar  */
134071d4279SBram Moolenaar # define PyArg_Parse dll_PyArg_Parse
135071d4279SBram Moolenaar # define PyArg_ParseTuple dll_PyArg_ParseTuple
13619e60943SBram Moolenaar # define PyMem_Free dll_PyMem_Free
137071d4279SBram Moolenaar # define PyDict_SetItemString dll_PyDict_SetItemString
138071d4279SBram Moolenaar # define PyErr_BadArgument dll_PyErr_BadArgument
139071d4279SBram Moolenaar # define PyErr_Clear dll_PyErr_Clear
140071d4279SBram Moolenaar # define PyErr_NoMemory dll_PyErr_NoMemory
141071d4279SBram Moolenaar # define PyErr_Occurred dll_PyErr_Occurred
142071d4279SBram Moolenaar # define PyErr_SetNone dll_PyErr_SetNone
143071d4279SBram Moolenaar # define PyErr_SetString dll_PyErr_SetString
144071d4279SBram Moolenaar # define PyEval_InitThreads dll_PyEval_InitThreads
145071d4279SBram Moolenaar # define PyEval_RestoreThread dll_PyEval_RestoreThread
146071d4279SBram Moolenaar # define PyEval_SaveThread dll_PyEval_SaveThread
147071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE
148071d4279SBram Moolenaar #  define PyGILState_Ensure dll_PyGILState_Ensure
149071d4279SBram Moolenaar #  define PyGILState_Release dll_PyGILState_Release
150071d4279SBram Moolenaar # endif
151071d4279SBram Moolenaar # define PyInt_AsLong dll_PyInt_AsLong
152071d4279SBram Moolenaar # define PyInt_FromLong dll_PyInt_FromLong
153071d4279SBram Moolenaar # define PyInt_Type (*dll_PyInt_Type)
154071d4279SBram Moolenaar # define PyList_GetItem dll_PyList_GetItem
1550ac9379aSBram Moolenaar # define PyList_Append dll_PyList_Append
156071d4279SBram Moolenaar # define PyList_New dll_PyList_New
157071d4279SBram Moolenaar # define PyList_SetItem dll_PyList_SetItem
158071d4279SBram Moolenaar # define PyList_Size dll_PyList_Size
159071d4279SBram Moolenaar # define PyList_Type (*dll_PyList_Type)
160071d4279SBram Moolenaar # define PyImport_ImportModule dll_PyImport_ImportModule
1610ac9379aSBram Moolenaar # define PyDict_New dll_PyDict_New
162071d4279SBram Moolenaar # define PyDict_GetItemString dll_PyDict_GetItemString
163071d4279SBram Moolenaar # define PyModule_GetDict dll_PyModule_GetDict
164071d4279SBram Moolenaar # define PyRun_SimpleString dll_PyRun_SimpleString
165071d4279SBram Moolenaar # define PyString_AsString dll_PyString_AsString
166071d4279SBram Moolenaar # define PyString_FromString dll_PyString_FromString
167071d4279SBram Moolenaar # define PyString_FromStringAndSize dll_PyString_FromStringAndSize
168071d4279SBram Moolenaar # define PyString_Size dll_PyString_Size
169071d4279SBram Moolenaar # define PyString_Type (*dll_PyString_Type)
170071d4279SBram Moolenaar # define PySys_SetObject dll_PySys_SetObject
171071d4279SBram Moolenaar # define PySys_SetArgv dll_PySys_SetArgv
172071d4279SBram Moolenaar # define PyType_Type (*dll_PyType_Type)
17330fec7bcSBram Moolenaar # define PyType_Ready (*dll_PyType_Ready)
174071d4279SBram Moolenaar # define Py_BuildValue dll_Py_BuildValue
175071d4279SBram Moolenaar # define Py_FindMethod dll_Py_FindMethod
176071d4279SBram Moolenaar # define Py_InitModule4 dll_Py_InitModule4
177644d37b8SBram Moolenaar # define Py_SetPythonHome dll_Py_SetPythonHome
178071d4279SBram Moolenaar # define Py_Initialize dll_Py_Initialize
1790e21a3f6SBram Moolenaar # define Py_Finalize dll_Py_Finalize
1800e21a3f6SBram Moolenaar # define Py_IsInitialized dll_Py_IsInitialized
181071d4279SBram Moolenaar # define _PyObject_New dll__PyObject_New
182071d4279SBram Moolenaar # define _Py_NoneStruct (*dll__Py_NoneStruct)
183071d4279SBram Moolenaar # define PyObject_Init dll__PyObject_Init
184071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
185071d4279SBram Moolenaar #  define PyType_IsSubtype dll_PyType_IsSubtype
186071d4279SBram Moolenaar # endif
187071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
188071d4279SBram Moolenaar #  define PyObject_Malloc dll_PyObject_Malloc
189071d4279SBram Moolenaar #  define PyObject_Free dll_PyObject_Free
190071d4279SBram Moolenaar # endif
191071d4279SBram Moolenaar 
192071d4279SBram Moolenaar /*
193071d4279SBram Moolenaar  * Pointers for dynamic link
194071d4279SBram Moolenaar  */
195071d4279SBram Moolenaar static int(*dll_PyArg_Parse)(PyObject *, char *, ...);
196071d4279SBram Moolenaar static int(*dll_PyArg_ParseTuple)(PyObject *, char *, ...);
19719e60943SBram Moolenaar static int(*dll_PyMem_Free)(void *);
198071d4279SBram Moolenaar static int(*dll_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item);
199071d4279SBram Moolenaar static int(*dll_PyErr_BadArgument)(void);
200071d4279SBram Moolenaar static void(*dll_PyErr_Clear)(void);
201071d4279SBram Moolenaar static PyObject*(*dll_PyErr_NoMemory)(void);
202071d4279SBram Moolenaar static PyObject*(*dll_PyErr_Occurred)(void);
203071d4279SBram Moolenaar static void(*dll_PyErr_SetNone)(PyObject *);
204071d4279SBram Moolenaar static void(*dll_PyErr_SetString)(PyObject *, const char *);
205071d4279SBram Moolenaar static void(*dll_PyEval_InitThreads)(void);
206071d4279SBram Moolenaar static void(*dll_PyEval_RestoreThread)(PyThreadState *);
207071d4279SBram Moolenaar static PyThreadState*(*dll_PyEval_SaveThread)(void);
208071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE
209071d4279SBram Moolenaar static PyGILState_STATE	(*dll_PyGILState_Ensure)(void);
210071d4279SBram Moolenaar static void (*dll_PyGILState_Release)(PyGILState_STATE);
211071d4279SBram Moolenaar #endif
212071d4279SBram Moolenaar static long(*dll_PyInt_AsLong)(PyObject *);
213071d4279SBram Moolenaar static PyObject*(*dll_PyInt_FromLong)(long);
214071d4279SBram Moolenaar static PyTypeObject* dll_PyInt_Type;
2152c45e945SBram Moolenaar static PyObject*(*dll_PyList_GetItem)(PyObject *, PyInt);
2160ac9379aSBram Moolenaar static PyObject*(*dll_PyList_Append)(PyObject *, PyObject *);
2172c45e945SBram Moolenaar static PyObject*(*dll_PyList_New)(PyInt size);
2182c45e945SBram Moolenaar static int(*dll_PyList_SetItem)(PyObject *, PyInt, PyObject *);
2192c45e945SBram Moolenaar static PyInt(*dll_PyList_Size)(PyObject *);
220071d4279SBram Moolenaar static PyTypeObject* dll_PyList_Type;
221071d4279SBram Moolenaar static PyObject*(*dll_PyImport_ImportModule)(const char *);
2220ac9379aSBram Moolenaar static PyObject*(*dll_PyDict_New)(void);
223071d4279SBram Moolenaar static PyObject*(*dll_PyDict_GetItemString)(PyObject *, const char *);
224071d4279SBram Moolenaar static PyObject*(*dll_PyModule_GetDict)(PyObject *);
225071d4279SBram Moolenaar static int(*dll_PyRun_SimpleString)(char *);
226071d4279SBram Moolenaar static char*(*dll_PyString_AsString)(PyObject *);
227071d4279SBram Moolenaar static PyObject*(*dll_PyString_FromString)(const char *);
2282c45e945SBram Moolenaar static PyObject*(*dll_PyString_FromStringAndSize)(const char *, PyInt);
2292c45e945SBram Moolenaar static PyInt(*dll_PyString_Size)(PyObject *);
230071d4279SBram Moolenaar static PyTypeObject* dll_PyString_Type;
231071d4279SBram Moolenaar static int(*dll_PySys_SetObject)(char *, PyObject *);
232071d4279SBram Moolenaar static int(*dll_PySys_SetArgv)(int, char **);
233071d4279SBram Moolenaar static PyTypeObject* dll_PyType_Type;
23430fec7bcSBram Moolenaar static int (*dll_PyType_Ready)(PyTypeObject *type);
235071d4279SBram Moolenaar static PyObject*(*dll_Py_BuildValue)(char *, ...);
236071d4279SBram Moolenaar static PyObject*(*dll_Py_FindMethod)(struct PyMethodDef[], PyObject *, char *);
237071d4279SBram Moolenaar static PyObject*(*dll_Py_InitModule4)(char *, struct PyMethodDef *, char *, PyObject *, int);
238644d37b8SBram Moolenaar static void(*dll_Py_SetPythonHome)(char *home);
239071d4279SBram Moolenaar static void(*dll_Py_Initialize)(void);
2400e21a3f6SBram Moolenaar static void(*dll_Py_Finalize)(void);
2410e21a3f6SBram Moolenaar static int(*dll_Py_IsInitialized)(void);
242071d4279SBram Moolenaar static PyObject*(*dll__PyObject_New)(PyTypeObject *, PyObject *);
243071d4279SBram Moolenaar static PyObject*(*dll__PyObject_Init)(PyObject *, PyTypeObject *);
244071d4279SBram Moolenaar static PyObject* dll__Py_NoneStruct;
245071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
246071d4279SBram Moolenaar static int (*dll_PyType_IsSubtype)(PyTypeObject *, PyTypeObject *);
247071d4279SBram Moolenaar # endif
248071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
249071d4279SBram Moolenaar static void* (*dll_PyObject_Malloc)(size_t);
250071d4279SBram Moolenaar static void (*dll_PyObject_Free)(void*);
251071d4279SBram Moolenaar # endif
252071d4279SBram Moolenaar 
253071d4279SBram Moolenaar static HINSTANCE hinstPython = 0; /* Instance of python.dll */
254071d4279SBram Moolenaar 
255071d4279SBram Moolenaar /* Imported exception objects */
256071d4279SBram Moolenaar static PyObject *imp_PyExc_AttributeError;
257071d4279SBram Moolenaar static PyObject *imp_PyExc_IndexError;
258071d4279SBram Moolenaar static PyObject *imp_PyExc_KeyboardInterrupt;
259071d4279SBram Moolenaar static PyObject *imp_PyExc_TypeError;
260071d4279SBram Moolenaar static PyObject *imp_PyExc_ValueError;
261071d4279SBram Moolenaar 
262071d4279SBram Moolenaar # define PyExc_AttributeError imp_PyExc_AttributeError
263071d4279SBram Moolenaar # define PyExc_IndexError imp_PyExc_IndexError
264071d4279SBram Moolenaar # define PyExc_KeyboardInterrupt imp_PyExc_KeyboardInterrupt
265071d4279SBram Moolenaar # define PyExc_TypeError imp_PyExc_TypeError
266071d4279SBram Moolenaar # define PyExc_ValueError imp_PyExc_ValueError
267071d4279SBram Moolenaar 
268071d4279SBram Moolenaar /*
269071d4279SBram Moolenaar  * Table of name to function pointer of python.
270071d4279SBram Moolenaar  */
271071d4279SBram Moolenaar # define PYTHON_PROC FARPROC
272071d4279SBram Moolenaar static struct
273071d4279SBram Moolenaar {
274071d4279SBram Moolenaar     char *name;
275071d4279SBram Moolenaar     PYTHON_PROC *ptr;
276071d4279SBram Moolenaar } python_funcname_table[] =
277071d4279SBram Moolenaar {
278071d4279SBram Moolenaar     {"PyArg_Parse", (PYTHON_PROC*)&dll_PyArg_Parse},
279071d4279SBram Moolenaar     {"PyArg_ParseTuple", (PYTHON_PROC*)&dll_PyArg_ParseTuple},
28019e60943SBram Moolenaar     {"PyMem_Free", (PYTHON_PROC*)&dll_PyMem_Free},
281071d4279SBram Moolenaar     {"PyDict_SetItemString", (PYTHON_PROC*)&dll_PyDict_SetItemString},
282071d4279SBram Moolenaar     {"PyErr_BadArgument", (PYTHON_PROC*)&dll_PyErr_BadArgument},
283071d4279SBram Moolenaar     {"PyErr_Clear", (PYTHON_PROC*)&dll_PyErr_Clear},
284071d4279SBram Moolenaar     {"PyErr_NoMemory", (PYTHON_PROC*)&dll_PyErr_NoMemory},
285071d4279SBram Moolenaar     {"PyErr_Occurred", (PYTHON_PROC*)&dll_PyErr_Occurred},
286071d4279SBram Moolenaar     {"PyErr_SetNone", (PYTHON_PROC*)&dll_PyErr_SetNone},
287071d4279SBram Moolenaar     {"PyErr_SetString", (PYTHON_PROC*)&dll_PyErr_SetString},
288071d4279SBram Moolenaar     {"PyEval_InitThreads", (PYTHON_PROC*)&dll_PyEval_InitThreads},
289071d4279SBram Moolenaar     {"PyEval_RestoreThread", (PYTHON_PROC*)&dll_PyEval_RestoreThread},
290071d4279SBram Moolenaar     {"PyEval_SaveThread", (PYTHON_PROC*)&dll_PyEval_SaveThread},
291071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE
292071d4279SBram Moolenaar     {"PyGILState_Ensure", (PYTHON_PROC*)&dll_PyGILState_Ensure},
293071d4279SBram Moolenaar     {"PyGILState_Release", (PYTHON_PROC*)&dll_PyGILState_Release},
294071d4279SBram Moolenaar # endif
295071d4279SBram Moolenaar     {"PyInt_AsLong", (PYTHON_PROC*)&dll_PyInt_AsLong},
296071d4279SBram Moolenaar     {"PyInt_FromLong", (PYTHON_PROC*)&dll_PyInt_FromLong},
297071d4279SBram Moolenaar     {"PyInt_Type", (PYTHON_PROC*)&dll_PyInt_Type},
298071d4279SBram Moolenaar     {"PyList_GetItem", (PYTHON_PROC*)&dll_PyList_GetItem},
2990ac9379aSBram Moolenaar     {"PyList_Append", (PYTHON_PROC*)&dll_PyList_Append},
300071d4279SBram Moolenaar     {"PyList_New", (PYTHON_PROC*)&dll_PyList_New},
301071d4279SBram Moolenaar     {"PyList_SetItem", (PYTHON_PROC*)&dll_PyList_SetItem},
302071d4279SBram Moolenaar     {"PyList_Size", (PYTHON_PROC*)&dll_PyList_Size},
303071d4279SBram Moolenaar     {"PyList_Type", (PYTHON_PROC*)&dll_PyList_Type},
304071d4279SBram Moolenaar     {"PyImport_ImportModule", (PYTHON_PROC*)&dll_PyImport_ImportModule},
305071d4279SBram Moolenaar     {"PyDict_GetItemString", (PYTHON_PROC*)&dll_PyDict_GetItemString},
3060ac9379aSBram Moolenaar     {"PyDict_New", (PYTHON_PROC*)&dll_PyDict_New},
307071d4279SBram Moolenaar     {"PyModule_GetDict", (PYTHON_PROC*)&dll_PyModule_GetDict},
308071d4279SBram Moolenaar     {"PyRun_SimpleString", (PYTHON_PROC*)&dll_PyRun_SimpleString},
309071d4279SBram Moolenaar     {"PyString_AsString", (PYTHON_PROC*)&dll_PyString_AsString},
310071d4279SBram Moolenaar     {"PyString_FromString", (PYTHON_PROC*)&dll_PyString_FromString},
311071d4279SBram Moolenaar     {"PyString_FromStringAndSize", (PYTHON_PROC*)&dll_PyString_FromStringAndSize},
312071d4279SBram Moolenaar     {"PyString_Size", (PYTHON_PROC*)&dll_PyString_Size},
313071d4279SBram Moolenaar     {"PyString_Type", (PYTHON_PROC*)&dll_PyString_Type},
314071d4279SBram Moolenaar     {"PySys_SetObject", (PYTHON_PROC*)&dll_PySys_SetObject},
315071d4279SBram Moolenaar     {"PySys_SetArgv", (PYTHON_PROC*)&dll_PySys_SetArgv},
316071d4279SBram Moolenaar     {"PyType_Type", (PYTHON_PROC*)&dll_PyType_Type},
31730fec7bcSBram Moolenaar     {"PyType_Ready", (PYTHON_PROC*)&dll_PyType_Ready},
318071d4279SBram Moolenaar     {"Py_BuildValue", (PYTHON_PROC*)&dll_Py_BuildValue},
319071d4279SBram Moolenaar     {"Py_FindMethod", (PYTHON_PROC*)&dll_Py_FindMethod},
320e7cb9cf6SBram Moolenaar # if (PY_VERSION_HEX >= 0x02050000) && SIZEOF_SIZE_T != SIZEOF_INT
321e7cb9cf6SBram Moolenaar     {"Py_InitModule4_64", (PYTHON_PROC*)&dll_Py_InitModule4},
322e7cb9cf6SBram Moolenaar # else
323071d4279SBram Moolenaar     {"Py_InitModule4", (PYTHON_PROC*)&dll_Py_InitModule4},
324e7cb9cf6SBram Moolenaar # endif
325644d37b8SBram Moolenaar     {"Py_SetPythonHome", (PYTHON_PROC*)&dll_Py_SetPythonHome},
326071d4279SBram Moolenaar     {"Py_Initialize", (PYTHON_PROC*)&dll_Py_Initialize},
3270e21a3f6SBram Moolenaar     {"Py_Finalize", (PYTHON_PROC*)&dll_Py_Finalize},
3280e21a3f6SBram Moolenaar     {"Py_IsInitialized", (PYTHON_PROC*)&dll_Py_IsInitialized},
329071d4279SBram Moolenaar     {"_PyObject_New", (PYTHON_PROC*)&dll__PyObject_New},
330071d4279SBram Moolenaar     {"PyObject_Init", (PYTHON_PROC*)&dll__PyObject_Init},
331071d4279SBram Moolenaar     {"_Py_NoneStruct", (PYTHON_PROC*)&dll__Py_NoneStruct},
332071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
333071d4279SBram Moolenaar     {"PyType_IsSubtype", (PYTHON_PROC*)&dll_PyType_IsSubtype},
334071d4279SBram Moolenaar # endif
335071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
336071d4279SBram Moolenaar     {"PyObject_Malloc", (PYTHON_PROC*)&dll_PyObject_Malloc},
337071d4279SBram Moolenaar     {"PyObject_Free", (PYTHON_PROC*)&dll_PyObject_Free},
338071d4279SBram Moolenaar # endif
339071d4279SBram Moolenaar     {"", NULL},
340071d4279SBram Moolenaar };
341071d4279SBram Moolenaar 
342071d4279SBram Moolenaar /*
343071d4279SBram Moolenaar  * Free python.dll
344071d4279SBram Moolenaar  */
345071d4279SBram Moolenaar     static void
346071d4279SBram Moolenaar end_dynamic_python(void)
347071d4279SBram Moolenaar {
348071d4279SBram Moolenaar     if (hinstPython)
349071d4279SBram Moolenaar     {
350bd5e15fdSBram Moolenaar 	close_dll(hinstPython);
351071d4279SBram Moolenaar 	hinstPython = 0;
352071d4279SBram Moolenaar     }
353071d4279SBram Moolenaar }
354071d4279SBram Moolenaar 
355071d4279SBram Moolenaar /*
356071d4279SBram Moolenaar  * Load library and get all pointers.
357071d4279SBram Moolenaar  * Parameter 'libname' provides name of DLL.
358071d4279SBram Moolenaar  * Return OK or FAIL.
359071d4279SBram Moolenaar  */
360071d4279SBram Moolenaar     static int
361071d4279SBram Moolenaar python_runtime_link_init(char *libname, int verbose)
362071d4279SBram Moolenaar {
363071d4279SBram Moolenaar     int i;
364071d4279SBram Moolenaar 
365644d37b8SBram Moolenaar #if !(defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)) && defined(UNIX) && defined(FEAT_PYTHON3)
366b744b2faSBram Moolenaar     /* Can't have Python and Python3 loaded at the same time.
367b744b2faSBram Moolenaar      * It cause a crash, because RTLD_GLOBAL is needed for
368b744b2faSBram Moolenaar      * standard C extension libraries of one or both python versions. */
3694c3a326cSBram Moolenaar     if (python3_loaded())
3704c3a326cSBram Moolenaar     {
371*9dc93ae4SBram Moolenaar 	if (verbose)
372b744b2faSBram Moolenaar 	    EMSG(_("E836: This Vim cannot execute :python after using :py3"));
3734c3a326cSBram Moolenaar 	return FAIL;
3744c3a326cSBram Moolenaar     }
3754c3a326cSBram Moolenaar #endif
3764c3a326cSBram Moolenaar 
377071d4279SBram Moolenaar     if (hinstPython)
378071d4279SBram Moolenaar 	return OK;
379bd5e15fdSBram Moolenaar     hinstPython = load_dll(libname);
380071d4279SBram Moolenaar     if (!hinstPython)
381071d4279SBram Moolenaar     {
382071d4279SBram Moolenaar 	if (verbose)
383071d4279SBram Moolenaar 	    EMSG2(_(e_loadlib), libname);
384071d4279SBram Moolenaar 	return FAIL;
385071d4279SBram Moolenaar     }
386071d4279SBram Moolenaar 
387071d4279SBram Moolenaar     for (i = 0; python_funcname_table[i].ptr; ++i)
388071d4279SBram Moolenaar     {
389bd5e15fdSBram Moolenaar 	if ((*python_funcname_table[i].ptr = symbol_from_dll(hinstPython,
390071d4279SBram Moolenaar 			python_funcname_table[i].name)) == NULL)
391071d4279SBram Moolenaar 	{
392bd5e15fdSBram Moolenaar 	    close_dll(hinstPython);
393071d4279SBram Moolenaar 	    hinstPython = 0;
394071d4279SBram Moolenaar 	    if (verbose)
395071d4279SBram Moolenaar 		EMSG2(_(e_loadfunc), python_funcname_table[i].name);
396071d4279SBram Moolenaar 	    return FAIL;
397071d4279SBram Moolenaar 	}
398071d4279SBram Moolenaar     }
399071d4279SBram Moolenaar     return OK;
400071d4279SBram Moolenaar }
401071d4279SBram Moolenaar 
402071d4279SBram Moolenaar /*
403071d4279SBram Moolenaar  * If python is enabled (there is installed python on Windows system) return
404071d4279SBram Moolenaar  * TRUE, else FALSE.
405071d4279SBram Moolenaar  */
406071d4279SBram Moolenaar     int
407e7cb9cf6SBram Moolenaar python_enabled(int verbose)
408071d4279SBram Moolenaar {
409071d4279SBram Moolenaar     return python_runtime_link_init(DYNAMIC_PYTHON_DLL, verbose) == OK;
410071d4279SBram Moolenaar }
411071d4279SBram Moolenaar 
412ca8a4dfeSBram Moolenaar /*
413ca8a4dfeSBram Moolenaar  * Load the standard Python exceptions - don't import the symbols from the
414071d4279SBram Moolenaar  * DLL, as this can cause errors (importing data symbols is not reliable).
415071d4279SBram Moolenaar  */
416071d4279SBram Moolenaar     static void
417ca8a4dfeSBram Moolenaar get_exceptions(void)
418071d4279SBram Moolenaar {
419071d4279SBram Moolenaar     PyObject *exmod = PyImport_ImportModule("exceptions");
420071d4279SBram Moolenaar     PyObject *exdict = PyModule_GetDict(exmod);
421071d4279SBram Moolenaar     imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError");
422071d4279SBram Moolenaar     imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError");
423071d4279SBram Moolenaar     imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt");
424071d4279SBram Moolenaar     imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError");
425071d4279SBram Moolenaar     imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError");
426071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_AttributeError);
427071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_IndexError);
428071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_KeyboardInterrupt);
429071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_TypeError);
430071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_ValueError);
431071d4279SBram Moolenaar     Py_XDECREF(exmod);
432071d4279SBram Moolenaar }
433071d4279SBram Moolenaar #endif /* DYNAMIC_PYTHON */
434071d4279SBram Moolenaar 
435ca8a4dfeSBram Moolenaar static PyObject *BufferNew (buf_T *);
436ca8a4dfeSBram Moolenaar static PyObject *WindowNew(win_T *);
437ca8a4dfeSBram Moolenaar static PyObject *LineToString(const char *);
438ca8a4dfeSBram Moolenaar 
439ca8a4dfeSBram Moolenaar static PyTypeObject RangeType;
440ca8a4dfeSBram Moolenaar 
441170bf1aeSBram Moolenaar /*
442170bf1aeSBram Moolenaar  * Include the code shared with if_python3.c
443170bf1aeSBram Moolenaar  */
444170bf1aeSBram Moolenaar #include "if_py_both.h"
445170bf1aeSBram Moolenaar 
446170bf1aeSBram Moolenaar 
447071d4279SBram Moolenaar /******************************************************
448071d4279SBram Moolenaar  * Internal function prototypes.
449071d4279SBram Moolenaar  */
450071d4279SBram Moolenaar 
451e7cb9cf6SBram Moolenaar static PyInt RangeStart;
452e7cb9cf6SBram Moolenaar static PyInt RangeEnd;
453071d4279SBram Moolenaar 
454071d4279SBram Moolenaar static void PythonIO_Flush(void);
455071d4279SBram Moolenaar static int PythonIO_Init(void);
456071d4279SBram Moolenaar static int PythonMod_Init(void);
457071d4279SBram Moolenaar 
458071d4279SBram Moolenaar /* Utility functions for the vim/python interface
459071d4279SBram Moolenaar  * ----------------------------------------------
460071d4279SBram Moolenaar  */
461071d4279SBram Moolenaar 
462e7cb9cf6SBram Moolenaar static int SetBufferLineList(buf_T *, PyInt, PyInt, PyObject *, PyInt *);
463071d4279SBram Moolenaar 
464071d4279SBram Moolenaar 
465071d4279SBram Moolenaar /******************************************************
466071d4279SBram Moolenaar  * 1. Python interpreter main program.
467071d4279SBram Moolenaar  */
468071d4279SBram Moolenaar 
469071d4279SBram Moolenaar static int initialised = 0;
470071d4279SBram Moolenaar 
471071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */
472071d4279SBram Moolenaar typedef PyObject PyThreadState;
4739ba0eb85SBram Moolenaar #endif
474071d4279SBram Moolenaar 
4759ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE
4769ba0eb85SBram Moolenaar static PyGILState_STATE pygilstate = PyGILState_UNLOCKED;
4779ba0eb85SBram Moolenaar #else
478071d4279SBram Moolenaar static PyThreadState *saved_python_thread = NULL;
4799ba0eb85SBram Moolenaar #endif
480071d4279SBram Moolenaar 
481071d4279SBram Moolenaar /*
482071d4279SBram Moolenaar  * Suspend a thread of the Python interpreter, other threads are allowed to
483071d4279SBram Moolenaar  * run.
484071d4279SBram Moolenaar  */
485293ee4d4SBram Moolenaar     static void
486293ee4d4SBram Moolenaar Python_SaveThread(void)
487071d4279SBram Moolenaar {
4889ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE
4899ba0eb85SBram Moolenaar     PyGILState_Release(pygilstate);
4909ba0eb85SBram Moolenaar #else
491071d4279SBram Moolenaar     saved_python_thread = PyEval_SaveThread();
4929ba0eb85SBram Moolenaar #endif
493071d4279SBram Moolenaar }
494071d4279SBram Moolenaar 
495071d4279SBram Moolenaar /*
496071d4279SBram Moolenaar  * Restore a thread of the Python interpreter, waits for other threads to
497071d4279SBram Moolenaar  * block.
498071d4279SBram Moolenaar  */
499293ee4d4SBram Moolenaar     static void
500293ee4d4SBram Moolenaar Python_RestoreThread(void)
501071d4279SBram Moolenaar {
5029ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE
5039ba0eb85SBram Moolenaar     pygilstate = PyGILState_Ensure();
5049ba0eb85SBram Moolenaar #else
505071d4279SBram Moolenaar     PyEval_RestoreThread(saved_python_thread);
506071d4279SBram Moolenaar     saved_python_thread = NULL;
507071d4279SBram Moolenaar #endif
5089ba0eb85SBram Moolenaar }
509071d4279SBram Moolenaar 
510071d4279SBram Moolenaar     void
511071d4279SBram Moolenaar python_end()
512071d4279SBram Moolenaar {
513a5792f58SBram Moolenaar     static int recurse = 0;
514a5792f58SBram Moolenaar 
515a5792f58SBram Moolenaar     /* If a crash occurs while doing this, don't try again. */
516a5792f58SBram Moolenaar     if (recurse != 0)
517a5792f58SBram Moolenaar 	return;
518a5792f58SBram Moolenaar 
519a5792f58SBram Moolenaar     ++recurse;
520a5792f58SBram Moolenaar 
521071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
5220e21a3f6SBram Moolenaar     if (hinstPython && Py_IsInitialized())
5239ba0eb85SBram Moolenaar     {
5249ba0eb85SBram Moolenaar 	Python_RestoreThread();	    /* enter python */
5250e21a3f6SBram Moolenaar 	Py_Finalize();
5269ba0eb85SBram Moolenaar     }
527071d4279SBram Moolenaar     end_dynamic_python();
5280e21a3f6SBram Moolenaar #else
5290e21a3f6SBram Moolenaar     if (Py_IsInitialized())
5309ba0eb85SBram Moolenaar     {
5319ba0eb85SBram Moolenaar 	Python_RestoreThread();	    /* enter python */
5320e21a3f6SBram Moolenaar 	Py_Finalize();
5339ba0eb85SBram Moolenaar     }
534071d4279SBram Moolenaar #endif
535a5792f58SBram Moolenaar 
536a5792f58SBram Moolenaar     --recurse;
537071d4279SBram Moolenaar }
538071d4279SBram Moolenaar 
5394c3a326cSBram Moolenaar #if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
5404c3a326cSBram Moolenaar     int
5414c3a326cSBram Moolenaar python_loaded()
5424c3a326cSBram Moolenaar {
5434c3a326cSBram Moolenaar     return (hinstPython != 0);
5444c3a326cSBram Moolenaar }
5454c3a326cSBram Moolenaar #endif
5464c3a326cSBram Moolenaar 
547071d4279SBram Moolenaar     static int
548071d4279SBram Moolenaar Python_Init(void)
549071d4279SBram Moolenaar {
550071d4279SBram Moolenaar     if (!initialised)
551071d4279SBram Moolenaar     {
552071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
553071d4279SBram Moolenaar 	if (!python_enabled(TRUE))
554071d4279SBram Moolenaar 	{
555071d4279SBram Moolenaar 	    EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded."));
556071d4279SBram Moolenaar 	    goto fail;
557071d4279SBram Moolenaar 	}
558071d4279SBram Moolenaar #endif
559071d4279SBram Moolenaar 
560644d37b8SBram Moolenaar #ifdef PYTHON_HOME
561644d37b8SBram Moolenaar 	Py_SetPythonHome(PYTHON_HOME);
562644d37b8SBram Moolenaar #endif
563644d37b8SBram Moolenaar 
564170bf1aeSBram Moolenaar 	init_structs();
565170bf1aeSBram Moolenaar 
566071d4279SBram Moolenaar #if !defined(MACOS) || defined(MACOS_X_UNIX)
567071d4279SBram Moolenaar 	Py_Initialize();
568071d4279SBram Moolenaar #else
569071d4279SBram Moolenaar 	PyMac_Initialize();
570071d4279SBram Moolenaar #endif
571071d4279SBram Moolenaar 	/* initialise threads */
572071d4279SBram Moolenaar 	PyEval_InitThreads();
573071d4279SBram Moolenaar 
574071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
575071d4279SBram Moolenaar 	get_exceptions();
576071d4279SBram Moolenaar #endif
577071d4279SBram Moolenaar 
578071d4279SBram Moolenaar 	if (PythonIO_Init())
579071d4279SBram Moolenaar 	    goto fail;
580071d4279SBram Moolenaar 
581071d4279SBram Moolenaar 	if (PythonMod_Init())
582071d4279SBram Moolenaar 	    goto fail;
583071d4279SBram Moolenaar 
5849774ecc8SBram Moolenaar 	/* Remove the element from sys.path that was added because of our
5859774ecc8SBram Moolenaar 	 * argv[0] value in PythonMod_Init().  Previously we used an empty
5869774ecc8SBram Moolenaar 	 * string, but dependinding on the OS we then get an empty entry or
5879774ecc8SBram Moolenaar 	 * the current directory in sys.path. */
5889774ecc8SBram Moolenaar 	PyRun_SimpleString("import sys; sys.path = filter(lambda x: x != '/must>not&exist', sys.path)");
5899774ecc8SBram Moolenaar 
590293ee4d4SBram Moolenaar 	/* the first python thread is vim's, release the lock */
591071d4279SBram Moolenaar 	Python_SaveThread();
592071d4279SBram Moolenaar 
593071d4279SBram Moolenaar 	initialised = 1;
594071d4279SBram Moolenaar     }
595071d4279SBram Moolenaar 
596071d4279SBram Moolenaar     return 0;
597071d4279SBram Moolenaar 
598071d4279SBram Moolenaar fail:
599071d4279SBram Moolenaar     /* We call PythonIO_Flush() here to print any Python errors.
600071d4279SBram Moolenaar      * This is OK, as it is possible to call this function even
601071d4279SBram Moolenaar      * if PythonIO_Init() has not completed successfully (it will
602071d4279SBram Moolenaar      * not do anything in this case).
603071d4279SBram Moolenaar      */
604071d4279SBram Moolenaar     PythonIO_Flush();
605071d4279SBram Moolenaar     return -1;
606071d4279SBram Moolenaar }
607071d4279SBram Moolenaar 
608071d4279SBram Moolenaar /*
609071d4279SBram Moolenaar  * External interface
610071d4279SBram Moolenaar  */
611071d4279SBram Moolenaar     static void
612071d4279SBram Moolenaar DoPythonCommand(exarg_T *eap, const char *cmd)
613071d4279SBram Moolenaar {
6149ba0eb85SBram Moolenaar #ifndef PY_CAN_RECURSE
615071d4279SBram Moolenaar     static int		recursive = 0;
616071d4279SBram Moolenaar #endif
617071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
618071d4279SBram Moolenaar     GrafPtr		oldPort;
619071d4279SBram Moolenaar #endif
620071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
621071d4279SBram Moolenaar     char		*saved_locale;
622071d4279SBram Moolenaar #endif
623071d4279SBram Moolenaar 
624071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE
625071d4279SBram Moolenaar     if (recursive)
626071d4279SBram Moolenaar     {
627071d4279SBram Moolenaar 	EMSG(_("E659: Cannot invoke Python recursively"));
628071d4279SBram Moolenaar 	return;
629071d4279SBram Moolenaar     }
630071d4279SBram Moolenaar     ++recursive;
631071d4279SBram Moolenaar #endif
632071d4279SBram Moolenaar 
633071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
634071d4279SBram Moolenaar     GetPort(&oldPort);
635071d4279SBram Moolenaar     /* Check if the Python library is available */
636071d4279SBram Moolenaar     if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress)
637071d4279SBram Moolenaar 	goto theend;
638071d4279SBram Moolenaar #endif
639071d4279SBram Moolenaar     if (Python_Init())
640071d4279SBram Moolenaar 	goto theend;
641071d4279SBram Moolenaar 
642071d4279SBram Moolenaar     RangeStart = eap->line1;
643071d4279SBram Moolenaar     RangeEnd = eap->line2;
644071d4279SBram Moolenaar     Python_Release_Vim();	    /* leave vim */
645071d4279SBram Moolenaar 
646071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
647071d4279SBram Moolenaar     /* Python only works properly when the LC_NUMERIC locale is "C". */
648071d4279SBram Moolenaar     saved_locale = setlocale(LC_NUMERIC, NULL);
649071d4279SBram Moolenaar     if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0)
650071d4279SBram Moolenaar 	saved_locale = NULL;
651071d4279SBram Moolenaar     else
652071d4279SBram Moolenaar     {
653071d4279SBram Moolenaar 	/* Need to make a copy, value may change when setting new locale. */
654071d4279SBram Moolenaar 	saved_locale = (char *)vim_strsave((char_u *)saved_locale);
655071d4279SBram Moolenaar 	(void)setlocale(LC_NUMERIC, "C");
656071d4279SBram Moolenaar     }
657071d4279SBram Moolenaar #endif
658071d4279SBram Moolenaar 
659071d4279SBram Moolenaar     Python_RestoreThread();	    /* enter python */
660071d4279SBram Moolenaar 
661071d4279SBram Moolenaar     PyRun_SimpleString((char *)(cmd));
662071d4279SBram Moolenaar 
663071d4279SBram Moolenaar     Python_SaveThread();	    /* leave python */
664071d4279SBram Moolenaar 
665071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
666071d4279SBram Moolenaar     if (saved_locale != NULL)
667071d4279SBram Moolenaar     {
668071d4279SBram Moolenaar 	(void)setlocale(LC_NUMERIC, saved_locale);
669071d4279SBram Moolenaar 	vim_free(saved_locale);
670071d4279SBram Moolenaar     }
671071d4279SBram Moolenaar #endif
672071d4279SBram Moolenaar 
673071d4279SBram Moolenaar     Python_Lock_Vim();		    /* enter vim */
674071d4279SBram Moolenaar     PythonIO_Flush();
675071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
676071d4279SBram Moolenaar     SetPort(oldPort);
677071d4279SBram Moolenaar #endif
678071d4279SBram Moolenaar 
679071d4279SBram Moolenaar theend:
680071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE
681071d4279SBram Moolenaar     --recursive;
682071d4279SBram Moolenaar #endif
683071d4279SBram Moolenaar     return;	    /* keeps lint happy */
684071d4279SBram Moolenaar }
685071d4279SBram Moolenaar 
686071d4279SBram Moolenaar /*
687071d4279SBram Moolenaar  * ":python"
688071d4279SBram Moolenaar  */
689071d4279SBram Moolenaar     void
690071d4279SBram Moolenaar ex_python(exarg_T *eap)
691071d4279SBram Moolenaar {
692071d4279SBram Moolenaar     char_u *script;
693071d4279SBram Moolenaar 
694071d4279SBram Moolenaar     script = script_get(eap, eap->arg);
695071d4279SBram Moolenaar     if (!eap->skip)
696071d4279SBram Moolenaar     {
697071d4279SBram Moolenaar 	if (script == NULL)
698071d4279SBram Moolenaar 	    DoPythonCommand(eap, (char *)eap->arg);
699071d4279SBram Moolenaar 	else
700071d4279SBram Moolenaar 	    DoPythonCommand(eap, (char *)script);
701071d4279SBram Moolenaar     }
702071d4279SBram Moolenaar     vim_free(script);
703071d4279SBram Moolenaar }
704071d4279SBram Moolenaar 
705071d4279SBram Moolenaar #define BUFFER_SIZE 1024
706071d4279SBram Moolenaar 
707071d4279SBram Moolenaar /*
708071d4279SBram Moolenaar  * ":pyfile"
709071d4279SBram Moolenaar  */
710071d4279SBram Moolenaar     void
711071d4279SBram Moolenaar ex_pyfile(exarg_T *eap)
712071d4279SBram Moolenaar {
713071d4279SBram Moolenaar     static char buffer[BUFFER_SIZE];
714071d4279SBram Moolenaar     const char *file = (char *)eap->arg;
715071d4279SBram Moolenaar     char *p;
716071d4279SBram Moolenaar 
717071d4279SBram Moolenaar     /* Have to do it like this. PyRun_SimpleFile requires you to pass a
718071d4279SBram Moolenaar      * stdio file pointer, but Vim and the Python DLL are compiled with
719071d4279SBram Moolenaar      * different options under Windows, meaning that stdio pointers aren't
720071d4279SBram Moolenaar      * compatible between the two. Yuk.
721071d4279SBram Moolenaar      *
722071d4279SBram Moolenaar      * Put the string "execfile('file')" into buffer. But, we need to
723071d4279SBram Moolenaar      * escape any backslashes or single quotes in the file name, so that
724071d4279SBram Moolenaar      * Python won't mangle the file name.
725071d4279SBram Moolenaar      */
726071d4279SBram Moolenaar     strcpy(buffer, "execfile('");
727071d4279SBram Moolenaar     p = buffer + 10; /* size of "execfile('" */
728071d4279SBram Moolenaar 
729071d4279SBram Moolenaar     while (*file && p < buffer + (BUFFER_SIZE - 3))
730071d4279SBram Moolenaar     {
731071d4279SBram Moolenaar 	if (*file == '\\' || *file == '\'')
732071d4279SBram Moolenaar 	    *p++ = '\\';
733071d4279SBram Moolenaar 	*p++ = *file++;
734071d4279SBram Moolenaar     }
735071d4279SBram Moolenaar 
736071d4279SBram Moolenaar     /* If we didn't finish the file name, we hit a buffer overflow */
737071d4279SBram Moolenaar     if (*file != '\0')
738071d4279SBram Moolenaar 	return;
739071d4279SBram Moolenaar 
740071d4279SBram Moolenaar     /* Put in the terminating "')" and a null */
741071d4279SBram Moolenaar     *p++ = '\'';
742071d4279SBram Moolenaar     *p++ = ')';
743071d4279SBram Moolenaar     *p++ = '\0';
744071d4279SBram Moolenaar 
745071d4279SBram Moolenaar     /* Execute the file */
746071d4279SBram Moolenaar     DoPythonCommand(eap, buffer);
747071d4279SBram Moolenaar }
748071d4279SBram Moolenaar 
749071d4279SBram Moolenaar /******************************************************
750071d4279SBram Moolenaar  * 2. Python output stream: writes output via [e]msg().
751071d4279SBram Moolenaar  */
752071d4279SBram Moolenaar 
753071d4279SBram Moolenaar /* Implementation functions
754071d4279SBram Moolenaar  */
755071d4279SBram Moolenaar 
756071d4279SBram Moolenaar     static PyObject *
757071d4279SBram Moolenaar OutputGetattr(PyObject *self, char *name)
758071d4279SBram Moolenaar {
759071d4279SBram Moolenaar     if (strcmp(name, "softspace") == 0)
760071d4279SBram Moolenaar 	return PyInt_FromLong(((OutputObject *)(self))->softspace);
761071d4279SBram Moolenaar 
762071d4279SBram Moolenaar     return Py_FindMethod(OutputMethods, self, name);
763071d4279SBram Moolenaar }
764071d4279SBram Moolenaar 
765071d4279SBram Moolenaar     static int
766071d4279SBram Moolenaar OutputSetattr(PyObject *self, char *name, PyObject *val)
767071d4279SBram Moolenaar {
768071d4279SBram Moolenaar     if (val == NULL) {
769071d4279SBram Moolenaar 	PyErr_SetString(PyExc_AttributeError, _("can't delete OutputObject attributes"));
770071d4279SBram Moolenaar 	return -1;
771071d4279SBram Moolenaar     }
772071d4279SBram Moolenaar 
773071d4279SBram Moolenaar     if (strcmp(name, "softspace") == 0)
774071d4279SBram Moolenaar     {
775071d4279SBram Moolenaar 	if (!PyInt_Check(val)) {
776071d4279SBram Moolenaar 	    PyErr_SetString(PyExc_TypeError, _("softspace must be an integer"));
777071d4279SBram Moolenaar 	    return -1;
778071d4279SBram Moolenaar 	}
779071d4279SBram Moolenaar 
780071d4279SBram Moolenaar 	((OutputObject *)(self))->softspace = PyInt_AsLong(val);
781071d4279SBram Moolenaar 	return 0;
782071d4279SBram Moolenaar     }
783071d4279SBram Moolenaar 
784071d4279SBram Moolenaar     PyErr_SetString(PyExc_AttributeError, _("invalid attribute"));
785071d4279SBram Moolenaar     return -1;
786071d4279SBram Moolenaar }
787071d4279SBram Moolenaar 
788071d4279SBram Moolenaar /***************/
789071d4279SBram Moolenaar 
790071d4279SBram Moolenaar     static int
791071d4279SBram Moolenaar PythonIO_Init(void)
792071d4279SBram Moolenaar {
793071d4279SBram Moolenaar     /* Fixups... */
79421377c8dSBram Moolenaar     PyType_Ready(&OutputType);
795071d4279SBram Moolenaar 
796170bf1aeSBram Moolenaar     return PythonIO_Init_io();
797071d4279SBram Moolenaar }
798071d4279SBram Moolenaar 
799071d4279SBram Moolenaar /******************************************************
800071d4279SBram Moolenaar  * 3. Implementation of the Vim module for Python
801071d4279SBram Moolenaar  */
802071d4279SBram Moolenaar 
803071d4279SBram Moolenaar /* Window type - Implementation functions
804071d4279SBram Moolenaar  * --------------------------------------
805071d4279SBram Moolenaar  */
806071d4279SBram Moolenaar 
807071d4279SBram Moolenaar #define WindowType_Check(obj) ((obj)->ob_type == &WindowType)
808071d4279SBram Moolenaar 
809071d4279SBram Moolenaar static void WindowDestructor(PyObject *);
810071d4279SBram Moolenaar static PyObject *WindowGetattr(PyObject *, char *);
811071d4279SBram Moolenaar 
812071d4279SBram Moolenaar /* Buffer type - Implementation functions
813071d4279SBram Moolenaar  * --------------------------------------
814071d4279SBram Moolenaar  */
815071d4279SBram Moolenaar 
816071d4279SBram Moolenaar #define BufferType_Check(obj) ((obj)->ob_type == &BufferType)
817071d4279SBram Moolenaar 
818071d4279SBram Moolenaar static void BufferDestructor(PyObject *);
819071d4279SBram Moolenaar static PyObject *BufferGetattr(PyObject *, char *);
820071d4279SBram Moolenaar static PyObject *BufferRepr(PyObject *);
821071d4279SBram Moolenaar 
8222c45e945SBram Moolenaar static PyInt BufferLength(PyObject *);
8232c45e945SBram Moolenaar static PyObject *BufferItem(PyObject *, PyInt);
8242c45e945SBram Moolenaar static PyObject *BufferSlice(PyObject *, PyInt, PyInt);
8252c45e945SBram Moolenaar static PyInt BufferAssItem(PyObject *, PyInt, PyObject *);
8262c45e945SBram Moolenaar static PyInt BufferAssSlice(PyObject *, PyInt, PyInt, PyObject *);
827071d4279SBram Moolenaar 
828071d4279SBram Moolenaar /* Line range type - Implementation functions
829071d4279SBram Moolenaar  * --------------------------------------
830071d4279SBram Moolenaar  */
831071d4279SBram Moolenaar 
832071d4279SBram Moolenaar #define RangeType_Check(obj) ((obj)->ob_type == &RangeType)
833071d4279SBram Moolenaar 
8342c45e945SBram Moolenaar static PyInt RangeAssItem(PyObject *, PyInt, PyObject *);
8352c45e945SBram Moolenaar static PyInt RangeAssSlice(PyObject *, PyInt, PyInt, PyObject *);
836071d4279SBram Moolenaar 
837071d4279SBram Moolenaar /* Current objects type - Implementation functions
838071d4279SBram Moolenaar  * -----------------------------------------------
839071d4279SBram Moolenaar  */
840071d4279SBram Moolenaar 
841071d4279SBram Moolenaar static PyObject *CurrentGetattr(PyObject *, char *);
842071d4279SBram Moolenaar static int CurrentSetattr(PyObject *, char *, PyObject *);
843071d4279SBram Moolenaar 
844071d4279SBram Moolenaar static PySequenceMethods BufferAsSeq = {
8452c45e945SBram Moolenaar     (PyInquiry)		BufferLength,	    /* sq_length,    len(x)   */
846071d4279SBram Moolenaar     (binaryfunc)	0, /* BufferConcat, */	     /* sq_concat,    x+y      */
8472c45e945SBram Moolenaar     (PyIntArgFunc)	0, /* BufferRepeat, */	     /* sq_repeat,    x*n      */
8482c45e945SBram Moolenaar     (PyIntArgFunc)	BufferItem,	    /* sq_item,      x[i]     */
8492c45e945SBram Moolenaar     (PyIntIntArgFunc)	BufferSlice,	    /* sq_slice,     x[i:j]   */
8502c45e945SBram Moolenaar     (PyIntObjArgProc)	BufferAssItem,	    /* sq_ass_item,  x[i]=v   */
8512c45e945SBram Moolenaar     (PyIntIntObjArgProc)	BufferAssSlice,     /* sq_ass_slice, x[i:j]=v */
852071d4279SBram Moolenaar };
853071d4279SBram Moolenaar 
854071d4279SBram Moolenaar static PyTypeObject BufferType = {
855071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
856071d4279SBram Moolenaar     0,
857071d4279SBram Moolenaar     "buffer",
858071d4279SBram Moolenaar     sizeof(BufferObject),
859071d4279SBram Moolenaar     0,
860071d4279SBram Moolenaar 
861071d4279SBram Moolenaar     (destructor)    BufferDestructor,	/* tp_dealloc,	refcount==0  */
862071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
863071d4279SBram Moolenaar     (getattrfunc)   BufferGetattr,	/* tp_getattr,	x.attr	     */
864071d4279SBram Moolenaar     (setattrfunc)   0,			/* tp_setattr,	x.attr=v     */
865071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
866071d4279SBram Moolenaar     (reprfunc)	    BufferRepr,		/* tp_repr,	`x`, print x */
867071d4279SBram Moolenaar 
868071d4279SBram Moolenaar     0,		    /* as number */
869071d4279SBram Moolenaar     &BufferAsSeq,   /* as sequence */
870071d4279SBram Moolenaar     0,		    /* as mapping */
871071d4279SBram Moolenaar 
872071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
873071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
874071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
875071d4279SBram Moolenaar };
876071d4279SBram Moolenaar 
877071d4279SBram Moolenaar /* Buffer object - Implementation
878071d4279SBram Moolenaar  */
879071d4279SBram Moolenaar 
880071d4279SBram Moolenaar     static PyObject *
881071d4279SBram Moolenaar BufferNew(buf_T *buf)
882071d4279SBram Moolenaar {
883071d4279SBram Moolenaar     /* We need to handle deletion of buffers underneath us.
884e344beadSBram Moolenaar      * If we add a "b_python_ref" field to the buf_T structure,
885071d4279SBram Moolenaar      * then we can get at it in buf_freeall() in vim. We then
886071d4279SBram Moolenaar      * need to create only ONE Python object per buffer - if
887071d4279SBram Moolenaar      * we try to create a second, just INCREF the existing one
888071d4279SBram Moolenaar      * and return it. The (single) Python object referring to
889e344beadSBram Moolenaar      * the buffer is stored in "b_python_ref".
890071d4279SBram Moolenaar      * Question: what to do on a buf_freeall(). We'll probably
891071d4279SBram Moolenaar      * have to either delete the Python object (DECREF it to
892071d4279SBram Moolenaar      * zero - a bad idea, as it leaves dangling refs!) or
893071d4279SBram Moolenaar      * set the buf_T * value to an invalid value (-1?), which
894071d4279SBram Moolenaar      * means we need checks in all access functions... Bah.
895071d4279SBram Moolenaar      */
896071d4279SBram Moolenaar 
897071d4279SBram Moolenaar     BufferObject *self;
898071d4279SBram Moolenaar 
899e344beadSBram Moolenaar     if (buf->b_python_ref != NULL)
900071d4279SBram Moolenaar     {
901e344beadSBram Moolenaar 	self = buf->b_python_ref;
902071d4279SBram Moolenaar 	Py_INCREF(self);
903071d4279SBram Moolenaar     }
904071d4279SBram Moolenaar     else
905071d4279SBram Moolenaar     {
906071d4279SBram Moolenaar 	self = PyObject_NEW(BufferObject, &BufferType);
907071d4279SBram Moolenaar 	if (self == NULL)
908071d4279SBram Moolenaar 	    return NULL;
909071d4279SBram Moolenaar 	self->buf = buf;
910e344beadSBram Moolenaar 	buf->b_python_ref = self;
911071d4279SBram Moolenaar     }
912071d4279SBram Moolenaar 
913071d4279SBram Moolenaar     return (PyObject *)(self);
914071d4279SBram Moolenaar }
915071d4279SBram Moolenaar 
916071d4279SBram Moolenaar     static void
917071d4279SBram Moolenaar BufferDestructor(PyObject *self)
918071d4279SBram Moolenaar {
919071d4279SBram Moolenaar     BufferObject *this = (BufferObject *)(self);
920071d4279SBram Moolenaar 
921071d4279SBram Moolenaar     if (this->buf && this->buf != INVALID_BUFFER_VALUE)
922e344beadSBram Moolenaar 	this->buf->b_python_ref = NULL;
923071d4279SBram Moolenaar 
924658ada69SBram Moolenaar     Py_DECREF(self);
925071d4279SBram Moolenaar }
926071d4279SBram Moolenaar 
927071d4279SBram Moolenaar     static PyObject *
928071d4279SBram Moolenaar BufferGetattr(PyObject *self, char *name)
929071d4279SBram Moolenaar {
930071d4279SBram Moolenaar     BufferObject *this = (BufferObject *)(self);
931071d4279SBram Moolenaar 
932071d4279SBram Moolenaar     if (CheckBuffer(this))
933071d4279SBram Moolenaar 	return NULL;
934071d4279SBram Moolenaar 
935071d4279SBram Moolenaar     if (strcmp(name, "name") == 0)
936071d4279SBram Moolenaar 	return Py_BuildValue("s", this->buf->b_ffname);
937071d4279SBram Moolenaar     else if (strcmp(name, "number") == 0)
938e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, this->buf->b_fnum);
939071d4279SBram Moolenaar     else if (strcmp(name,"__members__") == 0)
940071d4279SBram Moolenaar 	return Py_BuildValue("[ss]", "name", "number");
941071d4279SBram Moolenaar     else
942071d4279SBram Moolenaar 	return Py_FindMethod(BufferMethods, self, name);
943071d4279SBram Moolenaar }
944071d4279SBram Moolenaar 
945071d4279SBram Moolenaar     static PyObject *
946071d4279SBram Moolenaar BufferRepr(PyObject *self)
947071d4279SBram Moolenaar {
948555b280fSBram Moolenaar     static char repr[100];
949071d4279SBram Moolenaar     BufferObject *this = (BufferObject *)(self);
950071d4279SBram Moolenaar 
951071d4279SBram Moolenaar     if (this->buf == INVALID_BUFFER_VALUE)
952071d4279SBram Moolenaar     {
953e7cb9cf6SBram Moolenaar 	vim_snprintf(repr, 100, _("<buffer object (deleted) at %p>"), (self));
954071d4279SBram Moolenaar 	return PyString_FromString(repr);
955071d4279SBram Moolenaar     }
956071d4279SBram Moolenaar     else
957071d4279SBram Moolenaar     {
958071d4279SBram Moolenaar 	char *name = (char *)this->buf->b_fname;
959e7cb9cf6SBram Moolenaar 	PyInt len;
960071d4279SBram Moolenaar 
961071d4279SBram Moolenaar 	if (name == NULL)
962071d4279SBram Moolenaar 	    name = "";
963071d4279SBram Moolenaar 	len = strlen(name);
964071d4279SBram Moolenaar 
965071d4279SBram Moolenaar 	if (len > 35)
966071d4279SBram Moolenaar 	    name = name + (35 - len);
967071d4279SBram Moolenaar 
968555b280fSBram Moolenaar 	vim_snprintf(repr, 100, "<buffer %s%s>", len > 35 ? "..." : "", name);
969071d4279SBram Moolenaar 
970071d4279SBram Moolenaar 	return PyString_FromString(repr);
971071d4279SBram Moolenaar     }
972071d4279SBram Moolenaar }
973071d4279SBram Moolenaar 
974071d4279SBram Moolenaar /******************/
975071d4279SBram Moolenaar 
9762c45e945SBram Moolenaar     static PyInt
977071d4279SBram Moolenaar BufferLength(PyObject *self)
978071d4279SBram Moolenaar {
979071d4279SBram Moolenaar     /* HOW DO WE SIGNAL AN ERROR FROM THIS FUNCTION? */
980071d4279SBram Moolenaar     if (CheckBuffer((BufferObject *)(self)))
981071d4279SBram Moolenaar 	return -1; /* ??? */
982071d4279SBram Moolenaar 
983071d4279SBram Moolenaar     return (((BufferObject *)(self))->buf->b_ml.ml_line_count);
984071d4279SBram Moolenaar }
985071d4279SBram Moolenaar 
986071d4279SBram Moolenaar     static PyObject *
9872c45e945SBram Moolenaar BufferItem(PyObject *self, PyInt n)
988071d4279SBram Moolenaar {
989071d4279SBram Moolenaar     return RBItem((BufferObject *)(self), n, 1,
990071d4279SBram Moolenaar 		  (int)((BufferObject *)(self))->buf->b_ml.ml_line_count);
991071d4279SBram Moolenaar }
992071d4279SBram Moolenaar 
993071d4279SBram Moolenaar     static PyObject *
9942c45e945SBram Moolenaar BufferSlice(PyObject *self, PyInt lo, PyInt hi)
995071d4279SBram Moolenaar {
996071d4279SBram Moolenaar     return RBSlice((BufferObject *)(self), lo, hi, 1,
997071d4279SBram Moolenaar 		   (int)((BufferObject *)(self))->buf->b_ml.ml_line_count);
998071d4279SBram Moolenaar }
999071d4279SBram Moolenaar 
10002c45e945SBram Moolenaar     static PyInt
10012c45e945SBram Moolenaar BufferAssItem(PyObject *self, PyInt n, PyObject *val)
1002071d4279SBram Moolenaar {
1003ca8a4dfeSBram Moolenaar     return RBAsItem((BufferObject *)(self), n, val, 1,
1004e7cb9cf6SBram Moolenaar 		     (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1005071d4279SBram Moolenaar 		     NULL);
1006071d4279SBram Moolenaar }
1007071d4279SBram Moolenaar 
10082c45e945SBram Moolenaar     static PyInt
10092c45e945SBram Moolenaar BufferAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
1010071d4279SBram Moolenaar {
101119e60943SBram Moolenaar     return RBAsSlice((BufferObject *)(self), lo, hi, val, 1,
1012e7cb9cf6SBram Moolenaar 		      (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1013071d4279SBram Moolenaar 		      NULL);
1014071d4279SBram Moolenaar }
1015071d4279SBram Moolenaar 
1016071d4279SBram Moolenaar static PySequenceMethods RangeAsSeq = {
10172c45e945SBram Moolenaar     (PyInquiry)		RangeLength,	    /* sq_length,    len(x)   */
1018071d4279SBram Moolenaar     (binaryfunc)	0, /* RangeConcat, */	     /* sq_concat,    x+y      */
10192c45e945SBram Moolenaar     (PyIntArgFunc)	0, /* RangeRepeat, */	     /* sq_repeat,    x*n      */
10202c45e945SBram Moolenaar     (PyIntArgFunc)	RangeItem,	    /* sq_item,      x[i]     */
10212c45e945SBram Moolenaar     (PyIntIntArgFunc)	RangeSlice,	    /* sq_slice,     x[i:j]   */
10222c45e945SBram Moolenaar     (PyIntObjArgProc)	RangeAssItem,	    /* sq_ass_item,  x[i]=v   */
10232c45e945SBram Moolenaar     (PyIntIntObjArgProc)	RangeAssSlice,	    /* sq_ass_slice, x[i:j]=v */
1024071d4279SBram Moolenaar };
1025071d4279SBram Moolenaar 
1026071d4279SBram Moolenaar /* Line range object - Implementation
1027071d4279SBram Moolenaar  */
1028071d4279SBram Moolenaar 
1029071d4279SBram Moolenaar     static void
1030071d4279SBram Moolenaar RangeDestructor(PyObject *self)
1031071d4279SBram Moolenaar {
1032071d4279SBram Moolenaar     Py_DECREF(((RangeObject *)(self))->buf);
1033658ada69SBram Moolenaar     Py_DECREF(self);
1034071d4279SBram Moolenaar }
1035071d4279SBram Moolenaar 
1036071d4279SBram Moolenaar     static PyObject *
1037071d4279SBram Moolenaar RangeGetattr(PyObject *self, char *name)
1038071d4279SBram Moolenaar {
1039071d4279SBram Moolenaar     if (strcmp(name, "start") == 0)
1040e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->start - 1);
1041071d4279SBram Moolenaar     else if (strcmp(name, "end") == 0)
1042e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->end - 1);
1043071d4279SBram Moolenaar     else
1044071d4279SBram Moolenaar 	return Py_FindMethod(RangeMethods, self, name);
1045071d4279SBram Moolenaar }
1046071d4279SBram Moolenaar 
1047071d4279SBram Moolenaar /****************/
1048071d4279SBram Moolenaar 
10492c45e945SBram Moolenaar     static PyInt
10502c45e945SBram Moolenaar RangeAssItem(PyObject *self, PyInt n, PyObject *val)
1051071d4279SBram Moolenaar {
1052ca8a4dfeSBram Moolenaar     return RBAsItem(((RangeObject *)(self))->buf, n, val,
1053071d4279SBram Moolenaar 		     ((RangeObject *)(self))->start,
1054071d4279SBram Moolenaar 		     ((RangeObject *)(self))->end,
1055071d4279SBram Moolenaar 		     &((RangeObject *)(self))->end);
1056071d4279SBram Moolenaar }
1057071d4279SBram Moolenaar 
10582c45e945SBram Moolenaar     static PyInt
10592c45e945SBram Moolenaar RangeAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
1060071d4279SBram Moolenaar {
106119e60943SBram Moolenaar     return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val,
1062071d4279SBram Moolenaar 		      ((RangeObject *)(self))->start,
1063071d4279SBram Moolenaar 		      ((RangeObject *)(self))->end,
1064071d4279SBram Moolenaar 		      &((RangeObject *)(self))->end);
1065071d4279SBram Moolenaar }
1066071d4279SBram Moolenaar 
1067071d4279SBram Moolenaar /* Buffer list object - Definitions
1068071d4279SBram Moolenaar  */
1069071d4279SBram Moolenaar 
1070071d4279SBram Moolenaar typedef struct
1071071d4279SBram Moolenaar {
1072071d4279SBram Moolenaar     PyObject_HEAD
1073ca8a4dfeSBram Moolenaar } BufListObject;
1074071d4279SBram Moolenaar 
1075071d4279SBram Moolenaar static PySequenceMethods BufListAsSeq = {
10762c45e945SBram Moolenaar     (PyInquiry)		BufListLength,	    /* sq_length,    len(x)   */
1077071d4279SBram Moolenaar     (binaryfunc)	0,		    /* sq_concat,    x+y      */
10782c45e945SBram Moolenaar     (PyIntArgFunc)	0,		    /* sq_repeat,    x*n      */
10792c45e945SBram Moolenaar     (PyIntArgFunc)	BufListItem,	    /* sq_item,      x[i]     */
10802c45e945SBram Moolenaar     (PyIntIntArgFunc)	0,		    /* sq_slice,     x[i:j]   */
10812c45e945SBram Moolenaar     (PyIntObjArgProc)	0,		    /* sq_ass_item,  x[i]=v   */
10822c45e945SBram Moolenaar     (PyIntIntObjArgProc)	0,		    /* sq_ass_slice, x[i:j]=v */
1083071d4279SBram Moolenaar };
1084071d4279SBram Moolenaar 
1085071d4279SBram Moolenaar static PyTypeObject BufListType = {
1086071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
1087071d4279SBram Moolenaar     0,
1088071d4279SBram Moolenaar     "buffer list",
1089071d4279SBram Moolenaar     sizeof(BufListObject),
1090071d4279SBram Moolenaar     0,
1091071d4279SBram Moolenaar 
1092071d4279SBram Moolenaar     (destructor)    0,			/* tp_dealloc,	refcount==0  */
1093071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
1094071d4279SBram Moolenaar     (getattrfunc)   0,			/* tp_getattr,	x.attr	     */
1095071d4279SBram Moolenaar     (setattrfunc)   0,			/* tp_setattr,	x.attr=v     */
1096071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
1097071d4279SBram Moolenaar     (reprfunc)	    0,			/* tp_repr,	`x`, print x */
1098071d4279SBram Moolenaar 
1099071d4279SBram Moolenaar     0,		    /* as number */
1100071d4279SBram Moolenaar     &BufListAsSeq,  /* as sequence */
1101071d4279SBram Moolenaar     0,		    /* as mapping */
1102071d4279SBram Moolenaar 
1103071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
1104071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
1105071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
1106071d4279SBram Moolenaar };
1107071d4279SBram Moolenaar 
1108071d4279SBram Moolenaar /* Window object - Definitions
1109071d4279SBram Moolenaar  */
1110071d4279SBram Moolenaar 
1111071d4279SBram Moolenaar static struct PyMethodDef WindowMethods[] = {
1112071d4279SBram Moolenaar     /* name,	    function,		calling,    documentation */
1113071d4279SBram Moolenaar     { NULL,	    NULL,		0,	    NULL }
1114071d4279SBram Moolenaar };
1115071d4279SBram Moolenaar 
1116071d4279SBram Moolenaar static PyTypeObject WindowType = {
1117071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
1118071d4279SBram Moolenaar     0,
1119071d4279SBram Moolenaar     "window",
1120071d4279SBram Moolenaar     sizeof(WindowObject),
1121071d4279SBram Moolenaar     0,
1122071d4279SBram Moolenaar 
1123071d4279SBram Moolenaar     (destructor)    WindowDestructor,	/* tp_dealloc,	refcount==0  */
1124071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
1125071d4279SBram Moolenaar     (getattrfunc)   WindowGetattr,	/* tp_getattr,	x.attr	     */
1126071d4279SBram Moolenaar     (setattrfunc)   WindowSetattr,	/* tp_setattr,	x.attr=v     */
1127071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
1128071d4279SBram Moolenaar     (reprfunc)	    WindowRepr,		/* tp_repr,	`x`, print x */
1129071d4279SBram Moolenaar 
1130071d4279SBram Moolenaar     0,		    /* as number */
1131071d4279SBram Moolenaar     0,		    /* as sequence */
1132071d4279SBram Moolenaar     0,		    /* as mapping */
1133071d4279SBram Moolenaar 
1134071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
1135071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
1136071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
1137071d4279SBram Moolenaar };
1138071d4279SBram Moolenaar 
1139071d4279SBram Moolenaar /* Window object - Implementation
1140071d4279SBram Moolenaar  */
1141071d4279SBram Moolenaar 
1142071d4279SBram Moolenaar     static PyObject *
1143071d4279SBram Moolenaar WindowNew(win_T *win)
1144071d4279SBram Moolenaar {
1145071d4279SBram Moolenaar     /* We need to handle deletion of windows underneath us.
1146e344beadSBram Moolenaar      * If we add a "w_python_ref" field to the win_T structure,
1147071d4279SBram Moolenaar      * then we can get at it in win_free() in vim. We then
1148071d4279SBram Moolenaar      * need to create only ONE Python object per window - if
1149071d4279SBram Moolenaar      * we try to create a second, just INCREF the existing one
1150071d4279SBram Moolenaar      * and return it. The (single) Python object referring to
1151e344beadSBram Moolenaar      * the window is stored in "w_python_ref".
1152071d4279SBram Moolenaar      * On a win_free() we set the Python object's win_T* field
1153071d4279SBram Moolenaar      * to an invalid value. We trap all uses of a window
1154071d4279SBram Moolenaar      * object, and reject them if the win_T* field is invalid.
1155071d4279SBram Moolenaar      */
1156071d4279SBram Moolenaar 
1157071d4279SBram Moolenaar     WindowObject *self;
1158071d4279SBram Moolenaar 
1159e344beadSBram Moolenaar     if (win->w_python_ref)
1160071d4279SBram Moolenaar     {
1161e344beadSBram Moolenaar 	self = win->w_python_ref;
1162071d4279SBram Moolenaar 	Py_INCREF(self);
1163071d4279SBram Moolenaar     }
1164071d4279SBram Moolenaar     else
1165071d4279SBram Moolenaar     {
1166071d4279SBram Moolenaar 	self = PyObject_NEW(WindowObject, &WindowType);
1167071d4279SBram Moolenaar 	if (self == NULL)
1168071d4279SBram Moolenaar 	    return NULL;
1169071d4279SBram Moolenaar 	self->win = win;
1170e344beadSBram Moolenaar 	win->w_python_ref = self;
1171071d4279SBram Moolenaar     }
1172071d4279SBram Moolenaar 
1173071d4279SBram Moolenaar     return (PyObject *)(self);
1174071d4279SBram Moolenaar }
1175071d4279SBram Moolenaar 
1176071d4279SBram Moolenaar     static void
1177071d4279SBram Moolenaar WindowDestructor(PyObject *self)
1178071d4279SBram Moolenaar {
1179071d4279SBram Moolenaar     WindowObject *this = (WindowObject *)(self);
1180071d4279SBram Moolenaar 
1181071d4279SBram Moolenaar     if (this->win && this->win != INVALID_WINDOW_VALUE)
1182e344beadSBram Moolenaar 	this->win->w_python_ref = NULL;
1183071d4279SBram Moolenaar 
1184658ada69SBram Moolenaar     Py_DECREF(self);
1185071d4279SBram Moolenaar }
1186071d4279SBram Moolenaar 
1187071d4279SBram Moolenaar     static PyObject *
1188071d4279SBram Moolenaar WindowGetattr(PyObject *self, char *name)
1189071d4279SBram Moolenaar {
1190071d4279SBram Moolenaar     WindowObject *this = (WindowObject *)(self);
1191071d4279SBram Moolenaar 
1192071d4279SBram Moolenaar     if (CheckWindow(this))
1193071d4279SBram Moolenaar 	return NULL;
1194071d4279SBram Moolenaar 
1195071d4279SBram Moolenaar     if (strcmp(name, "buffer") == 0)
1196071d4279SBram Moolenaar 	return (PyObject *)BufferNew(this->win->w_buffer);
1197071d4279SBram Moolenaar     else if (strcmp(name, "cursor") == 0)
1198071d4279SBram Moolenaar     {
1199071d4279SBram Moolenaar 	pos_T *pos = &this->win->w_cursor;
1200071d4279SBram Moolenaar 
1201071d4279SBram Moolenaar 	return Py_BuildValue("(ll)", (long)(pos->lnum), (long)(pos->col));
1202071d4279SBram Moolenaar     }
1203071d4279SBram Moolenaar     else if (strcmp(name, "height") == 0)
1204071d4279SBram Moolenaar 	return Py_BuildValue("l", (long)(this->win->w_height));
1205071d4279SBram Moolenaar #ifdef FEAT_VERTSPLIT
1206071d4279SBram Moolenaar     else if (strcmp(name, "width") == 0)
1207071d4279SBram Moolenaar 	return Py_BuildValue("l", (long)(W_WIDTH(this->win)));
1208071d4279SBram Moolenaar #endif
1209071d4279SBram Moolenaar     else if (strcmp(name,"__members__") == 0)
1210071d4279SBram Moolenaar 	return Py_BuildValue("[sss]", "buffer", "cursor", "height");
1211071d4279SBram Moolenaar     else
1212071d4279SBram Moolenaar 	return Py_FindMethod(WindowMethods, self, name);
1213071d4279SBram Moolenaar }
1214071d4279SBram Moolenaar 
1215071d4279SBram Moolenaar /* Window list object - Definitions
1216071d4279SBram Moolenaar  */
1217071d4279SBram Moolenaar 
1218071d4279SBram Moolenaar typedef struct
1219071d4279SBram Moolenaar {
1220071d4279SBram Moolenaar     PyObject_HEAD
1221071d4279SBram Moolenaar }
1222071d4279SBram Moolenaar WinListObject;
1223071d4279SBram Moolenaar 
1224071d4279SBram Moolenaar static PySequenceMethods WinListAsSeq = {
12252c45e945SBram Moolenaar     (PyInquiry)		WinListLength,	    /* sq_length,    len(x)   */
1226071d4279SBram Moolenaar     (binaryfunc)	0,		    /* sq_concat,    x+y      */
12272c45e945SBram Moolenaar     (PyIntArgFunc)	0,		    /* sq_repeat,    x*n      */
12282c45e945SBram Moolenaar     (PyIntArgFunc)	WinListItem,	    /* sq_item,      x[i]     */
12292c45e945SBram Moolenaar     (PyIntIntArgFunc)	0,		    /* sq_slice,     x[i:j]   */
12302c45e945SBram Moolenaar     (PyIntObjArgProc)	0,		    /* sq_ass_item,  x[i]=v   */
12312c45e945SBram Moolenaar     (PyIntIntObjArgProc)	0,		    /* sq_ass_slice, x[i:j]=v */
1232071d4279SBram Moolenaar };
1233071d4279SBram Moolenaar 
1234071d4279SBram Moolenaar static PyTypeObject WinListType = {
1235071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
1236071d4279SBram Moolenaar     0,
1237071d4279SBram Moolenaar     "window list",
1238071d4279SBram Moolenaar     sizeof(WinListObject),
1239071d4279SBram Moolenaar     0,
1240071d4279SBram Moolenaar 
1241071d4279SBram Moolenaar     (destructor)    0,			/* tp_dealloc,	refcount==0  */
1242071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
1243071d4279SBram Moolenaar     (getattrfunc)   0,			/* tp_getattr,	x.attr	     */
1244071d4279SBram Moolenaar     (setattrfunc)   0,			/* tp_setattr,	x.attr=v     */
1245071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
1246071d4279SBram Moolenaar     (reprfunc)	    0,			/* tp_repr,	`x`, print x */
1247071d4279SBram Moolenaar 
1248071d4279SBram Moolenaar     0,		    /* as number */
1249071d4279SBram Moolenaar     &WinListAsSeq,  /* as sequence */
1250071d4279SBram Moolenaar     0,		    /* as mapping */
1251071d4279SBram Moolenaar 
1252071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
1253071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
1254071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
1255071d4279SBram Moolenaar };
1256071d4279SBram Moolenaar 
1257071d4279SBram Moolenaar /* Current items object - Definitions
1258071d4279SBram Moolenaar  */
1259071d4279SBram Moolenaar 
1260071d4279SBram Moolenaar typedef struct
1261071d4279SBram Moolenaar {
1262071d4279SBram Moolenaar     PyObject_HEAD
1263ca8a4dfeSBram Moolenaar } CurrentObject;
1264071d4279SBram Moolenaar 
1265071d4279SBram Moolenaar static PyTypeObject CurrentType = {
1266071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
1267071d4279SBram Moolenaar     0,
1268071d4279SBram Moolenaar     "current data",
1269071d4279SBram Moolenaar     sizeof(CurrentObject),
1270071d4279SBram Moolenaar     0,
1271071d4279SBram Moolenaar 
1272071d4279SBram Moolenaar     (destructor)    0,			/* tp_dealloc,	refcount==0  */
1273071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
1274071d4279SBram Moolenaar     (getattrfunc)   CurrentGetattr,	/* tp_getattr,	x.attr	     */
1275071d4279SBram Moolenaar     (setattrfunc)   CurrentSetattr,	/* tp_setattr,	x.attr=v     */
1276071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
1277071d4279SBram Moolenaar     (reprfunc)	    0,			/* tp_repr,	`x`, print x */
1278071d4279SBram Moolenaar 
1279071d4279SBram Moolenaar     0,		    /* as number */
1280071d4279SBram Moolenaar     0,		    /* as sequence */
1281071d4279SBram Moolenaar     0,		    /* as mapping */
1282071d4279SBram Moolenaar 
1283071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
1284071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
1285071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
1286071d4279SBram Moolenaar };
1287071d4279SBram Moolenaar 
1288071d4279SBram Moolenaar /* Current items object - Implementation
1289071d4279SBram Moolenaar  */
1290071d4279SBram Moolenaar     static PyObject *
12914bdbbf70SBram Moolenaar CurrentGetattr(PyObject *self UNUSED, char *name)
1292071d4279SBram Moolenaar {
1293071d4279SBram Moolenaar     if (strcmp(name, "buffer") == 0)
1294071d4279SBram Moolenaar 	return (PyObject *)BufferNew(curbuf);
1295071d4279SBram Moolenaar     else if (strcmp(name, "window") == 0)
1296071d4279SBram Moolenaar 	return (PyObject *)WindowNew(curwin);
1297071d4279SBram Moolenaar     else if (strcmp(name, "line") == 0)
1298e7cb9cf6SBram Moolenaar 	return GetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum);
1299071d4279SBram Moolenaar     else if (strcmp(name, "range") == 0)
1300071d4279SBram Moolenaar 	return RangeNew(curbuf, RangeStart, RangeEnd);
1301071d4279SBram Moolenaar     else if (strcmp(name,"__members__") == 0)
1302071d4279SBram Moolenaar 	return Py_BuildValue("[ssss]", "buffer", "window", "line", "range");
1303071d4279SBram Moolenaar     else
1304071d4279SBram Moolenaar     {
1305071d4279SBram Moolenaar 	PyErr_SetString(PyExc_AttributeError, name);
1306071d4279SBram Moolenaar 	return NULL;
1307071d4279SBram Moolenaar     }
1308071d4279SBram Moolenaar }
1309071d4279SBram Moolenaar 
1310071d4279SBram Moolenaar     static int
13114bdbbf70SBram Moolenaar CurrentSetattr(PyObject *self UNUSED, char *name, PyObject *value)
1312071d4279SBram Moolenaar {
1313071d4279SBram Moolenaar     if (strcmp(name, "line") == 0)
1314071d4279SBram Moolenaar     {
1315e7cb9cf6SBram Moolenaar 	if (SetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum, value, NULL) == FAIL)
1316071d4279SBram Moolenaar 	    return -1;
1317071d4279SBram Moolenaar 
1318071d4279SBram Moolenaar 	return 0;
1319071d4279SBram Moolenaar     }
1320071d4279SBram Moolenaar     else
1321071d4279SBram Moolenaar     {
1322071d4279SBram Moolenaar 	PyErr_SetString(PyExc_AttributeError, name);
1323071d4279SBram Moolenaar 	return -1;
1324071d4279SBram Moolenaar     }
1325071d4279SBram Moolenaar }
1326071d4279SBram Moolenaar 
1327071d4279SBram Moolenaar /* External interface
1328071d4279SBram Moolenaar  */
1329071d4279SBram Moolenaar 
1330071d4279SBram Moolenaar     void
1331071d4279SBram Moolenaar python_buffer_free(buf_T *buf)
1332071d4279SBram Moolenaar {
1333e344beadSBram Moolenaar     if (buf->b_python_ref != NULL)
1334071d4279SBram Moolenaar     {
1335e344beadSBram Moolenaar 	BufferObject *bp = buf->b_python_ref;
1336071d4279SBram Moolenaar 	bp->buf = INVALID_BUFFER_VALUE;
1337e344beadSBram Moolenaar 	buf->b_python_ref = NULL;
1338071d4279SBram Moolenaar     }
1339071d4279SBram Moolenaar }
1340071d4279SBram Moolenaar 
1341071d4279SBram Moolenaar #if defined(FEAT_WINDOWS) || defined(PROTO)
1342071d4279SBram Moolenaar     void
1343071d4279SBram Moolenaar python_window_free(win_T *win)
1344071d4279SBram Moolenaar {
1345e344beadSBram Moolenaar     if (win->w_python_ref != NULL)
1346071d4279SBram Moolenaar     {
1347e344beadSBram Moolenaar 	WindowObject *wp = win->w_python_ref;
1348071d4279SBram Moolenaar 	wp->win = INVALID_WINDOW_VALUE;
1349e344beadSBram Moolenaar 	win->w_python_ref = NULL;
1350071d4279SBram Moolenaar     }
1351071d4279SBram Moolenaar }
1352071d4279SBram Moolenaar #endif
1353071d4279SBram Moolenaar 
1354071d4279SBram Moolenaar static BufListObject TheBufferList =
1355071d4279SBram Moolenaar {
1356071d4279SBram Moolenaar     PyObject_HEAD_INIT(&BufListType)
1357071d4279SBram Moolenaar };
1358071d4279SBram Moolenaar 
1359071d4279SBram Moolenaar static WinListObject TheWindowList =
1360071d4279SBram Moolenaar {
1361071d4279SBram Moolenaar     PyObject_HEAD_INIT(&WinListType)
1362071d4279SBram Moolenaar };
1363071d4279SBram Moolenaar 
1364071d4279SBram Moolenaar static CurrentObject TheCurrent =
1365071d4279SBram Moolenaar {
1366071d4279SBram Moolenaar     PyObject_HEAD_INIT(&CurrentType)
1367071d4279SBram Moolenaar };
1368071d4279SBram Moolenaar 
1369071d4279SBram Moolenaar     static int
1370071d4279SBram Moolenaar PythonMod_Init(void)
1371071d4279SBram Moolenaar {
1372071d4279SBram Moolenaar     PyObject *mod;
1373071d4279SBram Moolenaar     PyObject *dict;
13749774ecc8SBram Moolenaar     /* The special value is removed from sys.path in Python_Init(). */
13759774ecc8SBram Moolenaar     static char *(argv[2]) = {"/must>not&exist/foo", NULL};
1376071d4279SBram Moolenaar 
1377071d4279SBram Moolenaar     /* Fixups... */
137821377c8dSBram Moolenaar     PyType_Ready(&BufferType);
137921377c8dSBram Moolenaar     PyType_Ready(&RangeType);
138021377c8dSBram Moolenaar     PyType_Ready(&WindowType);
138121377c8dSBram Moolenaar     PyType_Ready(&BufListType);
138221377c8dSBram Moolenaar     PyType_Ready(&WinListType);
138321377c8dSBram Moolenaar     PyType_Ready(&CurrentType);
1384071d4279SBram Moolenaar 
1385071d4279SBram Moolenaar     /* Set sys.argv[] to avoid a crash in warn(). */
1386071d4279SBram Moolenaar     PySys_SetArgv(1, argv);
1387071d4279SBram Moolenaar 
1388e7cb9cf6SBram Moolenaar     mod = Py_InitModule4("vim", VimMethods, (char *)NULL, (PyObject *)NULL, PYTHON_API_VERSION);
1389071d4279SBram Moolenaar     dict = PyModule_GetDict(mod);
1390071d4279SBram Moolenaar 
1391071d4279SBram Moolenaar     VimError = Py_BuildValue("s", "vim.error");
1392071d4279SBram Moolenaar 
1393071d4279SBram Moolenaar     PyDict_SetItemString(dict, "error", VimError);
13947df2d662SBram Moolenaar     PyDict_SetItemString(dict, "buffers", (PyObject *)(void *)&TheBufferList);
13957df2d662SBram Moolenaar     PyDict_SetItemString(dict, "current", (PyObject *)(void *)&TheCurrent);
13967df2d662SBram Moolenaar     PyDict_SetItemString(dict, "windows", (PyObject *)(void *)&TheWindowList);
1397071d4279SBram Moolenaar 
1398071d4279SBram Moolenaar     if (PyErr_Occurred())
1399071d4279SBram Moolenaar 	return -1;
1400071d4279SBram Moolenaar 
1401071d4279SBram Moolenaar     return 0;
1402071d4279SBram Moolenaar }
1403071d4279SBram Moolenaar 
1404071d4279SBram Moolenaar /*************************************************************************
1405071d4279SBram Moolenaar  * 4. Utility functions for handling the interface between Vim and Python.
1406071d4279SBram Moolenaar  */
1407071d4279SBram Moolenaar 
1408071d4279SBram Moolenaar /* Convert a Vim line into a Python string.
1409071d4279SBram Moolenaar  * All internal newlines are replaced by null characters.
1410071d4279SBram Moolenaar  *
1411071d4279SBram Moolenaar  * On errors, the Python exception data is set, and NULL is returned.
1412071d4279SBram Moolenaar  */
1413071d4279SBram Moolenaar     static PyObject *
1414071d4279SBram Moolenaar LineToString(const char *str)
1415071d4279SBram Moolenaar {
1416071d4279SBram Moolenaar     PyObject *result;
14172c45e945SBram Moolenaar     PyInt len = strlen(str);
1418071d4279SBram Moolenaar     char *p;
1419071d4279SBram Moolenaar 
1420071d4279SBram Moolenaar     /* Allocate an Python string object, with uninitialised contents. We
1421071d4279SBram Moolenaar      * must do it this way, so that we can modify the string in place
1422071d4279SBram Moolenaar      * later. See the Python source, Objects/stringobject.c for details.
1423071d4279SBram Moolenaar      */
1424071d4279SBram Moolenaar     result = PyString_FromStringAndSize(NULL, len);
1425071d4279SBram Moolenaar     if (result == NULL)
1426071d4279SBram Moolenaar 	return NULL;
1427071d4279SBram Moolenaar 
1428071d4279SBram Moolenaar     p = PyString_AsString(result);
1429071d4279SBram Moolenaar 
1430071d4279SBram Moolenaar     while (*str)
1431071d4279SBram Moolenaar     {
1432071d4279SBram Moolenaar 	if (*str == '\n')
1433071d4279SBram Moolenaar 	    *p = '\0';
1434071d4279SBram Moolenaar 	else
1435071d4279SBram Moolenaar 	    *p = *str;
1436071d4279SBram Moolenaar 
1437071d4279SBram Moolenaar 	++p;
1438071d4279SBram Moolenaar 	++str;
1439071d4279SBram Moolenaar     }
1440071d4279SBram Moolenaar 
1441071d4279SBram Moolenaar     return result;
1442071d4279SBram Moolenaar }
1443071d4279SBram Moolenaar 
1444071d4279SBram Moolenaar 
1445071d4279SBram Moolenaar /* Don't generate a prototype for the next function, it generates an error on
1446071d4279SBram Moolenaar  * newer Python versions. */
1447071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ && !defined(PROTO)
1448071d4279SBram Moolenaar 
1449071d4279SBram Moolenaar     char *
1450071d4279SBram Moolenaar Py_GetProgramName(void)
1451071d4279SBram Moolenaar {
1452071d4279SBram Moolenaar     return "vim";
1453071d4279SBram Moolenaar }
1454071d4279SBram Moolenaar #endif /* Python 1.4 */
1455170bf1aeSBram Moolenaar 
1456170bf1aeSBram Moolenaar     static void
1457170bf1aeSBram Moolenaar init_structs(void)
1458170bf1aeSBram Moolenaar {
1459170bf1aeSBram Moolenaar     vim_memset(&OutputType, 0, sizeof(OutputType));
1460170bf1aeSBram Moolenaar     OutputType.tp_name = "message";
1461170bf1aeSBram Moolenaar     OutputType.tp_basicsize = sizeof(OutputObject);
1462170bf1aeSBram Moolenaar     OutputType.tp_getattr = OutputGetattr;
1463170bf1aeSBram Moolenaar     OutputType.tp_setattr = OutputSetattr;
1464ca8a4dfeSBram Moolenaar 
1465ca8a4dfeSBram Moolenaar     vim_memset(&RangeType, 0, sizeof(RangeType));
1466ca8a4dfeSBram Moolenaar     RangeType.tp_name = "range";
1467ca8a4dfeSBram Moolenaar     RangeType.tp_basicsize = sizeof(RangeObject);
1468ca8a4dfeSBram Moolenaar     RangeType.tp_dealloc = RangeDestructor;
1469ca8a4dfeSBram Moolenaar     RangeType.tp_getattr = RangeGetattr;
1470ca8a4dfeSBram Moolenaar     RangeType.tp_repr = RangeRepr;
1471ca8a4dfeSBram Moolenaar     RangeType.tp_as_sequence = &RangeAsSeq;
1472170bf1aeSBram Moolenaar }
1473