eat-it/eatit.lua

--[[
Eat It - a Mpv plugin manager

Copyright © 2023 squibid

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
]]

local mp = require('mp')

-- load the config file
dofile(mp.command_native({'expand-path', '~~/eatit-cfg.lua'}))

-- helper functions --
local function tablelength(T)
  local count = 0
  for _ in pairs(T) do count = count + 1 end
  return count
end

local function fileexists(name)
   local f = io.open(name, 'r')
   if f ~= nil then io.close(f) return true else return false end
end

local function logwrite(string)
  if opts.logging.log then
    io.write(os.date(opts.logging.logdate) .. ' ' .. string .. '\n')
  end
end

local function openlog()
  if opts.logging.log then -- log if asked to
    -- get our logfile's full path
    fn = mp.command_native({'expand-path', opts.logging.logfile})

    f = io.open(fn, 'a') -- open file buffer
    io.output(f) -- set it as default
  end
end

local function closelog()
  if opts.logging.log then
    io.close(f)
  end
end

-- get the requested git repos
local function clonegit(i)
    logwrite('downloading ' .. plugins[i][1])

    -- construct the git cmd
    local cmd = 'git -C ' .. opts.dl.dir .. ' clone ' .. plugins[i][1]

    -- run the command and get the result for logging
    local run = io.popen(cmd)
    -- BUG: logwriting the git command doens't actually log the output
    -- (commented out for now)
    -- logwrite(run:read('*a'))
    run:close()
end

-- check for updates
local function checkupdates(i)
  local plugdir = opts.dl.dir .. '/' ..
    string.match(plugins[i][1], '/([^/]+)$'):gsub('.git', '')

  local localhashcmd = 'git -C ' .. plugdir .. ' log -1 --format=format:"%H"'

  local remotehashcmd = 'git -C ' .. plugdir ..
    ' log -n 1 $(git -C ' .. plugdir ..
    ' branch -r | head -1 | cut -d " " -f 3)' ..
    ' | head -1 | cut -d " " -f 2'

  local a = io.popen(localhashcmd)
  local localhash = string.gsub(a:read("*a"), '[\n\r]', '')
  a:close()

  a = io.popen(remotehashcmd)
  local remotehash = string.gsub(a:read("*a"), '[\n\r]', '')
  a:close()

  if localhash ~= remotehash then
    return true
  else
    return false
  end
end

-- start install
local function startinstall()
  logwrite('Starting Download...')
  for i = 1, tablelength(plugins) do
    -- check if the user has defined a file for the current plugin
    if not plugins[i]['file'] then
      logwrite('WARNING! Git repo "' .. plugins[i][1] ..
        '" doesn\'t have a specified file. Skipping download')
      -- FIXME: this results in the last plugin being downloaded & copied twice
      i = i + 1 -- if not we go to the next plugin
    end

    -- if no destdir defined we set it to the root of the mpv config dir
    if not plugins[i]['dir'] then
      plugins[i]['dir'] = '~~/'
    end

    -- get the file's dir
    local pluginfile = opts.dl.dir .. '/' ..
      string.match(plugins[i][1], '/([^/]+)$'):gsub('.git', '') ..
      '/' .. plugins[i]['file']

    -- get the dest dir
    local destfile = mp.command_native({'expand-path', '~~/'}) ..
      '/' .. plugins[i]['dir'] ..
      '/' .. plugins[i]['file']

    if fileexists(pluginfile) then
      -- if we need to update, update
      if checkupdates(i) then
        logwrite(plugins[i]['file'] .. ' is updating.')
        clonegit(i)
      else
        logwrite(plugins[i]['file'] .. ' is up to date!')
      end
    else
      clonegit(i)
    end

    -- copy the file contents over to the desired location
    local infile = io.open(pluginfile, 'r')
    local outfile = io.open(destfile, 'w')
    outfile:write(infile:read('*a'))
    outfile:close()
    infile:close()
  end
end

local function initupdate()
  openlog()

  logwrite('# of plugins defined in table: ' .. tablelength(plugins))
  os.execute('mkdir -p ' .. opts.dl.dir) -- make download dir
  startinstall()

  -- closing/removing everything
  if opts.dl.powerwash == true then
    logwrite('powerwashing the tmp dir (' .. opts.dl.dir .. ')')
    os.execute('rm -rf ' .. opts.dl.dir)
  end

  closelog()
end

-- run any user defined configs at startup
local function runopts()
  openlog()
  for i = 1, tablelength(plugins) do
    if plugins[i]['cfg'] then
      logwrite('Running ' .. plugins[i]['file'] .. '\'s config')
      plugins[i]['cfg']()
    end
  end
  closelog()
end

-- remove logfile on startup
openlog()
os.remove(fn)
closelog()

runopts()
mp.add_key_binding('U', 'UpdatePlugins', initupdate)