aws-ssm-env 0.1.0

data/lib/aws-ssm-env/loader.rb

@@ -0,0 +1,67 @@
+require 'aws-ssm-env/fetchers/factory'
+require 'aws-ssm-env/naming_strategies/factory'
+
+module AwsSsmEnv
+  # このgemのエントリポイントとなるクラス。メイン処理を行う。
+  # AWS EC2 Parameters Storeからパラメータを取得してENVに書き込む。
+  #
+  # @author Ryohei Sonoda
+  # @since 0.1.0
+  class Loader
+    # メイン処理。引数の詳細は AwsSsmEnv#load を参照。
+    def self.load(**args)
+      new(args).load
+    end
+
+    def initialize(**args)
+      parse_options(args)
+      if @logger
+        @logger.debug("#{self.class.name} overwrite: #{@overwrite}")
+        @logger.debug("#{self.class.name} fetcher: #{@fetcher}")
+        @logger.debug("#{self.class.name} naming_strategy: #{@naming_strategy}")
+      end
+    end
+
+    def load
+      @fetcher.each do |parameter|
+        var_name = @naming_strategy.parse_name(parameter)
+        @logger.debug("#{self.class.name} #{parameter.name} parameter value into ENV['#{var_name}']") if @logger
+        send(@applier, var_name, parameter.value)
+      end
+    end
+
+    private
+
+    def parse_options(**options)
+      @logger = options[:logger]
+      @fetcher = AwsSsmEnv::FetcherFactory.create_fetcher(options)
+      @naming_strategy = AwsSsmEnv::NamingStrategyFactory.create_naming_strategy(options)
+      @overwrite = overwrite?(options)
+      if @overwrite
+        @applier = :apply!
+      else
+        @applier = :apply
+      end
+    end
+
+    # overwriteフラグが指定されているかどうかを返す。
+    def overwrite?(overwrite: nil, **)
+      if overwrite.nil?
+        false
+      else
+        overwrite.to_s.downcase == 'true'
+      end
+    end
+
+    def apply(name, value)
+      if ENV[name]
+        return
+      end
+      apply!(name, value)
+    end
+
+    def apply!(name, value)
+      ENV[name] = value
+    end
+  end
+end

data/lib/aws-ssm-env/naming_strategies/basename.rb

@@ -0,0 +1,19 @@
+require 'aws-ssm-env/naming_strategy'
+
+module AwsSsmEnv
+  # 環境変数名にパラメータ名の階層表現のbasenameを利用するようにするNamingStrategy実装クラス。
+  # AwsSsmEnv#load で`naming`を指定しなかった場合にはこのクラスのインスタンスが利用される。
+  # 例えば、`/path/to/ENV_NAME`というパラメータ名であればENV['ENV_NAME']にパラメータ値がインジェクションされる。
+  #
+  # @author Ryohei Sonoda
+  # @since 0.1.0
+  class BasenameNamingStrategy < NamingStrategy
+    # @see AwsSsmEnv::NamingStrategy#parse_name
+    #
+    # パラメータ名の最後の階層を変数名として返す。
+    # @return [String] 環境変数名
+    def parse_name(parameter)
+      File.basename(parameter.name)
+    end
+  end
+end

data/lib/aws-ssm-env/naming_strategies/factory.rb

@@ -0,0 +1,68 @@
+require 'aws-ssm-env/naming_strategy'
+
+module AwsSsmEnv
+  # 環境変数名を導出するためのNamingStrategyクラスを取得もしくは生成するファクトリクラス。
+  #
+  # @author Ryohei Sonoda
+  # @since 0.1.0
+  class NamingStrategyFactory
+    BASENAME_STRATEGY = :basename
+    SNAKE_CASE_STRATEGY = :snakecase
+
+    class << self
+      # 環境変数名を導出するためのNamingStrategyクラスを取得もしくは生成する。
+      #
+      # @param [Hash] args AwsSsmEnv#load に渡された引数がそのまま渡される。
+      # @option args [Symbol, AwsSsmEnv::NamingStrategy, Object] naming
+      #   引数の詳細は AwsSsmEnv#load の説明を参照。
+      def create_naming_strategy(**args)
+        naming_strategy = args[:naming]
+        if naming_strategy.nil?
+          return default_strategy(args)
+        end
+        case naming_strategy
+        when BASENAME_STRATEGY
+          create_basename_strategy(args)
+        when SNAKE_CASE_STRATEGY
+          create_snakecase_strategy(args)
+        else
+          unknown_naming_strategy(naming_strategy)
+        end
+      end
+
+      private
+
+      def default_strategy(**args)
+        create_basename_strategy(args)
+      end
+
+      def create_basename_strategy(**args)
+        require 'aws-ssm-env/naming_strategies/basename'
+        AwsSsmEnv::BasenameNamingStrategy.new(args)
+      end
+
+      def create_snakecase_strategy(**args)
+        require 'aws-ssm-env/naming_strategies/snakecase'
+        AwsSsmEnv::SnakeCaseNamingStrategy.new(args)
+      end
+
+      def unknown_naming_strategy(naming_strategy)
+        unless naming_strategy_instance?(naming_strategy)
+          raise ArgumentError, 'Possible values for :naming are either :basename, :snakecase, ' \
+                + '"AwsSsmEnv::NamingStrategy" implementation class, an object with "parse_name" method.'
+        end
+        naming_strategy
+      end
+
+      def naming_strategy_instance?(object)
+        if object.is_a?(AwsSsmEnv::NamingStrategy)
+          true
+        elsif object.respond_to?(:parse_name)
+          true
+        else
+          false
+        end
+      end
+    end
+  end
+end

