1 /* vi:set ts=8 sts=4 sw=4 noet: 2 * 3 * VIM - Vi IMproved by Bram Moolenaar 4 * 5 * Ruby interface by Shugo Maeda 6 * with improvements by SegPhault (Ryan Paul) 7 * with improvements by Jon Maken 8 * 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 14 #ifdef HAVE_CONFIG_H 15 # include "auto/config.h" 16 #endif 17 18 #include <stdio.h> 19 #include <string.h> 20 21 #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 34 #if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 24 35 # define USE_RUBY_INTEGER 36 #endif 37 38 #ifdef DYNAMIC_RUBY 39 /* 40 * This is tricky. In ruby.h there is (inline) function rb_class_of() 41 * definition. This function use these variables. But we want function to 42 * use dll_* variables. 43 */ 44 # define rb_cFalseClass (*dll_rb_cFalseClass) 45 # define rb_cFixnum (*dll_rb_cFixnum) 46 # if defined(USE_RUBY_INTEGER) 47 # define rb_cInteger (*dll_rb_cInteger) 48 # endif 49 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20 50 # define rb_cFloat (*dll_rb_cFloat) 51 # endif 52 # define rb_cNilClass (*dll_rb_cNilClass) 53 # define rb_cSymbol (*dll_rb_cSymbol) 54 # define rb_cTrueClass (*dll_rb_cTrueClass) 55 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 56 /* 57 * On ver 1.8, all Ruby functions are exported with "__declspec(dllimport)" 58 * in ruby.h. But it causes trouble for these variables, because it is 59 * defined in this file. When defined this RUBY_EXPORT it modified to 60 * "extern" and be able to avoid this problem. 61 */ 62 # define RUBY_EXPORT 63 # endif 64 65 #if !(defined(WIN32) || defined(_WIN64)) 66 # include <dlfcn.h> 67 # define HINSTANCE void* 68 # define RUBY_PROC void* 69 # define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL) 70 # define symbol_from_dll dlsym 71 # define close_dll dlclose 72 #else 73 # define RUBY_PROC FARPROC 74 # define load_dll vimLoadLib 75 # define symbol_from_dll GetProcAddress 76 # define close_dll FreeLibrary 77 #endif 78 79 #endif /* ifdef DYNAMIC_RUBY */ 80 81 /* suggested by Ariya Mizutani */ 82 #if (_MSC_VER == 1200) 83 # undef _WIN32_WINNT 84 #endif 85 86 #if (defined(RUBY_VERSION) && RUBY_VERSION >= 19) \ 87 || (defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19) 88 # define RUBY19_OR_LATER 1 89 #endif 90 91 #if (defined(RUBY_VERSION) && RUBY_VERSION >= 20) \ 92 || (defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20) 93 # define RUBY20_OR_LATER 1 94 #endif 95 96 #if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19 97 /* Ruby 1.9 defines a number of static functions which use rb_num2long and 98 * rb_int2big */ 99 # define rb_num2long rb_num2long_stub 100 # define rb_int2big rb_int2big_stub 101 #endif 102 103 #if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19 \ 104 && VIM_SIZEOF_INT < VIM_SIZEOF_LONG 105 /* Ruby 1.9 defines a number of static functions which use rb_fix2int and 106 * rb_num2int if VIM_SIZEOF_INT < VIM_SIZEOF_LONG (64bit) */ 107 # define rb_fix2int rb_fix2int_stub 108 # define rb_num2int rb_num2int_stub 109 #endif 110 111 #if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER == 21 112 /* Ruby 2.1 adds new GC called RGenGC and RARRAY_PTR uses 113 * rb_gc_writebarrier_unprotect_promoted if USE_RGENGC */ 114 # define rb_gc_writebarrier_unprotect_promoted rb_gc_writebarrier_unprotect_promoted_stub 115 #endif 116 #if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 22 117 # define rb_gc_writebarrier_unprotect rb_gc_writebarrier_unprotect_stub 118 #endif 119 120 #include <ruby.h> 121 #ifdef RUBY19_OR_LATER 122 # include <ruby/encoding.h> 123 #endif 124 125 #undef off_t /* ruby defines off_t as _int64, Mingw uses long */ 126 #undef EXTERN 127 #undef _ 128 129 /* T_DATA defined both by Ruby and Mac header files, hack around it... */ 130 #if defined(MACOS_X) 131 # define __OPENTRANSPORT__ 132 # define __OPENTRANSPORTPROTOCOL__ 133 # define __OPENTRANSPORTPROVIDERS__ 134 #endif 135 136 /* 137 * The TypedData_XXX macro family can be used since Ruby 1.9.2 but 138 * rb_data_type_t changed in 1.9.3, therefore require at least 2.0. 139 * The old Data_XXX macro family was deprecated on Ruby 2.2. 140 * Use TypedData_XXX if available. 141 */ 142 #if defined(TypedData_Wrap_Struct) && defined(RUBY20_OR_LATER) 143 # define USE_TYPEDDATA 1 144 #endif 145 146 /* 147 * Backward compatibility for Ruby 1.8 and earlier. 148 * Ruby 1.9 does not provide STR2CSTR, instead StringValuePtr is provided. 149 * Ruby 1.9 does not provide RXXX(s)->len and RXXX(s)->ptr, instead 150 * RXXX_LEN(s) and RXXX_PTR(s) are provided. 151 */ 152 #ifndef StringValuePtr 153 # define StringValuePtr(s) STR2CSTR(s) 154 #endif 155 #ifndef RARRAY_LEN 156 # define RARRAY_LEN(s) RARRAY(s)->len 157 #endif 158 #ifndef RARRAY_PTR 159 # define RARRAY_PTR(s) RARRAY(s)->ptr 160 #endif 161 #ifndef RSTRING_LEN 162 # define RSTRING_LEN(s) RSTRING(s)->len 163 #endif 164 #ifndef RSTRING_PTR 165 # define RSTRING_PTR(s) RSTRING(s)->ptr 166 #endif 167 168 #ifdef HAVE_DUP 169 # undef HAVE_DUP 170 #endif 171 172 #include "vim.h" 173 #include "version.h" 174 175 #if defined(PROTO) && !defined(FEAT_RUBY) 176 /* Define these to be able to generate the function prototypes. */ 177 # define VALUE int 178 # define RUBY_DATA_FUNC int 179 #endif 180 181 static int ruby_initialized = 0; 182 static void *ruby_stack_start; 183 static VALUE objtbl; 184 185 static VALUE mVIM; 186 static VALUE cBuffer; 187 static VALUE cVimWindow; 188 static VALUE eDeletedBufferError; 189 static VALUE eDeletedWindowError; 190 191 static int ensure_ruby_initialized(void); 192 static void error_print(int); 193 static void ruby_io_init(void); 194 static void ruby_vim_init(void); 195 196 #if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK) 197 # if defined(__ia64) && !defined(ruby_init_stack) 198 # define ruby_init_stack(addr) ruby_init_stack((addr), rb_ia64_bsp()) 199 # endif 200 #endif 201 202 #if defined(DYNAMIC_RUBY) || defined(PROTO) 203 # if defined(PROTO) && !defined(HINSTANCE) 204 # define HINSTANCE int /* for generating prototypes */ 205 # endif 206 207 /* 208 * Wrapper defines 209 */ 210 # define rb_assoc_new dll_rb_assoc_new 211 # define rb_cObject (*dll_rb_cObject) 212 # define rb_check_type dll_rb_check_type 213 # ifdef USE_TYPEDDATA 214 # define rb_check_typeddata dll_rb_check_typeddata 215 # endif 216 # define rb_class_path dll_rb_class_path 217 # ifdef USE_TYPEDDATA 218 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 23 219 # define rb_data_typed_object_wrap dll_rb_data_typed_object_wrap 220 # else 221 # define rb_data_typed_object_alloc dll_rb_data_typed_object_alloc 222 # endif 223 # else 224 # define rb_data_object_alloc dll_rb_data_object_alloc 225 # endif 226 # define rb_define_class_under dll_rb_define_class_under 227 # define rb_define_const dll_rb_define_const 228 # define rb_define_global_function dll_rb_define_global_function 229 # define rb_define_method dll_rb_define_method 230 # define rb_define_module dll_rb_define_module 231 # define rb_define_module_function dll_rb_define_module_function 232 # define rb_define_singleton_method dll_rb_define_singleton_method 233 # define rb_define_virtual_variable dll_rb_define_virtual_variable 234 # define rb_stdout (*dll_rb_stdout) 235 # define rb_eArgError (*dll_rb_eArgError) 236 # define rb_eIndexError (*dll_rb_eIndexError) 237 # define rb_eRuntimeError (*dll_rb_eRuntimeError) 238 # define rb_eStandardError (*dll_rb_eStandardError) 239 # define rb_eval_string_protect dll_rb_eval_string_protect 240 # define rb_global_variable dll_rb_global_variable 241 # define rb_hash_aset dll_rb_hash_aset 242 # define rb_hash_new dll_rb_hash_new 243 # define rb_inspect dll_rb_inspect 244 # define rb_int2inum dll_rb_int2inum 245 # if VIM_SIZEOF_INT < VIM_SIZEOF_LONG /* 64 bits only */ 246 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER <= 18 247 # define rb_fix2int dll_rb_fix2int 248 # define rb_num2int dll_rb_num2int 249 # endif 250 # define rb_num2uint dll_rb_num2uint 251 # endif 252 # define rb_lastline_get dll_rb_lastline_get 253 # define rb_lastline_set dll_rb_lastline_set 254 # define rb_protect dll_rb_protect 255 # define rb_load dll_rb_load 256 # ifndef RUBY19_OR_LATER 257 # define rb_num2long dll_rb_num2long 258 # endif 259 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER <= 19 260 # define rb_num2ulong dll_rb_num2ulong 261 # endif 262 # define rb_obj_alloc dll_rb_obj_alloc 263 # define rb_obj_as_string dll_rb_obj_as_string 264 # define rb_obj_id dll_rb_obj_id 265 # define rb_raise dll_rb_raise 266 # define rb_str_cat dll_rb_str_cat 267 # define rb_str_concat dll_rb_str_concat 268 # undef rb_str_new 269 # define rb_str_new dll_rb_str_new 270 # ifdef rb_str_new2 271 /* Ruby may #define rb_str_new2 to use rb_str_new_cstr. */ 272 # define need_rb_str_new_cstr 1 273 /* Ruby's headers #define rb_str_new_cstr to make use of GCC's 274 * __builtin_constant_p extension. */ 275 # undef rb_str_new_cstr 276 # define rb_str_new_cstr dll_rb_str_new_cstr 277 # else 278 # define rb_str_new2 dll_rb_str_new2 279 # endif 280 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 281 # define rb_string_value dll_rb_string_value 282 # define rb_string_value_ptr dll_rb_string_value_ptr 283 # define rb_float_new dll_rb_float_new 284 # define rb_ary_new dll_rb_ary_new 285 # define rb_ary_push dll_rb_ary_push 286 # if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK) 287 # ifdef __ia64 288 # define rb_ia64_bsp dll_rb_ia64_bsp 289 # undef ruby_init_stack 290 # define ruby_init_stack(addr) dll_ruby_init_stack((addr), rb_ia64_bsp()) 291 # else 292 # define ruby_init_stack dll_ruby_init_stack 293 # endif 294 # endif 295 # else 296 # define rb_str2cstr dll_rb_str2cstr 297 # endif 298 # ifdef RUBY19_OR_LATER 299 # define rb_errinfo dll_rb_errinfo 300 # else 301 # define ruby_errinfo (*dll_ruby_errinfo) 302 # endif 303 # define ruby_init dll_ruby_init 304 # define ruby_init_loadpath dll_ruby_init_loadpath 305 # ifdef WIN3264 306 # define NtInitialize dll_NtInitialize 307 # define ruby_sysinit dll_ruby_sysinit 308 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 309 # define rb_w32_snprintf dll_rb_w32_snprintf 310 # endif 311 # endif 312 313 # ifdef RUBY19_OR_LATER 314 # define ruby_script dll_ruby_script 315 # define rb_enc_find_index dll_rb_enc_find_index 316 # define rb_enc_find dll_rb_enc_find 317 # undef rb_enc_str_new 318 # define rb_enc_str_new dll_rb_enc_str_new 319 # define rb_sprintf dll_rb_sprintf 320 # define rb_require dll_rb_require 321 # define ruby_options dll_ruby_options 322 # endif 323 324 /* 325 * Pointers for dynamic link 326 */ 327 static VALUE (*dll_rb_assoc_new) (VALUE, VALUE); 328 VALUE *dll_rb_cFalseClass; 329 VALUE *dll_rb_cFixnum; 330 # if defined(USE_RUBY_INTEGER) 331 VALUE *dll_rb_cInteger; 332 # endif 333 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20 334 VALUE *dll_rb_cFloat; 335 # endif 336 VALUE *dll_rb_cNilClass; 337 static VALUE *dll_rb_cObject; 338 VALUE *dll_rb_cSymbol; 339 VALUE *dll_rb_cTrueClass; 340 static void (*dll_rb_check_type) (VALUE,int); 341 # ifdef USE_TYPEDDATA 342 static void *(*dll_rb_check_typeddata) (VALUE,const rb_data_type_t *); 343 # endif 344 static VALUE (*dll_rb_class_path) (VALUE); 345 # ifdef USE_TYPEDDATA 346 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 23 347 static VALUE (*dll_rb_data_typed_object_wrap) (VALUE, void*, const rb_data_type_t *); 348 # else 349 static VALUE (*dll_rb_data_typed_object_alloc) (VALUE, void*, const rb_data_type_t *); 350 # endif 351 # else 352 static VALUE (*dll_rb_data_object_alloc) (VALUE, void*, RUBY_DATA_FUNC, RUBY_DATA_FUNC); 353 # endif 354 static VALUE (*dll_rb_define_class_under) (VALUE, const char*, VALUE); 355 static void (*dll_rb_define_const) (VALUE,const char*,VALUE); 356 static void (*dll_rb_define_global_function) (const char*,VALUE(*)(),int); 357 static void (*dll_rb_define_method) (VALUE,const char*,VALUE(*)(),int); 358 static VALUE (*dll_rb_define_module) (const char*); 359 static void (*dll_rb_define_module_function) (VALUE,const char*,VALUE(*)(),int); 360 static void (*dll_rb_define_singleton_method) (VALUE,const char*,VALUE(*)(),int); 361 static void (*dll_rb_define_virtual_variable) (const char*,VALUE(*)(),void(*)()); 362 static VALUE *dll_rb_stdout; 363 static VALUE *dll_rb_eArgError; 364 static VALUE *dll_rb_eIndexError; 365 static VALUE *dll_rb_eRuntimeError; 366 static VALUE *dll_rb_eStandardError; 367 static VALUE (*dll_rb_eval_string_protect) (const char*, int*); 368 static void (*dll_rb_global_variable) (VALUE*); 369 static VALUE (*dll_rb_hash_aset) (VALUE, VALUE, VALUE); 370 static VALUE (*dll_rb_hash_new) (void); 371 static VALUE (*dll_rb_inspect) (VALUE); 372 static VALUE (*dll_rb_int2inum) (long); 373 # if VIM_SIZEOF_INT < VIM_SIZEOF_LONG /* 64 bits only */ 374 static long (*dll_rb_fix2int) (VALUE); 375 static long (*dll_rb_num2int) (VALUE); 376 static unsigned long (*dll_rb_num2uint) (VALUE); 377 # endif 378 static VALUE (*dll_rb_lastline_get) (void); 379 static void (*dll_rb_lastline_set) (VALUE); 380 static VALUE (*dll_rb_protect) (VALUE (*)(VALUE), VALUE, int*); 381 static void (*dll_rb_load) (VALUE, int); 382 static long (*dll_rb_num2long) (VALUE); 383 static unsigned long (*dll_rb_num2ulong) (VALUE); 384 static VALUE (*dll_rb_obj_alloc) (VALUE); 385 static VALUE (*dll_rb_obj_as_string) (VALUE); 386 static VALUE (*dll_rb_obj_id) (VALUE); 387 static void (*dll_rb_raise) (VALUE, const char*, ...); 388 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 389 static VALUE (*dll_rb_string_value) (volatile VALUE*); 390 # else 391 static char *(*dll_rb_str2cstr) (VALUE,int*); 392 # endif 393 static VALUE (*dll_rb_str_cat) (VALUE, const char*, long); 394 static VALUE (*dll_rb_str_concat) (VALUE, VALUE); 395 static VALUE (*dll_rb_str_new) (const char*, long); 396 # ifdef need_rb_str_new_cstr 397 /* Ruby may #define rb_str_new2 to use rb_str_new_cstr. */ 398 static VALUE (*dll_rb_str_new_cstr) (const char*); 399 # else 400 static VALUE (*dll_rb_str_new2) (const char*); 401 # endif 402 # ifdef RUBY19_OR_LATER 403 static VALUE (*dll_rb_errinfo) (void); 404 # else 405 static VALUE *dll_ruby_errinfo; 406 # endif 407 static void (*dll_ruby_init) (void); 408 static void (*dll_ruby_init_loadpath) (void); 409 # ifdef WIN3264 410 static void (*dll_NtInitialize) (int*, char***); 411 static void (*dll_ruby_sysinit) (int*, char***); 412 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 413 static int (*dll_rb_w32_snprintf)(char*, size_t, const char*, ...); 414 # endif 415 # endif 416 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 417 static char * (*dll_rb_string_value_ptr) (volatile VALUE*); 418 static VALUE (*dll_rb_float_new) (double); 419 static VALUE (*dll_rb_ary_new) (void); 420 static VALUE (*dll_rb_ary_push) (VALUE, VALUE); 421 # if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK) 422 # ifdef __ia64 423 static void * (*dll_rb_ia64_bsp) (void); 424 static void (*dll_ruby_init_stack)(VALUE*, void*); 425 # else 426 static void (*dll_ruby_init_stack)(VALUE*); 427 # endif 428 # endif 429 # endif 430 # ifdef RUBY19_OR_LATER 431 static VALUE (*dll_rb_int2big)(SIGNED_VALUE); 432 # endif 433 434 # ifdef RUBY19_OR_LATER 435 static void (*dll_ruby_script) (const char*); 436 static int (*dll_rb_enc_find_index) (const char*); 437 static rb_encoding* (*dll_rb_enc_find) (const char*); 438 static VALUE (*dll_rb_enc_str_new) (const char*, long, rb_encoding*); 439 static VALUE (*dll_rb_sprintf) (const char*, ...); 440 static VALUE (*dll_rb_require) (const char*); 441 static void* (*ruby_options)(int, char**); 442 # endif 443 444 # if defined(USE_RGENGC) && USE_RGENGC 445 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER == 21 446 static void (*dll_rb_gc_writebarrier_unprotect_promoted)(VALUE); 447 # else 448 static void (*dll_rb_gc_writebarrier_unprotect)(VALUE obj); 449 # endif 450 # endif 451 452 # if defined(RUBY19_OR_LATER) && !defined(PROTO) 453 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 22 454 long rb_num2long_stub(VALUE x) 455 # else 456 SIGNED_VALUE rb_num2long_stub(VALUE x) 457 # endif 458 { 459 return dll_rb_num2long(x); 460 } 461 VALUE rb_int2big_stub(SIGNED_VALUE x) 462 { 463 return dll_rb_int2big(x); 464 } 465 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19 \ 466 && VIM_SIZEOF_INT < VIM_SIZEOF_LONG 467 long rb_fix2int_stub(VALUE x) 468 { 469 return dll_rb_fix2int(x); 470 } 471 long rb_num2int_stub(VALUE x) 472 { 473 return dll_rb_num2int(x); 474 } 475 # endif 476 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20 477 VALUE 478 rb_float_new_in_heap(double d) 479 { 480 return dll_rb_float_new(d); 481 } 482 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 22 483 unsigned long rb_num2ulong(VALUE x) 484 # else 485 VALUE rb_num2ulong(VALUE x) 486 # endif 487 { 488 return (long)RSHIFT((SIGNED_VALUE)(x),1); 489 } 490 # endif 491 # endif 492 493 /* Do not generate a prototype here, VALUE isn't always defined. */ 494 # if defined(USE_RGENGC) && USE_RGENGC && !defined(PROTO) 495 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER == 21 496 void rb_gc_writebarrier_unprotect_promoted_stub(VALUE obj) 497 { 498 dll_rb_gc_writebarrier_unprotect_promoted(obj); 499 } 500 # else 501 void rb_gc_writebarrier_unprotect_stub(VALUE obj) 502 { 503 dll_rb_gc_writebarrier_unprotect(obj); 504 } 505 # endif 506 # endif 507 508 static HINSTANCE hinstRuby = NULL; /* Instance of ruby.dll */ 509 510 /* 511 * Table of name to function pointer of ruby. 512 */ 513 static struct 514 { 515 char *name; 516 RUBY_PROC *ptr; 517 } ruby_funcname_table[] = 518 { 519 {"rb_assoc_new", (RUBY_PROC*)&dll_rb_assoc_new}, 520 {"rb_cFalseClass", (RUBY_PROC*)&dll_rb_cFalseClass}, 521 # if defined(USE_RUBY_INTEGER) 522 {"rb_cInteger", (RUBY_PROC*)&dll_rb_cInteger}, 523 # else 524 {"rb_cFixnum", (RUBY_PROC*)&dll_rb_cFixnum}, 525 # endif 526 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 20 527 {"rb_cFloat", (RUBY_PROC*)&dll_rb_cFloat}, 528 # endif 529 {"rb_cNilClass", (RUBY_PROC*)&dll_rb_cNilClass}, 530 {"rb_cObject", (RUBY_PROC*)&dll_rb_cObject}, 531 {"rb_cSymbol", (RUBY_PROC*)&dll_rb_cSymbol}, 532 {"rb_cTrueClass", (RUBY_PROC*)&dll_rb_cTrueClass}, 533 {"rb_check_type", (RUBY_PROC*)&dll_rb_check_type}, 534 # ifdef USE_TYPEDDATA 535 {"rb_check_typeddata", (RUBY_PROC*)&dll_rb_check_typeddata}, 536 # endif 537 {"rb_class_path", (RUBY_PROC*)&dll_rb_class_path}, 538 # ifdef USE_TYPEDDATA 539 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 23 540 {"rb_data_typed_object_wrap", (RUBY_PROC*)&dll_rb_data_typed_object_wrap}, 541 # else 542 {"rb_data_typed_object_alloc", (RUBY_PROC*)&dll_rb_data_typed_object_alloc}, 543 # endif 544 # else 545 {"rb_data_object_alloc", (RUBY_PROC*)&dll_rb_data_object_alloc}, 546 # endif 547 {"rb_define_class_under", (RUBY_PROC*)&dll_rb_define_class_under}, 548 {"rb_define_const", (RUBY_PROC*)&dll_rb_define_const}, 549 {"rb_define_global_function", (RUBY_PROC*)&dll_rb_define_global_function}, 550 {"rb_define_method", (RUBY_PROC*)&dll_rb_define_method}, 551 {"rb_define_module", (RUBY_PROC*)&dll_rb_define_module}, 552 {"rb_define_module_function", (RUBY_PROC*)&dll_rb_define_module_function}, 553 {"rb_define_singleton_method", (RUBY_PROC*)&dll_rb_define_singleton_method}, 554 {"rb_define_virtual_variable", (RUBY_PROC*)&dll_rb_define_virtual_variable}, 555 {"rb_stdout", (RUBY_PROC*)&dll_rb_stdout}, 556 {"rb_eArgError", (RUBY_PROC*)&dll_rb_eArgError}, 557 {"rb_eIndexError", (RUBY_PROC*)&dll_rb_eIndexError}, 558 {"rb_eRuntimeError", (RUBY_PROC*)&dll_rb_eRuntimeError}, 559 {"rb_eStandardError", (RUBY_PROC*)&dll_rb_eStandardError}, 560 {"rb_eval_string_protect", (RUBY_PROC*)&dll_rb_eval_string_protect}, 561 {"rb_global_variable", (RUBY_PROC*)&dll_rb_global_variable}, 562 {"rb_hash_aset", (RUBY_PROC*)&dll_rb_hash_aset}, 563 {"rb_hash_new", (RUBY_PROC*)&dll_rb_hash_new}, 564 {"rb_inspect", (RUBY_PROC*)&dll_rb_inspect}, 565 {"rb_int2inum", (RUBY_PROC*)&dll_rb_int2inum}, 566 # if VIM_SIZEOF_INT < VIM_SIZEOF_LONG /* 64 bits only */ 567 {"rb_fix2int", (RUBY_PROC*)&dll_rb_fix2int}, 568 {"rb_num2int", (RUBY_PROC*)&dll_rb_num2int}, 569 {"rb_num2uint", (RUBY_PROC*)&dll_rb_num2uint}, 570 # endif 571 {"rb_lastline_get", (RUBY_PROC*)&dll_rb_lastline_get}, 572 {"rb_lastline_set", (RUBY_PROC*)&dll_rb_lastline_set}, 573 {"rb_protect", (RUBY_PROC*)&dll_rb_protect}, 574 {"rb_load", (RUBY_PROC*)&dll_rb_load}, 575 {"rb_num2long", (RUBY_PROC*)&dll_rb_num2long}, 576 {"rb_num2ulong", (RUBY_PROC*)&dll_rb_num2ulong}, 577 {"rb_obj_alloc", (RUBY_PROC*)&dll_rb_obj_alloc}, 578 {"rb_obj_as_string", (RUBY_PROC*)&dll_rb_obj_as_string}, 579 {"rb_obj_id", (RUBY_PROC*)&dll_rb_obj_id}, 580 {"rb_raise", (RUBY_PROC*)&dll_rb_raise}, 581 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 582 {"rb_string_value", (RUBY_PROC*)&dll_rb_string_value}, 583 # else 584 {"rb_str2cstr", (RUBY_PROC*)&dll_rb_str2cstr}, 585 # endif 586 {"rb_str_cat", (RUBY_PROC*)&dll_rb_str_cat}, 587 {"rb_str_concat", (RUBY_PROC*)&dll_rb_str_concat}, 588 {"rb_str_new", (RUBY_PROC*)&dll_rb_str_new}, 589 # ifdef need_rb_str_new_cstr 590 {"rb_str_new_cstr", (RUBY_PROC*)&dll_rb_str_new_cstr}, 591 # else 592 {"rb_str_new2", (RUBY_PROC*)&dll_rb_str_new2}, 593 # endif 594 # ifdef RUBY19_OR_LATER 595 {"rb_errinfo", (RUBY_PROC*)&dll_rb_errinfo}, 596 # else 597 {"ruby_errinfo", (RUBY_PROC*)&dll_ruby_errinfo}, 598 # endif 599 {"ruby_init", (RUBY_PROC*)&dll_ruby_init}, 600 {"ruby_init_loadpath", (RUBY_PROC*)&dll_ruby_init_loadpath}, 601 # ifdef WIN3264 602 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER < 19 603 {"NtInitialize", (RUBY_PROC*)&dll_NtInitialize}, 604 # else 605 {"ruby_sysinit", (RUBY_PROC*)&dll_ruby_sysinit}, 606 # endif 607 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 608 {"rb_w32_snprintf", (RUBY_PROC*)&dll_rb_w32_snprintf}, 609 # endif 610 # endif 611 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 18 612 {"rb_string_value_ptr", (RUBY_PROC*)&dll_rb_string_value_ptr}, 613 # if DYNAMIC_RUBY_VER <= 19 614 {"rb_float_new", (RUBY_PROC*)&dll_rb_float_new}, 615 # else 616 {"rb_float_new_in_heap", (RUBY_PROC*)&dll_rb_float_new}, 617 # endif 618 {"rb_ary_new", (RUBY_PROC*)&dll_rb_ary_new}, 619 {"rb_ary_push", (RUBY_PROC*)&dll_rb_ary_push}, 620 # endif 621 # ifdef RUBY19_OR_LATER 622 {"rb_int2big", (RUBY_PROC*)&dll_rb_int2big}, 623 {"ruby_script", (RUBY_PROC*)&dll_ruby_script}, 624 {"rb_enc_find_index", (RUBY_PROC*)&dll_rb_enc_find_index}, 625 {"rb_enc_find", (RUBY_PROC*)&dll_rb_enc_find}, 626 {"rb_enc_str_new", (RUBY_PROC*)&dll_rb_enc_str_new}, 627 {"rb_sprintf", (RUBY_PROC*)&dll_rb_sprintf}, 628 {"rb_require", (RUBY_PROC*)&dll_rb_require}, 629 {"ruby_options", (RUBY_PROC*)&dll_ruby_options}, 630 # endif 631 # if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK) 632 # ifdef __ia64 633 {"rb_ia64_bsp", (RUBY_PROC*)&dll_rb_ia64_bsp}, 634 # endif 635 {"ruby_init_stack", (RUBY_PROC*)&dll_ruby_init_stack}, 636 # endif 637 # if defined(USE_RGENGC) && USE_RGENGC 638 # if defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER == 21 639 {"rb_gc_writebarrier_unprotect_promoted", (RUBY_PROC*)&dll_rb_gc_writebarrier_unprotect_promoted}, 640 # else 641 {"rb_gc_writebarrier_unprotect", (RUBY_PROC*)&dll_rb_gc_writebarrier_unprotect}, 642 # endif 643 # endif 644 {"", NULL}, 645 }; 646 647 /* 648 * Free ruby.dll 649 */ 650 static void 651 end_dynamic_ruby(void) 652 { 653 if (hinstRuby) 654 { 655 close_dll(hinstRuby); 656 hinstRuby = NULL; 657 } 658 } 659 660 /* 661 * Load library and get all pointers. 662 * Parameter 'libname' provides name of DLL. 663 * Return OK or FAIL. 664 */ 665 static int 666 ruby_runtime_link_init(char *libname, int verbose) 667 { 668 int i; 669 670 if (hinstRuby) 671 return OK; 672 hinstRuby = load_dll(libname); 673 if (!hinstRuby) 674 { 675 if (verbose) 676 EMSG2(_(e_loadlib), libname); 677 return FAIL; 678 } 679 680 for (i = 0; ruby_funcname_table[i].ptr; ++i) 681 { 682 if (!(*ruby_funcname_table[i].ptr = symbol_from_dll(hinstRuby, 683 ruby_funcname_table[i].name))) 684 { 685 close_dll(hinstRuby); 686 hinstRuby = NULL; 687 if (verbose) 688 EMSG2(_(e_loadfunc), ruby_funcname_table[i].name); 689 return FAIL; 690 } 691 } 692 return OK; 693 } 694 695 /* 696 * If ruby is enabled (there is installed ruby on Windows system) return TRUE, 697 * else FALSE. 698 */ 699 int 700 ruby_enabled(int verbose) 701 { 702 return ruby_runtime_link_init((char *)p_rubydll, verbose) == OK; 703 } 704 #endif /* defined(DYNAMIC_RUBY) || defined(PROTO) */ 705 706 void 707 ruby_end(void) 708 { 709 #ifdef DYNAMIC_RUBY 710 end_dynamic_ruby(); 711 #endif 712 } 713 714 void ex_ruby(exarg_T *eap) 715 { 716 int state; 717 char *script = NULL; 718 719 script = (char *)script_get(eap, eap->arg); 720 if (!eap->skip && ensure_ruby_initialized()) 721 { 722 if (script == NULL) 723 rb_eval_string_protect((char *)eap->arg, &state); 724 else 725 rb_eval_string_protect(script, &state); 726 if (state) 727 error_print(state); 728 } 729 vim_free(script); 730 } 731 732 /* 733 * In Ruby 1.9 or later, ruby String object has encoding. 734 * conversion buffer string of vim to ruby String object using 735 * VIM encoding option. 736 */ 737 static VALUE 738 vim_str2rb_enc_str(const char *s) 739 { 740 #ifdef RUBY19_OR_LATER 741 int isnum; 742 long lval; 743 char_u *sval; 744 rb_encoding *enc; 745 746 isnum = get_option_value((char_u *)"enc", &lval, &sval, 0); 747 if (isnum == 0) 748 { 749 enc = rb_enc_find((char *)sval); 750 vim_free(sval); 751 if (enc) 752 { 753 return rb_enc_str_new(s, (long)strlen(s), enc); 754 } 755 } 756 #endif 757 return rb_str_new2(s); 758 } 759 760 static VALUE 761 eval_enc_string_protect(const char *str, int *state) 762 { 763 #ifdef RUBY19_OR_LATER 764 int isnum; 765 long lval; 766 char_u *sval; 767 rb_encoding *enc; 768 VALUE v; 769 770 isnum = get_option_value((char_u *)"enc", &lval, &sval, 0); 771 if (isnum == 0) 772 { 773 enc = rb_enc_find((char *)sval); 774 vim_free(sval); 775 if (enc) 776 { 777 v = rb_sprintf("#-*- coding:%s -*-\n%s", rb_enc_name(enc), str); 778 return rb_eval_string_protect(StringValuePtr(v), state); 779 } 780 } 781 #endif 782 return rb_eval_string_protect(str, state); 783 } 784 785 void ex_rubydo(exarg_T *eap) 786 { 787 int state; 788 linenr_T i; 789 buf_T *was_curbuf = curbuf; 790 791 if (ensure_ruby_initialized()) 792 { 793 if (u_save(eap->line1 - 1, eap->line2 + 1) != OK) 794 return; 795 for (i = eap->line1; i <= eap->line2; i++) 796 { 797 VALUE line; 798 799 if (i > curbuf->b_ml.ml_line_count) 800 break; 801 line = vim_str2rb_enc_str((char *)ml_get(i)); 802 rb_lastline_set(line); 803 eval_enc_string_protect((char *) eap->arg, &state); 804 if (state) 805 { 806 error_print(state); 807 break; 808 } 809 if (was_curbuf != curbuf) 810 break; 811 line = rb_lastline_get(); 812 if (!NIL_P(line)) 813 { 814 if (TYPE(line) != T_STRING) 815 { 816 EMSG(_("E265: $_ must be an instance of String")); 817 return; 818 } 819 ml_replace(i, (char_u *) StringValuePtr(line), 1); 820 changed(); 821 #ifdef SYNTAX_HL 822 syn_changed(i); /* recompute syntax hl. for this line */ 823 #endif 824 } 825 } 826 check_cursor(); 827 update_curbuf(NOT_VALID); 828 } 829 } 830 831 static VALUE rb_load_wrap(VALUE file_to_load) 832 { 833 rb_load(file_to_load, 0); 834 return Qnil; 835 } 836 837 void ex_rubyfile(exarg_T *eap) 838 { 839 int state; 840 841 if (ensure_ruby_initialized()) 842 { 843 VALUE file_to_load = rb_str_new2((const char *)eap->arg); 844 rb_protect(rb_load_wrap, file_to_load, &state); 845 if (state) 846 error_print(state); 847 } 848 } 849 850 void ruby_buffer_free(buf_T *buf) 851 { 852 if (buf->b_ruby_ref) 853 { 854 rb_hash_aset(objtbl, rb_obj_id((VALUE) buf->b_ruby_ref), Qnil); 855 RDATA(buf->b_ruby_ref)->data = NULL; 856 } 857 } 858 859 void ruby_window_free(win_T *win) 860 { 861 if (win->w_ruby_ref) 862 { 863 rb_hash_aset(objtbl, rb_obj_id((VALUE) win->w_ruby_ref), Qnil); 864 RDATA(win->w_ruby_ref)->data = NULL; 865 } 866 } 867 868 static int ensure_ruby_initialized(void) 869 { 870 if (!ruby_initialized) 871 { 872 #ifdef DYNAMIC_RUBY 873 if (ruby_enabled(TRUE)) 874 { 875 #endif 876 #ifdef _WIN32 877 /* suggested by Ariya Mizutani */ 878 int argc = 1; 879 char *argv[] = {"gvim.exe"}; 880 char **argvp = argv; 881 # ifdef RUBY19_OR_LATER 882 ruby_sysinit(&argc, &argvp); 883 # else 884 NtInitialize(&argc, &argvp); 885 # endif 886 #endif 887 { 888 #if defined(RUBY19_OR_LATER) || defined(RUBY_INIT_STACK) 889 ruby_init_stack(ruby_stack_start); 890 #endif 891 ruby_init(); 892 } 893 #ifdef RUBY19_OR_LATER 894 { 895 int dummy_argc = 2; 896 char *dummy_argv[] = {"vim-ruby", "-e_=0"}; 897 ruby_options(dummy_argc, dummy_argv); 898 } 899 ruby_script("vim-ruby"); 900 #else 901 ruby_init_loadpath(); 902 #endif 903 ruby_io_init(); 904 ruby_vim_init(); 905 ruby_initialized = 1; 906 #ifdef DYNAMIC_RUBY 907 } 908 else 909 { 910 EMSG(_("E266: Sorry, this command is disabled, the Ruby library could not be loaded.")); 911 return 0; 912 } 913 #endif 914 } 915 return ruby_initialized; 916 } 917 918 static void error_print(int state) 919 { 920 #ifndef DYNAMIC_RUBY 921 #if !(defined(RUBY_VERSION) && RUBY_VERSION >= 19) \ 922 && !(defined(DYNAMIC_RUBY_VER) && DYNAMIC_RUBY_VER >= 19) 923 RUBYEXTERN VALUE ruby_errinfo; 924 #endif 925 #endif 926 VALUE eclass; 927 VALUE einfo; 928 char buff[BUFSIZ]; 929 930 #define TAG_RETURN 0x1 931 #define TAG_BREAK 0x2 932 #define TAG_NEXT 0x3 933 #define TAG_RETRY 0x4 934 #define TAG_REDO 0x5 935 #define TAG_RAISE 0x6 936 #define TAG_THROW 0x7 937 #define TAG_FATAL 0x8 938 #define TAG_MASK 0xf 939 940 switch (state) 941 { 942 case TAG_RETURN: 943 EMSG(_("E267: unexpected return")); 944 break; 945 case TAG_NEXT: 946 EMSG(_("E268: unexpected next")); 947 break; 948 case TAG_BREAK: 949 EMSG(_("E269: unexpected break")); 950 break; 951 case TAG_REDO: 952 EMSG(_("E270: unexpected redo")); 953 break; 954 case TAG_RETRY: 955 EMSG(_("E271: retry outside of rescue clause")); 956 break; 957 case TAG_RAISE: 958 case TAG_FATAL: 959 #ifdef RUBY19_OR_LATER 960 eclass = CLASS_OF(rb_errinfo()); 961 einfo = rb_obj_as_string(rb_errinfo()); 962 #else 963 eclass = CLASS_OF(ruby_errinfo); 964 einfo = rb_obj_as_string(ruby_errinfo); 965 #endif 966 if (eclass == rb_eRuntimeError && RSTRING_LEN(einfo) == 0) 967 { 968 EMSG(_("E272: unhandled exception")); 969 } 970 else 971 { 972 VALUE epath; 973 char *p; 974 975 epath = rb_class_path(eclass); 976 vim_snprintf(buff, BUFSIZ, "%s: %s", 977 RSTRING_PTR(epath), RSTRING_PTR(einfo)); 978 p = strchr(buff, '\n'); 979 if (p) *p = '\0'; 980 EMSG(buff); 981 } 982 break; 983 default: 984 vim_snprintf(buff, BUFSIZ, _("E273: unknown longjmp status %d"), state); 985 EMSG(buff); 986 break; 987 } 988 } 989 990 static VALUE vim_message(VALUE self UNUSED, VALUE str) 991 { 992 char *buff, *p; 993 994 str = rb_obj_as_string(str); 995 if (RSTRING_LEN(str) > 0) 996 { 997 /* Only do this when the string isn't empty, alloc(0) causes trouble. */ 998 buff = ALLOCA_N(char, RSTRING_LEN(str) + 1); 999 strcpy(buff, RSTRING_PTR(str)); 1000 p = strchr(buff, '\n'); 1001 if (p) *p = '\0'; 1002 MSG(buff); 1003 } 1004 else 1005 { 1006 MSG(""); 1007 } 1008 return Qnil; 1009 } 1010 1011 static VALUE vim_set_option(VALUE self UNUSED, VALUE str) 1012 { 1013 do_set((char_u *)StringValuePtr(str), 0); 1014 update_screen(NOT_VALID); 1015 return Qnil; 1016 } 1017 1018 static VALUE vim_command(VALUE self UNUSED, VALUE str) 1019 { 1020 do_cmdline_cmd((char_u *)StringValuePtr(str)); 1021 return Qnil; 1022 } 1023 1024 #ifdef FEAT_EVAL 1025 static VALUE vim_to_ruby(typval_T *tv) 1026 { 1027 VALUE result = Qnil; 1028 1029 if (tv->v_type == VAR_STRING) 1030 { 1031 result = rb_str_new2(tv->vval.v_string == NULL 1032 ? "" : (char *)(tv->vval.v_string)); 1033 } 1034 else if (tv->v_type == VAR_NUMBER) 1035 { 1036 result = INT2NUM(tv->vval.v_number); 1037 } 1038 # ifdef FEAT_FLOAT 1039 else if (tv->v_type == VAR_FLOAT) 1040 { 1041 result = rb_float_new(tv->vval.v_float); 1042 } 1043 # endif 1044 else if (tv->v_type == VAR_LIST) 1045 { 1046 list_T *list = tv->vval.v_list; 1047 listitem_T *curr; 1048 1049 result = rb_ary_new(); 1050 1051 if (list != NULL) 1052 { 1053 for (curr = list->lv_first; curr != NULL; curr = curr->li_next) 1054 { 1055 rb_ary_push(result, vim_to_ruby(&curr->li_tv)); 1056 } 1057 } 1058 } 1059 else if (tv->v_type == VAR_DICT) 1060 { 1061 result = rb_hash_new(); 1062 1063 if (tv->vval.v_dict != NULL) 1064 { 1065 hashtab_T *ht = &tv->vval.v_dict->dv_hashtab; 1066 long_u todo = ht->ht_used; 1067 hashitem_T *hi; 1068 dictitem_T *di; 1069 1070 for (hi = ht->ht_array; todo > 0; ++hi) 1071 { 1072 if (!HASHITEM_EMPTY(hi)) 1073 { 1074 --todo; 1075 1076 di = dict_lookup(hi); 1077 rb_hash_aset(result, rb_str_new2((char *)hi->hi_key), 1078 vim_to_ruby(&di->di_tv)); 1079 } 1080 } 1081 } 1082 } 1083 else if (tv->v_type == VAR_SPECIAL) 1084 { 1085 if (tv->vval.v_number <= VVAL_TRUE) 1086 result = INT2NUM(tv->vval.v_number); 1087 } /* else return Qnil; */ 1088 1089 return result; 1090 } 1091 #endif 1092 1093 static VALUE vim_evaluate(VALUE self UNUSED, VALUE str) 1094 { 1095 #ifdef FEAT_EVAL 1096 typval_T *tv; 1097 VALUE result; 1098 1099 tv = eval_expr((char_u *)StringValuePtr(str), NULL); 1100 if (tv == NULL) 1101 { 1102 return Qnil; 1103 } 1104 result = vim_to_ruby(tv); 1105 1106 free_tv(tv); 1107 1108 return result; 1109 #else 1110 return Qnil; 1111 #endif 1112 } 1113 1114 #ifdef USE_TYPEDDATA 1115 static size_t buffer_dsize(const void *buf); 1116 1117 static const rb_data_type_t buffer_type = { 1118 "vim_buffer", 1119 {0, 0, buffer_dsize, {0, 0}}, 1120 0, 0, 1121 # ifdef RUBY_TYPED_FREE_IMMEDIATELY 1122 0, 1123 # endif 1124 }; 1125 1126 static size_t buffer_dsize(const void *buf UNUSED) 1127 { 1128 return sizeof(buf_T); 1129 } 1130 #endif 1131 1132 static VALUE buffer_new(buf_T *buf) 1133 { 1134 if (buf->b_ruby_ref) 1135 { 1136 return (VALUE) buf->b_ruby_ref; 1137 } 1138 else 1139 { 1140 #ifdef USE_TYPEDDATA 1141 VALUE obj = TypedData_Wrap_Struct(cBuffer, &buffer_type, buf); 1142 #else 1143 VALUE obj = Data_Wrap_Struct(cBuffer, 0, 0, buf); 1144 #endif 1145 buf->b_ruby_ref = (void *) obj; 1146 rb_hash_aset(objtbl, rb_obj_id(obj), obj); 1147 return obj; 1148 } 1149 } 1150 1151 static buf_T *get_buf(VALUE obj) 1152 { 1153 buf_T *buf; 1154 1155 #ifdef USE_TYPEDDATA 1156 TypedData_Get_Struct(obj, buf_T, &buffer_type, buf); 1157 #else 1158 Data_Get_Struct(obj, buf_T, buf); 1159 #endif 1160 if (buf == NULL) 1161 rb_raise(eDeletedBufferError, "attempt to refer to deleted buffer"); 1162 return buf; 1163 } 1164 1165 static VALUE buffer_s_current(void) 1166 { 1167 return buffer_new(curbuf); 1168 } 1169 1170 static VALUE buffer_s_count(void) 1171 { 1172 buf_T *b; 1173 int n = 0; 1174 1175 FOR_ALL_BUFFERS(b) 1176 { 1177 /* Deleted buffers should not be counted 1178 * SegPhault - 01/07/05 */ 1179 if (b->b_p_bl) 1180 n++; 1181 } 1182 1183 return INT2NUM(n); 1184 } 1185 1186 static VALUE buffer_s_aref(VALUE self UNUSED, VALUE num) 1187 { 1188 buf_T *b; 1189 int n = NUM2INT(num); 1190 1191 FOR_ALL_BUFFERS(b) 1192 { 1193 /* Deleted buffers should not be counted 1194 * SegPhault - 01/07/05 */ 1195 if (!b->b_p_bl) 1196 continue; 1197 1198 if (n == 0) 1199 return buffer_new(b); 1200 1201 n--; 1202 } 1203 return Qnil; 1204 } 1205 1206 static VALUE buffer_name(VALUE self) 1207 { 1208 buf_T *buf = get_buf(self); 1209 1210 return buf->b_ffname ? rb_str_new2((char *)buf->b_ffname) : Qnil; 1211 } 1212 1213 static VALUE buffer_number(VALUE self) 1214 { 1215 buf_T *buf = get_buf(self); 1216 1217 return INT2NUM(buf->b_fnum); 1218 } 1219 1220 static VALUE buffer_count(VALUE self) 1221 { 1222 buf_T *buf = get_buf(self); 1223 1224 return INT2NUM(buf->b_ml.ml_line_count); 1225 } 1226 1227 static VALUE get_buffer_line(buf_T *buf, linenr_T n) 1228 { 1229 if (n <= 0 || n > buf->b_ml.ml_line_count) 1230 rb_raise(rb_eIndexError, "line number %ld out of range", (long)n); 1231 return vim_str2rb_enc_str((char *)ml_get_buf(buf, n, FALSE)); 1232 } 1233 1234 static VALUE buffer_aref(VALUE self, VALUE num) 1235 { 1236 buf_T *buf = get_buf(self); 1237 1238 if (buf != NULL) 1239 return get_buffer_line(buf, (linenr_T)NUM2LONG(num)); 1240 return Qnil; /* For stop warning */ 1241 } 1242 1243 static VALUE set_buffer_line(buf_T *buf, linenr_T n, VALUE str) 1244 { 1245 char *line = StringValuePtr(str); 1246 aco_save_T aco; 1247 1248 if (n > 0 && n <= buf->b_ml.ml_line_count && line != NULL) 1249 { 1250 /* set curwin/curbuf for "buf" and save some things */ 1251 aucmd_prepbuf(&aco, buf); 1252 1253 if (u_savesub(n) == OK) 1254 { 1255 ml_replace(n, (char_u *)line, TRUE); 1256 changed(); 1257 #ifdef SYNTAX_HL 1258 syn_changed(n); /* recompute syntax hl. for this line */ 1259 #endif 1260 } 1261 1262 /* restore curwin/curbuf and a few other things */ 1263 aucmd_restbuf(&aco); 1264 /* Careful: autocommands may have made "buf" invalid! */ 1265 1266 update_curbuf(NOT_VALID); 1267 } 1268 else 1269 { 1270 rb_raise(rb_eIndexError, "line number %ld out of range", (long)n); 1271 } 1272 return str; 1273 } 1274 1275 static VALUE buffer_aset(VALUE self, VALUE num, VALUE str) 1276 { 1277 buf_T *buf = get_buf(self); 1278 1279 if (buf != NULL) 1280 return set_buffer_line(buf, (linenr_T)NUM2LONG(num), str); 1281 return str; 1282 } 1283 1284 static VALUE buffer_delete(VALUE self, VALUE num) 1285 { 1286 buf_T *buf = get_buf(self); 1287 long n = NUM2LONG(num); 1288 aco_save_T aco; 1289 1290 if (n > 0 && n <= buf->b_ml.ml_line_count) 1291 { 1292 /* set curwin/curbuf for "buf" and save some things */ 1293 aucmd_prepbuf(&aco, buf); 1294 1295 if (u_savedel(n, 1) == OK) 1296 { 1297 ml_delete(n, 0); 1298 1299 /* Changes to non-active buffers should properly refresh 1300 * SegPhault - 01/09/05 */ 1301 deleted_lines_mark(n, 1L); 1302 1303 changed(); 1304 } 1305 1306 /* restore curwin/curbuf and a few other things */ 1307 aucmd_restbuf(&aco); 1308 /* Careful: autocommands may have made "buf" invalid! */ 1309 1310 update_curbuf(NOT_VALID); 1311 } 1312 else 1313 { 1314 rb_raise(rb_eIndexError, "line number %ld out of range", n); 1315 } 1316 return Qnil; 1317 } 1318 1319 static VALUE buffer_append(VALUE self, VALUE num, VALUE str) 1320 { 1321 buf_T *buf = get_buf(self); 1322 char *line = StringValuePtr(str); 1323 long n = NUM2LONG(num); 1324 aco_save_T aco; 1325 1326 if (line == NULL) 1327 { 1328 rb_raise(rb_eIndexError, "NULL line"); 1329 } 1330 else if (n >= 0 && n <= buf->b_ml.ml_line_count) 1331 { 1332 /* set curwin/curbuf for "buf" and save some things */ 1333 aucmd_prepbuf(&aco, buf); 1334 1335 if (u_inssub(n + 1) == OK) 1336 { 1337 ml_append(n, (char_u *) line, (colnr_T) 0, FALSE); 1338 1339 /* Changes to non-active buffers should properly refresh screen 1340 * SegPhault - 12/20/04 */ 1341 appended_lines_mark(n, 1L); 1342 1343 changed(); 1344 } 1345 1346 /* restore curwin/curbuf and a few other things */ 1347 aucmd_restbuf(&aco); 1348 /* Careful: autocommands may have made "buf" invalid! */ 1349 1350 update_curbuf(NOT_VALID); 1351 } 1352 else 1353 { 1354 rb_raise(rb_eIndexError, "line number %ld out of range", n); 1355 } 1356 return str; 1357 } 1358 1359 #ifdef USE_TYPEDDATA 1360 static size_t window_dsize(const void *buf); 1361 1362 static const rb_data_type_t window_type = { 1363 "vim_window", 1364 {0, 0, window_dsize, {0, 0}}, 1365 0, 0, 1366 # ifdef RUBY_TYPED_FREE_IMMEDIATELY 1367 0, 1368 # endif 1369 }; 1370 1371 static size_t window_dsize(const void *win UNUSED) 1372 { 1373 return sizeof(win_T); 1374 } 1375 #endif 1376 1377 static VALUE window_new(win_T *win) 1378 { 1379 if (win->w_ruby_ref) 1380 { 1381 return (VALUE) win->w_ruby_ref; 1382 } 1383 else 1384 { 1385 #ifdef USE_TYPEDDATA 1386 VALUE obj = TypedData_Wrap_Struct(cVimWindow, &window_type, win); 1387 #else 1388 VALUE obj = Data_Wrap_Struct(cVimWindow, 0, 0, win); 1389 #endif 1390 win->w_ruby_ref = (void *) obj; 1391 rb_hash_aset(objtbl, rb_obj_id(obj), obj); 1392 return obj; 1393 } 1394 } 1395 1396 static win_T *get_win(VALUE obj) 1397 { 1398 win_T *win; 1399 1400 #ifdef USE_TYPEDDATA 1401 TypedData_Get_Struct(obj, win_T, &window_type, win); 1402 #else 1403 Data_Get_Struct(obj, win_T, win); 1404 #endif 1405 if (win == NULL) 1406 rb_raise(eDeletedWindowError, "attempt to refer to deleted window"); 1407 return win; 1408 } 1409 1410 static VALUE window_s_current(void) 1411 { 1412 return window_new(curwin); 1413 } 1414 1415 /* 1416 * Added line manipulation functions 1417 * SegPhault - 03/07/05 1418 */ 1419 static VALUE line_s_current(void) 1420 { 1421 return get_buffer_line(curbuf, curwin->w_cursor.lnum); 1422 } 1423 1424 static VALUE set_current_line(VALUE self UNUSED, VALUE str) 1425 { 1426 return set_buffer_line(curbuf, curwin->w_cursor.lnum, str); 1427 } 1428 1429 static VALUE current_line_number(void) 1430 { 1431 return INT2FIX((int)curwin->w_cursor.lnum); 1432 } 1433 1434 1435 1436 static VALUE window_s_count(void) 1437 { 1438 win_T *w; 1439 int n = 0; 1440 1441 FOR_ALL_WINDOWS(w) 1442 n++; 1443 return INT2NUM(n); 1444 } 1445 1446 static VALUE window_s_aref(VALUE self UNUSED, VALUE num) 1447 { 1448 win_T *w; 1449 int n = NUM2INT(num); 1450 1451 for (w = firstwin; w != NULL; w = w->w_next, --n) 1452 if (n == 0) 1453 return window_new(w); 1454 return Qnil; 1455 } 1456 1457 static VALUE window_buffer(VALUE self) 1458 { 1459 win_T *win = get_win(self); 1460 1461 return buffer_new(win->w_buffer); 1462 } 1463 1464 static VALUE window_height(VALUE self) 1465 { 1466 win_T *win = get_win(self); 1467 1468 return INT2NUM(win->w_height); 1469 } 1470 1471 static VALUE window_set_height(VALUE self, VALUE height) 1472 { 1473 win_T *win = get_win(self); 1474 win_T *savewin = curwin; 1475 1476 curwin = win; 1477 win_setheight(NUM2INT(height)); 1478 curwin = savewin; 1479 return height; 1480 } 1481 1482 static VALUE window_width(VALUE self UNUSED) 1483 { 1484 return INT2NUM(get_win(self)->w_width); 1485 } 1486 1487 static VALUE window_set_width(VALUE self UNUSED, VALUE width) 1488 { 1489 win_T *win = get_win(self); 1490 win_T *savewin = curwin; 1491 1492 curwin = win; 1493 win_setwidth(NUM2INT(width)); 1494 curwin = savewin; 1495 return width; 1496 } 1497 1498 static VALUE window_cursor(VALUE self) 1499 { 1500 win_T *win = get_win(self); 1501 1502 return rb_assoc_new(INT2NUM(win->w_cursor.lnum), INT2NUM(win->w_cursor.col)); 1503 } 1504 1505 static VALUE window_set_cursor(VALUE self, VALUE pos) 1506 { 1507 VALUE lnum, col; 1508 win_T *win = get_win(self); 1509 1510 Check_Type(pos, T_ARRAY); 1511 if (RARRAY_LEN(pos) != 2) 1512 rb_raise(rb_eArgError, "array length must be 2"); 1513 lnum = RARRAY_PTR(pos)[0]; 1514 col = RARRAY_PTR(pos)[1]; 1515 win->w_cursor.lnum = NUM2LONG(lnum); 1516 win->w_cursor.col = NUM2UINT(col); 1517 check_cursor(); /* put cursor on an existing line */ 1518 update_screen(NOT_VALID); 1519 return Qnil; 1520 } 1521 1522 static VALUE f_nop(VALUE self UNUSED) 1523 { 1524 return Qnil; 1525 } 1526 1527 static VALUE f_p(int argc, VALUE *argv, VALUE self UNUSED) 1528 { 1529 int i; 1530 VALUE str = rb_str_new("", 0); 1531 1532 for (i = 0; i < argc; i++) 1533 { 1534 if (i > 0) rb_str_cat(str, ", ", 2); 1535 rb_str_concat(str, rb_inspect(argv[i])); 1536 } 1537 MSG(RSTRING_PTR(str)); 1538 return Qnil; 1539 } 1540 1541 static void ruby_io_init(void) 1542 { 1543 #ifndef DYNAMIC_RUBY 1544 RUBYEXTERN VALUE rb_stdout; 1545 #endif 1546 1547 rb_stdout = rb_obj_alloc(rb_cObject); 1548 rb_define_singleton_method(rb_stdout, "write", vim_message, 1); 1549 rb_define_singleton_method(rb_stdout, "flush", f_nop, 0); 1550 rb_define_global_function("p", f_p, -1); 1551 } 1552 1553 static void ruby_vim_init(void) 1554 { 1555 objtbl = rb_hash_new(); 1556 rb_global_variable(&objtbl); 1557 1558 /* The Vim module used to be called "VIM", but "Vim" is better. Make an 1559 * alias "VIM" for backwards compatibility. */ 1560 mVIM = rb_define_module("Vim"); 1561 rb_define_const(rb_cObject, "VIM", mVIM); 1562 rb_define_const(mVIM, "VERSION_MAJOR", INT2NUM(VIM_VERSION_MAJOR)); 1563 rb_define_const(mVIM, "VERSION_MINOR", INT2NUM(VIM_VERSION_MINOR)); 1564 rb_define_const(mVIM, "VERSION_BUILD", INT2NUM(VIM_VERSION_BUILD)); 1565 rb_define_const(mVIM, "VERSION_PATCHLEVEL", INT2NUM(VIM_VERSION_PATCHLEVEL)); 1566 rb_define_const(mVIM, "VERSION_SHORT", rb_str_new2(VIM_VERSION_SHORT)); 1567 rb_define_const(mVIM, "VERSION_MEDIUM", rb_str_new2(VIM_VERSION_MEDIUM)); 1568 rb_define_const(mVIM, "VERSION_LONG", rb_str_new2(VIM_VERSION_LONG)); 1569 rb_define_const(mVIM, "VERSION_LONG_DATE", rb_str_new2(VIM_VERSION_LONG_DATE)); 1570 rb_define_module_function(mVIM, "message", vim_message, 1); 1571 rb_define_module_function(mVIM, "set_option", vim_set_option, 1); 1572 rb_define_module_function(mVIM, "command", vim_command, 1); 1573 rb_define_module_function(mVIM, "evaluate", vim_evaluate, 1); 1574 1575 eDeletedBufferError = rb_define_class_under(mVIM, "DeletedBufferError", 1576 rb_eStandardError); 1577 eDeletedWindowError = rb_define_class_under(mVIM, "DeletedWindowError", 1578 rb_eStandardError); 1579 1580 cBuffer = rb_define_class_under(mVIM, "Buffer", rb_cObject); 1581 rb_define_singleton_method(cBuffer, "current", buffer_s_current, 0); 1582 rb_define_singleton_method(cBuffer, "count", buffer_s_count, 0); 1583 rb_define_singleton_method(cBuffer, "[]", buffer_s_aref, 1); 1584 rb_define_method(cBuffer, "name", buffer_name, 0); 1585 rb_define_method(cBuffer, "number", buffer_number, 0); 1586 rb_define_method(cBuffer, "count", buffer_count, 0); 1587 rb_define_method(cBuffer, "length", buffer_count, 0); 1588 rb_define_method(cBuffer, "[]", buffer_aref, 1); 1589 rb_define_method(cBuffer, "[]=", buffer_aset, 2); 1590 rb_define_method(cBuffer, "delete", buffer_delete, 1); 1591 rb_define_method(cBuffer, "append", buffer_append, 2); 1592 1593 /* Added line manipulation functions 1594 * SegPhault - 03/07/05 */ 1595 rb_define_method(cBuffer, "line_number", current_line_number, 0); 1596 rb_define_method(cBuffer, "line", line_s_current, 0); 1597 rb_define_method(cBuffer, "line=", set_current_line, 1); 1598 1599 1600 cVimWindow = rb_define_class_under(mVIM, "Window", rb_cObject); 1601 rb_define_singleton_method(cVimWindow, "current", window_s_current, 0); 1602 rb_define_singleton_method(cVimWindow, "count", window_s_count, 0); 1603 rb_define_singleton_method(cVimWindow, "[]", window_s_aref, 1); 1604 rb_define_method(cVimWindow, "buffer", window_buffer, 0); 1605 rb_define_method(cVimWindow, "height", window_height, 0); 1606 rb_define_method(cVimWindow, "height=", window_set_height, 1); 1607 rb_define_method(cVimWindow, "width", window_width, 0); 1608 rb_define_method(cVimWindow, "width=", window_set_width, 1); 1609 rb_define_method(cVimWindow, "cursor", window_cursor, 0); 1610 rb_define_method(cVimWindow, "cursor=", window_set_cursor, 1); 1611 1612 rb_define_virtual_variable("$curbuf", buffer_s_current, 0); 1613 rb_define_virtual_variable("$curwin", window_s_current, 0); 1614 } 1615 1616 void vim_ruby_init(void *stack_start) 1617 { 1618 /* should get machine stack start address early in main function */ 1619 ruby_stack_start = stack_start; 1620 } 1621