mcrawl

local event = require("event")
local term = require("term")
local component = require("component")
local gpu = component.gpu
local mry
local mrx
mrx, mry = gpu.maxResolution()
gpu.setResolution(mrx,mry)

if ( mrx > 40 ) then mrx = 40 end
if ( mry > 20 ) then mry = 20 end

local quit_program = false

local mapTiles
local viewChange = true
local infoChange = true
local logChange = true


local log_y_size = math.ceil(mry / 3)
local INFO_X_SIZE = 20

local view_max_x = mrx - INFO_X_SIZE
local view_max_y = mry - ( log_y_size + 1 )

local map_max_x = 50
local map_max_y = 50

local max_rooms = 10
local rooms = {}
local floorItems = {}

local moved = false
local endTurn = false


local player = {
   x = 10,
   y = 10,
   name = "Steve",
   maxHP = 10,
   HP = 10,
   level = 1,
   xp = 0,
   defence = 0,
   food = 10,
   maxFood = 10,
   emeralds = 0,


   weapon = nil,
   armor = nil,

   inventory = {}

}

-- MAP GENERATION

function drawSquare(x,y,x2,y2)
   xadder = 1
   yadder = 1
   if ( x > x2 ) then xadder = -1 end
   if ( y > y2 ) then yadder = -1 end

   for mx = x,x2,xadder do
      for my = y,y2,yadder do
         if ( my >= 1 and my <= map_max_y and mx >= 1 and mx <= map_max_x ) then
            if ( math.random() > 0.1 ) then mapTiles[mx][my] = "."
            else mapTiles[mx][my] = "," end
         end
      end
   end
end

function doRoomsIntersect(room1,room2)
   if ( room1.x-2 < room2.x2 and room1.x2+2 > room2.x and room1.y-2 < room2.y2 and room1.y2+2 > room2.y ) then return true end
   return false
end

function isRoomOK(room)
   for i,croom in pairs(rooms)    do

      if ( doRoomsIntersect(room,croom)) then return false end
   end
   return true
end


function drawRoom(room)
   x = room.x
   y = room.y
   x2 = room.x2
   y2 = room.y2
   drawSquare(x,y,x2,y2)
end


function NewRoom()
   local width = math.random(4,8)
   local height = math.random(4,8)
   local mx = math.random(2,map_max_x-width-1)
   local my = math.random(2,map_max_y-height-1)
   room = {
      x = mx,
      y = my,
      w = width,
      h = height,
      x2 = mx + width - 1,
      y2 = my + height - 1
   }
   return room
end

function getRandomSpot(room)
   local x = room.x + math.random(0,room.w-1)
   local y = room.y + math.random(0,room.h-1)
   return x,y
end

function drawPath(room1,room2)
   startX, startY = getRandomSpot(room1)
   endX, endY = getRandomSpot(room2)

   if ( math.random() > 0.5 ) then
      drawSquare(startX,startY,endX,startY)
      drawSquare(endX,startY,endX,endY)
   else
      drawSquare(startX,startY,startX,endY)
      drawSquare(startX,endY,endX,endY)
   end

end

-- FOG OF WAR

function getPlayerRoom()
   for i,room in pairs(rooms) do
      if ( player.x >= room.x and player.x <= room.x2 and player.y >= room.y and player.y <= room.y2 ) then return room end
   end
   return 0
end

function revealRoom(room)
   room.revealed = true
   for x=room.x-1,room.x2+1 do
      for y=room.y-1,room.y2+1 do
         revealMap[x][y] = 1
      end
   end
end

function revealTunnel()
   for x=player.x-1,player.x+1 do
      for y=player.y-1,player.y+1 do
         revealMap[x][y] = 1
      end
   end
end

function generateMap()
   mapTiles = {}
   revealMap = {}
   for x=1,map_max_x do  -- Clear Map
      mapTiles[x] = {}
     revealMap[x] = {}
      for y=1,map_max_y do
         mapTiles[x][y] = "#"
       revealMap[x][y] = 0
      end
   end

   local totalRooms = 0
   for i=1,max_rooms do
      local newRoom = NewRoom()
      local tries = 1
      while ( not isRoomOK(newRoom) and tries < 10 ) do
         tries = tries + 1
         newRoom = NewRoom()
      end
      if ( isRoomOK(newRoom) ) then
      table.insert(rooms,newRoom)
      totalRooms = totalRooms + 1
     local ix, iy = getRandomSpot(newRoom)
     addNewFloorItem("Apple",ix,iy)
   end
   end

   max_rooms = totalRooms

   startRoom = rooms[1]
   player.x, player.y = getRandomSpot(startRoom)
   revealRoom(startRoom)

   for i=1,max_rooms do
     drawRoom(rooms[i])
   end
   for i=1,max_rooms-1 do
     drawPath(rooms[i],rooms[i+1])
   end
end



