blob: 023a773e21827ba4e929b2601d1365082848d58d [file] [log] [blame]
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9/*
10 * Python extensions by Paul Moore.
11 * Changes for Unix by David Leonard.
12 *
13 * This consists of four parts:
14 * 1. Python interpreter main program
15 * 2. Python output stream: writes output via [e]msg().
16 * 3. Implementation of the Vim module for Python
17 * 4. Utility functions for handling the interface between Vim and Python.
18 */
19
20/*
21 * Roland Puntaier 2009/sept/16:
22 * Adaptations to support both python3.x and python2.x
23 */
24
Bram Moolenaar0c1f3f42011-02-25 15:18:50 +010025/* uncomment this if used with the debug version of python */
26/* #define Py_DEBUG */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +020027
28#include "vim.h"
29
30#include <limits.h>
31
32/* Python.h defines _POSIX_THREADS itself (if needed) */
33#ifdef _POSIX_THREADS
34# undef _POSIX_THREADS
35#endif
36
Bram Moolenaard68554d2010-07-25 13:43:20 +020037#if defined(_WIN32) && defined(HAVE_FCNTL_H)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +020038# undef HAVE_FCNTL_H
39#endif
40
41#ifdef _DEBUG
42# undef _DEBUG
43#endif
44
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +020045#define PY_SSIZE_T_CLEAN
46
47#ifdef F_BLANK
48# undef F_BLANK
49#endif
50
Bram Moolenaar6df6f472010-07-18 18:04:50 +020051#ifdef HAVE_STDARG_H
52# undef HAVE_STDARG_H /* Python's config.h defines it as well. */
53#endif
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +020054#ifdef _POSIX_C_SOURCE /* defined in feature.h */
55# undef _POSIX_C_SOURCE
56#endif
Bram Moolenaar6df6f472010-07-18 18:04:50 +020057#ifdef _XOPEN_SOURCE
58# undef _XOPEN_SOURCE /* pyconfig.h defines it as well. */
59#endif
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +020060
61#include <Python.h>
62#if defined(MACOS) && !defined(MACOS_X_UNIX)
63# include "macglue.h"
64# include <CodeFragments.h>
65#endif
66#undef main /* Defined in python.h - aargh */
67#undef HAVE_FCNTL_H /* Clash with os_win32.h */
68
69static void init_structs(void);
70
Bram Moolenaar3d64a312011-07-15 15:54:44 +020071/* The "surrogateescape" error handler is new in Python 3.1 */
72#if PY_VERSION_HEX >= 0x030100f0
73# define CODEC_ERROR_HANDLER "surrogateescape"
74#else
75# define CODEC_ERROR_HANDLER NULL
76#endif
77
Bram Moolenaar170bf1a2010-07-24 23:51:45 +020078#define PyInt Py_ssize_t
Bram Moolenaarca8a4df2010-07-31 19:54:14 +020079#define PyString_Check(obj) PyUnicode_Check(obj)
Bram Moolenaar3d64a312011-07-15 15:54:44 +020080#define PyString_AsBytes(obj) PyUnicode_AsEncodedString(obj, (char *)ENC_OPT, CODEC_ERROR_HANDLER);
Bram Moolenaar19e60942011-06-19 00:27:51 +020081#define PyString_FreeBytes(obj) Py_XDECREF(bytes)
82#define PyString_AsString(obj) PyBytes_AsString(obj)
83#define PyString_Size(obj) PyBytes_GET_SIZE(bytes)
Bram Moolenaarca8a4df2010-07-31 19:54:14 +020084#define PyString_FromString(repr) PyUnicode_FromString(repr)
Bram Moolenaar170bf1a2010-07-24 23:51:45 +020085
Bram Moolenaar0c1f3f42011-02-25 15:18:50 +010086#if defined(DYNAMIC_PYTHON3) || defined(PROTO)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +020087
Bram Moolenaarb61f95c2010-08-09 22:06:13 +020088# ifndef WIN3264
89# include <dlfcn.h>
90# define FARPROC void*
91# define HINSTANCE void*
Bram Moolenaar644d37b2010-11-16 19:26:02 +010092# if defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)
Bram Moolenaarb61f95c2010-08-09 22:06:13 +020093# define load_dll(n) dlopen((n), RTLD_LAZY)
94# else
95# define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL)
96# endif
97# define close_dll dlclose
98# define symbol_from_dll dlsym
99# else
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200100# define load_dll vimLoadLib
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200101# define close_dll FreeLibrary
102# define symbol_from_dll GetProcAddress
103# endif
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200104/*
105 * Wrapper defines
106 */
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200107# undef PyArg_Parse
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200108# define PyArg_Parse py3_PyArg_Parse
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200109# undef PyArg_ParseTuple
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200110# define PyArg_ParseTuple py3_PyArg_ParseTuple
Bram Moolenaar19e60942011-06-19 00:27:51 +0200111# define PyMem_Free py3_PyMem_Free
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200112# define PyDict_SetItemString py3_PyDict_SetItemString
113# define PyErr_BadArgument py3_PyErr_BadArgument
114# define PyErr_Clear py3_PyErr_Clear
115# define PyErr_NoMemory py3_PyErr_NoMemory
116# define PyErr_Occurred py3_PyErr_Occurred
117# define PyErr_SetNone py3_PyErr_SetNone
118# define PyErr_SetString py3_PyErr_SetString
119# define PyEval_InitThreads py3_PyEval_InitThreads
120# define PyEval_RestoreThread py3_PyEval_RestoreThread
121# define PyEval_SaveThread py3_PyEval_SaveThread
122# define PyGILState_Ensure py3_PyGILState_Ensure
123# define PyGILState_Release py3_PyGILState_Release
124# define PyLong_AsLong py3_PyLong_AsLong
125# define PyLong_FromLong py3_PyLong_FromLong
126# define PyList_GetItem py3_PyList_GetItem
127# define PyList_Append py3_PyList_Append
128# define PyList_New py3_PyList_New
129# define PyList_SetItem py3_PyList_SetItem
130# define PyList_Size py3_PyList_Size
131# define PySlice_GetIndicesEx py3_PySlice_GetIndicesEx
132# define PyImport_ImportModule py3_PyImport_ImportModule
133# define PyObject_Init py3__PyObject_Init
134# define PyDict_New py3_PyDict_New
135# define PyDict_GetItemString py3_PyDict_GetItemString
136# define PyModule_GetDict py3_PyModule_GetDict
137#undef PyRun_SimpleString
138# define PyRun_SimpleString py3_PyRun_SimpleString
139# define PySys_SetObject py3_PySys_SetObject
140# define PySys_SetArgv py3_PySys_SetArgv
141# define PyType_Type (*py3_PyType_Type)
142# define PyType_Ready py3_PyType_Ready
143#undef Py_BuildValue
144# define Py_BuildValue py3_Py_BuildValue
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100145# define Py_SetPythonHome py3_Py_SetPythonHome
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200146# define Py_Initialize py3_Py_Initialize
147# define Py_Finalize py3_Py_Finalize
148# define Py_IsInitialized py3_Py_IsInitialized
149# define _Py_NoneStruct (*py3__Py_NoneStruct)
150# define PyModule_AddObject py3_PyModule_AddObject
151# define PyImport_AppendInittab py3_PyImport_AppendInittab
152# define _PyUnicode_AsString py3__PyUnicode_AsString
Bram Moolenaar19e60942011-06-19 00:27:51 +0200153# undef PyUnicode_AsEncodedString
154# define PyUnicode_AsEncodedString py3_PyUnicode_AsEncodedString
155# undef PyBytes_AsString
156# define PyBytes_AsString py3_PyBytes_AsString
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200157# define PyObject_GenericGetAttr py3_PyObject_GenericGetAttr
158# define PySlice_Type (*py3_PySlice_Type)
Bram Moolenaar19e60942011-06-19 00:27:51 +0200159# define PyErr_NewException py3_PyErr_NewException
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200160# ifdef Py_DEBUG
161# define _Py_NegativeRefcount py3__Py_NegativeRefcount
162# define _Py_RefTotal (*py3__Py_RefTotal)
163# define _Py_Dealloc py3__Py_Dealloc
164# define _PyObject_DebugMalloc py3__PyObject_DebugMalloc
165# define _PyObject_DebugFree py3__PyObject_DebugFree
166# else
167# define PyObject_Malloc py3_PyObject_Malloc
168# define PyObject_Free py3_PyObject_Free
169# endif
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200170# define PyType_GenericAlloc py3_PyType_GenericAlloc
171# define PyType_GenericNew py3_PyType_GenericNew
172# define PyModule_Create2 py3_PyModule_Create2
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200173# undef PyUnicode_FromString
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200174# define PyUnicode_FromString py3_PyUnicode_FromString
Bram Moolenaar19e60942011-06-19 00:27:51 +0200175# undef PyUnicode_Decode
176# define PyUnicode_Decode py3_PyUnicode_Decode
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200177
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200178# ifdef Py_DEBUG
179# undef PyObject_NEW
180# define PyObject_NEW(type, typeobj) \
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200181( (type *) PyObject_Init( \
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200182 (PyObject *) _PyObject_DebugMalloc( _PyObject_SIZE(typeobj) ), (typeobj)) )
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200183# endif
184
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200185/*
186 * Pointers for dynamic link
187 */
188static int (*py3_PySys_SetArgv)(int, wchar_t **);
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100189static void (*py3_Py_SetPythonHome)(wchar_t *home);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200190static void (*py3_Py_Initialize)(void);
191static PyObject* (*py3_PyList_New)(Py_ssize_t size);
192static PyGILState_STATE (*py3_PyGILState_Ensure)(void);
193static void (*py3_PyGILState_Release)(PyGILState_STATE);
194static int (*py3_PySys_SetObject)(char *, PyObject *);
195static PyObject* (*py3_PyList_Append)(PyObject *, PyObject *);
196static Py_ssize_t (*py3_PyList_Size)(PyObject *);
197static int (*py3_PySlice_GetIndicesEx)(PySliceObject *r, Py_ssize_t length,
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200198 Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step, Py_ssize_t *slicelength);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200199static PyObject* (*py3_PyErr_NoMemory)(void);
200static void (*py3_Py_Finalize)(void);
201static void (*py3_PyErr_SetString)(PyObject *, const char *);
202static int (*py3_PyRun_SimpleString)(char *);
203static PyObject* (*py3_PyList_GetItem)(PyObject *, Py_ssize_t);
204static PyObject* (*py3_PyImport_ImportModule)(const char *);
205static int (*py3_PyErr_BadArgument)(void);
206static PyTypeObject* py3_PyType_Type;
207static PyObject* (*py3_PyErr_Occurred)(void);
208static PyObject* (*py3_PyModule_GetDict)(PyObject *);
209static int (*py3_PyList_SetItem)(PyObject *, Py_ssize_t, PyObject *);
210static PyObject* (*py3_PyDict_GetItemString)(PyObject *, const char *);
211static PyObject* (*py3_PyLong_FromLong)(long);
212static PyObject* (*py3_PyDict_New)(void);
213static PyObject* (*py3_Py_BuildValue)(char *, ...);
214static int (*py3_PyType_Ready)(PyTypeObject *type);
215static int (*py3_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item);
216static PyObject* (*py3_PyUnicode_FromString)(const char *u);
Bram Moolenaar19e60942011-06-19 00:27:51 +0200217static PyObject* (*py3_PyUnicode_Decode)(const char *u, Py_ssize_t size,
218 const char *encoding, const char *errors);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200219static long (*py3_PyLong_AsLong)(PyObject *);
220static void (*py3_PyErr_SetNone)(PyObject *);
221static void (*py3_PyEval_InitThreads)(void);
222static void(*py3_PyEval_RestoreThread)(PyThreadState *);
223static PyThreadState*(*py3_PyEval_SaveThread)(void);
224static int (*py3_PyArg_Parse)(PyObject *, char *, ...);
225static int (*py3_PyArg_ParseTuple)(PyObject *, char *, ...);
Bram Moolenaar19e60942011-06-19 00:27:51 +0200226static int (*py3_PyMem_Free)(void *);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200227static int (*py3_Py_IsInitialized)(void);
228static void (*py3_PyErr_Clear)(void);
229static PyObject*(*py3__PyObject_Init)(PyObject *, PyTypeObject *);
230static PyObject* py3__Py_NoneStruct;
231static int (*py3_PyModule_AddObject)(PyObject *m, const char *name, PyObject *o);
232static int (*py3_PyImport_AppendInittab)(const char *name, PyObject* (*initfunc)(void));
233static char* (*py3__PyUnicode_AsString)(PyObject *unicode);
Bram Moolenaar19e60942011-06-19 00:27:51 +0200234static PyObject* (*py3_PyUnicode_AsEncodedString)(PyObject *unicode, const char* encoding, const char* errors);
235static char* (*py3_PyBytes_AsString)(PyObject *bytes);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200236static PyObject* (*py3_PyObject_GenericGetAttr)(PyObject *obj, PyObject *name);
237static PyObject* (*py3_PyModule_Create2)(struct PyModuleDef* module, int module_api_version);
238static PyObject* (*py3_PyType_GenericAlloc)(PyTypeObject *type, Py_ssize_t nitems);
239static PyObject* (*py3_PyType_GenericNew)(PyTypeObject *type, PyObject *args, PyObject *kwds);
240static PyTypeObject* py3_PySlice_Type;
Bram Moolenaar19e60942011-06-19 00:27:51 +0200241static PyObject* (*py3_PyErr_NewException)(char *name, PyObject *base, PyObject *dict);
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200242# ifdef Py_DEBUG
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200243 static void (*py3__Py_NegativeRefcount)(const char *fname, int lineno, PyObject *op);
244 static Py_ssize_t* py3__Py_RefTotal;
245 static void (*py3__Py_Dealloc)(PyObject *obj);
246 static void (*py3__PyObject_DebugFree)(void*);
247 static void* (*py3__PyObject_DebugMalloc)(size_t);
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200248# else
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200249 static void (*py3_PyObject_Free)(void*);
250 static void* (*py3_PyObject_Malloc)(size_t);
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200251# endif
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200252
253static HINSTANCE hinstPy3 = 0; /* Instance of python.dll */
254
255/* Imported exception objects */
256static PyObject *p3imp_PyExc_AttributeError;
257static PyObject *p3imp_PyExc_IndexError;
258static PyObject *p3imp_PyExc_KeyboardInterrupt;
259static PyObject *p3imp_PyExc_TypeError;
260static PyObject *p3imp_PyExc_ValueError;
261
262# define PyExc_AttributeError p3imp_PyExc_AttributeError
263# define PyExc_IndexError p3imp_PyExc_IndexError
264# define PyExc_KeyboardInterrupt p3imp_PyExc_KeyboardInterrupt
265# define PyExc_TypeError p3imp_PyExc_TypeError
266# define PyExc_ValueError p3imp_PyExc_ValueError
267
268/*
269 * Table of name to function pointer of python.
270 */
271# define PYTHON_PROC FARPROC
272static struct
273{
274 char *name;
275 PYTHON_PROC *ptr;
276} py3_funcname_table[] =
277{
278 {"PySys_SetArgv", (PYTHON_PROC*)&py3_PySys_SetArgv},
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100279 {"Py_SetPythonHome", (PYTHON_PROC*)&py3_Py_SetPythonHome},
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200280 {"Py_Initialize", (PYTHON_PROC*)&py3_Py_Initialize},
281 {"PyArg_ParseTuple", (PYTHON_PROC*)&py3_PyArg_ParseTuple},
Bram Moolenaar19e60942011-06-19 00:27:51 +0200282 {"PyMem_Free", (PYTHON_PROC*)&py3_PyMem_Free},
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200283 {"PyList_New", (PYTHON_PROC*)&py3_PyList_New},
284 {"PyGILState_Ensure", (PYTHON_PROC*)&py3_PyGILState_Ensure},
285 {"PyGILState_Release", (PYTHON_PROC*)&py3_PyGILState_Release},
286 {"PySys_SetObject", (PYTHON_PROC*)&py3_PySys_SetObject},
287 {"PyList_Append", (PYTHON_PROC*)&py3_PyList_Append},
288 {"PyList_Size", (PYTHON_PROC*)&py3_PyList_Size},
289 {"PySlice_GetIndicesEx", (PYTHON_PROC*)&py3_PySlice_GetIndicesEx},
290 {"PyErr_NoMemory", (PYTHON_PROC*)&py3_PyErr_NoMemory},
291 {"Py_Finalize", (PYTHON_PROC*)&py3_Py_Finalize},
292 {"PyErr_SetString", (PYTHON_PROC*)&py3_PyErr_SetString},
293 {"PyRun_SimpleString", (PYTHON_PROC*)&py3_PyRun_SimpleString},
294 {"PyList_GetItem", (PYTHON_PROC*)&py3_PyList_GetItem},
295 {"PyImport_ImportModule", (PYTHON_PROC*)&py3_PyImport_ImportModule},
296 {"PyErr_BadArgument", (PYTHON_PROC*)&py3_PyErr_BadArgument},
297 {"PyType_Type", (PYTHON_PROC*)&py3_PyType_Type},
298 {"PyErr_Occurred", (PYTHON_PROC*)&py3_PyErr_Occurred},
299 {"PyModule_GetDict", (PYTHON_PROC*)&py3_PyModule_GetDict},
300 {"PyList_SetItem", (PYTHON_PROC*)&py3_PyList_SetItem},
301 {"PyDict_GetItemString", (PYTHON_PROC*)&py3_PyDict_GetItemString},
302 {"PyLong_FromLong", (PYTHON_PROC*)&py3_PyLong_FromLong},
303 {"PyDict_New", (PYTHON_PROC*)&py3_PyDict_New},
304 {"Py_BuildValue", (PYTHON_PROC*)&py3_Py_BuildValue},
305 {"PyType_Ready", (PYTHON_PROC*)&py3_PyType_Ready},
306 {"PyDict_SetItemString", (PYTHON_PROC*)&py3_PyDict_SetItemString},
307 {"PyLong_AsLong", (PYTHON_PROC*)&py3_PyLong_AsLong},
308 {"PyErr_SetNone", (PYTHON_PROC*)&py3_PyErr_SetNone},
309 {"PyEval_InitThreads", (PYTHON_PROC*)&py3_PyEval_InitThreads},
310 {"PyEval_RestoreThread", (PYTHON_PROC*)&py3_PyEval_RestoreThread},
311 {"PyEval_SaveThread", (PYTHON_PROC*)&py3_PyEval_SaveThread},
312 {"PyArg_Parse", (PYTHON_PROC*)&py3_PyArg_Parse},
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200313 {"Py_IsInitialized", (PYTHON_PROC*)&py3_Py_IsInitialized},
314 {"_Py_NoneStruct", (PYTHON_PROC*)&py3__Py_NoneStruct},
315 {"PyErr_Clear", (PYTHON_PROC*)&py3_PyErr_Clear},
316 {"PyObject_Init", (PYTHON_PROC*)&py3__PyObject_Init},
317 {"PyModule_AddObject", (PYTHON_PROC*)&py3_PyModule_AddObject},
318 {"PyImport_AppendInittab", (PYTHON_PROC*)&py3_PyImport_AppendInittab},
319 {"_PyUnicode_AsString", (PYTHON_PROC*)&py3__PyUnicode_AsString},
Bram Moolenaar19e60942011-06-19 00:27:51 +0200320 {"PyBytes_AsString", (PYTHON_PROC*)&py3_PyBytes_AsString},
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200321 {"PyObject_GenericGetAttr", (PYTHON_PROC*)&py3_PyObject_GenericGetAttr},
322 {"PyModule_Create2", (PYTHON_PROC*)&py3_PyModule_Create2},
323 {"PyType_GenericAlloc", (PYTHON_PROC*)&py3_PyType_GenericAlloc},
324 {"PyType_GenericNew", (PYTHON_PROC*)&py3_PyType_GenericNew},
325 {"PySlice_Type", (PYTHON_PROC*)&py3_PySlice_Type},
Bram Moolenaar19e60942011-06-19 00:27:51 +0200326 {"PyErr_NewException", (PYTHON_PROC*)&py3_PyErr_NewException},
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200327# ifdef Py_DEBUG
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200328 {"_Py_NegativeRefcount", (PYTHON_PROC*)&py3__Py_NegativeRefcount},
329 {"_Py_RefTotal", (PYTHON_PROC*)&py3__Py_RefTotal},
330 {"_Py_Dealloc", (PYTHON_PROC*)&py3__Py_Dealloc},
331 {"_PyObject_DebugFree", (PYTHON_PROC*)&py3__PyObject_DebugFree},
332 {"_PyObject_DebugMalloc", (PYTHON_PROC*)&py3__PyObject_DebugMalloc},
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200333# else
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200334 {"PyObject_Malloc", (PYTHON_PROC*)&py3_PyObject_Malloc},
335 {"PyObject_Free", (PYTHON_PROC*)&py3_PyObject_Free},
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200336# endif
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200337 {"", NULL},
338};
339
340/*
341 * Free python.dll
342 */
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200343 static void
344end_dynamic_python3(void)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200345{
Bram Moolenaar4c3a3262010-07-24 15:42:14 +0200346 if (hinstPy3 != 0)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200347 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200348 close_dll(hinstPy3);
349 hinstPy3 = 0;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200350 }
351}
352
353/*
354 * Load library and get all pointers.
355 * Parameter 'libname' provides name of DLL.
356 * Return OK or FAIL.
357 */
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200358 static int
359py3_runtime_link_init(char *libname, int verbose)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200360{
361 int i;
Bram Moolenaar19e60942011-06-19 00:27:51 +0200362 void *ucs_from_string, *ucs_decode, *ucs_as_encoded_string;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200363
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100364# if !(defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)) && defined(UNIX) && defined(FEAT_PYTHON)
Bram Moolenaarb744b2f2010-08-13 16:22:57 +0200365 /* Can't have Python and Python3 loaded at the same time.
366 * It cause a crash, because RTLD_GLOBAL is needed for
367 * standard C extension libraries of one or both python versions. */
Bram Moolenaar4c3a3262010-07-24 15:42:14 +0200368 if (python_loaded())
369 {
Bram Moolenaarb744b2f2010-08-13 16:22:57 +0200370 EMSG(_("E837: This Vim cannot execute :py3 after using :python"));
Bram Moolenaar4c3a3262010-07-24 15:42:14 +0200371 return FAIL;
372 }
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200373# endif
Bram Moolenaar4c3a3262010-07-24 15:42:14 +0200374
375 if (hinstPy3 != 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200376 return OK;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200377 hinstPy3 = load_dll(libname);
378
379 if (!hinstPy3)
380 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200381 if (verbose)
382 EMSG2(_(e_loadlib), libname);
383 return FAIL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200384 }
385
386 for (i = 0; py3_funcname_table[i].ptr; ++i)
387 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200388 if ((*py3_funcname_table[i].ptr = symbol_from_dll(hinstPy3,
389 py3_funcname_table[i].name)) == NULL)
390 {
391 close_dll(hinstPy3);
392 hinstPy3 = 0;
393 if (verbose)
394 EMSG2(_(e_loadfunc), py3_funcname_table[i].name);
395 return FAIL;
396 }
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200397 }
398
Bram Moolenaar69154f22010-07-18 21:42:34 +0200399 /* Load unicode functions separately as only the ucs2 or the ucs4 functions
400 * will be present in the library. */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200401 ucs_from_string = symbol_from_dll(hinstPy3, "PyUnicodeUCS2_FromString");
Bram Moolenaar19e60942011-06-19 00:27:51 +0200402 ucs_decode = symbol_from_dll(hinstPy3,
403 "PyUnicodeUCS2_Decode");
404 ucs_as_encoded_string = symbol_from_dll(hinstPy3,
405 "PyUnicodeUCS2_AsEncodedString");
406 if (!ucs_from_string || !ucs_decode || !ucs_as_encoded_string)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200407 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200408 ucs_from_string = symbol_from_dll(hinstPy3,
409 "PyUnicodeUCS4_FromString");
Bram Moolenaar19e60942011-06-19 00:27:51 +0200410 ucs_decode = symbol_from_dll(hinstPy3,
411 "PyUnicodeUCS4_Decode");
412 ucs_as_encoded_string = symbol_from_dll(hinstPy3,
413 "PyUnicodeUCS4_AsEncodedString");
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200414 }
Bram Moolenaar19e60942011-06-19 00:27:51 +0200415 if (ucs_from_string && ucs_decode && ucs_as_encoded_string)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200416 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200417 py3_PyUnicode_FromString = ucs_from_string;
Bram Moolenaar19e60942011-06-19 00:27:51 +0200418 py3_PyUnicode_Decode = ucs_decode;
419 py3_PyUnicode_AsEncodedString = ucs_as_encoded_string;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200420 }
421 else
422 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200423 close_dll(hinstPy3);
424 hinstPy3 = 0;
425 if (verbose)
426 EMSG2(_(e_loadfunc), "PyUnicode_UCSX_*");
427 return FAIL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200428 }
429
430 return OK;
431}
432
433/*
434 * If python is enabled (there is installed python on Windows system) return
435 * TRUE, else FALSE.
436 */
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200437 int
438python3_enabled(int verbose)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200439{
440 return py3_runtime_link_init(DYNAMIC_PYTHON3_DLL, verbose) == OK;
441}
442
443/* Load the standard Python exceptions - don't import the symbols from the
444 * DLL, as this can cause errors (importing data symbols is not reliable).
445 */
446static void get_py3_exceptions __ARGS((void));
447
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200448 static void
449get_py3_exceptions()
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200450{
451 PyObject *exmod = PyImport_ImportModule("builtins");
452 PyObject *exdict = PyModule_GetDict(exmod);
453 p3imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError");
454 p3imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError");
455 p3imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt");
456 p3imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError");
457 p3imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError");
458 Py_XINCREF(p3imp_PyExc_AttributeError);
459 Py_XINCREF(p3imp_PyExc_IndexError);
460 Py_XINCREF(p3imp_PyExc_KeyboardInterrupt);
461 Py_XINCREF(p3imp_PyExc_TypeError);
462 Py_XINCREF(p3imp_PyExc_ValueError);
463 Py_XDECREF(exmod);
464}
465#endif /* DYNAMIC_PYTHON3 */
466
Bram Moolenaarca8a4df2010-07-31 19:54:14 +0200467static PyObject *BufferNew (buf_T *);
468static PyObject *WindowNew(win_T *);
469static PyObject *LineToString(const char *);
470
471static PyTypeObject RangeType;
472
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200473/*
474 * Include the code shared with if_python.c
475 */
476#include "if_py_both.h"
477
478 static void
479call_PyObject_Free(void *p)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200480{
481#ifdef Py_DEBUG
482 _PyObject_DebugFree(p);
483#else
484 PyObject_Free(p);
485#endif
486}
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200487
488 static PyObject *
489call_PyType_GenericNew(PyTypeObject *type, PyObject *args, PyObject *kwds)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200490{
491 return PyType_GenericNew(type,args,kwds);
492}
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200493
494 static PyObject *
495call_PyType_GenericAlloc(PyTypeObject *type, Py_ssize_t nitems)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200496{
497 return PyType_GenericAlloc(type,nitems);
498}
499
500/******************************************************
501 * Internal function prototypes.
502 */
503
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200504static Py_ssize_t RangeStart;
505static Py_ssize_t RangeEnd;
506
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200507static int PythonIO_Init(void);
508static void PythonIO_Fini(void);
Bram Moolenaar69154f22010-07-18 21:42:34 +0200509PyMODINIT_FUNC Py3Init_vim(void);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200510
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200511/******************************************************
512 * 1. Python interpreter main program.
513 */
514
515static int py3initialised = 0;
516
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200517static PyGILState_STATE pygilstate = PyGILState_UNLOCKED;
518
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200519 void
520python3_end()
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200521{
522 static int recurse = 0;
523
524 /* If a crash occurs while doing this, don't try again. */
525 if (recurse != 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200526 return;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200527
528 ++recurse;
529
530#ifdef DYNAMIC_PYTHON3
531 if (hinstPy3)
532#endif
533 if (Py_IsInitialized())
534 {
535 // acquire lock before finalizing
536 pygilstate = PyGILState_Ensure();
537
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200538 PythonIO_Fini();
539 Py_Finalize();
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200540 }
541
542#ifdef DYNAMIC_PYTHON3
543 end_dynamic_python3();
544#endif
545
546 --recurse;
547}
548
Bram Moolenaar4c3a3262010-07-24 15:42:14 +0200549#if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON)) || defined(PROTO)
550 int
551python3_loaded()
552{
553 return (hinstPy3 != 0);
554}
555#endif
556
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200557 static int
558Python3_Init(void)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200559{
560 if (!py3initialised)
561 {
562#ifdef DYNAMIC_PYTHON3
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200563 if (!python3_enabled(TRUE))
564 {
565 EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded."));
566 goto fail;
567 }
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200568#endif
569
570 init_structs();
571
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100572
573#ifdef PYTHON3_HOME
574 Py_SetPythonHome(PYTHON3_HOME);
575#endif
576
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200577#if !defined(MACOS) || defined(MACOS_X_UNIX)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200578 Py_Initialize();
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200579#else
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200580 PyMac_Initialize();
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200581#endif
Bram Moolenaar456f2bb2011-06-12 21:37:13 +0200582 /* initialise threads, must be after Py_Initialize() */
583 PyEval_InitThreads();
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200584
585#ifdef DYNAMIC_PYTHON3
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200586 get_py3_exceptions();
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200587#endif
588
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200589 if (PythonIO_Init())
590 goto fail;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200591
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200592 PyImport_AppendInittab("vim", Py3Init_vim);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200593
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200594 /* Remove the element from sys.path that was added because of our
595 * argv[0] value in Py3Init_vim(). Previously we used an empty
596 * string, but dependinding on the OS we then get an empty entry or
Bram Moolenaar19e60942011-06-19 00:27:51 +0200597 * the current directory in sys.path.
598 * Only after vim has been imported, the element does exist in
599 * sys.path.
600 */
601 PyRun_SimpleString("import vim; import sys; sys.path = list(filter(lambda x: not x.endswith('must>not&exist'), sys.path))");
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200602
603 // lock is created and acquired in PyEval_InitThreads() and thread
604 // state is created in Py_Initialize()
605 // there _PyGILState_NoteThreadState() also sets gilcounter to 1
606 // (python must have threads enabled!)
607 // so the following does both: unlock GIL and save thread state in TLS
608 // without deleting thread state
609 PyGILState_Release(pygilstate);
610
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200611 py3initialised = 1;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200612 }
613
614 return 0;
615
616fail:
617 /* We call PythonIO_Flush() here to print any Python errors.
618 * This is OK, as it is possible to call this function even
619 * if PythonIO_Init() has not completed successfully (it will
620 * not do anything in this case).
621 */
622 PythonIO_Flush();
623 return -1;
624}
625
626/*
627 * External interface
628 */
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200629 static void
630DoPy3Command(exarg_T *eap, const char *cmd)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200631{
632#if defined(MACOS) && !defined(MACOS_X_UNIX)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200633 GrafPtr oldPort;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200634#endif
635#if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200636 char *saved_locale;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200637#endif
Bram Moolenaar19e60942011-06-19 00:27:51 +0200638 PyObject *cmdstr;
639 PyObject *cmdbytes;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200640
641#if defined(MACOS) && !defined(MACOS_X_UNIX)
642 GetPort(&oldPort);
643 /* Check if the Python library is available */
644 if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200645 goto theend;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200646#endif
647 if (Python3_Init())
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200648 goto theend;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200649
650 RangeStart = eap->line1;
651 RangeEnd = eap->line2;
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200652 Python_Release_Vim(); /* leave vim */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200653
654#if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
655 /* Python only works properly when the LC_NUMERIC locale is "C". */
656 saved_locale = setlocale(LC_NUMERIC, NULL);
657 if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200658 saved_locale = NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200659 else
660 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200661 /* Need to make a copy, value may change when setting new locale. */
662 saved_locale = (char *)vim_strsave((char_u *)saved_locale);
663 (void)setlocale(LC_NUMERIC, "C");
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200664 }
665#endif
666
667 pygilstate = PyGILState_Ensure();
668
Bram Moolenaar19e60942011-06-19 00:27:51 +0200669 /* PyRun_SimpleString expects a UTF-8 string. Wrong encoding may cause
670 * SyntaxError (unicode error). */
Bram Moolenaar3d64a312011-07-15 15:54:44 +0200671 cmdstr = PyUnicode_Decode(cmd, strlen(cmd),
672 (char *)ENC_OPT, CODEC_ERROR_HANDLER);
673 cmdbytes = PyUnicode_AsEncodedString(cmdstr, "utf-8", CODEC_ERROR_HANDLER);
Bram Moolenaar19e60942011-06-19 00:27:51 +0200674 Py_XDECREF(cmdstr);
675 PyRun_SimpleString(PyBytes_AsString(cmdbytes));
676 Py_XDECREF(cmdbytes);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200677
678 PyGILState_Release(pygilstate);
679
680#if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
681 if (saved_locale != NULL)
682 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200683 (void)setlocale(LC_NUMERIC, saved_locale);
684 vim_free(saved_locale);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200685 }
686#endif
687
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200688 Python_Lock_Vim(); /* enter vim */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200689 PythonIO_Flush();
690#if defined(MACOS) && !defined(MACOS_X_UNIX)
691 SetPort(oldPort);
692#endif
693
694theend:
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200695 return; /* keeps lint happy */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200696}
697
698/*
Bram Moolenaar368373e2010-07-19 20:46:22 +0200699 * ":py3"
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200700 */
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200701 void
702ex_py3(exarg_T *eap)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200703{
704 char_u *script;
705
706 script = script_get(eap, eap->arg);
707 if (!eap->skip)
708 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200709 if (script == NULL)
710 DoPy3Command(eap, (char *)eap->arg);
711 else
712 DoPy3Command(eap, (char *)script);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200713 }
714 vim_free(script);
715}
716
717#define BUFFER_SIZE 2048
718
719/*
Bram Moolenaar6df6f472010-07-18 18:04:50 +0200720 * ":py3file"
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200721 */
722 void
723ex_py3file(exarg_T *eap)
724{
725 static char buffer[BUFFER_SIZE];
726 const char *file;
727 char *p;
728 int i;
729
730 /* Have to do it like this. PyRun_SimpleFile requires you to pass a
731 * stdio file pointer, but Vim and the Python DLL are compiled with
732 * different options under Windows, meaning that stdio pointers aren't
733 * compatible between the two. Yuk.
734 *
Bram Moolenaar19e60942011-06-19 00:27:51 +0200735 * construct: exec(compile(open('a_filename', 'rb').read(), 'a_filename', 'exec'))
736 *
737 * Using bytes so that Python can detect the source encoding as it normally
738 * does. The doc does not say "compile" accept bytes, though.
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200739 *
740 * We need to escape any backslashes or single quotes in the file name, so that
741 * Python won't mangle the file name.
742 */
743
744 strcpy(buffer, "exec(compile(open('");
745 p = buffer + 19; /* size of "exec(compile(open('" */
746
747 for (i=0; i<2; ++i)
748 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200749 file = (char *)eap->arg;
750 while (*file && p < buffer + (BUFFER_SIZE - 3))
751 {
752 if (*file == '\\' || *file == '\'')
753 *p++ = '\\';
754 *p++ = *file++;
755 }
756 /* If we didn't finish the file name, we hit a buffer overflow */
757 if (*file != '\0')
758 return;
759 if (i==0)
760 {
Bram Moolenaar19e60942011-06-19 00:27:51 +0200761 strcpy(p,"','rb').read(),'");
762 p += 16;
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200763 }
764 else
765 {
766 strcpy(p,"','exec'))");
767 p += 10;
768 }
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200769 }
770
771
772 /* Execute the file */
773 DoPy3Command(eap, buffer);
774}
775
776/******************************************************
777 * 2. Python output stream: writes output via [e]msg().
778 */
779
780/* Implementation functions
781 */
782
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200783 static PyObject *
784OutputGetattro(PyObject *self, PyObject *nameobj)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200785{
786 char *name = "";
787 if (PyUnicode_Check(nameobj))
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200788 name = _PyUnicode_AsString(nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200789
790 if (strcmp(name, "softspace") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200791 return PyLong_FromLong(((OutputObject *)(self))->softspace);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200792
793 return PyObject_GenericGetAttr(self, nameobj);
794}
795
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200796 static int
797OutputSetattro(PyObject *self, PyObject *nameobj, PyObject *val)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200798{
799 char *name = "";
800 if (PyUnicode_Check(nameobj))
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200801 name = _PyUnicode_AsString(nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200802
803 if (val == NULL) {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200804 PyErr_SetString(PyExc_AttributeError, _("can't delete OutputObject attributes"));
805 return -1;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200806 }
807
808 if (strcmp(name, "softspace") == 0)
809 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200810 if (!PyLong_Check(val)) {
811 PyErr_SetString(PyExc_TypeError, _("softspace must be an integer"));
812 return -1;
813 }
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200814
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200815 ((OutputObject *)(self))->softspace = PyLong_AsLong(val);
816 return 0;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200817 }
818
819 PyErr_SetString(PyExc_AttributeError, _("invalid attribute"));
820 return -1;
821}
822
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200823/***************/
824
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200825 static int
826PythonIO_Init(void)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200827{
828 PyType_Ready(&OutputType);
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200829 return PythonIO_Init_io();
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200830}
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200831
832 static void
833PythonIO_Fini(void)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200834{
835 PySys_SetObject("stdout", NULL);
836 PySys_SetObject("stderr", NULL);
837}
838
839/******************************************************
840 * 3. Implementation of the Vim module for Python
841 */
842
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200843/* Window type - Implementation functions
844 * --------------------------------------
845 */
846
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200847#define WindowType_Check(obj) ((obj)->ob_base.ob_type == &WindowType)
848
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200849/* Buffer type - Implementation functions
850 * --------------------------------------
851 */
852
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200853#define BufferType_Check(obj) ((obj)->ob_base.ob_type == &BufferType)
854
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200855static Py_ssize_t BufferLength(PyObject *);
856static PyObject *BufferItem(PyObject *, Py_ssize_t);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200857static PyObject* BufferSubscript(PyObject *self, PyObject* idx);
Bram Moolenaar19e60942011-06-19 00:27:51 +0200858static Py_ssize_t BufferAsSubscript(PyObject *self, PyObject* idx, PyObject* val);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200859
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200860
861/* Line range type - Implementation functions
862 * --------------------------------------
863 */
864
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200865#define RangeType_Check(obj) ((obj)->ob_base.ob_type == &RangeType)
866
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200867static PyObject* RangeSubscript(PyObject *self, PyObject* idx);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200868static Py_ssize_t RangeAsItem(PyObject *, Py_ssize_t, PyObject *);
869
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200870/* Current objects type - Implementation functions
871 * -----------------------------------------------
872 */
873
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200874static PySequenceMethods BufferAsSeq = {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200875 (lenfunc) BufferLength, /* sq_length, len(x) */
876 (binaryfunc) 0, /* sq_concat, x+y */
877 (ssizeargfunc) 0, /* sq_repeat, x*n */
878 (ssizeargfunc) BufferItem, /* sq_item, x[i] */
879 0, /* was_sq_slice, x[i:j] */
Bram Moolenaar19e60942011-06-19 00:27:51 +0200880 0, /* sq_ass_item, x[i]=v */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200881 0, /* sq_ass_slice, x[i:j]=v */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200882 0, /* sq_contains */
883 0, /* sq_inplace_concat */
884 0, /* sq_inplace_repeat */
885};
886
887PyMappingMethods BufferAsMapping = {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200888 /* mp_length */ (lenfunc)BufferLength,
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200889 /* mp_subscript */ (binaryfunc)BufferSubscript,
Bram Moolenaar19e60942011-06-19 00:27:51 +0200890 /* mp_ass_subscript */ (objobjargproc)BufferAsSubscript,
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200891};
892
893
894/* Buffer object - Definitions
895 */
896
897static PyTypeObject BufferType;
898
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200899 static PyObject *
900BufferNew(buf_T *buf)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200901{
902 /* We need to handle deletion of buffers underneath us.
903 * If we add a "b_python3_ref" field to the buf_T structure,
904 * then we can get at it in buf_freeall() in vim. We then
905 * need to create only ONE Python object per buffer - if
906 * we try to create a second, just INCREF the existing one
907 * and return it. The (single) Python object referring to
908 * the buffer is stored in "b_python3_ref".
909 * Question: what to do on a buf_freeall(). We'll probably
910 * have to either delete the Python object (DECREF it to
911 * zero - a bad idea, as it leaves dangling refs!) or
912 * set the buf_T * value to an invalid value (-1?), which
913 * means we need checks in all access functions... Bah.
914 */
915
916 BufferObject *self;
917
918 if (buf->b_python3_ref != NULL)
919 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200920 self = buf->b_python3_ref;
921 Py_INCREF(self);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200922 }
923 else
924 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200925 self = PyObject_NEW(BufferObject, &BufferType);
926 buf->b_python3_ref = self;
927 if (self == NULL)
928 return NULL;
929 self->buf = buf;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200930 }
931
932 return (PyObject *)(self);
933}
934
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200935 static void
936BufferDestructor(PyObject *self)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200937{
938 BufferObject *this = (BufferObject *)(self);
939
940 if (this->buf && this->buf != INVALID_BUFFER_VALUE)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200941 this->buf->b_python3_ref = NULL;
Bram Moolenaar19e60942011-06-19 00:27:51 +0200942
943 Py_TYPE(self)->tp_free((PyObject*)self);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200944}
945
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200946 static PyObject *
947BufferGetattro(PyObject *self, PyObject*nameobj)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200948{
949 BufferObject *this = (BufferObject *)(self);
950
951 char *name = "";
952 if (PyUnicode_Check(nameobj))
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200953 name = _PyUnicode_AsString(nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200954
955 if (CheckBuffer(this))
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200956 return NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200957
958 if (strcmp(name, "name") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200959 return Py_BuildValue("s", this->buf->b_ffname);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200960 else if (strcmp(name, "number") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200961 return Py_BuildValue("n", this->buf->b_fnum);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200962 else if (strcmp(name,"__members__") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200963 return Py_BuildValue("[ss]", "name", "number");
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200964 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200965 return PyObject_GenericGetAttr(self, nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200966}
967
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200968 static PyObject *
969BufferRepr(PyObject *self)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200970{
971 static char repr[100];
972 BufferObject *this = (BufferObject *)(self);
973
974 if (this->buf == INVALID_BUFFER_VALUE)
975 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200976 vim_snprintf(repr, 100, _("<buffer object (deleted) at %p>"), (self));
977 return PyUnicode_FromString(repr);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200978 }
979 else
980 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200981 char *name = (char *)this->buf->b_fname;
982 Py_ssize_t len;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200983
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200984 if (name == NULL)
985 name = "";
986 len = strlen(name);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200987
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200988 if (len > 35)
989 name = name + (35 - len);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200990
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200991 vim_snprintf(repr, 100, "<buffer %s%s>", len > 35 ? "..." : "", name);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200992
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200993 return PyUnicode_FromString(repr);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200994 }
995}
996
997/******************/
998
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200999 static Py_ssize_t
1000BufferLength(PyObject *self)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001001{
1002 if (CheckBuffer((BufferObject *)(self)))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001003 return -1;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001004
1005 return (Py_ssize_t)(((BufferObject *)(self))->buf->b_ml.ml_line_count);
1006}
1007
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001008 static PyObject *
1009BufferItem(PyObject *self, Py_ssize_t n)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001010{
1011 return RBItem((BufferObject *)(self), n, 1,
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001012 (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001013}
1014
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001015 static PyObject *
1016BufferSlice(PyObject *self, Py_ssize_t lo, Py_ssize_t hi)
1017{
1018 return RBSlice((BufferObject *)(self), lo, hi, 1,
1019 (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count);
1020}
1021
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001022 static PyObject *
1023BufferSubscript(PyObject *self, PyObject* idx)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001024{
1025 if (PyLong_Check(idx)) {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001026 long _idx = PyLong_AsLong(idx);
1027 return BufferItem(self,_idx);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001028 } else if (PySlice_Check(idx)) {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001029 Py_ssize_t start, stop, step, slicelen;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001030
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001031 if (PySlice_GetIndicesEx((PySliceObject *)idx,
1032 (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count+1,
1033 &start, &stop,
1034 &step, &slicelen) < 0) {
1035 return NULL;
1036 }
Bram Moolenaar19e60942011-06-19 00:27:51 +02001037 return BufferSlice(self,start,stop);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001038 } else {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001039 PyErr_SetString(PyExc_IndexError, "Index must be int or slice");
1040 return NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001041 }
1042}
1043
Bram Moolenaar19e60942011-06-19 00:27:51 +02001044 static Py_ssize_t
1045BufferAsSubscript(PyObject *self, PyObject* idx, PyObject* val)
1046{
1047 if (PyLong_Check(idx)) {
1048 long n = PyLong_AsLong(idx);
1049 return RBAsItem((BufferObject *)(self), n, val, 1,
1050 (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1051 NULL);
1052 } else if (PySlice_Check(idx)) {
1053 Py_ssize_t start, stop, step, slicelen;
1054
1055 if (PySlice_GetIndicesEx((PySliceObject *)idx,
1056 (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count+1,
1057 &start, &stop,
1058 &step, &slicelen) < 0) {
1059 return -1;
1060 }
1061 return RBAsSlice((BufferObject *)(self), start, stop, val, 1,
1062 (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1063 NULL);
1064 } else {
1065 PyErr_SetString(PyExc_IndexError, "Index must be int or slice");
1066 return -1;
1067 }
1068}
1069
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001070static PySequenceMethods RangeAsSeq = {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001071 (lenfunc) RangeLength, /* sq_length, len(x) */
1072 (binaryfunc) 0, /* RangeConcat, sq_concat, x+y */
1073 (ssizeargfunc) 0, /* RangeRepeat, sq_repeat, x*n */
1074 (ssizeargfunc) RangeItem, /* sq_item, x[i] */
1075 0, /* was_sq_slice, x[i:j] */
1076 (ssizeobjargproc) RangeAsItem, /* sq_as_item, x[i]=v */
1077 0, /* sq_ass_slice, x[i:j]=v */
1078 0, /* sq_contains */
1079 0, /* sq_inplace_concat */
1080 0, /* sq_inplace_repeat */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001081};
1082
1083PyMappingMethods RangeAsMapping = {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001084 /* mp_length */ (lenfunc)RangeLength,
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001085 /* mp_subscript */ (binaryfunc)RangeSubscript,
1086 /* mp_ass_subscript */ (objobjargproc)0,
1087};
1088
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001089/* Line range object - Implementation
1090 */
1091
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001092 static void
1093RangeDestructor(PyObject *self)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001094{
1095 Py_DECREF(((RangeObject *)(self))->buf);
Bram Moolenaar19e60942011-06-19 00:27:51 +02001096 Py_TYPE(self)->tp_free((PyObject*)self);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001097}
1098
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001099 static PyObject *
1100RangeGetattro(PyObject *self, PyObject *nameobj)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001101{
1102 char *name = "";
1103 if (PyUnicode_Check(nameobj))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001104 name = _PyUnicode_AsString(nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001105
1106 if (strcmp(name, "start") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001107 return Py_BuildValue("n", ((RangeObject *)(self))->start - 1);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001108 else if (strcmp(name, "end") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001109 return Py_BuildValue("n", ((RangeObject *)(self))->end - 1);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001110 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001111 return PyObject_GenericGetAttr(self, nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001112}
1113
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001114/****************/
1115
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001116 static Py_ssize_t
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001117RangeAsItem(PyObject *self, Py_ssize_t n, PyObject *val)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001118{
1119 return RBAsItem(((RangeObject *)(self))->buf, n, val,
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001120 ((RangeObject *)(self))->start,
1121 ((RangeObject *)(self))->end,
1122 &((RangeObject *)(self))->end);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001123}
1124
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001125 static PyObject *
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001126RangeSubscript(PyObject *self, PyObject* idx)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001127{
1128 if (PyLong_Check(idx)) {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001129 long _idx = PyLong_AsLong(idx);
1130 return RangeItem(self,_idx);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001131 } else if (PySlice_Check(idx)) {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001132 Py_ssize_t start, stop, step, slicelen;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001133
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001134 if (PySlice_GetIndicesEx((PySliceObject *)idx,
1135 ((RangeObject *)(self))->end-((RangeObject *)(self))->start+1,
1136 &start, &stop,
1137 &step, &slicelen) < 0) {
1138 return NULL;
1139 }
1140 return RangeSlice(self,start,stop+1);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001141 } else {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001142 PyErr_SetString(PyExc_IndexError, "Index must be int or slice");
1143 return NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001144 }
1145}
1146
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001147/* Buffer list object - Definitions
1148 */
1149
1150typedef struct
1151{
1152 PyObject_HEAD
Bram Moolenaarca8a4df2010-07-31 19:54:14 +02001153} BufListObject;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001154
1155static PySequenceMethods BufListAsSeq = {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001156 (lenfunc) BufListLength, /* sq_length, len(x) */
1157 (binaryfunc) 0, /* sq_concat, x+y */
1158 (ssizeargfunc) 0, /* sq_repeat, x*n */
1159 (ssizeargfunc) BufListItem, /* sq_item, x[i] */
1160 0, /* was_sq_slice, x[i:j] */
1161 (ssizeobjargproc) 0, /* sq_as_item, x[i]=v */
1162 0, /* sq_ass_slice, x[i:j]=v */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001163 0, /* sq_contains */
1164 0, /* sq_inplace_concat */
1165 0, /* sq_inplace_repeat */
1166};
1167
1168static PyTypeObject BufListType;
1169
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001170/* Window object - Definitions
1171 */
1172
1173static struct PyMethodDef WindowMethods[] = {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001174 /* name, function, calling, documentation */
1175 { NULL, NULL, 0, NULL }
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001176};
1177
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001178static PyTypeObject WindowType;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001179
1180/* Window object - Implementation
1181 */
1182
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001183 static PyObject *
1184WindowNew(win_T *win)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001185{
1186 /* We need to handle deletion of windows underneath us.
1187 * If we add a "w_python3_ref" field to the win_T structure,
1188 * then we can get at it in win_free() in vim. We then
1189 * need to create only ONE Python object per window - if
1190 * we try to create a second, just INCREF the existing one
1191 * and return it. The (single) Python object referring to
1192 * the window is stored in "w_python3_ref".
1193 * On a win_free() we set the Python object's win_T* field
1194 * to an invalid value. We trap all uses of a window
1195 * object, and reject them if the win_T* field is invalid.
1196 */
1197
1198 WindowObject *self;
1199
1200 if (win->w_python3_ref)
1201 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001202 self = win->w_python3_ref;
1203 Py_INCREF(self);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001204 }
1205 else
1206 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001207 self = PyObject_NEW(WindowObject, &WindowType);
1208 if (self == NULL)
1209 return NULL;
1210 self->win = win;
1211 win->w_python3_ref = self;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001212 }
1213
1214 return (PyObject *)(self);
1215}
1216
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001217 static void
1218WindowDestructor(PyObject *self)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001219{
1220 WindowObject *this = (WindowObject *)(self);
1221
1222 if (this->win && this->win != INVALID_WINDOW_VALUE)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001223 this->win->w_python3_ref = NULL;
Bram Moolenaar19e60942011-06-19 00:27:51 +02001224
1225 Py_TYPE(self)->tp_free((PyObject*)self);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001226}
1227
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001228 static PyObject *
1229WindowGetattro(PyObject *self, PyObject *nameobj)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001230{
1231 WindowObject *this = (WindowObject *)(self);
1232
1233 char *name = "";
1234 if (PyUnicode_Check(nameobj))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001235 name = _PyUnicode_AsString(nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001236
1237
1238 if (CheckWindow(this))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001239 return NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001240
1241 if (strcmp(name, "buffer") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001242 return (PyObject *)BufferNew(this->win->w_buffer);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001243 else if (strcmp(name, "cursor") == 0)
1244 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001245 pos_T *pos = &this->win->w_cursor;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001246
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001247 return Py_BuildValue("(ll)", (long)(pos->lnum), (long)(pos->col));
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001248 }
1249 else if (strcmp(name, "height") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001250 return Py_BuildValue("l", (long)(this->win->w_height));
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001251#ifdef FEAT_VERTSPLIT
1252 else if (strcmp(name, "width") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001253 return Py_BuildValue("l", (long)(W_WIDTH(this->win)));
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001254#endif
1255 else if (strcmp(name,"__members__") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001256 return Py_BuildValue("[sss]", "buffer", "cursor", "height");
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001257 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001258 return PyObject_GenericGetAttr(self, nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001259}
1260
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001261 static int
1262WindowSetattro(PyObject *self, PyObject *nameobj, PyObject *val)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001263{
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001264 char *name = "";
Bram Moolenaarca8a4df2010-07-31 19:54:14 +02001265
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001266 if (PyUnicode_Check(nameobj))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001267 name = _PyUnicode_AsString(nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001268
Bram Moolenaarca8a4df2010-07-31 19:54:14 +02001269 return WindowSetattr(self, name, val);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001270}
1271
1272/* Window list object - Definitions
1273 */
1274
1275typedef struct
1276{
1277 PyObject_HEAD
1278}
1279WinListObject;
1280
1281static PySequenceMethods WinListAsSeq = {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001282 (lenfunc) WinListLength, /* sq_length, len(x) */
1283 (binaryfunc) 0, /* sq_concat, x+y */
1284 (ssizeargfunc) 0, /* sq_repeat, x*n */
1285 (ssizeargfunc) WinListItem, /* sq_item, x[i] */
1286 0, /* sq_slice, x[i:j] */
1287 (ssizeobjargproc)0, /* sq_as_item, x[i]=v */
1288 0, /* sq_ass_slice, x[i:j]=v */
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001289 0, /* sq_contains */
1290 0, /* sq_inplace_concat */
1291 0, /* sq_inplace_repeat */
1292};
1293
1294static PyTypeObject WinListType;
1295
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001296/* Current items object - Definitions
1297 */
1298
1299typedef struct
1300{
1301 PyObject_HEAD
Bram Moolenaarca8a4df2010-07-31 19:54:14 +02001302} CurrentObject;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001303
1304static PyTypeObject CurrentType;
1305
1306/* Current items object - Implementation
1307 */
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001308 static PyObject *
1309CurrentGetattro(PyObject *self UNUSED, PyObject *nameobj)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001310{
1311 char *name = "";
1312 if (PyUnicode_Check(nameobj))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001313 name = _PyUnicode_AsString(nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001314
1315 if (strcmp(name, "buffer") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001316 return (PyObject *)BufferNew(curbuf);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001317 else if (strcmp(name, "window") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001318 return (PyObject *)WindowNew(curwin);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001319 else if (strcmp(name, "line") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001320 return GetBufferLine(curbuf, (Py_ssize_t)curwin->w_cursor.lnum);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001321 else if (strcmp(name, "range") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001322 return RangeNew(curbuf, RangeStart, RangeEnd);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001323 else if (strcmp(name,"__members__") == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001324 return Py_BuildValue("[ssss]", "buffer", "window", "line", "range");
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001325 else
1326 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001327 PyErr_SetString(PyExc_AttributeError, name);
1328 return NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001329 }
1330}
1331
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001332 static int
1333CurrentSetattro(PyObject *self UNUSED, PyObject *nameobj, PyObject *value)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001334{
1335 char *name = "";
1336 if (PyUnicode_Check(nameobj))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001337 name = _PyUnicode_AsString(nameobj);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001338
1339 if (strcmp(name, "line") == 0)
1340 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001341 if (SetBufferLine(curbuf, (Py_ssize_t)curwin->w_cursor.lnum, value, NULL) == FAIL)
1342 return -1;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001343
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001344 return 0;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001345 }
1346 else
1347 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001348 PyErr_SetString(PyExc_AttributeError, name);
1349 return -1;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001350 }
1351}
1352
1353/* External interface
1354 */
1355
1356 void
1357python3_buffer_free(buf_T *buf)
1358{
1359 if (buf->b_python3_ref != NULL)
1360 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001361 BufferObject *bp = buf->b_python3_ref;
1362 bp->buf = INVALID_BUFFER_VALUE;
1363 buf->b_python3_ref = NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001364 }
1365}
1366
1367#if defined(FEAT_WINDOWS) || defined(PROTO)
1368 void
1369python3_window_free(win_T *win)
1370{
1371 if (win->w_python3_ref != NULL)
1372 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001373 WindowObject *wp = win->w_python3_ref;
1374 wp->win = INVALID_WINDOW_VALUE;
1375 win->w_python3_ref = NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001376 }
1377}
1378#endif
1379
1380static BufListObject TheBufferList =
1381{
1382 PyObject_HEAD_INIT(&BufListType)
1383};
1384
1385static WinListObject TheWindowList =
1386{
1387 PyObject_HEAD_INIT(&WinListType)
1388};
1389
1390static CurrentObject TheCurrent =
1391{
1392 PyObject_HEAD_INIT(&CurrentType)
1393};
1394
1395PyDoc_STRVAR(vim_module_doc,"vim python interface\n");
1396
1397static struct PyModuleDef vimmodule;
1398
Bram Moolenaar69154f22010-07-18 21:42:34 +02001399#ifndef PROTO
1400PyMODINIT_FUNC Py3Init_vim(void)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001401{
1402 PyObject *mod;
1403 /* The special value is removed from sys.path in Python3_Init(). */
1404 static wchar_t *(argv[2]) = {L"/must>not&exist/foo", NULL};
1405
1406 PyType_Ready(&BufferType);
1407 PyType_Ready(&RangeType);
1408 PyType_Ready(&WindowType);
1409 PyType_Ready(&BufListType);
1410 PyType_Ready(&WinListType);
1411 PyType_Ready(&CurrentType);
1412
1413 /* Set sys.argv[] to avoid a crash in warn(). */
1414 PySys_SetArgv(1, argv);
1415
1416 mod = PyModule_Create(&vimmodule);
Bram Moolenaar19e60942011-06-19 00:27:51 +02001417 if (mod == NULL)
1418 return NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001419
Bram Moolenaar19e60942011-06-19 00:27:51 +02001420 VimError = PyErr_NewException("vim.error", NULL, NULL);
1421 Py_INCREF(VimError);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001422
1423 PyModule_AddObject(mod, "error", VimError);
1424 Py_INCREF((PyObject *)(void *)&TheBufferList);
1425 PyModule_AddObject(mod, "buffers", (PyObject *)(void *)&TheBufferList);
1426 Py_INCREF((PyObject *)(void *)&TheCurrent);
1427 PyModule_AddObject(mod, "current", (PyObject *)(void *)&TheCurrent);
1428 Py_INCREF((PyObject *)(void *)&TheWindowList);
1429 PyModule_AddObject(mod, "windows", (PyObject *)(void *)&TheWindowList);
1430
1431 if (PyErr_Occurred())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001432 return NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001433
1434 return mod;
1435}
Bram Moolenaar69154f22010-07-18 21:42:34 +02001436#endif
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001437
1438/*************************************************************************
1439 * 4. Utility functions for handling the interface between Vim and Python.
1440 */
1441
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001442/* Convert a Vim line into a Python string.
1443 * All internal newlines are replaced by null characters.
1444 *
1445 * On errors, the Python exception data is set, and NULL is returned.
1446 */
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001447 static PyObject *
1448LineToString(const char *str)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001449{
1450 PyObject *result;
1451 Py_ssize_t len = strlen(str);
1452 char *tmp,*p;
1453
1454 tmp = (char *)alloc((unsigned)(len+1));
1455 p = tmp;
1456 if (p == NULL)
1457 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001458 PyErr_NoMemory();
1459 return NULL;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001460 }
1461
1462 while (*str)
1463 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001464 if (*str == '\n')
1465 *p = '\0';
1466 else
1467 *p = *str;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001468
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001469 ++p;
1470 ++str;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001471 }
1472 *p = '\0';
1473
Bram Moolenaar3d64a312011-07-15 15:54:44 +02001474 result = PyUnicode_Decode(tmp, len, (char *)ENC_OPT, CODEC_ERROR_HANDLER);
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001475
1476 vim_free(tmp);
1477 return result;
1478}
1479
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001480 static void
1481init_structs(void)
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001482{
1483 vim_memset(&OutputType, 0, sizeof(OutputType));
1484 OutputType.tp_name = "vim.message";
1485 OutputType.tp_basicsize = sizeof(OutputObject);
1486 OutputType.tp_getattro = OutputGetattro;
1487 OutputType.tp_setattro = OutputSetattro;
1488 OutputType.tp_flags = Py_TPFLAGS_DEFAULT;
1489 OutputType.tp_doc = "vim message object";
1490 OutputType.tp_methods = OutputMethods;
1491 OutputType.tp_alloc = call_PyType_GenericAlloc;
1492 OutputType.tp_new = call_PyType_GenericNew;
1493 OutputType.tp_free = call_PyObject_Free;
1494
1495 vim_memset(&BufferType, 0, sizeof(BufferType));
1496 BufferType.tp_name = "vim.buffer";
1497 BufferType.tp_basicsize = sizeof(BufferType);
1498 BufferType.tp_dealloc = BufferDestructor;
1499 BufferType.tp_repr = BufferRepr;
1500 BufferType.tp_as_sequence = &BufferAsSeq;
1501 BufferType.tp_as_mapping = &BufferAsMapping;
1502 BufferType.tp_getattro = BufferGetattro;
1503 BufferType.tp_flags = Py_TPFLAGS_DEFAULT;
1504 BufferType.tp_doc = "vim buffer object";
1505 BufferType.tp_methods = BufferMethods;
1506 BufferType.tp_alloc = call_PyType_GenericAlloc;
1507 BufferType.tp_new = call_PyType_GenericNew;
1508 BufferType.tp_free = call_PyObject_Free;
1509
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001510 vim_memset(&WindowType, 0, sizeof(WindowType));
1511 WindowType.tp_name = "vim.window";
1512 WindowType.tp_basicsize = sizeof(WindowObject);
1513 WindowType.tp_dealloc = WindowDestructor;
1514 WindowType.tp_repr = WindowRepr;
1515 WindowType.tp_getattro = WindowGetattro;
1516 WindowType.tp_setattro = WindowSetattro;
1517 WindowType.tp_flags = Py_TPFLAGS_DEFAULT;
1518 WindowType.tp_doc = "vim Window object";
1519 WindowType.tp_methods = WindowMethods;
1520 WindowType.tp_alloc = call_PyType_GenericAlloc;
1521 WindowType.tp_new = call_PyType_GenericNew;
1522 WindowType.tp_free = call_PyObject_Free;
1523
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +02001524 vim_memset(&BufListType, 0, sizeof(BufListType));
1525 BufListType.tp_name = "vim.bufferlist";
1526 BufListType.tp_basicsize = sizeof(BufListObject);
1527 BufListType.tp_as_sequence = &BufListAsSeq;
1528 BufListType.tp_flags = Py_TPFLAGS_DEFAULT;
1529 BufferType.tp_doc = "vim buffer list";
1530
1531 vim_memset(&WinListType, 0, sizeof(WinListType));
1532 WinListType.tp_name = "vim.windowlist";
1533 WinListType.tp_basicsize = sizeof(WinListType);
1534 WinListType.tp_as_sequence = &WinListAsSeq;
1535 WinListType.tp_flags = Py_TPFLAGS_DEFAULT;
1536 WinListType.tp_doc = "vim window list";
1537
1538 vim_memset(&RangeType, 0, sizeof(RangeType));
1539 RangeType.tp_name = "vim.range";
1540 RangeType.tp_basicsize = sizeof(RangeObject);
1541 RangeType.tp_dealloc = RangeDestructor;
1542 RangeType.tp_repr = RangeRepr;
1543 RangeType.tp_as_sequence = &RangeAsSeq;
1544 RangeType.tp_as_mapping = &RangeAsMapping;
1545 RangeType.tp_getattro = RangeGetattro;
1546 RangeType.tp_flags = Py_TPFLAGS_DEFAULT;
1547 RangeType.tp_doc = "vim Range object";
1548 RangeType.tp_methods = RangeMethods;
1549 RangeType.tp_alloc = call_PyType_GenericAlloc;
1550 RangeType.tp_new = call_PyType_GenericNew;
1551 RangeType.tp_free = call_PyObject_Free;
1552
1553 vim_memset(&CurrentType, 0, sizeof(CurrentType));
1554 CurrentType.tp_name = "vim.currentdata";
1555 CurrentType.tp_basicsize = sizeof(CurrentObject);
1556 CurrentType.tp_getattro = CurrentGetattro;
1557 CurrentType.tp_setattro = CurrentSetattro;
1558 CurrentType.tp_flags = Py_TPFLAGS_DEFAULT;
1559 CurrentType.tp_doc = "vim current object";
1560
1561 vim_memset(&vimmodule, 0, sizeof(vimmodule));
1562 vimmodule.m_name = "vim";
1563 vimmodule.m_doc = vim_module_doc;
1564 vimmodule.m_size = -1;
1565 vimmodule.m_methods = VimMethods;
1566}