From 1da7b933895c66e8411395df696a95e3fbe52e97 Mon Sep 17 00:00:00 2001
From: aleidk <ale.navarro.parra@gmail.com>
Date: Sat, 8 Apr 2023 17:33:33 -0400
Subject: [PATCH] Add undo tree telescope extension

---
 .../astronvim/lua/user/plugins/telescope.lua  | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/config/astronvim/lua/user/plugins/telescope.lua b/config/astronvim/lua/user/plugins/telescope.lua
index 5e920d9..eb9900f 100644
--- a/config/astronvim/lua/user/plugins/telescope.lua
+++ b/config/astronvim/lua/user/plugins/telescope.lua
@@ -1,5 +1,8 @@
 return {
 	"nvim-telescope/telescope.nvim",
+	dependencies = {
+		"debugloop/telescope-undo.nvim",
+	},
 	config = function(plugin, opts)
 		local actions = require("telescope.actions")
 		opts.defaults.mappings = {
@@ -11,6 +14,25 @@ return {
 				["<ESC>"] = actions.close,
 			},
 		}
+		opts.extensions = {
+			undo = {
+				entry_format = "state #$ID, $STAT, $TIME",
+				mappings = {
+					i = {
+						["<cr>"] = require("telescope-undo.actions").yank_additions,
+						["<S-cr>"] = require("telescope-undo.actions").yank_deletions,
+						["<C-cr>"] = require("telescope-undo.actions").restore,
+					},
+				},
+			},
+		}
 		require("plugins.configs.telescope")(plugin, opts)
+
+		-- require telescope and load extensions as necessary
+		local telescope = require("telescope")
+		telescope.load_extension("undo")
+		vim.keymap.set("n", "<leader>fu", function()
+			telescope.extensions.undo.undo()
+		end, { desc = "Undo tree" })
 	end,
 }