path: root/server/opensubtitles.lua

#!/usr/bin/env lua

local attr = require("lib.attr")
local curl = require("lib.curl")
local util = require("lib.util")

local domain = "https://www.opensubtitles.org"

local ids_fetch = function(page)
	local iter, no_name, line, id, name, tab

	tab = {}
	no_name = 0
	iter = page:gmatch("[^\n\r]+")
	while true do
		line = iter()
		if not line then
			break
		end

		id = line:match("/en/subtitles/%d*")
		if id then
			id = id:match("%d+$")

			line = iter() -- movie
			if line:find("%.%.%.$") then
				-- name cuts off...
				name = line:gsub('"[^"]*$', "")
				name = name:match('[^"]+$')
			else
				name = line:gsub("<br/><a rel.*$", "")
				name = name:match("[^>]+$")
			end

			if not name then
				line = iter()

				if line:find("^%[S%d%dE%d%d%]$") then
					-- it's a series
					line = iter()
					if line:find("%.%.%.$") then
						name = line:gsub('^.*title="', "")
						name = name:match('[^"]+')
					else
						name = line:match("[^<]+")
					end
				else
					-- no name
					name = tostring(no_name)
					no_name = no_name + 1
				end
			end

			tab[name] = id
		end
	end

	return tab
end

local search_ohash = function(ohash, name, lang)
	local fetch, hcode, url, id

	url = domain
		.. "/en"
		.. "/search/sublanguageid-"
		.. lang
		.. "/moviehash-"
		.. ohash
	fetch, hcode = curl.get(url, nil, nil)

	id = attr.fuzzy(name, ids_fetch(fetch))
	if hcode and not id then
		util.error("opensubtitles: search_ohash failed")
	end

	if id then
		return domain .. "/en/subtitleserve/sub/" .. id
	end
end

local search_filesize = function(filesize, name, lang)
	local fetch, hcode, url, id, a

	a = attr.build(name)

	url = domain .. "/en" .. "/search/sublanguageid-" .. lang
	if a.season and a.episode then
		url = url .. "/season-" .. a.season .. "/episode-" .. a.episode
	end
	url = url .. "/moviebytesize-" .. filesize

	fetch, hcode = curl.get(url, nil, nil)
	if not hcode then
		return nil
	end

	id = attr.fuzzy(name, ids_fetch(fetch))
	if hcode and not id then
		util.error("opensubtitles: search_filesize failed")
	end

	if id then
		return domain .. "/en/subtitleserve/sub/" .. id
	end
end

local url_encode = function(str)
	local url = ""

	for c in str:gmatch(".") do
		if c:find("%w") then
			url = url .. c
		else
			url = url .. "%" .. string.format("%x", c:byte()):upper()
		end
	end

	return url
end

local suggest_id = function(name)
	local url, fetch, hcode, id

	url = domain
		.. "/libs/suggest.php?format=json3&MovieName="
		.. url_encode(name)

	fetch, hcode = curl.get(url, nil, nil)
	if not hcode then
		return nil
	end

	id = fetch:match('"id":%d+')
	if not id then
		return nil
	end

	return id:match("%d+")
end

local search_name = function(name, lang)
	local fetch, hcode, movie_id, url, id

	movie_id = suggest_id(name)
	if not movie_id then
		return nil
	end

	url = domain
		.. "/en/search/sublanguageid-"
		.. lang
		.. "/idmovie-"
		.. movie_id
	fetch, hcode = curl.get(url, nil, nil)
	if not hcode then
		return nil
	end

	id = attr.fuzzy(name, ids_fetch(fetch))
	if not id then
		util.error("opensubtitles: search_name failed")
	end

	if id then
		return domain .. "/en/subtitleserve/sub/" .. id
	end
end

local search = function(path, out, info)
	local ohash, link, name, lang

	lang = info.iso639_2_lang or "eng"
	name = info.name or util.string_vid_path_to_name(path)

	if util.file_exists(path) then
		ohash = util.opensubtitles_hash(path)
		link = search_ohash(ohash, name, lang)
	end

	if not link then
		link = search_filesize(info.filesize, name, lang)
	end

	if not link then
		link = search_name(name, lang)
	end

	if link then
		return curl.zip_link_to_file(link, nil, out)
	end
end

return {
	search_ohash = search_ohash,
	search_filesize = search_filesize,
	search_name = search_name,
	search = search,
}