blob: 06465bd5a7b2fa86ab621301483936a5c7f911e1 [file] [log] [blame]
Bram Moolenaar071d4272004-06-13 20:20:40 +00001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00004 *
5 * Ruby interface by Shugo Maeda
6 * with improvements by SegPhault (Ryan Paul)
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +02007 * with improvements by Jon Maken
Bram Moolenaar071d4272004-06-13 20:20:40 +00008 *
9 * Do ":help uganda" in Vim to read copying and usage conditions.
10 * Do ":help credits" in Vim to see a list of people who contributed.
11 * See README.txt for an overview of the Vim source code.
12 */
13
Bram Moolenaar2d73ff42010-10-27 17:40:59 +020014#ifdef HAVE_CONFIG_H
15# include "auto/config.h"
16#endif
Bram Moolenaar3ca71f12010-10-27 16:49:47 +020017
Bram Moolenaare2793352011-01-17 19:53:27 +010018#include <stdio.h>
19#include <string.h>
20
Bram Moolenaar071d4272004-06-13 20:20:40 +000021#ifdef _WIN32
22# if !defined(DYNAMIC_RUBY_VER) || (DYNAMIC_RUBY_VER < 18)
23# define NT
24# endif
25# ifndef DYNAMIC_RUBY
26# define IMPORT /* For static dll usage __declspec(dllimport) */
27# define RUBYEXTERN __declspec(dllimport)
28# endif
29#endif
30#ifndef RUBYEXTERN
31# define RUBYEXTERN extern
32#endif
33
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +020034#ifdef DYNAMIC_RUBY
Bram Moolenaar071d4272004-06-13 20:20:40 +000035/*
36 * This is tricky. In ruby.h there is (inline) function rb_class_of()
37 * definition. This function use these variables. But we want function to
38 * use dll_* variables.
39 */
Bram Moolenaar071d4272004-06-13 20:20:40 +000040# define rb_cFalseClass (*dll_rb_cFalseClass)
41# define rb_cFixnum (*dll_rb_cFixnum)
Bram Moolenaardb3fbe52013-03-07 15:16:21 +010042# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20
43# define rb_cFloat (*dll_rb_cFloat)
44# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000045# define rb_cNilClass (*dll_rb_cNilClass)
46# define rb_cSymbol (*dll_rb_cSymbol)
47# define rb_cTrueClass (*dll_rb_cTrueClass)
48# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
49/*
Bram Moolenaar42d57f02010-03-10 12:47:00 +010050 * On ver 1.8, all Ruby functions are exported with "__declspec(dllimport)"
51 * in ruby.h. But it causes trouble for these variables, because it is
Bram Moolenaar071d4272004-06-13 20:20:40 +000052 * defined in this file. When defined this RUBY_EXPORT it modified to
53 * "extern" and be able to avoid this problem.
54 */
55# define RUBY_EXPORT
56# endif
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +020057
Bram Moolenaar2d73ff42010-10-27 17:40:59 +020058#if !(defined(WIN32) || defined(_WIN64))
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +020059# include <dlfcn.h>
Bram Moolenaar3ca71f12010-10-27 16:49:47 +020060# define HINSTANCE void*
61# define RUBY_PROC void*
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +020062# define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL)
63# define symbol_from_dll dlsym
64# define close_dll dlclose
65#else
Bram Moolenaar3ca71f12010-10-27 16:49:47 +020066# define RUBY_PROC FARPROC
Bram Moolenaarebbcb822010-10-23 14:02:54 +020067# define load_dll vimLoadLib
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +020068# define symbol_from_dll GetProcAddress
69# define close_dll FreeLibrary
Bram Moolenaar071d4272004-06-13 20:20:40 +000070#endif
71
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +020072#endif /* ifdef DYNAMIC_RUBY */
73
Bram Moolenaar0b69c732010-02-17 15:11:50 +010074/* suggested by Ariya Mizutani */
75#if (_MSC_VER == 1200)
76# undef _WIN32_WINNT
77#endif
78
Bram Moolenaar639a2552010-03-17 18:15:23 +010079#if (defined(RUBY_VERSION) && RUBY_VERSION >= 19) \
80 || (defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19)
81# define RUBY19_OR_LATER 1
82#endif
83
Bram Moolenaar42d57f02010-03-10 12:47:00 +010084#if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19
85/* Ruby 1.9 defines a number of static functions which use rb_num2long and
86 * rb_int2big */
87# define rb_num2long rb_num2long_stub
88# define rb_int2big rb_int2big_stub
89#endif
90
Bram Moolenaar498af702014-03-28 21:58:21 +010091#if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19 \
92 && VIM_SIZEOF_INT < VIM_SIZEOF_LONG
93/* Ruby 1.9 defines a number of static functions which use rb_fix2int and
Bram Moolenaara2aa31a2014-02-23 22:52:40 +010094 * rb_num2int if VIM_SIZEOF_INT < VIM_SIZEOF_LONG (64bit) */
Bram Moolenaar0bcdd6e2013-04-14 16:19:03 +020095# define rb_fix2int rb_fix2int_stub
96# define rb_num2int rb_num2int_stub
97#endif
98
Bram Moolenaara1a118b2014-02-05 22:41:15 +010099# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 21
100/* Ruby 2.1 adds new GC called RGenGC and RARRAY_PTR uses
101 * rb_gc_writebarrier_unprotect_promoted if USE_RGENGC */
102# define rb_gc_writebarrier_unprotect_promoted rb_gc_writebarrier_unprotect_promoted_stub
103# endif
104
Bram Moolenaar071d4272004-06-13 20:20:40 +0000105#include <ruby.h>
Bram Moolenaar639a2552010-03-17 18:15:23 +0100106#ifdef RUBY19_OR_LATER
Bram Moolenaar165641d2010-02-17 16:23:09 +0100107# include <ruby/encoding.h>
108#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000109
Bram Moolenaar7a8ef142010-12-24 13:39:35 +0100110#undef off_t /* ruby defines off_t as _int64, Mingw uses long */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000111#undef EXTERN
112#undef _
113
114/* T_DATA defined both by Ruby and Mac header files, hack around it... */
Bram Moolenaarcb635362007-05-12 13:02:42 +0000115#if defined(MACOS_X_UNIX) || defined(macintosh)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000116# define __OPENTRANSPORT__
117# define __OPENTRANSPORTPROTOCOL__
118# define __OPENTRANSPORTPROVIDERS__
119#endif
120
Bram Moolenaar165641d2010-02-17 16:23:09 +0100121/*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +0200122 * Backward compatibility for Ruby 1.8 and earlier.
Bram Moolenaar165641d2010-02-17 16:23:09 +0100123 * Ruby 1.9 does not provide STR2CSTR, instead StringValuePtr is provided.
124 * Ruby 1.9 does not provide RXXX(s)->len and RXXX(s)->ptr, instead
125 * RXXX_LEN(s) and RXXX_PTR(s) are provided.
126 */
127#ifndef StringValuePtr
128# define StringValuePtr(s) STR2CSTR(s)
129#endif
130#ifndef RARRAY_LEN
131# define RARRAY_LEN(s) RARRAY(s)->len
132#endif
133#ifndef RARRAY_PTR
134# define RARRAY_PTR(s) RARRAY(s)->ptr
135#endif
136#ifndef RSTRING_LEN
137# define RSTRING_LEN(s) RSTRING(s)->len
138#endif
139#ifndef RSTRING_PTR
140# define RSTRING_PTR(s) RSTRING(s)->ptr
141#endif
142
Bram Moolenaar071d4272004-06-13 20:20:40 +0000143#include "vim.h"
144#include "version.h"
145
146#if defined(PROTO) && !defined(FEAT_RUBY)
147/* Define these to be able to generate the function prototypes. */
148# define VALUE int
149# define RUBY_DATA_FUNC int
150#endif
151
152static int ruby_initialized = 0;
Bram Moolenaar99685e62013-05-11 13:56:18 +0200153static void *ruby_stack_start;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000154static VALUE objtbl;
155
156static VALUE mVIM;
157static VALUE cBuffer;
158static VALUE cVimWindow;
159static VALUE eDeletedBufferError;
160static VALUE eDeletedWindowError;
161
162static int ensure_ruby_initialized(void);
163static void error_print(int);
164static void ruby_io_init(void);
165static void ruby_vim_init(void);
166
Bram Moolenaar10f3a792013-05-20 12:52:29 +0200167#if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK)
168# if defined(__ia64) && !defined(ruby_init_stack)
169# define ruby_init_stack(addr) ruby_init_stack((addr), rb_ia64_bsp())
170# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000171#endif
172
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200173#if defined(DYNAMIC_RUBY) || defined(PROTO)
174# ifdef PROTO
175# define HINSTANCE int /* for generating prototypes */
176# endif
177
Bram Moolenaar071d4272004-06-13 20:20:40 +0000178/*
179 * Wrapper defines
180 */
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200181# define rb_assoc_new dll_rb_assoc_new
182# define rb_cObject (*dll_rb_cObject)
183# define rb_check_type dll_rb_check_type
184# define rb_class_path dll_rb_class_path
185# define rb_data_object_alloc dll_rb_data_object_alloc
186# define rb_define_class_under dll_rb_define_class_under
187# define rb_define_const dll_rb_define_const
188# define rb_define_global_function dll_rb_define_global_function
189# define rb_define_method dll_rb_define_method
190# define rb_define_module dll_rb_define_module
191# define rb_define_module_function dll_rb_define_module_function
192# define rb_define_singleton_method dll_rb_define_singleton_method
193# define rb_define_virtual_variable dll_rb_define_virtual_variable
194# define rb_stdout (*dll_rb_stdout)
195# define rb_eArgError (*dll_rb_eArgError)
196# define rb_eIndexError (*dll_rb_eIndexError)
197# define rb_eRuntimeError (*dll_rb_eRuntimeError)
198# define rb_eStandardError (*dll_rb_eStandardError)
199# define rb_eval_string_protect dll_rb_eval_string_protect
200# define rb_global_variable dll_rb_global_variable
201# define rb_hash_aset dll_rb_hash_aset
202# define rb_hash_new dll_rb_hash_new
203# define rb_inspect dll_rb_inspect
204# define rb_int2inum dll_rb_int2inum
Bram Moolenaara2aa31a2014-02-23 22:52:40 +0100205# if VIM_SIZEOF_INT < VIM_SIZEOF_LONG /* 64 bits only */
Bram Moolenaar498af702014-03-28 21:58:21 +0100206# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER <= 18
207# define rb_fix2int dll_rb_fix2int
208# define rb_num2int dll_rb_num2int
209# endif
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200210# define rb_num2uint dll_rb_num2uint
211# endif
212# define rb_lastline_get dll_rb_lastline_get
213# define rb_lastline_set dll_rb_lastline_set
214# define rb_load_protect dll_rb_load_protect
215# ifndef RUBY19_OR_LATER
216# define rb_num2long dll_rb_num2long
217# endif
218# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER <= 19
219# define rb_num2ulong dll_rb_num2ulong
220# endif
221# define rb_obj_alloc dll_rb_obj_alloc
222# define rb_obj_as_string dll_rb_obj_as_string
223# define rb_obj_id dll_rb_obj_id
224# define rb_raise dll_rb_raise
225# define rb_str_cat dll_rb_str_cat
226# define rb_str_concat dll_rb_str_concat
227# define rb_str_new dll_rb_str_new
228# ifdef rb_str_new2
Bram Moolenaar1d2beae2010-05-22 21:56:55 +0200229/* Ruby may #define rb_str_new2 to use rb_str_new_cstr. */
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200230# define need_rb_str_new_cstr 1
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200231/* Ruby's headers #define rb_str_new_cstr to make use of GCC's
232 * __builtin_constant_p extension. */
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200233# undef rb_str_new_cstr
234# define rb_str_new_cstr dll_rb_str_new_cstr
Bram Moolenaar76a86062013-05-11 17:45:48 +0200235# else
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200236# define rb_str_new2 dll_rb_str_new2
Bram Moolenaar76a86062013-05-11 17:45:48 +0200237# endif
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200238# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200239# define rb_string_value dll_rb_string_value
240# define rb_string_value_ptr dll_rb_string_value_ptr
241# define rb_float_new dll_rb_float_new
242# define rb_ary_new dll_rb_ary_new
243# define rb_ary_push dll_rb_ary_push
Bram Moolenaar10f3a792013-05-20 12:52:29 +0200244# if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK)
245# ifdef __ia64
246# define rb_ia64_bsp dll_rb_ia64_bsp
247# undef ruby_init_stack
248# define ruby_init_stack(addr) dll_ruby_init_stack((addr), rb_ia64_bsp())
249# else
250# define ruby_init_stack dll_ruby_init_stack
251# endif
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200252# endif
253# else
254# define rb_str2cstr dll_rb_str2cstr
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200255# endif
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200256# ifdef RUBY19_OR_LATER
257# define rb_errinfo dll_rb_errinfo
258# else
259# define ruby_errinfo (*dll_ruby_errinfo)
260# endif
261# define ruby_init dll_ruby_init
262# define ruby_init_loadpath dll_ruby_init_loadpath
263# ifdef WIN3264
264# define NtInitialize dll_NtInitialize
265# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
266# define rb_w32_snprintf dll_rb_w32_snprintf
267# endif
268# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000269
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200270# ifdef RUBY19_OR_LATER
271# define ruby_script dll_ruby_script
272# define rb_enc_find_index dll_rb_enc_find_index
273# define rb_enc_find dll_rb_enc_find
274# define rb_enc_str_new dll_rb_enc_str_new
275# define rb_sprintf dll_rb_sprintf
276# define rb_require dll_rb_require
277# define ruby_process_options dll_ruby_process_options
278# endif
Bram Moolenaar165641d2010-02-17 16:23:09 +0100279
Bram Moolenaar071d4272004-06-13 20:20:40 +0000280/*
281 * Pointers for dynamic link
282 */
283static VALUE (*dll_rb_assoc_new) (VALUE, VALUE);
Bram Moolenaarba52cde2010-06-25 04:29:11 +0200284VALUE *dll_rb_cFalseClass;
285VALUE *dll_rb_cFixnum;
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200286# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20
Bram Moolenaardb3fbe52013-03-07 15:16:21 +0100287VALUE *dll_rb_cFloat;
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200288# endif
Bram Moolenaarba52cde2010-06-25 04:29:11 +0200289VALUE *dll_rb_cNilClass;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000290static VALUE *dll_rb_cObject;
Bram Moolenaarba52cde2010-06-25 04:29:11 +0200291VALUE *dll_rb_cSymbol;
292VALUE *dll_rb_cTrueClass;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000293static void (*dll_rb_check_type) (VALUE,int);
294static VALUE (*dll_rb_class_path) (VALUE);
295static VALUE (*dll_rb_data_object_alloc) (VALUE, void*, RUBY_DATA_FUNC, RUBY_DATA_FUNC);
296static VALUE (*dll_rb_define_class_under) (VALUE, const char*, VALUE);
297static void (*dll_rb_define_const) (VALUE,const char*,VALUE);
298static void (*dll_rb_define_global_function) (const char*,VALUE(*)(),int);
299static void (*dll_rb_define_method) (VALUE,const char*,VALUE(*)(),int);
300static VALUE (*dll_rb_define_module) (const char*);
301static void (*dll_rb_define_module_function) (VALUE,const char*,VALUE(*)(),int);
302static void (*dll_rb_define_singleton_method) (VALUE,const char*,VALUE(*)(),int);
303static void (*dll_rb_define_virtual_variable) (const char*,VALUE(*)(),void(*)());
304static VALUE *dll_rb_stdout;
305static VALUE *dll_rb_eArgError;
306static VALUE *dll_rb_eIndexError;
307static VALUE *dll_rb_eRuntimeError;
308static VALUE *dll_rb_eStandardError;
309static VALUE (*dll_rb_eval_string_protect) (const char*, int*);
310static void (*dll_rb_global_variable) (VALUE*);
311static VALUE (*dll_rb_hash_aset) (VALUE, VALUE, VALUE);
312static VALUE (*dll_rb_hash_new) (void);
313static VALUE (*dll_rb_inspect) (VALUE);
314static VALUE (*dll_rb_int2inum) (long);
Bram Moolenaara2aa31a2014-02-23 22:52:40 +0100315# if VIM_SIZEOF_INT < VIM_SIZEOF_LONG /* 64 bits only */
Bram Moolenaar2623b4f2012-09-18 16:36:32 +0200316static long (*dll_rb_fix2int) (VALUE);
317static long (*dll_rb_num2int) (VALUE);
318static unsigned long (*dll_rb_num2uint) (VALUE);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200319# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000320static VALUE (*dll_rb_lastline_get) (void);
321static void (*dll_rb_lastline_set) (VALUE);
322static void (*dll_rb_load_protect) (VALUE, int, int*);
323static long (*dll_rb_num2long) (VALUE);
324static unsigned long (*dll_rb_num2ulong) (VALUE);
325static VALUE (*dll_rb_obj_alloc) (VALUE);
326static VALUE (*dll_rb_obj_as_string) (VALUE);
327static VALUE (*dll_rb_obj_id) (VALUE);
328static void (*dll_rb_raise) (VALUE, const char*, ...);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200329# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +0200330static VALUE (*dll_rb_string_value) (volatile VALUE*);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200331# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000332static char *(*dll_rb_str2cstr) (VALUE,int*);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200333# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000334static VALUE (*dll_rb_str_cat) (VALUE, const char*, long);
335static VALUE (*dll_rb_str_concat) (VALUE, VALUE);
336static VALUE (*dll_rb_str_new) (const char*, long);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200337# ifdef need_rb_str_new_cstr
Bram Moolenaar1d2beae2010-05-22 21:56:55 +0200338/* Ruby may #define rb_str_new2 to use rb_str_new_cstr. */
339static VALUE (*dll_rb_str_new_cstr) (const char*);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200340# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000341static VALUE (*dll_rb_str_new2) (const char*);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200342# endif
343# ifdef RUBY19_OR_LATER
Bram Moolenaar165641d2010-02-17 16:23:09 +0100344static VALUE (*dll_rb_errinfo) (void);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200345# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000346static VALUE *dll_ruby_errinfo;
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200347# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000348static void (*dll_ruby_init) (void);
349static void (*dll_ruby_init_loadpath) (void);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200350# ifdef WIN3264
Bram Moolenaar0b69c732010-02-17 15:11:50 +0100351static void (*dll_NtInitialize) (int*, char***);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200352# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200353static int (*dll_rb_w32_snprintf)(char*, size_t, const char*, ...);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200354# endif
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200355# endif
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200356# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100357static char * (*dll_rb_string_value_ptr) (volatile VALUE*);
358static VALUE (*dll_rb_float_new) (double);
359static VALUE (*dll_rb_ary_new) (void);
360static VALUE (*dll_rb_ary_push) (VALUE, VALUE);
Bram Moolenaar10f3a792013-05-20 12:52:29 +0200361# if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK)
362# ifdef __ia64
Bram Moolenaar76a86062013-05-11 17:45:48 +0200363static void * (*dll_rb_ia64_bsp) (void);
364static void (*dll_ruby_init_stack)(VALUE*, void*);
Bram Moolenaar10f3a792013-05-20 12:52:29 +0200365# else
Bram Moolenaar76a86062013-05-11 17:45:48 +0200366static void (*dll_ruby_init_stack)(VALUE*);
Bram Moolenaar10f3a792013-05-20 12:52:29 +0200367# endif
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200368# endif
Bram Moolenaar76a86062013-05-11 17:45:48 +0200369# endif
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200370# ifdef RUBY19_OR_LATER
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100371static VALUE (*dll_rb_int2big)(SIGNED_VALUE);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200372# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000373
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200374# ifdef RUBY19_OR_LATER
Bram Moolenaar165641d2010-02-17 16:23:09 +0100375static void (*dll_ruby_script) (const char*);
376static int (*dll_rb_enc_find_index) (const char*);
377static rb_encoding* (*dll_rb_enc_find) (const char*);
378static VALUE (*dll_rb_enc_str_new) (const char*, long, rb_encoding*);
379static VALUE (*dll_rb_sprintf) (const char*, ...);
Bram Moolenaar7a8ef142010-12-24 13:39:35 +0100380static VALUE (*dll_rb_require) (const char*);
Bram Moolenaar7a8ef142010-12-24 13:39:35 +0100381static void* (*ruby_process_options)(int, char**);
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200382# endif
Bram Moolenaar165641d2010-02-17 16:23:09 +0100383
Bram Moolenaara1a118b2014-02-05 22:41:15 +0100384# if defined(USE_RGENGC) && USE_RGENGC
385static void (*dll_rb_gc_writebarrier_unprotect_promoted)(VALUE);
386# endif
387
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200388# if defined(RUBY19_OR_LATER) && !defined(PROTO)
Bram Moolenaarff8cf2b2012-11-24 13:39:00 +0100389SIGNED_VALUE rb_num2long_stub(VALUE x)
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100390{
391 return dll_rb_num2long(x);
392}
Bram Moolenaarff8cf2b2012-11-24 13:39:00 +0100393VALUE rb_int2big_stub(SIGNED_VALUE x)
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100394{
395 return dll_rb_int2big(x);
396}
Bram Moolenaar498af702014-03-28 21:58:21 +0100397# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19 \
398 && VIM_SIZEOF_INT < VIM_SIZEOF_LONG
Bram Moolenaar0bcdd6e2013-04-14 16:19:03 +0200399long rb_fix2int_stub(VALUE x)
400{
401 return dll_rb_fix2int(x);
402}
403long rb_num2int_stub(VALUE x)
404{
405 return dll_rb_num2int(x);
406}
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200407# endif
408# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20
Bram Moolenaar886ed692013-02-26 13:41:35 +0100409VALUE
410rb_float_new_in_heap(double d)
411{
412 return dll_rb_float_new(d);
413}
Bram Moolenaardb3fbe52013-03-07 15:16:21 +0100414VALUE rb_num2ulong(VALUE x)
Bram Moolenaar886ed692013-02-26 13:41:35 +0100415{
416 return (long)RSHIFT((SIGNED_VALUE)(x),1);
417}
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200418# endif
419# endif
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100420
Bram Moolenaara1a118b2014-02-05 22:41:15 +0100421# if defined(USE_RGENGC) && USE_RGENGC
422void rb_gc_writebarrier_unprotect_promoted_stub(VALUE obj)
423{
424 return dll_rb_gc_writebarrier_unprotect_promoted(obj);
425}
426# endif
427
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200428static HINSTANCE hinstRuby = NULL; /* Instance of ruby.dll */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000429
430/*
Bram Moolenaarda2303d2005-08-30 21:55:26 +0000431 * Table of name to function pointer of ruby.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000432 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000433static struct
434{
435 char *name;
436 RUBY_PROC *ptr;
437} ruby_funcname_table[] =
438{
439 {"rb_assoc_new", (RUBY_PROC*)&dll_rb_assoc_new},
440 {"rb_cFalseClass", (RUBY_PROC*)&dll_rb_cFalseClass},
441 {"rb_cFixnum", (RUBY_PROC*)&dll_rb_cFixnum},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200442# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20
Bram Moolenaardb3fbe52013-03-07 15:16:21 +0100443 {"rb_cFloat", (RUBY_PROC*)&dll_rb_cFloat},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200444# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000445 {"rb_cNilClass", (RUBY_PROC*)&dll_rb_cNilClass},
446 {"rb_cObject", (RUBY_PROC*)&dll_rb_cObject},
447 {"rb_cSymbol", (RUBY_PROC*)&dll_rb_cSymbol},
448 {"rb_cTrueClass", (RUBY_PROC*)&dll_rb_cTrueClass},
449 {"rb_check_type", (RUBY_PROC*)&dll_rb_check_type},
450 {"rb_class_path", (RUBY_PROC*)&dll_rb_class_path},
451 {"rb_data_object_alloc", (RUBY_PROC*)&dll_rb_data_object_alloc},
452 {"rb_define_class_under", (RUBY_PROC*)&dll_rb_define_class_under},
453 {"rb_define_const", (RUBY_PROC*)&dll_rb_define_const},
454 {"rb_define_global_function", (RUBY_PROC*)&dll_rb_define_global_function},
455 {"rb_define_method", (RUBY_PROC*)&dll_rb_define_method},
456 {"rb_define_module", (RUBY_PROC*)&dll_rb_define_module},
457 {"rb_define_module_function", (RUBY_PROC*)&dll_rb_define_module_function},
458 {"rb_define_singleton_method", (RUBY_PROC*)&dll_rb_define_singleton_method},
459 {"rb_define_virtual_variable", (RUBY_PROC*)&dll_rb_define_virtual_variable},
460 {"rb_stdout", (RUBY_PROC*)&dll_rb_stdout},
461 {"rb_eArgError", (RUBY_PROC*)&dll_rb_eArgError},
462 {"rb_eIndexError", (RUBY_PROC*)&dll_rb_eIndexError},
463 {"rb_eRuntimeError", (RUBY_PROC*)&dll_rb_eRuntimeError},
464 {"rb_eStandardError", (RUBY_PROC*)&dll_rb_eStandardError},
465 {"rb_eval_string_protect", (RUBY_PROC*)&dll_rb_eval_string_protect},
466 {"rb_global_variable", (RUBY_PROC*)&dll_rb_global_variable},
467 {"rb_hash_aset", (RUBY_PROC*)&dll_rb_hash_aset},
468 {"rb_hash_new", (RUBY_PROC*)&dll_rb_hash_new},
469 {"rb_inspect", (RUBY_PROC*)&dll_rb_inspect},
470 {"rb_int2inum", (RUBY_PROC*)&dll_rb_int2inum},
Bram Moolenaara2aa31a2014-02-23 22:52:40 +0100471# if VIM_SIZEOF_INT < VIM_SIZEOF_LONG /* 64 bits only */
Bram Moolenaar2623b4f2012-09-18 16:36:32 +0200472 {"rb_fix2int", (RUBY_PROC*)&dll_rb_fix2int},
473 {"rb_num2int", (RUBY_PROC*)&dll_rb_num2int},
474 {"rb_num2uint", (RUBY_PROC*)&dll_rb_num2uint},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200475# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000476 {"rb_lastline_get", (RUBY_PROC*)&dll_rb_lastline_get},
477 {"rb_lastline_set", (RUBY_PROC*)&dll_rb_lastline_set},
478 {"rb_load_protect", (RUBY_PROC*)&dll_rb_load_protect},
479 {"rb_num2long", (RUBY_PROC*)&dll_rb_num2long},
480 {"rb_num2ulong", (RUBY_PROC*)&dll_rb_num2ulong},
481 {"rb_obj_alloc", (RUBY_PROC*)&dll_rb_obj_alloc},
482 {"rb_obj_as_string", (RUBY_PROC*)&dll_rb_obj_as_string},
483 {"rb_obj_id", (RUBY_PROC*)&dll_rb_obj_id},
484 {"rb_raise", (RUBY_PROC*)&dll_rb_raise},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200485# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +0200486 {"rb_string_value", (RUBY_PROC*)&dll_rb_string_value},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200487# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000488 {"rb_str2cstr", (RUBY_PROC*)&dll_rb_str2cstr},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200489# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000490 {"rb_str_cat", (RUBY_PROC*)&dll_rb_str_cat},
491 {"rb_str_concat", (RUBY_PROC*)&dll_rb_str_concat},
492 {"rb_str_new", (RUBY_PROC*)&dll_rb_str_new},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200493# ifdef need_rb_str_new_cstr
Bram Moolenaar1d2beae2010-05-22 21:56:55 +0200494 {"rb_str_new_cstr", (RUBY_PROC*)&dll_rb_str_new_cstr},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200495# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000496 {"rb_str_new2", (RUBY_PROC*)&dll_rb_str_new2},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200497# endif
498# ifdef RUBY19_OR_LATER
Bram Moolenaar165641d2010-02-17 16:23:09 +0100499 {"rb_errinfo", (RUBY_PROC*)&dll_rb_errinfo},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200500# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000501 {"ruby_errinfo", (RUBY_PROC*)&dll_ruby_errinfo},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200502# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000503 {"ruby_init", (RUBY_PROC*)&dll_ruby_init},
504 {"ruby_init_loadpath", (RUBY_PROC*)&dll_ruby_init_loadpath},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200505# ifdef WIN3264
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100506 {
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200507# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER < 19
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100508 "NtInitialize",
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200509# else
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100510 "ruby_sysinit",
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200511# endif
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100512 (RUBY_PROC*)&dll_NtInitialize},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200513# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
Bram Moolenaar071d4272004-06-13 20:20:40 +0000514 {"rb_w32_snprintf", (RUBY_PROC*)&dll_rb_w32_snprintf},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200515# endif
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200516# endif
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200517# if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100518 {"rb_string_value_ptr", (RUBY_PROC*)&dll_rb_string_value_ptr},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200519# if DYNAMIC_RUBY_VER <= 19
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100520 {"rb_float_new", (RUBY_PROC*)&dll_rb_float_new},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200521# else
Bram Moolenaar886ed692013-02-26 13:41:35 +0100522 {"rb_float_new_in_heap", (RUBY_PROC*)&dll_rb_float_new},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200523# endif
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100524 {"rb_ary_new", (RUBY_PROC*)&dll_rb_ary_new},
525 {"rb_ary_push", (RUBY_PROC*)&dll_rb_ary_push},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200526# endif
527# ifdef RUBY19_OR_LATER
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100528 {"rb_int2big", (RUBY_PROC*)&dll_rb_int2big},
Bram Moolenaar165641d2010-02-17 16:23:09 +0100529 {"ruby_script", (RUBY_PROC*)&dll_ruby_script},
530 {"rb_enc_find_index", (RUBY_PROC*)&dll_rb_enc_find_index},
531 {"rb_enc_find", (RUBY_PROC*)&dll_rb_enc_find},
532 {"rb_enc_str_new", (RUBY_PROC*)&dll_rb_enc_str_new},
533 {"rb_sprintf", (RUBY_PROC*)&dll_rb_sprintf},
Bram Moolenaar7a8ef142010-12-24 13:39:35 +0100534 {"rb_require", (RUBY_PROC*)&dll_rb_require},
Bram Moolenaar7a8ef142010-12-24 13:39:35 +0100535 {"ruby_process_options", (RUBY_PROC*)&dll_ruby_process_options},
Bram Moolenaar3b9abb62013-05-12 14:11:17 +0200536# endif
Bram Moolenaar10f3a792013-05-20 12:52:29 +0200537# if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK)
538# ifdef __ia64
539 {"rb_ia64_bsp", (RUBY_PROC*)&dll_rb_ia64_bsp},
540# endif
541 {"ruby_init_stack", (RUBY_PROC*)&dll_ruby_init_stack},
542# endif
Bram Moolenaara1a118b2014-02-05 22:41:15 +0100543# if defined(USE_RGENGC) && USE_RGENGC
544 {"rb_gc_writebarrier_unprotect_promoted", (RUBY_PROC*)&dll_rb_gc_writebarrier_unprotect_promoted},
545# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000546 {"", NULL},
547};
548
549/*
550 * Free ruby.dll
551 */
552 static void
553end_dynamic_ruby()
554{
555 if (hinstRuby)
556 {
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +0200557 close_dll(hinstRuby);
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200558 hinstRuby = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000559 }
560}
561
562/*
563 * Load library and get all pointers.
564 * Parameter 'libname' provides name of DLL.
565 * Return OK or FAIL.
566 */
567 static int
568ruby_runtime_link_init(char *libname, int verbose)
569{
570 int i;
571
572 if (hinstRuby)
573 return OK;
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +0200574 hinstRuby = load_dll(libname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000575 if (!hinstRuby)
576 {
577 if (verbose)
578 EMSG2(_(e_loadlib), libname);
579 return FAIL;
580 }
581
582 for (i = 0; ruby_funcname_table[i].ptr; ++i)
583 {
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +0200584 if (!(*ruby_funcname_table[i].ptr = symbol_from_dll(hinstRuby,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000585 ruby_funcname_table[i].name)))
586 {
Bram Moolenaarf9b5ef82010-09-29 13:02:53 +0200587 close_dll(hinstRuby);
Bram Moolenaar3ca71f12010-10-27 16:49:47 +0200588 hinstRuby = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000589 if (verbose)
590 EMSG2(_(e_loadfunc), ruby_funcname_table[i].name);
591 return FAIL;
592 }
593 }
594 return OK;
595}
596
597/*
598 * If ruby is enabled (there is installed ruby on Windows system) return TRUE,
599 * else FALSE.
600 */
601 int
602ruby_enabled(verbose)
603 int verbose;
604{
605 return ruby_runtime_link_init(DYNAMIC_RUBY_DLL, verbose) == OK;
606}
607#endif /* defined(DYNAMIC_RUBY) || defined(PROTO) */
608
609 void
610ruby_end()
611{
612#ifdef DYNAMIC_RUBY
613 end_dynamic_ruby();
614#endif
615}
616
617void ex_ruby(exarg_T *eap)
618{
619 int state;
620 char *script = NULL;
621
Bram Moolenaar35a2e192006-03-13 22:07:11 +0000622 script = (char *)script_get(eap, eap->arg);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000623 if (!eap->skip && ensure_ruby_initialized())
624 {
625 if (script == NULL)
626 rb_eval_string_protect((char *)eap->arg, &state);
627 else
628 rb_eval_string_protect(script, &state);
629 if (state)
630 error_print(state);
631 }
632 vim_free(script);
633}
634
Bram Moolenaar165641d2010-02-17 16:23:09 +0100635/*
636 * In Ruby 1.9 or later, ruby String object has encoding.
637 * conversion buffer string of vim to ruby String object using
638 * VIM encoding option.
639 */
640 static VALUE
641vim_str2rb_enc_str(const char *s)
642{
Bram Moolenaar639a2552010-03-17 18:15:23 +0100643#ifdef RUBY19_OR_LATER
Bram Moolenaar165641d2010-02-17 16:23:09 +0100644 int isnum;
645 long lval;
646 char_u *sval;
647 rb_encoding *enc;
648
649 isnum = get_option_value((char_u *)"enc", &lval, &sval, 0);
650 if (isnum == 0)
651 {
652 enc = rb_enc_find((char *)sval);
653 vim_free(sval);
654 if (enc) {
655 return rb_enc_str_new(s, strlen(s), enc);
656 }
657 }
658#endif
659 return rb_str_new2(s);
660}
661
662 static VALUE
663eval_enc_string_protect(const char *str, int *state)
664{
Bram Moolenaar639a2552010-03-17 18:15:23 +0100665#ifdef RUBY19_OR_LATER
Bram Moolenaar165641d2010-02-17 16:23:09 +0100666 int isnum;
667 long lval;
668 char_u *sval;
669 rb_encoding *enc;
670 VALUE v;
671
672 isnum = get_option_value((char_u *)"enc", &lval, &sval, 0);
673 if (isnum == 0)
674 {
675 enc = rb_enc_find((char *)sval);
676 vim_free(sval);
677 if (enc)
678 {
679 v = rb_sprintf("#-*- coding:%s -*-\n%s", rb_enc_name(enc), str);
680 return rb_eval_string_protect(StringValuePtr(v), state);
681 }
682 }
683#endif
684 return rb_eval_string_protect(str, state);
685}
686
Bram Moolenaar071d4272004-06-13 20:20:40 +0000687void ex_rubydo(exarg_T *eap)
688{
689 int state;
690 linenr_T i;
691
692 if (ensure_ruby_initialized())
693 {
694 if (u_save(eap->line1 - 1, eap->line2 + 1) != OK)
695 return;
696 for (i = eap->line1; i <= eap->line2; i++) {
Bram Moolenaare980d8a2010-12-08 13:11:21 +0100697 VALUE line;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000698
Bram Moolenaare980d8a2010-12-08 13:11:21 +0100699 line = vim_str2rb_enc_str((char *)ml_get(i));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000700 rb_lastline_set(line);
Bram Moolenaar165641d2010-02-17 16:23:09 +0100701 eval_enc_string_protect((char *) eap->arg, &state);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000702 if (state) {
703 error_print(state);
704 break;
705 }
706 line = rb_lastline_get();
707 if (!NIL_P(line)) {
708 if (TYPE(line) != T_STRING) {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +0000709 EMSG(_("E265: $_ must be an instance of String"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000710 return;
711 }
Bram Moolenaar165641d2010-02-17 16:23:09 +0100712 ml_replace(i, (char_u *) StringValuePtr(line), 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713 changed();
714#ifdef SYNTAX_HL
715 syn_changed(i); /* recompute syntax hl. for this line */
716#endif
717 }
718 }
719 check_cursor();
720 update_curbuf(NOT_VALID);
721 }
722}
723
724void ex_rubyfile(exarg_T *eap)
725{
726 int state;
727
728 if (ensure_ruby_initialized())
729 {
730 rb_load_protect(rb_str_new2((char *) eap->arg), 0, &state);
731 if (state) error_print(state);
732 }
733}
734
735void ruby_buffer_free(buf_T *buf)
736{
Bram Moolenaare344bea2005-09-01 20:46:49 +0000737 if (buf->b_ruby_ref)
738 {
739 rb_hash_aset(objtbl, rb_obj_id((VALUE) buf->b_ruby_ref), Qnil);
740 RDATA(buf->b_ruby_ref)->data = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741 }
742}
743
744void ruby_window_free(win_T *win)
745{
Bram Moolenaare344bea2005-09-01 20:46:49 +0000746 if (win->w_ruby_ref)
747 {
748 rb_hash_aset(objtbl, rb_obj_id((VALUE) win->w_ruby_ref), Qnil);
749 RDATA(win->w_ruby_ref)->data = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000750 }
751}
752
753static int ensure_ruby_initialized(void)
754{
755 if (!ruby_initialized)
756 {
757#ifdef DYNAMIC_RUBY
758 if (ruby_enabled(TRUE))
759 {
760#endif
Bram Moolenaar0b69c732010-02-17 15:11:50 +0100761#ifdef _WIN32
762 /* suggested by Ariya Mizutani */
763 int argc = 1;
764 char *argv[] = {"gvim.exe"};
765 NtInitialize(&argc, &argv);
766#endif
Bram Moolenaarb2c03502010-07-02 20:20:09 +0200767 {
Bram Moolenaar10f3a792013-05-20 12:52:29 +0200768#if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK)
Bram Moolenaar99685e62013-05-11 13:56:18 +0200769 ruby_init_stack(ruby_stack_start);
Bram Moolenaar165641d2010-02-17 16:23:09 +0100770#endif
Bram Moolenaarb2c03502010-07-02 20:20:09 +0200771 ruby_init();
772 }
Bram Moolenaar639a2552010-03-17 18:15:23 +0100773#ifdef RUBY19_OR_LATER
Bram Moolenaar7a8ef142010-12-24 13:39:35 +0100774 {
775 int dummy_argc = 2;
776 char *dummy_argv[] = {"vim-ruby", "-e0"};
777 ruby_process_options(dummy_argc, dummy_argv);
778 }
Bram Moolenaar165641d2010-02-17 16:23:09 +0100779 ruby_script("vim-ruby");
Bram Moolenaar7a8ef142010-12-24 13:39:35 +0100780#else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000781 ruby_init_loadpath();
Bram Moolenaar165641d2010-02-17 16:23:09 +0100782#endif
Bram Moolenaar7a8ef142010-12-24 13:39:35 +0100783 ruby_io_init();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000784 ruby_vim_init();
785 ruby_initialized = 1;
786#ifdef DYNAMIC_RUBY
787 }
788 else
789 {
790 EMSG(_("E266: Sorry, this command is disabled, the Ruby library could not be loaded."));
791 return 0;
792 }
793#endif
794 }
795 return ruby_initialized;
796}
797
798static void error_print(int state)
799{
800#ifndef DYNAMIC_RUBY
Bram Moolenaar42d57f02010-03-10 12:47:00 +0100801#if !(defined(RUBY_VERSION) && RUBY_VERSION >= 19) \
802 && !(defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000803 RUBYEXTERN VALUE ruby_errinfo;
804#endif
Bram Moolenaar165641d2010-02-17 16:23:09 +0100805#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000806 VALUE eclass;
807 VALUE einfo;
808 char buff[BUFSIZ];
809
810#define TAG_RETURN 0x1
811#define TAG_BREAK 0x2
812#define TAG_NEXT 0x3
813#define TAG_RETRY 0x4
814#define TAG_REDO 0x5
815#define TAG_RAISE 0x6
816#define TAG_THROW 0x7
817#define TAG_FATAL 0x8
818#define TAG_MASK 0xf
819
820 switch (state) {
821 case TAG_RETURN:
Bram Moolenaara93fa7e2006-04-17 22:14:47 +0000822 EMSG(_("E267: unexpected return"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000823 break;
824 case TAG_NEXT:
Bram Moolenaara93fa7e2006-04-17 22:14:47 +0000825 EMSG(_("E268: unexpected next"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000826 break;
827 case TAG_BREAK:
Bram Moolenaara93fa7e2006-04-17 22:14:47 +0000828 EMSG(_("E269: unexpected break"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000829 break;
830 case TAG_REDO:
Bram Moolenaara93fa7e2006-04-17 22:14:47 +0000831 EMSG(_("E270: unexpected redo"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000832 break;
833 case TAG_RETRY:
Bram Moolenaara93fa7e2006-04-17 22:14:47 +0000834 EMSG(_("E271: retry outside of rescue clause"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000835 break;
836 case TAG_RAISE:
837 case TAG_FATAL:
Bram Moolenaar639a2552010-03-17 18:15:23 +0100838#ifdef RUBY19_OR_LATER
Bram Moolenaar165641d2010-02-17 16:23:09 +0100839 eclass = CLASS_OF(rb_errinfo());
840 einfo = rb_obj_as_string(rb_errinfo());
841#else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000842 eclass = CLASS_OF(ruby_errinfo);
843 einfo = rb_obj_as_string(ruby_errinfo);
Bram Moolenaar165641d2010-02-17 16:23:09 +0100844#endif
845 if (eclass == rb_eRuntimeError && RSTRING_LEN(einfo) == 0) {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +0000846 EMSG(_("E272: unhandled exception"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000847 }
848 else {
849 VALUE epath;
850 char *p;
851
852 epath = rb_class_path(eclass);
Bram Moolenaar9c13b352005-05-19 20:53:52 +0000853 vim_snprintf(buff, BUFSIZ, "%s: %s",
Bram Moolenaar165641d2010-02-17 16:23:09 +0100854 RSTRING_PTR(epath), RSTRING_PTR(einfo));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000855 p = strchr(buff, '\n');
856 if (p) *p = '\0';
857 EMSG(buff);
858 }
859 break;
860 default:
Bram Moolenaar9c13b352005-05-19 20:53:52 +0000861 vim_snprintf(buff, BUFSIZ, _("E273: unknown longjmp status %d"), state);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000862 EMSG(buff);
863 break;
864 }
865}
866
Bram Moolenaarcd8b20a2009-05-22 16:20:57 +0000867static VALUE vim_message(VALUE self UNUSED, VALUE str)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000868{
869 char *buff, *p;
870
871 str = rb_obj_as_string(str);
Bram Moolenaar3f5f7952011-08-04 19:34:59 +0200872 if (RSTRING_LEN(str) > 0)
873 {
874 /* Only do this when the string isn't empty, alloc(0) causes trouble. */
875 buff = ALLOCA_N(char, RSTRING_LEN(str));
876 strcpy(buff, RSTRING_PTR(str));
877 p = strchr(buff, '\n');
878 if (p) *p = '\0';
879 MSG(buff);
880 }
881 else
882 {
883 MSG("");
884 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000885 return Qnil;
886}
887
Bram Moolenaarcd8b20a2009-05-22 16:20:57 +0000888static VALUE vim_set_option(VALUE self UNUSED, VALUE str)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000889{
Bram Moolenaar165641d2010-02-17 16:23:09 +0100890 do_set((char_u *)StringValuePtr(str), 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000891 update_screen(NOT_VALID);
892 return Qnil;
893}
894
Bram Moolenaarcd8b20a2009-05-22 16:20:57 +0000895static VALUE vim_command(VALUE self UNUSED, VALUE str)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000896{
Bram Moolenaar165641d2010-02-17 16:23:09 +0100897 do_cmdline_cmd((char_u *)StringValuePtr(str));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000898 return Qnil;
899}
900
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100901#ifdef FEAT_EVAL
902static VALUE vim_to_ruby(typval_T *tv)
903{
904 VALUE result = Qnil;
905
906 if (tv->v_type == VAR_STRING)
907 {
Bram Moolenaar94127e42010-03-19 23:08:48 +0100908 result = rb_str_new2(tv->vval.v_string == NULL
909 ? "" : (char *)(tv->vval.v_string));
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100910 }
911 else if (tv->v_type == VAR_NUMBER)
912 {
Bram Moolenaar639a2552010-03-17 18:15:23 +0100913 result = INT2NUM(tv->vval.v_number);
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100914 }
915# ifdef FEAT_FLOAT
916 else if (tv->v_type == VAR_FLOAT)
917 {
Bram Moolenaar639a2552010-03-17 18:15:23 +0100918 result = rb_float_new(tv->vval.v_float);
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100919 }
920# endif
921 else if (tv->v_type == VAR_LIST)
922 {
Bram Moolenaar639a2552010-03-17 18:15:23 +0100923 list_T *list = tv->vval.v_list;
924 listitem_T *curr;
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100925
Bram Moolenaar639a2552010-03-17 18:15:23 +0100926 result = rb_ary_new();
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100927
Bram Moolenaar639a2552010-03-17 18:15:23 +0100928 if (list != NULL)
929 {
930 for (curr = list->lv_first; curr != NULL; curr = curr->li_next)
931 {
932 rb_ary_push(result, vim_to_ruby(&curr->li_tv));
933 }
934 }
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100935 }
936 else if (tv->v_type == VAR_DICT)
937 {
Bram Moolenaar639a2552010-03-17 18:15:23 +0100938 result = rb_hash_new();
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100939
Bram Moolenaar639a2552010-03-17 18:15:23 +0100940 if (tv->vval.v_dict != NULL)
941 {
942 hashtab_T *ht = &tv->vval.v_dict->dv_hashtab;
943 long_u todo = ht->ht_used;
944 hashitem_T *hi;
945 dictitem_T *di;
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100946
Bram Moolenaar639a2552010-03-17 18:15:23 +0100947 for (hi = ht->ht_array; todo > 0; ++hi)
948 {
949 if (!HASHITEM_EMPTY(hi))
950 {
951 --todo;
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100952
Bram Moolenaar639a2552010-03-17 18:15:23 +0100953 di = dict_lookup(hi);
954 rb_hash_aset(result, rb_str_new2((char *)hi->hi_key),
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100955 vim_to_ruby(&di->di_tv));
Bram Moolenaar639a2552010-03-17 18:15:23 +0100956 }
957 }
958 }
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100959 } /* else return Qnil; */
960
961 return result;
962}
963#endif
964
Bram Moolenaarcd8b20a2009-05-22 16:20:57 +0000965static VALUE vim_evaluate(VALUE self UNUSED, VALUE str)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000966{
967#ifdef FEAT_EVAL
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100968 typval_T *tv;
969 VALUE result;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000970
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100971 tv = eval_expr((char_u *)StringValuePtr(str), NULL);
972 if (tv == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000973 {
Bram Moolenaar639a2552010-03-17 18:15:23 +0100974 return Qnil;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000975 }
Bram Moolenaar3fac56e2010-02-24 15:48:04 +0100976 result = vim_to_ruby(tv);
977
978 free_tv(tv);
979
980 return result;
981#else
982 return Qnil;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000983#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000984}
985
986static VALUE buffer_new(buf_T *buf)
987{
Bram Moolenaare344bea2005-09-01 20:46:49 +0000988 if (buf->b_ruby_ref)
989 {
990 return (VALUE) buf->b_ruby_ref;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000991 }
Bram Moolenaare344bea2005-09-01 20:46:49 +0000992 else
993 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000994 VALUE obj = Data_Wrap_Struct(cBuffer, 0, 0, buf);
Bram Moolenaare344bea2005-09-01 20:46:49 +0000995 buf->b_ruby_ref = (void *) obj;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000996 rb_hash_aset(objtbl, rb_obj_id(obj), obj);
997 return obj;
998 }
999}
1000
1001static buf_T *get_buf(VALUE obj)
1002{
1003 buf_T *buf;
1004
1005 Data_Get_Struct(obj, buf_T, buf);
1006 if (buf == NULL)
1007 rb_raise(eDeletedBufferError, "attempt to refer to deleted buffer");
1008 return buf;
1009}
1010
1011static VALUE buffer_s_current()
1012{
1013 return buffer_new(curbuf);
1014}
1015
1016static VALUE buffer_s_count()
1017{
1018 buf_T *b;
1019 int n = 0;
1020
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001021 for (b = firstbuf; b != NULL; b = b->b_next)
1022 {
Bram Moolenaarc9b4b052006-04-30 18:54:39 +00001023 /* Deleted buffers should not be counted
1024 * SegPhault - 01/07/05 */
1025 if (b->b_p_bl)
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001026 n++;
1027 }
1028
Bram Moolenaar071d4272004-06-13 20:20:40 +00001029 return INT2NUM(n);
1030}
1031
Bram Moolenaarcd8b20a2009-05-22 16:20:57 +00001032static VALUE buffer_s_aref(VALUE self UNUSED, VALUE num)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001033{
1034 buf_T *b;
1035 int n = NUM2INT(num);
1036
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001037 for (b = firstbuf; b != NULL; b = b->b_next)
1038 {
Bram Moolenaarc9b4b052006-04-30 18:54:39 +00001039 /* Deleted buffers should not be counted
1040 * SegPhault - 01/07/05 */
1041 if (!b->b_p_bl)
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001042 continue;
1043
Bram Moolenaarc9b4b052006-04-30 18:54:39 +00001044 if (n == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001045 return buffer_new(b);
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001046
Bram Moolenaarc9b4b052006-04-30 18:54:39 +00001047 n--;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001048 }
1049 return Qnil;
1050}
1051
1052static VALUE buffer_name(VALUE self)
1053{
1054 buf_T *buf = get_buf(self);
1055
Bram Moolenaar35a2e192006-03-13 22:07:11 +00001056 return buf->b_ffname ? rb_str_new2((char *)buf->b_ffname) : Qnil;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001057}
1058
1059static VALUE buffer_number(VALUE self)
1060{
1061 buf_T *buf = get_buf(self);
1062
1063 return INT2NUM(buf->b_fnum);
1064}
1065
1066static VALUE buffer_count(VALUE self)
1067{
1068 buf_T *buf = get_buf(self);
1069
1070 return INT2NUM(buf->b_ml.ml_line_count);
1071}
1072
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001073static VALUE get_buffer_line(buf_T *buf, linenr_T n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001074{
Bram Moolenaar3c531602010-11-16 14:46:19 +01001075 if (n <= 0 || n > buf->b_ml.ml_line_count)
1076 rb_raise(rb_eIndexError, "line number %ld out of range", (long)n);
1077 return vim_str2rb_enc_str((char *)ml_get_buf(buf, n, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001078}
1079
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001080static VALUE buffer_aref(VALUE self, VALUE num)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001081{
1082 buf_T *buf = get_buf(self);
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001083
1084 if (buf != NULL)
1085 return get_buffer_line(buf, (linenr_T)NUM2LONG(num));
1086 return Qnil; /* For stop warning */
1087}
1088
1089static VALUE set_buffer_line(buf_T *buf, linenr_T n, VALUE str)
1090{
Bram Moolenaar165641d2010-02-17 16:23:09 +01001091 char *line = StringValuePtr(str);
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001092 aco_save_T aco;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001093
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001094 if (n > 0 && n <= buf->b_ml.ml_line_count && line != NULL)
1095 {
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001096 /* set curwin/curbuf for "buf" and save some things */
1097 aucmd_prepbuf(&aco, buf);
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001098
Bram Moolenaar071d4272004-06-13 20:20:40 +00001099 if (u_savesub(n) == OK) {
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001100 ml_replace(n, (char_u *)line, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001101 changed();
1102#ifdef SYNTAX_HL
1103 syn_changed(n); /* recompute syntax hl. for this line */
1104#endif
1105 }
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001106
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001107 /* restore curwin/curbuf and a few other things */
1108 aucmd_restbuf(&aco);
1109 /* Careful: autocommands may have made "buf" invalid! */
Bram Moolenaarf30e74c2006-08-16 17:35:00 +00001110
Bram Moolenaar071d4272004-06-13 20:20:40 +00001111 update_curbuf(NOT_VALID);
1112 }
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001113 else
1114 {
Bram Moolenaar165641d2010-02-17 16:23:09 +01001115 rb_raise(rb_eIndexError, "line number %ld out of range", (long)n);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001116 }
1117 return str;
1118}
1119
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001120static VALUE buffer_aset(VALUE self, VALUE num, VALUE str)
1121{
1122 buf_T *buf = get_buf(self);
1123
1124 if (buf != NULL)
1125 return set_buffer_line(buf, (linenr_T)NUM2LONG(num), str);
1126 return str;
1127}
1128
Bram Moolenaar071d4272004-06-13 20:20:40 +00001129static VALUE buffer_delete(VALUE self, VALUE num)
1130{
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001131 buf_T *buf = get_buf(self);
1132 long n = NUM2LONG(num);
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001133 aco_save_T aco;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001134
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001135 if (n > 0 && n <= buf->b_ml.ml_line_count)
1136 {
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001137 /* set curwin/curbuf for "buf" and save some things */
1138 aucmd_prepbuf(&aco, buf);
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001139
Bram Moolenaar071d4272004-06-13 20:20:40 +00001140 if (u_savedel(n, 1) == OK) {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001141 ml_delete(n, 0);
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001142
Bram Moolenaarc9b4b052006-04-30 18:54:39 +00001143 /* Changes to non-active buffers should properly refresh
1144 * SegPhault - 01/09/05 */
1145 deleted_lines_mark(n, 1L);
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001146
Bram Moolenaar071d4272004-06-13 20:20:40 +00001147 changed();
1148 }
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001149
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001150 /* restore curwin/curbuf and a few other things */
1151 aucmd_restbuf(&aco);
1152 /* Careful: autocommands may have made "buf" invalid! */
Bram Moolenaarf30e74c2006-08-16 17:35:00 +00001153
Bram Moolenaar071d4272004-06-13 20:20:40 +00001154 update_curbuf(NOT_VALID);
1155 }
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001156 else
1157 {
Bram Moolenaar165641d2010-02-17 16:23:09 +01001158 rb_raise(rb_eIndexError, "line number %ld out of range", n);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001159 }
1160 return Qnil;
1161}
1162
1163static VALUE buffer_append(VALUE self, VALUE num, VALUE str)
1164{
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001165 buf_T *buf = get_buf(self);
Bram Moolenaar165641d2010-02-17 16:23:09 +01001166 char *line = StringValuePtr(str);
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001167 long n = NUM2LONG(num);
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001168 aco_save_T aco;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001169
Bram Moolenaar3c531602010-11-16 14:46:19 +01001170 if (line == NULL)
1171 {
Bram Moolenaar165641d2010-02-17 16:23:09 +01001172 rb_raise(rb_eIndexError, "NULL line");
1173 }
1174 else if (n >= 0 && n <= buf->b_ml.ml_line_count)
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001175 {
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001176 /* set curwin/curbuf for "buf" and save some things */
1177 aucmd_prepbuf(&aco, buf);
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001178
Bram Moolenaar071d4272004-06-13 20:20:40 +00001179 if (u_inssub(n + 1) == OK) {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001180 ml_append(n, (char_u *) line, (colnr_T) 0, FALSE);
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001181
Bram Moolenaarc9b4b052006-04-30 18:54:39 +00001182 /* Changes to non-active buffers should properly refresh screen
1183 * SegPhault - 12/20/04 */
1184 appended_lines_mark(n, 1L);
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001185
Bram Moolenaarc9b4b052006-04-30 18:54:39 +00001186 changed();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001187 }
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001188
Bram Moolenaar20ff7922006-06-20 19:10:43 +00001189 /* restore curwin/curbuf and a few other things */
1190 aucmd_restbuf(&aco);
1191 /* Careful: autocommands may have made "buf" invalid! */
Bram Moolenaarf30e74c2006-08-16 17:35:00 +00001192
Bram Moolenaar071d4272004-06-13 20:20:40 +00001193 update_curbuf(NOT_VALID);
1194 }
Bram Moolenaar3c531602010-11-16 14:46:19 +01001195 else
1196 {
Bram Moolenaar165641d2010-02-17 16:23:09 +01001197 rb_raise(rb_eIndexError, "line number %ld out of range", n);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001198 }
1199 return str;
1200}
1201
1202static VALUE window_new(win_T *win)
1203{
Bram Moolenaare344bea2005-09-01 20:46:49 +00001204 if (win->w_ruby_ref)
1205 {
1206 return (VALUE) win->w_ruby_ref;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001207 }
Bram Moolenaare344bea2005-09-01 20:46:49 +00001208 else
1209 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001210 VALUE obj = Data_Wrap_Struct(cVimWindow, 0, 0, win);
Bram Moolenaare344bea2005-09-01 20:46:49 +00001211 win->w_ruby_ref = (void *) obj;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001212 rb_hash_aset(objtbl, rb_obj_id(obj), obj);
1213 return obj;
1214 }
1215}
1216
1217static win_T *get_win(VALUE obj)
1218{
1219 win_T *win;
1220
1221 Data_Get_Struct(obj, win_T, win);
1222 if (win == NULL)
1223 rb_raise(eDeletedWindowError, "attempt to refer to deleted window");
1224 return win;
1225}
1226
1227static VALUE window_s_current()
1228{
1229 return window_new(curwin);
1230}
1231
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001232/*
1233 * Added line manipulation functions
1234 * SegPhault - 03/07/05
1235 */
1236static VALUE line_s_current()
1237{
1238 return get_buffer_line(curbuf, curwin->w_cursor.lnum);
1239}
1240
Bram Moolenaarcd8b20a2009-05-22 16:20:57 +00001241static VALUE set_current_line(VALUE self UNUSED, VALUE str)
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001242{
1243 return set_buffer_line(curbuf, curwin->w_cursor.lnum, str);
1244}
1245
1246static VALUE current_line_number()
1247{
1248 return INT2FIX((int)curwin->w_cursor.lnum);
1249}
1250
1251
1252
Bram Moolenaar071d4272004-06-13 20:20:40 +00001253static VALUE window_s_count()
1254{
1255#ifdef FEAT_WINDOWS
1256 win_T *w;
1257 int n = 0;
1258
Bram Moolenaarf740b292006-02-16 22:11:02 +00001259 for (w = firstwin; w != NULL; w = w->w_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001260 n++;
1261 return INT2NUM(n);
1262#else
1263 return INT2NUM(1);
1264#endif
1265}
1266
Bram Moolenaarcd8b20a2009-05-22 16:20:57 +00001267static VALUE window_s_aref(VALUE self UNUSED, VALUE num)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001268{
1269 win_T *w;
1270 int n = NUM2INT(num);
1271
1272#ifndef FEAT_WINDOWS
1273 w = curwin;
1274#else
1275 for (w = firstwin; w != NULL; w = w->w_next, --n)
1276#endif
1277 if (n == 0)
1278 return window_new(w);
1279 return Qnil;
1280}
1281
1282static VALUE window_buffer(VALUE self)
1283{
1284 win_T *win = get_win(self);
1285
1286 return buffer_new(win->w_buffer);
1287}
1288
1289static VALUE window_height(VALUE self)
1290{
1291 win_T *win = get_win(self);
1292
1293 return INT2NUM(win->w_height);
1294}
1295
1296static VALUE window_set_height(VALUE self, VALUE height)
1297{
1298 win_T *win = get_win(self);
1299 win_T *savewin = curwin;
1300
1301 curwin = win;
1302 win_setheight(NUM2INT(height));
1303 curwin = savewin;
1304 return height;
1305}
1306
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001307static VALUE window_width(VALUE self UNUSED)
Bram Moolenaarda2303d2005-08-30 21:55:26 +00001308{
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001309 return INT2NUM(W_WIDTH(get_win(self)));
Bram Moolenaarda2303d2005-08-30 21:55:26 +00001310}
1311
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001312static VALUE window_set_width(VALUE self UNUSED, VALUE width)
Bram Moolenaarda2303d2005-08-30 21:55:26 +00001313{
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001314#ifdef FEAT_VERTSPLIT
Bram Moolenaarda2303d2005-08-30 21:55:26 +00001315 win_T *win = get_win(self);
1316 win_T *savewin = curwin;
1317
1318 curwin = win;
1319 win_setwidth(NUM2INT(width));
1320 curwin = savewin;
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001321#endif
Bram Moolenaarda2303d2005-08-30 21:55:26 +00001322 return width;
1323}
1324
Bram Moolenaar071d4272004-06-13 20:20:40 +00001325static VALUE window_cursor(VALUE self)
1326{
1327 win_T *win = get_win(self);
1328
1329 return rb_assoc_new(INT2NUM(win->w_cursor.lnum), INT2NUM(win->w_cursor.col));
1330}
1331
1332static VALUE window_set_cursor(VALUE self, VALUE pos)
1333{
1334 VALUE lnum, col;
1335 win_T *win = get_win(self);
1336
1337 Check_Type(pos, T_ARRAY);
Bram Moolenaar165641d2010-02-17 16:23:09 +01001338 if (RARRAY_LEN(pos) != 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001339 rb_raise(rb_eArgError, "array length must be 2");
Bram Moolenaar165641d2010-02-17 16:23:09 +01001340 lnum = RARRAY_PTR(pos)[0];
1341 col = RARRAY_PTR(pos)[1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001342 win->w_cursor.lnum = NUM2LONG(lnum);
1343 win->w_cursor.col = NUM2UINT(col);
1344 check_cursor(); /* put cursor on an existing line */
1345 update_screen(NOT_VALID);
1346 return Qnil;
1347}
1348
Bram Moolenaar6217cdc2012-04-25 12:28:09 +02001349static VALUE f_nop(VALUE self UNUSED)
Bram Moolenaar35df7d22012-04-20 18:05:47 +02001350{
1351 return Qnil;
1352}
1353
Bram Moolenaarcd8b20a2009-05-22 16:20:57 +00001354static VALUE f_p(int argc, VALUE *argv, VALUE self UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001355{
1356 int i;
1357 VALUE str = rb_str_new("", 0);
1358
1359 for (i = 0; i < argc; i++) {
1360 if (i > 0) rb_str_cat(str, ", ", 2);
1361 rb_str_concat(str, rb_inspect(argv[i]));
1362 }
Bram Moolenaar165641d2010-02-17 16:23:09 +01001363 MSG(RSTRING_PTR(str));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001364 return Qnil;
1365}
1366
1367static void ruby_io_init(void)
1368{
1369#ifndef DYNAMIC_RUBY
1370 RUBYEXTERN VALUE rb_stdout;
1371#endif
1372
1373 rb_stdout = rb_obj_alloc(rb_cObject);
1374 rb_define_singleton_method(rb_stdout, "write", vim_message, 1);
Bram Moolenaar35df7d22012-04-20 18:05:47 +02001375 rb_define_singleton_method(rb_stdout, "flush", f_nop, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001376 rb_define_global_function("p", f_p, -1);
1377}
1378
1379static void ruby_vim_init(void)
1380{
1381 objtbl = rb_hash_new();
1382 rb_global_variable(&objtbl);
1383
Bram Moolenaarf711faf2007-05-10 16:48:19 +00001384 /* The Vim module used to be called "VIM", but "Vim" is better. Make an
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001385 * alias "VIM" for backwards compatibility. */
Bram Moolenaarf711faf2007-05-10 16:48:19 +00001386 mVIM = rb_define_module("Vim");
1387 rb_define_const(rb_cObject, "VIM", mVIM);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001388 rb_define_const(mVIM, "VERSION_MAJOR", INT2NUM(VIM_VERSION_MAJOR));
1389 rb_define_const(mVIM, "VERSION_MINOR", INT2NUM(VIM_VERSION_MINOR));
1390 rb_define_const(mVIM, "VERSION_BUILD", INT2NUM(VIM_VERSION_BUILD));
1391 rb_define_const(mVIM, "VERSION_PATCHLEVEL", INT2NUM(VIM_VERSION_PATCHLEVEL));
1392 rb_define_const(mVIM, "VERSION_SHORT", rb_str_new2(VIM_VERSION_SHORT));
1393 rb_define_const(mVIM, "VERSION_MEDIUM", rb_str_new2(VIM_VERSION_MEDIUM));
1394 rb_define_const(mVIM, "VERSION_LONG", rb_str_new2(VIM_VERSION_LONG));
1395 rb_define_const(mVIM, "VERSION_LONG_DATE", rb_str_new2(VIM_VERSION_LONG_DATE));
1396 rb_define_module_function(mVIM, "message", vim_message, 1);
1397 rb_define_module_function(mVIM, "set_option", vim_set_option, 1);
1398 rb_define_module_function(mVIM, "command", vim_command, 1);
1399 rb_define_module_function(mVIM, "evaluate", vim_evaluate, 1);
1400
1401 eDeletedBufferError = rb_define_class_under(mVIM, "DeletedBufferError",
1402 rb_eStandardError);
1403 eDeletedWindowError = rb_define_class_under(mVIM, "DeletedWindowError",
1404 rb_eStandardError);
1405
1406 cBuffer = rb_define_class_under(mVIM, "Buffer", rb_cObject);
1407 rb_define_singleton_method(cBuffer, "current", buffer_s_current, 0);
1408 rb_define_singleton_method(cBuffer, "count", buffer_s_count, 0);
1409 rb_define_singleton_method(cBuffer, "[]", buffer_s_aref, 1);
1410 rb_define_method(cBuffer, "name", buffer_name, 0);
1411 rb_define_method(cBuffer, "number", buffer_number, 0);
1412 rb_define_method(cBuffer, "count", buffer_count, 0);
1413 rb_define_method(cBuffer, "length", buffer_count, 0);
1414 rb_define_method(cBuffer, "[]", buffer_aref, 1);
1415 rb_define_method(cBuffer, "[]=", buffer_aset, 2);
1416 rb_define_method(cBuffer, "delete", buffer_delete, 1);
1417 rb_define_method(cBuffer, "append", buffer_append, 2);
1418
Bram Moolenaarbe4d5062006-03-18 21:30:13 +00001419 /* Added line manipulation functions
1420 * SegPhault - 03/07/05 */
1421 rb_define_method(cBuffer, "line_number", current_line_number, 0);
1422 rb_define_method(cBuffer, "line", line_s_current, 0);
1423 rb_define_method(cBuffer, "line=", set_current_line, 1);
1424
1425
Bram Moolenaar071d4272004-06-13 20:20:40 +00001426 cVimWindow = rb_define_class_under(mVIM, "Window", rb_cObject);
1427 rb_define_singleton_method(cVimWindow, "current", window_s_current, 0);
1428 rb_define_singleton_method(cVimWindow, "count", window_s_count, 0);
1429 rb_define_singleton_method(cVimWindow, "[]", window_s_aref, 1);
1430 rb_define_method(cVimWindow, "buffer", window_buffer, 0);
1431 rb_define_method(cVimWindow, "height", window_height, 0);
1432 rb_define_method(cVimWindow, "height=", window_set_height, 1);
Bram Moolenaarda2303d2005-08-30 21:55:26 +00001433 rb_define_method(cVimWindow, "width", window_width, 0);
1434 rb_define_method(cVimWindow, "width=", window_set_width, 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001435 rb_define_method(cVimWindow, "cursor", window_cursor, 0);
1436 rb_define_method(cVimWindow, "cursor=", window_set_cursor, 1);
1437
1438 rb_define_virtual_variable("$curbuf", buffer_s_current, 0);
1439 rb_define_virtual_variable("$curwin", window_s_current, 0);
1440}
Bram Moolenaar99685e62013-05-11 13:56:18 +02001441
1442void vim_ruby_init(void *stack_start)
1443{
1444 /* should get machine stack start address early in main function */
1445 ruby_stack_start = stack_start;
1446}