feat: prune tag containers based on ttl

Author: cbochs

lua/grapple.lua

@@ -424,6 +424,20 @@ function Grapple.reset(opts)
     end
 end
 
+---@param opts { ttl: integer | string }
+---@return string[] | nil, string? error
+function Grapple.prune(opts)
+    local App = require("grapple.app")
+    local app = App.get()
+
+    local pruned_ids, err = app.tag_manager:prune(opts.ttl)
+    if not pruned_ids then
+        return nil, err
+    end
+
+    return pruned_ids, nil
+end
+
 ---Create a user-defined scope
 ---@param definition grapple.scope_definition
 ---@return string? error

lua/grapple/state.lua

@@ -82,6 +82,53 @@ function State:remove(name)
     end
 end
 
+---@return string[] | nil pruned, string? error, string? error_kind
+function State:prune(ttl_msec)
+    local function current_time()
+        local name = os.tmpname()
+        local fd, err, err_kind = vim.loop.fs_open(name, "r", 438)
+        if not fd then
+            return nil, err, err_kind
+        end
+
+        local stat, err, err_kind = vim.loop.fs_fstat(fd)
+        if not stat then
+            return nil, err, err_kind
+        end
+
+        assert(vim.loop.fs_close(fd))
+        assert(vim.loop.fs_unlink(name))
+
+        return stat.mtime.sec, nil, nil
+    end
+
+    local now, err, err_kind = current_time()
+    if not now then
+        return nil, err, err_kind
+    end
+
+    local pruned = {}
+
+    for name, _ in vim.fs.dir(self.save_dir) do
+        local path = Path.join(self.save_dir, name)
+
+        ---@diagnostic disable-next-line: redefined-local
+        local stat, err, err_kind = vim.loop.fs_stat(path)
+        if not stat then
+            return err, err_kind
+        end
+
+        local elapsed_msec = now - stat.mtime.sec
+        if elapsed_msec > ttl_msec then
+            name = path_decode(name)
+            name = string.gsub(name, "%.json", "")
+            table.insert(pruned, path_decode(name))
+        end
+    end
+
+    return pruned, nil, nil
+end
+
 ---@param name string
 ---@return any decoded, string? error, string? error_kind
 function State:read(name)

lua/grapple/tag_manager.lua

@@ -152,6 +152,44 @@ function TagManager:reset(id)
     end
 end
 
+---@param ttl integer | string
+---@return string[] | nil pruned, string? error
+function TagManager:prune(ttl)
+    vim.validate({
+        ttl = { ttl, { "number", "string" } },
+    })
+
+    local ttl_msec
+    if type(ttl) == "number" then
+        ttl_msec = ttl
+    elseif type(ttl) == "string" then
+        local n, kind = string.match(ttl, "^(%d+)(%S)$")
+        if not n or not kind then
+            return nil, string.format("Could not parse time-to-live: %s", ttl)
+        end
+
+        n = assert(tonumber(n))
+        if kind == "d" then
+            ttl_msec = n * 24 * 60 * 60 * 1000
+        elseif kind == "h" then
+            ttl_msec = n * 60 * 60 * 1000
+        elseif kind == "m" then
+            ttl_msec = n * 60 * 1000
+        else
+            return nil, string.format("Invalid time-to-live kind: %s", kind)
+        end
+    else
+        return nil, string.format("Invalid time-to-live: %s", vim.inspect(ttl))
+    end
+
+    local pruned_ids, err = self.state:prune(ttl_msec)
+    if not pruned_ids then
+        return nil, err
+    end
+
+    return pruned_ids, nil
+end
+
 ---@param id string
 ---@return string? error
 function TagManager:sync(id)