module Legion
  module Data
    class Models
      attr_reader :loaded_models
      def initialize(options = {})
        options.merge!(default_options) { |_key, v1, _v2| v1 }
        @loaded_models = []
        require_sequel_models if options[:auto_load]
      end

      def default_options
        { continue_on_model_fail: false,
          auto_load:              true }
      end

      def models
        %w[chain relationship function task namespace task_log datacenter environment node]
      end

      def require_sequel_models(files = models)
        files.each { |file| load_sequel_model(file) }
      end

      def load_sequel_model(model)
        Legion::Logging.debug("Trying to load #{model}.rb")
        require_relative "models/#{model}"
        @loaded_models << model
        Legion::Logging.debug("Successfully loaded #{model}")
      rescue LoadError => e
        Legion::Logging.fatal("Failed to load #{model}")
        raise e unless Legion::Settings[:data][:models][:continue_on_fail]
      end
    end
  end
end