spwaves

--Tank
--[[ Info:
  Welcome to World of Tanks. Your objective is to
  eliminate all tanks from the playing field.

  Enemys:
  Red - Moves
  Green - Shoots at you
  Orange - Shoots and moves

  Help:
  Ammo Packs
  Health Packs
  Extra Support (Not Yet Implemented)

  Information about the player.stat file:
  The player.stat file holds the following info
  in the file in the following order:
    Name
    Campaign Level
    Highscore
    Total Lifetime Shots
    Total Lifetime Enemies Killed
    Total Lifetime Deaths
    Wave Number
]]--

local w,h = term.getSize()

running = true

refresh = 0.10
gtId = os.startTimer(refresh)

brefresh = 1.5
gtId2 = os.startTimer(brefresh)

brand = math.random(0.10, 1.5)
gtId3 = os.startTimer(brand)

hrefresh = 3
gtId4 = os.startTimer(hrefresh)

levelprogress = 1
tries = 1

--Opens the player stat file--
h = fs.open("Tanks/player.stat", "r")
name = h.readLine()
level = h.readLine()
highscore = h.readLine()
tlbullets = h.readLine()
tlenemies = h.readLine()
tldeaths = h.readLine()
wavenum = h.readLine()

lbullets = tlbullets
lenemies = tlenemies
ldeaths = tldeaths

local w,h = term.getSize()

wave = 1

tankInfo = {
  x = 22,
  y = h - 5,
  hp = 15,
  maxhp = 15,
  ammo = 10
}

ammoamount = tostring(tankInfo.ammo)

bullets = { }
ent = { }
ammopack = { }
healthpack = { }

bulletColor = {
  colors.white,
  colors.gray,
  colors.red,
  colors.green,
  colors.orange
}

tankBullet = bulletColor[1]
enemyBullet = bulletColor[4]
ent3Bullet = bulletColor[5]

function createBullets(x, y, xvel, yvel, typeBullets, bdmg)
  table.insert(bullets, {
    x = x,
    y = y,
    xstep = xvel,
    ystep = yvel,
    type = typeBullets,
    dmg = bdmg
   })
end

function createEnt(x, y, typeEnt, entNumber, xvel, yvel, entId)
  table.insert(ent, {
  x = x,
  y = y,
  color = typeEnt,
  newEnt = entNumber,
  xstep = xvel,
  ystep = yvel,
  ID = entId
  })
end

function createAmmo(x, y, amount)
  table.insert(ammopack, {
  x = x,
  y = y,
  amount = amount
  })
end

function createHPack(x, y, amount)
  table.insert(healthpack, {
  x = x,
  y = y,
  amount = amount
  })
end

function tank()
term.setTextColor(colors.white)
  term.setCursorPos(tankInfo.x, tankInfo.y)
  print(" | ")
  term.setCursorPos(tankInfo.x, tankInfo.y + 1)
  print("*-*")
  term.setCursorPos(tankInfo.x, tankInfo.y + 2)
  print("| |")
  term.setCursorPos(tankInfo.x, tankInfo.y + 3)
  print("*-*")
end

function enemy()
for _, v in pairs(ent) do
term.setTextColor(v.color)
  term.setCursorPos(v.x, v.y)
  print(" | ")
  term.setCursorPos(v.x, v.y - 1)
  print("*-*")
  term.setCursorPos(v.x, v.y - 2)
  print("| |")
  term.setCursorPos(v.x, v.y - 3)
  print("*-*")
 end
end

function ammoPack()
for _, v in pairs(ammopack) do
  term.setTextColor(colors.yellow)
  term.setCursorPos(v.x, v.y)
  print("[]")
end
end

function healthPack()
for _, v in pairs(healthpack) do
  term.setTextColor(colors.red)
  term.setCursorPos(v.x, v.y)
  print("[]")
end
end

function updateEnt()
  if #ent > 0 then
    for i = 1, #ent do
      if ent[i] then
        local ents = ent[i]
        if ents.ID == 1 then
          moveEnemy1(i, 5)
        elseif ents.ID == 3 then
          moveEnemy1(i, 10)
        end
      end
    end
  end
end

function updateEnt2()
  if #ent > 0 then
    for i = 1, #ent do
      if ent[i] then
        local ents = ent[i]
        if ents.ID == 2 then
          enemyBullets(i)
        end
      end
    end
  end
