local eapi = ...
local MapObj = require("Libraries.map")

local t = {} --The tilemap editor

--=====-- Spritesheet Values --=====--

local imgw, imgh = 8, 8 --The sprite size (MUST NOT CHANGE)

local swidth, sheight = screenSize() --The screen size
local sheetW, sheetH = math.floor(swidth/imgw), math.floor(sheight/imgh) --The size of the spritessheet in cells (sprites)
local bankH = sheetH/4 --The height of each bank in cells (sprites)
local bsizeH = bankH*imgh --The height of each bank in pixels
local sizeW, sizeH = sheetW*imgw, sheetH*imgh --The size of the spritessheet in pixels

local SpriteMap --We be recieved later in t:entered()

--========-- Map Values --========--

local MapW, MapH = math.floor(swidth*0.75), sheight --The size of the map in cells.

local MapPW, MapPH = MapW*8, MapH*8 --The map size in pixels.

local MapVW, MapVH = swidth/8, sheight/8 --The visible map space in cells.

local MapVPW, MapVPH = MapVW*8, MapVH*8 --The visible map space in pixels.

local Map = MapObj(MapW, MapH)

local mapdx, mapdy = 0,0 --Map drawing offsets.

local bgsprite = eapi.editorsheet:extract(59):image() --The background image sprite.
local bgquad = bgsprite:quad(0,0,MapVPW,MapVPH) --The quad of the background image

local isquad = bgsprite:quad(0,0,8*8,8)

local SpritesMap = MapObj(22,12)

SpritesMap:map(function(x,y)
  local tid = y*22+x
  if tid > 255 then tid = 0 end
  return tid
end)

--=======-- GUI VARIABLES --=======--

local selectedTool = 0
local selectedSlot = 2

local toolbarGrid = {swidth-9,8,9,sheight-8,1,15}

local hotbarTiles = {0,0,0,0,0,0,0,0,0,0}

local mapRect = {0,8,swidth-9,sheight-8}

local spritesGrid = {3,11,22*8,12*8,22,12}

local selPattern = imagedata("LK12;GPUIMG;2x2;0110")
local selsx, selsy, selex, seley

local darkScreenshot

--=========-- Functions --=========--

local function queuedFill(img,sx,sy,rcol)
  
  local cell = img.cell
  
  local tcol = cell(img,sx,sy) --The target color
  
  if tcol == rcol then return end
  
  --Queue, QueueSize, QueuePosition
  local q, qs, qp = {}, 0,0
  
  cell(img,sx,sy,rcol)
  qs = qs + 1
  q[qs] = {sx,sy}
  
  local function test(x,y)
    if minx and (x < minx or y < miny or x > maxx or y > maxy) then return end
    if cell(img,x,y) == tcol then
      cell(img,x,y,rcol)
      qs = qs + 1
      q[qs] = {x,y}
    end
  end
  
  while qp < qs do --While there are items in the queue.
    qp = qp + 1
    local n = q[qp]
    local x,y = n[1], n[2]
    test(x-1,y) test(x+1,y)
    test(x,y-1) test(x,y+1)
  end
end

function t:entered()
  SpriteMap = eapi.leditors[eapi.editors.sprite].SpriteMap
  self:redraw()
end

function t:redraw()
  if selectedTool == 4 then
    self:drawMenu()
  else
    self:drawMap()
  end
end

function t:drawToolbar()
  eapi:drawTopBar()
  
  --Draw the background
  rect(swidth-9,8, 9,sheight-8, false, 0)
  
  --Draw the hotbar tiles
  for i=0,9 do
    local sprid = hotbarTiles[i+1]
    
    if sprid == 0 then
      eapi.editorsheet:draw(120, swidth-8,8+i*8)
    else
      SpriteMap:draw(sprid, swidth-8,8+i*8)
    end
  end
  
  --Draw the tools
  rect(swidth-9,sheight-5*8,8,5*8, false, 9)
  for i=0,4 do
    local sprid = 114+i
    
    if i == selectedTool then sprid = sprid + 24 end
    
    eapi.editorsheet:draw(sprid,swidth-8,sheight-5*8+i*8)
  end
  
  --Draw hotbar selection box
  rect(swidth-9,8*selectedSlot-1, 10,10, true, 1)
  rect(swidth-11,8*selectedSlot-3, 12,14, true, 1)
  rect(swidth-10,8*selectedSlot-2, 11,12, true, 7)
