vscripts 0.0.1

data/lib/vscripts/util/local_system.rb

@@ -0,0 +1,91 @@
+require 'fileutils'
+require 'yaml'
+require 'English'
+
+module VScripts
+  module Util
+    # Local system functions library
+    module LocalSystem
+      # Hosts file path
+      def hosts_path
+        '/etc/hosts'
+      end
+
+      # Hostname file path
+      def hostname_path
+        '/etc/hostname'
+      end
+
+      # Returns the current fully qualified domain
+      def local_fqdn
+        `hostname -f`.strip
+      end
+
+      # Returns the local host name
+      def local_host_name
+        `hostname`.strip
+      end
+
+      # Returns the local domain name
+      def local_domain_name
+        `dnsdomainname`.strip
+      end
+
+      # Tries to get the reverse dns
+      def external_dns
+        ext_ip = `wget -q -O - checkip.dyndns.org \
+          | sed -e 's/[^[:digit:]|.]//g'`
+        `dig +short -x #{ext_ip}`.strip
+      rescue
+        false
+      end
+
+      # Creates the directory for the specified file
+      def ensure_file_dir(file)
+        path = File.dirname(file)
+        `mkdir -p #{path}`
+      end
+
+      # Writes to file
+      def write_file(file, body)
+        ensure_file_dir(file)
+        File.open(file, 'w') do |newfile|
+          newfile.write(body)
+        end
+      rescue Errno::EACCES
+        puts "FATAL: You need to be root in order to write to #{file}"
+      end
+
+      # Ensures the specified file has the specified content
+      def ensure_file_content(file, body)
+        write = write_file(file, body)
+        read  = IO.read(file)
+        read == body || write
+      rescue
+        write
+      end
+
+      # Gets system checks from environment variables
+      def checks
+        config['SystemChecks']
+      rescue
+        {}
+      end
+
+      # Runs each command specified and returns the name and exit status
+      def process_checks
+        codes = {}
+        checks.each do |name, command|
+          system("#{command} > /dev/null 2>&1")
+          codes[name] = $CHILD_STATUS.exitstatus
+        end
+        codes
+      end
+
+      # Extract the codes for each check
+      def status_codes
+        process_checks.values.compact
+      end
+    end # module LocalSystem
+  end # module Util
+end # module VScripts

data/lib/vscripts/version.rb

@@ -0,0 +1,6 @@
+# Version number
+module VScripts
+  VERSION = File.read(File.expand_path('../../../VERSION', __FILE__)).strip
+  VERSION_MAJOR, VERSION_MINOR, VERSION_PATCH = VERSION.split('.')
+  VERSION_C = "VScripts #{VERSION} (c) #{Time.new.year} Vlad Ghinea"
+end

data/spec/spec_helper.rb

@@ -0,0 +1,2 @@
+class DummyClass
+end

data/spec/vscripts/aws/ec2_spec.rb

