feat: question-tabs picker

Author: kawre

lua/leetcode-ui/component/button.lua

@@ -14,16 +14,13 @@ setmetatable(button, component)
 --- @param sc string|nil
 --- @param on_press function? optional
 --- @param expandable boolean? optional
---- @param keybind string? optional
---- @param keybind_opts table? optional
-function button:init(text, sc, on_press, expandable, keybind, keybind_opts)
+function button:init(text, sc, on_press, expandable)
     local opts = {
         position = "center",
         on_press = on_press or function() end,
         sc = sc,
     }
     sc = sc and "" or ""
-    -- log.debug(opts)
 
     local width = 50
     local expand = ""

lua/leetcode/command.lua

@@ -105,25 +105,7 @@ function cmd.menu() vim.api.nvim_set_current_tabpage(_Lc_MenuTabPage) end
 ---@param layout layouts
 function cmd.menu_layout(layout) _Lc_Menu:set_layout(layout) end
 
-function cmd.list_questions()
-    local utils = require("leetcode.utils")
-    local questions = utils.get_current_questions()
-
-    if not vim.tbl_isempty(questions) then
-        local ui = require("leetcode.ui")
-        local curr_tabp = vim.api.nvim_get_current_tabpage()
-
-        ui.pick_one(questions, "Select a Question", function(item)
-            ---@type question_response
-            local question = item.question.q
-            local text = question.frontend_id .. ". " .. question.title
-
-            return (curr_tabp == item.tabpage and "(C) " or "    ") .. text
-        end, function(res) pcall(vim.cmd.tabnext, res.tabpage) end)
-    else
-        log.warn("No questions opened")
-    end
-end
+function cmd.question_tabs() require("leetcode.pickers.question-tabs").pick() end
 
 function cmd.change_lang() require("leetcode.pickers.language").pick() end
 

lua/leetcode/init.lua

@@ -43,7 +43,7 @@ local function setup_cmds()
 
     vim.api.nvim_create_user_command("LcConsole", function() cmd.console() end, {})
     vim.api.nvim_create_user_command("LcMenu", function() cmd.menu() end, {})
-    vim.api.nvim_create_user_command("LcQuestionTabs", function() cmd.list_questions() end, {})
+    vim.api.nvim_create_user_command("LcQuestionTabs", function() cmd.question_tabs() end, {})
     vim.api.nvim_create_user_command("LcLanguage", function() cmd.change_lang() end, {})
     vim.api.nvim_create_user_command("LcDescriptionToggle", function() cmd.desc_toggle() end, {})
 end

lua/leetcode/pickers/init.lua

@@ -98,7 +98,7 @@ return {
                         local selection = action_state.get_selected_entry()
 
                         if not selection then return end
-                        config.lang = selection.slug
+                        config.lang = selection.value.slug
                         log.info("Language set to " .. selection.lang)
                     end)
                     return true

lua/leetcode/pickers/language.lua

@@ -0,0 +1,123 @@
+local log = require("leetcode.logger")
+local utils = require("leetcode.utils")
+
+local Question = require("leetcode.ui.question")
+
+local pickers = require("telescope.pickers")
+local finders = require("telescope.finders")
+local conf = require("telescope.config").values
+local config = require("leetcode.config")
+
+local entry_display = require("telescope.pickers.entry_display")
+local actions = require("telescope.actions")
+local action_state = require("telescope.actions.state")
+
+---@param question lc.Cache.Question
+---
+---@return string
+local function question_formatter(question)
+    return string.format("%d. %s", question.frontend_id, question.title)
+end
+
+local function display_current(entry)
+    local tabp = vim.api.nvim_get_current_tabpage()
+    if tabp ~= entry.tabpage then return unpack({ "", "" }) end
+
+    return { "", "" }
+end
+
+local function display_difficulty(q)
+    local filetypes = {
+        cpp = "cpp",
+        java = "java",
+        python = "py",
+        python3 = "py",
+        c = "c",
+        csharp = "cs",
+        javascript = "js",
+        typescript = "ts",
+        php = "php",
+        swift = "swift",
+        kotlin = "kt",
+        dart = "dart",
+        golang = "go",
+        ruby = "rb",
+        scala = "scala",
+        rust = "rs",
+        racket = "rkt",
+        erlang = "erl",
+        elixir = "ex",
+    }
+
+    return { filetypes[q.lang], "LeetCode" .. q.q.difficulty }
+end
+
+---@param question lc.Cache.Question
+local function display_question(question)
+    local index = { question.frontend_id .. ".", "LeetCodeNormal" }
+    local title = { question.title }
+
+    return unpack({ index, title })
+end
+
+local displayer = entry_display.create({
+    separator = " ",
+    items = {
+        { width = 1 },
+        { width = 5 },
+        { width = 5 },
+        { remaining = true },
+    },
+})
+
+local function make_display(entry)
+    ---@type lc.Cache.Question
+    local q = entry.value.question.q
+
+    return displayer({
+        display_current(entry.value),
+        display_difficulty(entry.value.question),
+        display_question(q),
+    })
+end
+
+local function entry_maker(entry)
+    return {
+        value = entry,
+        display = make_display,
+        ordinal = question_formatter(entry.question.q),
+    }
+end
+
+local opts = require("telescope.themes").get_dropdown()
+
+return {
+    pick = function()
+        local tabs = utils.get_current_question_tabs()
+        if vim.tbl_isempty(tabs) then
+            log.warn("No questions opened")
+            return
+        end
+
+        pickers
+            .new(opts, {
+                prompt_title = "Select a Question",
+                finder = finders.new_table({
+                    results = tabs,
+                    entry_maker = entry_maker,
+                }),
+                sorter = conf.generic_sorter(opts),
+                attach_mappings = function(prompt_bufnr, map)
+                    actions.select_default:replace(function()
+                        actions.close(prompt_bufnr)
+                        local selection = action_state.get_selected_entry()
+
+                        if not selection then return end
+                        pcall(vim.cmd.tabnext, selection.value.tabpage)
+                    end)
+                    return true
+                end,
+            })
+            :find()
+    end,
+}

lua/leetcode/pickers/question-tabs.lua

@@ -45,8 +45,7 @@ end
 
 ---@param question lc.Cache.Question
 local function display_question(question)
-    -- local title = question_formatter(question)
-    local index = { question.frontend_id .. "." }
+    local index = { question.frontend_id .. ".", "LeetCodeNormal" }
     local title = { question.title }
 
     return unpack({ index, title })

lua/leetcode/pickers/question.lua

@@ -61,7 +61,7 @@ function question:autocmds()
     vim.api.nvim_create_autocmd("TabEnter", {
         group = group_id,
         callback = function()
-            local questions = utils.get_current_questions()
+            local questions = utils.get_current_question_tabs()
 
             local tabpage = vim.api.nvim_get_current_tabpage()
             for _, q in ipairs(questions) do

lua/leetcode/ui/question.lua

@@ -44,7 +44,12 @@ function utils.map(mode, lhs, rhs, opts)
     vim.api.nvim_set_keymap(mode, lhs, rhs, options)
 end
 
-function utils.get_current_questions()
+function utils.get_current_question_tabs()
+    ---@class lc.Question.Tab
+    ---@field tabpage integer
+    ---@field question lc.Question
+
+    ---@type lc.Question.Tab[]
     local questions = {}
 
     for _, tabp in ipairs(vim.api.nvim_list_tabpages()) do
@@ -91,7 +96,7 @@ function utils.filetype(lang)
     }
 
     local ft = filetypes[lang]
-    assert(ft)
+    assert(ft, "Unknown language: " .. lang)
     return ft
 end