extractors: Add support for richGridRenderer

This commit is contained in:
Samantaz Fox 2022-10-31 20:40:43 +01:00
parent 84cd4d6a5b
commit f267394bbe
No known key found for this signature in database
GPG Key ID: F42821059186176E

View File

@ -436,7 +436,18 @@ private module Extractors
content = extract_selected_tab(target["tabs"])["content"] content = extract_selected_tab(target["tabs"])["content"]
if section_list_contents = content.dig?("sectionListRenderer", "contents") if section_list_contents = content.dig?("sectionListRenderer", "contents")
section_list_contents.as_a.each do |renderer_container| raw_items = unpack_section_list(section_list_contents)
elsif rich_grid_contents = content.dig?("richGridRenderer", "contents")
raw_items = rich_grid_contents.as_a
end
return raw_items
end
private def self.unpack_section_list(contents)
raw_items = [] of JSON::Any
contents.as_a.each do |renderer_container|
renderer_container_contents = renderer_container["itemSectionRenderer"]["contents"][0] renderer_container_contents = renderer_container["itemSectionRenderer"]["contents"][0]
# Category extraction # Category extraction
@ -452,7 +463,6 @@ private module Extractors
raw_items << item raw_items << item
end end
end end
end
return raw_items return raw_items
end end
@ -525,14 +535,11 @@ private module Extractors
end end
private def self.extract(target) private def self.extract(target)
raw_items = [] of JSON::Any content = target["continuationItems"]?
if content = target["gridContinuation"]? content ||= target.dig?("gridContinuation", "items")
raw_items = content["items"].as_a content ||= target.dig?("richGridContinuation", "contents")
elsif content = target["continuationItems"]?
raw_items = content.as_a
end
return raw_items return content.nil? ? [] of JSON::Any : content.as_a
end end
def self.extractor_name def self.extractor_name