mio-config 1.3.0

data/lib/mio.rb

@@ -0,0 +1,24 @@
+require 'mio/client'
+require 'mio/search'
+require 'mio/errors'
+
+require 'mio/model'
+require 'mio/model/autoload'
+
+class Mio
+
+  attr_accessor :base_uri, :username, :password
+  attr_reader :client
+  def initialize base_uri=nil, username=nil, password=nil
+    @base_uri = base_uri
+    @username = username
+    @password = password
+
+    if block_given?
+      yield self
+    end
+
+    @client = Mio::Client.new @base_uri, @username, @password
+  end
+
+end

data/lib/mio/#model.rb#

@@ -0,0 +1,130 @@
+class Mio
+  class Model
+    class << self
+      attr_accessor :fields, :resource_name
+      def set_resource r
+        @resource_name = r.to_s
+      end
+
+      def field key, type, desc, default=nil, matcher=nil
+        @fields ||= []
+        @fields << {name: key,
+                    type: type,
+                    default: default,
+                    matcher: matcher,
+                    desc: desc,}
+
+      end
+
+      def nested val=nil
+        if val.nil?
+          @nested_value || false
+        else
+          @nested_value = val
+        end
+      end
+      alias_method :nested?, :nested
+
+      def mappings
+        m = {}
+        ObjectSpace.each_object(Class).each do |k|
+          if k < Mio::Model
+            m[ k.to_s.split('::').last.downcase ] = k
+          end
+        end
+        m
+      end
+    end
+
+    def initialize client, args
+      @client = client
+      @args   = args
+      @search = Mio::Search.new @client
+    end
+
+    def go
+      unless look_up
+        @object = create
+      else
+        @object = look_up
+
+        # We can't edit a running resource
+        set_start :stop
+      end
+
+      configure if self.respond_to? :config_hash
+      set_enable
+      set_start
+
+      return @object
+    end
+
+    def create
+      @client.create self.class.resource_name, create_hash
+    end
+
+    def configure
+      @client.configure self.class.resource_name,
+                        @object['id'],
+                        config_hash
+    end
+
+    def set_enable a=nil
+      if a.nil?
+        action = @args.enable == :true ? 'enable' : 'disable'
+      else
+        action = a.to_s
+      end
+      @client.action self.class.resource_name,
+                     @object['id'],
+                     {action: action}
+    end
+    alias_method :disable!, :set_enable
+    alias_method :enable!, :set_enable
+
+    def set_start a=nil
+      if a.nil?
+        action = @args.start == :true ? 'start' : 'stop'
+      else
+        action = a.to_s
+      end
+      @client.action self.class.resource_name,
+                     @object['id'],
+                     {action: action}
+    end
+    alias_method :stop!, :set_start
+    alias_method :start!, :set_start
+
+    def validate
+      testable = @args.dup.to_h
+
+      self.class.fields.each do |f|
+        unless testable.key? f[:name]
+          raise Mio::Model::MissingField, "Missing field #{f[:name]} to #{self}"
+        end
+
+        extracted_field = testable.delete f[:name]
+        unless extracted_field.is_a? f[:type]
+          raise Mio::Model::DataTypeError, "#{f[:name]} should be of type #{f[:type]} for #{self}"
+        end
+
+        unless f['matcher'].nil? or extracted_field.to_s.match(f[:matcher])
+          raise Mio::Model::DataValueError, "#{self} #{f[:name]} value '#{extracted_field}' does not match #{f[:matcher]}"
+        end
+      end
+
+      testable.keys.each do |k|
+        raise Mio::Model::NoSuchField, "#{k} for #{self}"
+      end
+      true
+    end
+    alias_method :valid?, :validate
+
+    private
+    def look_up
+      r = self.class.resource_name
+      @client.find_all(r)[r].find{|o| o['name'] == @args.name}
+    end
+  end
+
+end

data/lib/mio/client.rb

