chambermaid 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ceb4e76a467a39e91b32a40efff57e8fe8d213e28ea38c9742569af98eac7e0f
-  data.tar.gz: dfdb963dece1c57711fae828eaeebf53490ce2569f6fa339839d2c01a24787ca
+  metadata.gz: ed3bfd126164a3fd94e16ccdf453af29f9091f9139074bb80b922fd237db3453
+  data.tar.gz: d609e14f50b7b085a38dfbcc4379920015d2b184e11fa0dbf2c26e294a238990
 SHA512:
-  metadata.gz: 2997b5bf940b31802cb900d50b042820885da4209fff8e2231228994fbb3daf3efa4c979fb6a41f37a1512e3cea96440e3744b8ca2939dcd2543cc1287c1e960
-  data.tar.gz: cc518b62b452ea0c63e83b04fd3535345ed335a671fffd6fba6c184addfdaa45a9f29553a825d55e97921598300675e1c054d496dd30a850ca520968170f15be
+  metadata.gz: 567267cf7ea6992aa545ad5e745eb6f6c5653366526bfee94873894a31d8c235bd2ebc1ed3c54d79fc362b46093a0fa004c4e79ee77dbfcfd9b371cc36439875
+  data.tar.gz: d65da6cc45e6fbfc6c53e09a3de5a55a89a9d707385944c8868db6a50de6b7b83d30b235ff2084e50dff27daab9a1d668b2d4c1bb4789640031425b1b33f0918

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    chambermaid (0.2.0)
+    chambermaid (0.3.0)
       aws-sdk-ssm (~> 1.85)
 
 GEM

data/README.md

@@ -26,6 +26,7 @@ Or install it yourself as:
 
 ```ruby
 Chambermaid.add_namespace("/my/param/namespace")
+Chambermaid.add_service("my-chamber-service")
 ```
 
 **Configuration Block**
@@ -35,6 +36,7 @@ Chambermaid.add_namespace("/my/param/namespace")
 
 Chambermaid.configure do |config|
   config.add_namespace("/my/param/namespace")
+  config.add_service("my-chamber-service")
 
   # Set `overload: true` to choose these params over existing
   # ones in ENV when they are merged together
@@ -42,6 +44,11 @@ Chambermaid.configure do |config|
 end
 ```
 
+**Reload SSM into ENV**
+```ruby
+Chambermaid.reload!
+```
+
 **Restore ENV to original state**
 ```ruby
 Chambermaid.restore!

data/lib/chambermaid/base.rb

@@ -1,16 +1,19 @@
+require "chambermaid/environment"
+require "chambermaid/namespace"
 require "chambermaid/parameter_store"
 
 module Chambermaid
   module Base
     def self.extended(base)
       # Make a copy of ENV before we mess it all up
-      @@_original_env = ENV.to_h.dup
+      @@_original_env = ENV.to_h.dup.freeze
     end
 
     extend self
 
     def configure
       yield self
+      load!
     end
 
     # @todo
@@ -18,16 +21,18 @@ module Chambermaid
       raise "Namespaces must be defined" unless @namespaces
     end
 
+    # Load SSM into ENV
+    def load!
+      @namespaces.each(&:load!)
+    end
+
     # @todo
     def reload!
-      restore!
-      @namespaces.each do |ns|
-        ns[:store].reload!
-        update_env!(
-          params: ns[:store].params,
-          overload: ns[:overload]
-        )
-      end
+      @namespaces.each(&:reload!)
+    end
+
+    def unload!
+      @namespaces.each(&:unload!)
     end
 
     # Restore ENV to its original state
@@ -43,30 +48,66 @@ module Chambermaid
     #   true  - replace any duplicate ENV keys with new params
     #   false - keep any existing duplicate ENV key values
     #
-    # @raise
+    # @raise [ArgumentError]
+    #   when `path` is not a string
+    #
+    # @example
+    #   Chambermaid.add_namespace("/my/param/namespace")
+    #
+    # @example overload duplicate ENV vars
+    #   Chambermaid.add_namespace("/my/param/namespace", overload: true)
     def add_namespace(path, overload: false)
-      @namespaces ||= []
-      # raise "namespace already included in ENV" unless @namespaces[path].nil?
+      raise ArgumentError.new("`path` must be a string") unless path.is_a?(String)
+      raise ArgumentError.new("`overload` must be a boolean") unless [true, false].include?(overload)
 
-      store = ParameterStore.load!(path: path)
-      @namespaces << { store: store, overload: overload }
-      update_env!(params: store.params, overload: overload)
+      namespaces << Namespace.new(path: path, overload: overload)
     end
 
-    # Inject into ENV
+    # Immediately load an AWS SSM parameter namespace to ENV
     #
-    # @param [Hash] params
+    # @param [String] path
     # @param [Boolean] overload
     #   true  - replace any duplicate ENV keys with new params
     #   false - keep any existing duplicate ENV key values
