skirnir 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a0e1b499c4beb1871eca335e45ce3d84a596da80
+  data.tar.gz: 55d96459c0281137094fba9155ebc272ba0ebcce
+SHA512:
+  metadata.gz: 1f1b164087f21715af27d1b4a34a6c990662a66a4ce11b1a90c0f730c49d21f66fedf678affd7495de3c671bdc3f626426176ab30b57cfff2c845052fb91c0df
+  data.tar.gz: d8236dca09bc58d2e14673e6a3cc8a951a6ce868cedf9a200cadaeb9a85748a744002987229fa532cbe2f96cce0e0bbe07c07035a966d8574196fba9028b6d29

data/.gitignore

@@ -0,0 +1 @@
+.idea

data/Gemfile

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

data/Gemfile.lock

@@ -0,0 +1,19 @@
+PATH
+  remote: .
+  specs:
+    skirnir (0.0.2)
+      deep_merge (~> 1.0.1)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    deep_merge (1.0.1)
+    rake (10.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.6)
+  rake
+  skirnir!

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 irontoby
+
+MIT License
+
+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/README.md

@@ -0,0 +1,29 @@
+# Skirnir
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'skirnir'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install skirnir
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it ( https://github.com/[my-github-username]/skirnir/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+

data/bin/ski

@@ -0,0 +1,72 @@
+#!/usr/bin/env ruby
+
+require 'skirnir'
+require 'pp'
+
+def show_box_info node, env
+  #file = File.expand_path('config.yml')
+
+  info = Skirnir::Site.new
+
+  if node.nil?
+    info.read_default
+  else
+    info.read(env)
+  end
+
+  pp info.chef_hash_for_node_name node
+end
+
+def list_boxes(env='all')
+  info = Skirnir::Site.new
+
+  if env.nil? || env == '' || env == 'all'
+    info.silent = true
+
+    info.all_environments.each do |env|
+      list_nodes info, env
+      puts "\n"
+    end
+  else
+    list_nodes info, env
+  end
+
+end
+
+def list_nodes(info, env)
+  info.read env
+  puts "Nodes in #{env}:"
+  info.nodes.map {|node| puts format("  %8s %s\n", node.name, node.ip_int)}
+end
+
+def give_help
+  puts <<-eos
+ski [command] [node] [env]
+
+AVAILABLE COMMANDS:
+  show <node> [env]   Outputs Chef/node JSON
+  list [env]          List node names, or nodes for env
+  help                Print this help message.
+  eos
+
+  exit
+end
+
+command, node, env = ARGF.argv
+
+give_help if command.nil?
+
+case command
+  when 'help'
+    give_help
+  when 'show'
+    show_box_info node, env
+  when 'list'
+    env = node
+    list_boxes env
+  else
+    give_help
+end
+
+exit
+

data/lib/skirnir.rb

@@ -0,0 +1,4 @@
+require 'skirnir/version'
+require 'skirnir/site'
+require 'skirnir/node'
+require 'skirnir/chef'

data/lib/skirnir/chef.rb

@@ -0,0 +1,31 @@
+class Skirnir::Chef
+  def initialize(defs)
+    @defs = defs
+  end
+
+  def set(role, *path, value)
+    #where = path.join('/')
+    #puts "Set for role #{role}: #{where} = #{value}"
+
+    h = Hash.construct(path, value)
+    @defs[role] = {} unless @defs.has_key?(role)
+
+    @defs[role].deep_merge!(h)
+  end
+
+  # @param roles [Array]
+  def get_for_roles(*roles)
+    roles.flatten!
+
+    roles.unshift('all') unless roles.include?('all')
+
+    result = {}
+
+    roles.each do |role|
+      result.deep_merge!(@defs[role]) if @defs.has_key?(role)
+    end
+
+    result
+  end
+end
+

data/lib/skirnir/node.rb

@@ -0,0 +1,42 @@
+class Skirnir::Node
+  attr_accessor :name, :roles, :chef, :run_list, :ip_int, :ip_ext
+
+  def initialize(site_name, config)
+    @name = config['name']
+    @roles = config['roles']
+    @run_list = @roles.map{|role| "recipe[#{site_name}::#{role}_box]"}
+    @ip_int = config['ip_int']
+    @ip_ext = config['ip_ext'] || @ip_int
+
+    chef_config = config['chef'] || {}
+
+    @chef = Skirnir::Chef.new({'all' => chef_config})
+
+    to_h.each_pair do |key, value|
+      @chef.set('all', 'box', key, value)
+    end
+
+  end
+
+  def has_role?(role)
+    @roles.include? role
+  end
+
+  # @return [Hash]
+  def to_h
+    {
+      'name' => @name,
+      'roles' => @roles,
+      'ip_int' => @ip_int,
+      'ip_ext' => @ip_ext
+    }
+  end
+
+  # returns the chef definitions specific for this node
+  def chef_hash
+    @chef.get_for_roles('all')
+  end
+
+end
+
+

data/lib/skirnir/site.rb

@@ -0,0 +1,163 @@
+require 'yaml'
+require 'deep_merge'
+
+DEFAULT_FILENAME = 'config.yml'
+LOCAL_FILENAME = 'local.yml'
+
+class Skirnir::Site
+  attr_reader :info, :nodes, :chef, :vagrant, :config
+  attr_accessor :silent
+
+  def initialize
+    @silent = false
+  end
+
+  def read_default
+    env=get_env_from_local_section('vagrant')
+
+    read(env, 'config.yml')
+  end
+
+  def read_for_update
+    env = get_env_from_local_section('update')
+
+    read(env, 'config.yml')
+  end
+
+  def read(env, filename = DEFAULT_FILENAME)
+    all = get_config filename
+    puts "- Section: #{env}\n\n" unless @silent
+
+    config = all['common'].deep_merge!(all[env])
+
+    if File.exists?(LOCAL_FILENAME)
+      local = YAML.load(IO.read(LOCAL_FILENAME))
+      config.deep_merge!(local)
+    end
+
+    @config = config
+    @info = config['info']
+    @vagrant = config['vagrant']
+
+    @chef = Skirnir::Chef.new(config['chef'])
+    @nodes = config['nodes'].map {|node| Skirnir::Node.new(site_name, node)}
+
+    setup_role_chefs
+
+    @nodes
+  end
+
+  # @return [Array<Node>]
+  def nodes_with_role(role)
+    @nodes.select {|node| node.has_role? role}
+  end
+
+  def chef_hash_for_node_name(name)
+    node = (@nodes.select {|node| node.name == name}).first
+
+    raise "Unknown node name ''#{name}''" if node.nil?
+
+    chef_hash_for_node(node)
+  end
+
+  # @param node [Node]
+  def chef_hash_for_node(node)
+    result = chef.get_for_roles(node.roles)
+    result.deep_merge!(node.chef_hash)
+
+    nodes_info = @nodes.map {|n| n.to_h}
+    nodes_hash = Hash.construct(%w(site nodes), nodes_info)
+    result.deep_merge!(nodes_hash)
+
+    result
+  end
+
+  def chef_data_for_all_nodes
+    result = {}
+
+    @nodes.each do |node|
+      name = node.name
+
+      name += @info['node_suffix'] unless @info['node_suffix'].empty?
+
+      attribs = chef_hash_for_node(node)
+      run_list = node.run_list
+
+      result[name] = { 'attributes' => attribs, 'run_list' => run_list}
+    end
+
+    result
+  end
+
+  def site_name
+    @info['site_name']
+  end
+
+  def all_environments(filename = DEFAULT_FILENAME)
+    all = get_config
+
+    all.keys.select {|x| x != 'common'}
+  end
+
+private
+  def get_config(filename = DEFAULT_FILENAME)
+    puts "Reading config from #{filename}\n" unless @silent
+    YAML.load(IO.read(filename))
+  end
+
+  def get_env_from_local_section(section_name)
+    raise "Can't determine default environment; #{LOCAL_FILENAME} missing" unless File.exists?(LOCAL_FILENAME)
+
+    local = YAML.load(IO.read(LOCAL_FILENAME))
+
+    unless local.has_key?(section_name) && local[section_name].has_key?('env')
+      raise "Can't determine default environment; YAML #{section_name}/env entry missing"
+    end
+
+    local[section_name]['env']
+  end
+
+  def setup_role_chefs
+    setup_chef_db_host
+    setup_chef_nginx
+    setup_chef_app_should_migrate
+  end
+
+  def setup_chef_db_host
+    db_nodes = nodes_with_role('db')
+
+    raise "Expected single node with role 'db'; found #{db_nodes.count}" if db_nodes.count != 1
+
+    db_ip = db_nodes[0].ip_int
+    app_nodes = nodes_with_role('app')
+
+    is_local = app_nodes.all? {|node| node.ip_int == db_ip}
+
+    @chef.set('all', %w(site db host), is_local ? 'localhost' : db_ip)
+    @chef.set('all', %w(site db is_local), is_local)
+    @chef.set('db', %w(site db bind_address), is_local ? '127.0.0.1' : db_ip)
+  end
+
+  def setup_chef_nginx
+    # TODO this is no good for staging, need is_local for web too
+    web_ips = nodes_with_role('web').map{|node| node.ip_int}.map{|ip| "#{ip}/32"}
+
+    @chef.set('app', ['nginx'], { 'authorized_ips' => web_ips, 'realip' => { 'addresses' => web_ips}})
+  end
+
+  def setup_chef_app_should_migrate
+    should_migrate = true
+
+    nodes_with_role('app').each do |node|
+      node.chef.set('all', %w(site app should_migrate), should_migrate)
+      should_migrate = false
+    end
+  end
+
+end
+
+class Hash
+  def self.construct(*path, value)
+    path.flatten.reverse.inject(value) {|a, n| { n => a}}
+  end
+end

data/lib/skirnir/version.rb

@@ -0,0 +1,3 @@
+module Skirnir
+  VERSION = '0.0.2'
+end

data/skirnir.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'skirnir/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'skirnir'
+  spec.version       = Skirnir::VERSION
+  spec.date          = '2014-06-16'
+  spec.authors       = ['Toby Johnson']
+  spec.email         = ['irontoby@gmail.com']
+  spec.summary       = 'Tools for working with Chef, Vagrant and Digital Ocean'
+  spec.description   = 'Coming soon...'
+  spec.homepage      = 'http://rubygems.org/gems/skirnir'
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ['lib']
+
+  spec.add_development_dependency 'bundler', '~> 1.6'
+  spec.add_development_dependency 'rake'
+
+  spec.add_runtime_dependency 'deep_merge', '~> 1.0.1'
+end

metadata

@@ -0,0 +1,101 @@
+--- !ruby/object:Gem::Specification
+name: skirnir
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Toby Johnson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-06-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: deep_merge
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.1
+description: Coming soon...
+email:
+- irontoby@gmail.com
+executables:
+- ski
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/ski
+- lib/skirnir.rb
+- lib/skirnir/chef.rb
+- lib/skirnir/node.rb
+- lib/skirnir/site.rb
+- lib/skirnir/version.rb
+- pkg/skirnir-0.0.2.gem
+- skirnir.gemspec
+homepage: http://rubygems.org/gems/skirnir
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Tools for working with Chef, Vagrant and Digital Ocean
+test_files: []