snged

-- snged v1.0
--  by chiloxsan

function prompt(promptString)
  local width, height = term.getSize()
  term.setCursorPos(1, height)
  term.setBackgroundColor(gui_colours.highlight)
  term.setTextColor(gui_colours.highlight_text)
  term.clearLine()
  write(promptString)
  return read()
end

statusbar = {
  ["new"] = function(songAreaObj)
    return {
      ["songAreaObj"] = songAreaObj,

      ["draw"] = function(self)
        local width, height = term.getSize()
        term.setCursorPos(width - 10, height)
        term.setBackgroundColour(gui_colours.highlight)
        term.setTextColour(gui_colours.highlight_text)
        term.clearLine()

        if self.songAreaObj.playing then
          write("|>|")
        else
          write("|||")
        end

        if songAreaObj.highNotes then
          write("OCT2|")
        else
          write("OCT1|")
        end
        if songAreaObj.recording then
          term.setTextColour(colours.red)
          write("REC")
        else
          term.setTextColour(colours.lime)
          write("NAV")
        end
        term.setCursorPos(1, height)
        term.setTextColour(gui_colours.highlight_text)
        write("Pos: " .. songAreaObj.currentPos .. " | Length: " .. (songAreaObj.songObj.length + 1) .. " | Ins: " .. songAreaObj.recordIns .. " | Skip: " .. songAreaObj.skip)
      end;

      ["tick"] = function(self, e, ...)
        if e == "saved" then
          local width, height = term.getSize()
          term.setCursorPos(1, height)
          term.setBackgroundColour(gui_colours.highlight)
          term.setTextColour(gui_colours.highlight_text)
          term.clearLine()
          write("Saved.")
          sleep(1)
          os.queueEvent("update")
        end
      end
    }
  end
}

