Sha256: 243c9b944cb9088a75791963017c0d11d440d3f243d13a8aa25003e8721feb20

Contents?: true

Size: 1.38 KB

Versions: 2

Compression:

Stored size: 1.38 KB

Contents

require 'json'
require 'active_support/core_ext/hash'
require 'rest-client'
require 'yaml'
require_relative 'macro'
require_relative 'argbucket'

DEFAULT_CONFIG = '/tmp/apidragonconf.yaml'
PLUGINS = '/tmp/apidragonplugins/'
LOG = '/tmp/apidragon.log'

class Api < ArgBucket
  # Uses the configuration data in config.yaml to run a sequence of api calls.
  # All variables are dumped into a variable bucket (@arg_bucket)
  # so that they can be used as parameters for function calls.

  def initialize(macro_name, config)
    `mkdir #{PLUGINS}` unless Dir.exist? PLUGINS
    @arg_bucket = {}
    @config = load_config config
    import @config['vars']
    @macro = @config['macros'][macro_name]
    if @macro.nil? then fail "Command: '#{macro_name}' is not defined." end
  end

  def load_config(config)
    if config.nil? then config = DEFAULT_CONFIG end
    YAML.load_file config
  end

  def go
    @macro.each_pair do |_key, value|
      call = Call.new(value, @arg_bucket)
      @arg_bucket = call.run
      if !value['record'].nil?
        value['record'].each do |var|
          add_config_var var, get(var)
        end
      end
    end
  end

  def import(args)
    args.each_pair do |key, value|
      set key, value
    end
  end

  def add_config_var(key,value)
    @config['vars'][key] = value
    file = File.open CONFIG, 'w'
    file.write(@config.to_yaml.gsub("\n-", "\n\n-"))
    file.close
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version Path
apidragon-1.4.3 lib/apidragon/api.rb
apidragon-1.4.1 lib/apidragon/api.rb