safety_razor 0.1.0

data/Vagrantfile

@@ -0,0 +1,27 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+Vagrant.configure("2") do |config|
+  config.vm.box = "opscode-ubuntu-12.04"
+  config.vm.box_url = "https://opscode-vm.s3.amazonaws.com/vagrant/opscode_ubuntu-12.04_provisionerless.box"
+
+  config.vm.network :forwarded_port, guest: 8026, host: 8026
+
+  config.omnibus.chef_version = :latest
+  config.berkshelf.enabled = true
+
+  config.vm.provision :chef_solo do |chef|
+    chef.run_list = ["recipe[razor]"]
+    chef.json = {
+      :razor => {
+        :images => {
+          'ubuntu-minimal-10.04' => {
+            'url' => 'http://archive.ubuntu.com/ubuntu/dists/lucid/main/installer-amd64/current/images/netboot/mini.iso',
+            'checksum' => '72602f91a85a856248e519c6446d303fa8990b4328899385990a95177681dc58',
+            'version' => '10.04'
+          }
+        }
+      }
+    }
+  end
+end

data/lib/safety_razor.rb

@@ -0,0 +1,7 @@
+# -*- encoding: utf-8 -*-
+
+require "safety_razor/version"
+require "safety_razor/client"
+
+module SafetyRazor
+end

data/lib/safety_razor/client.rb

@@ -0,0 +1,58 @@
+# -*- encoding: utf-8 -*-
+
+require 'faraday'
+
+require 'safety_razor/slice/active_model'
+require 'safety_razor/slice/broker'
+require 'safety_razor/slice/model'
+require 'safety_razor/slice/node'
+require 'safety_razor/slice/policy'
+require 'safety_razor/slice/tag'
+require 'safety_razor/slice/tag_matcher'
+
+module SafetyRazor
+
+  # Client object which manages the connection to the Razor API endpoint.
+  #
+  # @author Fletcher Nichol <fnichol@nichol.ca>
+  #
+  class Client
+
+    attr_reader :connection
+
+    def initialize(options = {})
+      @connection = Faraday.new(:url => options[:uri]) do |faraday|
+        faraday.request :url_encoded
+        faraday.adapter  Faraday.default_adapter
+      end
+    end
+
+    def active_model
+      @active_model ||= Slice::ActiveModel.new(self)
+    end
+
+    def broker
+      @broker ||= Slice::Broker.new(self)
+    end
+
+    def model
+      @model ||= Slice::Model.new(self)
+    end
+
+    def node
+      @node ||= Slice::Node.new(self)
+    end
+
+    def policy
+      @policy ||= Slice::Policy.new(self)
+    end
+
+    def tag
+      @tag ||= Slice::Tag.new(self)
+    end
+
+    def tag_matcher
+      @tag_matcher ||= Slice::TagMatcher.new(self)
+    end
+  end
+end

data/lib/safety_razor/slice/active_model.rb

@@ -0,0 +1,30 @@
+# -*- encoding: utf-8 -*-
+
+require 'safety_razor/slice/base'
+
+module SafetyRazor
+
+  module Slice
+
+    # Client API for Razor's active model slice.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    #
+    class ActiveModel < Base
+
+      def create(params)
+        raise NoMethodError, "Node#create is not defined"
+      end
+
+      def update(params)
+        raise NoMethodError, "Node#update is not defined"
+      end
+
+      private
+
+      def slice_name
+        "active_model"
+      end
+    end
+  end
+end

data/lib/safety_razor/slice/base.rb

@@ -0,0 +1,89 @@
+# -*- encoding: utf-8 -*-
+
+require 'json'
+require 'hashie'
+
+module SafetyRazor
+
+  module Slice
+
+    # Common behavior for Razor's model slices.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    #
+    class Base
+
+      def initialize(client)
+        @client = client
+      end
+
+      def create(params)
+        payload = JSON.generate(params)
+        response = connection.post(slice_path, 'json_hash' => payload)
+        parse(response).first
+      end
+
+      def get(uuid)
+        response = connection.get(slice_path(uuid))
+        parse(response).first
+      end
+
+      def all
+        response = connection.get(slice_path)
+        parse(response)
+      end
+
+      def update(params)
+        payload = JSON.generate(params)
+        uuid = params[:uuid]
+        response = connection.put(slice_path(uuid), 'json_hash' => payload)
+        parse(response).first
+      end
+
+      def destroy(uuid)
+        connection.delete(slice_path(uuid))
+      end
+
+      protected
+
+      attr_reader :client
+
+      def connection
+        client.connection
+      end
+
+      def slice_path(uuid = nil)
+        path = "/razor/api/#{slice_name}"
+        path += "/#{uuid}" if uuid
+        path
+      end
+
+      def parse(response)
+        collection = JSON.parse(response.body)["response"]
+        collection = [collection] if collection.is_a?(Hash)
+
+        Array(collection).map { |obj| new_mash(strip_ivars(obj)) }
+      end
+
+      def new_mash(obj)
+        Hashie::Mash.new(obj)
+      end
+
+      def strip_ivars(obj)
+        case obj
+        when Hash
+          stripped = Hash.new
+          obj.each_pair do |key, value|
+            new_key = key.is_a?(String) ? key.sub(/^@/, '') : key
+            stripped[new_key] = strip_ivars(value)
+          end
+          stripped
+        when Array
+          obj.map { |value| strip_ivars(value) }
+        else
+          obj
+        end
+      end
+    end
+  end
+end

