From 05a9be65f8e0d97fd36711acfe5d626e316722ae Mon Sep 17 00:00:00 2001 From: sinanmohd Date: Fri, 29 Sep 2023 18:02:59 +0530 Subject: repo: init --- server/subscene.lua | 205 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 205 insertions(+) create mode 100644 server/subscene.lua (limited to 'server') diff --git a/server/subscene.lua b/server/subscene.lua new file mode 100644 index 0000000..a51b0d6 --- /dev/null +++ b/server/subscene.lua @@ -0,0 +1,205 @@ +#!/usr/bin/env lua + +local curl = require 'lib/curl' +local util = require 'lib/util' + +-- [[ languages supported by subscene ]] -- +local languages = { + ['arabic'] = 2, + ['bengali'] = 54, + ['brazillian-portuguese'] = 4, + ['chinese-bg-code'] = 7, + ['czech'] = 9, + ['danish'] = 10, + ['dutch'] = 11, + ['english'] = 13, + ['farsipersian'] = 46, + ['finnish'] = 17, + ['french'] = 18, + ['german'] = 19, + ['greek'] = 21, + ['hebrew'] = 22, + ['indonesian'] = 44, + ['italian'] = 26, + ['japanese'] = 27, + ['korean'] = 28, + ['malay'] = 50, + ['norwegian'] = 30, + ['polish'] = 31, + ['portuguese'] = 32, + ['romanian'] = 33, + ['spanish'] = 38, + ['swedish'] = 39, + ['thai'] = 40, + ['turkish'] = 41, + ['vietnamese'] = 45, + ['albanian'] = 1, + ['armenian'] = 73, + ['azerbaijani'] = 55, + ['basque'] = 74, + ['belarusian'] = 68, + ['big-5-code'] = 3, + ['bosnian'] = 60, + ['bulgarian'] = 5, + ['bulgarian_english'] = 6, + ['burmese'] = 61, + ['cambodian_khmer'] = 79, + ['catalan'] = 49, + ['croatian'] = 8, + ['dutch_english'] = 12, + ['english_german'] = 15, + ['esperanto'] = 47, + ['estonian'] = 16, + ['georgian'] = 62, + ['greenlandic'] = 57, + ['hindi'] = 51, + ['hungarian'] = 23, + ['hungarian_english'] = 24, + ['icelandic'] = 25, + ['kannada'] = 78, + ['kinyarwanda'] = 81, + ['kurdish'] = 52, + ['latvian'] = 29, + ['lithuanian'] = 43, + ['macedonian'] = 48, + ['malayalam'] = 64, + ['manipuri'] = 65, + ['mongolian'] = 72, + ['nepali'] = 80, + ['pashto'] = 67, + ['punjabi'] = 66, + ['russian'] = 34, + ['serbian'] = 35, + ['sinhala'] = 58, + ['slovak'] = 36, + ['slovenian'] = 37, + ['somali'] = 70, + ['sundanese'] = 76, + ['swahili'] = 75, + ['tagalog'] = 53, + ['tamil'] = 59, + ['telugu'] = 63, + ['ukrainian'] = 56, + ['urdu'] = 42, + ['yoruba'] = 71, +} + +local language = 'english' +local domain = 'https://subscene.com' +local headr = {['Cookie'] = 'LanguageFilter=' ..languages[language]} +local retries = 10 + +local title_search = function (key) + local url, args, fetch, hcode, tries, title + + url = domain .. '/subtitles/searchbytitle' + args = "--data-raw query='" .. key .. "' " + + tries = 0 + repeat + fetch, hcode = curl.get(url, headr, args) + tries = tries + 1 + until hcode == 200 or tries > retries + + title = fetch:match('href="/subtitles/[^"]*') + if title then + title = title:gsub('href="', domain) + end + + return title, (hcode == 200 and title ~= nil) +end + +local id_fetch = function (title) + local tab, id, name, fetch, hcode, tries, iter, line + + tries = 0 + repeat + fetch, hcode = curl.get(title, headr, nil) + tries = tries + 1 + until hcode == 200 or tries > retries + + tab = {} + iter = fetch:gmatch('[^\n\r]+') + while true do + line = iter() + if not line then + break + end + + id = line:match('/subtitles/[^/]*/[^/]*/%d*') + if id then + iter() + iter() + iter() + iter() + + name = iter():match('%S[^\n\r]*%S') + tab[name] = domain .. id + id = nil + end + end + + return tab +end + +local link_fetch = function (id) + local fetch, tries, hcode, link + + tries = 0 + repeat + fetch, hcode = curl.get(id, headr, nil) + tries = tries + 1 + until hcode == 200 or tries > retries + + link = domain .. fetch:match('/subtitles/' .. language .. '%-text/[^"]*') + return link, (hcode == 200) +end + +local sub_fetch = function(link, out) + local tries, hcode, zip, zcode + + tries = 0 + zip = os.tmpname() + + repeat + _, hcode = curl.get(link, headr, '-o ' .. zip) + tries = tries + 1 + until hcode == 200 or tries > retries + + zcode = util.zip_ext_first(zip, out) + os.remove(zip) + + return (hcode == 200) and zcode +end + +local search = function (key, out) + local title, id, link, rc + + key = util.string_rm_vid_ext(key) + title, rc = title_search(key) + if not rc then + return nil + end + + id = id_fetch(title) + id = util.table_match_or_any(id, key) + if not id then + return nil + end + + link, rc = link_fetch(id) + if not rc then + return nil + end + + rc = sub_fetch(link, out) + return rc +end + +return { + title_search = title_search, + id_fetch = id_fetch, + link_fetch = link_fetch, + sub_fetch = sub_fetch, + search = search, +} -- cgit v1.2.3