Added in tetris. (#276)

* Added in tetris.

* Updated key combos.

* Made non-resizable.

* Forgot the t-block.

* Upped default size.

* Added in controllable down speed.

* Added in SRS rotations, pausing, holding.

* Added tag.

* Added drop shadow and lock delay.

2 files changed, 422 insertions, 0 deletions

diff --git a/manifest.json b/manifest.json
index 320f1bb..81e06c9 100644
--- a/manifest.json
+++ b/manifest.json
@@ -1305,6 +1305,14 @@
       "remote": "https://github.com/adamharrison/lite-xl-terminal.git:89788ede70768726cf60b5c3049f1fc67bea46cd"
     },
     {
+      "description": "Allows you to play tetris directly in the editor.",
+      "version": "0.1",
+      "path": "plugins/tetris.lua",
+      "id": "tetris",
+      "tags": ["game"],
+      "mod_version": 3
+    },
+    {
       "description": "Compile Tex files into PDF",
       "version": "0.1",
       "path": "plugins/texcompile.lua",
diff --git a/plugins/tetris.lua b/plugins/tetris.lua
new file mode 100644
index 0000000..b7b2258
--- /dev/null
+++ b/plugins/tetris.lua
@@ -0,0 +1,414 @@
+-- mod-version:3
+
+local core = require "core"
+local config = require "core.config"
+local style = require "core.style"
+local common = require "core.common"
+local View = require "core.view"
+local command = require "core.command"
+local keymap = require "core.keymap"
+
+local TetrisView = View:extend()
+
+config.plugins.tetris = common.merge({
+  tick = 0.5, -- The amount of time in seconds it takes for a piece to fall one line at 0 score.
+  height = 30, -- The amount of cells of height.
+  width = 10, -- The amount of cells of width.
+  cell_size = 18, -- The size in pixels of each cell.
+  cell_padding = 2, -- pixels between each cell
+  drop_shadow = true, -- Should cast a drop shadow.
+  lock_delay = 3, -- the multiplier for lock delay over a normal tick. set to 0 to disable
+  down_amount = 1 -- the amount we move a tetronimo down when you hit the down key; change to math.huge for instant.
+}, config.plugins.tetris)
+
+function TetrisView:new(options)
+  TetrisView.super.new(self)
+  self.cell_size = options.cell_size
+  self.cell_padding = options.cell_padding
+  self.grid = { x = options.width, y = options.height }
+  self.size.x = self.grid.x * (self.cell_size + self.cell_padding) + style.padding.x * 2
+  self.cells = { }
+  self.score = 0
+  self.paused = false
+  self.initial_tick = options.tick
+  self.lock_delay = options.lock_delay
+  self.drop_shadow = options.drop_shadow
+  self.tick = self:calculate_tick(self.score)
+  self.finished = false
+  self.thread = core.add_thread(function()
+    while not self.finished do
+      self:step()
+      core.redraw = true
+      coroutine.yield(self.tick)
+    end
+  end)
+  -- easier to specify rotations than to start doing matrix multiplication
+  self.tetronimos = {
+    {
+      color = { common.color "#ff0000" },
+      shape = { {
+        0,1,1,0,
+        0,1,1,0,
+        0,0,0,0,
+        0,0,0,0
+      } }
+    },
+    {
+      color = { common.color "#00ff00" },
+      shape = { {
+        1,1,0,0,
+        1,0,0,0,
+        1,0,0,0,
+        0,0,0,0
+      }, {
+        1,1,1,0,
+        0,0,1,0,
+        0,0,0,0,
+        0,0,0,0
+      }, {
+        0,0,1,0,
+        0,0,1,0,
+        0,1,1,0,
+        0,0,0,0
+      }, {
+        0,0,0,0,
+        1,0,0,0,
+        1,1,1,0,
+        0,0,0,0
+      } }
+    },
+    {
+      color = { common.color "#0000ff" },
+      shape = { {
+        1,0,0,0,
+        1,0,0,0,
+        1,1,0,0,
+        0,0,0,0
+      }, {
+        1,1,1,0,
+        1,0,0,0,
+        0,0,0,0,
+        0,0,0,0
+      }, {
+        0,1,1,0,
+        0,0,1,0,
+        0,0,1,0,
+        0,0,0,0
+      }, {
+        0,0,0,0,
+        0,0,1,0,
+        1,1,1,0,
+        0,0,0,0
+      } }
+    },
+    {
+      color = { common.color "#00ffff" },
+      shape = { {
+        1,1,0,0,
+        0,1,1,0,
+        0,0,0,0,
+        0,0,0,0
+      }, {
+        0,0,1,0,
+        0,1,1,0,
+        0,1,0,0,
+        0,0,0,0
+      }, {
+        0,  0,0,0,
+        1,1,0,0,
+        0,1,1,0,
+        0,0,0,0
+      }, {
+        0,1,0,0,
+        1,1,0,0,
+        1,0,0,0,
+        0,0,0,0
+      } }
+    },
+    {
+      color = { common.color "#ffff00" },
+      shape = { {
+        0,1,1,0,
+        1,1,0,0,
+        0,0,0,0,
+        0,0,0,0
+      }, {
+        0,1,0,0,
+        0,1,1,0,
+        0,0,1,0,
+        0,0,0,0
+      }, {
+        0,0,0,0,
+        0,1,1,0,
+        1,1,0,0,
+        0,0,0,0
+      }, {
+        1,0,0,0,
+        1,1,0,0,
+        0,1,0,0,
+        0,0,0,0
+      } }
+    },
+    {
+      color = { common.color "#ff00ff" },
+      shape = { {
+        0,1,0,0,
+        0,1,0,0,
+        0,1,0,0,
+        0,1,0,0
+      }, {
+        0,0,0,0,
+        1,1,1,1,
+        0,0,0,0,
+        0,0,0,0
+      }, {
+        0,0,1,0,
+        0,0,1,0,
+        0,0,1,0,
+        0,0,1,0
+      }, {
+        0,0,0,0,
+        0,0,0,0,
+        1,1,1,1,
+        0,0,0,0
+      } }
+    },
+    {
+      color = { common.color "#ffffff" },
+      shape = { {
+        1,1,1,0,
+        0,1,0,0,
+        0,0,0,0,
+        0,0,0,0
+      }, {
+        0,0,1,0,
+        0,1,1,0,
+        0,0,1,0,
+        0,0,0,0
+      }, {
+        0,0,0,0,
+        0,1,0,0,
+        1,1,1,0,
+        0,0,0,0
+      }, {
+        1,0,0,0,
+        1,1,0,0,
+        1,0,0,0,
+        0,0,0,0
+      } }
+    }
+  }
+  self.live_piece = nil
+  self.hold_piece = nil
+end
+
+function TetrisView:calculate_tick(score)
+  return self.initial_tick / (math.floor(score / 10) + 1)
+end
+
+
+function TetrisView:does_collide(x, y, tetronimo, rot)
+  local shape = tetronimo.shape[rot]
+  for i = 0, 3 do
+    for j = 0, 3 do
+      local ny = y + i
+      local nx = x + j
+      if (nx >= self.grid.x or ny >= self.grid.y or nx < 0 or ny < 0 or self.cells[self.grid.x * ny + nx + 1]) and shape[i * 4 + j + 1] == 1 then
+        return true
+      end
+    end
+  end
+  return false
+end
+
+function TetrisView:finalize_live_piece()
+  assert(self.live_piece)
+  local shape = self.live_piece.tetronimo.shape[self.live_piece.rot]
+  for i = 0, 3 do
+    for j = 0, 3 do
+      local ny = self.live_piece.y + i
+      local nx = self.live_piece.x + j
+      if shape[(i * 4 + j) + 1] == 1 then
+        self.cells[ny * self.grid.x + nx + 1] = self.live_piece.idx
+      end
+    end
+  end
+  for y = self.live_piece.y, math.min(self.live_piece.y + 4, self.grid.y - 1) do
+    local all_present = true
+    for x = 0, self.grid.x - 1 do
+      if not self.cells[y * self.grid.x + x + 1] then all_present = false end
+    end
+    if all_present then
+      self.score = self.score + 1
+      self.tick = self:calculate_tick(self.score)
+      for ny = y, 2, -1 do
+        for nx = 0, self.grid.x - 1 do
+          self.cells[ny * self.grid.x + nx + 1] = self.cells[(ny - 1) * self.grid.x + nx + 1]
+        end
+      end
+    end
+  end
+  self.live_piece = nil
+end
+
+
+function TetrisView:step()
+  if not self.finished and not self.paused then
+    if not self.live_piece then
+      local idx = self.next_piece or math.floor(math.random() * #self.tetronimos) + 1
+      self.live_piece = { tetronimo = self.tetronimos[idx], idx = idx, x = math.floor(self.grid.x / 2), y = 0, rot = 1 }
+      self.next_piece = math.floor(math.random() * #self.tetronimos) + 1
+      if (self:does_collide(self.live_piece.x, self.live_piece.y + 1, self.live_piece.tetronimo, self.live_piece.rot)) then
+        self:finalize_live_piece()
+        self.finished = true
+      end
+    else
+      if (self:does_collide(self.live_piece.x, self.live_piece.y + 1, self.live_piece.tetronimo, self.live_piece.rot)) then
+        self.live_piece.countup = (self.live_piece.countup or 0) + 1
+        if self.live_piece.countup > self.lock_delay then
+          self:finalize_live_piece()
+        end
+      else
+        self.live_piece.y = self.live_piece.y + 1
+        self.live_piece.countup = 0
+      end
+    end
+  end
+end
+
+function TetrisView:draw_tetronimo(posx, posy, tetronimo, rot, color)
+  local shape = tetronimo.shape[rot]
+  for y = 0, 3 do
+    for x = 0, 3 do
+      if shape[y * 4 + x + 1] == 1 then
+        renderer.draw_rect(posx + x * (self.cell_size + self.cell_padding), posy + y * (self.cell_size + self.cell_padding), self.cell_size, self.cell_size, color or tetronimo.color)
+      end
+    end
+  end
+end
+
+function TetrisView:draw()
+  self:draw_background(style.background3)
+  local lh = style.font:get_height()
+  local tx = self.position.x + style.padding.x
+  local ty = self.position.y + style.padding.y
+
+  renderer.draw_text(style.font, "Score: " .. self.score, tx, self.position.y + style.padding.y, style.normal)
+  local w = renderer.draw_text(style.font, "Next Piece", tx, self.position.y + style.padding.y + lh, style.normal)
+  if self.next_piece then
+    self:draw_tetronimo(tx, self.position.y + style.padding.y + lh * 2, self.tetronimos[self.next_piece], 1)
+  end
+  if self.held_piece then
+    self:draw_tetronimo(w + style.padding.x, self.position.y + style.padding.y + lh * 2, self.tetronimos[self.held_piece], 1)
+  end
+  renderer.draw_text(style.font, "Held Piece", w + style.padding.x, self.position.y + style.padding.y + lh, style.normal)
+  ty = ty + lh * 2 + (self.cell_size + self.cell_padding) * 4 + style.padding.y
+
+  renderer.draw_rect(tx, ty, (self.cell_size + self.cell_padding) * self.grid.x, (self.cell_size + self.cell_padding) * self.grid.y, style.background)
+  for y = 0, self.grid.y - 1 do
+    for x = 0, self.grid.x - 1 do
+      if self.cells[y * self.grid.x + x + 1] then
+        local color = self.tetronimos[self.cells[y * self.grid.x + x + 1]].color
+        renderer.draw_rect(tx + x * (self.cell_size + self.cell_padding), ty + y * (self.cell_size + self.cell_padding), self.cell_size, self.cell_size, color)
+      end
+    end
+  end
+  if self.live_piece then
+    self:draw_tetronimo(tx + self.live_piece.x * (self.cell_size + self.cell_padding), ty + self.live_piece.y * (self.cell_size + self.cell_padding), self.live_piece.tetronimo, self.live_piece.rot)
+    if self.drop_shadow then
+      local y = self:get_max_drop(math.huge)
+      if y ~= self.live_piece.y then
+        self:draw_tetronimo(tx + self.live_piece.x * (self.cell_size + self.cell_padding), ty + y * (self.cell_size + self.cell_padding), self.live_piece.tetronimo, self.live_piece.rot, { self.live_piece.tetronimo.color[1], self.live_piece.tetronimo.color[2], self.live_piece.tetronimo.color[3], 50 })
+      end
+    end
+  end
+  if self.finished or self.paused then renderer.draw_rect(tx, ty, self.grid.x * (self.cell_size + self.cell_padding), self.grid.y * (self.cell_size + self.cell_padding), { common.color "rgba(255, 255, 255, 0.5)" }) end
+  if self.finished then common.draw_text(style.font, style.error, "GAME OVER", "center", tx, ty, self.grid.x * (self.cell_size + self.cell_padding), self.grid.y * (self.cell_size + self.cell_padding)) end
+  if self.paused then common.draw_text(style.font, style.warn, "PAUSED", "center", tx, ty, self.grid.x * (self.cell_size + self.cell_padding), self.grid.y * (self.cell_size + self.cell_padding)) end
+end
+
+function TetrisView:rotate()
+  if self.live_piece and not self.paused then
+    local new_rot = (self.live_piece.rot % #self.live_piece.tetronimo.shape) + 1
+    if not self:does_collide(self.live_piece.x, self.live_piece.y, self.live_piece.tetronimo, new_rot) then
+      self.live_piece.rot = new_rot
+    end
+  end
+end
+
+function TetrisView:hold()
+  if self.live_piece and not self.paused then
+    if self.held_piece then
+      if not self:does_collide(self.live_piece.x, self.live_piece.y, self.tetronimos[self.held_piece], 1) then
+        local live_piece = self.live_piece.idx
+        self.live_piece = { x = self.live_piece.x, y = self.live_piece.y, rot = 1, idx = self.held_piece, tetronimo = self.tetronimos[self.held_piece] }
+        self.held_piece = live_piece
+      end
+    else
+      self.held_piece = self.live_piece.idx
+      self.live_piece = nil
+    end
+  end
+end
+
+function TetrisView:get_max_drop(amount)
+  if self.live_piece then
+    for y = self.live_piece.y, math.min(self.grid.y, self.live_piece.y + amount) do
+      if self:does_collide(self.live_piece.x, y + 1, self.live_piece.tetronimo, self.live_piece.rot) then
+        return y, true
+      end
+    end
+  end
+  return self.live_piece.y + amount, false
+end
+
+function TetrisView:drop(amount)
+  if self.live_piece and not self.paused then
+    local y, collides = self:get_max_drop(amount)
+    self.live_piece.y = y
+    if collides then
+      self:finalize_live_piece()
+    end
+  end
+end
+
+function TetrisView:shift(delta)
+  if self.live_piece and not self.paused and not self:does_collide(self.live_piece.x + delta, self.live_piece.y, self.live_piece.tetronimo, self.live_piece.rot) then
+    self.live_piece.x = self.live_piece.x + delta
+  end
+end
+
+command.add(TetrisView, {
+  ["tetris:rotate"] = function() core.active_view:rotate() end,
+  ["tetris:shift-left"] = function() core.active_view:shift(-1) end,
+  ["tetris:shift-right"] = function() core.active_view:shift(1) end,
+  ["tetris:drop"] = function() core.active_view:drop(config.plugins.tetris.down_amount) end,
+  ["tetris:hard-drop"] = function() core.active_view:drop(math.huge) end,
+  ["tetris:hold"] = function() core.active_view:hold() end,
+  ["tetris:toggle-pause"] = function() core.active_view.paused = not core.active_view.paused end,
+  ["tetris:quit"] = function()
+    core.active_view.finished = true
+    core.active_view.node:close_view(core.root_view.root_node, core.active_view)
+  end
+})
+command.add(nil, {
+  ["tetris:start"] = function()
+    local view = TetrisView(config.plugins.tetris)
+    local node = core.root_view:get_active_node()
+    view.node = node:split("right", view, { x = true }, false)
+    core.set_active_view(view)
+  end
+})
+
+keymap.add {
+  ["up"] = "tetris:rotate",
+  ["left"] = "tetris:shift-left",
+  ["right"] = "tetris:shift-right",
+  ["down"] = "tetris:drop",
+  ["space"] = "tetris:hard-drop",
+  ["tab"] = "tetris:hold",
+  ["escape"] = "tetris:quit",
+  ["ctrl+e"] = { "tetris:quit", "tetris:start" },
+  ["p"] = "tetris:toggle-pause"
+}
+return { view = TetrisView }