@@ -0,0 +1,81 @@
+require 'spec_helper'
+require 'vscripts/aws/ec2'
+require 'ostruct'
+
+describe VScripts::AWS::EC2 do
+  before :each do
+    ::AWS.config({
+      :access_key_id => '1234',
+      :secret_access_key => '5678',
+      :region => 'us-east-1',
+      :logger => nil,
+      :stub_requests => true
+    })
+
+    @tags = {'Name' => 'test'}
+
+    @inst1 = OpenStruct.new(id: '1111', status: :running)
+    @inst2 = OpenStruct.new(id: '2222', status: :running, tags: @tags)
+    @inst3 = OpenStruct.new(id: '3333', status: :terminated)
+
+    @ec2 = VScripts::AWS::EC2
+    @ec2.any_instance.stub(:check_instance) { true }
+    @ec2.any_instance.stub(:instance_id) { @inst1.id }
+  end
+
+  describe '#instance' do
+    it 'returns AWS::EC2' do
+      expect(subject.ec2).to be_an_instance_of ::AWS::EC2
+    end
+  end
+
+  describe '#new' do
+    it 'returns AWS::EC2::Instance' do
+      expect(subject.instance).to be_an_instance_of ::AWS::EC2::Instance
+      expect(subject.instance.id).to eq(@inst1.id)
+    end
+  end
+
+  describe '#all_tags' do
+    it 'returns AWS::EC2::ResourceTagCollection' do
+      expect(subject.all_tags)
+        .to be_an_instance_of ::AWS::EC2::ResourceTagCollection
+    end
+  end
+
+  describe '#tags_without' do
+    it 'returns a Hash' do
+      expect(subject.tags_without)
+        .to be_an_instance_of ::AWS::EC2::ResourceTagCollection
+    end
+  end
+
+  describe '#name' do
+    it 'returns Hash value' do
+      subject.stub(:all_tags_hash) {@tags}
+      expect(subject.name).to eq(@tags['Name'])
+    end
+  end
+
+  describe '#named_instances' do
+    it 'returns AWS::EC2::InstanceCollection' do
+      expect(subject.named_instances)
+        .to be_an_instance_of ::AWS::EC2::InstanceCollection
+    end
+  end
+
+  describe '#functional_instances' do
+    it 'returns an Array' do
+      subject.stub(:named_instances) { [@inst1, @inst2, @inst3] }
+      expect(subject.functional_instances).to be_an Array
+      expect(subject.functional_instances).to include(@inst1, @inst2)
+    end
+  end
+
+  describe '#similar_instances' do
+    it 'returns an Array' do
+      subject.stub(:functional_instances) { [@inst1, @inst2] }
+      expect(subject.similar_instances).to include(@tags['Name'])
+    end
+  end
+end

data/spec/vscripts/aws/metadata_spec.rb

@@ -0,0 +1,48 @@
+require 'spec_helper'
+require 'vscripts/aws/metadata'
+
+describe VScripts::AWS::Metadata do
+  before(:each) do
+    @dummy = DummyClass.new
+    @dummy.extend VScripts::AWS::Metadata
+    @dummy.stub_chain(:open, :read).and_return('Remote server response')
+  end
+
+  describe '#metadata_url' do
+    it 'returns metadata url string' do
+      expect(@dummy.metadata_url).to be_a String
+    end
+  end
+
+  describe '#zone' do
+    it 'returns zone string' do
+      expect(@dummy.zone).to eq('Remote server response')
+    end
+  end
+
+  describe '#region' do
+    it 'returns region string' do
+      @dummy.stub(:zone).and_return('us-test-1a')
+      expect(@dummy.region).to eq('us-test-1')
+    end
+  end
+
+  describe '#instance_id' do
+    it 'returns instance id string' do
+      expect(@dummy.instance_id).to eq('Remote server response')
+    end
+  end
+
+  describe '#public_hostname' do
+    it 'returns public hostname string' do
+      expect(@dummy.public_hostname).to eq('Remote server response')
+    end
+  end
+
+  describe '#ec2_instance?' do
+    it 'returns true' do
+      Net::HTTP.stub(:get_response).and_return(true)
+      expect(@dummy.ec2_instance?).to be_true
+    end
+  end
+end

data/spec/vscripts/aws_spec.rb

@@ -0,0 +1,7 @@
+require 'vscripts/aws'
+
+describe VScripts::AWS do
+  it 'Should load default configuration for AWS SDK' do
+    expect(::AWS.config.stub_requests).to be_true
+  end
+end

data/spec/vscripts/command_line_spec.rb

@@ -0,0 +1,34 @@
+require 'vscripts/command_line'
+
+describe VScripts::CommandLine do
+  before :each do
+    @cmd = VScripts::Command.list.first.to_s
+    @cli = VScripts::CommandLine.new([@cmd, 'extra_args'])
+  end
+
+  describe '#new' do
+    it 'returns the command line arguments' do
+      expect(@cli).to be_an_instance_of VScripts::CommandLine
+    end
+  end
+
+  describe '#global' do
+    it 'returns the global options as a Hash' do
+      expect(@cli.global).to be_an_instance_of Hash
+    end
+  end
+
+  describe '#command' do
+    it 'returns the command name' do
+      expect(@cli.command).to be_a Symbol
+      expect(@cli.command).to eql @cmd.capitalize.to_sym
+    end
+  end
+
+  describe '#extra' do
+    it 'returns the rest of the arguments as an Array' do
+      expect(@cli.extra).to be_an_instance_of Array
+      expect(@cli.extra).to eql ['extra_args']
+    end
+  end
+end

