go_figure 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

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

data/README.textile

@@ -0,0 +1,45 @@
+h2. Go Figure Gem
+
+This gem generates configuration for Go server - basically given a stage/pipeline 
+group this gem will configure for default pipeline and a simple test stage
+
+h2. Installation
+
+<pre>$ [sudo] gem install go_figure</pre>
+
+h2. Usage
+
+<pre>
+ require 'go_figure'
+ endpoint = GoFigure::GoConfigEndpoint.new(:host => 'example.com', :port => 8153)
+ config = endpoint.get_config
+ config.set_pipeline('http://github.com/user/repo.git', 'rails_working_dir')
+ endpoint.update_config(config)
+</pre>
+
+h2. License
+
+Go Figure Gem is MIT Licensed
+
+The MIT License
+
+Copyright (c) 2012 ThoughtWorks, Inc. (http://thoughtworks.com)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+

data/Rakefile

@@ -0,0 +1,13 @@
+require "bundler/gem_tasks"
+require 'rake/testtask'
+
+task :test do
+  Rake::TestTask.new do |t|
+     t.libs << "test"
+     t.test_files = FileList['test/**/*test.rb']
+     t.verbose = true
+   end
+end
+
+task :default => :test
+

data/go_figure.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "go_figure/version"
+
+Gem::Specification.new do |s|
+  s.name        = "go_figure"
+  s.version     = GoFigure::VERSION
+  s.authors     = ["Ketan Padegaonkar"]
+  s.email       = ["KetanPadegaonkar@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{A Ruby DSL to write a configuration file for the a go server.}
+  s.description = %q{This provides a ruby DSL to create a configuration file for the go server (thoughtworks-studios.com/go)}
+
+  s.rubyforge_project = "go_figure"
+
+  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"]
+  
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  s.add_runtime_dependency "nokogiri", "~> 1.5.0"
+end

data/lib/go-pipelines.xml.erb

@@ -0,0 +1,32 @@
+<pipelines group="defaultGroup">
+  <pipeline name="defaultPipeline">
+    <materials>
+      <git url="<%= @git_url %>" />
+    </materials>
+    <stage name="Units">
+      <jobs>
+        <job name="Test">
+          <tasks>
+            <exec command="bundle" workingdir="<%= @working_dir %>">
+              <arg>install</arg>
+              <arg>--local</arg>
+              <runif status="passed" />
+            </exec>
+            <exec command="bundle" workingdir="<%= @working_dir %>">
+              <arg>exec</arg>
+              <arg>rake</arg>
+              <arg>db:create</arg>
+              <arg>db:migrate</arg>
+              <runif status="passed" />
+            </exec>
+            <exec command="bundle" workingdir="<%= @working_dir %>">
+              <arg>exec</arg>
+              <arg>rake</arg>
+              <runif status="passed" />
+            </exec>
+          </tasks>
+        </job>
+      </jobs>
+    </stage>
+  </pipeline>
+</pipelines>

data/lib/go_figure/go_config.rb

@@ -0,0 +1,51 @@
+require 'nokogiri'
+require 'erb'
+
+module GoFigure
+  class GoConfig
+    attr_accessor :original_md5
+    attr_reader   :original_xml
+
+    def initialize(options = {})
+      @original_md5 = options[:md5]
+      @original_xml = options[:xml]
+      @doc = Nokogiri.XML(@original_xml, nil, 'utf-8')
+    end
+
+    def set_pipeline(git_url, working_dir)
+      @doc.root.xpath('pipelines').remove
+      agents = @doc.root.xpath('agents')
+
+      if agents.any?
+        agents.before(pipeline_template(git_url, working_dir))
+      else
+        @doc.root << pipeline_template(git_url, working_dir)
+      end
+
+      @doc = Nokogiri.XML(@doc.to_s) do |config|
+        config.default_xml.noblanks
+      end
+    end
+
+    def pipeline_template(git_url, working_dir)
+      template = ERB.new(File.read(File.expand_path('../../go-pipelines.xml.erb', __FILE__)))
+      template.result(PipelineConfig.new(git_url, working_dir).get_binding)
+    end
+
+    def xml_content
+      @doc.to_s
+    end
+
+    class PipelineConfig
+      def initialize(git_url, working_dir)
+        @git_url = git_url
+        @working_dir = working_dir
+      end
+
+      def get_binding
+        binding
+      end
+    end
+
+  end
+end

data/lib/go_figure/go_config_endpoint.rb

@@ -0,0 +1,43 @@
+module GoFigure
+
+  class GoConfigEndpoint
+
+    DEFAULT_OPTIONS = {
+      :host => 'localhost',
+      :port => 8153
+    }
+
+    attr_accessor *DEFAULT_OPTIONS.keys
+    attr_accessor :http_fetcher
+
+    def initialize(options)
+      DEFAULT_OPTIONS.each do |k, v|
+        self.send("#{k}=", options[k] || v)
+      end
+    end
+
+    def get_config
+      response = http_fetcher.get(config_xml_url)
+      if response.status == 200
+        return GoConfig.new(:md5 => response['X-CRUISE-CONFIG-MD5'], :xml => response.body)
+      else
+        raise "Could not fetch the go config file"
+      end
+    end
+
+    def update_config(go_config)
+      http_fetcher.post(config_xml_url, :xmlFile => go_config.xml_content, :md5 => go_config.original_md5)
+    end
+
+    def http_fetcher
+      @http_fetcher || HttpFetcher.new
+    end
+
+    private
+    def config_xml_url
+      "http://#{host}:#{port}/go/api/admin/config.xml"
+    end
+
+  end
+
+end

data/lib/go_figure/http_fetcher.rb

@@ -0,0 +1,27 @@
+require 'net/http'
+require 'net/https'
+
+module UserImport
+  class HttpFetcher
+
+    def fetch(url, user, pass)
+      url = URI.parse(url)
+
+      http = Net::HTTP.new(url.host, url.port)
+      http.use_ssl = url.scheme == 'https'
+
+      res = http.start do |http|
+        req = Net::HTTP::Get.new(url.path)
+        req.basic_auth user, pass
+        http.request(req)
+      end
+
+      case res
+      when Net::HTTPSuccess
+        return res.body
+      end
+      res.error!
+    end
+  end
+end
+

data/lib/go_figure/version.rb

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

data/lib/go_figure.rb

@@ -0,0 +1,8 @@
+require "go_figure/version"
+require "go_figure/go_config"
+require "go_figure/http_fetcher"
+require "go_figure/go_config_endpoint"
+
+module GoFigure
+  # Your code goes here...
+end

data/test/fixtures/foo.xml

@@ -0,0 +1,35 @@
+<?xml version="1.0" encoding="utf-8"?>
+<cruise xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:say="com.thoughtworks.cruise.task.plugin.say" xsi:noNamespaceSchemaLocation="cruise-config.xsd" xsi:schemaLocation="com.thoughtworks.cruise.task.plugin.say bundle://1.0:1/say-config.xsd" schemaVersion="52">
+  <pipelines group="defaultGroup1">
+    <pipeline name="MyPipeline">
+      <materials>
+        <git url="git@go-saas.github.com:ThoughtWorksInc/go-saas.git" />
+      </materials>
+      <stage name="Units">
+        <jobs>
+          <job name="UnitTest">
+            <tasks>
+              <exec command="bundle" workingdir="rails">
+                <arg>install</arg>
+                <arg>--local</arg>
+                <runif status="passed" />
+              </exec>
+              <exec command="bundle" workingdir="rails">
+                <arg>exec</arg>
+                <arg>rake</arg>
+                <arg>db:create</arg>
+                <arg>db:migrate</arg>
+                <runif status="passed" />
+              </exec>
+              <rake workingdir="rails" />
+            </tasks>
+          </job>
+        </jobs>
+      </stage>
+    </pipeline>
+  </pipelines>
+  <agents>
+    <agent hostname="ika.corporate.thoughtworks.com" ipaddress="10.12.1.229" uuid="de494931-c26c-44ef-a94d-8835166a3e88" />
+  </agents>
+</cruise>
+

data/test/go_figure/go_config_endpoint_test.rb

@@ -0,0 +1,46 @@
+require 'test_helper'
+
+
+module GoFigure
+  class GoConfigEndpointTest < Test::Unit::TestCase
+
+    def test_should_fetch_xml_config_on_localhost
+      md5 = '7455edff001e2f262beb7c13f10ff7cb'
+      endpoint = GoConfigEndpoint.new(:host => 'example.com', :port => 1234)
+      endpoint.http_fetcher = fetcher_with_config_file('foo.xml', md5, config_endpoint)
+
+      config = endpoint.get_config
+      assert_equal config_file('foo.xml'), config.original_xml
+      assert_equal md5, config.original_md5
+    end
+
+    def test_should_post_back_new_config_xml_content_with_original_md5
+      md5 = '7455edff001e2f262beb7c13f10ff7cb'
+      endpoint = GoConfigEndpoint.new(:host => 'example.com', :port => 1234)
+      endpoint.http_fetcher = fetcher_with_config_file('foo.xml', md5, config_endpoint)
+
+      config = endpoint.get_config
+      config.set_pipeline("http://git.example.com/foo/bar.git", "my_rails_app")
+
+      endpoint.http_fetcher.register_content("OK", config_endpoint, :post, :md5 => config.original_md5, :xmlFile => config.xml_content)
+
+      endpoint.update_config(config)
+      assert endpoint.http_fetcher.invoked?(config_endpoint, :post, :md5 => config.original_md5, :xmlFile => config.xml_content)
+    end
+
+    def config_endpoint
+      "http://example.com:1234/go/api/admin/config.xml"
+    end
+
+    def fetcher_with_config_file(file_name, md5, url)
+      endpoint = StubHttpFetcher.new
+      endpoint.register_content(config_file(file_name), config_endpoint, :get, 'X-CRUISE-CONFIG-MD5' => md5)
+      endpoint
+    end
+
+    def config_file(file_name)
+      File.read("test/fixtures/#{file_name}")
+    end
+
+  end
+end

data/test/go_figure/go_config_test.rb

@@ -0,0 +1,55 @@
+require 'test_helper'
+
+module GoFigure
+  class GoConfigTest < Test::Unit::TestCase
+
+    def test_should_set_pipeline_in_a_config_file_with_no_pipelines_and_no_agents
+      assert_pipeline_template %Q{<?xml version="1.0" encoding="utf-8"?>
+          <cruise />
+      }
+    end
+
+    def test_should_set_pipeline_in_a_config_file_with_pipelines_and_no_agents
+
+      assert_pipeline_template %Q{<?xml version="1.0" encoding="utf-8"?>
+          <cruise>
+            <pipelines group="1">
+            </pipelines>
+
+            <pipelines group="2">
+            </pipelines>
+          </cruise>
+      }
+    end
+
+    def test_should_set_pipeline_in_a_config_file_with_no_pipelines_and_agents
+      assert_pipeline_template %Q{<?xml version="1.0" encoding="utf-8"?>
+          <cruise>
+            <agents />
+         </cruise>
+      }
+    end
+
+    def test_should_set_pipelines_in_a_config_file_with_pipelines_and_agents
+      assert_pipeline_template %Q{<?xml version="1.0" encoding="utf-8"?>
+          <cruise>
+            <pipelines group="1">
+            </pipelines>
+
+            <pipelines group="2">
+            </pipelines>
+             <agents />
+          </cruise>
+      }
+    end
+
+
+    def assert_pipeline_template(xml)
+      config = GoConfig.new(:xml => xml)
+      config.set_pipeline('http://git.example.com/my_project/atlas.git', 'atlas_rails')
+      assert config.xml_content =~ %r{<pipelines group="defaultGroup">}
+      assert config.xml_content =~ %r{<git url="http://git.example.com/my_project/atlas.git"/>}
+    end
+
+  end
+end

data/test/stub_http_fetcher.rb

@@ -0,0 +1,88 @@
+require 'ostruct'
+class StubHttpFetcher
+  # Careful : this will respond to any call with nil
+  class HashStruct < OpenStruct
+    def [](key)
+      self.headers[key]
+    end
+  end
+
+  module Response
+    def invoked!
+      @invoked = true
+    end
+
+    def invoked?
+      !!@invoked
+    end
+  end
+
+  class StringResponse
+    include Response
+    def initialize(content, status = 200, headers = {})
+      @content = content
+      @status = 200
+      @headers = headers
+    end
+
+    def execute
+      HashStruct.new(:body => @content, :status => @status, :headers => @headers)
+    end
+
+  end
+
+  class ErrorResponse
+    include Response
+    def initialize(error)
+      @error = error
+    end
+
+    def execute
+      raise @error
+    end
+  end
+
+  def get(url)
+    if body = interweb[:get][url]
+      body.invoked!
+      return body.execute
+    else
+      raise "404 - Could not find #{url}. Available URLs are #{@interweb[:get].keys.inspect}"
+    end
+  end
+
+  def post(url, params = {})
+    if body = interweb[:post][url]
+      response = body.execute
+      if response.headers == params
+        body.invoked!
+        return response
+      end
+    end
+
+    raise "404 - Could not find a url listening to #{url.inspect} that responds to post params #{params.inspect}. Available URLs are #{@interweb[:post].keys.inspect}"
+  end
+
+  def invoked?(url, type = :get, params = {})
+    if body = interweb[type][url]
+      response = body.execute
+      if response.headers == params
+        body.invoked?
+      end
+    end
+  end
+
+  def register_content(content, url, type = :get, headers = {})
+    interweb[type][url] = StringResponse.new(content, 200, headers)
+  end
+
+  def register_error(url, type = :get, error)
+    interweb[type][url] = ErrorResponse.new(error)
+  end
+
+  def interweb
+    @interweb ||= {:get => {}, :post => {}}
+  end
+
+end
+

data/test/test_helper.rb

@@ -0,0 +1,5 @@
+require 'test/unit'
+require 'stub_http_fetcher'
+
+require 'go_figure'
+

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification
+name: go_figure
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Ketan Padegaonkar
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-14 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: &70114218851080 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70114218851080
+description: This provides a ruby DSL to create a configuration file for the go server
+  (thoughtworks-studios.com/go)
+email:
+- KetanPadegaonkar@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.textile
+- Rakefile
+- go_figure.gemspec
+- lib/go-pipelines.xml.erb
+- lib/go_figure.rb
+- lib/go_figure/go_config.rb
+- lib/go_figure/go_config_endpoint.rb
+- lib/go_figure/http_fetcher.rb
+- lib/go_figure/version.rb
+- test/fixtures/foo.xml
+- test/go_figure/go_config_endpoint_test.rb
+- test/go_figure/go_config_test.rb
+- test/stub_http_fetcher.rb
+- test/test_helper.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: go_figure
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: A Ruby DSL to write a configuration file for the a go server.
+test_files:
+- test/fixtures/foo.xml
+- test/go_figure/go_config_endpoint_test.rb
+- test/go_figure/go_config_test.rb
+- test/stub_http_fetcher.rb
+- test/test_helper.rb