@@ -0,0 +1,106 @@
+require 'faraday'
+require 'faraday/detailed_logger'
+require 'hashie/mash'
+require 'net/http/persistent'
+require 'json'
+
+require 'pp'
+
+require 'mio/requests'
+
+class Mio
+  class Client
+
+    def initialize base_uri, username, password
+      @base_uri = base_uri
+      @agent = Faraday.new(url: base_uri) do |f|
+        f.adapter :net_http_persistent
+        f.response :detailed_logger if ENV.fetch('VERBOSE', 'false').to_s.downcase == 'true'
+      end
+      @agent.basic_auth(username, password)
+    end
+
+    def find_all resource, opts={}
+      url = path(resource)
+      response = get url, opts
+
+      unless response.success?
+        raise Mio::Client::LoadOfBollocks, "GET on #{url} returned #{response.status}"
+      end
+      make_object response.body
+    end
+
+    def create resource, payload, opts={}
+      url = path(resource)
+      response = post url, payload, opts
+      unless response.success?
+        raise Mio::Client::LoadOfBollocks, "POST on #{url}, with #{payload.inspect} returned #{response.status}"
+      end
+
+      make_object response.body
+    end
+
+    def update resource, payload, opts={}
+      url = path(resource)
+      response = put url, payload, opts
+      unless response.success?
+        raise Mio::Client::LoadOfBollocks, "PUT on #{url}, with #{payload.inspect} returned #{response.status}"
+      end
+
+      make_object response.body
+    end
+
+
+    def configure resource, id, payload, opts={}
+      url = path(resource, id, :configuration)
+      response = put url, payload, opts
+      unless response.success?
+        raise Mio::Client::LoadOfBollocks, "PUT on #{url}, with #{payload.inspect} returned #{response.status}"
+      end
+
+      make_object response.body
+    end
+
+    def action resource, id, payload, opts={}
+      url = path(resource, id, :actions)
+      statuses = get url, opts
+      unless statuses.success?
+        raise Mio::Client::LoadOfBollocks, "GET on #{url} returned #{statuses.status}"
+      end
+
+      if JSON.parse(statuses.body).find{|h| h['action'] == payload[:action]}
+        response = post url, payload, opts
+
+        unless response.success?
+          raise Mio::Client::LoadOfBollocks, "PUT on #{url}, with #{payload.inspect} returned #{response.status}"
+        end
+        return make_object response.body
+      end
+    end
+
+    private
+    def get url, opts
+      Mio::Requests.make_request :get, @agent, url, opts
+    end
+
+    def post url, payload, opts
+      Mio::Requests.make_request :post, @agent, url, opts, payload
+    end
+
+    def put url, payload, opts
+      Mio::Requests.make_request :put, @agent, url, opts, payload
+    end
+
+    def make_object response
+      JSON.parse(response)
+    end
+
+    def path resource, id=nil, endpoint=nil
+      path_string = "#{@base_uri}/#{resource.sub(/^\//, '')}"
+      path_string += "/#{id}" if id
+      path_string += "/#{endpoint.to_s}" if endpoint
+      path_string
+    end
+
+  end
+end

data/lib/mio/config.rb

@@ -0,0 +1,14 @@
+require 'hashie/mash'
+require 'yaml'
+require 'erb'
+
+class Mio
+  class Config
+    def self.read filename
+      unless File.exist? filename
+        raise Mio::Config::FileMissing, "The config file '#{filename}' is missing"
+      end
+      OpenStruct.new YAML.load( ERB.new(File.read(filename)).result )
+    end
+  end
+end

data/lib/mio/errors.rb

@@ -0,0 +1,18 @@
+class Mio
+  class Client
+    class LoadOfBollocks < ArgumentError ; end
+  end
+
+  class Model
+    class NoSuchField < ArgumentError ; end
+    class MissingField < ArgumentError ; end
+    class DataTypeError < ArgumentError ; end
+    class DataValueError < ArgumentError ; end
+    class NoSuchResource < ArgumentError ; end
+    class EmptyField < ArgumentError ; end
+  end
+
+  class Config
+    class FileMissing < IOError ; end
+  end
+end

data/lib/mio/migrations.rb

