aboutsummaryrefslogtreecommitdiff
path: root/init.lua
blob: 408d20c50b017cd0279784eef2ed6486fa763f28 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
-- Copyright (c) 2024 Florian Fischer. All rights reserved.
--
-- vis-lockfiles is free software: you can redistribute it and/or modify it under
-- the terms of the GNU General Public License as published by the Free Software
-- Foundation, either version 3 of the License, or (at your option) any later
-- version.
--
-- vis-lockfiles is distributed in the hope that it will be useful, but WITHOUT
-- ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
-- FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License along with
-- vis-lockfiles found in the LICENSE file.
-- If not, see <https://www.gnu.org/licenses/>.
local source_str = debug.getinfo(1, 'S').source:sub(2)
local script_path = source_str:match('(.*/)')

local lockfiles = dofile(script_path .. 'lockfiles.lua')

local M = {}
M.directory = '${XDG_STATE_HOME:-${HOME}/.local/state}/vis/locks'
M.directory_expanded = nil

local function get_lock_dir()
  if M.directory_expanded ~= M.directory then
    M.directory = lockfiles.capture_cmd('printf %s "' .. M.directory .. '"')
    M.directory_expanded = M.directory
  end
  return M.directory
end

-- get a user confirmation
-- return true if user selected yes, false otherwise
local function prompt(path, lfile_path, pid)
  local choices = 'a\ne\nq'
  local msg = 'Lock file found!\n\t lock file: "' .. lfile_path .. '"\n\tfile name: ' .. path ..
                  '\n\tprocess ID: ' .. pid

  vis:command('open')
  local prompt_win = vis.win
  prompt_win.file:insert(0, msg)
  prompt_win.selection.pos = 0

  vis:redraw()

  local menu_prompt = '(E)dit anyway, (Q)uit, [A]bort'
  local cmd = 'printf %s \'' .. choices .. '\' | vis-menu -b -p \'' .. menu_prompt .. '\''

  local status, output = vis:pipe(prompt_win.file, {start = 0, finish = 0}, cmd)
  vis:redraw()

  local result = false
  if status == 0 then
    if output:sub(1, 1) == 'e' then
      result = true
    elseif output:sub(1, 1) == 'q' then
      vis:exit(1)
    end
  end

  -- TODO: Auto close the prompt window
  -- Currently it fails because it is last only open window
  -- prompt_win:close(true)
  return result
end

local created_locks = {}

vis.events.subscribe(vis.events.FILE_OPEN, function(file)
  if created_locks[file.path] or not file.path then
    -- either a noname buffer, or already locked by us
    return
  end
  local lfile_path = lockfiles.lock_file_path(get_lock_dir(), file.path)
  local lock_pid = lockfiles.create_lock_file(get_lock_dir(), lfile_path)
  if lockfiles.vis_pid == lock_pid then
    created_locks[file.path] = lfile_path
  elseif not prompt(file.path, lfile_path, lock_pid) then
    -- TODO: abort
    vis:exit(1)
  end
end)

vis.events.subscribe(vis.events.FILE_CLOSE, function(file)
  if not file.path then
    return
  end
  local lfile_path = created_locks[file.path]
  if lfile_path then
    lockfiles.delete_lock_file(lfile_path)
    created_locks[file.path] = nil
  end
end)

vis.events.subscribe(vis.events.QUIT, function()
  for _, lfile_path in pairs(created_locks) do
    if lfile_path then
      lockfiles.delete_lock_file(lfile_path)
    end
  end
  created_locks = {}
end)

return M