function getMapTile(x,y)

   if ( x == player.x and y == player.y ) then return "@" end

   if ( revealMap[x][y] == 0 ) then return "*" end

   local item = getItemAt(x,y)
   if ( item ~= nil ) then return item.icon end

   return mapTiles[x][y]
end




function changeReveal()
   local playerRoom = getPlayerRoom()
   if ( playerRoom ~= 0 ) then
      if ( playerRoom.revealed ~= true ) then
         revealRoom(playerRoom)
      end
   else
      revealTunnel()
   end
end

-- SCREEN DRAWING

function drawView()
   local px = math.ceil(view_max_x / 2 )
   local py = math.ceil(view_max_y / 2 )


   local x_start = 1 + player.x - px
   if ( x_start < 1 ) then x_start = 1 end
   if ( x_start + view_max_x > map_max_x ) then x_start = map_max_x - view_max_x end
   local x_end = x_start + view_max_x

   local dy = player.y - py
   if ( dy < 0 ) then dy = 0 end
   if ( dy + view_max_y > map_max_y ) then dy = map_max_y - view_max_y end

   for y = 1, view_max_y, 1 do
     --local line = mapstring[y+dy]
    --local display = string.sub(line,x_start,x_end)
    term.setCursor(1,y)

    for x = x_start, x_end do term.write(getMapTile(x,y+dy)) end
   end

   --term.setCursor(player.x - ( x_start - 1 ), player.y - dy )
   --term.write("@")

end

function drawInfo()
   local sx = view_max_x+4
   term.setCursor(sx,1)
   term.write("Player : " .. player.name)
   term.setCursor(sx,2)
   term.write("Health : " .. player.HP .. "/" .. player.maxHP)
   term.setCursor(sx,3)
   term.write("Defence : " .. player.defence)
   term.setCursor(sx,4)
   term.write("Food : " .. math.ceil(player.food) .. "/" .. player.maxFood)
   term.setCursor(sx,5)
   term.write("Level : " .. player.level)
   term.setCursor(sx,6)
   term.write("EXP : " .. player.xp )
   term.setCursor(sx,7)
   term.write("Weapon: ")
   if ( player.weapon == nil ) then term.write("Fists") else term.write(weapon.name) end
   term.setCursor(sx,8)
   term.write("Armor: ")
   if ( player.armor == nil ) then term.write("Clothes") else term.write(armor.name) end
   term.setCursor(sx,9)
   term.write("Emeralds: " .. player.emeralds)


end

function drawLog()
   local logI = 0
   for y = view_max_y+3,mry do
     term.setCursor(2,y)
    term.clearLine()
    term.setCursor(2,y)
     logI = logI + 1

     if ( mLog[logI] ~= nil ) then term.write(mLog[logI]) end
   end
end

function drawScreen()
   if (viewChange) then drawView() end
   if (infoChange) then drawInfo() end
   if (logChange) then drawLog() end
end

local showingInv = false

function showInventory()
   showingInv = true

   top = string.rep("=",mrx-6)
   mid = "|" .. string.rep(" ",mrx-8) .. "|"
   term.setCursor(3,3)
   term.write(top)
   term.setCursor(3,mry-3)
   term.write(top)
   for y=4,mry-4 do
      term.setCursor(3,y)
      term.write(mid)
   end
   term.setCursor(math.ceil((mrx/2) - 4), 4)
   term.write("INVENTORY")
   for i,item in pairs(player.inventory) do
      if ( i < mry-6 ) then
         term.setCursor(6, i+5)
         term.write(item.name)
         if ( item.quant > 1 ) then term.write(" x".. item.quant) end
      end

   end


end

-- PLAYER MOVEMENT

function movePlayer(direction)
   if ( direction == "up"    and player.y > 1  ) then tryToMovePlayer(player.x,player.y-1) end
   if ( direction == "down"  and player.y < map_max_y ) then tryToMovePlayer(player.x,player.y+1) end
   if ( direction == "left"  and player.x > 1  ) then tryToMovePlayer(player.x-1,player.y) end
   if ( direction == "right" and player.x < map_max_x ) then tryToMovePlayer(player.x+1,player.y)  end
   if ( direction == "nw"    and player.x > 1 and
                                 player.y > 1 ) then tryToMovePlayer(player.x-1,player.y-1) end
   if ( direction == "ne"    and player.x < map_max_x and
                                 player.y > 1 ) then tryToMovePlayer(player.x+1,player.y-1) end
   if ( direction == "sw"    and player.x > 1 and
                                 player.y < map_max_y ) then tryToMovePlayer(player.x-1,player.y+1) end
   if ( direction == "se"    and player.x < map_max_x and
                                 player.y < map_max_y ) then tryToMovePlayer(player.x+1,player.y+1) end



end

function tryToMovePlayer(newx,newy)
   tile = getMapTile(newx,newy)

   if ( tile == "#" ) then return end
   player.x = newx
   player.y = newy
   viewChange = true
   changeReveal()
   moved = true
   endTurn = true

end