end

function t:drawMap()
  --Clear the darkScreenshot
  darkScreenshot = false
  
  --Clip to map area
  clip(0,8,swidth-9,sheight-8)
  
  --Draw the background.
  rect(0,8,swidth-9,sheight-8, false,0)
  pal(1,2) --Change blue to red.
  bgsprite:draw(0,8, 0,1,1, bgquad)
  pal() --Reset blue to blue.
  local bgx, bgy = 0,0
  if -mapdx < 0 then bgx = mapdx end
  if -mapdy < 0 then bgy = mapdy end
  if -mapdx > MapPW-MapVPW then bgx = MapPW-MapVPW+mapdx-1 end
  if -mapdy > MapPH-MapVPH then bgy = MapPH-MapVPH+mapdy-1 end
  
  clip(bgx,bgy+8,MapVPW,MapVPH)
  bgsprite:draw(0,0, 0,1,1, bgquad)
  clip(0,8,swidth-9,sheight-8)
  
  --Draw the map
  palt(0,false)
  Map:draw(math.floor(mapdx%8-8),math.floor(mapdy%8),-math.floor(mapdx/8)-1,-math.floor(mapdy/8)-1,MapVW+1,MapVH+1, 1,1, SpriteMap)
  palt()
  
  --Draw selection
  if selsx then
    local seldx = selsx*8+mapdx-1
    local seldy = selsy*8+mapdy-1+8
    
    local selw = (selex-selsx)*8+10
    local selh = (seley-selsy)*8+10
    rect(seldx-2,seldy-2,selw+4,selh+4,true,0)
    rect(seldx-1,seldy-1,selw+2,selh+2,true,0)
    rect(seldx,seldy,selw,selh,true,0)
    patternFill(selPattern)
    rect(seldx-1,seldy-1,selw+2,selh+2,true,7)
    patternFill()
  end
  
  --Declip
  clip()
  
  --Draw the toolbar
  self:drawToolbar()
end

function t:drawMenu()
  
  if selectedTool ~= 4 then --Menu open by holding alt
    if not darkScreenshot then
      darkScreenshot = screenshot(0,8,swidth-8,sheight-8)
      ImageUtils.darken(darkScreenshot)
      darkScreenshot = darkScreenshot:image()
    end
    palt(0,false); darkScreenshot:draw(0,8); palt()
  else --The menu tool is selected
    --Clear the screen
    rect(0,8,swidth,sheight-8, false, 5)
    
    --Delete the darkScreenshot
    darkScreenshot = false
  end
  
  --Draw the toolbar
  self:drawToolbar()
  
  --Draw the sprites
  local sdx, sdy = 1,9
  rect(sdx,sdy,22*8+4,12*8+4,true,7)
  rect(sdx+1,sdy+1,22*8+2,12*8+2,true,1)
  rect(sdx+2,sdy+2,22*8,12*8,false,0)
  SpritesMap:draw(sdx+2,sdy+2,false,false,false,false,false,false,SpriteMap)
  eapi.editorsheet:draw(120,sdx+2,sdy+2)
  
  --Invalid tiles
  pal(1,2)
  bgsprite:draw(sdx+2+8*14,sdy+2+8*11,0,1,1,isquad)
  pal()
  
  --Sprite selection box
  local sprid = hotbarTiles[selectedSlot]
  local sscx, sscy = sprid%22, math.floor(sprid/22)
  
  local ssx, ssy = sdx+sscx*8, sdy+sscy*8
  rect(ssx-1,ssy-1,14,14,true,1)
  rect(ssx,ssy,12,12,true,7)
  rect(ssx+1,ssy+1,10,10,true,0)
end