end

function updateEnt3()
  if #ent > 0 then
    for i = 1, #ent do
      if ent[i] then
        local ents = ent[i]
        if ents.ID == 3 then
          enemy3Bullets(i)
        end
      end
    end
  end
end


function checkForCollison()
  if #ent > 0 then
    for i = 1, #ent do
      if ent[i] then
        local ents = ent[i]
        if (ents.x <= tankInfo.x + 2 and ents.x + 2 >= tankInfo.x) and (ents.x + 2 <= tankInfo.x + 2 and ents.x >= tankInfo.x)
         and (ents.y >= tankInfo.y and ents.y + 3 <= tankInfo.y - 3) then
          running = false
          endGame("Tip: You should not run head first into battle")
        end
      end
    end
  end
end

function updateAmmo()
  for _, v in pairs(ammopack) do
    if v.x > tankInfo.x then
      if tankInfo.x + 2 >= v.x and v.y >= tankInfo.y and v.y <= tankInfo.y + 2 then
        tankInfo.ammo = tankInfo.ammo + v.amount
        table.remove(ammopack, 1)
        break
      end
    elseif v.x < tankInfo.x then
      if tankInfo.x <= v.x and v.y >= tankInfo.y and v.y <= tankInfo.y + 2 then
        tankInfo.ammo = tankInfo.ammo + v.amount
        table.remove(ammopack, 1)
        break
      end
    end
  end
end

function updateHealth()
  for _, v in pairs(healthpack) do
    if v.x > tankInfo.x then
      if tankInfo.x + 2 >= v.x and v.y >= tankInfo.y and v.y <= tankInfo.y + 2 then
        tankInfo.hp = tankInfo.hp + v.amount
        table.remove(healthpack, 1)
        break
      end
    elseif v.x < tankInfo.x then
      if tankInfo.x <= v.x and v.y >= tankInfo.y and v.y <= tankInfo.y + 2 then
        tankInfo.hp = tankInfo.hp + v.amount
        table.remove(healthpack, 1)
        break
      end
    end
  end
end

function endGame(deathmsg)
    local select = 1
  function optionsdraw()
    term.clear()
    term.setTextColor(colors.black)
    term.setBackgroundColor(colors.white)
    term.clear()
    bg = paintutils.loadImage("Tanks/bgt")
    paintutils.drawImage(bg, 1,1)
    term.setBackgroundColor(colors.white)
    term.setCursorPos(w/2 - 4, 11)
    print("GAME OVER")
    term.setCursorPos(4, 13)
    print(deathmsg)
    term.setCursorPos(w/2 - 8, 15)
    print("Retry Last Level")
    term.setCursorPos(w/2 - 4, 17)
    print("Quit Game")
    term.setTextColor(colors.black)
      if select == 1 then
        term.setCursorPos(w/2 - 9, 15)
        print("[")
        term.setCursorPos(w/2 + 8, 15)
        print("]")
      else
        term.setCursorPos(w/2 - 5, 17)
        print("[")
        term.setCursorPos(w/2 + 5, 17)
        print("]")
      end
  end
  optionsdraw()
    while true do
      local event, p1 = os.pullEventRaw()
        if event == "key" then
          if p1 == keys.w or p1 == keys.s or p1 == keys.up or p1 == keys.down then
            select = -select
            optionsdraw()
          elseif p1 == keys.enter or p1 == keys.space then
            if select == 1 and tries <= 2 then
              running = true
              tankInfo.x = w/2
              tankInfo.y = h - 5
              tankInfo.hp = tankInfo.maxhp
              tankInfo.ammo = 10
              term.setBackgroundColor(colors.black)
              term.setTextColor(colors.white)
                for i = 1, #bullets do
                  table.remove(bullets, i)
                end
                for i = 1, #healthpack do
                  table.remove(healthpack, i)
                end
                gtId = os.startTimer(refresh)
                gtId2 = os.startTimer(brefresh)
                gtId3 = os.startTimer(brand)
                gtId4 = os.startTimer(hrefresh)
              drawBg()
              tries = tries + 1
              ldeaths = ldeaths + 1
              break
            else
            ldeaths = ldeaths + 1
            w = fs.open("Tanks/player.stat", "w")
            w.writeLine(name)
            w.writeLine(levelprogress)
            w.writeLine(highscore)
            w.writeLine(lbullets)
            w.writeLine(lenemies)
            w.writeLine(ldeaths)
            w.writeLine(wavenum)
            w.flush()
            w.close()
              term.clear()
              term.setTextColor(colors.white)
              term.setBackgroundColor(colors.black)
              term.clear()
              term.setCursorPos(1,1)
              print("Thank you for playing")
              return
            end
          end
        end
    end
  end