function isKey(chara,str)
   return ( chara == string.byte(str) or chara == string.byte(string.upper(str)) )
end

mLog = {}
logSize = 0
function addLog(message)
    logSize = logSize + 1
   table.insert(mLog,message)
    if ( logSize >= log_y_size ) then
      logSize = logSize - 1
      table.remove(mLog,1)
   end
   logChange = true
end

function handleKey(address,chara,code,pname)
   --print(chara,code)
   if ( player.name == "Steve" ) then
      player.name = pname
      addLog(player.name .. " has entered the dungeon!")
      infoChange = true
   end
   if ( isKey(chara,"1") ) then quit_program = true end

   if ( showingInv ) then
     showingInv = false
     viewChange = true
     logChange = true
     infoChange = true
     term.clear()
     return
   end

   if ( isKey(chara,"w") or code == 200 ) then movePlayer("up") end
   if ( isKey(chara,"x") or code == 208 ) then movePlayer("down") end
   if ( isKey(chara,"a") or code == 203 ) then movePlayer("left") end
   if ( isKey(chara,"d") or code == 205 ) then movePlayer("right") end
   if ( isKey(chara,"q") or code == 199 ) then movePlayer("nw") end
   if ( isKey(chara,"e") or code == 201 ) then movePlayer("ne") end
   if ( isKey(chara,"z") or code == 207 ) then movePlayer("sw") end
   if ( isKey(chara,"c") or code == 209 ) then movePlayer("se") end
   if ( isKey(chara,"g") ) then playerGet() end
   if ( isKey(chara,"i") ) then showInventory() end
end

function eventHandler(eventID,...)
   if ( eventID == "key_down" ) then handleKey(...) end
end

-- ITEMS

function getItemAt(x,y)
   for i,item in pairs(floorItems) do
      if ( item.x == x and item.y == y ) then
      item.id = i
      return item
     end
   end
end

function addNewFloorItem(itemType,x,y,quantity)
   local newItem = newItem(itemType,quantity)
   newItem.x = x
   newItem.y = y
   table.insert(floorItems,newItem)
end

function playerDropItem(item)
   table.remove(player.inventory,item.id)
   item.x = player.x
   item.y = player.y
   table.insert(floorItems,item)
   if ( player.weapon == item ) then player.weapon = nil end
   if ( player.armor == item ) then player.armor = nil end
end

function getPlayerItem(item)
   for i,sitem in pairs(player.inventory) do

      if ( item.name == sitem.name ) then
         sitem.id = i
         return sitem
      end
   end
end

function playerGetItem(item)

   table.remove(floorItems,item.id)
   if ( item.iType == "emerald" ) then
      player.emeralds = player.emeralds + item.quantity
      return
   end

   if ( isItemStackable(item) ) then
      existingItem = getPlayerItem(item)
      if ( existingItem ~= nil ) then
        if ( existingItem.quant >= 64 ) then return false end
        existingItem.quant = math.min(existingItem.quant + item.quant,64)
        return true
      end
   end
   table.insert(player.inventory,item)
   return true
end

function newItem(itemType,quantity)
   if ( quantity == nil or quantity < 1 ) then quantity = 1 end
   local retItem = {
      name = itemType,
      quant = quantity
   }

   if ( itemType == "Emerald" ) then retItem.iType = "emerald" end

   if ( itemType == "Apple" ) then
      retItem.iType = "food"
      retItem.value = 2
   end


   if ( retItem.iType == "food" ) then retItem.icon = "%" end
   if ( retItem.iType == "weapon" ) then retItem.icon = "/" end
   if ( retItem.iType == "armor" ) then retItem.icon = "]" end
   if ( retItem.iType == "potion" ) then retItem.icon = "!" end
   if ( retItem.iType == "emerald" ) then retItem.icon = "$" end
   if ( retItem.iType == "craft" ) then retItem.icon = "^" end

   if ( retItem.iType == nil ) then addLog("Bad item: ".. itemType) end
   if ( retItem.icon == nil ) then addLog("Bad itemtype: " .. retItem.iType) end

   return retItem

end

function isItemStackable(item)
   if ( item.name == "Arrows" ) then return true end
   if ( item.iType == "weapon" or item.iType == "armor" ) then return false end
   return true

end

-- player actions
function resolveMovement()
end

function playerGet()
   item = getItemAt(player.x,player.y)
   if ( item == nil ) then
      addLog("Nothing to pick up here.")
      return
   end
   if ( playerGetItem(item) ) then
     addLog("You picked up the " .. item.name)
   else
     addLog("You can't hold any more.")
   end
end

-- Monsters?
function resolveTurn()
end

-- MAIN LOOP


generateMap()
term.clear()

while ( quit_program == false ) do
   drawScreen()
   viewChange = false
   logChange = false
   infoChange = false
   moved = false
   endTurn = false
   eventHandler(event.pull())
   if ( moved == true ) then resolveMovement() end
   if ( endTurn == true ) then resolveTurn() end

end