data/lib/safety_razor/slice/broker.rb

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+
+require 'safety_razor/slice/base'
+
+module SafetyRazor
+
+  module Slice
+
+    # Client API for Razor's broker slice.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    #
+    class Broker < Base
+
+      def plugins
+        response = connection.get(slice_path("plugins"))
+        parse(response)
+      end
+
+      private
+
+      def slice_name
+        "broker"
+      end
+    end
+  end
+end

data/lib/safety_razor/slice/model.rb

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+
+require 'safety_razor/slice/base'
+
+module SafetyRazor
+
+  module Slice
+
+    # Client API for Razor's model slice.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    #
+    class Model < Base
+
+      def templates
+        response = connection.get(slice_path("templates"))
+        parse(response)
+      end
+
+      private
+
+      def slice_name
+        "model"
+      end
+    end
+  end
+end

data/lib/safety_razor/slice/node.rb

@@ -0,0 +1,58 @@
+# -*- encoding: utf-8 -*-
+
+require 'safety_razor/slice/base'
+
+module SafetyRazor
+
+  module Slice
+
+    # Client API for Razor's node slice.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    #
+    class Node < Base
+
+      def register(params)
+        payload = JSON.generate(params)
+        response = connection.post(slice_path("register"),
+          'json_hash' => payload)
+        parse(response).first
+      end
+
+      def checkin(params)
+        payload = JSON.generate(params)
+        response = connection.post(slice_path("checkin"),
+          'json_hash' => payload)
+        parse(response).first
+      end
+
+      def get_attributes(uuid)
+        response = connection.get(slice_path(uuid), "field" => "attributes")
+        parse(response).first
+      end
+
+      def get_hardware_ids(uuid)
+        response = connection.get(slice_path(uuid), "field" => "hardware_ids")
+        parse(response).first["hw_id"]
+      end
+
+      def create(params)
+        raise NoMethodError, "Node#create is not defined"
+      end
+
+      def update(params)
+        raise NoMethodError, "Node#update is not defined"
+      end
+
+      def destroy(uuid)
+        raise NoMethodError, "Node#destroy is not defined"
+      end
+
+      private
+
+      def slice_name
+        "node"
+      end
+    end
+  end
+end

data/lib/safety_razor/slice/policy.rb

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+
+require 'safety_razor/slice/base'
+
+module SafetyRazor
+
+  module Slice
+
+    # Client API for Razor's policy slice.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    #
+    class Policy < Base
+
+      def templates
+        response = connection.get(slice_path("templates"))
+        parse(response)
+      end
+
+      private
+
+      def slice_name
+        "policy"
+      end
+    end
+  end
+end

data/lib/safety_razor/slice/tag.rb

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+
+require 'safety_razor/slice/base'
+
+module SafetyRazor
+
+  module Slice
+
+    # Client API for Razor's tag slice.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    #
+    class Tag < Base
+
+      private
+
+      def slice_name
+        "tag"
+      end
+    end
+  end
+end

data/lib/safety_razor/slice/tag_matcher.rb

@@ -0,0 +1,65 @@
+# -*- encoding: utf-8 -*-
+
+require 'safety_razor/slice/base'
+
+module SafetyRazor
+
+  module Slice
+
+    # Client API for Razor's tag matcher slice.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    #
+    class TagMatcher < Base
+
+      def create(tag_uuid, params)
+        payload = JSON.generate(params)
+        response = connection.post(slice_path(tag_uuid), 'json_hash' => payload)
+        parse(response).first
+      end
+
+      def get(tag_uuid, uuid)
+        response = connection.get(slice_path(tag_uuid, uuid))
+        parse(response).first
+      end
+
+      def all
+        raise NoMethodError, "TagMatcher#all is not defined"
+      end
+
+      def update(tag_uuid, params)
+        payload = JSON.generate(params)
+        uuid = params[:uuid]
+        response = connection.put(slice_path(tag_uuid, uuid),
+          'json_hash' => payload)
+        parse(response).first
+      end
+
+      def destroy(tag_uuid, uuid)
+        connection.delete(slice_path(tag_uuid, uuid))
+      end
+
+      protected
+
+      def slice_path(tag_uuid, uuid = nil)
+        path = "/razor/api/tag/#{tag_uuid}/matcher"
+        path += "/#{uuid}" if uuid
+        path
+      end
+
+      def new_mash(obj)
+        Hashie::Mash.new(obj).extend(KeyReaderMethod)
+      end
+
+      # Overrides a Hash method collision with the tag matcher attribute of
+      # `key'.
+      #
+      module KeyReaderMethod
+
+        def key
+          self["key"]
+        end
+      end
+    end
+  end
+end