-- Some simple dialog screens

local _ENV = mkmodule('gui.dialogs')

local gui = require('gui')
local utils = require('utils')

local dscreen = dfhack.screen

MessageBox = defclass(MessageBox, gui.FramedScreen)

MessageBox.focus_path = 'MessageBox'
MessageBox.frame_style = gui.GREY_LINE_FRAME

function MessageBox:init(info)
    info = info or {}
    self:init_fields{
        text = info.text or {},
        frame_title = info.title,
        frame_width = info.frame_width,
        on_accept = info.on_accept,
        on_cancel = info.on_cancel,
        on_close = info.on_close,
        text_pen = info.text_pen
    }
    if type(self.text) == 'string' then
        self.text = utils.split_string(self.text, "\n")
    end
    gui.FramedScreen.init(self, info)
    return self
end

function MessageBox:getWantedFrameSize()
    local text = self.text
    local w = #(self.frame_title or '') + 4
    w = math.max(w, 20)
    w = math.max(self.frame_width or w, w)
    for _, l in ipairs(text) do
        w = math.max(w, #l)
    end
    local h = #text+1
    if h > 1 then
        h = h+1
    end
    return w+2, #text+2
end

function MessageBox:onRenderBody(dc)
    if #self.text > 0 then
        dc:newline(1):pen(self.text_pen or COLOR_GREY)
        for _, l in ipairs(self.text or {}) do
            dc:string(l):newline(1)
        end
    end

    if self.on_accept then
        local x,y = self.frame_rect.x1+1, self.frame_rect.y2+1
        dscreen.paintString({fg=COLOR_LIGHTGREEN},x,y,'ESC')
        dscreen.paintString({fg=COLOR_GREY},x+3,y,'/')
        dscreen.paintString({fg=COLOR_LIGHTGREEN},x+4,y,'y')
    end
end

function MessageBox:onDestroy()
    if self.on_close then
        self.on_close()
    end
end

function MessageBox:onInput(keys)
    if keys.MENU_CONFIRM then
        self:dismiss()
        if self.on_accept then
            self.on_accept()
        end
    elseif keys.LEAVESCREEN or (keys.SELECT and not self.on_accept) then
        self:dismiss()
        if self.on_cancel then
            self.on_cancel()
        end
    end
end

function showMessage(title, text, tcolor, on_close)
    mkinstance(MessageBox):init{
        text = text,
        title = title,
        text = text,
        text_pen = tcolor,
        on_close = on_close
    }:show()
end

function showYesNoPrompt(title, text, tcolor, on_accept, on_cancel)
    mkinstance(MessageBox):init{
        title = title,
        text = text,
        text_pen = tcolor,
        on_accept = on_accept,
        on_cancel = on_cancel,
    }:show()
end

InputBox = defclass(InputBox, MessageBox)

InputBox.focus_path = 'InputBox'

function InputBox:init(info)
    info = info or {}
    self:init_fields{
        input = info.input or '',
        input_pen = info.input_pen,
        on_input = info.on_input,
    }
    MessageBox.init(self, info)
    self.on_accept = nil
    return self
end

function InputBox:getWantedFrameSize()
    local mw, mh = MessageBox.getWantedFrameSize(self)
    return mw, mh+2
end

function InputBox:onRenderBody(dc)
    MessageBox.onRenderBody(self, dc)

    dc:newline(1)
    dc:pen(self.input_pen or COLOR_LIGHTCYAN)
    dc:fill(dc.x1+1,dc.y,dc.x2-1,dc.y)

    local cursor = '_'
    if math.floor(dfhack.getTickCount()/300) % 2 == 0 then
        cursor = ' '
    end
    local txt = self.input .. cursor
    if #txt > dc.width-2 then
        txt = string.sub(txt, #txt-dc.width+3)
        -- Add prefix arrow
        dc:advance(-1):char(27)
    end
    dc:string(txt)
end

function InputBox:onInput(keys)
    if keys.SELECT then
        self:dismiss()
        if self.on_input then
            self.on_input(self.input)
        end
    elseif keys.LEAVESCREEN then
        self:dismiss()
        if self.on_cancel then
            self.on_cancel()
        end
    elseif keys._STRING then
        if keys._STRING == 0 then
            self.input = string.sub(self.input, 1, #self.input-1)
        else
            self.input = self.input .. string.char(keys._STRING)
        end
    end
end

function showInputPrompt(title, text, tcolor, input, on_input, on_cancel, min_width)
    mkinstance(InputBox):init{
        title = title,
        text = text,
        text_pen = tcolor,
        input = input,
        on_input = on_input,
        on_cancel = on_cancel,
        frame_width = min_width,
    }:show()
end


return _ENV