data/lib/aws-ssm-env/naming_strategies/snakecase.rb

@@ -0,0 +1,58 @@
+require 'aws-ssm-env/naming_strategy'
+
+module AwsSsmEnv
+  # パラメータ名の階層表現をスネークケースに変換した値を環境変数名とする。
+  # 例えば、`removed_prefix`が`/path`で`/path/to/environment_name`というパラメータ名なら
+  # ENV['TO_ENVIRONMENT_NAME']にパラメータ値がインジェクションされる。
+  #
+  # @author Ryohei Sonoda
+  # @since 0.1.0
+  class SnakeCaseNamingStrategy < NamingStrategy
+    # ここの引数は AwsSsmEnv#load の呼び出し時に渡された引数がそのまま渡される。
+    #
+    # @param [Hash] args AwsSsmEnv#load の呼び出し時に渡された引数。
+    # @option args [String] :removed_prefix
+    #   パラメータ名から除去するプレフィクス。この文字列は導出される環境変数名に含まない。
+    #   :removed_prefixが指定されておらず、:begins_with または :path が指定されていた場合はそれを利用する。 TODO: AwsSsmEnv#loadとREADMEに反映
+    # @option args [String, Regexp] :delimiter
+    #   アンダースコアに変換する区切り文字。デフォルトはスラッシュ('/')。 TODO: AwsSsmEnv#loadとREADMEに反映
+    def initialize(**args)
+      @logger = args[:logger]
+      @delimiter = detect_delimiter(args)
+      removed_prefix = detect_prefix(args).sub(%r{/\z}, '')
+      @removed_prefix = /\A#{Regexp.escape(removed_prefix)}/
+      @logger.debug("#{self.class.name} removed_prefix is #{@removed_prefix}") if @logger
+    end
+
+    # @see AwsSsmEnv::NamingStrategy#parse_name
+    #
+    # パラメータ名からプレフィクスを除去してパス区切りをアンダースコアに変換後、大文字にして返す。
+    # @return [String] 環境変数名
+    def parse_name(parameter)
+      name_without_prefix = parameter.name.gsub(@removed_prefix, '')
+      name_without_prefix.gsub(@delimiter, '_').upcase
+    end
+
+    private
+
+    def detect_delimiter(**args)
+      if args[:delimiter].nil?
+        '/'
+      else
+        args[:delimiter]
+      end
+    end
+
+    def detect_prefix(**args)
+      if args[:removed_prefix]
+        args[:removed_prefix]
+      elsif args[:begins_with]
+        args[:begins_with]
+      elsif args[:path]
+        args[:path]
+      else
+        ''
+      end
+    end
+  end
+end

data/lib/aws-ssm-env/naming_strategy.rb

@@ -0,0 +1,23 @@
+module AwsSsmEnv
+  # パラメータの値を設定する環境変数名を決定するためのStrategyクラス。
+  # 実装クラスを AwsSsmEnv#load の引数で`naming`パラメータとして渡すことにより
+  # インジェクションされる環境変数名の命名ルールを切り替えられるようにする。
+  #
+  # @abstract
+  # @author Ryohei Sonoda
+  # @since 0.1.0
+  class NamingStrategy
+    # ここの引数は AwsSsmEnv#load の呼び出し時に渡された引数がそのまま渡される。
+    # サブクラスでは必要に応じて使う引数をインスタンス変数に保持しておく。
+    #
+    # @param [Hash] ** AwsSsmEnv#load の呼び出し時に渡された引数。
+    def initialize(**); end
+
+    # パラメータから環境変数名を導出するメソッド。
+    # @abstract
+    # @return [String] 環境変数名
+    def parse_name(_parameter)
+      raise NotImplementedError, 'parse_name'
+    end
+  end
+end

data/lib/aws-ssm-env/version.rb

@@ -0,0 +1,3 @@
+module AwsSsmEnv
+  VERSION = '0.1.0'.freeze
+end

metadata

@@ -0,0 +1,156 @@
+--- !ruby/object:Gem::Specification
+name: aws-ssm-env
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Ryohei Sonoda
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-12-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ssm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+- !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: rspec
+  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: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.48.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.48.1
+- !ruby/object:Gem::Dependency
+  name: rubocop-rspec
+  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: simplecov
+  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: simplecov-console
+  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: Loads environment variables from AWS EC2 System Parameters.
+email:
+- ryohei-sonoda@m3.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- README_en.md
+- lib/aws-ssm-env.rb
+- lib/aws-ssm-env/fetcher.rb
+- lib/aws-ssm-env/fetchers/begins_with.rb
+- lib/aws-ssm-env/fetchers/factory.rb
+- lib/aws-ssm-env/fetchers/path.rb
+- lib/aws-ssm-env/loader.rb
+- lib/aws-ssm-env/naming_strategies/basename.rb
+- lib/aws-ssm-env/naming_strategies/factory.rb
+- lib/aws-ssm-env/naming_strategies/snakecase.rb
+- lib/aws-ssm-env/naming_strategy.rb
+- lib/aws-ssm-env/version.rb
+homepage: https://github.com/sonodar/aws-ssm-env-ruby
+licenses:
+- Apache License 2.0
+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.6.13
+signing_key: 
+specification_version: 4
+summary: Loads environment variables from AWS EC2 System Parameters.
+test_files: []