continuum-stager-api 0.1.0

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    MWRiOTA4YWU1ZGNlZDA4YzI4MTMwMjRmYjA2ZjlmYTZiYjMxOTZlMg==
+  data.tar.gz: !binary |-
+    ZDE3NWEwODI5MGIyNjMyODcyYTNmNDU4NjZmNmI5ZmM4NWY2ODkzOA==
+!binary "U0hBNTEy":
+  metadata.gz: !binary |-
+    MjUwY2Q0YzVkYzc4MTYxYTVmMTMwNGUxYzIxNGIzMzZiZjQ4MzVmM2JmMjAw
+    ODFhNTJmOTE2Yzc4YmMzNjMyMGRmOTUzZjlkYTY2NjIxYWVlZjJiOWIxZjRk
+    ZGRlODlmZDJmZDVmOTU0YjlhZmQ3ODU4YzQwZDlhMWQzYjUzNzM=
+  data.tar.gz: !binary |-
+    MjQwMzdkODU0YzI3NGQxNTIwMzgxZmRmYWM5NzcyMDY2Mjk4ZTk1MDNkODgx
+    OGM0MjBlYzI5Y2NlYzdkZTA5ZTI5NmM0ODNkODczMTVmNjVkM2RmYWU2OGYy
+    ZWUwMjk0NTI2OTZjY2VhZjVkOTExMWQ2OTc5NGU2MjUwYzUwOGU=

data/.gitignore

@@ -0,0 +1,22 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+.ruby-gemset
+.ruby-version
+.rvmrc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+.powenv
+.idea/

data/.travis.yml

@@ -0,0 +1,15 @@
+before_install:
+  - gem update --system 2.1.11
+language: ruby
+rvm:
+  - "1.8.7"
+  - "1.9.2"
+  - "1.9.3"
+  - "2.0.0"
+  - "2.1.0"
+  - "rbx"
+  - "jruby"
+matrix:
+  allow_failures:
+    - rvm: "rbx"
+    - rvm: "jruby"

data/CHANGELOG.md

