class DataFiles::REPL
Loads all yaml files in given directory, creates ActiveData
subclass for each file and provides an interactive shell.
Public Class Methods
new(directory)
click to toggle source
# File lib/data_files/repl.rb, line 13 def initialize(directory) @klass_names = [] parse_data(directory) end
Public Instance Methods
create_class(klass_name, data)
click to toggle source
# File lib/data_files/repl.rb, line 31 def create_class(klass_name, data) @klass_names << klass_name types = parse_types(data) klass = Class.new(ActiveData) do class_variable_set(:@@data, data) class_variable_set(:@@attributes, data.first.keys) class_variable_set(:@@types, types) attr_accessor(*data.first.keys) end Object.const_set(klass_name, klass) end
initial_prompt()
click to toggle source
# File lib/data_files/repl.rb, line 51 def initial_prompt puts 'Available data models:' @klass_names.sort.each do |klass_name| puts " - #{klass_name}" end end
parse_data(directory)
click to toggle source
# File lib/data_files/repl.rb, line 18 def parse_data(directory) Dir.foreach(File.join(directory, 'data')) do |filename| next unless filename.end_with?('.yml') filepath = File.join(directory, 'data', filename) key = File.basename(filepath, File.extname(filepath)) data = load_yaml(filepath) klass_name = key.capitalize.delete_suffix('s') create_class(klass_name, data) end end
prompt()
click to toggle source
# File lib/data_files/repl.rb, line 45 def prompt initial_prompt DataFiles::Autocompletion.new read_input end
read_input()
click to toggle source
# File lib/data_files/repl.rb, line 58 def read_input bnd = binding while (input = Readline.readline('> ', true)) begin puts bnd.eval(input).to_s rescue StandardError => e puts "\e[31m#{e.class}:\e[0m #{e.message}" end end end
Private Instance Methods
load_yaml(filepath)
click to toggle source
# File lib/data_files/repl.rb, line 71 def load_yaml(filepath) YAML .safe_load(File.read(filepath), permitted_classes: [Date]) .map .each_with_index { |item, index| item.merge('_id' => index + 1) } end
parse_types(data)
click to toggle source
# File lib/data_files/repl.rb, line 78 def parse_types(data) types = {} data.first.keys.each do |attr| types[attr] = data.collect { |item| item[attr].class.name } types[attr] << 'TrueClass' if types[attr].include?('FalseClass') types[attr] << 'FalseClass' if types[attr].include?('TrueClass') types[attr].uniq! end types end