function drawBg()
term.clear()
  for i = 1, w do
  term.setTextColor(colors.white)
    term.setCursorPos(w - (i), h - 1)
    write("-")
  end
  for i = 1, h do
    term.setTextColor(colors.white)
      term.setCursorPos(5, h - (i))
      write("|")
  end
  function hpbar()
  for i = 1, (tankInfo.hp/tankInfo.maxhp)*5 do
    term.setBackgroundColor(colors.red)
    term.setCursorPos(2, 15 - (i))
    write("  ")
    term.setBackgroundColor(colors.black)
  end
  end
  term.setCursorPos(1, 9)
  print("*--*")
  for i = 1, 5 do
    term.setTextColor(colors.white)
    term.setCursorPos(1, 15 - (i))
    write("|  |")
    hpbar()
  end
  term.setCursorPos(1,15)
  print("*--*")
  term.setCursorPos(1,2)
  print("Tank")
  print("    ")
  print("Ammo")
  print(tankInfo.ammo)
  print("    ")
  print("HP ")
  term.setCursorPos(2,8)
  print(tankInfo.hp)
term.setCursorPos(1,h)
print("Tank Info -     Name: "..name.." Wave: "..wave)
tank()

  for _, v in pairs(bullets) do
    term.setCursorPos(v.x, v.y)
    term.setTextColor(v.type)
    term.write(".")
    term.setTextColor(colors.black)
  end
  --Draws all the desired enemies on the screen
    enemy()
    ammoPack()
    healthPack()
end

function createNewWave()
wave = wave + 1
wavenum = wave
entrand = math.random(1,8)
 for i = 1, entrand do
   entnum = 1
   idrand = math.random(1,3)
   xrand = math.random(5, 40)
     if idrand == 1 then
     xrand = math.random(5, 40)
       createEnt(xrand, 5, colors.red, entnum, 1,0,1)
     elseif idrand == 2 then
       createEnt(xrand, 10, colors.green, entnum, 0,0,2)
     elseif idrand == 3 then
       createEnt(xrand, 10, colors.orange, entnum, 1,0,3)
     end
     entnum = entnum + 1
 end
end

tank()
drawBg()
  enemy()