-    def update_env!(params:, overload:)
-      if overload
-        ENV.update(params)
-      else
-        current_env = ENV.to_h.dup
-        new_env = params.merge(current_env)
-        ENV.replace(new_env)
-      end
+    #
+    # @raise [ArgumentError]
+    #   when `path` is not a string
+    #
+    # @example
+    #   Chambermaid.add_namespace!("/my/param/namespace")
+    #
+    # @example overload duplicate ENV vars
+    #   Chambermaid.add_namespace("/my/param/namespace", overload: true)
+    def add_namespace!(path, overload: false)
+      raise ArgumentError.new("`path` must be a string") unless path.is_a?(String)
+      raise ArgumentError.new("`overload` must be a boolean") unless [true, false].include?(overload)
+
+      namespaces << Namespace.load!(path: path, overload: overload)
+    end
+
+    # Add all secrets from Chamber service to ENV
+    #
+    # @param [String] service
+    # @param [Boolean] overload
+    #   true  - replace any duplicate ENV keys with new params
+    #   false - keep any existing duplicate ENV key values
+    #
+    # @example
+    #   Chambermaid.add_service("my-chamber-service")
+    #
+    # @example overload duplicate ENV vars
+    #   Chambermaid.add_service("my-chamber-service", overload: true)
+    #
+    # @see {Chambermaid::Base.add_namespace}
+    def add_service(service, overload: false)
+      service = "/#{service}" unless service[0] == "/"
+      add_namespace(service)
+    end
+
+    private
+
+    def namespaces
+      @namespaces ||= []
     end
   end
 end

data/lib/chambermaid/environment.rb

@@ -0,0 +1,68 @@
+module Chambermaid
+  class Environment < Hash
+    attr_reader :params
+
+    # Create new Chambermaid::Environment
+    #
+    # @param [Hash] params
+    #
+    # @raise [ArgumentError]
+    #   if params is not type Hash
+    #
+    # @return [Chambermaid::Environment]
+    def initialize(params)
+      validate_params!(params)
+      stash_current_env!
+      update(format_env(params))
+    end
+
+    # Generate a dotenv (.env) compatible string
+    #
+    # @return [String] dotenv compatible string
+    def to_dotenv
+      to_h.inject("") do |env_str, param|
+        env_str + "#{param[0]}=#{param[1]}\n"
+      end
+    end
+
+    # Write a .env file
+    #
+    # @param [String] file_path
+    def to_file!(file_path)
+      File.open(file_path, "wb") do |f|
+        f.write(to_dotenv)
+      end
+    end
+
+    # Inject into ENV without overwriting duplicates
+    def load!
+      each { |k, v| ENV[k] ||= v }
+    end
+
+    # Inject into ENV and overwrite duplicates
+    def overload!
+      each { |k, v| ENV[k] = v }
+    end
+
+    # Restore to original ENV
+    def unload!
+      ENV.replace(@_original_env)
+    end
+
+    private
+
+    def stash_current_env!
+      @_original_env ||= ENV.to_h.dup.freeze
+    end
+
+    def format_env(params)
+      params.map{|k,v| [k.upcase, v]}.to_h
+    end
+
+    def validate_params!(params)
+      unless params.is_a?(Hash)
+        raise ArgumentError.new("`params` must be a hash")
+      end
+    end
+  end
+end

data/lib/chambermaid/namespace.rb

@@ -0,0 +1,42 @@
+module Chambermaid
+  class Namespace
+    # @param [String] path
+    # @param [Boolean] overload
+    def initialize(path:, overload: false)
+      @path = path
+      @overload = overload
+
+      @store = ParameterStore.new(path: path)
+      @env = Environment.new({})
+    end
+
+    def self.load!(path:, overload: false)
+      namespace = new(path: path, overload: overload)
+      namespace.load!
+      namespace
+    end
+
+    def load!
+      @store.load!
+      load_env!
+    end
+
+    def reload!
+      @env.unload!
+      @store.reload!
+      load_env!
+    end
+
+    def unload!
+      @env.unload!
+    end
+
+    private
+
+    # Inject into ENV
+    def load_env!
+      @env.replace(@store.params)
+      @overload ? @env.overload! : @env.load!
+    end
+  end
+end

data/lib/chambermaid/parameter_store.rb

@@ -15,6 +15,10 @@ module Chambermaid
       fetch_ssm_params!
     end
 
+    def loaded?
+      !@params_list.empty?
+    end
+
     def self.load!(path:)
       store = new(path: path)
       store.load!
@@ -29,12 +33,6 @@ module Chambermaid
 
     alias :to_h :params
 
-    def to_env
-      params.inject("") do |env_str, param|
-        env_str + "#{param[0]}=#{param[1]}\n"
-      end
-    end
-
     private
 
     def client
@@ -63,7 +61,7 @@ module Chambermaid
 
     def clear_params!
       @params = nil
-      @params_list = nil
+      @params_list = []
     end
   end
 end

data/lib/chambermaid/version.rb

@@ -1,3 +1,3 @@
 module Chambermaid
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chambermaid
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Miles Zimmerman
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-08-01 00:00:00.000000000 Z
+date: 2020-08-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-ssm
@@ -45,6 +45,8 @@ files:
 - chambermaid.gemspec
 - lib/chambermaid.rb
 - lib/chambermaid/base.rb
+- lib/chambermaid/environment.rb
+- lib/chambermaid/namespace.rb
 - lib/chambermaid/parameter_store.rb
 - lib/chambermaid/version.rb
 homepage: https://github.com/mileszim/chambermaid