From a8d388f9568908634170fbf8f808334052ea496e Mon Sep 17 00:00:00 2001
From: Marco Hinz <mh.codebro@gmail.com>
Date: Tue, 23 Apr 2013 18:00:42 +0200
Subject: [PATCH] refactor load/save_session()

---
 autoload/startify.vim | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/autoload/startify.vim b/autoload/startify.vim
index 19e23fd..2596dc9 100644
--- a/autoload/startify.vim
+++ b/autoload/startify.vim
@@ -8,26 +8,26 @@ if exists('g:autoloaded_startify') || &cp
 endif
 let g:autoloaded_startify = 1
 
-function! startify#save_session() abort
-  let spath = g:startify_session_dir .'/'. input('Save under this session name: ') | redraw
-  if filereadable(spath)
-    echo 'Session already exists. Overwrite?  [y/n]' | redraw
-    if nr2char(getchar()) == 'y'
-      execute 'mksession! '. spath | echo 'Session saved under: '. spath
-    else
-      echo 'Did NOT save the session!'
-    endif
-  else
+function! startify#save_session(...) abort
+  let spath = g:startify_session_dir .'/'. (exists('a:1') ? a:1 : input('Save under this session name: ')) | redraw
+  if !filereadable(spath)
     execute 'mksession '. spath | echo 'Session saved under: '. spath
+    return
+  endif
+  echo 'Session already exists. Overwrite?  [y/n]' | redraw
+  if nr2char(getchar()) == 'y'
+    execute 'mksession! '. spath | echo 'Session saved under: '. spath
+  else
+    echo 'Did NOT save the session!'
   endif
 endfunction
 
-function! startify#load_session() abort
-  let spath = g:startify_session_dir .'/'. input('Load this session: ') | redraw
+function! startify#load_session(...) abort
+  let spath = g:startify_session_dir .'/'. (exists('a:1') ? a:1 : input('Load this session: ')) | redraw
   if filereadable(spath)
     execute 'source '. spath
   else
-    echo 'No such file!'
+    echo 'No such file: '. spath
   endif
 endfunction