fix(project_diff_preview): fix toggle issues

tanvirtin · Nov 21, 2024 · f30e62a · f30e62a
1 parent 3e3403c
commit f30e62a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lua/vgit/features/screens/ProjectDiffScreen/Store.lua b/lua/vgit/features/screens/ProjectDiffScreen/Store.lua
@@ -142,6 +142,7 @@ end
 function Store:get_filepath()
   local reponame = self.state.reponame
   local filename = self:get_filename()
+  if not filename then return nil, { 'entry not found' } end
 
   filename = fs.make_relative(reponame, filename)
   filename = string.format('%s/%s', reponame, filename)

diff --git a/lua/vgit/features/screens/ProjectDiffScreen/init.lua b/lua/vgit/features/screens/ProjectDiffScreen/init.lua
@@ -395,7 +395,7 @@ function ProjectDiffScreen:show()
       mode = 'n',
       key = '<enter>',
       handler = loop.coroutine(function()
-        local mark, _ = self.diff_view:get_current_mark_under_cursor()
+        local mark = self.diff_view:get_current_mark_under_cursor()
         if not mark then return end
 
         local filepath = self.store:get_filepath()
@@ -479,7 +479,6 @@ function ProjectDiffScreen:show()
       mode = 'n',
       key = '<enter>',
       handler = loop.coroutine(function()
-        local mark, _ = self.diff_view:get_current_mark_under_cursor()
         local filename = self.store:get_filepath()
         loop.free_textlock()
 
@@ -489,6 +488,8 @@ function ProjectDiffScreen:show()
           return
         end
 
+        local mark = self.diff_view:get_current_mark_under_cursor()
+
         self:destroy()
         fs.open(filename)