data/spec/vscripts/command_spec.rb

@@ -0,0 +1,9 @@
+require 'vscripts/command'
+
+describe VScripts::Command do
+  describe '#list' do
+    it 'returns a list of available commands' do
+      expect(VScripts::Command.list).to be_an_instance_of Array
+    end
+  end
+end

data/spec/vscripts/commands/tags2facts_spec.rb

@@ -0,0 +1,34 @@
+require 'vscripts/commands/tags2facts'
+
+describe VScripts::Commands::Tags2facts do
+
+  before :each do
+    @tags2facts = VScripts::Commands::Tags2facts.new(['extra_args'])
+  end
+
+  describe '#new' do
+    it 'returns cli arguments and loads EC2 SDK' do
+      expect(@tags2facts.cli).to be_a Hash
+      expect(@tags2facts.ec2).to be_an_instance_of VScripts::AWS::EC2
+    end
+  end
+
+  describe '#parser' do
+    it 'returns parser' do
+      expect(@tags2facts.parser).to be_an_instance_of Trollop::Parser
+    end
+  end
+
+  describe '#exclude_list' do
+    it 'returns exclude list' do
+      expect(@tags2facts.exclude_list).to eq(['Name', 'Domain'])
+    end
+  end
+
+  describe '#tags_json' do
+    it 'returns JSON formatted string' do
+      @tags2facts.stub_chain(:tags_hash, :to_h) {{ key: 'value' }}
+      expect(@tags2facts.tags_json).to eq("{\n  \"key\": \"value\"\n}")
+    end
+  end
+end

data/spec/vscripts/util/local_system_spec.rb

@@ -0,0 +1,80 @@
+require 'spec_helper'
+require 'tempfile'
+require 'vscripts/util/local_system'
+
+describe VScripts::Util::LocalSystem do
+  before(:all) do
+    @dummy = DummyClass.new
+    @dummy.extend VScripts::Util::LocalSystem
+  end
+
+  describe '#hosts_path' do
+    it 'returns the path to hosts file' do
+      expect(@dummy.hosts_path).to be_a String
+    end
+  end
+  describe '#hostname_path' do
+    it 'returns the path to the hostname file' do
+      expect(@dummy.hostname_path).to be_a String
+    end
+  end
+  describe '#local_fqdn' do
+    it 'returns the local FQDN' do
+      expect(@dummy.local_fqdn).to be_a String
+    end
+  end
+  describe '#local_host_name' do
+    it 'returns the local host name' do
+      expect(@dummy.local_host_name).to be_a String
+    end
+  end
+  describe '#local_domain_name' do
+    it 'returns the local domain name' do
+      expect(@dummy.local_domain_name).to be_a String
+    end
+  end
+  describe '#ensure_file_dir' do
+    it 'create a directory for the specified files' do
+      test_dir  = Dir::Tmpname.make_tmpname '/tmp/vscripts', nil
+      test_file = 'test_file'
+      @dummy.ensure_file_dir("#{test_dir}/#{test_file}")
+      expect(Dir.exists?(test_dir)).to be_true
+      `rm -r #{test_dir}`
+    end
+  end
+  describe '#write_file' do
+    it 'should write to a file' do
+      test_file  = Dir::Tmpname.make_tmpname '/tmp/vscripts', nil
+      @dummy.write_file(test_file, 'test')
+      expect(IO.read(test_file)).to eq('test')
+      `rm #{test_file}`
+    end
+  end
+  describe '#ensure_file_content' do
+    it 'should ensure content of file' do
+      test_dir  = Dir::Tmpname.make_tmpname '/tmp/vscripts', nil
+      test_file = 'test_file'
+      test      = "#{test_dir}/#{test_file}"
+      @dummy.ensure_file_content(test, 'test')
+      expect(IO.read(test)).to eq('test')
+      `rm -r #{test_dir}`
+    end
+  end
+  describe '#checks' do
+    it 'returns an array of checks' do
+      expect(@dummy.checks).to be_a Hash
+    end
+  end
+  describe '#process_checks' do
+    it 'should execute system command and return exit code' do
+      @dummy.stub(:checks) {{'test command' => 'exit 5'}}
+      expect(@dummy.process_checks).to eq({"test command"=>5})
+    end
+  end
+  describe '#status_codes' do
+    it 'should return an Array of exit codes' do
+      @dummy.stub(:checks) {{'test command' => 'exit 5'}}
+      expect(@dummy.status_codes).to eq([5])
+    end
+  end
+end

