yaml_master 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 34993300413ebe4676e7cf80c868d219262334dc
-  data.tar.gz: 6ab8999f19b39b2d61116afea9bd67f406bc7b94
+  metadata.gz: 53a07ec3606692dbdf0cdc7c8250bbacf399097d
+  data.tar.gz: f1dc0b2f19e919f1ea6d2d8dcd921320a0fa8643
 SHA512:
-  metadata.gz: 1096589dade1789f856a57e3bddd8b05ccf4a89c97965743a530b5f6d92efda5a9b25290c6260d0a66fd436ac57872a002b62f613fe9f07ba9f2668206326005
-  data.tar.gz: 282ed2538a8ef0c9fbc2deb54581f7402cc89d7805c788ccba37d8a604b486055debaf6ffe214a6d29efe00d649fabd7393da295fed56ae838c3cc613c0ad89f
+  metadata.gz: 513e962e3febc9658017f1095a9ce800308b1bcb285f105057419edc32757ecfbb11249a81cb8642e02fd9f3b39d26805cfbb304418f57166a4c005c96339f91
+  data.tar.gz: 5fb39d061b168fba6056db7dcf3aee0063e51f66e38c66b4a1d902ce33778af129795d3884495f718ce2bb6a9869c23d908d1166fc7f416d854b9c0a7e193ea3

data/lib/yaml_master.rb

@@ -3,12 +3,19 @@ require "yaml_master/version"
 require "yaml"
 require "erb"
 require "pathname"
+require "pp"
 
 YAML.add_domain_type(nil, "include") do |type, val|
   YAML.load_file(val)
 end
 
 class YamlMaster
+  class KeyFetchError < StandardError
+    def initialize(data, key)
+      super("cannot fetch key \"#{key}\" from\n#{data.pretty_inspect}")
+    end
+  end
+
   attr_reader :master, :master_path
 
   def initialize(io_or_filename)
@@ -27,7 +34,7 @@ class YamlMaster
     raise "data key is necessary on toplevel" unless @master["data"]
   end
 
-  def generate(key, output, options = {})
+  def generate(key, output = nil, options = {})
     puts "gen: #{output}" if options[:verbose]
     yaml = YAML.dump(fetch_data_from_master(key))
 
@@ -47,7 +54,24 @@ class YamlMaster
   private
 
   def fetch_data_from_master(key)
-    @master["data"].fetch(key)
+    keys = split_key(key)
+    keys.inject(@master["data"]) do |data, k|
+      data.fetch(k)
+    end
+  rescue
+    raise KeyFetchError.new(@master["data"], key)
+  end
+
+  def split_key(key)
+    keys = key.split(".")
+    array_pattern = /\[(\d+)\]/
+    keys.map do |k|
+      if k.match(array_pattern)
+        Regexp.last_match[1].to_i
+      else
+        k
+      end
+    end
   end
 
   class EmbeddedMethods

data/lib/yaml_master/version.rb

@@ -1,3 +1,3 @@
 class YamlMaster
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: yaml_master
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - joker1007