From a93748acdb2e7bc4389b3738b4c787b764c3b2a6 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Sun, 7 Jul 2024 19:01:17 +0200
Subject: [PATCH] feat(load): fallback to regular session when branch session
 does not exist (yet)

---
 lua/persistence/init.lua | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/lua/persistence/init.lua b/lua/persistence/init.lua
index bf0be75..0f57b0e 100644
--- a/lua/persistence/init.lua
+++ b/lua/persistence/init.lua
@@ -7,9 +7,11 @@ M._active = false
 
 local e = vim.fn.fnameescape
 
-function M.current()
+---@param opts? {branch?: boolean}
+function M.current(opts)
+  opts = opts or {}
   local name = vim.fn.getcwd():gsub("[\\/:]+", "%%")
-  if Config.options.branch then
+  if Config.options.branch and opts.branch ~= false then
     local branch = M.branch()
     if branch and branch ~= "main" and branch ~= "master" then
       name = name .. "%%" .. branch:gsub("[\\/:]+", "%%")
@@ -71,7 +73,16 @@ end
 ---@param opts? { last?: boolean }
 function M.load(opts)
   opts = opts or {}
-  local file = opts.last and M.last() or M.current()
+  ---@type string
+  local file
+  if opts.last then
+    file = M.last()
+  else
+    file = M.current()
+    if vim.fn.filereadable(file) == 0 then
+      file = M.current({ branch = false })
+    end
+  end
   if file and vim.fn.filereadable(file) ~= 0 then
     M.fire("LoadPre")
     vim.cmd("silent! source " .. e(file))