[Computer Craft] Mouse File Browser 0.1a by BigSHinyToys

--[[
        file selection and dilog box API / lib / other
        by BigSHinyToys
        note: send link to nightin9ale on CC forums
]]--

local tArgs = {...}
local ver = "0.1a"

local bugTest = false

if tArgs[1] and tArgs[1] == "-d" then
    bugTest = true
end

local function clear()
    term.setBackgroundColor(colors.black)
    term.setTextColor(colors.white)
    term.clear()
    term.setCursorBlink(false)
    term.setCursorPos(1,1)
end

local function dialogBox(title,message,options)
    term.setCursorBlink(false)
    local selected = 1
    local boxW,boxH = 26,3
    local termX,termY = term.getSize()
    local ofsX,ofsY = (termX/2) - (boxW/2) , (termY/2) - (boxH/2) -- offset from top left

    local space = 0
    local range = {}
    for i = 1,#options do
        range[i] = {s = space,f = space + string.len(options[i])}
        space = space + string.len(options[i])+3
    end
    local ofC = (boxW/2) - (space/2)

    local function drawBox()
        if term.isColor() then
            term.setTextColor(colors.black)
            term.setCursorPos(ofsX,ofsY)
            term.setBackgroundColor(colors.blue)
            term.write(" "..title..string.rep(" ",boxW-#title-5).."_[]")
            term.setBackgroundColor(colors.red)
            term.setTextColor(colors.white)
            term.write("X")
            term.setCursorPos(ofsX,ofsY+1)
            term.setBackgroundColor(colors.white)
            term.setTextColor(colors.black)
            term.write(" "..message..string.rep(" ",boxW-#message-1))
            term.setCursorPos(ofsX,ofsY+2)
            term.write(string.rep(" ",boxW))
            term.setCursorPos(ofsX,ofsY+3)

            term.write(string.rep(" ",boxW))
            for i = 1,#options do
                if i == selected then
                    term.setBackgroundColor(colors.lightGray)
                    term.setCursorPos(range[i].s + ofC + ofsX - 1,ofsY + 3)
                    term.write("["..options[i].."]")
                    term.setBackgroundColor(colors.white)
                    term.write(" ")
                else
                    term.setCursorPos(range[i].s + ofC + ofsX - 1,ofsY + 3)
                    term.write(" "..options[i].." ")
                end
            end
            term.setCursorPos(ofsX + ofC + space,ofsY + 3)
            term.write(string.rep(" ",boxW - (ofC + space)))
            term.setBackgroundColor(colors.black)
            term.setTextColor(colors.white)
        else

        end
    end
    while true do
        drawBox()
        event = {os.pullEvent()}
        if event[1] == "key" then
            if event[2] == 203 then -- left
                selected = selected - 1
                if selected < 1 then
                    selected = #options
                end
            elseif event[2] == 205 then -- right
                selected = selected + 1
                if selected > #options then
                    selected = 1
                end
            elseif event[2] == 28 then -- enter
                return selected , options[selected]
            end
        elseif event[1] == "mouse_click" then
            term.setCursorPos(1,1)
            term.clearLine()

            if bugTest then term.write("M "..event[2].." X "..event[3].." Y "..event[4]) end

            if event[2] == 1 then
                if event[4] == ofsY + 3 then
                    for i = 1,#options do
                        if event[3] >= range[i].s + ofC + ofsX - 1 and event[3] <= range[i].f + ofC + ofsX then
                            return i , options[i]
                        end
                    end
                end
            end
        end
    end
end

-- modified read made to play nice with coroutines

local sLine = ""

local function readMOD( _sReplaceChar, _tHistory,_wdth)
    term.setCursorBlink( true )

    local nHistoryPos = nil
    local nPos = 0
    if _sReplaceChar then
        _sReplaceChar = string.sub( _sReplaceChar, 1, 1 )
    end

    local w, h = term.getSize()
    if _wdth and type(_wdth) == "number" then
        w = _wdth
    end
    local sx, sy = term.getCursorPos()

    local function redraw( _sCustomReplaceChar )
        local nScroll = 0
        if sx + nPos >= w then
            nScroll = (sx + nPos) - w
        end

        term.setCursorPos( sx, sy )
        local sReplace = _sCustomReplaceChar or _sReplaceChar
        if sReplace then
            term.write( string.rep(sReplace, string.len(sLine) - nScroll) )
        else
            term.write( string.sub( sLine, nScroll + 1 ) )
        end
        term.setCursorPos( sx + nPos - nScroll, sy )
    end

    while true do
        local sEvent, param = os.pullEvent()
        if sEvent == "char" then
            sLine = string.sub( sLine, 1, nPos ) .. param .. string.sub( sLine, nPos + 1 )
            nPos = nPos + 1
            redraw()

        elseif sEvent == "key" then
            if param == keys.enter then
                -- Enter
                -- break

            elseif param == keys.left then
                -- Left
                if nPos > 0 then
                    nPos = nPos - 1
                    redraw()
                end

            elseif param == keys.right then
                -- Right
                if nPos < string.len(sLine) then
                    nPos = nPos + 1
                    redraw()
                end

            elseif param == keys.up or param == keys.down then
                -- Up or down
                if _tHistory then
                    redraw(" ");
                    if param == keys.up then
                        -- Up
                        if nHistoryPos == nil then
                            if #_tHistory > 0 then
                                nHistoryPos = #_tHistory
                            end
                        elseif nHistoryPos > 1 then
                            nHistoryPos = nHistoryPos - 1
                        end
                    else
                        -- Down
                        if nHistoryPos == #_tHistory then
                            nHistoryPos = nil
                        elseif nHistoryPos ~= nil then
                            nHistoryPos = nHistoryPos + 1
                        end
                    end

                    if nHistoryPos then
                        sLine = _tHistory[nHistoryPos]
                        nPos = string.len( sLine )
                    else
                        sLine = ""
                        nPos = 0
                    end
                    redraw()
                end
            elseif param == keys.backspace then
                -- Backspace
                if nPos > 0 then
                    redraw(" ");
                    sLine = string.sub( sLine, 1, nPos - 1 ) .. string.sub( sLine, nPos + 1 )
                    nPos = nPos - 1
                    redraw()
                end
            elseif param == keys.home then
                -- Home
                nPos = 0
                redraw()
            elseif param == keys.delete then
                if nPos < string.len(sLine) then
                    redraw(" ");
                    sLine = string.sub( sLine, 1, nPos ) .. string.sub( sLine, nPos + 2 )
                    redraw()
                end
            elseif param == keys["end"] then
                -- End
                nPos = string.len(sLine)
                redraw()
            end
        elseif sEvent == "redraw" then
            redraw()
        end
    end

    term.setCursorBlink( false )
    -- term.setCursorPos( w + 1, sy )
    -- print()

    return sLine
end

-- end modified read

local function printC(posX,posY,textCol,backCol,text)
    term.setCursorPos(posX,posY)
    term.setTextColor(textCol)
    term.setBackgroundColor(backCol)
    term.write(text)
end

local function rClickMenu(title,tList,posX,posY)

    term.setCursorBlink(false)
    local BoxTitle = title
    local choices = tList
    local termX,termY = term.getSize()
    local offX,offY

    local width = #BoxTitle + 2
    local hight = #choices + 1

    for i = 1,#choices do
        if width < #choices[i] + 2 then
            width = #choices[i] + 2
        end
    end

    offX,offY = math.ceil((termX/2) - (width/2)),math.ceil((termY/2) - (hight/2))

    if posX and posY then -- offX,offY = posX,posY
        if posX >= termX - width - 1 then
            offX = termX - width - 1
        else
            offX = posX
        end
        if posY >= termY - hight then
            offY = termY - hight
        else
            offY = posY
        end
    end

    local function reDrawer()
        printC(offX,offY,colors.black,colors.blue," "..BoxTitle..string.rep(" ",width - #BoxTitle - 1))
        for i = 1,#choices do
            printC(offX,offY + i,colors.black,colors.white," "..choices[i]..string.rep(" ",width - #choices[i] - 1))
        end
    end

    while true do
        reDrawer()
        local event = {os.pullEvent()}
        if event[1] == "mouse_click" then
            if event[2] == 1 then -- event[3] = x event[4] = y
                if event[4] > offY and event[4] < hight + offY and event[3] >= offX and event[3] < width + offX then
                    return choices[event[4] - offY]
                else
                    return
                end
            elseif event[2] == 2 then
                return
            end
        end
    end

end

local function fileSelect(mode) -- save_file open_file browse < not yet implemented

    local title = "File Browser "..ver
    local bRun = true

    local termX,termY = term.getSize()
    local offsetX,offsetY = 1,1
    local hight,width = math.ceil(termY-2),math.ceil(termX-2)

    local boxOffX,boxOffY = offsetX,offsetY + 2
    local boxH,boxW = hight - 3 ,width - 2

    local barX,barY = offsetX + 1,offsetY + 2
    local barH,barW = 1,width - 1

    local tbarX,tbarY = offsetX + 1,offsetY + 1
    local tbarH,tbarW = 1,width - 1

    local exitX,exitY = offsetX + width - 1 ,offsetY + 1

    local parth = {}
    local list

    local fSlash = [[\]]
    local listOff = 0

    local function stringParth()
        local temp = fSlash
        for i = 1,#parth do
            if i == #parth then
                temp = temp..parth[i]
            else
                temp = temp..parth[i]..fSlash
            end
        end
        return temp
    end

    local sParth = stringParth()
    local files = {}
    local folders = {}
    local disks = {}

    local function NewList()
        listOff = 0
        sParth = stringParth()
        files = {}
        folders = {}
        disks = {}
        test,list = pcall(fs.list,sParth)
        -- print(" + "..tostring(test).." "..tostring(list))
        for i = 1,#list do
            if fs.isDir(sParth..fSlash..list[i]) then
                table.insert(folders,list[i])
            else
                table.insert(files,list[i])
            end
        end
        if #parth == 0 then
            for i,v in pairs(rs.getSides()) do
                if disk.isPresent(v) and disk.hasData(v) then
                    disks[disk.getMountPath(v)] = v
                end
            end
        end
        table.sort(disks)
        table.sort(folders)
        table.sort(files)
    end

    NewList()

    local function size(test)
        if #test > boxW - 3 then
            return string.sub(test,1,boxW-3)
        end
        return test
    end

    local tRead = coroutine.create(function()
        term.setCursorPos(boxOffX + 2,boxOffY + boxH + 1)
        readMOD(nil,nil,boxOffX + boxW)
    end)

    local function resume(...)
        term.setTextColor(colors.black)
        term.setBackgroundColor(colors.white)
        term.setCursorPos(boxOffX + 1,boxOffY + boxH + 1)
        term.write(string.rep(" ",boxW+1))
        coroutine.resume(tRead,...)
        term.setTextColor(colors.white)
        term.setBackgroundColor(colors.black)
    end


    while bRun do

        -- draw top title bar
        term.setCursorPos(tbarX,tbarY)
        term.setTextColor(colors.black)
        term.setBackgroundColor(colors.blue)
        local b = tbarW - #title -2
        if b < 0 then
            b = 0
        end
        term.write(string.sub(" "..title,1,tbarW)..string.rep(" ",b))
        term.setTextColor(colors.white)
        term.setBackgroundColor(colors.red)
        term.write("X")

        -- draw location bar

        term.setCursorPos(barX,barY)
        term.setTextColor(colors.black)
        term.setBackgroundColor(colors.lightGray)
        local a = barW - #sParth - 1
        if a < 0 then
            a = 0
        end
        term.write(string.sub(" "..sParth,1,barW)..string.rep(" ",a))

        -- draw scroll bar

        if #folders + #files > boxH then
            term.setBackgroundColor(colors.lightGray)
            for i = 1,boxH do
                term.setCursorPos(boxOffX+boxW+1,i + boxOffY)
                local scroll = math.floor( boxH* (listOff/(#folders + #files-boxH+2)) )+1
                if i == scroll then
                    term.setBackgroundColor(colors.gray)
                    term.write(" ")
                    term.setBackgroundColor(colors.lightGray)
                else
                    term.write(" ")
                end
            end
        else
            term.setBackgroundColor(colors.gray)
            for i = 1,boxH do
                term.setCursorPos(boxOffX+boxW+1,i + boxOffY)
                term.write(" ")
            end
        end

        -- draw main section

        term.setTextColor(colors.black)
        term.setBackgroundColor(colors.cyan)
        for i = 1,boxH do
            term.setCursorPos(1+boxOffX,i+boxOffY)
            if folders[i+listOff] then
                if disks[folders[i+listOff]] then
                    term.setTextColor(colors.orange)
                    term.setBackgroundColor(colors.blue)
                    term.write("D!")
                else
                    term.setTextColor(colors.blue)
                    term.setBackgroundColor(colors.lightBlue)
                    term.write("[]")
                end
                term.setTextColor(colors.black)
                term.setBackgroundColor(colors.cyan)
                local repTimes = boxW - #folders[i+listOff] - 3
                if repTimes < 0 then
                    repTimes = 0
                end
                term.write(" "..size(folders[i+listOff])..string.rep(" ",repTimes))
            elseif files[i-#folders+listOff] then
                local repTimes = boxW - #files[i-#folders+listOff] - 3
                if repTimes < 0 then
                    repTimes = 0
                end
                term.write("   "..size(files[i-#folders+listOff])..string.rep(" ",repTimes))
            else
                term.write(string.rep(" ",boxW))
            end
        end
        term.setTextColor(colors.white)
        term.setBackgroundColor(colors.black)

        if bugTest then
            term.setCursorPos(1,1)
            term.write(listOff.." "..boxOffY.." "..boxH)
        end

        resume("redraw")

        -- react to events

        local event = {os.pullEvent()}

        if event[1] == "mouse_click" then
            if event[2] == 1 then -- left mouse
                local temp = nil
                if event[4] > boxOffY and event[4] <= boxH + boxOffY then
                    temp = folders[event[4]+listOff-boxOffY] or files[event[4]-#folders+listOff-boxOffY]
                end
                if temp and event[3] > boxOffX and event[3] <= #temp + 3 + boxOffX and event[3] < boxOffX + boxW then
                    if fs.isDir(stringParth()..fSlash..temp) then
                        table.insert(parth,temp)
                        NewList()
                    else
                        if fs.exists(stringParth()..fSlash..temp) then
                            if dialogBox("Run file ?",temp,{"yes","no"}) == 1 then
                                clear()
                                local test,info = shell.run(stringParth()..fSlash..temp)
                                term.setCursorBlink(false)
                                if not test then
                                    dialogBox("ERROR",tostring(info),{"ok"})
                                end
                            end
                        clear()
                        end
                    end
                elseif event[3] == boxOffX+boxW+1 and event[4] > boxOffY and event[4] <= boxOffY+boxH then
                    if #folders + #files > boxH then
                        if event[4] == boxOffY + 1 then
                            listOff = 0
                        elseif event[4] == boxOffY+boxH then
                            listOff = #folders + #files + 1 - boxH
                        else
                            listOff = math.ceil((event[4] - boxOffY - 1 )*(((#folders + #files - boxH+2)/boxH)))
                        end
                    end
                elseif event[3] == exitX and event[4] == exitY then
                    if dialogBox("Confirm","Exit application",{"yes","no"}) == 1 then
                        bRun = false
                    end
                end
            elseif event[2] == 2 then -- right mouse
                local temp = nil
                if event[4] > boxOffY and event[4] <= boxH + boxOffY then
                    temp = folders[event[4]+listOff-boxOffY] or files[event[4]-#folders+listOff-boxOffY]
                end
                if temp and event[3] > boxOffX and event[3] <= #temp + 3 + boxOffX and event[3] < boxOffX + boxW then
                    local sType = "File"

                    if fs.isDir(stringParth()..fSlash..temp) then
                        sType = "Folder"
                    end

                    local sChoice = nil

                    if disks[temp] then
                        sChoice = rClickMenu("Options",{"Open","Eject"},event[3]+1,event[4]+1)
                    else
                        if sType == "Folder" then
                            sChoice = rClickMenu("Options",{"Open","Delete"},event[3]+1,event[4]+1)
                        else
                            sChoice = rClickMenu("Options",{"Run","Run CMD","Edit","Paint","Delete"},event[3]+1,event[4]+1)
                        end
                    end

                    if sChoice == "Open" then
                        table.insert(parth,temp)
                        NewList()
                    elseif sChoice == "Run" then
                        clear()
                        local test,info = shell.run(stringParth()..fSlash..temp)
                        term.setCursorBlink(false)
                        if not test then
                            dialogBox("ERROR",tostring(info),{"ok"})
                        end
                    elseif sChoice == "Run CMD" then
                        clear()
                        print("Enter Commands")
                        local cmd = read()
                        clear()
                        local test,info = shell.run(stringParth()..fSlash..temp,cmd)
                        term.setCursorBlink(false)
                        if not test then
                            dialogBox("ERROR",tostring(info),{"ok"})
                        end
                    elseif sChoice == "Edit" then
                        if sType == "File" then
                            shell.run("edit",stringParth()..fSlash..temp) -- stringParth()..fSlash..temp
                        else
                            dialogBox("ERROR","Can't edit a Folder",{"ok"})
                        end
                    elseif sChoice == "Delete" then
                        if dialogBox("Confirm","Delete "..sType.." "..temp,{"yes","no"}) == 1 then
                            if fs.isReadOnly(stringParth()..fSlash..temp) then
                                dialogBox("ERROR",sType.." Is read Only",{"ok"})
                            else
                                fs.delete(stringParth()..fSlash..temp)
                                NewList()
                            end
                        end
                    elseif sChoice == "Paint" then
                        if sType == "File" then
                            shell.run("paint",stringParth()..fSlash..temp)
                        else
                            dialogBox("ERROR","Can't edit a Folder",{"ok"})
                        end
                    elseif sChoice == "Eject" then
                        if dialogBox("Confirm","Eject disk "..disks[temp].." "..fSlash..temp,{"yes","no"}) == 1 then
                            disk.eject(disks[temp])
                            NewList()
                        end
                    end
                else
                    table.remove(parth,#parth)
                    NewList()
                end
            end
        elseif event[1] == "mouse_scroll" then
            listOff = listOff + event[2]
            if listOff < 0 then
                listOff = 0
            end
            if #folders + #files + 1 - boxH > 0 and listOff > #folders + #files + 1 - boxH then
                listOff = #folders + #files + 1 - boxH
            elseif listOff > 0 and #folders + #files + 1 - boxH < 0 then
                listOff = 0
            end

        elseif event[1] == "key" then
            --[[
            if event[2] == 200 then boxOffY = boxOffY -1 -- up
            elseif event[2] == 208 then boxOffY = boxOffY +1 -- down
            elseif event[2] == 203 then boxOffX = boxOffX -1 -- left
            elseif event[2] == 205 then boxOffX = boxOffX +1 -- right
            end
            ]]--
            resume(unpack(event))

        elseif event[1] == "char" then
            --[[
            if event[2] == "w" then boxH = boxH + 1
            elseif event[2] == "s" then boxH = boxH - 1
            elseif event[2] == "a" then boxW = boxW - 1
            elseif event[2] == "d" then boxW = boxW + 1
            end
            ]]--
            resume(unpack(event))
        elseif event[1] == "disk" or event[1] == "disk_eject" then
            NewList()
        end
        term.setBackgroundColor(colors.black)
        term.clear()
    end
end



clear()

fileSelect(mode)
clear()


--[[
local test = {
{t = "test1",m = "is the working",s = {"yes","no"}},
{t = "test2",m = "is everthing ok",s = {"ok"}},
}

for i = 1, #test do
    local n,word = dialogBox(test[i].t,test[i].m,test[i].s)
    print(n.." "..word)
end
]]--