Contents

require 'psych'

module Canals
  class CanalOptionError < StandardError; end

  class CanalOptions
    BIND_ADDRESS = "127.0.0.1"
    attr_reader :name, :remote_host, :remote_port, :local_port, :env_name, :env

    def initialize(args)
      @args = validate?(args)
      @name = @args["name"]
      @remote_host = @args["remote_host"]
      @remote_port = @args["remote_port"]
      @local_port = @args["local_port"]
      @env_name = @args['env']
      @env = Canals.repository.environment(@env_name)
    end

    def bind_address
      return @args["bind_address"] if @args["bind_address"]
      return BIND_ADDRESS
    end

    def hostname
      get_env_var("hostname")
    end

    def user
      get_env_var("user")
    end

    def pem
      get_env_var("pem")
    end

    def proxy
      prxy = ""
      prxy += "-i #{pem} " if pem
      prxy += "#{user}@#{hostname}"
      prxy
    end

    def to_yaml
      Psych.dump(@args)
    end

    def to_hash
      @args.dup
    end

    private

    def validate?(args)
      vargs = args.dup
      raise CanalOptionError.new("Missing option: \"name\" in canal creation") if args["name"].nil?
      raise CanalOptionError.new("Missing option: \"remote_host\" in canal creation") if args["remote_host"].nil?
      raise CanalOptionError.new("Missing option: \"remote_port\" in canal creation") if args["remote_port"].nil?
      vargs["remote_port"] = vargs["remote_port"].to_i
      if vargs["local_port"].nil?
        vargs["local_port"] = vargs["remote_port"]
      else
        vargs["local_port"] = vargs["local_port"].to_i
      end
      vargs
    end

    def get_env_var(var)
      return @args[var] if @args[var]
      return @env.send var.to_sym if @env
      nil
    end

  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
canals-0.8.2	lib/canals/options.rb
canals-0.8.1	lib/canals/options.rb
canals-0.8.0	lib/canals/options.rb