xref: /vim-8.2.3635/src/if_python.c (revision 19e60943)
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 
59*19e60943SBram Moolenaar /* No-op conversion functions, use with care! */
60*19e60943SBram Moolenaar #define PyString_AsBytes(obj) (obj)
61*19e60943SBram Moolenaar #define PyString_FreeBytes(obj)
62*19e60943SBram 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
136*19e60943SBram 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 *, ...);
197*19e60943SBram 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},
280*19e60943SBram 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     {
371b744b2faSBram Moolenaar 	EMSG(_("E836: This Vim cannot execute :python after using :py3"));
3724c3a326cSBram Moolenaar 	return FAIL;
3734c3a326cSBram Moolenaar     }
3744c3a326cSBram Moolenaar #endif
3754c3a326cSBram Moolenaar 
376071d4279SBram Moolenaar     if (hinstPython)
377071d4279SBram Moolenaar 	return OK;
378bd5e15fdSBram Moolenaar     hinstPython = load_dll(libname);
379071d4279SBram Moolenaar     if (!hinstPython)
380071d4279SBram Moolenaar     {
381071d4279SBram Moolenaar 	if (verbose)
382071d4279SBram Moolenaar 	    EMSG2(_(e_loadlib), libname);
383071d4279SBram Moolenaar 	return FAIL;
384071d4279SBram Moolenaar     }
385071d4279SBram Moolenaar 
386071d4279SBram Moolenaar     for (i = 0; python_funcname_table[i].ptr; ++i)
387071d4279SBram Moolenaar     {
388bd5e15fdSBram Moolenaar 	if ((*python_funcname_table[i].ptr = symbol_from_dll(hinstPython,
389071d4279SBram Moolenaar 			python_funcname_table[i].name)) == NULL)
390071d4279SBram Moolenaar 	{
391bd5e15fdSBram Moolenaar 	    close_dll(hinstPython);
392071d4279SBram Moolenaar 	    hinstPython = 0;
393071d4279SBram Moolenaar 	    if (verbose)
394071d4279SBram Moolenaar 		EMSG2(_(e_loadfunc), python_funcname_table[i].name);
395071d4279SBram Moolenaar 	    return FAIL;
396071d4279SBram Moolenaar 	}
397071d4279SBram Moolenaar     }
398071d4279SBram Moolenaar     return OK;
399071d4279SBram Moolenaar }
400071d4279SBram Moolenaar 
401071d4279SBram Moolenaar /*
402071d4279SBram Moolenaar  * If python is enabled (there is installed python on Windows system) return
403071d4279SBram Moolenaar  * TRUE, else FALSE.
404071d4279SBram Moolenaar  */
405071d4279SBram Moolenaar     int
406e7cb9cf6SBram Moolenaar python_enabled(int verbose)
407071d4279SBram Moolenaar {
408071d4279SBram Moolenaar     return python_runtime_link_init(DYNAMIC_PYTHON_DLL, verbose) == OK;
409071d4279SBram Moolenaar }
410071d4279SBram Moolenaar 
411ca8a4dfeSBram Moolenaar /*
412ca8a4dfeSBram Moolenaar  * Load the standard Python exceptions - don't import the symbols from the
413071d4279SBram Moolenaar  * DLL, as this can cause errors (importing data symbols is not reliable).
414071d4279SBram Moolenaar  */
415071d4279SBram Moolenaar     static void
416ca8a4dfeSBram Moolenaar get_exceptions(void)
417071d4279SBram Moolenaar {
418071d4279SBram Moolenaar     PyObject *exmod = PyImport_ImportModule("exceptions");
419071d4279SBram Moolenaar     PyObject *exdict = PyModule_GetDict(exmod);
420071d4279SBram Moolenaar     imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError");
421071d4279SBram Moolenaar     imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError");
422071d4279SBram Moolenaar     imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt");
423071d4279SBram Moolenaar     imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError");
424071d4279SBram Moolenaar     imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError");
425071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_AttributeError);
426071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_IndexError);
427071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_KeyboardInterrupt);
428071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_TypeError);
429071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_ValueError);
430071d4279SBram Moolenaar     Py_XDECREF(exmod);
431071d4279SBram Moolenaar }
432071d4279SBram Moolenaar #endif /* DYNAMIC_PYTHON */
433071d4279SBram Moolenaar 
434ca8a4dfeSBram Moolenaar static PyObject *BufferNew (buf_T *);
435ca8a4dfeSBram Moolenaar static PyObject *WindowNew(win_T *);
436ca8a4dfeSBram Moolenaar static PyObject *LineToString(const char *);
437ca8a4dfeSBram Moolenaar 
438ca8a4dfeSBram Moolenaar static PyTypeObject RangeType;
439ca8a4dfeSBram Moolenaar 
440170bf1aeSBram Moolenaar /*
441170bf1aeSBram Moolenaar  * Include the code shared with if_python3.c
442170bf1aeSBram Moolenaar  */
443170bf1aeSBram Moolenaar #include "if_py_both.h"
444170bf1aeSBram Moolenaar 
445170bf1aeSBram Moolenaar 
446071d4279SBram Moolenaar /******************************************************
447071d4279SBram Moolenaar  * Internal function prototypes.
448071d4279SBram Moolenaar  */
449071d4279SBram Moolenaar 
450e7cb9cf6SBram Moolenaar static PyInt RangeStart;
451e7cb9cf6SBram Moolenaar static PyInt RangeEnd;
452071d4279SBram Moolenaar 
453071d4279SBram Moolenaar static void PythonIO_Flush(void);
454071d4279SBram Moolenaar static int PythonIO_Init(void);
455071d4279SBram Moolenaar static int PythonMod_Init(void);
456071d4279SBram Moolenaar 
457071d4279SBram Moolenaar /* Utility functions for the vim/python interface
458071d4279SBram Moolenaar  * ----------------------------------------------
459071d4279SBram Moolenaar  */
460071d4279SBram Moolenaar 
461e7cb9cf6SBram Moolenaar static int SetBufferLineList(buf_T *, PyInt, PyInt, PyObject *, PyInt *);
462071d4279SBram Moolenaar 
463071d4279SBram Moolenaar 
464071d4279SBram Moolenaar /******************************************************
465071d4279SBram Moolenaar  * 1. Python interpreter main program.
466071d4279SBram Moolenaar  */
467071d4279SBram Moolenaar 
468071d4279SBram Moolenaar static int initialised = 0;
469071d4279SBram Moolenaar 
470071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */
471071d4279SBram Moolenaar typedef PyObject PyThreadState;
4729ba0eb85SBram Moolenaar #endif
473071d4279SBram Moolenaar 
4749ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE
4759ba0eb85SBram Moolenaar static PyGILState_STATE pygilstate = PyGILState_UNLOCKED;
4769ba0eb85SBram Moolenaar #else
477071d4279SBram Moolenaar static PyThreadState *saved_python_thread = NULL;
4789ba0eb85SBram Moolenaar #endif
479071d4279SBram Moolenaar 
480071d4279SBram Moolenaar /*
481071d4279SBram Moolenaar  * Suspend a thread of the Python interpreter, other threads are allowed to
482071d4279SBram Moolenaar  * run.
483071d4279SBram Moolenaar  */
484293ee4d4SBram Moolenaar     static void
485293ee4d4SBram Moolenaar Python_SaveThread(void)
486071d4279SBram Moolenaar {
4879ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE
4889ba0eb85SBram Moolenaar     PyGILState_Release(pygilstate);
4899ba0eb85SBram Moolenaar #else
490071d4279SBram Moolenaar     saved_python_thread = PyEval_SaveThread();
4919ba0eb85SBram Moolenaar #endif
492071d4279SBram Moolenaar }
493071d4279SBram Moolenaar 
494071d4279SBram Moolenaar /*
495071d4279SBram Moolenaar  * Restore a thread of the Python interpreter, waits for other threads to
496071d4279SBram Moolenaar  * block.
497071d4279SBram Moolenaar  */
498293ee4d4SBram Moolenaar     static void
499293ee4d4SBram Moolenaar Python_RestoreThread(void)
500071d4279SBram Moolenaar {
5019ba0eb85SBram Moolenaar #ifdef PY_CAN_RECURSE
5029ba0eb85SBram Moolenaar     pygilstate = PyGILState_Ensure();
5039ba0eb85SBram Moolenaar #else
504071d4279SBram Moolenaar     PyEval_RestoreThread(saved_python_thread);
505071d4279SBram Moolenaar     saved_python_thread = NULL;
506071d4279SBram Moolenaar #endif
5079ba0eb85SBram Moolenaar }
508071d4279SBram Moolenaar 
509071d4279SBram Moolenaar     void
510071d4279SBram Moolenaar python_end()
511071d4279SBram Moolenaar {
512a5792f58SBram Moolenaar     static int recurse = 0;
513a5792f58SBram Moolenaar 
514a5792f58SBram Moolenaar     /* If a crash occurs while doing this, don't try again. */
515a5792f58SBram Moolenaar     if (recurse != 0)
516a5792f58SBram Moolenaar 	return;
517a5792f58SBram Moolenaar 
518a5792f58SBram Moolenaar     ++recurse;
519a5792f58SBram Moolenaar 
520071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
5210e21a3f6SBram Moolenaar     if (hinstPython && Py_IsInitialized())
5229ba0eb85SBram Moolenaar     {
5239ba0eb85SBram Moolenaar 	Python_RestoreThread();	    /* enter python */
5240e21a3f6SBram Moolenaar 	Py_Finalize();
5259ba0eb85SBram Moolenaar     }
526071d4279SBram Moolenaar     end_dynamic_python();
5270e21a3f6SBram Moolenaar #else
5280e21a3f6SBram Moolenaar     if (Py_IsInitialized())
5299ba0eb85SBram Moolenaar     {
5309ba0eb85SBram Moolenaar 	Python_RestoreThread();	    /* enter python */
5310e21a3f6SBram Moolenaar 	Py_Finalize();
5329ba0eb85SBram Moolenaar     }
533071d4279SBram Moolenaar #endif
534a5792f58SBram Moolenaar 
535a5792f58SBram Moolenaar     --recurse;
536071d4279SBram Moolenaar }
537071d4279SBram Moolenaar 
5384c3a326cSBram Moolenaar #if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
5394c3a326cSBram Moolenaar     int
5404c3a326cSBram Moolenaar python_loaded()
5414c3a326cSBram Moolenaar {
5424c3a326cSBram Moolenaar     return (hinstPython != 0);
5434c3a326cSBram Moolenaar }
5444c3a326cSBram Moolenaar #endif
5454c3a326cSBram Moolenaar 
546071d4279SBram Moolenaar     static int
547071d4279SBram Moolenaar Python_Init(void)
548071d4279SBram Moolenaar {
549071d4279SBram Moolenaar     if (!initialised)
550071d4279SBram Moolenaar     {
551071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
552071d4279SBram Moolenaar 	if (!python_enabled(TRUE))
553071d4279SBram Moolenaar 	{
554071d4279SBram Moolenaar 	    EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded."));
555071d4279SBram Moolenaar 	    goto fail;
556071d4279SBram Moolenaar 	}
557071d4279SBram Moolenaar #endif
558071d4279SBram Moolenaar 
559644d37b8SBram Moolenaar #ifdef PYTHON_HOME
560644d37b8SBram Moolenaar 	Py_SetPythonHome(PYTHON_HOME);
561644d37b8SBram Moolenaar #endif
562644d37b8SBram Moolenaar 
563170bf1aeSBram Moolenaar 	init_structs();
564170bf1aeSBram Moolenaar 
565071d4279SBram Moolenaar #if !defined(MACOS) || defined(MACOS_X_UNIX)
566071d4279SBram Moolenaar 	Py_Initialize();
567071d4279SBram Moolenaar #else
568071d4279SBram Moolenaar 	PyMac_Initialize();
569071d4279SBram Moolenaar #endif
570071d4279SBram Moolenaar 	/* initialise threads */
571071d4279SBram Moolenaar 	PyEval_InitThreads();
572071d4279SBram Moolenaar 
573071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
574071d4279SBram Moolenaar 	get_exceptions();
575071d4279SBram Moolenaar #endif
576071d4279SBram Moolenaar 
577071d4279SBram Moolenaar 	if (PythonIO_Init())
578071d4279SBram Moolenaar 	    goto fail;
579071d4279SBram Moolenaar 
580071d4279SBram Moolenaar 	if (PythonMod_Init())
581071d4279SBram Moolenaar 	    goto fail;
582071d4279SBram Moolenaar 
5839774ecc8SBram Moolenaar 	/* Remove the element from sys.path that was added because of our
5849774ecc8SBram Moolenaar 	 * argv[0] value in PythonMod_Init().  Previously we used an empty
5859774ecc8SBram Moolenaar 	 * string, but dependinding on the OS we then get an empty entry or
5869774ecc8SBram Moolenaar 	 * the current directory in sys.path. */
5879774ecc8SBram Moolenaar 	PyRun_SimpleString("import sys; sys.path = filter(lambda x: x != '/must>not&exist', sys.path)");
5889774ecc8SBram Moolenaar 
589293ee4d4SBram Moolenaar 	/* the first python thread is vim's, release the lock */
590071d4279SBram Moolenaar 	Python_SaveThread();
591071d4279SBram Moolenaar 
592071d4279SBram Moolenaar 	initialised = 1;
593071d4279SBram Moolenaar     }
594071d4279SBram Moolenaar 
595071d4279SBram Moolenaar     return 0;
596071d4279SBram Moolenaar 
597071d4279SBram Moolenaar fail:
598071d4279SBram Moolenaar     /* We call PythonIO_Flush() here to print any Python errors.
599071d4279SBram Moolenaar      * This is OK, as it is possible to call this function even
600071d4279SBram Moolenaar      * if PythonIO_Init() has not completed successfully (it will
601071d4279SBram Moolenaar      * not do anything in this case).
602071d4279SBram Moolenaar      */
603071d4279SBram Moolenaar     PythonIO_Flush();
604071d4279SBram Moolenaar     return -1;
605071d4279SBram Moolenaar }
606071d4279SBram Moolenaar 
607071d4279SBram Moolenaar /*
608071d4279SBram Moolenaar  * External interface
609071d4279SBram Moolenaar  */
610071d4279SBram Moolenaar     static void
611071d4279SBram Moolenaar DoPythonCommand(exarg_T *eap, const char *cmd)
612071d4279SBram Moolenaar {
6139ba0eb85SBram Moolenaar #ifndef PY_CAN_RECURSE
614071d4279SBram Moolenaar     static int		recursive = 0;
615071d4279SBram Moolenaar #endif
616071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
617071d4279SBram Moolenaar     GrafPtr		oldPort;
618071d4279SBram Moolenaar #endif
619071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
620071d4279SBram Moolenaar     char		*saved_locale;
621071d4279SBram Moolenaar #endif
622071d4279SBram Moolenaar 
623071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE
624071d4279SBram Moolenaar     if (recursive)
625071d4279SBram Moolenaar     {
626071d4279SBram Moolenaar 	EMSG(_("E659: Cannot invoke Python recursively"));
627071d4279SBram Moolenaar 	return;
628071d4279SBram Moolenaar     }
629071d4279SBram Moolenaar     ++recursive;
630071d4279SBram Moolenaar #endif
631071d4279SBram Moolenaar 
632071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
633071d4279SBram Moolenaar     GetPort(&oldPort);
634071d4279SBram Moolenaar     /* Check if the Python library is available */
635071d4279SBram Moolenaar     if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress)
636071d4279SBram Moolenaar 	goto theend;
637071d4279SBram Moolenaar #endif
638071d4279SBram Moolenaar     if (Python_Init())
639071d4279SBram Moolenaar 	goto theend;
640071d4279SBram Moolenaar 
641071d4279SBram Moolenaar     RangeStart = eap->line1;
642071d4279SBram Moolenaar     RangeEnd = eap->line2;
643071d4279SBram Moolenaar     Python_Release_Vim();	    /* leave vim */
644071d4279SBram Moolenaar 
645071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
646071d4279SBram Moolenaar     /* Python only works properly when the LC_NUMERIC locale is "C". */
647071d4279SBram Moolenaar     saved_locale = setlocale(LC_NUMERIC, NULL);
648071d4279SBram Moolenaar     if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0)
649071d4279SBram Moolenaar 	saved_locale = NULL;
650071d4279SBram Moolenaar     else
651071d4279SBram Moolenaar     {
652071d4279SBram Moolenaar 	/* Need to make a copy, value may change when setting new locale. */
653071d4279SBram Moolenaar 	saved_locale = (char *)vim_strsave((char_u *)saved_locale);
654071d4279SBram Moolenaar 	(void)setlocale(LC_NUMERIC, "C");
655071d4279SBram Moolenaar     }
656071d4279SBram Moolenaar #endif
657071d4279SBram Moolenaar 
658071d4279SBram Moolenaar     Python_RestoreThread();	    /* enter python */
659071d4279SBram Moolenaar 
660071d4279SBram Moolenaar     PyRun_SimpleString((char *)(cmd));
661071d4279SBram Moolenaar 
662071d4279SBram Moolenaar     Python_SaveThread();	    /* leave python */
663071d4279SBram Moolenaar 
664071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
665071d4279SBram Moolenaar     if (saved_locale != NULL)
666071d4279SBram Moolenaar     {
667071d4279SBram Moolenaar 	(void)setlocale(LC_NUMERIC, saved_locale);
668071d4279SBram Moolenaar 	vim_free(saved_locale);
669071d4279SBram Moolenaar     }
670071d4279SBram Moolenaar #endif
671071d4279SBram Moolenaar 
672071d4279SBram Moolenaar     Python_Lock_Vim();		    /* enter vim */
673071d4279SBram Moolenaar     PythonIO_Flush();
674071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
675071d4279SBram Moolenaar     SetPort(oldPort);
676071d4279SBram Moolenaar #endif
677071d4279SBram Moolenaar 
678071d4279SBram Moolenaar theend:
679071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE
680071d4279SBram Moolenaar     --recursive;
681071d4279SBram Moolenaar #endif
682071d4279SBram Moolenaar     return;	    /* keeps lint happy */
683071d4279SBram Moolenaar }
684071d4279SBram Moolenaar 
685071d4279SBram Moolenaar /*
686071d4279SBram Moolenaar  * ":python"
687071d4279SBram Moolenaar  */
688071d4279SBram Moolenaar     void
689071d4279SBram Moolenaar ex_python(exarg_T *eap)
690071d4279SBram Moolenaar {
691071d4279SBram Moolenaar     char_u *script;
692071d4279SBram Moolenaar 
693071d4279SBram Moolenaar     script = script_get(eap, eap->arg);
694071d4279SBram Moolenaar     if (!eap->skip)
695071d4279SBram Moolenaar     {
696071d4279SBram Moolenaar 	if (script == NULL)
697071d4279SBram Moolenaar 	    DoPythonCommand(eap, (char *)eap->arg);
698071d4279SBram Moolenaar 	else
699071d4279SBram Moolenaar 	    DoPythonCommand(eap, (char *)script);
700071d4279SBram Moolenaar     }
701071d4279SBram Moolenaar     vim_free(script);
702071d4279SBram Moolenaar }
703071d4279SBram Moolenaar 
704071d4279SBram Moolenaar #define BUFFER_SIZE 1024
705071d4279SBram Moolenaar 
706071d4279SBram Moolenaar /*
707071d4279SBram Moolenaar  * ":pyfile"
708071d4279SBram Moolenaar  */
709071d4279SBram Moolenaar     void
710071d4279SBram Moolenaar ex_pyfile(exarg_T *eap)
711071d4279SBram Moolenaar {
712071d4279SBram Moolenaar     static char buffer[BUFFER_SIZE];
713071d4279SBram Moolenaar     const char *file = (char *)eap->arg;
714071d4279SBram Moolenaar     char *p;
715071d4279SBram Moolenaar 
716071d4279SBram Moolenaar     /* Have to do it like this. PyRun_SimpleFile requires you to pass a
717071d4279SBram Moolenaar      * stdio file pointer, but Vim and the Python DLL are compiled with
718071d4279SBram Moolenaar      * different options under Windows, meaning that stdio pointers aren't
719071d4279SBram Moolenaar      * compatible between the two. Yuk.
720071d4279SBram Moolenaar      *
721071d4279SBram Moolenaar      * Put the string "execfile('file')" into buffer. But, we need to
722071d4279SBram Moolenaar      * escape any backslashes or single quotes in the file name, so that
723071d4279SBram Moolenaar      * Python won't mangle the file name.
724071d4279SBram Moolenaar      */
725071d4279SBram Moolenaar     strcpy(buffer, "execfile('");
726071d4279SBram Moolenaar     p = buffer + 10; /* size of "execfile('" */
727071d4279SBram Moolenaar 
728071d4279SBram Moolenaar     while (*file && p < buffer + (BUFFER_SIZE - 3))
729071d4279SBram Moolenaar     {
730071d4279SBram Moolenaar 	if (*file == '\\' || *file == '\'')
731071d4279SBram Moolenaar 	    *p++ = '\\';
732071d4279SBram Moolenaar 	*p++ = *file++;
733071d4279SBram Moolenaar     }
734071d4279SBram Moolenaar 
735071d4279SBram Moolenaar     /* If we didn't finish the file name, we hit a buffer overflow */
736071d4279SBram Moolenaar     if (*file != '\0')
737071d4279SBram Moolenaar 	return;
738071d4279SBram Moolenaar 
739071d4279SBram Moolenaar     /* Put in the terminating "')" and a null */
740071d4279SBram Moolenaar     *p++ = '\'';
741071d4279SBram Moolenaar     *p++ = ')';
742071d4279SBram Moolenaar     *p++ = '\0';
743071d4279SBram Moolenaar 
744071d4279SBram Moolenaar     /* Execute the file */
745071d4279SBram Moolenaar     DoPythonCommand(eap, buffer);
746071d4279SBram Moolenaar }
747071d4279SBram Moolenaar 
748071d4279SBram Moolenaar /******************************************************
749071d4279SBram Moolenaar  * 2. Python output stream: writes output via [e]msg().
750071d4279SBram Moolenaar  */
751071d4279SBram Moolenaar 
752071d4279SBram Moolenaar /* Implementation functions
753071d4279SBram Moolenaar  */
754071d4279SBram Moolenaar 
755071d4279SBram Moolenaar     static PyObject *
756071d4279SBram Moolenaar OutputGetattr(PyObject *self, char *name)
757071d4279SBram Moolenaar {
758071d4279SBram Moolenaar     if (strcmp(name, "softspace") == 0)
759071d4279SBram Moolenaar 	return PyInt_FromLong(((OutputObject *)(self))->softspace);
760071d4279SBram Moolenaar 
761071d4279SBram Moolenaar     return Py_FindMethod(OutputMethods, self, name);
762071d4279SBram Moolenaar }
763071d4279SBram Moolenaar 
764071d4279SBram Moolenaar     static int
765071d4279SBram Moolenaar OutputSetattr(PyObject *self, char *name, PyObject *val)
766071d4279SBram Moolenaar {
767071d4279SBram Moolenaar     if (val == NULL) {
768071d4279SBram Moolenaar 	PyErr_SetString(PyExc_AttributeError, _("can't delete OutputObject attributes"));
769071d4279SBram Moolenaar 	return -1;
770071d4279SBram Moolenaar     }
771071d4279SBram Moolenaar 
772071d4279SBram Moolenaar     if (strcmp(name, "softspace") == 0)
773071d4279SBram Moolenaar     {
774071d4279SBram Moolenaar 	if (!PyInt_Check(val)) {
775071d4279SBram Moolenaar 	    PyErr_SetString(PyExc_TypeError, _("softspace must be an integer"));
776071d4279SBram Moolenaar 	    return -1;
777071d4279SBram Moolenaar 	}
778071d4279SBram Moolenaar 
779071d4279SBram Moolenaar 	((OutputObject *)(self))->softspace = PyInt_AsLong(val);
780071d4279SBram Moolenaar 	return 0;
781071d4279SBram Moolenaar     }
782071d4279SBram Moolenaar 
783071d4279SBram Moolenaar     PyErr_SetString(PyExc_AttributeError, _("invalid attribute"));
784071d4279SBram Moolenaar     return -1;
785071d4279SBram Moolenaar }
786071d4279SBram Moolenaar 
787071d4279SBram Moolenaar /***************/
788071d4279SBram Moolenaar 
789071d4279SBram Moolenaar     static int
790071d4279SBram Moolenaar PythonIO_Init(void)
791071d4279SBram Moolenaar {
792071d4279SBram Moolenaar     /* Fixups... */
79321377c8dSBram Moolenaar     PyType_Ready(&OutputType);
794071d4279SBram Moolenaar 
795170bf1aeSBram Moolenaar     return PythonIO_Init_io();
796071d4279SBram Moolenaar }
797071d4279SBram Moolenaar 
798071d4279SBram Moolenaar /******************************************************
799071d4279SBram Moolenaar  * 3. Implementation of the Vim module for Python
800071d4279SBram Moolenaar  */
801071d4279SBram Moolenaar 
802071d4279SBram Moolenaar /* Window type - Implementation functions
803071d4279SBram Moolenaar  * --------------------------------------
804071d4279SBram Moolenaar  */
805071d4279SBram Moolenaar 
806071d4279SBram Moolenaar #define WindowType_Check(obj) ((obj)->ob_type == &WindowType)
807071d4279SBram Moolenaar 
808071d4279SBram Moolenaar static void WindowDestructor(PyObject *);
809071d4279SBram Moolenaar static PyObject *WindowGetattr(PyObject *, char *);
810071d4279SBram Moolenaar 
811071d4279SBram Moolenaar /* Buffer type - Implementation functions
812071d4279SBram Moolenaar  * --------------------------------------
813071d4279SBram Moolenaar  */
814071d4279SBram Moolenaar 
815071d4279SBram Moolenaar #define BufferType_Check(obj) ((obj)->ob_type == &BufferType)
816071d4279SBram Moolenaar 
817071d4279SBram Moolenaar static void BufferDestructor(PyObject *);
818071d4279SBram Moolenaar static PyObject *BufferGetattr(PyObject *, char *);
819071d4279SBram Moolenaar static PyObject *BufferRepr(PyObject *);
820071d4279SBram Moolenaar 
8212c45e945SBram Moolenaar static PyInt BufferLength(PyObject *);
8222c45e945SBram Moolenaar static PyObject *BufferItem(PyObject *, PyInt);
8232c45e945SBram Moolenaar static PyObject *BufferSlice(PyObject *, PyInt, PyInt);
8242c45e945SBram Moolenaar static PyInt BufferAssItem(PyObject *, PyInt, PyObject *);
8252c45e945SBram Moolenaar static PyInt BufferAssSlice(PyObject *, PyInt, PyInt, PyObject *);
826071d4279SBram Moolenaar 
827071d4279SBram Moolenaar /* Line range type - Implementation functions
828071d4279SBram Moolenaar  * --------------------------------------
829071d4279SBram Moolenaar  */
830071d4279SBram Moolenaar 
831071d4279SBram Moolenaar #define RangeType_Check(obj) ((obj)->ob_type == &RangeType)
832071d4279SBram Moolenaar 
8332c45e945SBram Moolenaar static PyInt RangeAssItem(PyObject *, PyInt, PyObject *);
8342c45e945SBram Moolenaar static PyInt RangeAssSlice(PyObject *, PyInt, PyInt, PyObject *);
835071d4279SBram Moolenaar 
836071d4279SBram Moolenaar /* Current objects type - Implementation functions
837071d4279SBram Moolenaar  * -----------------------------------------------
838071d4279SBram Moolenaar  */
839071d4279SBram Moolenaar 
840071d4279SBram Moolenaar static PyObject *CurrentGetattr(PyObject *, char *);
841071d4279SBram Moolenaar static int CurrentSetattr(PyObject *, char *, PyObject *);
842071d4279SBram Moolenaar 
843071d4279SBram Moolenaar static PySequenceMethods BufferAsSeq = {
8442c45e945SBram Moolenaar     (PyInquiry)		BufferLength,	    /* sq_length,    len(x)   */
845071d4279SBram Moolenaar     (binaryfunc)	0, /* BufferConcat, */	     /* sq_concat,    x+y      */
8462c45e945SBram Moolenaar     (PyIntArgFunc)	0, /* BufferRepeat, */	     /* sq_repeat,    x*n      */
8472c45e945SBram Moolenaar     (PyIntArgFunc)	BufferItem,	    /* sq_item,      x[i]     */
8482c45e945SBram Moolenaar     (PyIntIntArgFunc)	BufferSlice,	    /* sq_slice,     x[i:j]   */
8492c45e945SBram Moolenaar     (PyIntObjArgProc)	BufferAssItem,	    /* sq_ass_item,  x[i]=v   */
8502c45e945SBram Moolenaar     (PyIntIntObjArgProc)	BufferAssSlice,     /* sq_ass_slice, x[i:j]=v */
851071d4279SBram Moolenaar };
852071d4279SBram Moolenaar 
853071d4279SBram Moolenaar static PyTypeObject BufferType = {
854071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
855071d4279SBram Moolenaar     0,
856071d4279SBram Moolenaar     "buffer",
857071d4279SBram Moolenaar     sizeof(BufferObject),
858071d4279SBram Moolenaar     0,
859071d4279SBram Moolenaar 
860071d4279SBram Moolenaar     (destructor)    BufferDestructor,	/* tp_dealloc,	refcount==0  */
861071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
862071d4279SBram Moolenaar     (getattrfunc)   BufferGetattr,	/* tp_getattr,	x.attr	     */
863071d4279SBram Moolenaar     (setattrfunc)   0,			/* tp_setattr,	x.attr=v     */
864071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
865071d4279SBram Moolenaar     (reprfunc)	    BufferRepr,		/* tp_repr,	`x`, print x */
866071d4279SBram Moolenaar 
867071d4279SBram Moolenaar     0,		    /* as number */
868071d4279SBram Moolenaar     &BufferAsSeq,   /* as sequence */
869071d4279SBram Moolenaar     0,		    /* as mapping */
870071d4279SBram Moolenaar 
871071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
872071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
873071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
874071d4279SBram Moolenaar };
875071d4279SBram Moolenaar 
876071d4279SBram Moolenaar /* Buffer object - Implementation
877071d4279SBram Moolenaar  */
878071d4279SBram Moolenaar 
879071d4279SBram Moolenaar     static PyObject *
880071d4279SBram Moolenaar BufferNew(buf_T *buf)
881071d4279SBram Moolenaar {
882071d4279SBram Moolenaar     /* We need to handle deletion of buffers underneath us.
883e344beadSBram Moolenaar      * If we add a "b_python_ref" field to the buf_T structure,
884071d4279SBram Moolenaar      * then we can get at it in buf_freeall() in vim. We then
885071d4279SBram Moolenaar      * need to create only ONE Python object per buffer - if
886071d4279SBram Moolenaar      * we try to create a second, just INCREF the existing one
887071d4279SBram Moolenaar      * and return it. The (single) Python object referring to
888e344beadSBram Moolenaar      * the buffer is stored in "b_python_ref".
889071d4279SBram Moolenaar      * Question: what to do on a buf_freeall(). We'll probably
890071d4279SBram Moolenaar      * have to either delete the Python object (DECREF it to
891071d4279SBram Moolenaar      * zero - a bad idea, as it leaves dangling refs!) or
892071d4279SBram Moolenaar      * set the buf_T * value to an invalid value (-1?), which
893071d4279SBram Moolenaar      * means we need checks in all access functions... Bah.
894071d4279SBram Moolenaar      */
895071d4279SBram Moolenaar 
896071d4279SBram Moolenaar     BufferObject *self;
897071d4279SBram Moolenaar 
898e344beadSBram Moolenaar     if (buf->b_python_ref != NULL)
899071d4279SBram Moolenaar     {
900e344beadSBram Moolenaar 	self = buf->b_python_ref;
901071d4279SBram Moolenaar 	Py_INCREF(self);
902071d4279SBram Moolenaar     }
903071d4279SBram Moolenaar     else
904071d4279SBram Moolenaar     {
905071d4279SBram Moolenaar 	self = PyObject_NEW(BufferObject, &BufferType);
906071d4279SBram Moolenaar 	if (self == NULL)
907071d4279SBram Moolenaar 	    return NULL;
908071d4279SBram Moolenaar 	self->buf = buf;
909e344beadSBram Moolenaar 	buf->b_python_ref = self;
910071d4279SBram Moolenaar     }
911071d4279SBram Moolenaar 
912071d4279SBram Moolenaar     return (PyObject *)(self);
913071d4279SBram Moolenaar }
914071d4279SBram Moolenaar 
915071d4279SBram Moolenaar     static void
916071d4279SBram Moolenaar BufferDestructor(PyObject *self)
917071d4279SBram Moolenaar {
918071d4279SBram Moolenaar     BufferObject *this = (BufferObject *)(self);
919071d4279SBram Moolenaar 
920071d4279SBram Moolenaar     if (this->buf && this->buf != INVALID_BUFFER_VALUE)
921e344beadSBram Moolenaar 	this->buf->b_python_ref = NULL;
922071d4279SBram Moolenaar 
923658ada69SBram Moolenaar     Py_DECREF(self);
924071d4279SBram Moolenaar }
925071d4279SBram Moolenaar 
926071d4279SBram Moolenaar     static PyObject *
927071d4279SBram Moolenaar BufferGetattr(PyObject *self, char *name)
928071d4279SBram Moolenaar {
929071d4279SBram Moolenaar     BufferObject *this = (BufferObject *)(self);
930071d4279SBram Moolenaar 
931071d4279SBram Moolenaar     if (CheckBuffer(this))
932071d4279SBram Moolenaar 	return NULL;
933071d4279SBram Moolenaar 
934071d4279SBram Moolenaar     if (strcmp(name, "name") == 0)
935071d4279SBram Moolenaar 	return Py_BuildValue("s", this->buf->b_ffname);
936071d4279SBram Moolenaar     else if (strcmp(name, "number") == 0)
937e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, this->buf->b_fnum);
938071d4279SBram Moolenaar     else if (strcmp(name,"__members__") == 0)
939071d4279SBram Moolenaar 	return Py_BuildValue("[ss]", "name", "number");
940071d4279SBram Moolenaar     else
941071d4279SBram Moolenaar 	return Py_FindMethod(BufferMethods, self, name);
942071d4279SBram Moolenaar }
943071d4279SBram Moolenaar 
944071d4279SBram Moolenaar     static PyObject *
945071d4279SBram Moolenaar BufferRepr(PyObject *self)
946071d4279SBram Moolenaar {
947555b280fSBram Moolenaar     static char repr[100];
948071d4279SBram Moolenaar     BufferObject *this = (BufferObject *)(self);
949071d4279SBram Moolenaar 
950071d4279SBram Moolenaar     if (this->buf == INVALID_BUFFER_VALUE)
951071d4279SBram Moolenaar     {
952e7cb9cf6SBram Moolenaar 	vim_snprintf(repr, 100, _("<buffer object (deleted) at %p>"), (self));
953071d4279SBram Moolenaar 	return PyString_FromString(repr);
954071d4279SBram Moolenaar     }
955071d4279SBram Moolenaar     else
956071d4279SBram Moolenaar     {
957071d4279SBram Moolenaar 	char *name = (char *)this->buf->b_fname;
958e7cb9cf6SBram Moolenaar 	PyInt len;
959071d4279SBram Moolenaar 
960071d4279SBram Moolenaar 	if (name == NULL)
961071d4279SBram Moolenaar 	    name = "";
962071d4279SBram Moolenaar 	len = strlen(name);
963071d4279SBram Moolenaar 
964071d4279SBram Moolenaar 	if (len > 35)
965071d4279SBram Moolenaar 	    name = name + (35 - len);
966071d4279SBram Moolenaar 
967555b280fSBram Moolenaar 	vim_snprintf(repr, 100, "<buffer %s%s>", len > 35 ? "..." : "", name);
968071d4279SBram Moolenaar 
969071d4279SBram Moolenaar 	return PyString_FromString(repr);
970071d4279SBram Moolenaar     }
971071d4279SBram Moolenaar }
972071d4279SBram Moolenaar 
973071d4279SBram Moolenaar /******************/
974071d4279SBram Moolenaar 
9752c45e945SBram Moolenaar     static PyInt
976071d4279SBram Moolenaar BufferLength(PyObject *self)
977071d4279SBram Moolenaar {
978071d4279SBram Moolenaar     /* HOW DO WE SIGNAL AN ERROR FROM THIS FUNCTION? */
979071d4279SBram Moolenaar     if (CheckBuffer((BufferObject *)(self)))
980071d4279SBram Moolenaar 	return -1; /* ??? */
981071d4279SBram Moolenaar 
982071d4279SBram Moolenaar     return (((BufferObject *)(self))->buf->b_ml.ml_line_count);
983071d4279SBram Moolenaar }
984071d4279SBram Moolenaar 
985071d4279SBram Moolenaar     static PyObject *
9862c45e945SBram Moolenaar BufferItem(PyObject *self, PyInt n)
987071d4279SBram Moolenaar {
988071d4279SBram Moolenaar     return RBItem((BufferObject *)(self), n, 1,
989071d4279SBram Moolenaar 		  (int)((BufferObject *)(self))->buf->b_ml.ml_line_count);
990071d4279SBram Moolenaar }
991071d4279SBram Moolenaar 
992071d4279SBram Moolenaar     static PyObject *
9932c45e945SBram Moolenaar BufferSlice(PyObject *self, PyInt lo, PyInt hi)
994071d4279SBram Moolenaar {
995071d4279SBram Moolenaar     return RBSlice((BufferObject *)(self), lo, hi, 1,
996071d4279SBram Moolenaar 		   (int)((BufferObject *)(self))->buf->b_ml.ml_line_count);
997071d4279SBram Moolenaar }
998071d4279SBram Moolenaar 
9992c45e945SBram Moolenaar     static PyInt
10002c45e945SBram Moolenaar BufferAssItem(PyObject *self, PyInt n, PyObject *val)
1001071d4279SBram Moolenaar {
1002ca8a4dfeSBram Moolenaar     return RBAsItem((BufferObject *)(self), n, val, 1,
1003e7cb9cf6SBram Moolenaar 		     (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1004071d4279SBram Moolenaar 		     NULL);
1005071d4279SBram Moolenaar }
1006071d4279SBram Moolenaar 
10072c45e945SBram Moolenaar     static PyInt
10082c45e945SBram Moolenaar BufferAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
1009071d4279SBram Moolenaar {
1010*19e60943SBram Moolenaar     return RBAsSlice((BufferObject *)(self), lo, hi, val, 1,
1011e7cb9cf6SBram Moolenaar 		      (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1012071d4279SBram Moolenaar 		      NULL);
1013071d4279SBram Moolenaar }
1014071d4279SBram Moolenaar 
1015071d4279SBram Moolenaar static PySequenceMethods RangeAsSeq = {
10162c45e945SBram Moolenaar     (PyInquiry)		RangeLength,	    /* sq_length,    len(x)   */
1017071d4279SBram Moolenaar     (binaryfunc)	0, /* RangeConcat, */	     /* sq_concat,    x+y      */
10182c45e945SBram Moolenaar     (PyIntArgFunc)	0, /* RangeRepeat, */	     /* sq_repeat,    x*n      */
10192c45e945SBram Moolenaar     (PyIntArgFunc)	RangeItem,	    /* sq_item,      x[i]     */
10202c45e945SBram Moolenaar     (PyIntIntArgFunc)	RangeSlice,	    /* sq_slice,     x[i:j]   */
10212c45e945SBram Moolenaar     (PyIntObjArgProc)	RangeAssItem,	    /* sq_ass_item,  x[i]=v   */
10222c45e945SBram Moolenaar     (PyIntIntObjArgProc)	RangeAssSlice,	    /* sq_ass_slice, x[i:j]=v */
1023071d4279SBram Moolenaar };
1024071d4279SBram Moolenaar 
1025071d4279SBram Moolenaar /* Line range object - Implementation
1026071d4279SBram Moolenaar  */
1027071d4279SBram Moolenaar 
1028071d4279SBram Moolenaar     static void
1029071d4279SBram Moolenaar RangeDestructor(PyObject *self)
1030071d4279SBram Moolenaar {
1031071d4279SBram Moolenaar     Py_DECREF(((RangeObject *)(self))->buf);
1032658ada69SBram Moolenaar     Py_DECREF(self);
1033071d4279SBram Moolenaar }
1034071d4279SBram Moolenaar 
1035071d4279SBram Moolenaar     static PyObject *
1036071d4279SBram Moolenaar RangeGetattr(PyObject *self, char *name)
1037071d4279SBram Moolenaar {
1038071d4279SBram Moolenaar     if (strcmp(name, "start") == 0)
1039e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->start - 1);
1040071d4279SBram Moolenaar     else if (strcmp(name, "end") == 0)
1041e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->end - 1);
1042071d4279SBram Moolenaar     else
1043071d4279SBram Moolenaar 	return Py_FindMethod(RangeMethods, self, name);
1044071d4279SBram Moolenaar }
1045071d4279SBram Moolenaar 
1046071d4279SBram Moolenaar /****************/
1047071d4279SBram Moolenaar 
10482c45e945SBram Moolenaar     static PyInt
10492c45e945SBram Moolenaar RangeAssItem(PyObject *self, PyInt n, PyObject *val)
1050071d4279SBram Moolenaar {
1051ca8a4dfeSBram Moolenaar     return RBAsItem(((RangeObject *)(self))->buf, n, val,
1052071d4279SBram Moolenaar 		     ((RangeObject *)(self))->start,
1053071d4279SBram Moolenaar 		     ((RangeObject *)(self))->end,
1054071d4279SBram Moolenaar 		     &((RangeObject *)(self))->end);
1055071d4279SBram Moolenaar }
1056071d4279SBram Moolenaar 
10572c45e945SBram Moolenaar     static PyInt
10582c45e945SBram Moolenaar RangeAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
1059071d4279SBram Moolenaar {
1060*19e60943SBram Moolenaar     return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val,
1061071d4279SBram Moolenaar 		      ((RangeObject *)(self))->start,
1062071d4279SBram Moolenaar 		      ((RangeObject *)(self))->end,
1063071d4279SBram Moolenaar 		      &((RangeObject *)(self))->end);
1064071d4279SBram Moolenaar }
1065071d4279SBram Moolenaar 
1066071d4279SBram Moolenaar /* Buffer list object - Definitions
1067071d4279SBram Moolenaar  */
1068071d4279SBram Moolenaar 
1069071d4279SBram Moolenaar typedef struct
1070071d4279SBram Moolenaar {
1071071d4279SBram Moolenaar     PyObject_HEAD
1072ca8a4dfeSBram Moolenaar } BufListObject;
1073071d4279SBram Moolenaar 
1074071d4279SBram Moolenaar static PySequenceMethods BufListAsSeq = {
10752c45e945SBram Moolenaar     (PyInquiry)		BufListLength,	    /* sq_length,    len(x)   */
1076071d4279SBram Moolenaar     (binaryfunc)	0,		    /* sq_concat,    x+y      */
10772c45e945SBram Moolenaar     (PyIntArgFunc)	0,		    /* sq_repeat,    x*n      */
10782c45e945SBram Moolenaar     (PyIntArgFunc)	BufListItem,	    /* sq_item,      x[i]     */
10792c45e945SBram Moolenaar     (PyIntIntArgFunc)	0,		    /* sq_slice,     x[i:j]   */
10802c45e945SBram Moolenaar     (PyIntObjArgProc)	0,		    /* sq_ass_item,  x[i]=v   */
10812c45e945SBram Moolenaar     (PyIntIntObjArgProc)	0,		    /* sq_ass_slice, x[i:j]=v */
1082071d4279SBram Moolenaar };
1083071d4279SBram Moolenaar 
1084071d4279SBram Moolenaar static PyTypeObject BufListType = {
1085071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
1086071d4279SBram Moolenaar     0,
1087071d4279SBram Moolenaar     "buffer list",
1088071d4279SBram Moolenaar     sizeof(BufListObject),
1089071d4279SBram Moolenaar     0,
1090071d4279SBram Moolenaar 
1091071d4279SBram Moolenaar     (destructor)    0,			/* tp_dealloc,	refcount==0  */
1092071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
1093071d4279SBram Moolenaar     (getattrfunc)   0,			/* tp_getattr,	x.attr	     */
1094071d4279SBram Moolenaar     (setattrfunc)   0,			/* tp_setattr,	x.attr=v     */
1095071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
1096071d4279SBram Moolenaar     (reprfunc)	    0,			/* tp_repr,	`x`, print x */
1097071d4279SBram Moolenaar 
1098071d4279SBram Moolenaar     0,		    /* as number */
1099071d4279SBram Moolenaar     &BufListAsSeq,  /* as sequence */
1100071d4279SBram Moolenaar     0,		    /* as mapping */
1101071d4279SBram Moolenaar 
1102071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
1103071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
1104071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
1105071d4279SBram Moolenaar };
1106071d4279SBram Moolenaar 
1107071d4279SBram Moolenaar /* Window object - Definitions
1108071d4279SBram Moolenaar  */
1109071d4279SBram Moolenaar 
1110071d4279SBram Moolenaar static struct PyMethodDef WindowMethods[] = {
1111071d4279SBram Moolenaar     /* name,	    function,		calling,    documentation */
1112071d4279SBram Moolenaar     { NULL,	    NULL,		0,	    NULL }
1113071d4279SBram Moolenaar };
1114071d4279SBram Moolenaar 
1115071d4279SBram Moolenaar static PyTypeObject WindowType = {
1116071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
1117071d4279SBram Moolenaar     0,
1118071d4279SBram Moolenaar     "window",
1119071d4279SBram Moolenaar     sizeof(WindowObject),
1120071d4279SBram Moolenaar     0,
1121071d4279SBram Moolenaar 
1122071d4279SBram Moolenaar     (destructor)    WindowDestructor,	/* tp_dealloc,	refcount==0  */
1123071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
1124071d4279SBram Moolenaar     (getattrfunc)   WindowGetattr,	/* tp_getattr,	x.attr	     */
1125071d4279SBram Moolenaar     (setattrfunc)   WindowSetattr,	/* tp_setattr,	x.attr=v     */
1126071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
1127071d4279SBram Moolenaar     (reprfunc)	    WindowRepr,		/* tp_repr,	`x`, print x */
1128071d4279SBram Moolenaar 
1129071d4279SBram Moolenaar     0,		    /* as number */
1130071d4279SBram Moolenaar     0,		    /* as sequence */
1131071d4279SBram Moolenaar     0,		    /* as mapping */
1132071d4279SBram Moolenaar 
1133071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
1134071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
1135071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
1136071d4279SBram Moolenaar };
1137071d4279SBram Moolenaar 
1138071d4279SBram Moolenaar /* Window object - Implementation
1139071d4279SBram Moolenaar  */
1140071d4279SBram Moolenaar 
1141071d4279SBram Moolenaar     static PyObject *
1142071d4279SBram Moolenaar WindowNew(win_T *win)
1143071d4279SBram Moolenaar {
1144071d4279SBram Moolenaar     /* We need to handle deletion of windows underneath us.
1145e344beadSBram Moolenaar      * If we add a "w_python_ref" field to the win_T structure,
1146071d4279SBram Moolenaar      * then we can get at it in win_free() in vim. We then
1147071d4279SBram Moolenaar      * need to create only ONE Python object per window - if
1148071d4279SBram Moolenaar      * we try to create a second, just INCREF the existing one
1149071d4279SBram Moolenaar      * and return it. The (single) Python object referring to
1150e344beadSBram Moolenaar      * the window is stored in "w_python_ref".
1151071d4279SBram Moolenaar      * On a win_free() we set the Python object's win_T* field
1152071d4279SBram Moolenaar      * to an invalid value. We trap all uses of a window
1153071d4279SBram Moolenaar      * object, and reject them if the win_T* field is invalid.
1154071d4279SBram Moolenaar      */
1155071d4279SBram Moolenaar 
1156071d4279SBram Moolenaar     WindowObject *self;
1157071d4279SBram Moolenaar 
1158e344beadSBram Moolenaar     if (win->w_python_ref)
1159071d4279SBram Moolenaar     {
1160e344beadSBram Moolenaar 	self = win->w_python_ref;
1161071d4279SBram Moolenaar 	Py_INCREF(self);
1162071d4279SBram Moolenaar     }
1163071d4279SBram Moolenaar     else
1164071d4279SBram Moolenaar     {
1165071d4279SBram Moolenaar 	self = PyObject_NEW(WindowObject, &WindowType);
1166071d4279SBram Moolenaar 	if (self == NULL)
1167071d4279SBram Moolenaar 	    return NULL;
1168071d4279SBram Moolenaar 	self->win = win;
1169e344beadSBram Moolenaar 	win->w_python_ref = self;
1170071d4279SBram Moolenaar     }
1171071d4279SBram Moolenaar 
1172071d4279SBram Moolenaar     return (PyObject *)(self);
1173071d4279SBram Moolenaar }
1174071d4279SBram Moolenaar 
1175071d4279SBram Moolenaar     static void
1176071d4279SBram Moolenaar WindowDestructor(PyObject *self)
1177071d4279SBram Moolenaar {
1178071d4279SBram Moolenaar     WindowObject *this = (WindowObject *)(self);
1179071d4279SBram Moolenaar 
1180071d4279SBram Moolenaar     if (this->win && this->win != INVALID_WINDOW_VALUE)
1181e344beadSBram Moolenaar 	this->win->w_python_ref = NULL;
1182071d4279SBram Moolenaar 
1183658ada69SBram Moolenaar     Py_DECREF(self);
1184071d4279SBram Moolenaar }
1185071d4279SBram Moolenaar 
1186071d4279SBram Moolenaar     static PyObject *
1187071d4279SBram Moolenaar WindowGetattr(PyObject *self, char *name)
1188071d4279SBram Moolenaar {
1189071d4279SBram Moolenaar     WindowObject *this = (WindowObject *)(self);
1190071d4279SBram Moolenaar 
1191071d4279SBram Moolenaar     if (CheckWindow(this))
1192071d4279SBram Moolenaar 	return NULL;
1193071d4279SBram Moolenaar 
1194071d4279SBram Moolenaar     if (strcmp(name, "buffer") == 0)
1195071d4279SBram Moolenaar 	return (PyObject *)BufferNew(this->win->w_buffer);
1196071d4279SBram Moolenaar     else if (strcmp(name, "cursor") == 0)
1197071d4279SBram Moolenaar     {
1198071d4279SBram Moolenaar 	pos_T *pos = &this->win->w_cursor;
1199071d4279SBram Moolenaar 
1200071d4279SBram Moolenaar 	return Py_BuildValue("(ll)", (long)(pos->lnum), (long)(pos->col));
1201071d4279SBram Moolenaar     }
1202071d4279SBram Moolenaar     else if (strcmp(name, "height") == 0)
1203071d4279SBram Moolenaar 	return Py_BuildValue("l", (long)(this->win->w_height));
1204071d4279SBram Moolenaar #ifdef FEAT_VERTSPLIT
1205071d4279SBram Moolenaar     else if (strcmp(name, "width") == 0)
1206071d4279SBram Moolenaar 	return Py_BuildValue("l", (long)(W_WIDTH(this->win)));
1207071d4279SBram Moolenaar #endif
1208071d4279SBram Moolenaar     else if (strcmp(name,"__members__") == 0)
1209071d4279SBram Moolenaar 	return Py_BuildValue("[sss]", "buffer", "cursor", "height");
1210071d4279SBram Moolenaar     else
1211071d4279SBram Moolenaar 	return Py_FindMethod(WindowMethods, self, name);
1212071d4279SBram Moolenaar }
1213071d4279SBram Moolenaar 
1214071d4279SBram Moolenaar /* Window list object - Definitions
1215071d4279SBram Moolenaar  */
1216071d4279SBram Moolenaar 
1217071d4279SBram Moolenaar typedef struct
1218071d4279SBram Moolenaar {
1219071d4279SBram Moolenaar     PyObject_HEAD
1220071d4279SBram Moolenaar }
1221071d4279SBram Moolenaar WinListObject;
1222071d4279SBram Moolenaar 
1223071d4279SBram Moolenaar static PySequenceMethods WinListAsSeq = {
12242c45e945SBram Moolenaar     (PyInquiry)		WinListLength,	    /* sq_length,    len(x)   */
1225071d4279SBram Moolenaar     (binaryfunc)	0,		    /* sq_concat,    x+y      */
12262c45e945SBram Moolenaar     (PyIntArgFunc)	0,		    /* sq_repeat,    x*n      */
12272c45e945SBram Moolenaar     (PyIntArgFunc)	WinListItem,	    /* sq_item,      x[i]     */
12282c45e945SBram Moolenaar     (PyIntIntArgFunc)	0,		    /* sq_slice,     x[i:j]   */
12292c45e945SBram Moolenaar     (PyIntObjArgProc)	0,		    /* sq_ass_item,  x[i]=v   */
12302c45e945SBram Moolenaar     (PyIntIntObjArgProc)	0,		    /* sq_ass_slice, x[i:j]=v */
1231071d4279SBram Moolenaar };
1232071d4279SBram Moolenaar 
1233071d4279SBram Moolenaar static PyTypeObject WinListType = {
1234071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
1235071d4279SBram Moolenaar     0,
1236071d4279SBram Moolenaar     "window list",
1237071d4279SBram Moolenaar     sizeof(WinListObject),
1238071d4279SBram Moolenaar     0,
1239071d4279SBram Moolenaar 
1240071d4279SBram Moolenaar     (destructor)    0,			/* tp_dealloc,	refcount==0  */
1241071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
1242071d4279SBram Moolenaar     (getattrfunc)   0,			/* tp_getattr,	x.attr	     */
1243071d4279SBram Moolenaar     (setattrfunc)   0,			/* tp_setattr,	x.attr=v     */
1244071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
1245071d4279SBram Moolenaar     (reprfunc)	    0,			/* tp_repr,	`x`, print x */
1246071d4279SBram Moolenaar 
1247071d4279SBram Moolenaar     0,		    /* as number */
1248071d4279SBram Moolenaar     &WinListAsSeq,  /* as sequence */
1249071d4279SBram Moolenaar     0,		    /* as mapping */
1250071d4279SBram Moolenaar 
1251071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
1252071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
1253071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
1254071d4279SBram Moolenaar };
1255071d4279SBram Moolenaar 
1256071d4279SBram Moolenaar /* Current items object - Definitions
1257071d4279SBram Moolenaar  */
1258071d4279SBram Moolenaar 
1259071d4279SBram Moolenaar typedef struct
1260071d4279SBram Moolenaar {
1261071d4279SBram Moolenaar     PyObject_HEAD
1262ca8a4dfeSBram Moolenaar } CurrentObject;
1263071d4279SBram Moolenaar 
1264071d4279SBram Moolenaar static PyTypeObject CurrentType = {
1265071d4279SBram Moolenaar     PyObject_HEAD_INIT(0)
1266071d4279SBram Moolenaar     0,
1267071d4279SBram Moolenaar     "current data",
1268071d4279SBram Moolenaar     sizeof(CurrentObject),
1269071d4279SBram Moolenaar     0,
1270071d4279SBram Moolenaar 
1271071d4279SBram Moolenaar     (destructor)    0,			/* tp_dealloc,	refcount==0  */
1272071d4279SBram Moolenaar     (printfunc)     0,			/* tp_print,	print x      */
1273071d4279SBram Moolenaar     (getattrfunc)   CurrentGetattr,	/* tp_getattr,	x.attr	     */
1274071d4279SBram Moolenaar     (setattrfunc)   CurrentSetattr,	/* tp_setattr,	x.attr=v     */
1275071d4279SBram Moolenaar     (cmpfunc)	    0,			/* tp_compare,	x>y	     */
1276071d4279SBram Moolenaar     (reprfunc)	    0,			/* tp_repr,	`x`, print x */
1277071d4279SBram Moolenaar 
1278071d4279SBram Moolenaar     0,		    /* as number */
1279071d4279SBram Moolenaar     0,		    /* as sequence */
1280071d4279SBram Moolenaar     0,		    /* as mapping */
1281071d4279SBram Moolenaar 
1282071d4279SBram Moolenaar     (hashfunc) 0,			/* tp_hash, dict(x) */
1283071d4279SBram Moolenaar     (ternaryfunc) 0,			/* tp_call, x()     */
1284071d4279SBram Moolenaar     (reprfunc) 0,			/* tp_str,  str(x)  */
1285071d4279SBram Moolenaar };
1286071d4279SBram Moolenaar 
1287071d4279SBram Moolenaar /* Current items object - Implementation
1288071d4279SBram Moolenaar  */
1289071d4279SBram Moolenaar     static PyObject *
12904bdbbf70SBram Moolenaar CurrentGetattr(PyObject *self UNUSED, char *name)
1291071d4279SBram Moolenaar {
1292071d4279SBram Moolenaar     if (strcmp(name, "buffer") == 0)
1293071d4279SBram Moolenaar 	return (PyObject *)BufferNew(curbuf);
1294071d4279SBram Moolenaar     else if (strcmp(name, "window") == 0)
1295071d4279SBram Moolenaar 	return (PyObject *)WindowNew(curwin);
1296071d4279SBram Moolenaar     else if (strcmp(name, "line") == 0)
1297e7cb9cf6SBram Moolenaar 	return GetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum);
1298071d4279SBram Moolenaar     else if (strcmp(name, "range") == 0)
1299071d4279SBram Moolenaar 	return RangeNew(curbuf, RangeStart, RangeEnd);
1300071d4279SBram Moolenaar     else if (strcmp(name,"__members__") == 0)
1301071d4279SBram Moolenaar 	return Py_BuildValue("[ssss]", "buffer", "window", "line", "range");
1302071d4279SBram Moolenaar     else
1303071d4279SBram Moolenaar     {
1304071d4279SBram Moolenaar 	PyErr_SetString(PyExc_AttributeError, name);
1305071d4279SBram Moolenaar 	return NULL;
1306071d4279SBram Moolenaar     }
1307071d4279SBram Moolenaar }
1308071d4279SBram Moolenaar 
1309071d4279SBram Moolenaar     static int
13104bdbbf70SBram Moolenaar CurrentSetattr(PyObject *self UNUSED, char *name, PyObject *value)
1311071d4279SBram Moolenaar {
1312071d4279SBram Moolenaar     if (strcmp(name, "line") == 0)
1313071d4279SBram Moolenaar     {
1314e7cb9cf6SBram Moolenaar 	if (SetBufferLine(curbuf, (PyInt)curwin->w_cursor.lnum, value, NULL) == FAIL)
1315071d4279SBram Moolenaar 	    return -1;
1316071d4279SBram Moolenaar 
1317071d4279SBram Moolenaar 	return 0;
1318071d4279SBram Moolenaar     }
1319071d4279SBram Moolenaar     else
1320071d4279SBram Moolenaar     {
1321071d4279SBram Moolenaar 	PyErr_SetString(PyExc_AttributeError, name);
1322071d4279SBram Moolenaar 	return -1;
1323071d4279SBram Moolenaar     }
1324071d4279SBram Moolenaar }
1325071d4279SBram Moolenaar 
1326071d4279SBram Moolenaar /* External interface
1327071d4279SBram Moolenaar  */
1328071d4279SBram Moolenaar 
1329071d4279SBram Moolenaar     void
1330071d4279SBram Moolenaar python_buffer_free(buf_T *buf)
1331071d4279SBram Moolenaar {
1332e344beadSBram Moolenaar     if (buf->b_python_ref != NULL)
1333071d4279SBram Moolenaar     {
1334e344beadSBram Moolenaar 	BufferObject *bp = buf->b_python_ref;
1335071d4279SBram Moolenaar 	bp->buf = INVALID_BUFFER_VALUE;
1336e344beadSBram Moolenaar 	buf->b_python_ref = NULL;
1337071d4279SBram Moolenaar     }
1338071d4279SBram Moolenaar }
1339071d4279SBram Moolenaar 
1340071d4279SBram Moolenaar #if defined(FEAT_WINDOWS) || defined(PROTO)
1341071d4279SBram Moolenaar     void
1342071d4279SBram Moolenaar python_window_free(win_T *win)
1343071d4279SBram Moolenaar {
1344e344beadSBram Moolenaar     if (win->w_python_ref != NULL)
1345071d4279SBram Moolenaar     {
1346e344beadSBram Moolenaar 	WindowObject *wp = win->w_python_ref;
1347071d4279SBram Moolenaar 	wp->win = INVALID_WINDOW_VALUE;
1348e344beadSBram Moolenaar 	win->w_python_ref = NULL;
1349071d4279SBram Moolenaar     }
1350071d4279SBram Moolenaar }
1351071d4279SBram Moolenaar #endif
1352071d4279SBram Moolenaar 
1353071d4279SBram Moolenaar static BufListObject TheBufferList =
1354071d4279SBram Moolenaar {
1355071d4279SBram Moolenaar     PyObject_HEAD_INIT(&BufListType)
1356071d4279SBram Moolenaar };
1357071d4279SBram Moolenaar 
1358071d4279SBram Moolenaar static WinListObject TheWindowList =
1359071d4279SBram Moolenaar {
1360071d4279SBram Moolenaar     PyObject_HEAD_INIT(&WinListType)
1361071d4279SBram Moolenaar };
1362071d4279SBram Moolenaar 
1363071d4279SBram Moolenaar static CurrentObject TheCurrent =
1364071d4279SBram Moolenaar {
1365071d4279SBram Moolenaar     PyObject_HEAD_INIT(&CurrentType)
1366071d4279SBram Moolenaar };
1367071d4279SBram Moolenaar 
1368071d4279SBram Moolenaar     static int
1369071d4279SBram Moolenaar PythonMod_Init(void)
1370071d4279SBram Moolenaar {
1371071d4279SBram Moolenaar     PyObject *mod;
1372071d4279SBram Moolenaar     PyObject *dict;
13739774ecc8SBram Moolenaar     /* The special value is removed from sys.path in Python_Init(). */
13749774ecc8SBram Moolenaar     static char *(argv[2]) = {"/must>not&exist/foo", NULL};
1375071d4279SBram Moolenaar 
1376071d4279SBram Moolenaar     /* Fixups... */
137721377c8dSBram Moolenaar     PyType_Ready(&BufferType);
137821377c8dSBram Moolenaar     PyType_Ready(&RangeType);
137921377c8dSBram Moolenaar     PyType_Ready(&WindowType);
138021377c8dSBram Moolenaar     PyType_Ready(&BufListType);
138121377c8dSBram Moolenaar     PyType_Ready(&WinListType);
138221377c8dSBram Moolenaar     PyType_Ready(&CurrentType);
1383071d4279SBram Moolenaar 
1384071d4279SBram Moolenaar     /* Set sys.argv[] to avoid a crash in warn(). */
1385071d4279SBram Moolenaar     PySys_SetArgv(1, argv);
1386071d4279SBram Moolenaar 
1387e7cb9cf6SBram Moolenaar     mod = Py_InitModule4("vim", VimMethods, (char *)NULL, (PyObject *)NULL, PYTHON_API_VERSION);
1388071d4279SBram Moolenaar     dict = PyModule_GetDict(mod);
1389071d4279SBram Moolenaar 
1390071d4279SBram Moolenaar     VimError = Py_BuildValue("s", "vim.error");
1391071d4279SBram Moolenaar 
1392071d4279SBram Moolenaar     PyDict_SetItemString(dict, "error", VimError);
13937df2d662SBram Moolenaar     PyDict_SetItemString(dict, "buffers", (PyObject *)(void *)&TheBufferList);
13947df2d662SBram Moolenaar     PyDict_SetItemString(dict, "current", (PyObject *)(void *)&TheCurrent);
13957df2d662SBram Moolenaar     PyDict_SetItemString(dict, "windows", (PyObject *)(void *)&TheWindowList);
1396071d4279SBram Moolenaar 
1397071d4279SBram Moolenaar     if (PyErr_Occurred())
1398071d4279SBram Moolenaar 	return -1;
1399071d4279SBram Moolenaar 
1400071d4279SBram Moolenaar     return 0;
1401071d4279SBram Moolenaar }
1402071d4279SBram Moolenaar 
1403071d4279SBram Moolenaar /*************************************************************************
1404071d4279SBram Moolenaar  * 4. Utility functions for handling the interface between Vim and Python.
1405071d4279SBram Moolenaar  */
1406071d4279SBram Moolenaar 
1407071d4279SBram Moolenaar /* Convert a Vim line into a Python string.
1408071d4279SBram Moolenaar  * All internal newlines are replaced by null characters.
1409071d4279SBram Moolenaar  *
1410071d4279SBram Moolenaar  * On errors, the Python exception data is set, and NULL is returned.
1411071d4279SBram Moolenaar  */
1412071d4279SBram Moolenaar     static PyObject *
1413071d4279SBram Moolenaar LineToString(const char *str)
1414071d4279SBram Moolenaar {
1415071d4279SBram Moolenaar     PyObject *result;
14162c45e945SBram Moolenaar     PyInt len = strlen(str);
1417071d4279SBram Moolenaar     char *p;
1418071d4279SBram Moolenaar 
1419071d4279SBram Moolenaar     /* Allocate an Python string object, with uninitialised contents. We
1420071d4279SBram Moolenaar      * must do it this way, so that we can modify the string in place
1421071d4279SBram Moolenaar      * later. See the Python source, Objects/stringobject.c for details.
1422071d4279SBram Moolenaar      */
1423071d4279SBram Moolenaar     result = PyString_FromStringAndSize(NULL, len);
1424071d4279SBram Moolenaar     if (result == NULL)
1425071d4279SBram Moolenaar 	return NULL;
1426071d4279SBram Moolenaar 
1427071d4279SBram Moolenaar     p = PyString_AsString(result);
1428071d4279SBram Moolenaar 
1429071d4279SBram Moolenaar     while (*str)
1430071d4279SBram Moolenaar     {
1431071d4279SBram Moolenaar 	if (*str == '\n')
1432071d4279SBram Moolenaar 	    *p = '\0';
1433071d4279SBram Moolenaar 	else
1434071d4279SBram Moolenaar 	    *p = *str;
1435071d4279SBram Moolenaar 
1436071d4279SBram Moolenaar 	++p;
1437071d4279SBram Moolenaar 	++str;
1438071d4279SBram Moolenaar     }
1439071d4279SBram Moolenaar 
1440071d4279SBram Moolenaar     return result;
1441071d4279SBram Moolenaar }
1442071d4279SBram Moolenaar 
1443071d4279SBram Moolenaar 
1444071d4279SBram Moolenaar /* Don't generate a prototype for the next function, it generates an error on
1445071d4279SBram Moolenaar  * newer Python versions. */
1446071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ && !defined(PROTO)
1447071d4279SBram Moolenaar 
1448071d4279SBram Moolenaar     char *
1449071d4279SBram Moolenaar Py_GetProgramName(void)
1450071d4279SBram Moolenaar {
1451071d4279SBram Moolenaar     return "vim";
1452071d4279SBram Moolenaar }
1453071d4279SBram Moolenaar #endif /* Python 1.4 */
1454170bf1aeSBram Moolenaar 
1455170bf1aeSBram Moolenaar     static void
1456170bf1aeSBram Moolenaar init_structs(void)
1457170bf1aeSBram Moolenaar {
1458170bf1aeSBram Moolenaar     vim_memset(&OutputType, 0, sizeof(OutputType));
1459170bf1aeSBram Moolenaar     OutputType.tp_name = "message";
1460170bf1aeSBram Moolenaar     OutputType.tp_basicsize = sizeof(OutputObject);
1461170bf1aeSBram Moolenaar     OutputType.tp_getattr = OutputGetattr;
1462170bf1aeSBram Moolenaar     OutputType.tp_setattr = OutputSetattr;
1463ca8a4dfeSBram Moolenaar 
1464ca8a4dfeSBram Moolenaar     vim_memset(&RangeType, 0, sizeof(RangeType));
1465ca8a4dfeSBram Moolenaar     RangeType.tp_name = "range";
1466ca8a4dfeSBram Moolenaar     RangeType.tp_basicsize = sizeof(RangeObject);
1467ca8a4dfeSBram Moolenaar     RangeType.tp_dealloc = RangeDestructor;
1468ca8a4dfeSBram Moolenaar     RangeType.tp_getattr = RangeGetattr;
1469ca8a4dfeSBram Moolenaar     RangeType.tp_repr = RangeRepr;
1470ca8a4dfeSBram Moolenaar     RangeType.tp_as_sequence = &RangeAsSeq;
1471170bf1aeSBram Moolenaar }
1472