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