filedialog.lua

local imgui = require("imgui")
local nativefs = require("love-imgui-filedialog.love-nativefs.nativefs")

local dialog = {}

function dialog.new(mode, ...)
    local args = {...}
    local fileCallback, cancelCallback, path, defaultFilename, saveFile
    if mode == 'open' then
        fileCallback = args[1]
        cancelCallback = args[2]
        path = args[3]
    elseif mode == 'save' then
        saveFile = args[1]
        defaultFilename = args[2]
        successCallback = args[3]
        cancelCallback = args[4]
        path = args[5]
    end

    local self = {}
    setmetatable(self, {__index=dialog})

    self.path = path or nativefs.getWorkingDirectory()
    self.separator = "/"--string.sub(self.path, 1, 1) == '/' and '/' or '\\'
    self.files = {}
    self.fileCallback = fileCallback
    self.cancelCallback = cancelCallback
    self.saveFilename = 'untitled' or defaultFilename
    self.saveFile = saveFile
    self.view = 'list'
    self.mode = mode
    self.previewCanvas = nil
    self.previewImage = nil
    self.selectedTime = 0
    self.windowSizeSet = false
    self.columnWidthSet = false
    self.close = false
    self._pathbar = {}
    self._pathbar.viewButtonWidth = 0

    self:_updateFiles()

    return self
end

function dialog:draw()
    -- Create filedialog
    imgui.OpenPopup("File Chooser Dialog")

    -- Set dialog size
    if not self.defaultSizeSet then
        imgui.SetNextWindowSize(500, 300)
        self.defaultSizeSet = true
    end
    if imgui.BeginPopupModal("File Chooser Dialog") then
        self:_drawPathbar()
        self:_drawListView()
        self:_drawButtons()

        imgui.EndPopup()
    end

    if not self.close then
        return self
    end
end

function dialog:_drawPathbar()
    imgui.BeginChild("Pathbar", imgui.GetContentRegionMax()-self._pathbar.viewButtonWidth-10, imgui.GetFontSize() + 4) -- love-imgui doesn't support imgui.GetItemsLineHeightWithSpacing()
        local path = ''
        for i in self.path:gmatch("[^"..self.separator.."]+") do
            path = path .. (path ~= '' and self.separator or '') .. i
            if i ~= self.separator then
                if imgui.Button(i) then
                    self.path = path
                    self.tmpPath = self.path
                    self:_updateFiles()
                end
            end
            imgui.SameLine()
        end
    imgui.EndChild()
    if self.mode == 'save' then
        _, self.saveFilename = imgui.InputText("File Name", self.saveFilename, 64)
    end
end

function dialog:_drawListView()
    -- Draw list view
    imgui.BeginChild(0, 0, -imgui.GetFontSize() + 4)                        -- love-imgui doesn't support imgui.GetItemsLineHeightWithSpacing()
    imgui.Columns(3, nil, true)                                             -- Begin columns for file browser
    if not self.columnWidthSet then
        imgui.NextColumn()                                                  -- Move to Size column
        imgui.SetColumnOffset(-1, 170)                                      -- Set offset for Size column
        imgui.NextColumn()                                                  -- Move to Modified column
        imgui.SetColumnOffset(-1, 240)                                      -- Set offset for Modified column
        imgui.NextColumn()                                                  -- Return to Name column
    end
    imgui.Separator()

    -- Table Header
    imgui.Text("Name")
    imgui.NextColumn()
    imgui.Text("Size")
    imgui.NextColumn()
    imgui.Text("Modified")
    imgui.NextColumn()
    imgui.Separator()

    if not self.columnWidthSet then
        imgui.NextColumn()                                                  -- Move to Size column
        imgui.SetColumnOffset(-1, 170)                                      -- Set offset for Size column
        imgui.NextColumn()                                                  -- Move to Modified column
        imgui.SetColumnOffset(-1, 240)                                      -- Set offset for Modified column
        imgui.NextColumn()                                                  -- Return to Name column
        self.columnWidthSet = true
    end
    -- List files
    for i, file in ipairs(self.files) do
        -- Name
        if imgui.Selectable(file.name .. (file.attributes and file.attributes.type == "directory" and self.separator or ''), self.selected == i, "DontClosePopups") then
            self:_fileClicked(i, file)
        end
        imgui.NextColumn()

        -- Size
        if file.attributes and file.attributes.type ~= 'directory' then
            local divisor
            local suffix

            if file.attributes.size > 1073741824 then
                divisor = 1073741824
                suffix = 'GB'
            elseif file.attributes.size > 1048576 then
                divisor = 1048576
                suffix = 'MB'
            elseif file.attributes.size > 1024 then
                divisor = 1024
                suffix = 'KB'
            else
                divisor = 1
                suffix = 'B'
            end

            local size = tostring(file.attributes.size/divisor)
            local str = string.format("%.1f", size)..suffix
            imgui.Text(str)
        end
        imgui.NextColumn()

        -- Modified Date
        if file.attributes then
            local date = os.date("*t", file.attributes.modtime)
            if date then
                imgui.Text(string.format("%d-%02d-%02d", date.year, date.month, date.day))
            end
        end
        imgui.NextColumn()
    end

    imgui.Columns(1)
    imgui.EndChild()
