feat(ui): add vim.ui.select rendering option for questions (#300)

Author: khaitranhq

README.md

@@ -215,6 +215,9 @@ require('opencode').setup({
       preset = 'nerdfonts', -- 'nerdfonts' | 'text'. Choose UI icon style (default: 'nerdfonts')
       overrides = {}, -- Optional per-key overrides, see section below
     },
+    questions = {
+      use_vim_ui_select = false, -- If true, render questions/prompts with vim.ui.select instead of showing them inline in the output buffer.
+    },
     output = {
       tools = {
         show_output = true, -- Show tools output [diffs, cmd output, etc.] (default: true)

lua/opencode/config.lua

@@ -138,6 +138,9 @@ M.defaults = {
       },
       always_scroll_to_bottom = false,
     },
+    questions = {
+      use_vim_ui_select = false, -- If true, render questions with vim.ui.select instead of in the output buffer
+    },
     input = {
       min_height = 0.10,
       max_height = 0.25,

lua/opencode/ui/question_window.lua

@@ -2,6 +2,8 @@ local state = require('opencode.state')
 local icons = require('opencode.ui.icons')
 local Dialog = require('opencode.ui.dialog')
 
+local config = require('opencode.config')
+
 local M = {}
 
 M._current_question = nil
@@ -27,8 +29,13 @@ function M.show_question(question_request)
   M._current_question = question_request
   M._current_question_index = 1
   M._collected_answers = {}
-  M._setup_dialog()
-  render_question()
+
+  if config.ui.questions and config.ui.questions.use_vim_ui_select then
+    M._show_question_with_vim_ui_select()
+  else
+    M._setup_dialog()
+    render_question()
+  end
 end
 
 function M.clear_question()
@@ -247,6 +254,60 @@ function M._clear_dialog()
   end
 end
 
+---Show question using vim.ui.select
+function M._show_question_with_vim_ui_select()
+  if not M.has_question() then
+    return
+  end
+
+  local question_info = M.get_current_question_info()
+  if not question_info or not question_info.options then
+    return
+  end
+
+  local options_to_display = add_other_if_missing(question_info.options)
+  local progress = ''
+  if M._current_question and #M._current_question.questions > 1 then
+    progress = string.format(' (%d/%d)', M._current_question_index, #M._current_question.questions)
+  end
+
+  local prompt = question_info.question .. progress
+  local choices = {}
+  for i, option in ipairs(options_to_display) do
+    table.insert(choices, option.label)
+  end
+
+  vim.ui.select(choices, {
+    prompt = prompt,
+    format_item = function(item)
+      return item
+    end,
+  }, function(choice)
+    if not choice then
+      -- User cancelled
+      if M._current_question and M._current_question.id then
+        M._send_reject(M._current_question.id)
+      end
+      M.clear_question()
+      return
+    end
+
+    -- Find the selected option index
+    local selected_index = nil
+    for i, option in ipairs(options_to_display) do
+      if option.label == choice then
+        selected_index = i
+        break
+      end
+    end
+
+    if selected_index then
+      M._answering = true
+      M._answer_with_option(selected_index)
+    end
+  end)
+end
+
 ---@param request_id string
 ---@param answers string[][]
 function M._send_reply(request_id, answers)

lua/opencode/ui/renderer.lua

@@ -201,6 +201,16 @@ function M.render_permissions_display()
 end
 
 function M.clear_question_display()
+  local config_module = require('opencode.config')
+  local use_vim_ui = config_module.ui.questions and config_module.ui.questions.use_vim_ui_select
+
+  if use_vim_ui then
+    -- When using vim.ui.select, there's nothing to clear from the buffer
+    local question_window = require('opencode.ui.question_window')
+    question_window.clear_question()
+    return
+  end
+
   local question_window = require('opencode.ui.question_window')
   question_window.clear_question()
   M._remove_part_from_buffer('question-display-part')
@@ -209,6 +219,14 @@ end
 
 ---Render question display as a fake part
 function M.render_question_display()
+  local config_module = require('opencode.config')
+  local use_vim_ui = config_module.ui.questions and config_module.ui.questions.use_vim_ui_select
+
+  if use_vim_ui then
+    -- When using vim.ui.select, we don't render anything in the buffer
+    return
+  end
+
   local question_window = require('opencode.ui.question_window')
 
   local current_question = question_window._current_question