FindingMusic/search.lua

#!/usr/bin/env luajit

local function checkreq(name, display)
  local success, library = pcall(function() return require(name) end)
  if not success then
    print("'" .. display or name .. "' missing.")
    return nil
  else
    return library
  end
end

local urlencode = checkreq('urlencode')
checkreq('cjson', 'lua-cjson')
local music = checkreq('music', 'music.lua')

local help = [[Usage:

  ./search.lua <count> <funkwhale>

<count>: How many tracks to search for. Defaults to 5.
         (Opens a new tab per track in your default browser searching Google.)
<funkwhale>: true: Search my FunkWhale instance for the track.

Optionally requires urlencode to be installed from LuaRocks.
Currently only tested on MacOS 11.4 to 12.1.
]]
if arg[1]:find("h") then
  print(help)
  return false
end

local count = tonumber(arg[1]) or 5
local search_funkwhale = arg[2] == "true"

local results = music.random(count, nil, nil, {downloaded = true, searched = true})
local errors_occurred, name, encoded = false
for _,v in ipairs(results) do
  name = music.name(v) -- music.data[v].names[1]
  if not name then
    print("Track '" .. v .. "' does not exist?")
    errors_occurred = true
    break
  end
  if urlencode then
    encoded = urlencode.encode_url(name)
  else
    encoded = name:gsub("%s", "+"):gsub("&", "&amp;")
  end
  if search_funkwhale then
    local separator = name:find(" %- ")
    if separator then name = name:sub(separator + 3) end
    if urlencode then
      name = urlencode.encode_url(name)
    else
      name = name:gsub("%s", "+"):gsub("&", "&amp;")
    end
    os.execute("open \"https://funkwhale.tangentfox.com/search?q=" .. name .. "&type=tracks\"")
  end
  os.execute("open \"https://google.com/search?q=" .. encoded .. "\"")
end
if errors_occurred then
  print("Database not saved because errors occurred.")
else
  music.set(results, {searched = true})
  music.save()
end