niso 2.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 44a5eba8f95a0e95645da843a3f3f17c342bdc42
+  data.tar.gz: aa7b1186b90247e42dfc3af1086b8b554333fc6d
+SHA512:
+  metadata.gz: f3190d8deba79bb938f78fca3580a840a04906b83420819b023b75b90654487619b6cecc7e7b7ebbe9fd6eb4319297a80d19c72cd685e526248bd882f4bdccf5
+  data.tar.gz: 36244fbc3c8b5be59fd58afd3f4d82272b6a2584494f14e2101f85ef770b17180a1a842d5343ffb3f2d1cf491ac3572f72d6bdc80417dd0520ecdc378394b14b

data/.gitignore

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

data/.travis.yml

@@ -0,0 +1,3 @@
+language: ruby
+rvm:
+  - 2.4.0

data/CHANGELOG.md

@@ -0,0 +1,21 @@
+## 2.0, release 2017-03-28
+* updated to explicitly use digital ocean for `niso setup` and `niso teardown`
+* added `droplet_kit`
+* removed `digital_ocean`
+
+## 1.5, release 2013-09-27
+* `niso deploy [linode|digital_ocean] [name]` will deploy to the instance with additional attributes.
+
+## 1.4, release 2013-09-27
+* `niso teardown` no longer requires instance names as an argument, it lets you choose from a list instead.
+
+## Past Releases
+* v1.3: SSH config support. Thanks to @toooooooby
+* v1.2: Evaluate everything as ERB templates by default. Added "files" folder.
+* v1.1: "set -e" by default. apt-get everywhere in place of aptitude. Linode DNS support for DigitalOcean instances.
+* v1.0: System functions are refactored into niso.mute() and niso.install().
+* v0.9: Support for [DigitalOcean](https://www.digitalocean.com) setup / teardown.
+* v0.8: Added `--sudo` option to `niso deploy`.
+* v0.7: Added `erase_remote_folder` and `cache_remote_recipes` preferences for customized behavior.
+* v0.6: System function niso::silencer() added for succinct log messages.
+* v0.5: Role-based configuration supported. Reworked directory structure. **Incompatible with previous versions**.

data/Gemfile

@@ -0,0 +1,3 @@
+source 'http://rubygems.org'
+
+gemspec

data/README.md

@@ -0,0 +1,89 @@
+# nîso ᓃᓱ
+
+    niso is the number 2 (two) in Cree
+
+## Quickstart
+
+Install:
+
+    $ [sudo] gem install niso
+
+Go into your project directory (if it's a Rails project, `config` would be a good place to start with), then:
+
+    $ niso create
+
+It generates a `niso` folder along with subdirectories and templates. Inside `niso`, there are `niso.yml` and `install.sh`. Those two are the most important files that you mainly work on.
+
+Go into the `niso` directory, then run `niso deploy`:
+
+    $ cd niso
+    $ niso deploy example.com
+
+## Commands
+
+    $ niso                                           # Show command help
+    $ niso compile                                   # Compile niso project
+    $ niso create                                    # Create a new niso project
+    $ niso deploy [user@host:port] [role] [--sudo]   # Deploy niso project
+
+    $ niso setup                                     # Setup a new digital ocean droplet
+    $ niso teardown                                  # Teardown an existing digital ocean droplet
+
+## Directory structure
+
+Here's the directory structure that `niso create` automatically generates:
+
+    niso/
+      install.sh      # main script
+      niso.yml        # add custom attributes and remote recipes here
+
+      recipes/        # put commonly used scripts here, referred from install.sh
+        niso.sh
+      roles/          # when role is specified, scripts here will be concatenated
+        db.sh         # to install.sh in the compile phase
+        web.sh
+      files/          # put any files to be transferred
+      compiled/       # everything under this folder will be transferred to the
+                    # remote server (do not edit directly)
+
+## Dynamic values?
+
+There are two ways to pass dynamic values to the script - ruby and bash.
+
+Make sure `eval_erb: true` is set in `niso.yml`. In the compile phase, attributes defined in `niso.yml` are accessible from any files in the form of `<%= @attributes.ruby_version %>`.
+
+## Role-based configuration
+
+You probably have different configurations between **web servers** and **database servers**.
+
+No problem - how niso handles role-based configuration is refreshingly simple.
+
+Shell scripts under the `roles` directory, such as `web.sh` or `db.sh`, are automatically recognized as a role. The role script will be appended to `install.sh` at deploy, so you should put common configurations in `install.sh` and role specific procedures in the role script.
+
+For instance, when you set up a new web server, deploy with a role name:
+
+    niso deploy example.com web
+
+It is equivalent to running `install.sh`, followed by `web.sh`.
+
+### Requirements
+
+For `niso deploy`
+
+    $ [sudo] gem install droplet_kit
+    $ [sudo] gem install highline
+
+### DigitalOcean
+You can setup a new VM, or teardown an existing VM interactively. Use `niso setup` and `niso teardown` for that.
+
+[DigitalOcean](https://www.digitalocean.com) only.
+
+v2.0 of the DigitalOcean API is supported &mdash; get your [access token](https://cloud.digitalocean.com/api_access).
+
+### What's new:
+
+Please see the [CHANGELOG](https://github.com/dakotalightning/niso/blob/master/CHANGELOG.md).
+
+## Credits
+
+- This project is a hard fork of the proejct [niso](https://github.com/kenn/niso) by [kenn](https://github.com/kenn).

data/Rakefile

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

data/bin/niso

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+
+# Abort beautifully with ctrl+c.
+Signal.trap(:INT) { abort "\nAborting." }
+
+# Load the main lib and invoke CLI.
+require 'niso'
+Niso::Cli.start

data/lib/niso/cli.rb

@@ -0,0 +1,164 @@
+require 'open3'
+require 'ostruct'
+require 'net/ssh'
+
+module Niso
+  class Cli < Thor
+    include Thor::Actions
+
+    desc 'create', 'Create niso project'
+    def create(project = 'niso')
+      do_create(project)
+    end
+
+    desc 'deploy [user@host:port] [role] [--sudo] or deploy droplet [name] [role]  [--sudo]', 'Deploy niso project'
+    method_options :sudo => false
+    def deploy(first, *args)
+      do_deploy(first, *args)
+    end
+
+    desc 'compile', 'Compile niso project'
+    def compile(role = nil)
+      do_compile(role)
+    end
+
+    desc 'setup', 'Setup a new VM'
+    def setup(provider = "droplet_kit")
+      Niso::Cloud.new(self, provider).setup
+    end
+
+    desc 'teardown', 'Teardown an existing VM'
+    def teardown(provider = "droplet_kit")
+      Niso::Cloud.new(self, provider).teardown
+    end
+
+    desc 'version', 'Show version'
+    def version
+      puts Gem.loaded_specs['niso'].version.to_s
+    end
+
+    no_tasks do
+      include Niso::Utility
+
+      def self.source_root
+        File.expand_path('../../',__FILE__)
+      end
+
+      def do_create(project)
+        copy_file 'templates/create/.gitignore',         "#{project}/.gitignore"
+        copy_file 'templates/create/niso.yml',          "#{project}/niso.yml"
+        copy_file 'templates/create/install.sh',         "#{project}/install.sh"
+        copy_file 'templates/create/recipes/niso.sh',   "#{project}/recipes/niso.sh"
+        copy_file 'templates/create/roles/db.sh',        "#{project}/roles/db.sh"
+        copy_file 'templates/create/roles/web.sh',       "#{project}/roles/web.sh"
+        copy_file 'templates/create/files/.gitkeep',     "#{project}/files/.gitkeep"
+      end
+
+      def do_deploy(first, *args)
+        if ['droplet'].include?(first)
+          @instance_attributes = YAML.load(File.read("#{first}/instances/#{args[0]}.yml"))
+          target = @instance_attributes[:fqdn]
+          role = args[1]
+        else
+          target = first
+          role = args[0]
+        end
+
+        sudo = 'sudo ' if options.sudo?
+        user, host, port = parse_target(target)
+        endpoint = "#{user}@#{host}"
+
+        # compile attributes and recipes
+        do_compile(role)
+
+        # The host key might change when we instantiate a new VM, so
+        # we remove (-R) the old host key from known_hosts.
+        `ssh-keygen -R #{host} 2> /dev/null`
+
+        remote_commands = <<-EOS
+        rm -rf ~/niso &&
+        mkdir ~/niso &&
+        cd ~/niso &&
+        tar xz &&
+        #{sudo}bash install.sh
+        EOS
+
+        remote_commands.strip! << ' && rm -rf ~/niso' if @config['preferences'] and @config['preferences']['erase_remote_folder']
+
+        local_commands = <<-EOS
+        cd compiled
+        tar cz . | ssh -o 'StrictHostKeyChecking no' #{endpoint} -p #{port} '#{remote_commands}'
+        EOS
+
+        Open3.popen3(local_commands) do |stdin, stdout, stderr|
+          stdin.close
+          t = Thread.new do
+            while (line = stderr.gets)
+              print line.color(:red)
+            end
+          end
+          while (line = stdout.gets)
+            print line.color(:green)
+          end
+          t.join
+        end
+      end
+
+      def do_compile(role)
+        # Check if you're in the niso directory
+        abort_with 'You must be in the niso folder' unless File.exists?('niso.yml')
+        # Check if role exists
+        abort_with "#{role} doesn't exist!" if role and !File.exists?("roles/#{role}.sh")
+
+        # Load niso.yml
+        @config = YAML.load(File.read('niso.yml'))
+
+        # Merge instance attributes
+        @config['attributes'] ||= {}
+        @config['attributes'].update(Hash[@instance_attributes.map{|k,v| [k.to_s, v] }]) if @instance_attributes
+
+        # Break down attributes into individual files
+        (@config['attributes'] || {}).each {|key, value| create_file "compiled/attributes/#{key}", value }
+
+        # Retrieve remote recipes via HTTP
+        cache_remote_recipes = @config['preferences'] && @config['preferences']['cache_remote_recipes']
+        (@config['recipes'] || []).each do |key, value|
+          next if cache_remote_recipes and File.exists?("compiled/recipes/#{key}.sh")
+          get value, "compiled/recipes/#{key}.sh"
+        end
+
+        copy_or_template = (@config['preferences'] && @config['preferences']['eval_erb']) ? :template : :copy_file
+        copy_local_files(@config, copy_or_template)
+
+        # Build install.sh
+        if role
+          if copy_or_template == :template
+            template File.expand_path('install.sh'), 'compiled/_install.sh'
+            create_file 'compiled/install.sh', File.binread('compiled/_install.sh') << "\n" << File.binread("compiled/roles/#{role}.sh")
+          else
+            create_file 'compiled/install.sh', File.binread('install.sh') << "\n" << File.binread("roles/#{role}.sh")
+          end
+        else
+          send copy_or_template, File.expand_path('install.sh'), 'compiled/install.sh'
+        end
+      end
+
+      def parse_target(target)
+        target.match(/(.*@)?(.*?)(:.*)?$/)
+        # Load ssh config if it exists
+        config = Net::SSH::Config.for($2)
+        [ ($1 && $1.delete('@') || config[:user] || 'root'),
+          config[:host_name] || $2,
+          ($3 && $3.delete(':') || config[:port] && config[:port].to_s || '22') ]
+      end
+
+      def copy_local_files(config, copy_or_template)
+        @attributes = OpenStruct.new(config['attributes'])
+        files = Dir['{recipes,roles,files}/**/*'].select { |file| File.file?(file) }
+        files.each { |file| send copy_or_template, File.expand_path(file), File.expand_path("compiled/#{file}") }
+
+        (config['files'] || []).each {|file| send copy_or_template, File.expand_path(file), "compiled/files/#{File.basename(file)}" }
+      end
+    end
+  end
+end

data/lib/niso/cloud/base.rb

@@ -0,0 +1,103 @@
+Niso::Dependency.load('highline')
+
+module Niso
+  class Cloud
+    class Base
+      include Niso::Utility
+
+      def initialize(cli, provider)
+        @provider = provider
+        @cli = cli
+        @ui = HighLine.new
+      end
+
+      def setup
+        unless File.exist? provider_config_path
+          @cli.empty_directory "#{@provider}/instances"
+          @cli.template "templates/setup/#{@provider}.yml", provider_config_path
+          exit_with "Now go ahead and edit #{@provider}.yml, then run this command again!"
+        end
+
+        # get the config { provider }.yml
+        assign_config
+
+        if @config['name'] == 'example-droplet-01'
+          abort_with "You must have your own settings in #{@provider}.yml"
+        end
+
+        if @config['access_token'] == 'your_api_key'
+          abort_with "You must add a valid access_token in #{@provider}.yml"
+        end
+
+        # Ask environment and hostname
+        @env = ask_menu("environment?", @config['environments'])
+
+        @name = "#{@config['name']}-#{@env}"
+        abort_with "#{@name} already exists!" if instance_config_path.exist?
+
+        # get the client @client
+        setup_client
+
+        @attributes = {}
+        do_setup
+
+        # Save instance info
+        @cli.create_file instance_config_path, YAML.dump(@instance)
+      end
+
+      def teardown
+        names = Dir.glob("#{@provider}/instances/*.yml").map{|i| i.split('/').last.sub('.yml','') }
+        abort_with "No match found with #{@provider}/instances/*.yml" if names.empty?
+
+        @name = ask_menu("which instance?: ", names)
+
+        assign_config
+
+        @instance = YAML.load(instance_config_path.read)
+
+        # Are you sure?
+        moveon = ask("Are you sure about deleting #{@instance[:name]} permanently? (y/n) ", String) {|q| q.in = ['y','n']}
+        exit unless moveon == 'y'
+
+        # Run Linode / DigitalOcean specific tasks
+        setup_client
+        do_teardown
+
+        # Remove the instance config file
+        @cli.remove_file instance_config_path
+
+        say 'Done.'
+      end
+
+      def assign_config
+        @config = YAML.load(provider_config_path.read)
+      end
+
+      def provider_config_path
+        Pathname.new "#{@provider}/#{@provider}.yml"
+      end
+
+      def instance_config_path
+        Pathname.new "#{@provider}/instances/#{@name}.yml"
+      end
+
+      def ask(question, answer_type, &details)
+        @ui.ask(@ui.color(question, :green, :bold), answer_type, &details)
+      end
+
+      def ask_menu(question, choices)
+        answer = @ui.choose do |menu|
+          menu.prompt = "#{@ui.color(question, :green, :bold)} "
+          choices.each { |n| menu.choice(n) }
+        end
+        say("=> #{answer}")
+        answer
+      end
+
+      def proceed?
+        moveon = ask("Are you ready to go ahead and create #{@name}? (y/n) ", String) {|q| q.in = ['y','n']}
+        exit unless moveon == 'y'
+      end
+    end
+  end
+end

data/lib/niso/cloud/droplet_kit.rb

@@ -0,0 +1,92 @@
+Niso::Dependency.load('droplet_kit')
+
+module Niso
+  class Cloud
+    class DropletKit < Base
+      def do_setup
+
+        if @config["size"]
+          # Choose size
+          @attributes[:size] = @config["size"]
+          say "using config size"
+        else
+          choose(:size, @client.sizes.all())
+        end
+
+        if @config["region"]
+          # Choose region
+          @attributes[:region] = @config["region"]
+          say "using config region"
+        else
+          choose(:region, @client.regions.all())
+        end
+
+        # Choose an image
+        if @config['images_filter']
+          result = @client.images.all(type: @config['images_filter'])
+          result = result.select{|i| i.distribution.match Regexp.new(@config['images_filter'], Regexp::IGNORECASE) }
+        end
+        choose(:image, result)
+
+        # Go ahead?
+        moveon = ask("Are you ready to go ahead and create #{@attributes}? (y/n) ", ['y','n'])
+        exit unless moveon == 'y'
+
+        ssh_keys = @client.ssh_keys.all.collect { |key| key.fingerprint }
+
+        # Create
+        say "creating a new droplet..."
+        droplet = ::DropletKit::Droplet.new(name: @config["name"], region: @attributes[:region_slug], image: @attributes[:image_slug], size: @attributes[:size_slug], ssh_keys: ssh_keys)
+        result = @client.droplets.create(droplet)
+
+        @droplet_id = result.id
+        say "Created a new droplet (id: #{@droplet_id}). Booting..."
+
+        # Boot - we need this before getting public IP
+        while @client.droplets.find(id: @droplet_id).status.downcase != 'active'
+          sleep 3
+        end
+
+        @networks = @client.droplets.find(id: @droplet_id).networks
+        say "Done. networks = #{@networks.ip_address}"
+
+        @instance = {
+          :droplet_id => @droplet_id,
+          :env  => @env,
+          :host => @host,
+          :name => @name,
+          :networks => @networks,
+          :size_slug      => @attributes[:size_slug],
+          :region_slug    => @attributes[:region_slug],
+          :image_slug     => @attributes[:image_slug],
+        }
+      end
+
+      def choose(key, result)
+        abort "no #{key} found!" if result.first.nil?
+        # result.each{|i| say "#{i.slug}" }
+        choices = result.map(&:slug).compact
+        @attributes[:"#{key}_slug"] = ask_menu("which #{key}?: ", choices)
+      end
+
+      def do_teardown
+        say 'deleting droplet...'
+        @client.droplets.delete(id: @instance[:droplet_id])
+      end
+
+      # Set the client for DropletKit
+      #
+      # DropletKit::DropletResource
+      def setup_client
+        begin
+          @client = ::DropletKit::Client.new(access_token: @config['access_token'])
+          info = @client.account.info()
+          moveon = ask(" ... continue with #{info.email} account? (y/n) ", String) {|q| q.in = ['y','n']}
+          exit unless moveon == 'y'
+        rescue ::DropletKit::Error => e
+          abort_with e.message
+        end
+      end
+    end
+  end
+end

data/lib/niso/cloud.rb

@@ -0,0 +1,22 @@
+module Niso
+  class Cloud
+    include Niso::Utility
+
+    def initialize(cli, provider)
+      @subject = case provider
+      when 'droplet_kit'
+        Niso::Cloud::DropletKit.new(cli, provider)
+      else
+        abort_with "Provider #{provider} is not valid!"
+      end
+    end
+
+    def method_missing(sym, *args, &block)
+      @subject.send sym, *args, &block
+    end
+
+    def respond_to?(method)
+      @subject.respond_to?(sym) || super
+    end
+  end
+end

data/lib/niso/dependency.rb

@@ -0,0 +1,38 @@
+module Niso
+  class Dependency
+    def self.all
+      {
+        'highline' => { :require => 'highline', :version => '>= 1.7.8'},
+        'droplet_kit' =>  { :require => 'droplet_kit',  :version => '>= 2.1.0' },
+      }
+    end
+
+    def self.load(name)
+      begin
+        gem(name, all[name][:version])
+        require(all[name][:require])
+      rescue LoadError
+        if $!.to_s =~ /Gemfile/
+          Logger.error <<-EOS
+Dependency missing: #{name}
+Add this line to your application's Gemfile.
+
+    gem '#{name}', '#{all[name][:version]}'
+
+Please try again after running "bundle install".
+          EOS
+        else
+          Logger.error <<-EOS
+Dependency missing: #{name}
+To install the gem, issue the following command:
+
+    gem install #{name} -v '#{all[name][:version]}'
+
+Please try again after installing the missing dependency.
+          EOS
+        end
+        abort
+      end
+    end
+  end
+end

data/lib/niso/logger.rb

@@ -0,0 +1,17 @@
+module Niso
+  module Logger
+    class << self
+      def info(text)
+        puts text.bright
+      end
+
+      def success(text)
+        puts text.color(:green).bright
+      end
+
+      def error(text)
+        puts text.color(:red).bright
+      end
+    end
+  end
+end

data/lib/niso/utility.rb

@@ -0,0 +1,17 @@
+module Niso
+  module Utility
+    def abort_with(text)
+      Logger.error text
+      abort
+    end
+
+    def exit_with(text)
+      Logger.success text
+      exit
+    end
+
+    def say(text)
+      Logger.info text
+    end
+  end
+end

data/lib/niso.rb

@@ -0,0 +1,20 @@
+require 'thor'
+require 'rainbow'
+require 'yaml'
+
+# Starting 2.0.0, Rainbow no longer patches string with the color method by default.
+require 'rainbow/version'
+require 'rainbow/ext/string' unless Rainbow::VERSION < '2.0.0'
+
+module Niso
+  autoload :Cli,        'niso/cli'
+  autoload :Cloud,      'niso/cloud'
+  autoload :Dependency, 'niso/dependency'
+  autoload :Logger,     'niso/logger'
+  autoload :Utility,    'niso/utility'
+
+  class Cloud
+    autoload :Base,         'niso/cloud/base'
+    autoload :DropletKit,   'niso/cloud/droplet_kit'
+  end
+end

data/lib/templates/create/.gitignore

@@ -0,0 +1 @@
+/compiled

data/lib/templates/create/install.sh

@@ -0,0 +1,50 @@
+#!/bin/bash
+set -e
+
+# Load base utility functions like niso.mute() and niso.install()
+source recipes/.sh
+
+# This line is necessary for automated provisioning for Debian/Ubuntu.
+# Remove if you're not on Debian/Ubuntu.
+export DEBIAN_FRONTEND=noninteractive
+
+# Add Dotdeb repository. Recommended if you're using Debian. See http://www.dotdeb.org/about/
+# source recipes/dotdeb.sh
+# source recipes/backports.sh
+
+# Update apt catalog and upgrade installed packages
+niso.mute "apt-get update"
+niso.mute "apt-get -y upgrade"
+
+# Install packages
+apt-get -y install git-core ntp curl
+
+# Install sysstat, then configure if this is a new install.
+if niso.install "sysstat"; then
+  sed -i 's/ENABLED="false"/ENABLED="true"/' /etc/default/sysstat
+  /etc/init.d/sysstat restart
+fi
+
+# Set RAILS_ENV
+environment=<%= @attributes.environment %>
+
+if ! grep -Fq "RAILS_ENV" ~/.bash_profile; then
+  echo 'Setting up RAILS_ENV...'
+  echo "export RAILS_ENV=$environment" >> ~/.bash_profile
+  source ~/.bash_profile
+fi
+
+# Install Ruby using RVM
+source recipes/rvm.sh
+ruby_version=<%= @attributes.ruby_version %>
+
+if [[ "$(which ruby)" != /usr/local/rvm/rubies/ruby-$ruby_version* ]]; then
+  echo "Installing ruby-$ruby_version"
+  # Install dependencies using RVM autolibs - see https://blog.engineyard.com/2013/rvm-ruby-2-0
+  rvm install --autolibs=3 $ruby_version
+  rvm $ruby_version --default
+  echo 'gem: --no-ri --no-rdoc' > ~/.gemrc
+
+  # Install Bundler
+  gem install bundler
+fi

data/lib/templates/create/niso.yml

@@ -0,0 +1,30 @@
+---
+# Dynamic variables here will be compiled to individual files in compiled/attributes.
+attributes:
+  environment: production
+  ruby_version: 2.0.0
+
+# Remote recipes here will be downloaded to compiled/recipes.
+recipes:
+  rvm: https://raw.githubusercontent.com/kenn/niso-recipes/master/ruby/rvm.sh
+  # dotdeb: https://raw.githubusercontent.com/kenn/niso-recipes/master/debian/dotdeb-wheezy.sh
+  # backports: https://raw.githubusercontent.com/kenn/niso-recipes/master/debian/backports-wheezy.sh
+  # mongodb-10gen: https://raw.githubusercontent.com/kenn/niso-recipes/master/debian/mongodb-10gen.sh
+
+# Files specified here will be copied to compiled/files.
+# files:
+#   - ~/.ssh/id_rsa.pub
+
+# Fine tune how Niso should work.
+preferences:
+  # Erase the generated folder on the server after deploy.
+  # Turn on when you are done with testing and ready for production use.
+  erase_remote_folder: true
+
+  # Skip retrieving remote recipes when local copies already exist. This setting helps
+  # iterative deploy testing considerably faster, when you have a lot of remote recipes.
+  cache_remote_recipes: false
+
+  # Evaluate files as ERB templates. When enabled, you can pass dynamic values in the form
+  # of <%= @attributes.environment %> in recipes, roles, files and install.sh.
+  eval_erb: true

data/lib/templates/create/recipes/niso.sh

@@ -0,0 +1,47 @@
+# This file is used to define functions under the niso.* namespace.
+
+# Set $niso_pkg to "apt-get" or "yum", or abort.
+#
+if which apt-get >/dev/null 2>&1; then
+  export niso_pkg=apt-get
+elif which yum >/dev/null 2>&1; then
+  export niso_pkg=yum
+fi
+
+if [ "$niso_pkg" = '' ]; then
+  echo 'niso only supports apt-get or yum!' >&2
+  exit 1
+fi
+
+# Mute STDOUT and STDERR
+#
+function niso.mute() {
+  echo "Running \"$@\""
+  `$@ >/dev/null 2>&1`
+  return $?
+}
+
+# Installer
+#
+function niso.installed() {
+  if [ "$niso_pkg" = 'apt-get' ]; then
+    dpkg -s $@ >/dev/null 2>&1
+  elif [ "$niso_pkg" = 'yum' ]; then
+    rpm -qa | grep $@ >/dev/null
+  fi
+  return $?
+}
+
+# When there's "set -e" in install.sh, niso.install should be used with if statement,
+# otherwise the script may exit unexpectedly when the package is already installed.
+#
+function niso.install() {
+  if niso.installed "$@"; then
+    echo "$@ already installed"
+    return 1
+  else
+    echo "No packages found matching $@. Installing..."
+    niso.mute "$niso_pkg -y install $@"
+    return 0
+  fi
+}

data/lib/templates/create/roles/db.sh

@@ -0,0 +1,5 @@
+# Install Database Server
+
+# source recipes/mongodb-10gen.sh           # MongoDB
+# apt-get -q -y install mysql-server-5.5    # MySQL 5.5 - You may need to enable Dotdeb in install.sh first
+# apt-get -y install redis-server           # Redis - You may need to enable Dotdeb in install.sh first

data/lib/templates/create/roles/web.sh

@@ -0,0 +1,4 @@
+# Install Web server
+
+apt-get -y install nginx        # Nginx
+# apt-get -y install apache2    # Apache

data/lib/templates/setup/droplet_kit.yml

@@ -0,0 +1,15 @@
+---
+access_token: your_api_key
+
+name: example-droplet-01
+# region: sf01
+# image: ubuntu-14-04-x64
+# size: 512mb
+
+# filter out large lists by keyword
+images_filter: ubuntu
+
+# add / remove environments
+environments:
+  - production
+  - staging

data/niso.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+
+Gem::Specification.new do |spec|
+  spec.name          = 'niso'
+  spec.version       = '2.0.3'
+  spec.authors       = ['Dakota Lightning']
+  spec.email         = ['im@koda.io']
+  spec.homepage      = 'http://github.com/dakotalightning/niso'
+  spec.summary       = %q{Server provisioning utility}
+  spec.description   = %q{Server provisioning utility}
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files`.split($/)
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+
+  spec.add_runtime_dependency 'thor'
+  spec.add_runtime_dependency 'rainbow'
+  spec.add_runtime_dependency 'net-ssh'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'minitest'
+end

data/test/test_cli.rb

@@ -0,0 +1,52 @@
+require 'niso'
+require 'minitest/autorun'
+
+class TestCli < Minitest::Test
+  def setup
+    @cli = Niso::Cli.new
+  end
+
+  def test_parse_target
+    assert_equal ['user', 'example.com', '2222'], @cli.parse_target('user@example.com:2222')
+    assert_equal ['root', 'example.com', '2222'], @cli.parse_target('example.com:2222')
+    assert_equal ['user', 'example.com', '22'],   @cli.parse_target('user@example.com')
+    assert_equal ['root', 'example.com', '22'],   @cli.parse_target('example.com')
+    assert_equal ['root', '192.168.0.1', '22'],   @cli.parse_target('192.168.0.1')
+  end
+
+  def test_parse_target_with_ssh_config
+    ssh_config = lambda do |host|
+      if host == 'example.com'
+        { :host_name => "buzz.example.com", :user => "foobar", :port => 2222 }
+      else
+        {}
+      end
+    end
+
+    Net::SSH::Config.stub(:for, ssh_config) do
+      assert_equal ['foobar', 'buzz.example.com', '2222'], @cli.parse_target('example.com')
+      assert_equal ['foobar', 'buzz.example.com', '8080'], @cli.parse_target('example.com:8080')
+      assert_equal ['piyo', 'buzz.example.com', '2222'], @cli.parse_target('piyo@example.com')
+      assert_equal ['piyo', 'buzz.example.com', '8080'], @cli.parse_target('piyo@example.com:8080')
+      assert_equal ['root', '192.168.0.1', '22'], @cli.parse_target('192.168.0.1')
+    end
+  end
+
+  def test_create
+    @cli.create 'sandbox'
+    assert File.exist?('sandbox/niso.yml')
+    FileUtils.rm_rf 'sandbox'
+  end
+
+  def test_copy_local_files
+    pwd = Dir.pwd
+    Dir.chdir('test/niso_test_dir')
+
+    @cli.copy_local_files({}, :copy_file)
+    assert File.exists?('compiled/files/nginx/nginx.conf')
+    assert File.exists?('compiled/recipes/nginx.sh')
+    FileUtils.rm_rf 'compiled'
+
+    Dir.chdir(pwd)
+  end
+end

metadata

@@ -0,0 +1,145 @@
+--- !ruby/object:Gem::Specification
+name: niso
+version: !ruby/object:Gem::Version
+  version: 2.0.3
+platform: ruby
+authors:
+- Dakota Lightning
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-03-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rainbow
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !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: minitest
+  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'
+description: Server provisioning utility
+email:
+- im@koda.io
+executables:
+- niso
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- CHANGELOG.md
+- Gemfile
+- README.md
+- Rakefile
+- bin/niso
+- lib/niso.rb
+- lib/niso/cli.rb
+- lib/niso/cloud.rb
+- lib/niso/cloud/base.rb
+- lib/niso/cloud/droplet_kit.rb
+- lib/niso/dependency.rb
+- lib/niso/logger.rb
+- lib/niso/utility.rb
+- lib/templates/create/.gitignore
+- lib/templates/create/files/.gitkeep
+- lib/templates/create/install.sh
+- lib/templates/create/niso.yml
+- lib/templates/create/recipes/niso.sh
+- lib/templates/create/roles/db.sh
+- lib/templates/create/roles/web.sh
+- lib/templates/setup/droplet_kit.yml
+- niso.gemspec
+- test/niso_test_dir/files/nginx/nginx.conf
+- test/niso_test_dir/recipes/nginx.sh
+- test/test_cli.rb
+homepage: http://github.com/dakotalightning/niso
+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.5.1
+signing_key: 
+specification_version: 4
+summary: Server provisioning utility
+test_files:
+- test/niso_test_dir/files/nginx/nginx.conf
+- test/niso_test_dir/recipes/nginx.sh
+- test/test_cli.rb