local api = vim.api
local conf = require('Trans').conf
local hover = conf.hover
local buffer = require('Trans.buffer')()
local error_msg = conf.icon.notfound .. '    没有找到相关的翻译'

local node = require('Trans.node')
local it = node.item
local t = node.text
local f = node.format

local function handle_result(result)
    local icon = conf.icon
    local notfound = icon.notfound
    local indent = '    '

    local addtitle = function(title)
        buffer:addline {
            it('', 'TransTitleRound'),
            it(title, 'TransTitle'),
            it('', 'TransTitleRound'),
        }
    end

    local process = {
        title = function(title)
            local word     = title.word
            local oxford   = title.oxford
            local collins  = title.collins
            local phonetic = title.phonetic

            if not phonetic and not collins and not oxford then
                buffer:addline(it(result.word, 'TransWord'))

            else
                buffer:addline(f {
                    width = hover.width,
                    text = t {
                        it(word, 'TransWord'),
                        t {
                            it('['),
                            it((phonetic and phonetic ~= '') and phonetic or notfound, 'TransPhonetic'),
                            it(']')
                        },
                        it(collins and icon.star:rep(collins) or notfound, 'TransCollins'),
                        it(oxford == 1 and icon.yes or icon.no)
                    },
                })
            end
        end,

        tag = function(tag)
            addtitle('标签')
            local tag_map = {
                zk    = '中考',
                gk    = '高考',
                ky    = '考研',
                cet4  = '四级',
                cet6  = '六级',
                ielts = '雅思',
                toefl = '托福',
                gre   = 'gre ',
            }

            local tags = {}
            local size = 0
            local interval = '    '
            for _tag in vim.gsplit(tag, ' ', true) do
                size = size + 1
                tags[size] = tag_map[_tag]
            end


            for i = 1, size, 3 do
                buffer:addline(
                    it(
                        indent .. tags[i] ..
                        (tags[i + 1] and interval .. tags[i + 1] ..
                            (tags[i + 2] and interval .. tags[i + 2] or '') or ''),
                        'TransTag'
                    )
                )
            end

            buffer:addline('')
        end,

        pos = function(pos)
            addtitle('词性')
            local pos_map = {
                a = '代词pron         ',
                c = '连接词conj       ',
                i = '介词prep         ',
                j = '形容词adj        ',
                m = '数词num          ',
                n = '名词n            ',
                p = '代词pron         ',
                r = '副词adv          ',
                u = '感叹词int        ',
                v = '动词v            ',
                x = '否定标记not      ',
                t = '不定式标记infm   ',
                d = '限定词determiner ',
            }

            local s = '%s %2s%%'
            for _pos in vim.gsplit(pos, '/', true) do
                buffer:addline(
                    it(indent .. s:format(pos_map[_pos:sub(1, 1)], _pos:sub(3)), 'TransPos')
                )
            end

            buffer:addline('')
        end,

        exchange = function(exchange)
            addtitle('词形变化')
            local exchange_map = {
                ['p'] = '过去式      ',
                ['d'] = '过去分词    ',
                ['i'] = '现在分词    ',
                ['r'] = '比较级      ',
                ['t'] = '最高级      ',
                ['s'] = '复数        ',
                ['0'] = '原型        ',
                ['1'] = '类别        ',
                ['3'] = '第三人称单数',
                ['f'] = '第三人称单数',
            }
            local interval = '    '
            for exc in vim.gsplit(exchange, '/', true) do
                buffer:addline(
                    it(indent .. exchange_map[exc:sub(1, 1)] .. interval .. exc:sub(3), 'TransExchange')
                )
            end

            buffer:addline('')
        end,

        translation = function(translation)
            if hover.auto_play then
                result.title.word:play()
            end

            addtitle('中文翻译')

            for trs in vim.gsplit(translation, '\n', true) do
                buffer:addline(
                    it(indent .. trs, 'TransTranslation')
                )
            end

            buffer:addline('')
        end,

        definition = function(definition)
            addtitle('英文注释')

            for def in vim.gsplit(definition, '\n', true) do
                def = def:gsub('^%s+', '', 1) -- TODO :判断是否需要分割空格
                buffer:addline(
                    it(indent .. def, 'TransDefinition')
                )
            end

            buffer:addline('')
        end,
    }

    buffer:set('modifiable', true)
    for _, field in ipairs(conf.order) do
        local value = result[field]
        if value and value ~= '' then
            process[field](value)
        end
    end
    buffer:set('modifiable', false)
end

local function open_window(opts)
    opts           = opts or {}
    local col      = opts.col or 1
    local row      = opts.row or 1
    local width    = opts.width or hover.width
    local height   = opts.height or hover.height
    local relative = opts.relative or 'cursor'
    local task     = opts.task

    local win = require('Trans.window') {
        col       = col,
        row       = row,
        task      = task,
        buf       = buffer,
        relative  = relative,
        width     = width,
        height    = height,
        title     = hover.title,
        border    = hover.border,
        animation = hover.animation,
        zindex    = 100,
        enter     = false,
        ns        = require('Trans').ns,
    }
    return win