function t:selectTool(id)
  if selectedTool == id then return end
  if selectedTool == 4 and not isKDown("lalt","ralt") then
    selectedTool = id
    self:drawMap()
  elseif id == 4 or isKDown("lalt","ralt") then
    selectedTool = id
    self:drawMenu()
  else
    selectedTool = id
    self:drawToolbar()
  end
end

function t:selectSlot(id)
  selectedSlot = id
  if selectedTool == 4 or isKDown("lalt","ralt")  then
    self:drawMenu()
  else
    self:drawMap()
  end
end

function t:nextSlot()
  self:selectSlot(id%10+1)
end

function t:prevSlot()
  self:selectSlot((id+8)%10+1)
end

local tbmouse = false

function t:toolbarmouse(x,y,it,state)
  local cx, cy = whereInGrid(x,y,toolbarGrid)
  if cx then
    if isMDown(1) then
      cursor("handpress")
    else
      cursor("handrelease")
    end
    
    if state == "pressed" and not it then
      tbmouse = true
    end
    
    if not it and not tbmouse then return end
    
    if cy < 11 then --Tile slots
      self:selectSlot(cy)
    else --Tool
      self:selectTool(cy-11)
      
      if selectedTool < 3 and selsx then
        selsx, selsy, selex, seley = nil,nil,nil,nil
        self:drawMap()
      end
    end
  else
    local curCursor = cursor()
    if curCursor == "handpress" or curCursor == "handrelease" then
      cursor("normal")
    end
  end
  
  if state == "released" then tbmouse = false end
end

local mpmouse = false
local lastPX, lastPY = 0,0
local panflag = false
local selxf,selyf = false, false

function t:mapmouse(x,y,it,state,dx,dy)
  if selectedTool == 4 or isKDown("lalt","ralt") then return end
  if isInRect(x,y,mapRect) then
    
    if selectedTool == 2 then
      cursor("hand")
    else
      cursor("normal")
    end
    
    if state == "pressed" and not it then
      mpmouse = true
    end
    
    if not it and not mpmouse then return end
    
    local cx = math.floor((x-mapdx)/8)
    local cy = math.floor((y-8-mapdy)/8)
    
    --Pencil
    if selectedTool == 0 then
      Map:cell(cx,cy,hotbarTiles[selectedSlot])
      self:drawMap()
      
    --Bucket (Fill)
    elseif selectedTool == 1 then
      queuedFill(Map,cx,cy,hotbarTiles[selectedSlot])
      self:drawMap()
    
    --Pan (Hand)
    elseif selectedTool == 2 then
      if state == "pressed" then
        panflag = true
      elseif state == "moved" and panflag then
        local pdx, pdy = dx+lastPX, dy+lastPY
        
        lastPX, lastPY = pdx-math.floor(pdx), pdy-math.floor(pdy)
        
        pdx, pdy = math.floor(pdx), math.floor(pdy)
        
        mapdx, mapdy = mapdx+dx, mapdy+dy
        
        self:drawMap()
      elseif state == "released" then
        panflag = false
      end
      
    --Selection
    elseif selectedTool == 3 then
      if state == "pressed" or not selsx then
        if cx < 0 or cy < 0 or cx >= MapW or cy >= MapH then
          selsx, selsy, selex, seley = nil,nil,nil,nil
          self:drawMap()
        else
          selxf, selyf = false, false
          selsx, selsy = cx, cy
        end
      end
      
      if selsx then
        if selxf then
          selsx = cx
          if selsx == selex then
            selxf = false
          end
        else
          selex = cx
          if selex < selsx then
            selsx, selex = selex, selsx
            selxf = true
          end
        end
        
        if selyf then
          selsy = cy
          if selsy == seley then
            selyf = false
          end
        else
          seley = cy
          if seley < selsy then
            selsy, seley = seley, selsy
            selyf = true
          end
        end
        
        selex = math.max(0,selex)
        seley = math.max(0,seley)
        selex = math.min(MapW-1,selex)
        seley = math.min(MapH-1,seley)
        
        selsx = math.max(0,selsx)
        selsy = math.max(0,selsy)
        selsx = math.min(MapW-1,selsx)
        selsy = math.min(MapH-1,selsy)
      end
      
      self:drawMap()
    end
  else
    if cursor() == "hand" then
      cursor("normal")
    end
  end
  
  if state == "released" then mpmouse = false end