songArea = {
  ["new"] = function(songObj)
    return {
      ["currentPos"] = 0,
      ["songObj"]    = songObj,
      ["playing"]    = false,
      ["recording"]  = false,
      ["recordIns"]  = 0,
      ["recordCol"]  = 1,
      ["highNotes"]  = false,
      ["skip"]       = 1,

      ["draw"] = function(self)
        local width, height = term.getSize()
        local areaHeight = height - 4
        local rowPos = (self.currentPos - math.floor(areaHeight / 2))
        term.setCursorPos(1, 3)
        for y = 1, areaHeight do
          term.setBackgroundColor(gui_colours.background)
          if rowPos == self.currentPos then
            term.setCursorPos(1, 3 + y)
            for i = 1, width do
              if self.recording then
                term.setBackgroundColor(gui_colours.track_record)
              else
                term.setBackgroundColor(gui_colours.track_highlight)
              end
              write(" ")
            end
          end
          term.setCursorPos(1, 3 + y)
          if self.songObj.songData[rowPos] ~= nil then
            for i = 1, self.songObj.cols + 1 do
              term.setTextColor(gui_colours.ins)
              if self.songObj.songData[rowPos][(i * 2) - 1] ~= nil then
                term.setTextColor(gui_colours[self.songObj.songData[rowPos][(i * 2) - 1]])
              end
              if self.songObj.songData[rowPos][(i * 2) - 1] ~= nil then
                write(self.songObj.songData[rowPos][(i * 2) - 1])
              else
                write(".")
              end
              term.setTextColor(gui_colours.note)
              if self.songObj.songData[rowPos][i * 2] ~= nil then
                if #tostring(self.songObj.songData[rowPos][i * 2]) == 1 then
                  write("0" .. self.songObj.songData[rowPos][i * 2])
                else
                  write(self.songObj.songData[rowPos][i * 2])
                end
              else
                write("..")
              end
              term.setTextColor(gui_colours.track_border)
              if self.recording and self.recordCol == i and self.currentPos == rowPos then
                write("<")
              else
                write("|")
              end
            end
          end
          rowPos = rowPos + 1
        end
      end;

      ["tick"] = function(self, e, ...)
        local eventData = { ... }
        if e == "mouse_scroll" or e == "key" then
          if eventData[1] == -1 or eventData[1] == 200 then
            self.currentPos = self.currentPos - 1
          elseif eventData[1] == 1 or eventData[1] == 208 then
            self.currentPos = self.currentPos + 1
          elseif eventData[1] == 199 then
            self.currentPos = 0
          elseif eventData[1] == 207 then
            self.currentPos = self.songObj.length
          elseif eventData[1] == 57 then
            self.recording = not self.recording
          elseif eventData[1] == 203 and self.recording then
            if self.recordCol > 1 then self.recordCol = self.recordCol - 1 end
          elseif eventData[1] == 205 and self.recording then
            if self.recordCol < self.songObj.cols + 1 then
              self.recordCol = self.recordCol + 1
            end
          elseif (eventData[1] >= 16 and eventData[1] <= 27) or (e == "key" and eventData[1] == 0) then
            local note = 12
            if eventData[1] ~= 0 then
              note = eventData[1] - 16
            end

            if self.highNotes then
              note = note + 12
            end

            if vNoteblockObj ~= nil then
              vNoteblockObj.playNote(self.recordIns, note)
            else
              noteblocks[1].playNote(self.recordIns, note)
            end

            if self.recording and self.songObj.songData[self.currentPos] ~= nil then
              if self.songObj.songData[self.currentPos][(self.recordCol * 2) - 1] == nil then
                self.songObj.songData[self.currentPos][#self.songObj.songData[self.currentPos] + 1] = self.recordIns
                self.songObj.songData[self.currentPos][#self.songObj.songData[self.currentPos] + 1] = note
              else
                self.songObj.songData[self.currentPos][(self.recordCol * 2) - 1] = self.recordIns
                self.songObj.songData[self.currentPos][self.recordCol * 2] = note
              end
              self.songObj:rescan()
              self.currentPos = self.currentPos + self.skip
            end
          elseif eventData[1] == 15 then
            self.highNotes = (not self.highNotes)
          elseif eventData[1] >= 2 and eventData[1] <= 6 then
            self.recordIns = eventData[1] - 2
          elseif eventData[1] == 211 and self.recording then
            if self.songObj.songData[self.currentPos][self.recordCol * 2] ~= nil then
              table.remove(self.songObj.songData[self.currentPos], (self.recordCol * 2) - 1)
              table.remove(self.songObj.songData[self.currentPos], (self.recordCol * 2) - 1)
            end
            self.currentPos = self.currentPos + self.skip
          elseif eventData[1] == 12 and self.skip > 0 then
            self.skip = self.skip - 1
          elseif eventData[1] == 13 then
            self.skip = self.skip + 1
          elseif eventData[1] == 28 then
            self.playing = (not self.playing)
          end

          if self.currentPos > self.songObj.length then self.currentPos = 0 end
          if self.currentPos < 0 then self.currentPos = 0 end
          if self.recordCol < 1 then self.recordCol = 1 end
        end
      end
    }
  end
}

song = {
  ["new"] = function()
    return {
      ["title"] = "",
      ["length"] = 0,
      ["author"] = "",
      ["originalAuthour"] = "",
      ["description"] = "",
      ["interval"] = 0.1,
      ["songData"] = {},
      ["cols"] = 0,

      ["rescan"] = function(self)
        for i = 0, #self.songData do
          if math.floor(#self.songData[i] / 2) > self.cols then self.cols = math.floor(#self.songData[i] / 2) end
        end
      end;

      ["loadFromString"] = function(self, str)
        local data = textutils.unserialize(str)
        if type(data) ~= "table" then error("Invalid song file/string.") end
        self.title = data["title"]
        self.length = data["length"]
        self.author = data["author"]
        self.originalAuthor = data["originalAuthor"]
        self.description = data["description"]
        self.interval = data["interval"]
        self.songData = data["song"]

        self:rescan()
      end;

      ["loadFromFile"] = function(self, filename)
        local f = fs.open(filename, "r")
        self:loadFromString(f:readAll("*a"))
        f.close()
      end;

      ["writeToString"] = function(self)
        t = {
          ["title"] = self.title,
          ["length"] = self.length,
          ["author"] = self.author,
          ["originalAuthour"] = self.originalAuthor,
          ["description"] = self.description,
          ["interval"] = self.interval,
          ["song"] = self.songData
        }
        return textutils.serialize(t)
      end;

      ["writeToFile"] = function(self, filename)
        local f = fs.open(filename, "w")
        f.write(self:writeToString())
        f.close()
      end
    }
  end
}

button = {
  ["new"] = function(text, x, y, callback)
    return {
      ["text"] = text,
      ["x"]    = x,
      ["y"]    = y,
      ["callback"] = callback,

      ["draw"] = function(self)
        term.setCursorPos(self.x, self.y)
        term.setTextColour(gui_colours.button_text)
        term.setBackgroundColour(gui_colours.button)
        write(" " .. self.text .. " ")
      end;

      ["tick"] = function(self, e, ...)
        local eventData = { ... }
        local cursorX = eventData[2]
        local cursorY = eventData[3]
        if e == "mouse_click" and cursorY == self.y and cursorX >= self.x and cursorX <= self.x + #text + 1 then
          self.callback()
        end
      end
    }
  end
}

toolbar = {
  ["new"] = function(songAreaObj)
    return {
      songAreaObj = songAreaObj,

      ["buttons"] = {
        button.new("PlayPause", 12, 1, function() songAreaObj.playing = (not songAreaObj.playing) end),
        button.new("SetNoteObj", 24, 1, function()
          local obj = prompt("vNoteblockObj=")
          if _G[obj] ~= nil and _G[obj].playNote ~= nil then
            vNoteblockObj = _G[obj]
          else
            vNoteblockObj = nil
          end
        end),
        button.new("Top", 37, 1, function() songAreaObj.currentPos = 0 end),
        button.new("Exit", 43, 1, function() os.queueEvent("exit") end),
        button.new("EditInterval", 1, 2, function()
          songAreaObj.songObj.interval = tonumber(prompt("interval="))
          if songAreaObj.songObj.interval == nil or songAreaObj.songObj.interval < 0.05 then
            songAreaObj.songObj.interval = 0.05
          end
        end),
        button.new("EditAuthorData", 16, 2, function()
          songAreaObj.songObj.title = prompt("song.title=")
          songAreaObj.songObj.description = prompt("song.description=")
          songAreaObj.songObj.author = prompt("song.author=")
          songAreaObj.songObj.originalAuthor = prompt("song.originalAuthor=")
        end),
        button.new("EditLength", 33, 2, function()
          local length = tonumber(prompt("length="))
          if length ~= nil and length > 0 then
            songAreaObj.songObj.length = length - 1
          end
          for i = 0, length - 1 do
            if songAreaObj.songObj.songData[i] == nil then songAreaObj.songObj.songData[i] = {} end
          end
          songAreaObj.songObj:rescan()
        end),
        button.new("Save", 46, 2, function()
          songAreaObj.songObj:writeToFile(filename)
          os.queueEvent("saved")
        end)
      },

      ["draw"] = function(self)
        term.setCursorPos(1, 1)
        term.setBackgroundColour(gui_colours.highlight)
        term.setTextColour(gui_colours.highlight_text)
        local width, height = term.getSize()
        for y = 1, 3 do
          for x = 1, width do
            write(" ")
          end
          print("")
        end
        term.setCursorPos(1, 1)
        write("snged v1.0")
        for k, v in ipairs(self.buttons) do v:draw() end
        term.setCursorPos(1, 3)
        term.setBackgroundColour(gui_colours.highlight)
        term.setTextColour(gui_colours.highlight_text)
        write("Title: " .. self.songAreaObj.songObj.title .. " | Author: " .. self.songAreaObj.songObj.author)
      end;

      ["tick"] = function(self, e, ...)
        for k, v in ipairs(self.buttons) do v:tick(e, ...) end
      end
    }
  end
}

function cleanup()
  gui_colours   = nil
  cleanup       = nil
  button        = nil
  toolbar       = nil
  song          = nil
  songArea      = nil
  statusbar     = nil
  popup         = nil
  vNoteblockObj = nil
  noteblocks    = nil
  prompt        = nil
end

gui_colours = {
  ["highlight"] = colours.blue,
  ["highlight_text"] = colours.white,
  ["button"] = colours.lime,
  ["button_text"] = colours.white,
  ["button_click_text"] = colours.white,
  ["button_click"] = colours.red,
  ["text"] = colours.white,
  ["background"] = colours.black,
  ["note"] = colours.lime,
  ["ins"] = colours.yellow,
  ["track_border"] = colours.white,
  ["track_highlight"] = colours.grey,
  ["track_record"] = colours.red,

  [0] = colours.lightBlue,
  [1] = colours.lightGrey,
  [2] = colours.yellow,
  [3] = colours.white,
  [4] = colours.brown
}

local tArgs = { ... }

local width, height = term.getSize()

popup = nil

local songObj = song.new()

if tArgs[1] ~= nil then
  filename = shell.resolve(tArgs[1])

  if fs.isDir(filename) then
    error("You cannot edit directories.")
  end

  if fs.exists(filename) then
    songObj:loadFromFile(filename)
  end
else
  print("Usage: snged filename")
  return
end

local songAreaObj = songArea.new(songObj)
local toolbarObj = toolbar.new(songAreaObj)
local statusbarObj = statusbar.new(songAreaObj)

noteblocks = {}
vNoteblockObj = nil

for k, v in ipairs(rs.getSides()) do
  local testObj = peripheral.wrap(v)
  if type(testObj) == "table" and testObj.playNote ~= nil then
    table.insert(noteblocks, testObj)
  end
end

toolbarObj:draw()
songAreaObj:draw()
statusbarObj:draw()

parallel.waitForAny(function()
    while true do
      local e, p1, p2, p3 = os.pullEvent()
      if e == "exit" then break end
      toolbarObj:tick(e, p1, p2, p3)
      songAreaObj:tick(e, p1, p2, p3)
      statusbarObj:tick(e, p1, p2, p3)
      term.setBackgroundColor(gui_colours.background)
      term.clear()
      toolbarObj:draw()
      songAreaObj:draw()
      statusbarObj:draw()
    end
  end,

  function()
    while true do
      local slept = false
      local nb = 1
      if songAreaObj.playing and songAreaObj.songObj.songData[songAreaObj.currentPos] ~= nil then
        slept = true
        for i = 1, #songAreaObj.songObj.songData[songAreaObj.currentPos], 2 do
          if vNoteblockObj == nil then
            noteblocks[nb].playNote(songAreaObj.songObj.songData[songAreaObj.currentPos][i], songAreaObj.songObj.songData[songAreaObj.currentPos][i + 1])
          else
            vNoteblockObj.playNote(songAreaObj.songObj.songData[songAreaObj.currentPos][i], songAreaObj.songObj.songData[songAreaObj.currentPos][i + 1])
          end
          nb = nb + 1
          if nb > #noteblocks then
            nb = 1
          end
        end
--        songAreaObj.recordCol = math.floor(#songAreaObj.songObj.songData[songAreaObj.currentPos] / 2) + 1
        if songAreaObj.recordCol < 1 then songAreaObj.recordCol = 1 end
        os.queueEvent("update")
        sleep(songAreaObj.songObj.interval)
        songAreaObj.currentPos = songAreaObj.currentPos + 1
        if songAreaObj.currentPos > songAreaObj.songObj.length then
          songAreaObj.currentPos = 0
        end
      else
        songAreaObj.playing = false
      end
      if not slept then
        sleep(0.05)
      end
    end
  end
)

cleanup()
term.setCursorPos(1, 1)
term.setTextColor(colors.white)
term.setBackgroundColor(colors.black)
term.clear()