@@ -0,0 +1,96 @@
+require 'mio'
+require 'colorize'
+require 'mio/config'
+
+class Mio
+  class Migrations
+    def initialize base_uri, username, password, base_dir='./migrations'
+      @mio = Mio.new do |m|
+        m.base_uri = base_uri
+        m.username = username
+        m.password = password
+      end
+      @base = File.expand_path(base_dir)
+      @migrations = Dir.glob( File.join(@base, '*.rb') ).sort
+    end
+
+    def create_migration_file desc
+      File.join(@base,
+                "#{Time.now.strftime('%Y%m%d%H%M%S')}_#{desc.gsub(/\W/, '-')}.rb")
+    end
+
+    def run_migrations
+      @migrations.each do |migration|
+        unless has_run? migration
+          load_migration migration
+        end
+      end
+    end
+
+    def has_run? migration
+      false
+    end
+
+    def load_migration migration
+      eval File.read(migration)
+    end
+
+    # Helper methods for within the migrations
+    def migrate desc, &block
+      msg desc, 'starting'
+      block.call
+
+      msg desc, 'completed'
+    end
+
+    def type_migration klass, conf
+      if klass.nested?
+        get_it klass.new(@mio.client, conf)
+      else
+        do_it klass.new(@mio.client, conf)
+      end
+    end
+
+    def method_missing method_sym, *arguments, &block
+      mapped = Mio::Model.mappings[method_sym.to_s]
+
+      if mapped.nil?
+        super
+      else
+        conf = OpenStruct.new
+        yield conf
+        type_migration mapped, conf
+      end
+    end
+
+    def respond_to? method_sym, include_private=false
+      if Mio::Model.mappings[method_sym.to_s].nil?
+        super
+      else
+        true
+      end
+    end
+
+    private
+    def do_it thing
+      if thing.valid?
+        obj = thing.go
+      end
+      puts "Created '#{obj['name']}' with id '#{obj['id']}'"
+    end
+
+    def get_it thing
+      if thing.valid?
+        return thing.create_hash
+      end
+      return nil
+    end
+
+
+    def msg desc, state
+      print '======> '.magenta
+      print "#{desc}:\t".cyan
+      puts "#{state.upcase}".magenta
+    end
+  end
+end

data/lib/mio/model.rb

@@ -0,0 +1,133 @@
+class Mio
+  class Model
+    class << self
+      attr_accessor :fields, :resource_name
+      def set_resource r
+        @resource_name = r.to_s
+      end
+
+      def field key, type, desc, default=nil, matcher=nil
+        @fields ||= []
+        @fields << {name: key,
+                    type: type,
+                    default: default,
+                    matcher: matcher,
+                    desc: desc,}
+
+      end
+
+      def nested val=nil
+        if val.nil?
+          @nested_value || false
+        else
+          @nested_value = val
+        end
+      end
+      alias_method :nested?, :nested
+
+      def mappings
+        m = {}
+        ObjectSpace.each_object(Class).each do |k|
+          if k < Mio::Model
+            m[ k.to_s.split('::').last.downcase ] = k
+          end
+        end
+        m
+      end
+    end
+
+    def initialize client, args
+      @client = client
+      @args   = args
+      @search = Mio::Search.new @client
+    end
+
+    def go
+      unless look_up
+        @object = create
+      else
+        @object = look_up
+
+        # We can't edit a running resource
+        set_start :stop
+      end
+
+      configure if self.respond_to? :config_hash
+      set_enable
+      set_start
+
+      return @object
+    end
+
+    def create
+      @client.create self.class.resource_name, create_hash
+    end
+
+    def configure
+      @client.configure self.class.resource_name,
+                        @object['id'],
+                        config_hash
+    end
+
+    def set_enable a=nil
+      if a.nil?
+        action = @args.enable == :true ? 'enable' : 'disable'
+      else
+        action = a.to_s
+      end
+      @client.action self.class.resource_name,
+                     @object['id'],
+                     {action: action}
+    end
+    alias_method :disable!, :set_enable
+    alias_method :enable!, :set_enable
+
+    def set_start a=nil
+      if a.nil?
+        action = @args.start == :true ? 'start' : 'stop'
+      else
+        action = a.to_s
+      end
+      @client.action self.class.resource_name,
+                     @object['id'],
+                     {action: action}
+    end
+    alias_method :stop!, :set_start
+    alias_method :start!, :set_start
+
+    def validate
+      testable = @args.dup.to_h
+
+      self.class.fields.each do |f|
+        unless testable.key? f[:name]
+          raise Mio::Model::MissingField, "Missing field #{f[:name]} to #{self}"
+        end
+
+        extracted_field = testable.delete f[:name]
+        unless extracted_field.is_a? f[:type]
+          raise Mio::Model::DataTypeError, "#{f[:name]} should be of type #{f[:type]} for #{self}"
+        end
+
+        unless f['matcher'].nil? or extracted_field.to_s.match(f[:matcher])
+          raise Mio::Model::DataValueError, "#{self} #{f[:name]} value '#{extracted_field}' does not match #{f[:matcher]}"
+        end
+      end
+
+      testable.keys.each do |k|
+        raise Mio::Model::NoSuchField, "#{k} for #{self}"
+      end
+      true
+    end
+    alias_method :valid?, :validate
+
+    private
+    def look_up
+      r = self.class.resource_name
+      all_resources = @client.find_all(r)
+      return nil if all_resources['totalCount'] == 0
+
+      all_resources[r].find{|o| o['name'] == @args.name}
+    end
+  end
+
+end