end

local mmouse = false

function t:menumouse(x,y,it,state)
  if selectedTool ~= 4 and not isKDown("lalt","ralt")  then
    mmouse = false; return
  end
  
  local cx, cy = whereInGrid(x,y,spritesGrid)
  if cx then
    
    if cy > 11 and cx > 14 then
      cursor("cross")
    elseif isMDown(1) then
      cursor("handpress")
    else
      cursor("handrelease")
    end
    
    if state == "pressed" and not it then
      mmouse = true
    end
    
    if not it and not mmouse then return end
    
    local tid = cy*22+cx-23
    if tid <= 255 then
      hotbarTiles[selectedSlot] = tid
      self:drawMenu()
    end
  elseif x < swidth-9 and y >= 8 then
    local curCursor = cursor()
    if curCursor == "handpress" or curCursor == "handrelease" then
      cursor("normal")
    end
  end
  
  if state == "released" then mmouse = false end
end


function t:mousepressed(x,y,b,it)
  self:toolbarmouse(x,y,it,"pressed")
  self:mapmouse(x,y,it,"pressed")
  self:menumouse(x,y,it,"pressed")
end

function t:mousemoved(x,y,dx,dy,it)
  self:toolbarmouse(x,y,it,"moved",dx,dy)
  self:mapmouse(x,y,it,"moved",dx,dy)
  self:menumouse(x,y,it,"moved",dx,dy)
end

function t:mousereleased(x,y,b,it)
  self:toolbarmouse(x,y,it,"released")
  self:mapmouse(x,y,it,"released")
  self:menumouse(x,y,it,"released")
end

function t:keypressed(key, sc, isrep)
  if (key == "lalt" or key == "ralt") and not isrep then
    if selectedTool ~= 4 then self:drawMenu() end
    local mx, my = getMPos(); self:mousemoved(mx,my,0,0,isMobile())
  end
end

function t:keyreleased(key, sc)
  if key == "lalt" or key == "ralt" then
    if selectedTool ~= 4 then self:drawMap() end
    local mx, my = getMPos(); self:mousemoved(mx,my,0,0,isMobile())
  end
end

function t:export()
  return Map:export()
end

function t:encode()
  return BinUtils.mapToBin(Map)
end

function t:import(data)
  if data then
    Map:import(data)
  else
    Map = MapObj(MapW,MapH)
  end
end

function t:decode(data)
  BinUtils.binToMap(Map,data)
end

local mvspeed = 64

function t:update(dt)
  
  --Auto scroll in select tool.
  if selectedTool == 3 and isMDown(1) then
    local mx, my = getMPos()
    
    if mx < swidth-8 and my > 7 then
      local sd = false --should draw
      
      if mx < 7 then
        mapdx = mapdx + mvspeed*dt
        sd = true
      elseif mx > swidth-14 then
        mapdx = mapdx - mvspeed*dt
        sd = true
      end
      
      if my < 8+6 then
        mapdy = mapdy + mvspeed*dt
        sd = true
      elseif my > sheight-6 then
        mapdy = mapdy - mvspeed*dt
        sd = true
      end
      
      if sd then
        self:mousemoved(mx,my,0,0,isMobile())
      end
    end
  end
end

t.keymap = {
  --Hotbar
  ["1"] = function(self) self:selectSlot(1) end,
  ["2"] = function(self) self:selectSlot(2) end,
  ["3"] = function(self) self:selectSlot(3) end,
  ["4"] = function(self) self:selectSlot(4) end,
  ["5"] = function(self) self:selectSlot(5) end,
  ["6"] = function(self) self:selectSlot(6) end,
  ["7"] = function(self) self:selectSlot(7) end,
  ["8"] = function(self) self:selectSlot(8) end,
  ["9"] = function(self) self:selectSlot(9) end,
  ["0"] = function(self) self:selectSlot(10) end
}

return t