function input()
local event, key = os.pullEvent()
  if event == "key" then
    if key == keys.q then
    running = false
    elseif key == keys.w then
      tankInfo.y = tankInfo.y - 1
    elseif key == keys.s then
      tankInfo.y = tankInfo.y + 1
    elseif key == keys.a then
      tankInfo.x = tankInfo.x - 1
    elseif key == keys.d then
      tankInfo.x = tankInfo.x + 1
    elseif key == keys.up then
      tankInfo.y = tankInfo.y - 1
    elseif key == keys.down then
      tankInfo.y = tankInfo.y + 1
    elseif key == keys.left then
      tankInfo.x = tankInfo.x - 1
    elseif key == keys.right then
      tankInfo.x = tankInfo.x + 1
    elseif key == keys.space then
      if tankInfo.ammo > 0 then
        createBullets(tankInfo.x + 1, tankInfo.y - 1, 0, -1, tankBullet,0)
        tankInfo.ammo = tankInfo.ammo - 1
        lbullets = lbullets + 1
      end
      drawBg()
      tank()
        enemy()
        updateBullets()
        checkForCollison()
        updateAmmo()
    end
    term.setBackgroundColor(colors.black)
    term.setCursorPos(tankInfo.x, tankInfo.y)
    term.clearLine()
    term.setCursorPos(tankInfo.x, tankInfo.y + 1)
    term.clearLine()
    term.setCursorPos(tankInfo.x, tankInfo.y + 2)
    term.clear()
    tank()
    drawBg()
    if tankInfo.x == w - 3 then
      tankInfo.x = 22
      tankInfo.y = h - 5
    elseif tankInfo.x == 5 then
      tankInfo.x = 22
      tankInfo.y = h - 5
    elseif tankInfo.y == h - 3 then
      tankInfo.x = 22
      tankInfo.y = h - 5
    elseif tankInfo.y == 1 then
      tankInfo.x = 22
      tankInfo.y = h - 5
    end
  elseif event == "timer" then
    if key == gtId then
      updateBullets()
      updateEnt()
      updateAmmo()
      updateHealth()
      checkForCollison()
      gtId = os.startTimer(refresh)
      drawBg()
      if tankInfo.hp <= 0 then
        running = false
        endGame("You kind of got shot")
      end
      if levelprogress == 14 then
        createAmmo(20, 20, 20)
      elseif levelprogress == 16 then
        createAmmo(20, 20, 20)
      else
        if tankInfo.ammo <= 3 then
          if #ammopack == 0 then
          xrand = math.random(7,48)
          yrand = math.random(2,h-5)
          arand = math.random(0,20)
          if arand > 0 then
            createAmmo(xrand, yrand, arand)
          elseif arand == 0 then
            running = false
            endGame("You took all the bullets from your camp")
          end
          end
        end
      end
    elseif key == gtId2 then
      updateEnt2()
      gtId2 = os.startTimer(brefresh)
      if #ent <= 0 then
        createNewWave()
      end
    elseif key == gtId3 then
      updateEnt3()
      brand = math.random(0.10, 2.5)
      gtId3 = os.startTimer(brand)
    elseif key == gtId4 then
      updateHealth()
      if levelprogress == 14 then
        if tankInfo.hp <= 5 then
          createHPack(10, 10, 10)
        elseif tankInfo.hp == 10 then
          createHPack(10, 10, 5)
        end
      elseif levelprogress == 16 then
        if tankInfo.hp <= 5 then
          createHPack(10, 10, 10)
        elseif tankInfo.hp == 10 then
          createHPack(10, 10, 5)
        end
      else
      if tankInfo.hp <= 5 then
        chances = math.random(0, 50)
        if #healthpack == 0 then
          if chances < 20 then
          xrand = math.random(7, 49)
          yrand = math.random(2, (h - 5))
          createHPack(xrand, yrand, 5)
          end
        end
      end
      end
      gtId4 = os.startTimer(hrefresh)
    end
  end
end

function updateBullets()
  for _, v in pairs(bullets) do
    v.x = v.x + v.xstep
    v.y = v.y + v.ystep
    for i, k in pairs(ent) do
      if v.y == k.y and (v.x >= k.x - 2 and v.x <= k.x + 2) then
        entnum = #ent
        if entnum > 0 then
          table.remove(ent, i)
          lenemies = lenemies + 1
          break
        else
          ent = { }
        end
      elseif v.y == tankInfo.y and (v.x >= tankInfo.x and v.x <= tankInfo.x + 2) then
        if tankInfo.hp > 0 then
          tankInfo.hp = tankInfo.hp - v.dmg
          break
        elseif tankInfo.hp == 0 then
          running = false
          endGame("Your tank was obliterated")
        end
      end
    end
  end

  local blistl = #bullets
  for i = 1, blistl do
    local v = bullets[i]
    if v and (v.y < 1 or v.y > h or v.y > h - 4) then
      table.remove(bullets, i)
      blistl = blistl - 1
      i = i - 1
      break
    end
  end
end

function enemyBullets(nr)
local ents = ent[nr]
    createBullets(ents.x + 1, ents.y, 0, 1, enemyBullet, 5)
end

function enemy3Bullets(nr)
  local ents = ent[nr]
    createBullets(ents.x + 1, ents.y, 0, 1, ent3Bullet, 5)
end

function moveEnemy1(nr, xparam)

local ents = ent[nr]

  ents.x = ents.x + ents.xstep
  ents.y = ents.y + ents.ystep
  if ents.x == w - 2 then
    ents.xstep = -1
    ents.ystep = 0
    enemy()
  elseif ents.x == xparam then
    ents.xstep = 1
    ents.ystep = 0
  elseif (ents.y >= tankInfo.y and ents.y <= tankInfo.y + 2 and ents.x >= tankInfo.x and ents.x <= tankInfo.x + 2)
   and (ents.y - 2 >= tankInfo.y and ents.y - 2 <= tankInfo.y + 2 and ents.x + 2 >= tankInfo.x and ents.x + 2 <= tankInfo.x + 2) then
    running = false
    endGame("Your tank perished")
  end
end

  while running do
    input()
  end