F/ordered autoload

lua/session_manager/init.lua

@@ -46,7 +46,9 @@ function session_manager.load_last_session(discard_current)
   local last_session = utils.get_last_session_filename()
   if last_session then
     utils.load_session(last_session, discard_current)
+    return true
   end
+  return false
 end
 
 --- Loads a session for the current working directory.
@@ -56,8 +58,10 @@ function session_manager.load_current_dir_session(discard_current)
     local session = config.dir_to_session_filename(cwd)
     if session:exists() then
       utils.load_session(session.filename, discard_current)
+      return true
     end
   end
+  return false
 end
 
 --- Saves a session for the current working directory.
@@ -68,13 +72,31 @@ function session_manager.save_current_session()
   end
 end
 
+local get_autoload_mode = function()
+  if config.autoload_mode == AutoloadMode.Disabled or
+      vim.fn.argc() > 0
+      or vim.g.started_with_stdin then
+    return { AutoloadMode.Disabled }
+  end
+  if config.autoload_mode == AutoloadMode.CurrentDir or
+      config.autoload_mode == AutoloadMode.LastSession then
+    -- Don't break existing configurations
+    return { config.autoload_mode }
+  end
+  return config.autoload_mode
+end
+
+local autoloaders = {
+  [AutoloadMode.Disabled] = function() return true end,
+  [AutoloadMode.CurrentDir] = session_manager.load_current_dir_session,
+  [AutoloadMode.LastSession] = session_manager.load_last_session,
+}
+
 --- Loads a session based on settings. Executed after starting the editor.
 function session_manager.autoload_session()
-  if config.autoload_mode ~= AutoloadMode.Disabled and vim.fn.argc() == 0 and not vim.g.started_with_stdin then
-    if config.autoload_mode == AutoloadMode.CurrentDir then
-      session_manager.load_current_dir_session()
-    elseif config.autoload_mode == AutoloadMode.LastSession then
-      session_manager.load_last_session()
+  for _, mode in ipairs(get_autoload_mode()) do
+    if autoloaders[mode]() then
+      return
     end
   end
 end