moonshot 0.7.3 → 0.7.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4efa7393335490c5e80a03e6f3c733727c75acdf
-  data.tar.gz: 185c228d9dcec000b92fafca815cf9ac9b36ebf4
+  metadata.gz: 207bc0ba79e75fe91db4654c343970fa6d902503
+  data.tar.gz: d3dba39637da2478ea33d114ded4d5d386a6df38
 SHA512:
-  metadata.gz: 17624a530e39bbe1179a776c082d9f1b4e5d2b8ea68dd0c0fc811d3767b5a4b014721b6b41d7253ba973d33772ee1cfb2b4b7c9345bfeef5625dcbeb9e045646
-  data.tar.gz: ca1d90df3feba59452c6d0a5c6bf63467c73d7fc009c2fda144cfce826a31e7f6184c1042b6bfbf99256baaab37ae9a4cabcb8cd5de03e162c2fd5f21d0cd63f
+  metadata.gz: a85739273ac8d835db5402aa3ebf96c290e961c2ac4a4af6fa6825a99172dc0cec56d6d9c5d5a762cb4523e77d0ec5d76fac4f5422db7188b69c18b94f85a025
+  data.tar.gz: 3c51fe1fc53305da957c7bca13c5dc041d2e2cff497f2b93ed7f7dd8b28c9fe8a8048b5f17a89241a046557ba55e5e69885c918de1d84f07e6984518cec86efd

data/lib/moonshot.rb

@@ -10,6 +10,8 @@ module Moonshot
   end
   module DeploymentMechanism # rubocop:disable Documentation
   end
+  module Plugins # rubocop:disable Documentation
+  end
 end
 
 [
@@ -39,3 +41,8 @@ end
   'build_mechanism/version_proxy',
   'deployment_mechanism/code_deploy'
 ].each { |f| require_relative "moonshot/#{f}" }
+
+# Bundled plugins
+[
+  'backup'
+].each { |p| require_relative "plugins/#{p}" }

data/lib/plugins/backup.rb

@@ -0,0 +1,174 @@
+require 'rubygems/package'
+require 'zlib'
+require_relative '../moonshot/creds_helper'
+
+module Moonshot
+  module Plugins
+    # Moonshot plugin class for deflating and uploading files on given hooks
+    class Backup # rubocop:disable Metrics/ClassLength
+      include Moonshot::CredsHelper
+
+      attr_accessor :bucket,
+                    :buckets,
+                    :files,
+                    :hooks,
+                    :target_name
+
+      def initialize
+        yield self if block_given?
+        raise ArgumentError \
+          if @files.nil? || @files.empty? || @hooks.nil? || !(@bucket.nil? ^ @buckets.nil?)
+
+        @target_name ||= '%{app_name}_%{timestamp}_%{user}.tar.gz'
+      end
+
+      # Factory method to create preconfigured Backup plugins. Uploads current
+      # template and parameter files.
+      # @param backup [String] target bucket name
+      # @return [Backup] configured backup object
+      def self.to_bucket(bucket)
+        raise ArgumentError if bucket.nil? || bucket.empty?
+        Moonshot::Plugins::Backup.new do |b|
+          b.bucket = bucket
+          b.files = [
+            'cloud_formation/%{app_name}.json',
+            'cloud_formation/parameters/%{stack_name}.yml'
+          ]
+          b.hooks = [:post_create, :post_update]
+        end
+      end
+
+      # Main worker method, creates a tarball of the given files, and uploads
+      # to an S3 bucket.
+      #
+      # @param resources [Resources] injected Moonshot resources
+      def backup(resources) # rubocop:disable Metrics/AbcSize
+        raise ArgumentError if resources.nil?
+
+        @app_name = resources.stack.app_name
+        @stack_name = resources.stack.name
+        @target_name = render(@target_name)
+        @target_bucket = define_bucket
+
+        return if @target_bucket.nil?
+
+        resources.ilog.start("#{log_message} in progress.") do |s|
+          begin
+            tar_out = tar(@files)
+            zip_out = zip(tar_out)
+            upload(zip_out)
+
+            s.success("#{log_message} succeeded.")
+          rescue StandardError => e
+            s.failure("#{log_message} failed: #{e}")
+          ensure
+            tar_out.close unless tar_out.nil?
+            zip_out.close unless zip_out.nil?
+          end
+        end
+      end
+
+      # Dynamically responding to hooks supplied in the constructor
+      def method_missing(method_name, *args, &block)
+        @hooks.include?(method_name) ? backup(*args) : super
+      end
+
+      def respond_to?(method_name, include_private = false)
+        @hooks.include?(method_name) || super
+      end
+
+      private
+
+      attr_accessor :app_name,
+                    :stack_name,
+                    :target_bucket
+
+      # Create a tar archive in memory, returning the IO object pointing at the
+      # beginning of the archive.
+      #
+      # @param target_files [Array<String>]
+      # @return tar_stream [IO]
+      def tar(target_files)
+        tar_stream = StringIO.new
+        Gem::Package::TarWriter.new(tar_stream) do |writer|
+          target_files.each do |file|
+            file = render(file)
+
+            writer.add_file(File.basename(file), 0644) do |io|
+              File.open(file, 'r') { |f| io.write(f.read) }
+            end
+          end
+        end
+        tar_stream.seek(0)
+        tar_stream
+      end
+
+      # Create a zip archive in memory, returning the IO object pointing at the
+      # beginning of the zipfile.
+      #
+      # @param io_tar [IO] tar stream
+      # @return zip_stream [IO] IO stream of zipped file
+      def zip(io_tar)
+        zip_stream = StringIO.new
+        Zlib::GzipWriter.wrap(zip_stream) do |gz|
+          gz.write(io_tar.read)
+          gz.finish
+        end
+        zip_stream.seek(0)
+        zip_stream
+      end
+
+      # Uploads an object from the passed IO stream to the specified bucket
+      #
+      # @param io_zip [IO] tar stream
+      def upload(io_zip)
+        s3_client.put_object(
+          acl: 'private',
+          bucket: @target_bucket,
+          key: @target_name,
+          body: io_zip
+        )
+      end
+
+      # Renders string with the specified placeholders
+      #
+      # @param io_zip [String] raw string with placeholders
+      # @return [String] rendered string
+      def render(placeholder)
+        format(
+          placeholder,
+          app_name: @app_name,
+          stack_name: @stack_name,
+          timestamp: Time.now.to_i.to_s,
+          user: ENV['USER']
+        )
+      end
+
+      def log_message
+        "Uploading '#{@target_name}' to '#{@target_bucket}'"
+      end
+
+      def iam_account
+        iam_client.list_account_aliases.account_aliases.first
+      end
+
+      def define_bucket
+        case
+        # returning already calculated bucket name
+        when @target_bucket
+          @target_bucket
+        # single bucket for all accounts
+        when @bucket
+          @bucket
+        # calculating bucket based on account name
+        when @buckets
+          bucket_by_account(iam_account)
+        end
+      end
+
+      def bucket_by_account(account)
+        @buckets[account]
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: moonshot
 version: !ruby/object:Gem::Version
-  version: 0.7.3
+  version: 0.7.4
 platform: ruby
 authors:
 - Cloud Engineering <engineering@acquia.com>
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-06-10 00:00:00.000000000 Z
+date: 2016-06-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -213,6 +213,7 @@ files:
 - lib/moonshot/stack_parameter_printer.rb
 - lib/moonshot/stack_template.rb
 - lib/moonshot/unicode_table.rb
+- lib/plugins/backup.rb
 homepage: https://github.com/acquia/moonshot
 licenses:
 - Apache-2.0