From 1f725fca21f13b829bc731eadb46b408c1614e6a Mon Sep 17 00:00:00 2001
From: Inom Turdikulov <3938577+Inom-Turdikulov@users.noreply.github.com>
Date: Mon, 26 Feb 2024 16:02:22 +0600
Subject: [PATCH] feat: add custom media_extensions support (#306)

* Add custom media_extensions support

Customize insert image and preview image files list. This is useful to
add optional filetypes into filtered list (for example
telescope-media-files optionally supporting svg preview)

* Make media_extension option matching to default one
---
 doc/telekasten.txt | 14 ++++++++++++++
 lua/telekasten.lua | 42 +++++++++++++++---------------------------
 2 files changed, 29 insertions(+), 27 deletions(-)

diff --git a/doc/telekasten.txt b/doc/telekasten.txt
index 9723842..9bc088c 100644
--- a/doc/telekasten.txt
+++ b/doc/telekasten.txt
@@ -136,6 +136,20 @@ telekasten.setup({opts})
         -- "viu-previewer" if you have viu installed
       media_previewer = "telescope-media-files",
 
+      -- Customize insert image and preview image files list. This is useful
+      -- to add optional filetypes into filtered list (for example
+      -- telescope-media-files optionally supporting svg preview)
+      media_extensions = {
+        ".png",
+        ".jpg",
+        ".bmp",
+        ".gif",
+        ".pdf",
+        ".mp4",
+        ".webm",
+        ".webp",
+      },
+
       -- Calendar integration
       plug_into_calendar = true,         -- use calendar integration
       calendar_opts = {
diff --git a/lua/telekasten.lua b/lua/telekasten.lua
index 7ee5ca9..3a298de 100644
--- a/lua/telekasten.lua
+++ b/lua/telekasten.lua
@@ -148,6 +148,17 @@ local function defaultConfig(home)
         -- "catimg-previewer" if you have catimg installed
         -- "viu-previewer" if you have viu installed
         media_previewer = "telescope-media-files",
+        -- files which will be aviable in insert and preview images list
+        media_extensions = {
+            ".png",
+            ".jpg",
+            ".bmp",
+            ".gif",
+            ".pdf",
+            ".mp4",
+            ".webm",
+            ".webp",
+        },
         -- A customizable fallback handler for urls.
         follow_url_fallback = nil,
         -- Enable creation new notes with Ctrl-n when finding notes
@@ -1395,15 +1406,7 @@ local function PreviewImg(opts)
                 cwd = imageDir,
                 default_text = fname,
                 find_command = M.Cfg.find_command,
-                filter_extensions = {
-                    ".png",
-                    ".jpg",
-                    ".bmp",
-                    ".gif",
-                    ".pdf",
-                    ".mp4",
-                    ".webm",
-                },
+                filter_extensions = M.Cfg.media_extensions,
                 preview_type = "media",
                 attach_mappings = function(prompt_bufnr, map)
                     actions.select_default:replace(function()
@@ -1447,15 +1450,7 @@ local function BrowseImg(opts)
             prompt_title = "Preview image/media",
             cwd = M.Cfg.home,
             find_command = M.Cfg.find_command,
-            filter_extensions = {
-                ".png",
-                ".jpg",
-                ".bmp",
-                ".gif",
-                ".pdf",
-                ".mp4",
-                ".webm",
-            },
+            filter_extensions = M.Cfg.media_extensions,
             preview_type = "media",
             attach_mappings = function(prompt_bufnr, map)
                 actions.select_default:replace(function()
@@ -1784,15 +1779,7 @@ local function InsertImgLink(opts)
             prompt_title = "Find image/media",
             cwd = M.Cfg.home,
             find_command = M.Cfg.find_command,
-            filter_extensions = {
-                ".png",
-                ".jpg",
-                ".bmp",
-                ".gif",
-                ".pdf",
-                ".mp4",
-                ".webm",
-            },
+            filter_extensions = M.Cfg.media_extensions,
             preview_type = "media",
             attach_mappings = function(prompt_bufnr, map)
                 actions.select_default:replace(function()
@@ -2943,6 +2930,7 @@ local function Setup(cfg)
         M.Cfg.rg_pcre = true
     end
     M.Cfg.media_previewer = M.Cfg.media_previewer
+    M.Cfg.media_extensions = M.Cfg.media_extensions
 end
 
 local function _setup(cfg)