aboutsummaryrefslogtreecommitdiff
path: root/init.lua
blob: 5c85818b7d87c373713ecadc9ac76cfbd62c67b7 (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
-- 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 confirm(path)
  local choices = 'no\nyes'
  local prompt = path .. ' exists. Edit anyway:'
  local cmd = 'printf "' .. choices .. '" | vis-menu -b -p \'' .. prompt .. '\''

  local choice = nil
  -- local status, output = vis:pipe(nil, {start = 0, finish = 0}, cmd)
  local menu = assert(io.popen(cmd))
  local output = menu:read('*a')
  local _, _, status = menu:close()
  if status == 0 then
    -- trim newline from selection
    if output:sub(-1) == '\n' then
      choice = output:sub(1, -2)
    else
      choice = output
    end
  end
  vis:redraw()
  return choice == 'yes'
end

local created_locks = {}

vis.events.subscribe(vis.events.FILE_OPEN, function(file)
  if not file.path then
    return
  end
  local lfile_path = lockfiles.lock_file_path(get_lock_dir(), file.path)
  if lockfiles.create_lock_file(get_lock_dir(), lfile_path) then
    created_locks[lfile_path] = true
  elseif not confirm(lfile_path) then
    os.exit(1)
  end
end)

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

return M