local M = {}

local type_check = require("Trans.util.debug").type_check
local query = require("Trans.database").query

local function get_select()
    local s_start = vim.fn.getpos("'<")
    local s_end = vim.fn.getpos("'>")
    if s_start[2] ~= s_start[2] then
        error('TODO: multiline translate')
    end
    local lin = vim.api.nvim_buf_get_lines(0, s_start[2] - 1, s_end[2], false)[1]
    local word = string.sub(lin, s_start[3], s_end[3])
    return word
end

M.get_query_res = function(method)
    type_check {
        method = { method, 'string' },
    }

    -- NOTE : get query word
    local word = ''
    if method == 'cursor' then
        word = vim.fn.expand('<cword>')
    elseif method == 'select' then
        word = get_select():match('%s+')
    elseif method == 'input' then
        word = vim.fn.input('请输入您要查询的单词:') -- TODO Use Telescope with fuzzy finder
    else
        error('unknown method')
    end
    -- FIXME
    return query(word)
end

return M