feat: add error handling for display module loading

jellydn · Oct 31, 2024 · de1d1e8 · de1d1e8
1 parent 9b50a31
commit de1d1e8
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/lua/hurl/history.lua b/lua/hurl/history.lua
@@ -21,7 +21,11 @@ function M.show_last_response()
   end
 
   local last_response = response_history[1]
-  local display = require('hurl.' .. _HURL_GLOBAL_CONFIG.mode)
+  local ok, display = pcall(require, 'hurl.' .. (_HURL_GLOBAL_CONFIG.mode or 'split'))
+  if not ok then
+    utils.notify('Failed to load display module: ' .. display, vim.log.levels.ERROR)
+    return
+  end
 
   display.show(last_response, last_response.display_type or 'text')
 end

diff --git a/lua/hurl/lib/hurl_runner.lua b/lua/hurl/lib/hurl_runner.lua
@@ -296,7 +296,11 @@ function M.execute_hurl_cmd(opts, callback)
   save_last_hurl_command(cmd)
 
   -- Clear the display and show processing message with Hurl command
-  local display = require('hurl.' .. _HURL_GLOBAL_CONFIG.mode)
+  local ok, display = pcall(require, 'hurl.' .. (_HURL_GLOBAL_CONFIG.mode or 'split'))
+  if not ok then
+    utils.notify('Failed to load display module: ' .. display, vim.log.levels.ERROR)
+    return
+  end
   display.clear()
 
   local stdout_data = ''

diff --git a/lua/hurl/main.lua b/lua/hurl/main.lua
@@ -325,7 +325,11 @@ function M.setup()
     if last_response then
       -- Ensure response_time is a number
       last_response.response_time = tonumber(last_response.response_time) or '-'
-      local display = require('hurl.' .. _HURL_GLOBAL_CONFIG.mode)
+      local ok, display = pcall(require, 'hurl.' .. (_HURL_GLOBAL_CONFIG.mode or 'split'))
+      if not ok then
+        utils.notify('Failed to load display module: ' .. display, vim.log.levels.ERROR)
+        return
+      end
       display.show(last_response, 'json')
     else
       utils.notify('No response history available', vim.log.levels.INFO)