1 /* vi:set ts=8 sts=4 sw=4 noet: 2 * 3 * VIM - Vi IMproved by Bram Moolenaar 4 * 5 * Do ":help uganda" in Vim to read copying and usage conditions. 6 * Do ":help credits" in Vim to see a list of people who contributed. 7 * See README.txt for an overview of the Vim source code. 8 */ 9 10 /* 11 * session.c: session related functions 12 */ 13 14 #include "vim.h" 15 16 #if defined(FEAT_SESSION) || defined(PROTO) 17 18 static int did_lcd; // whether ":lcd" was produced for a session 19 20 /* 21 * Write a file name to the session file. 22 * Takes care of the "slash" option in 'sessionoptions' and escapes special 23 * characters. 24 * Returns FAIL if writing fails or out of memory. 25 */ 26 static int 27 ses_put_fname(FILE *fd, char_u *name, unsigned *flagp) 28 { 29 char_u *sname; 30 char_u *p; 31 int retval = OK; 32 33 sname = home_replace_save(NULL, name); 34 if (sname == NULL) 35 return FAIL; 36 37 if (*flagp & SSOP_SLASH) 38 { 39 // change all backslashes to forward slashes 40 for (p = sname; *p != NUL; MB_PTR_ADV(p)) 41 if (*p == '\\') 42 *p = '/'; 43 } 44 45 // escape special characters 46 p = vim_strsave_fnameescape(sname, FALSE); 47 vim_free(sname); 48 if (p == NULL) 49 return FAIL; 50 51 // write the result 52 if (fputs((char *)p, fd) < 0) 53 retval = FAIL; 54 55 vim_free(p); 56 return retval; 57 } 58 59 /* 60 * Write a buffer name to the session file. 61 * Also ends the line, if "add_eol" is TRUE. 62 * Returns FAIL if writing fails. 63 */ 64 static int 65 ses_fname(FILE *fd, buf_T *buf, unsigned *flagp, int add_eol) 66 { 67 char_u *name; 68 69 // Use the short file name if the current directory is known at the time 70 // the session file will be sourced. 71 // Don't do this for ":mkview", we don't know the current directory. 72 // Don't do this after ":lcd", we don't keep track of what the current 73 // directory is. 74 if (buf->b_sfname != NULL 75 && flagp == &ssop_flags 76 && (ssop_flags & (SSOP_CURDIR | SSOP_SESDIR)) 77 #ifdef FEAT_AUTOCHDIR 78 && !p_acd 79 #endif 80 && !did_lcd) 81 name = buf->b_sfname; 82 else 83 name = buf->b_ffname; 84 if (ses_put_fname(fd, name, flagp) == FAIL 85 || (add_eol && put_eol(fd) == FAIL)) 86 return FAIL; 87 return OK; 88 } 89 90 /* 91 * Write an argument list to the session file. 92 * Returns FAIL if writing fails. 93 */ 94 static int 95 ses_arglist( 96 FILE *fd, 97 char *cmd, 98 garray_T *gap, 99 int fullname, // TRUE: use full path name 100 unsigned *flagp) 101 { 102 int i; 103 char_u *buf = NULL; 104 char_u *s; 105 106 if (fputs(cmd, fd) < 0 || put_eol(fd) == FAIL) 107 return FAIL; 108 if (put_line(fd, "%argdel") == FAIL) 109 return FAIL; 110 for (i = 0; i < gap->ga_len; ++i) 111 { 112 // NULL file names are skipped (only happens when out of memory). 113 s = alist_name(&((aentry_T *)gap->ga_data)[i]); 114 if (s != NULL) 115 { 116 if (fullname) 117 { 118 buf = alloc(MAXPATHL); 119 if (buf != NULL) 120 { 121 (void)vim_FullName(s, buf, MAXPATHL, FALSE); 122 s = buf; 123 } 124 } 125 if (fputs("$argadd ", fd) < 0 126 || ses_put_fname(fd, s, flagp) == FAIL 127 || put_eol(fd) == FAIL) 128 { 129 vim_free(buf); 130 return FAIL; 131 } 132 vim_free(buf); 133 } 134 } 135 return OK; 136 } 137 138 /* 139 * Return non-zero if window "wp" is to be stored in the Session. 140 */ 141 static int 142 ses_do_win(win_T *wp) 143 { 144 #ifdef FEAT_TERMINAL 145 if (bt_terminal(wp->w_buffer)) 146 return !term_is_finished(wp->w_buffer) 147 && (ssop_flags & SSOP_TERMINAL) 148 && term_should_restore(wp->w_buffer); 149 #endif 150 if (wp->w_buffer->b_fname == NULL 151 #ifdef FEAT_QUICKFIX 152 // When 'buftype' is "nofile" can't restore the window contents. 153 || bt_nofilename(wp->w_buffer) 154 #endif 155 ) 156 return (ssop_flags & SSOP_BLANK); 157 if (bt_help(wp->w_buffer)) 158 return (ssop_flags & SSOP_HELP); 159 return TRUE; 160 } 161 162 /* 163 * Return TRUE if frame "fr" has a window somewhere that we want to save in 164 * the Session. 165 */ 166 static int 167 ses_do_frame(frame_T *fr) 168 { 169 frame_T *frc; 170 171 if (fr->fr_layout == FR_LEAF) 172 return ses_do_win(fr->fr_win); 173 FOR_ALL_FRAMES(frc, fr->fr_child) 174 if (ses_do_frame(frc)) 175 return TRUE; 176 return FALSE; 177 } 178 179 /* 180 * Skip frames that don't contain windows we want to save in the Session. 181 * Returns NULL when there none. 182 */ 183 static frame_T * 184 ses_skipframe(frame_T *fr) 185 { 186 frame_T *frc; 187 188 FOR_ALL_FRAMES(frc, fr) 189 if (ses_do_frame(frc)) 190 break; 191 return frc; 192 } 193 194 /* 195 * Write commands to "fd" to recursively create windows for frame "fr", 196 * horizontally and vertically split. 197 * After the commands the last window in the frame is the current window. 198 * Returns FAIL when writing the commands to "fd" fails. 199 */ 200 static int 201 ses_win_rec(FILE *fd, frame_T *fr) 202 { 203 frame_T *frc; 204 int count = 0; 205 206 if (fr->fr_layout != FR_LEAF) 207 { 208 // Find first frame that's not skipped and then create a window for 209 // each following one (first frame is already there). 210 frc = ses_skipframe(fr->fr_child); 211 if (frc != NULL) 212 while ((frc = ses_skipframe(frc->fr_next)) != NULL) 213 { 214 // Make window as big as possible so that we have lots of room 215 // to split. 216 if (put_line(fd, "wincmd _ | wincmd |") == FAIL 217 || put_line(fd, fr->fr_layout == FR_COL 218 ? "split" : "vsplit") == FAIL) 219 return FAIL; 220 ++count; 221 } 222 223 // Go back to the first window. 224 if (count > 0 && (fprintf(fd, fr->fr_layout == FR_COL 225 ? "%dwincmd k" : "%dwincmd h", count) < 0 226 || put_eol(fd) == FAIL)) 227 return FAIL; 228 229 // Recursively create frames/windows in each window of this column or 230 // row. 231 frc = ses_skipframe(fr->fr_child); 232 while (frc != NULL) 233 { 234 ses_win_rec(fd, frc); 235 frc = ses_skipframe(frc->fr_next); 236 // Go to next window. 237 if (frc != NULL && put_line(fd, "wincmd w") == FAIL) 238 return FAIL; 239 } 240 } 241 return OK; 242 } 243 244 static int 245 ses_winsizes( 246 FILE *fd, 247 int restore_size, 248 win_T *tab_firstwin) 249 { 250 int n = 0; 251 win_T *wp; 252 253 if (restore_size && (ssop_flags & SSOP_WINSIZE)) 254 { 255 for (wp = tab_firstwin; wp != NULL; wp = wp->w_next) 256 { 257 if (!ses_do_win(wp)) 258 continue; 259 ++n; 260 261 // restore height when not full height 262 if (wp->w_height + wp->w_status_height < topframe->fr_height 263 && (fprintf(fd, 264 "exe '%dresize ' . ((&lines * %ld + %ld) / %ld)", 265 n, (long)wp->w_height, Rows / 2, Rows) < 0 266 || put_eol(fd) == FAIL)) 267 return FAIL; 268 269 // restore width when not full width 270 if (wp->w_width < Columns && (fprintf(fd, 271 "exe 'vert %dresize ' . ((&columns * %ld + %ld) / %ld)", 272 n, (long)wp->w_width, Columns / 2, Columns) < 0 273 || put_eol(fd) == FAIL)) 274 return FAIL; 275 } 276 } 277 else 278 { 279 // Just equalise window sizes 280 if (put_line(fd, "wincmd =") == FAIL) 281 return FAIL; 282 } 283 return OK; 284 } 285 286 static int 287 put_view_curpos(FILE *fd, win_T *wp, char *spaces) 288 { 289 int r; 290 291 if (wp->w_curswant == MAXCOL) 292 r = fprintf(fd, "%snormal! $", spaces); 293 else 294 r = fprintf(fd, "%snormal! 0%d|", spaces, wp->w_virtcol + 1); 295 return r < 0 || put_eol(fd) == FAIL ? FALSE : OK; 296 } 297 298 /* 299 * Write commands to "fd" to restore the view of a window. 300 * Caller must make sure 'scrolloff' is zero. 301 */ 302 static int 303 put_view( 304 FILE *fd, 305 win_T *wp, 306 int add_edit, // add ":edit" command to view 307 unsigned *flagp, // vop_flags or ssop_flags 308 int current_arg_idx, // current argument index of the window, 309 // use -1 if unknown 310 hashtab_T *terminal_bufs UNUSED) // already encountered terminal buffers, 311 // can be NULL 312 { 313 win_T *save_curwin; 314 int f; 315 int do_cursor; 316 int did_next = FALSE; 317 318 // Always restore cursor position for ":mksession". For ":mkview" only 319 // when 'viewoptions' contains "cursor". 320 do_cursor = (flagp == &ssop_flags || *flagp & SSOP_CURSOR); 321 322 // Local argument list. 323 if (wp->w_alist == &global_alist) 324 { 325 if (put_line(fd, "argglobal") == FAIL) 326 return FAIL; 327 } 328 else 329 { 330 if (ses_arglist(fd, "arglocal", &wp->w_alist->al_ga, 331 flagp == &vop_flags 332 || !(*flagp & SSOP_CURDIR) 333 || wp->w_localdir != NULL, flagp) == FAIL) 334 return FAIL; 335 } 336 337 // Only when part of a session: restore the argument index. Some 338 // arguments may have been deleted, check if the index is valid. 339 if (wp->w_arg_idx != current_arg_idx && wp->w_arg_idx < WARGCOUNT(wp) 340 && flagp == &ssop_flags) 341 { 342 if (fprintf(fd, "%ldargu", (long)wp->w_arg_idx + 1) < 0 343 || put_eol(fd) == FAIL) 344 return FAIL; 345 did_next = TRUE; 346 } 347 348 // Edit the file. Skip this when ":next" already did it. 349 if (add_edit && (!did_next || wp->w_arg_idx_invalid)) 350 { 351 # ifdef FEAT_TERMINAL 352 if (bt_terminal(wp->w_buffer)) 353 { 354 if (term_write_session(fd, wp, terminal_bufs) == FAIL) 355 return FAIL; 356 } 357 else 358 # endif 359 // Load the file. 360 if (wp->w_buffer->b_ffname != NULL 361 # ifdef FEAT_QUICKFIX 362 && !bt_nofilename(wp->w_buffer) 363 # endif 364 ) 365 { 366 // Editing a file in this buffer: use ":edit file". 367 // This may have side effects! (e.g., compressed or network file). 368 // 369 // Note, if a buffer for that file already exists, use :badd to 370 // edit that buffer, to not lose folding information (:edit resets 371 // folds in other buffers) 372 if (fputs("if bufexists(\"", fd) < 0 373 || ses_fname(fd, wp->w_buffer, flagp, FALSE) == FAIL 374 || fputs("\") | buffer ", fd) < 0 375 || ses_fname(fd, wp->w_buffer, flagp, FALSE) == FAIL 376 || fputs(" | else | edit ", fd) < 0 377 || ses_fname(fd, wp->w_buffer, flagp, FALSE) == FAIL 378 || fputs(" | endif", fd) < 0 379 || put_eol(fd) == FAIL) 380 return FAIL; 381 } 382 else 383 { 384 // No file in this buffer, just make it empty. 385 if (put_line(fd, "enew") == FAIL) 386 return FAIL; 387 #ifdef FEAT_QUICKFIX 388 if (wp->w_buffer->b_ffname != NULL) 389 { 390 // The buffer does have a name, but it's not a file name. 391 if (fputs("file ", fd) < 0 392 || ses_fname(fd, wp->w_buffer, flagp, TRUE) == FAIL) 393 return FAIL; 394 } 395 #endif 396 do_cursor = FALSE; 397 } 398 } 399 400 if (wp->w_alt_fnum) 401 { 402 buf_T *alt = buflist_findnr(wp->w_alt_fnum); 403 404 // Set the alternate file. 405 if ((flagp == &ssop_flags) 406 && alt != NULL 407 && alt->b_fname != NULL 408 && *alt->b_fname != NUL 409 && (fputs("balt ", fd) < 0 410 || ses_fname(fd, alt, flagp, TRUE) == FAIL)) 411 return FAIL; 412 } 413 414 // Local mappings and abbreviations. 415 if ((*flagp & (SSOP_OPTIONS | SSOP_LOCALOPTIONS)) 416 && makemap(fd, wp->w_buffer) == FAIL) 417 return FAIL; 418 419 // Local options. Need to go to the window temporarily. 420 // Store only local values when using ":mkview" and when ":mksession" is 421 // used and 'sessionoptions' doesn't include "options". 422 // Some folding options are always stored when "folds" is included, 423 // otherwise the folds would not be restored correctly. 424 save_curwin = curwin; 425 curwin = wp; 426 curbuf = curwin->w_buffer; 427 if (*flagp & (SSOP_OPTIONS | SSOP_LOCALOPTIONS)) 428 f = makeset(fd, OPT_LOCAL, 429 flagp == &vop_flags || !(*flagp & SSOP_OPTIONS)); 430 #ifdef FEAT_FOLDING 431 else if (*flagp & SSOP_FOLDS) 432 f = makefoldset(fd); 433 #endif 434 else 435 f = OK; 436 curwin = save_curwin; 437 curbuf = curwin->w_buffer; 438 if (f == FAIL) 439 return FAIL; 440 441 #ifdef FEAT_FOLDING 442 // Save Folds when 'buftype' is empty and for help files. 443 if ((*flagp & SSOP_FOLDS) 444 && wp->w_buffer->b_ffname != NULL 445 && (bt_normal(wp->w_buffer) || bt_help(wp->w_buffer))) 446 { 447 if (put_folds(fd, wp) == FAIL) 448 return FAIL; 449 } 450 #endif 451 452 // Set the cursor after creating folds, since that moves the cursor. 453 if (do_cursor) 454 { 455 456 // Restore the cursor line in the file and relatively in the 457 // window. Don't use "G", it changes the jumplist. 458 if (fprintf(fd, "let s:l = %ld - ((%ld * winheight(0) + %ld) / %ld)", 459 (long)wp->w_cursor.lnum, 460 (long)(wp->w_cursor.lnum - wp->w_topline), 461 (long)wp->w_height / 2, (long)wp->w_height) < 0 462 || put_eol(fd) == FAIL 463 || put_line(fd, "if s:l < 1 | let s:l = 1 | endif") == FAIL 464 || put_line(fd, "keepjumps exe s:l") == FAIL 465 || put_line(fd, "normal! zt") == FAIL 466 || fprintf(fd, "keepjumps %ld", (long)wp->w_cursor.lnum) < 0 467 || put_eol(fd) == FAIL) 468 return FAIL; 469 // Restore the cursor column and left offset when not wrapping. 470 if (wp->w_cursor.col == 0) 471 { 472 if (put_line(fd, "normal! 0") == FAIL) 473 return FAIL; 474 } 475 else 476 { 477 if (!wp->w_p_wrap && wp->w_leftcol > 0 && wp->w_width > 0) 478 { 479 if (fprintf(fd, 480 "let s:c = %ld - ((%ld * winwidth(0) + %ld) / %ld)", 481 (long)wp->w_virtcol + 1, 482 (long)(wp->w_virtcol - wp->w_leftcol), 483 (long)wp->w_width / 2, (long)wp->w_width) < 0 484 || put_eol(fd) == FAIL 485 || put_line(fd, "if s:c > 0") == FAIL 486 || fprintf(fd, 487 " exe 'normal! ' . s:c . '|zs' . %ld . '|'", 488 (long)wp->w_virtcol + 1) < 0 489 || put_eol(fd) == FAIL 490 || put_line(fd, "else") == FAIL 491 || put_view_curpos(fd, wp, " ") == FAIL 492 || put_line(fd, "endif") == FAIL) 493 return FAIL; 494 } 495 else if (put_view_curpos(fd, wp, "") == FAIL) 496 return FAIL; 497 } 498 } 499 500 // Local directory, if the current flag is not view options or the "curdir" 501 // option is included. 502 if (wp->w_localdir != NULL 503 && (flagp != &vop_flags || (*flagp & SSOP_CURDIR))) 504 { 505 if (fputs("lcd ", fd) < 0 506 || ses_put_fname(fd, wp->w_localdir, flagp) == FAIL 507 || put_eol(fd) == FAIL) 508 return FAIL; 509 did_lcd = TRUE; 510 } 511 512 return OK; 513 } 514 515 #ifdef FEAT_EVAL 516 static int 517 store_session_globals(FILE *fd) 518 { 519 hashtab_T *gvht = get_globvar_ht(); 520 hashitem_T *hi; 521 dictitem_T *this_var; 522 int todo; 523 char_u *p, *t; 524 525 todo = (int)gvht->ht_used; 526 for (hi = gvht->ht_array; todo > 0; ++hi) 527 { 528 if (!HASHITEM_EMPTY(hi)) 529 { 530 --todo; 531 this_var = HI2DI(hi); 532 if ((this_var->di_tv.v_type == VAR_NUMBER 533 || this_var->di_tv.v_type == VAR_STRING) 534 && var_flavour(this_var->di_key) == VAR_FLAVOUR_SESSION) 535 { 536 // Escape special characters with a backslash. Turn a LF and 537 // CR into \n and \r. 538 p = vim_strsave_escaped(tv_get_string(&this_var->di_tv), 539 (char_u *)"\\\"\n\r"); 540 if (p == NULL) // out of memory 541 break; 542 for (t = p; *t != NUL; ++t) 543 if (*t == '\n') 544 *t = 'n'; 545 else if (*t == '\r') 546 *t = 'r'; 547 if ((fprintf(fd, "let %s = %c%s%c", 548 this_var->di_key, 549 (this_var->di_tv.v_type == VAR_STRING) ? '"' 550 : ' ', 551 p, 552 (this_var->di_tv.v_type == VAR_STRING) ? '"' 553 : ' ') < 0) 554 || put_eol(fd) == FAIL) 555 { 556 vim_free(p); 557 return FAIL; 558 } 559 vim_free(p); 560 } 561 #ifdef FEAT_FLOAT 562 else if (this_var->di_tv.v_type == VAR_FLOAT 563 && var_flavour(this_var->di_key) == VAR_FLAVOUR_SESSION) 564 { 565 float_T f = this_var->di_tv.vval.v_float; 566 int sign = ' '; 567 568 if (f < 0) 569 { 570 f = -f; 571 sign = '-'; 572 } 573 if ((fprintf(fd, "let %s = %c%f", 574 this_var->di_key, sign, f) < 0) 575 || put_eol(fd) == FAIL) 576 return FAIL; 577 } 578 #endif 579 } 580 } 581 return OK; 582 } 583 #endif 584 585 /* 586 * Write openfile commands for the current buffers to an .exrc file. 587 * Return FAIL on error, OK otherwise. 588 */ 589 static int 590 makeopens( 591 FILE *fd, 592 char_u *dirnow) // Current directory name 593 { 594 buf_T *buf; 595 int only_save_windows = TRUE; 596 int nr; 597 int restore_size = TRUE; 598 win_T *wp; 599 char_u *sname; 600 win_T *edited_win = NULL; 601 int tabnr; 602 int restore_stal = FALSE; 603 win_T *tab_firstwin; 604 frame_T *tab_topframe; 605 int cur_arg_idx = 0; 606 int next_arg_idx = 0; 607 int ret = FAIL; 608 #ifdef FEAT_TERMINAL 609 hashtab_T terminal_bufs; 610 611 hash_init(&terminal_bufs); 612 #endif 613 614 if (ssop_flags & SSOP_BUFFERS) 615 only_save_windows = FALSE; // Save ALL buffers 616 617 // Begin by setting the this_session variable, and then other 618 // sessionable variables. 619 #ifdef FEAT_EVAL 620 if (put_line(fd, "let v:this_session=expand(\"<sfile>:p\")") == FAIL) 621 goto fail; 622 if (ssop_flags & SSOP_GLOBALS) 623 if (store_session_globals(fd) == FAIL) 624 goto fail; 625 #endif 626 627 // Close all windows and tabs but one. 628 if (put_line(fd, "silent only") == FAIL) 629 goto fail; 630 if ((ssop_flags & SSOP_TABPAGES) 631 && put_line(fd, "silent tabonly") == FAIL) 632 goto fail; 633 634 // Now a :cd command to the session directory or the current directory 635 if (ssop_flags & SSOP_SESDIR) 636 { 637 if (put_line(fd, "exe \"cd \" . escape(expand(\"<sfile>:p:h\"), ' ')") 638 == FAIL) 639 goto fail; 640 } 641 else if (ssop_flags & SSOP_CURDIR) 642 { 643 sname = home_replace_save(NULL, globaldir != NULL ? globaldir : dirnow); 644 if (sname == NULL 645 || fputs("cd ", fd) < 0 646 || ses_put_fname(fd, sname, &ssop_flags) == FAIL 647 || put_eol(fd) == FAIL) 648 { 649 vim_free(sname); 650 goto fail; 651 } 652 vim_free(sname); 653 } 654 655 // If there is an empty, unnamed buffer we will wipe it out later. 656 // Remember the buffer number. 657 if (put_line(fd, "if expand('%') == '' && !&modified && line('$') <= 1 && getline(1) == ''") == FAIL) 658 goto fail; 659 if (put_line(fd, " let s:wipebuf = bufnr('%')") == FAIL) 660 goto fail; 661 if (put_line(fd, "endif") == FAIL) 662 goto fail; 663 664 // Now save the current files, current buffer first. 665 if (put_line(fd, "set shortmess=aoO") == FAIL) 666 goto fail; 667 668 // the global argument list 669 if (ses_arglist(fd, "argglobal", &global_alist.al_ga, 670 !(ssop_flags & SSOP_CURDIR), &ssop_flags) == FAIL) 671 goto fail; 672 673 if (ssop_flags & SSOP_RESIZE) 674 { 675 // Note: after the restore we still check it worked! 676 if (fprintf(fd, "set lines=%ld columns=%ld" , Rows, Columns) < 0 677 || put_eol(fd) == FAIL) 678 goto fail; 679 } 680 681 #ifdef FEAT_GUI 682 if (gui.in_use && (ssop_flags & SSOP_WINPOS)) 683 { 684 int x, y; 685 686 if (gui_mch_get_winpos(&x, &y) == OK) 687 { 688 // Note: after the restore we still check it worked! 689 if (fprintf(fd, "winpos %d %d", x, y) < 0 || put_eol(fd) == FAIL) 690 goto fail; 691 } 692 } 693 #endif 694 695 // When there are two or more tabpages and 'showtabline' is 1 the tabline 696 // will be displayed when creating the next tab. That resizes the windows 697 // in the first tab, which may cause problems. Set 'showtabline' to 2 698 // temporarily to avoid that. 699 if (p_stal == 1 && first_tabpage->tp_next != NULL) 700 { 701 if (put_line(fd, "set stal=2") == FAIL) 702 goto fail; 703 restore_stal = TRUE; 704 } 705 706 // May repeat putting Windows for each tab, when "tabpages" is in 707 // 'sessionoptions'. 708 // Don't use goto_tabpage(), it may change directory and trigger 709 // autocommands. 710 tab_firstwin = firstwin; // first window in tab page "tabnr" 711 tab_topframe = topframe; 712 if ((ssop_flags & SSOP_TABPAGES)) 713 { 714 tabpage_T *tp; 715 716 // Similar to ses_win_rec() below, populate the tab pages first so 717 // later local options won't be copied to the new tabs. 718 FOR_ALL_TABPAGES(tp) 719 if (tp->tp_next != NULL && put_line(fd, "tabnew") == FAIL) 720 goto fail; 721 if (first_tabpage->tp_next != NULL && put_line(fd, "tabrewind") == FAIL) 722 goto fail; 723 } 724 for (tabnr = 1; ; ++tabnr) 725 { 726 tabpage_T *tp = NULL; 727 int need_tabnext = FALSE; 728 int cnr = 1; 729 730 if ((ssop_flags & SSOP_TABPAGES)) 731 { 732 tp = find_tabpage(tabnr); 733 734 if (tp == NULL) 735 break; // done all tab pages 736 if (tp == curtab) 737 { 738 tab_firstwin = firstwin; 739 tab_topframe = topframe; 740 } 741 else 742 { 743 tab_firstwin = tp->tp_firstwin; 744 tab_topframe = tp->tp_topframe; 745 } 746 if (tabnr > 1) 747 need_tabnext = TRUE; 748 } 749 750 // Before creating the window layout, try loading one file. If this 751 // is aborted we don't end up with a number of useless windows. 752 // This may have side effects! (e.g., compressed or network file). 753 for (wp = tab_firstwin; wp != NULL; wp = wp->w_next) 754 { 755 if (ses_do_win(wp) 756 && wp->w_buffer->b_ffname != NULL 757 && !bt_help(wp->w_buffer) 758 #ifdef FEAT_QUICKFIX 759 && !bt_nofilename(wp->w_buffer) 760 #endif 761 ) 762 { 763 if (need_tabnext && put_line(fd, "tabnext") == FAIL) 764 goto fail; 765 need_tabnext = FALSE; 766 767 if (fputs("edit ", fd) < 0 768 || ses_fname(fd, wp->w_buffer, &ssop_flags, TRUE) 769 == FAIL) 770 goto fail; 771 if (!wp->w_arg_idx_invalid) 772 edited_win = wp; 773 break; 774 } 775 } 776 777 // If no file got edited create an empty tab page. 778 if (need_tabnext && put_line(fd, "tabnext") == FAIL) 779 goto fail; 780 781 // Save current window layout. 782 if (put_line(fd, "set splitbelow splitright") == FAIL) 783 goto fail; 784 if (ses_win_rec(fd, tab_topframe) == FAIL) 785 goto fail; 786 if (!p_sb && put_line(fd, "set nosplitbelow") == FAIL) 787 goto fail; 788 if (!p_spr && put_line(fd, "set nosplitright") == FAIL) 789 goto fail; 790 791 // Check if window sizes can be restored (no windows omitted). 792 // Remember the window number of the current window after restoring. 793 nr = 0; 794 for (wp = tab_firstwin; wp != NULL; wp = W_NEXT(wp)) 795 { 796 if (ses_do_win(wp)) 797 ++nr; 798 else 799 restore_size = FALSE; 800 if (curwin == wp) 801 cnr = nr; 802 } 803 804 // Go to the first window. 805 if (put_line(fd, "wincmd t") == FAIL) 806 goto fail; 807 808 // If more than one window, see if sizes can be restored. 809 // First set 'winheight' and 'winwidth' to 1 to avoid the windows being 810 // resized when moving between windows. 811 // Do this before restoring the view, so that the topline and the 812 // cursor can be set. This is done again below. 813 // winminheight and winminwidth need to be set to avoid an error if the 814 // user has set winheight or winwidth. 815 if (put_line(fd, "set winminheight=0") == FAIL 816 || put_line(fd, "set winheight=1") == FAIL 817 || put_line(fd, "set winminwidth=0") == FAIL 818 || put_line(fd, "set winwidth=1") == FAIL) 819 goto fail; 820 if (nr > 1 && ses_winsizes(fd, restore_size, tab_firstwin) == FAIL) 821 goto fail; 822 823 // Restore the tab-local working directory if specified 824 // Do this before the windows, so that the window-local directory can 825 // override the tab-local directory. 826 if (tp != NULL && tp->tp_localdir != NULL && ssop_flags & SSOP_CURDIR) 827 { 828 if (fputs("tcd ", fd) < 0 829 || ses_put_fname(fd, tp->tp_localdir, &ssop_flags) == FAIL 830 || put_eol(fd) == FAIL) 831 goto fail; 832 did_lcd = TRUE; 833 } 834 835 // Restore the view of the window (options, file, cursor, etc.). 836 for (wp = tab_firstwin; wp != NULL; wp = wp->w_next) 837 { 838 if (!ses_do_win(wp)) 839 continue; 840 if (put_view(fd, wp, wp != edited_win, &ssop_flags, cur_arg_idx, 841 #ifdef FEAT_TERMINAL 842 &terminal_bufs 843 #else 844 NULL 845 #endif 846 ) == FAIL) 847 goto fail; 848 if (nr > 1 && put_line(fd, "wincmd w") == FAIL) 849 goto fail; 850 next_arg_idx = wp->w_arg_idx; 851 } 852 853 // The argument index in the first tab page is zero, need to set it in 854 // each window. For further tab pages it's the window where we do 855 // "tabedit". 856 cur_arg_idx = next_arg_idx; 857 858 // Restore cursor to the current window if it's not the first one. 859 if (cnr > 1 && (fprintf(fd, "%dwincmd w", cnr) < 0 860 || put_eol(fd) == FAIL)) 861 goto fail; 862 863 // Restore window sizes again after jumping around in windows, because 864 // the current window has a minimum size while others may not. 865 if (nr > 1 && ses_winsizes(fd, restore_size, tab_firstwin) == FAIL) 866 goto fail; 867 868 // Don't continue in another tab page when doing only the current one 869 // or when at the last tab page. 870 if (!(ssop_flags & SSOP_TABPAGES)) 871 break; 872 } 873 874 if (ssop_flags & SSOP_TABPAGES) 875 { 876 if (fprintf(fd, "tabnext %d", tabpage_index(curtab)) < 0 877 || put_eol(fd) == FAIL) 878 goto fail; 879 } 880 if (restore_stal && put_line(fd, "set stal=1") == FAIL) 881 goto fail; 882 883 // Now put the remaining buffers into the buffer list. 884 // This is near the end, so that when 'hidden' is set we don't create extra 885 // buffers. If the buffer was already created with another command the 886 // ":badd" will have no effect. 887 FOR_ALL_BUFFERS(buf) 888 { 889 if (!(only_save_windows && buf->b_nwindows == 0) 890 && !(buf->b_help && !(ssop_flags & SSOP_HELP)) 891 #ifdef FEAT_TERMINAL 892 // Skip terminal buffers: finished ones are not useful, others 893 // will be resurrected and result in a new buffer. 894 && !bt_terminal(buf) 895 #endif 896 && buf->b_fname != NULL 897 && buf->b_p_bl) 898 { 899 if (fprintf(fd, "badd +%ld ", buf->b_wininfo == NULL ? 1L 900 : buf->b_wininfo->wi_fpos.lnum) < 0 901 || ses_fname(fd, buf, &ssop_flags, TRUE) == FAIL) 902 goto fail; 903 } 904 } 905 906 // Wipe out an empty unnamed buffer we started in. 907 if (put_line(fd, "if exists('s:wipebuf') && len(win_findbuf(s:wipebuf)) == 0") 908 == FAIL) 909 goto fail; 910 if (put_line(fd, " silent exe 'bwipe ' . s:wipebuf") == FAIL) 911 goto fail; 912 if (put_line(fd, "endif") == FAIL) 913 goto fail; 914 if (put_line(fd, "unlet! s:wipebuf") == FAIL) 915 goto fail; 916 917 // Re-apply 'winheight', 'winwidth' and 'shortmess'. 918 if (fprintf(fd, "set winheight=%ld winwidth=%ld shortmess=%s", 919 p_wh, p_wiw, p_shm) < 0 || put_eol(fd) == FAIL) 920 goto fail; 921 // Re-apply 'winminheight' and 'winminwidth'. 922 if (fprintf(fd, "set winminheight=%ld winminwidth=%ld", 923 p_wmh, p_wmw) < 0 || put_eol(fd) == FAIL) 924 goto fail; 925 926 // Lastly, execute the x.vim file if it exists. 927 if (put_line(fd, "let s:sx = expand(\"<sfile>:p:r\").\"x.vim\"") == FAIL 928 || put_line(fd, "if filereadable(s:sx)") == FAIL 929 || put_line(fd, " exe \"source \" . fnameescape(s:sx)") == FAIL 930 || put_line(fd, "endif") == FAIL) 931 goto fail; 932 933 ret = OK; 934 fail: 935 #ifdef FEAT_TERMINAL 936 hash_clear_all(&terminal_bufs, 0); 937 #endif 938 return ret; 939 } 940 941 /* 942 * Get the name of the view file for the current buffer. 943 */ 944 static char_u * 945 get_view_file(int c) 946 { 947 int len = 0; 948 char_u *p, *s; 949 char_u *retval; 950 char_u *sname; 951 952 if (curbuf->b_ffname == NULL) 953 { 954 emsg(_(e_noname)); 955 return NULL; 956 } 957 sname = home_replace_save(NULL, curbuf->b_ffname); 958 if (sname == NULL) 959 return NULL; 960 961 // We want a file name without separators, because we're not going to make 962 // a directory. 963 // "normal" path separator -> "=+" 964 // "=" -> "==" 965 // ":" path separator -> "=-" 966 for (p = sname; *p; ++p) 967 if (*p == '=' || vim_ispathsep(*p)) 968 ++len; 969 retval = alloc(STRLEN(sname) + len + STRLEN(p_vdir) + 9); 970 if (retval != NULL) 971 { 972 STRCPY(retval, p_vdir); 973 add_pathsep(retval); 974 s = retval + STRLEN(retval); 975 for (p = sname; *p; ++p) 976 { 977 if (*p == '=') 978 { 979 *s++ = '='; 980 *s++ = '='; 981 } 982 else if (vim_ispathsep(*p)) 983 { 984 *s++ = '='; 985 #if defined(BACKSLASH_IN_FILENAME) || defined(AMIGA) || defined(VMS) 986 if (*p == ':') 987 *s++ = '-'; 988 else 989 #endif 990 *s++ = '+'; 991 } 992 else 993 *s++ = *p; 994 } 995 *s++ = '='; 996 *s++ = c; 997 STRCPY(s, ".vim"); 998 } 999 1000 vim_free(sname); 1001 return retval; 1002 } 1003 1004 /* 1005 * ":loadview [nr]" 1006 */ 1007 void 1008 ex_loadview(exarg_T *eap) 1009 { 1010 char_u *fname; 1011 1012 fname = get_view_file(*eap->arg); 1013 if (fname != NULL) 1014 { 1015 do_source(fname, FALSE, DOSO_NONE, NULL); 1016 vim_free(fname); 1017 } 1018 } 1019 1020 # if defined(FEAT_GUI_GNOME) \ 1021 || (defined(GUI_MAY_SPAWN) && defined(EXPERIMENTAL_GUI_CMD)) \ 1022 || defined(PROTO) 1023 /* 1024 * Generate a script that can be used to restore the current editing session. 1025 * Save the value of v:this_session before running :mksession in order to make 1026 * automagic session save fully transparent. Return TRUE on success. 1027 */ 1028 int 1029 write_session_file(char_u *filename) 1030 { 1031 char_u *escaped_filename; 1032 char *mksession_cmdline; 1033 unsigned int save_ssop_flags; 1034 int failed; 1035 1036 // Build an ex command line to create a script that restores the current 1037 // session if executed. Escape the filename to avoid nasty surprises. 1038 escaped_filename = vim_strsave_escaped(filename, escape_chars); 1039 if (escaped_filename == NULL) 1040 return FALSE; 1041 mksession_cmdline = alloc(10 + (int)STRLEN(escaped_filename) + 1); 1042 if (mksession_cmdline == NULL) 1043 { 1044 vim_free(escaped_filename); 1045 return FALSE; 1046 } 1047 strcpy(mksession_cmdline, "mksession "); 1048 STRCAT(mksession_cmdline, escaped_filename); 1049 vim_free(escaped_filename); 1050 1051 // Use a reasonable hardcoded set of 'sessionoptions' flags to avoid 1052 // unpredictable effects when the session is saved automatically. Also, 1053 // we definitely need SSOP_GLOBALS to be able to restore v:this_session. 1054 // Don't use SSOP_BUFFERS to prevent the buffer list from becoming 1055 // enormously large if the GNOME session feature is used regularly. 1056 save_ssop_flags = ssop_flags; 1057 ssop_flags = (SSOP_BLANK|SSOP_CURDIR|SSOP_FOLDS|SSOP_GLOBALS 1058 |SSOP_HELP|SSOP_OPTIONS|SSOP_WINSIZE|SSOP_TABPAGES); 1059 1060 do_cmdline_cmd((char_u *)"let Save_VV_this_session = v:this_session"); 1061 failed = (do_cmdline_cmd((char_u *)mksession_cmdline) == FAIL); 1062 do_cmdline_cmd((char_u *)"let v:this_session = Save_VV_this_session"); 1063 do_unlet((char_u *)"Save_VV_this_session", TRUE); 1064 1065 ssop_flags = save_ssop_flags; 1066 vim_free(mksession_cmdline); 1067 1068 // Reopen the file and append a command to restore v:this_session, 1069 // as if this save never happened. This is to avoid conflicts with 1070 // the user's own sessions. FIXME: It's probably less hackish to add 1071 // a "stealth" flag to 'sessionoptions' -- gotta ask Bram. 1072 if (!failed) 1073 { 1074 FILE *fd; 1075 1076 fd = open_exfile(filename, TRUE, APPENDBIN); 1077 1078 failed = (fd == NULL 1079 || put_line(fd, "let v:this_session = Save_VV_this_session") 1080 == FAIL 1081 || put_line(fd, "unlet Save_VV_this_session") == FAIL); 1082 1083 if (fd != NULL && fclose(fd) != 0) 1084 failed = TRUE; 1085 1086 if (failed) 1087 mch_remove(filename); 1088 } 1089 1090 return !failed; 1091 } 1092 # endif 1093 1094 #endif // FEAT_SESSION 1095 1096 #if defined(FEAT_SESSION) && defined(USE_CRNL) 1097 # define MKSESSION_NL 1098 static int mksession_nl = FALSE; // use NL only in put_eol() 1099 #endif 1100 1101 /* 1102 * ":mkexrc", ":mkvimrc", ":mkview" and ":mksession". 1103 */ 1104 void 1105 ex_mkrc(exarg_T *eap) 1106 { 1107 FILE *fd; 1108 int failed = FALSE; 1109 char_u *fname; 1110 #ifdef FEAT_BROWSE 1111 char_u *browseFile = NULL; 1112 #endif 1113 #ifdef FEAT_SESSION 1114 int view_session = FALSE; 1115 int using_vdir = FALSE; // using 'viewdir'? 1116 char_u *viewFile = NULL; 1117 unsigned *flagp; 1118 #endif 1119 1120 if (eap->cmdidx == CMD_mksession || eap->cmdidx == CMD_mkview) 1121 { 1122 #ifdef FEAT_SESSION 1123 view_session = TRUE; 1124 #else 1125 ex_ni(eap); 1126 return; 1127 #endif 1128 } 1129 1130 #ifdef FEAT_SESSION 1131 // Use the short file name until ":lcd" is used. We also don't use the 1132 // short file name when 'acd' is set, that is checked later. 1133 did_lcd = FALSE; 1134 1135 // ":mkview" or ":mkview 9": generate file name with 'viewdir' 1136 if (eap->cmdidx == CMD_mkview 1137 && (*eap->arg == NUL 1138 || (vim_isdigit(*eap->arg) && eap->arg[1] == NUL))) 1139 { 1140 eap->forceit = TRUE; 1141 fname = get_view_file(*eap->arg); 1142 if (fname == NULL) 1143 return; 1144 viewFile = fname; 1145 using_vdir = TRUE; 1146 } 1147 else 1148 #endif 1149 if (*eap->arg != NUL) 1150 fname = eap->arg; 1151 else if (eap->cmdidx == CMD_mkvimrc) 1152 fname = (char_u *)VIMRC_FILE; 1153 #ifdef FEAT_SESSION 1154 else if (eap->cmdidx == CMD_mksession) 1155 fname = (char_u *)SESSION_FILE; 1156 #endif 1157 else 1158 fname = (char_u *)EXRC_FILE; 1159 1160 #ifdef FEAT_BROWSE 1161 if (cmdmod.cmod_flags & CMOD_BROWSE) 1162 { 1163 browseFile = do_browse(BROWSE_SAVE, 1164 # ifdef FEAT_SESSION 1165 eap->cmdidx == CMD_mkview ? (char_u *)_("Save View") : 1166 eap->cmdidx == CMD_mksession ? (char_u *)_("Save Session") : 1167 # endif 1168 (char_u *)_("Save Setup"), 1169 fname, (char_u *)"vim", NULL, 1170 (char_u *)_(BROWSE_FILTER_MACROS), NULL); 1171 if (browseFile == NULL) 1172 goto theend; 1173 fname = browseFile; 1174 eap->forceit = TRUE; // since dialog already asked 1175 } 1176 #endif 1177 1178 #if defined(FEAT_SESSION) && defined(vim_mkdir) 1179 // When using 'viewdir' may have to create the directory. 1180 if (using_vdir && !mch_isdir(p_vdir)) 1181 vim_mkdir_emsg(p_vdir, 0755); 1182 #endif 1183 1184 fd = open_exfile(fname, eap->forceit, WRITEBIN); 1185 if (fd != NULL) 1186 { 1187 #ifdef FEAT_SESSION 1188 if (eap->cmdidx == CMD_mkview) 1189 flagp = &vop_flags; 1190 else 1191 flagp = &ssop_flags; 1192 #endif 1193 1194 #ifdef MKSESSION_NL 1195 // "unix" in 'sessionoptions': use NL line separator 1196 if (view_session && (*flagp & SSOP_UNIX)) 1197 mksession_nl = TRUE; 1198 #endif 1199 1200 // Write the version command for :mkvimrc 1201 if (eap->cmdidx == CMD_mkvimrc) 1202 (void)put_line(fd, "version 6.0"); 1203 1204 #ifdef FEAT_SESSION 1205 if (eap->cmdidx == CMD_mksession) 1206 { 1207 if (put_line(fd, "let SessionLoad = 1") == FAIL) 1208 failed = TRUE; 1209 } 1210 1211 if (eap->cmdidx != CMD_mkview) 1212 #endif 1213 { 1214 // Write setting 'compatible' first, because it has side effects. 1215 // For that same reason only do it when needed. 1216 if (p_cp) 1217 (void)put_line(fd, "if !&cp | set cp | endif"); 1218 else 1219 (void)put_line(fd, "if &cp | set nocp | endif"); 1220 } 1221 1222 #ifdef FEAT_SESSION 1223 if (!view_session 1224 || (eap->cmdidx == CMD_mksession 1225 && (*flagp & SSOP_OPTIONS))) 1226 #endif 1227 failed |= (makemap(fd, NULL) == FAIL 1228 || makeset(fd, OPT_GLOBAL, FALSE) == FAIL); 1229 1230 #ifdef FEAT_SESSION 1231 if (!failed && view_session) 1232 { 1233 if (put_line(fd, "let s:so_save = &g:so | let s:siso_save = &g:siso | setg so=0 siso=0 | setl so=-1 siso=-1") == FAIL) 1234 failed = TRUE; 1235 if (eap->cmdidx == CMD_mksession) 1236 { 1237 char_u *dirnow; // current directory 1238 1239 dirnow = alloc(MAXPATHL); 1240 if (dirnow == NULL) 1241 failed = TRUE; 1242 else 1243 { 1244 // Change to session file's dir. 1245 if (mch_dirname(dirnow, MAXPATHL) == FAIL 1246 || mch_chdir((char *)dirnow) != 0) 1247 *dirnow = NUL; 1248 if (*dirnow != NUL && (ssop_flags & SSOP_SESDIR)) 1249 { 1250 if (vim_chdirfile(fname, NULL) == OK) 1251 shorten_fnames(TRUE); 1252 } 1253 else if (*dirnow != NUL 1254 && (ssop_flags & SSOP_CURDIR) && globaldir != NULL) 1255 { 1256 if (mch_chdir((char *)globaldir) == 0) 1257 shorten_fnames(TRUE); 1258 } 1259 1260 failed |= (makeopens(fd, dirnow) == FAIL); 1261 1262 // restore original dir 1263 if (*dirnow != NUL && ((ssop_flags & SSOP_SESDIR) 1264 || ((ssop_flags & SSOP_CURDIR) && globaldir != NULL))) 1265 { 1266 if (mch_chdir((char *)dirnow) != 0) 1267 emsg(_(e_prev_dir)); 1268 shorten_fnames(TRUE); 1269 } 1270 vim_free(dirnow); 1271 } 1272 } 1273 else 1274 { 1275 failed |= (put_view(fd, curwin, !using_vdir, flagp, -1, NULL) 1276 == FAIL); 1277 } 1278 if (put_line(fd, "let &g:so = s:so_save | let &g:siso = s:siso_save") 1279 == FAIL) 1280 failed = TRUE; 1281 #ifdef FEAT_SEARCH_EXTRA 1282 if (no_hlsearch && put_line(fd, "nohlsearch") == FAIL) 1283 failed = TRUE; 1284 #endif 1285 if (put_line(fd, "doautoall SessionLoadPost") == FAIL) 1286 failed = TRUE; 1287 if (eap->cmdidx == CMD_mksession) 1288 { 1289 if (put_line(fd, "unlet SessionLoad") == FAIL) 1290 failed = TRUE; 1291 } 1292 } 1293 #endif 1294 if (put_line(fd, "\" vim: set ft=vim :") == FAIL) 1295 failed = TRUE; 1296 1297 failed |= fclose(fd); 1298 1299 if (failed) 1300 emsg(_(e_write)); 1301 #if defined(FEAT_SESSION) 1302 else if (eap->cmdidx == CMD_mksession) 1303 { 1304 // successful session write - set this_session var 1305 char_u *tbuf; 1306 1307 tbuf = alloc(MAXPATHL); 1308 if (tbuf != NULL) 1309 { 1310 if (vim_FullName(fname, tbuf, MAXPATHL, FALSE) == OK) 1311 set_vim_var_string(VV_THIS_SESSION, tbuf, -1); 1312 vim_free(tbuf); 1313 } 1314 } 1315 #endif 1316 #ifdef MKSESSION_NL 1317 mksession_nl = FALSE; 1318 #endif 1319 } 1320 1321 #ifdef FEAT_BROWSE 1322 theend: 1323 vim_free(browseFile); 1324 #endif 1325 #ifdef FEAT_SESSION 1326 vim_free(viewFile); 1327 #endif 1328 } 1329 1330 #if (defined(FEAT_VIMINFO) || defined(FEAT_SESSION)) || defined(PROTO) 1331 var_flavour_T 1332 var_flavour(char_u *varname) 1333 { 1334 char_u *p = varname; 1335 1336 if (ASCII_ISUPPER(*p)) 1337 { 1338 while (*(++p)) 1339 if (ASCII_ISLOWER(*p)) 1340 return VAR_FLAVOUR_SESSION; 1341 return VAR_FLAVOUR_VIMINFO; 1342 } 1343 else 1344 return VAR_FLAVOUR_DEFAULT; 1345 } 1346 #endif 1347 1348 /* 1349 * Write end-of-line character(s) for ":mkexrc", ":mkvimrc" and ":mksession". 1350 * Return FAIL for a write error. 1351 */ 1352 int 1353 put_eol(FILE *fd) 1354 { 1355 if ( 1356 #ifdef USE_CRNL 1357 ( 1358 # ifdef MKSESSION_NL 1359 !mksession_nl && 1360 # endif 1361 (putc('\r', fd) < 0)) || 1362 #endif 1363 (putc('\n', fd) < 0)) 1364 return FAIL; 1365 return OK; 1366 } 1367 1368 /* 1369 * Write a line to "fd". 1370 * Return FAIL for a write error. 1371 */ 1372 int 1373 put_line(FILE *fd, char *s) 1374 { 1375 if (fputs(s, fd) < 0 || put_eol(fd) == FAIL) 1376 return FAIL; 1377 return OK; 1378 } 1379