iv-org/invidious
1
0
Fork 0
mirror of https://github.com/iv-org/invidious.git synced 2024-12-30 01:36:21 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add support for 'region' in search

Browse Source
This commit is contained in:
Omar Roth 2019-02-06 18:21:40 -06:00
parent d625d0ffbd
commit a7b79824de
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/invidious.cr
View File

@ -596,6 +596,7 @@ end
get "/search" do |env|
locale = LOCALES[env.get("locale").as(String)]?
region = env.params.query["region"]?
query = env.params.query["search_query"]?
query ||= env.params.query["q"]?
@ -671,7 +672,7 @@ get "/search" do |env|
next templated "error"
end
count, videos = search(search_query, page, search_params).as(Tuple)
count, videos = search(search_query, page, search_params, proxies, region).as(Tuple)
end
templated "search"
@ -3199,6 +3200,7 @@ end
get "/api/v1/search" do |env|
locale = LOCALES[env.get("locale").as(String)]?
region = env.params.query["region"]?
env.response.content_type = "application/json"
@ -3220,7 +3222,6 @@ get "/api/v1/search" do |env|
features = env.params.query["features"]?.try &.split(",").map { |feature| feature.downcase }
features ||= [] of String
# TODO: Support other content types
content_type = env.params.query["type"]?.try &.downcase
content_type ||= "video"
@ -3235,7 +3236,7 @@ get "/api/v1/search" do |env|
end
end
count, search_results = search(query, page, search_params).as(Tuple)
count, search_results = search(query, page, search_params, proxies, region).as(Tuple)
response = JSON.build do |json|
json.array do
search_results.each do |item|

4
src/invidious/search.cr
View File

@ -85,8 +85,8 @@ def channel_search(query, page, channel)
return count, items
end
def search(query, page = 1, search_params = produce_search_params(content_type: "all"))
client = make_client(YT_URL)
def search(query, page = 1, search_params = produce_search_params(content_type: "all"), proxies = nil, region = nil)
client = make_client(YT_URL, proxies, region)
if query.empty?
return {0, [] of SearchItem}
end