classify_cluster 0.0.1

data/.gitignore

@@ -0,0 +1,3 @@
+pkg/*
+*.gem
+.bundle

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in classify_cluster.gemspec
+gemspec

data/README

@@ -0,0 +1,5 @@
+Classify Cluster
+
+Useful if your wanting to have one location for specifying your capistrano and puppet configuration.
+
+This is still extremely alpha, do not recommend use on production systems.  More on documentation coming soon.

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/bin/classify

@@ -0,0 +1,23 @@
+#!/usr/bin/env ruby
+
+require "classify_cluster"
+require "classify_cluster/version"
+require "commander/import"
+require 'yaml'
+
+program :name, "Classify Cluster"
+program :version, ClassifyCluster::VERSION
+program :description, 'Generates configurations for different aspects of the cluster based off of one configuration.'
+default_command :puppet
+
+command :puppet do |c|
+  c.syntax = 'classify puppet <clustername> [options]'
+  c.description = 'Returns on STDOUT YAML dump of the clusters puppet config'
+  c.option '--config STRING', String, 'Location to config file.'
+  c.option '--export STRING', String, 'Location to output puppet cluster config'
+  c.action do |args, options|
+    options.default :config => ClassifyCluster::Base.default_config_file, :export => ENV['HOME']
+    ClassifyCluster::Writers::Puppet.export!(options.export, :cluster => args[0], :config_file => options.config)
+    say("Success!")
+  end
+end

data/classify_cluster.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "classify_cluster/version"
+
+Gem::Specification.new do |s|
+  s.name        = "classify_cluster"
+  s.version     = ClassifyCluster::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Sean Cashin"]
+  s.email       = ["sean@socialcast.com"]
+  s.homepage    = "http://rubygems.org/gems/classify_cluster"
+  s.summary     = %q{Contains several binaries for generating capistrano and puppet configurations}
+  s.description = %q{Reading from a YAML file will allow for consistent configuration between capistrano and puppet}
+  s.add_dependency(%q<commander>, ["= 4.0.3"])
+  s.add_dependency(%q<activesupport>, [">= 2.3.5"])
+
+  s.rubyforge_project = "classify_cluster"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

data/lib/classify_cluster/base.rb

@@ -0,0 +1,7 @@
+module ClassifyCluster
+  class Base
+    def self.default_config_file
+      File.join('/etc', 'cluster.rb')
+    end
+  end
+end

data/lib/classify_cluster/configurator.rb

@@ -0,0 +1,103 @@
+module ClassifyCluster
+  module Configurator
+    class Configuration
+      attr_reader :clusters
+      def initialize(config_path)
+        @clusters = {}
+        eval File.open(config_path).read
+      end
+      
+      def cluster(cluster_name, &block)
+        @clusters[cluster_name] = ClassifyCluster::Configurator::Cluster.new(cluster_name, &block)
+      end
+    end
+    class Cluster
+      attr_reader :nodes, :name
+      def initialize(*args, &block)
+        @nodes = {}
+        @name = args.first
+        block.call self
+      end
+      def name(value=nil)
+        return @name unless value
+        @name = value
+      end
+      def node(node_name, &block)
+        @nodes[node_name] = ClassifyCluster::Configurator::Node.new(node_name, &block)
+      end
+    end
+    class Node
+      attr_reader :fqdn, :variables, :resources, :classes, :public_ip, :private_ip, :roles, :default
+      def initialize(*args, &block)
+        @variables = {}
+        @resources = []
+        @classes = []
+        @roles = []
+        @fqdn = (args.first.to_s == 'default' ? '' : args.first)
+        @default = args.first.to_s == 'default'
+        block.call self
+      end
+      def default?
+        return @default
+      end
+      def fqdn(value=nil)
+        return @fqdn unless value
+        @fqdn = value
+      end
+      def public_ip(value = nil)
+        return @public_ip unless value
+        @public_ip = value
+      end
+      def private_ip(value = nil)
+        return @private_ip unless value
+        @private_ip = value
+      end
+      def role(&block)
+        @roles << ClassifyCluster::Configurator::Role.new(&block)
+      end
+      def variable(name, value)
+        @variables[name] = value
+      end
+      def resource(&block)
+        @resources << ClassifyCluster::Configurator::Resource.new(&block)
+      end
+      def klass(name)
+        @classes << name
+      end
+    end
+    class Resource
+      attr_reader :type, :name, :options
+      def initialize(*args, &block)
+        @options = {}
+        block.call self
+      end
+      def type(value = nil)
+        return @type unless value
+        @type = value
+      end
+      def name(value = nil)
+        return @name unless value
+        @name = value
+      end
+      def options(value = nil)
+        return @options unless value
+        @options = value
+      end
+    end
+    class Role
+      attr_reader :type, :options
+      def initialize(*args, &block)
+        @options = {}
+        block.call self
+      end
+      def type(value = nil)
+        return @type unless value
+        @type = value
+      end
+      def options(value = nil)
+        return @options unless value
+        @options = value
+      end
+    end
+  end
+end

data/lib/classify_cluster/version.rb

@@ -0,0 +1,3 @@
+module ClassifyCluster
+  VERSION = "0.0.1"
+end

data/lib/classify_cluster/writers/capistrano.rb

@@ -0,0 +1,16 @@
+module ClassifyCluster
+  module Writers
+    class Capistrano
+      def self.export!(capistrano_configurator, cluster, config_file=ClassifyCluster::Base.default_config_file)
+        config = ClassifyCluster::Configurator::Configuration.new(config_file).clusters[cluster]
+        config.nodes.each_pair do |name, node|
+          roles = node.roles
+          next if roles.empty?
+          roles.each do |role|
+            capistrano_configurator.role(role.type, node.public_ip, role.options)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/classify_cluster/writers/puppet.rb

@@ -0,0 +1,39 @@
+module ClassifyCluster
+  module Writers
+    class Puppet
+      def self.export!(export_to_folder, options={})
+        options.reverse_merge! :config_file => ClassifyCluster::Base.default_config_file
+        config = ClassifyCluster::Configurator::Configuration.new(options[:config_file])
+        config.clusters.each_pair do |name, cluster|
+          next if options[:cluster] && !(options[:cluster] == cluster.name.to_s)
+          File.open(File.join(export_to_folder, "#{cluster.name}.pp"), 'w') do |file|
+            cluster.nodes.each_pair do |fqdn, node|
+              file.write(output(%Q%node "#{node.default? ? 'default' : node.fqdn}" {%))
+              node.variables.each_pair do |key, value|
+                file.write(output("$#{key}=#{value.inspect}", :indent => 1))
+              end
+              node.resources.each do |resource|
+                file.write(output("#{resource.type} { #{resource.name.inspect}:", :indent => 1))
+                count = 0
+                resource.options.each_pair do |key, value|
+                  file.write(output("#{key} => #{value.inspect}#{"," if count >= resource.options.size}", :indent => 2))
+                  count += 1
+                end
+                file.write(output("}", :indent => 1))
+              end
+              node.classes.each do |klass|
+                file.write(output("include #{klass}", :indent => 1))
+              end
+            end
+            file.write(output("}"))
+          end
+        end; nil
+      end
+      private
+      def self.output(string, options={})
+        options.reverse_merge! :indent => 0
+        %Q%#{("\s" * 2) * options[:indent]}#{string}\n%
+      end
+    end
+  end
+end

data/lib/classify_cluster/writers.rb

@@ -0,0 +1,2 @@
+require 'classify_cluster/writers/capistrano'
+require 'classify_cluster/writers/puppet'

data/lib/classify_cluster.rb

@@ -0,0 +1,5 @@
+require 'yaml'
+require 'active_support'
+require 'classify_cluster/base'
+require 'classify_cluster/configurator'
+require 'classify_cluster/writers'

metadata

@@ -0,0 +1,111 @@
+--- !ruby/object:Gem::Specification 
+name: classify_cluster
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Sean Cashin
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-10-22 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: commander
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 57
+        segments: 
+        - 4
+        - 0
+        - 3
+        version: 4.0.3
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: activesupport
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 9
+        segments: 
+        - 2
+        - 3
+        - 5
+        version: 2.3.5
+  type: :runtime
+  version_requirements: *id002
+description: Reading from a YAML file will allow for consistent configuration between capistrano and puppet
+email: 
+- sean@socialcast.com
+executables: 
+- classify
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README
+- Rakefile
+- bin/classify
+- classify_cluster.gemspec
+- lib/classify_cluster.rb
+- lib/classify_cluster/base.rb
+- lib/classify_cluster/configurator.rb
+- lib/classify_cluster/version.rb
+- lib/classify_cluster/writers.rb
+- lib/classify_cluster/writers/capistrano.rb
+- lib/classify_cluster/writers/puppet.rb
+has_rdoc: true
+homepage: http://rubygems.org/gems/classify_cluster
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: classify_cluster
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Contains several binaries for generating capistrano and puppet configurations
+test_files: []
+