Backups/Mango
1
0
Fork 0
mirror of https://github.com/hkalexling/Mango.git synced 2025-08-02 19:05:32 -04:00
Code Issues Projects Releases Wiki Activity

Choose correct subclass based on YAML node

Browse Source
This commit is contained in:
Alex Ling 2022-05-29 05:24:41 +00:00
parent 5b23a112b2
commit 2fb620211d
1 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/library/entry.cr
View File

@ -1,5 +1,15 @@
require "image_size" require "image_size"
private def node_has_key(node : YAML::Nodes::Mapping, key : String)
node.nodes
.map_with_index { |n, i| {n, i} }
.select(&.[1].even?)
.map(&.[0])
.select(&.is_a?(YAML::Nodes::Scalar))
.map(&.as(YAML::Nodes::Scalar).value)
.includes? key
end
abstract class Entry abstract class Entry
getter id : String, book : Title, title : String, path : String, getter id : String, book : Title, title : String, path : String,
size : String, pages : Int32, mtime : Time, size : String, pages : Int32, mtime : Time,
@ -13,10 +23,20 @@ abstract class Entry
end end
def self.new(ctx : YAML::ParseContext, node : YAML::Nodes::Node) def self.new(ctx : YAML::ParseContext, node : YAML::Nodes::Node)
# TODO: check node? and select proper subclass unless node.is_a? YAML::Nodes::Mapping
ArchiveEntry.new ctx, node raise "Unexpected node type in YAML"
rescue e end
DirEntry.new ctx, node # Doing YAML::Any.new(ctx, node) here causes a weird error, so
# instead we are using a more hacky approach (see `node_has_key`).
# TODO: Use a more elegant approach
if node_has_key node, "zip_path"
ArchiveEntry.new ctx, node
elsif node_has_key node, "dir_path"
DirEntry.new ctx, node
else
raise "Unknown entry found in YAML cache. Try deleting the " \
"`library.yml.gz` file"
end
end end
def build_json(*, slim = false) def build_json(*, slim = false)