Contents

require 'etcdist/log'

module Etcdist
  # Reads data from file system into directories, keys and values.
  class Reader
    # @param [String] dir The path to the data directory
    def initialize(dir)
      @dir = dir
    end

    # Returns a hash of type { directory => { key => val } }
    def read
      @dir = File.expand_path(@dir)
      files = Dir[File.join(@dir, '**', '*')].reject { |p| File.directory? p }
      Log.info("found #{files.length} files in #{@dir}")

      files.reduce(Hash.new { |h, k| h[k] = {} }) do |h, f|
        directory = File.dirname(f).gsub(@dir, '')
        entries = Hash[IO.readlines(f).map { |e| e.chomp.split('=', 2) }.select { |k, _| valid_key?(k) }]
        Log.debug("found #{entries.length} entries in #{f.gsub(@dir, '')}: #{entries}")
        h[directory].merge!(entries)
        h
      end
    end

    private

    def valid_key?(key)
      is_valid = !(key.include? '/')
      Log.warn("ignoring invalid key #{key}") unless is_valid
      is_valid
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
etcdist-0.0.5	lib/etcdist/reader.rb
etcdist-0.0.4	lib/etcdist/reader.rb