data/spec/vscripts/version_spec.rb

@@ -0,0 +1,7 @@
+require 'vscripts/version'
+
+describe 'VScripts::VERSION' do
+  it 'Should be a string' do
+    expect(VScripts::VERSION).to be_an_instance_of String
+  end
+end

data/spec/vscripts_spec.rb

@@ -0,0 +1,4 @@
+require 'vscripts'
+
+describe VScripts do
+end

data/tasks/deploy.rake

@@ -0,0 +1,95 @@
+DEV_BRANCH     = 'develop'
+LOG_RANGE      = "master...#{DEV_BRANCH}"
+VERSION_FILE   = 'VERSION'
+CHANGELOG_FILE = 'CHANGELOG.md'
+
+desc 'Automated deployment'
+task :deploy do
+  check_branch
+  commit_changelog
+  merge_changes
+  push_release
+  package_and_push_gem
+end
+
+def check_branch
+  if ! `git ls-files --others --exclude-standard`.empty?
+    abort 'ERROR: There are untracked files!'
+  elsif ! system('git diff --quiet && git diff --cached --quiet')
+    abort 'ERROR: There are staged but uncommitted files!'
+  end
+  `git fetch`
+end
+
+def bump_version
+  @version = File.read(VERSION_FILE).strip
+  @version.gsub(/(\d+)\.(\d+)\.(\d+)/) {
+    @major, @minor, @patch = $1.to_i, $2.to_i, $3.to_i
+  }
+  @new_version = "#{@major}.#{@minor}.#{@patch + 1}"
+  File.write('VERSION', @new_version)
+end
+
+def changelog
+  File.read(CHANGELOG_FILE)
+end
+
+def changes
+  @changes ||= `git log #{LOG_RANGE} --no-merges`
+end
+
+def pretty_changes
+  `git log #{LOG_RANGE} --reverse --no-merges --pretty=format:'  * %s ([%cn - %h](https://github.com/vghn/vscripts/commit/%H))'`
+end
+
+def update_changelog
+  bump_version
+  puts 'Writing new changelog'
+  File.write(
+    CHANGELOG_FILE,
+    "Version #{@new_version}\n---\n" + pretty_changes + "\n\n" + changelog
+  )
+end
+
+def commit_changelog
+  if changes.lines.count > 0
+    update_changelog
+    puts 'Committing version and changelog'
+    `git commit -am \"Bump version.\"`
+  else
+    puts 'No changes committed. Exiting.'
+    exit 0
+  end
+end
+
+def branch
+  @branch ||= `git branch`.match(/^\* (\S+)/)[1]
+end
+
+def switch_to_branch(name)
+  `git checkout #{name}` unless branch == name
+end
+
+def merge
+  @merge ||= `git merge --no-ff #{DEV_BRANCH} -m \"v#{@new_version}\"`
+end
+
+def merge_changes
+  switch_to_branch('master')
+  puts "Merging \"#{DEV_BRANCH}\" branch"
+  abort 'ERROR: Conflicts found; Stopping!' if merge =~ /conflict/i
+end
+
+def push_release
+  puts "Tagging v#{@new_version} release."
+  `git tag -a v#{@new_version} -m 'Version #{@new_version}'`
+  puts 'Pushing all branches and tags'
+  `git push --all --follow-tags`
+end
+
+def package_and_push_gem
+  puts 'Building gem'
+  Rake::Task['build'].invoke
+  puts 'Pushing gem to Rubygems'
+  `gem push pkg/vscripts-#{@new_version}.gem`
+end