end

local function handle_keymap(win, word)
    local keymap = hover.keymap
    local cur_buf = api.nvim_get_current_buf()
    local del = vim.keymap.del
    local function try_del_keymap()
        for _, key in pairs(keymap) do
            pcall(del, 'n', key, { buffer = cur_buf })
        end
    end

    local lock = false
    local cmd_id
    local next = win.id
    local action = {
        pageup = function()
            buffer:normal('gg')
        end,

        pagedown = function()
            buffer:normal('G')
        end,

        pin = function()
            if lock then
                error('too many window')
            else
                lock = true
            end
            pcall(api.nvim_del_autocmd, cmd_id)
            local width = win.width
            local height = win.height
            local col = vim.o.columns - width - 3
            local buf = buffer.bufnr
            win:try_close()
            win.tasks:add(function()
                win = open_window {
                    width = width,
                    height = height,
                    relative = 'editor',
                    col = col,
                    task = function(self)
                        self:set('wrap', true)
                    end,
                }

                del('n', keymap.pin, { buffer = cur_buf })
                api.nvim_create_autocmd('BufWipeOut', {
                    callback = function(opt)
                        if opt.buf == buf or opt.buf == cur_buf then
                            lock = false
                            api.nvim_del_autocmd(opt.id)
                        end
                    end
                })
            end)
        end,

        close = function()
            pcall(api.nvim_del_autocmd, cmd_id)
            win:try_close()
            win.tasts:add(function()
                buffer:delete()
            end)
            try_del_keymap()
        end,

        toggle_entry = function()
            if lock and win:is_valid() then
                local prev = api.nvim_get_current_win()
                api.nvim_set_current_win(next)
                next = prev
            else
                del('n', keymap.toggle_entry, { buffer = cur_buf })
            end
        end,

        play = function()
            if word then
                word:play()
            end
        end,
    }

    local set = vim.keymap.set
    local opts = { buffer = cur_buf, silent = true }
    for act, key in pairs(hover.keymap) do
        set('n', key, action[act], opts)
    end

    if hover.auto_close_events then
        cmd_id = api.nvim_create_autocmd(
            hover.auto_close_events, {
            buffer = 0,
            callback = function(opt)
                win:try_close()
                win.tasks:add(function()
                    buffer:delete()
                    try_del_keymap()
                end)
                api.nvim_del_autocmd(opt.id)
            end,
        })
    end
end

local function online_query(win, word)
    -- FIXME :
    local lists = {
        remove = table.remove
    }
    local engines = conf.engines
    local size = #engines
    local icon = conf.icon
    local error_line = it(error_msg, 'TransFailed')

    if size == 0 then
        buffer:addline(error_line)

    else
        for i = 1, size do
            lists[size] = require('Trans.query.' .. engines[i])(word)
        end
        local win_width = win.width
        local cell = icon.cell
        local spinner = require('Trans.ui.spinner')[hover.spinner]
        local range = #spinner

        local timeout = hover.timeout
        local interval = math.floor(timeout / (win.width - spinner[1]:width()))

        local s = '%s %s'
        local width = hover.width
        local height = hover.height
        buffer:set('modifiable', true)

        require('Trans.util.display') {
            times = win_width,
            interval = interval,
            frame = function(self, times)
                for i, v in ipairs(lists) do
                    local res = v[1]
                    if res then
                        vim.pretty_print(res)
                        buffer:del(1)
                        win:set_width(width)
                        handle_result(res)
                        local actual_height = buffer:height {
                            width = width,
                            wrap = true,
                        }
                        height = math.min(height, actual_height)

                        win:expand {
                            field = 'height',
                            target = height,
                        }

                        win.tasks:add(function(this)
                            this:set('wrap', true)
                            handle_keymap(this, word)
                        end)

                        self.run = false
                        return

                    elseif res == false then
                        lists:remove(i)
                        size = size - 1
                    end
                end

                local line
                if size == 0 or times == win_width then
                    line = error_line
                    self.run = false
                    win:set('wrap', true)
                    handle_keymap(win, word)

                else
                    line = it(s:format(spinner[times % range + 1], cell:rep(times)), 'MoreMsg')
                end

                buffer:addline(line, 1)
            end,

            callback = function()
                buffer:set('modifiable', false)
            end,
        }
    end
end

return function(word)
    buffer:init()
    local result = require('Trans.query.offline')(word)

    local opts
    if result then
        handle_result(result)

        local width = hover.width
        local height = math.min(buffer:height {
            width = width,
            wrap = true,
        }, hover.height)

        opts = {
            width = width,
            height = height,
            task = function(self)
                self:set('wrap', true)
                handle_keymap(self, word)
            end
        }

    else
        opts = {
            width = error_msg:width(),
            height = 1,
            task = function(win)
                online_query(win, word)
            end
        }
    end

    open_window(opts)
end