@@ -0,0 +1,16 @@
+# Changelog
+All notable changes to this project will be documented in this file.
+
+## 0.0.1 Initial version
+
+### Added
+- Nothing.
+
+### Deprecated
+- Nothing.
+
+### Removed
+- Nothing.
+
+### Fixed
+- Nothing.

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2014 Apcera
+
+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,37 @@
+# continuum-stager-api
+
+Simple gem to assist users writing custom Continuum Stagers.
+
+## Installation
+
+Add to your `Gemfile`:
+
+```ruby
+gem "continuum-stager-api"
+```
+
+Then `bundle install`.
+
+## License
+
+The MIT License (MIT)
+
+Copyright (c) 2014 Apcera
+
+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,7 @@
+#!/usr/bin/env rake
+require File.join('bundler', 'gem_tasks')
+require File.join('rspec', 'core', 'rake_task')
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/apcera-stager-api-contrib.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+Gem::Specification.new do |gem|
+  gem.add_dependency "rest-client"
+  gem.add_dependency 'json'
+
+  gem.authors       = ["Josh Ellithorpe"]
+  gem.email         = ["josh@apcera.com"]
+  gem.description   = %q{Continuum Stager api library}
+  gem.summary       = %q{Continuum Stager api library which makes it super easy to write stagers for Apcera's Continuum.}
+  gem.homepage      = "http://apcera.com"
+  gem.license       = "MIT"
+
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.files         = `git ls-files`.split("\n")
+  gem.test_files    = `git ls-files -- {spec}/*`.split("\n")
+  gem.name          = "continuum-stager-api"
+  gem.require_paths = ["lib"]
+  gem.version       = "0.1.0"
+
+  gem.add_development_dependency 'rspec', '~> 2.6.0'
+  gem.add_development_dependency 'rake'
+  gem.add_development_dependency 'webmock', '1.11'
+  gem.add_development_dependency 'simplecov'
+  gem.add_development_dependency 'vcr'
+end

data/lib/apcera/stager/error.rb

@@ -0,0 +1,6 @@
+module Apcera
+  module Error
+    class StagerURLRequired < StandardError; end
+    class ExecuteError < StandardError; end
+  end
+end

data/lib/apcera/stager/loader.rb

@@ -0,0 +1,10 @@
+require "bundler"
+Bundler.setup
+require 'rest-client'
+require 'json'
+
+base = File.expand_path File.dirname(__FILE__)
+
+Dir[File.join(base, '*.rb')].each do |file|
+  require file
+end

data/lib/apcera/stager/stager.rb

@@ -0,0 +1,270 @@
+module Apcera
+  class Stager
+    attr_accessor :stager_url, :app_path, :root_path, :pkg_path, :updated_pkg_path, :system_options
+
+    PKG_NAME = "pkg.tar.gz"
+    UPDATED_PKG_NAME = "updated.tar.gz"
+
+    def initialize(options = {})
+      # Require stager url. Needed to talk to the Staging Coordinator.
+      @stager_url = options[:stager_url] || ENV["STAGER_URL"]
+      raise Apcera::Error::StagerURLRequired.new("stager_url required") unless @stager_url
+
+      # Setup the environment, some test items here.
+      setup_environment
+    end
+
+    # Download a package from the staging coordinator.
+    def download
+      response = RestClient.get(@stager_url + "/data")
+      File.open(@pkg_path, "wb") do |f|
+        f.write(response.to_str)
+      end
+    rescue => e
+      fail e
+    end
+
+    # Execute a command in the shell.
+    # We don't want real commands in tests.
+    def execute(cmd)
+      Bundler.with_clean_env do
+        result = system(cmd, @system_options)
+        if !result
+          raise Apcera::Error::ExecuteError.new("failed to execute: #{cmd}.\n")
+        end
+
+        result
+      end
+    rescue => e
+      fail e
+    end
+
+    # Execute a command in the app dir. Useful helper.
+    def execute_app(cmd)
+      Bundler.with_clean_env do
+        Dir.chdir(@app_path) do |app_path|
+          result = system(cmd, @system_options)
+          if !result
+            raise Apcera::Error::ExecuteError.new("failed to execute: #{cmd}.\n")
+          end
+
+          result
+        end
+      end
+    rescue => e
+      fail e
+    end
+
+    # Extract the package to a given location.
+    def extract(location)
+      @app_path = File.join(@root_path, location)
+      Dir.mkdir(@app_path) unless Dir.exists?(@app_path)
+
+      execute_app("tar -zxf #{@pkg_path}")
+    rescue => e
+      fail e
+    end
+
+    # Upload the new package to the staging coordinator
+    def upload
+      execute_app("tar czf #{@updated_pkg_path} .")
+
+      sha1 = Digest::SHA1.file(@updated_pkg_path)
+      File.open(@updated_pkg_path, "rb") do |f|
+        response = RestClient.post(@stager_url+"/data?sha1=#{sha1.to_s}", f.read, { :content_type => "application/octet-stream" } )
+      end
+    rescue => e
+      fail e
+    end
+
+    # Snapshot the stager filesystem for app
+    def snapshot
+      response = RestClient.post(@stager_url+"/snapshot", {})
+    rescue => e
+      fail e
+    end
+
+    # Add environment variable to package.
+    def environment_add(key, value)
+      response = RestClient.post(@stager_url+"/meta", {
+        :resource => "environment",
+        :action => "add",
+        :key => key,
+        :value => value
+      })
+    rescue => e
+      fail e
+    end
+
+    # Delete environment variable from package.
+    def environment_remove(key, value)
+      response = RestClient.post(@stager_url+"/meta", {
+        :resource => "environment",
+        :action => "remove",
+        :key => key,
+        :value => value
+      })
+    rescue => e
+      fail e
+    end
+
+    # Add provides to package.
+    def provides_add(type, name)
+      response = RestClient.post(@stager_url+"/meta", {
+        :resource => "provides",
+        :action => "add",
+        :type => type,
+        :name => name
+      })
+    rescue => e
+      fail e
+    end
+
+    # Delete provides from package.
+    def provides_remove(key, value)
+      response = RestClient.post(@stager_url+"/meta", {
+        :resource => "provides",
+        :action => "remove",
+        :type => type,
+        :name => name
+      })
+    rescue => e
+      fail e
+    end
+
+    # Add dependencies to package.
+    def dependencies_add(type, name)
+      response = RestClient.post(@stager_url+"/meta", {
+        :resource => "dependencies",
+        :action => "add",
+        :type => type,
+        :name => name
+      })
+    rescue => e
+      fail e
+    end
+
+    # Delete dependencies from package.
+    def dependencies_remove(type, name)
+      response = RestClient.post(@stager_url+"/meta", {
+        :resource => "dependencies",
+        :action => "remove",
+        :type => type,
+        :name => name
+      })
+    rescue => e
+      fail e
+    end
+
+    # Add template to package.
+    def templates_add(path, left_delimiter = "{{", right_delimiter = "}}")
+      response = RestClient.post(@stager_url+"/meta", {
+        :resource => "templates",
+        :action => "add",
+        :path => path,
+        :left_delimiter => left_delimiter,
+        :right_delimiter => right_delimiter
+      })
+    rescue => e
+      fail e
+    end
+
+    # Delete template from package.
+    def templates_remove(path, left_delimiter = "{{", right_delimiter = "}}")
+      response = RestClient.post(@stager_url+"/meta", {
+        :resource => "templates",
+        :action => "remove",
+        :path => path,
+        :left_delimiter => left_delimiter,
+        :right_delimiter => right_delimiter
+      })
+    rescue => e
+      fail e
+    end
+
+    # Get metadata for the package being staged.
+    def meta
+      response = RestClient.get(@stager_url+"/meta")
+      return JSON.parse(response.to_s)
+    rescue => e
+      output_error "Error: #{e.message}.\n"
+      raise e
+    end
+
+    # Tell the staging coordinator you are done.
+    def done
+      response = RestClient.post(@stager_url+"/done", {})
+      exit0r 0
+    rescue => e
+      fail e
+    end
+
+    # Tell the staging coordinator you need to relaunch.
+    def relaunch
+      response = RestClient.post(@stager_url+"/relaunch", {})
+      exit0r 0
+    rescue => e
+      fail e
+    end
+
+    # Finish staging, compress your app dir and send to the staging coordinator.
+    # Then tell the staging coordinator we are done.
+    def complete
+      upload
+      done
+    end
+
+    # Returns the start command for the package.
+    def start_command
+      self.meta["environment"]["START_COMMAND"]
+    end
+
+    # Easily set the start command
+    def start_command=(val)
+      self.environment_add("START_COMMAND", val)
+    end
+
+    # Returns the start path for the package.
+    def start_path
+      self.meta["environment"]["START_PATH"]
+    end
+
+    # Easily set the start path
+    def start_path=(val)
+      self.environment_add("START_PATH", val)
+    end
+
+    # Fail the stager, something went wrong.
+    def fail(error = nil)
+      output_error "Error: #{error.message}.\n" if error
+      RestClient.post(@stager_url+"/failed", {})
+    rescue => e
+      output_error "Error: #{e.message}.\n"
+    ensure
+      exit0r 1
+    end
+
+    # Exit, needed for tests to not quit.
+    def exit0r(code)
+      exit code
+    end
+
+    private
+
+    def output_error(text)
+      $stderr.puts text
+    end
+
+    def output(text)
+      $stdout.puts text
+    end
+
+    def setup_environment
+      # When staging we use the root path. These are overridden in tests.
+      @root_path = "/"
+      @pkg_path = File.join(@root_path, PKG_NAME)
+      @updated_pkg_path = File.join(@root_path, UPDATED_PKG_NAME)
+      @system_options = {}
+    end
+  end
+end