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