end

function dialog:_drawGridView()
end

function dialog:_drawButtons()
    if imgui.Button("Close") then
        if self.cancelCallback then
            self.cancelCallback()
        end
        self.close = true
    end
    imgui.SameLine()
    if self.mode == 'open' then
        if imgui.Button("Open") then
            if self.files[self.selected].attributes and self.files[self.selected].attributes.type == "directory" then
                self.path = self.path .. self.separator .. self.files[self.selected].name
                self:_updateFiles()
            elseif self.fileCallback then
                self.fileCallback(nativefs.newFile(self.path .. '/' .. self.files[self.selected].name))
                self.close = true
            end
        end
    elseif self.mode == 'save' then
        if imgui.Button('Save') then
            self:_saveFile()
            self.close = true
        end
    end
end

function dialog:_drawConfirm()
end

function dialog:_updateFiles()
    self.selected = nil
    self.files = {}
    for _, name in ipairs(nativefs.getDirectoryItems(self.path)) do
        local file = {}
        file.name = name
        file.attributes = nativefs.getInfo(self.path..self.separator..name)
        if file.name ~= "." and file.name ~= ".." then
            table.insert(self.files, file)
        end
    end
    table.sort(self.files, self._sortByName)
end

function dialog:_fileClicked(i, name)
    -- Detect double click
    if (self.selected == i) and (os.clock() - self.selectedTime) < 0.25 then
        if self.files[i].attributes.type == 'directory' then
            self.path = self.path..self.separator..self.files[i].name
            self.tmpPath = self.path
            self:_updateFiles()
        elseif self.files[i].attributes.type == 'file' then
            if self.mode == 'open' then
                if self.fileCallback then
                    self.fileCallback(nativefs.newFile(self.path .. self.separator .. self.files[i].name))
                end
                self.close = true
            elseif self.mode == 'save' then
                print("ARE YOU SURE YOU WANT TO OVERWRITE???!?!??!??!??!")
            end
        end
    else
        self.selected = i
    end
    if self.files[i] and self.files[i].attributes and self.files[i].attributes.type == 'file' then
        self.saveFilename = self.files[i].name
    end
    self.selectedTime = os.clock()
end

function dialog:_saveFile()
    local tmpFile = nativefs.getWorkingDirectory()..self.separator..self.saveFile
    local newFile = self.path..self.separator..self.saveFilename
    os.rename(tmpFile, newFile)
    print(tmpFile, newFile)
end

function dialog:_splitString(string, char)
    local strings = {}
    for substring in string.gmatch(string, '[^'..char..']+') do
        table.insert(strings, substring)
    end

    return strings
end

function dialog._sortByName(a, b)
    return string.lower(a.name) < string.lower(b.name)
end

return dialog