diff options
Diffstat (limited to 'server/opensubtitles.lua')
| -rw-r--r-- | server/opensubtitles.lua | 95 |
1 files changed, 51 insertions, 44 deletions
diff --git a/server/opensubtitles.lua b/server/opensubtitles.lua index 0d7af7b..5f73d7f 100644 --- a/server/opensubtitles.lua +++ b/server/opensubtitles.lua @@ -1,48 +1,48 @@ #!/usr/bin/env lua -local curl = require 'lib.curl' -local util = require 'lib.util' -local attr = require 'lib.attr' +local attr = require("lib.attr") +local curl = require("lib.curl") +local util = require("lib.util") -local domain = 'https://www.opensubtitles.org' +local domain = "https://www.opensubtitles.org" -local ids_fetch = function (page) +local ids_fetch = function(page) local iter, no_name, line, id, name, tab tab = {} no_name = 0 - iter = page:gmatch('[^\n\r]+') + iter = page:gmatch("[^\n\r]+") while true do line = iter() if not line then break end - id = line:match('/en/subtitles/%d*') + id = line:match("/en/subtitles/%d*") if id then - id = id:match('%d+$') + id = id:match("%d+$") line = iter() -- movie - if line:find('%.%.%.$') then + if line:find("%.%.%.$") then -- name cuts off... - name = line:gsub('"[^"]*$', '') + name = line:gsub('"[^"]*$', "") name = name:match('[^"]+$') else - name = line:gsub('<br/><a rel.*$', '') - name = name:match('[^>]+$') + 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 + if line:find("^%[S%d%dE%d%d%]$") then -- it's a series line = iter() - if line:find('%.%.%.$') then - name = line:gsub('^.*title="', '') + if line:find("%.%.%.$") then + name = line:gsub('^.*title="', "") name = name:match('[^"]+') else - name = line:match('[^<]+') + name = line:match("[^<]+") end else -- no name @@ -58,33 +58,37 @@ local ids_fetch = function (page) return tab end -local search_ohash = function (ohash, name, lang) +local search_ohash = function(ohash, name, lang) local fetch, hcode, url, id - url = domain .. '/en' .. '/search/sublanguageid-' .. lang .. - '/moviehash-' .. ohash + 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') + util.error("opensubtitles: search_ohash failed") end if id then - return domain .. '/en/subtitleserve/sub/' .. id + return domain .. "/en/subtitleserve/sub/" .. id end end -local search_filesize = function (filesize, name, lang) +local search_filesize = function(filesize, name, lang) local fetch, hcode, url, id, a a = attr.build(name) - url = domain .. '/en' .. '/search/sublanguageid-' .. lang + url = domain .. "/en" .. "/search/sublanguageid-" .. lang if a.season and a.episode then - url = url .. '/season-' .. a.season .. '/episode-' .. a.episode + url = url .. "/season-" .. a.season .. "/episode-" .. a.episode end - url = url .. '/moviebytesize-' .. filesize + url = url .. "/moviebytesize-" .. filesize fetch, hcode = curl.get(url, nil, nil) if not hcode then @@ -93,34 +97,34 @@ local search_filesize = function (filesize, name, lang) id = attr.fuzzy(name, ids_fetch(fetch)) if hcode and not id then - util.error('opensubtitles: search_filesize failed') + util.error("opensubtitles: search_filesize failed") end if id then - return domain .. '/en/subtitleserve/sub/' .. id + return domain .. "/en/subtitleserve/sub/" .. id end end -local url_encode = function (str) +local url_encode = function(str) local url = "" - for c in str:gmatch('.') do - if c:find('%w') then + for c in str:gmatch(".") do + if c:find("%w") then url = url .. c else - url = url .. '%' .. - string.format('%x', c:byte()):upper() + url = url .. "%" .. string.format("%x", c:byte()):upper() end end return url end -local suggest_id = function (name) +local suggest_id = function(name) local url, fetch, hcode, id - url = domain .. '/libs/suggest.php?format=json3&MovieName=' .. - url_encode(name) + url = domain + .. "/libs/suggest.php?format=json3&MovieName=" + .. url_encode(name) fetch, hcode = curl.get(url, nil, nil) if not hcode then @@ -132,10 +136,10 @@ local suggest_id = function (name) return nil end - return id:match('%d+') + return id:match("%d+") end -local search_name = function (name, lang) +local search_name = function(name, lang) local fetch, hcode, movie_id, url, id movie_id = suggest_id(name) @@ -143,8 +147,11 @@ local search_name = function (name, lang) return nil end - url = domain .. '/en/search/sublanguageid-' .. lang .. - '/idmovie-' .. movie_id + url = domain + .. "/en/search/sublanguageid-" + .. lang + .. "/idmovie-" + .. movie_id fetch, hcode = curl.get(url, nil, nil) if not hcode then return nil @@ -152,18 +159,18 @@ local search_name = function (name, lang) id = attr.fuzzy(name, ids_fetch(fetch)) if not id then - util.error('opensubtitles: search_name failed') + util.error("opensubtitles: search_name failed") end if id then - return domain .. '/en/subtitleserve/sub/' .. id + return domain .. "/en/subtitleserve/sub/" .. id end end -local search = function (path, out, info) +local search = function(path, out, info) local ohash, link, name, lang - lang = info.iso639_2_lang or 'eng' + lang = info.iso639_2_lang or "eng" name = info.name or util.string_vid_path_to_name(path) if util.file_exists(path) then @@ -188,5 +195,5 @@ return { search_ohash = search_ohash, search_filesize = search_filesize, search_name = search_name, - search = search + search = search, } |