small-ops 0.0.1.beta

data/bin/docker2etcd

@@ -0,0 +1,24 @@
+#!/usr/bin/env ruby
+
+require 'small-setup'
+
+`docker ps | tail -n+2 | awk '{ print $1 }'`.split("\n")
+    .map  { |id| `docker inspect #{id}` }
+    .map  { |inspect| JSON.parse(inspect)[0]}
+    .map  { |data| flatten(data,data["Name"]) }
+    .each { |container| 
+        name = container.keys.first.split("/")[1]
+        port = `docker inspect #{container["/#{name}/id"]} | grep -o '[0-9]\\+/tcp' | head -n1 | grep -o '[0-9]\\+'`.gsub("\n","")
+        hport = container["/#{name}/networksettings/ports/#{port}/tcp/hostport"]
+        container["/#{name}/name"] = name
+        container["/#{name}/port"] = hport
+        container["/#{name}/host"] = @host
+        container["/#{name}/url"] = "http://#{@host}:#{hport}"
+        container.keys.each {|key|
+            if container[key] != nil then
+                puts "#{key} = #{container[key]}"
+                http_put("#{@etcd}/v2/keys#{@prefix}#{key}","value=#{URI.encode(container[key])}")
+            end
+        }
+    }
+

data/bin/etcd2env

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+
+require 'small-setup'
+
+data = http_get("#{@etcd}/v2/keys/#{@prefix}?recursive=true")
+
+final = eflatten( data["node"] )
+final.keys.each {|k|
+  puts "export #{k}=\"#{final[k]}\""
+  puts "export #{k.upcase}=\"#{final[k]}\""
+}
+

data/lib/small-setup.rb

@@ -0,0 +1,107 @@
+
+require 'uri'
+require 'json'
+require 'net/http'
+require 'optparse'
+
+options = {}
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: command [options]"
+  opts.on("-h", "--host host", "Set host IP/domain") do |h|
+    options[:host] = h
+  end
+  opts.on("-e", "--etcd host:port", "Set etcd host and port") do |e|
+    options[:etcd] = e
+  end
+  opts.on("-p", "--prefix prefix", "Set etcd prefix path") do |p|
+    options[:prefix] = p
+  end
+  opts.on("-d","--daemon","Run in background") do |d|
+    options[:daemon]=d
+  end
+  opts.on("-o","--output file","Output file") do |o|
+    options[:output]=o
+  end
+  opts.on("-i","--input file","Input file") do |o|
+    options[:input]=o
+  end
+end.parse!
+
+
+@host = options[:host] || ENV['HOST'] || `hostname -I | awk '{ print $1 }'`.gsub("\n","")
+@etcd = options[:etcd] || ENV['ETCD'] || "http://#{@host}:4001"
+@prefix = options[:prefix] || ENV['PREFIX'] || ""
+@foreground = !options[:daemon]
+@output = options[:output] || false
+@input = options[:input] || false 
+
+def http_get(uri)
+    JSON.parse(Net::HTTP.get(URI(uri)))
+end
+
+def http_put(uri,doc) 
+    uri = URI.parse(uri)
+    header = {'Content-Type'=> 'application/x-www-form-urlencoded'}
+    if doc.class == Hash then
+        header = {'Content-Type'=> 'application/json'}
+    end
+    http = Net::HTTP.new(uri.host, uri.port)
+    request = Net::HTTP::Put.new(uri.request_uri, header)
+    request.body = doc
+    if doc.class == Hash then
+        request.body=doc.to_json
+    end
+    response = http.request(request)
+    if response.body.length >= 2 then
+        JSON.parse(response.body)
+    else
+        {}
+    end
+end
+
+def flatten(obj,sub)
+    flat={}
+    sub=sub.gsub("-","_").downcase
+    obj.keys.each {|k|
+        key=k.gsub("-","_").gsub("/","_").downcase
+        if obj[k].class == Array then
+            if(obj[k][0].class == Hash) then
+                flat=flat.merge(flatten(obj[k][0],"#{sub}/#{k}"))
+            end
+        elsif obj[k].class == Hash then
+            flat=flat.merge(flatten(obj[k],"#{sub}/#{k}"))
+        else
+            flat["#{sub}/#{key}"] = obj[k].to_s
+        end
+    }
+    flat
+end
+
+def eflatten(obj)
+  flat = {}
+  if obj["dir"] then
+    if obj["nodes"] then
+      obj["nodes"].each { |n|
+        flat = flat.merge(eflatten(n))
+      }
+    end
+  else
+    key = obj["key"].gsub("/","_").gsub("-","_")
+    flat[key[1..key.length]]=obj["value"]
+  end
+  flat
+end
+
+def nodes2obj(nodes,prefix)
+    obj={}
+    nodes.each {|node|
+      if node['dir'] && node['nodes'] then
+        obj[node['key'].gsub(prefix,'')]=nodes2obj( node['nodes'],"#{ node['key'] }/" )
+      elsif node['value'] then
+        obj[node['key'].gsub(prefix,'')]=node['value']
+      end
+    }
+    obj
+end
+

metadata

@@ -0,0 +1,50 @@
+--- !ruby/object:Gem::Specification
+name: small-ops
+version: !ruby/object:Gem::Version
+  version: 0.0.1.beta
+  prerelease: 6
+platform: ruby
+authors:
+- Diogo Silva
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-07-09 00:00:00.000000000 Z
+dependencies: []
+description: Docker, etcd, confd...
+email: diogo@diogok.net
+executables:
+- docker2etcd
+- etcd2env
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/small-setup.rb
+- bin/docker2etcd
+- bin/etcd2env
+homepage: https://github.com/
+licenses:
+- MIT
+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: 1.3.1
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Small Docker ops utils
+test_files: []