proctor 0.0.3 → 0.0.4

data/lib/proctor/command/helpers/command.rb

@@ -0,0 +1,27 @@
+module Proctor
+  module Command
+    module Helpers
+
+      module Command
+
+        def check_number_of_args(args, target_number, message, opts = {})
+          opts_txt = options_msg(opts)
+          if args.length != target_number
+            raise InvalidNumberOfArgs, "Usage '#{message}' #{opts_txt}"
+          end
+        end
+
+        def options_msg(hash)
+          hash.to_a.map do |i|
+            "\n  #{i[0]} [#{i[1].join('|')}]"
+          end.join
+        end
+
+        class InvalidNumberOfArgs < RuntimeError
+        end
+
+      end
+
+    end
+  end
+end

data/lib/proctor/command/render_group_cmd.rb

@@ -0,0 +1,39 @@
+require_relative "./helpers/command"
+require_relative "../resource/erb"
+require_relative "../resource/lookup"
+
+module Proctor
+  module Command
+    class RenderGroupCmd
+
+      include Helpers::Command
+      include Proctor::Resource::Erb
+      include Proctor::Resource::Lookup
+
+      def initialize(global, options, args)
+        @manager = args[0]
+        @node    = args[1]
+        @env     = options['env']
+        opts = {'MANAGERS' => @env.managers.names, 'NODES' => @env.nodes.names}
+        check_number_of_args(args, 2, 'render_group MANAGER NODE', opts)
+      end
+
+      def render_string
+        manager  = lookup_valid_manager(@manager, @env)
+        node     = lookup_valid_node(@node, @env)
+        template = lookup_valid_group_template(@manager, @node, @env)
+        services = lookup_valid_services_hash(@node, @env)
+        render_group_template(manager, node, services, template, @env)
+      end
+
+      def group
+        puts render_string
+      end
+
+      def render
+        puts render_string
+      end
+
+    end
+  end
+end

data/lib/proctor/command/render_service_cmd.rb

@@ -0,0 +1,34 @@
+require_relative "./helpers/command"
+require_relative "../resource/erb"
+require_relative "../resource/lookup"
+
+module Proctor
+  module Command
+    class RenderServiceCmd
+
+      include Helpers::Command
+      include Proctor::Resource::Erb
+      include Proctor::Resource::Lookup
+
+      def initialize(global, options, args)
+        @manager = args[0]
+        @service = args[1]
+        @env     = options['env']
+        opts = {'MANAGERS' => @env.managers.names, 'SERVICES' => @env.services.names}
+        check_number_of_args(args, 2, 'render_service MANAGER SERVICE', opts)
+      end
+
+      def render
+        puts render_string
+      end
+
+      def render_string
+        manager  = lookup_valid_manager(@manager, @env)
+        service  = lookup_valid_service(@service, @env)
+        template = lookup_valid_service_template(@manager, @service, @env)
+        render_service_template(manager, service, template, @env)
+      end
+
+    end
+  end
+end

data/lib/proctor/command/start_cmd.rb

@@ -0,0 +1,70 @@
+require_relative "./helpers/command"
+require_relative "../resource/erb"
+require_relative "../resource/lookup"
+
+module Proctor
+  module Command
+    class StartCmd
+
+      include Helpers::Command
+      include Proctor::Resource::Erb
+      include Proctor::Resource::Lookup
+
+      def initialize(global, options, args)
+        @env       = options['env']
+        @manager   = args[0]
+        @node      = args[1]
+        opts = {'MANAGERS' => @env.managers.names, 'NODES' => @env.nodes.names}
+        check_number_of_args(args, 2, 'start MANAGER NODE', opts)
+      end
+
+      def start
+        env = @env
+        _manager = lookup_valid_manager(@manager, @env)
+        #update_manager_with_cl_options(manager)
+        manager = erb_ify(_manager, {}, env)
+        node = lookup_valid_node(@node, @env)
+        if manager.data["group_data"]
+          start_group(manager)
+        end
+        if manager.data["service_data"]
+          node.services.each do |service_name|
+            service = lookup_valid_service(service_name, @env)
+            start_service(manager, service, @env)
+          end
+        end
+      end
+
+      private
+
+      def start_group(manager)
+        export_directory = manager.data["export_directory"]
+        start_command = manager.data["group_data"]["start_command"]
+        begin
+          cmd = "cd #{export_directory} ; #{start_command}"
+          system cmd
+        rescue
+          msg = "start problem"
+          raise StartError, msg
+        end
+      end
+
+      def start_service(manager, service, env)
+        export_directory = manager.data["export_directory"]
+        start_command = manager.data["service_data"]["start_command"]
+        begin
+          cmd = "cd #{export_directory} ; #{start_command}"
+          system cmd
+        rescue
+          msg = "start problem"
+          raise StartError, msg
+        end
+      end
+    end
+
+    class StartError < RuntimeError
+    end
+
+  end
+end
+

data/lib/proctor/config/app_file.rb

@@ -0,0 +1,54 @@
+require 'yaml'
+require_relative '../util/helpers'
+
+module Proctor
+  module Config
+    class AppFile
+
+      include Proctor::Config
+      include Util::Helpers
+
+      attr_reader :data, :handle, :name
+
+      def initialize(file_path)
+        validate_file_existence(file_path)
+        @name    = file_path
+        @data    = load_data(file_path)
+        @handle  = md5_handle(file_path)
+        validate_data_format(file_path)
+      end
+
+      private
+
+      def load_data(file)
+        begin
+          @data = YAML.load_file(file)
+        rescue Psych::SyntaxError
+          error_msg = "invalid file format (expecting YAML - #{file})"
+          raise InvalidYamlException.new, error_msg
+        end
+      end
+
+      def validate_data_format(file)
+        raise InvalidDataType, "expecting hash data (#{file})" unless @data.is_a?(Hash)
+      end
+
+      def validate_file_existence(file)
+        error_msg = "missing app file (#{file})"
+        raise MissingAppFileException, error_msg unless File.exist?(file)
+      end
+
+    end
+
+    class MissingAppFileException < RuntimeError
+    end
+
+    class InvalidYamlException < RuntimeError
+    end
+
+    class InvalidDataType < RuntimeError
+    end
+
+  end
+
+end

data/lib/proctor/config/app_files.rb

@@ -0,0 +1,50 @@
+require 'hash_deep_merge'
+require_relative './app_file'
+
+module Proctor
+  module Config
+    class AppFiles
+
+      def initialize(params)
+        @all = path_list(params).map {|f| AppFile.new(f)}
+      end
+
+      def all
+        @all.sort {|a,b| a.name <=> b.name}
+      end
+
+      def merged_data
+        @all.reduce({}) { |a,v| a.deep_merge(v.data) }
+      end
+
+      def path_list(params)
+        gem_app_path + cwd_app_path + commandline_app_path(params)
+      end
+
+      def find_by_handle(handle)
+        @all.find {|app_file| app_file.handle == handle}
+      end
+
+      private
+
+      def commandline_app_path(params)
+        cl_name = params[0][:f]
+        cl_name.nil? ? [] : [cl_name]
+      end
+
+      def cwd_app_path
+        tgt_file = "#{Dir.pwd}/Proctorfile"
+        File.exist?(tgt_file) ? [tgt_file] : []
+      end
+
+      def gem_app_path
+        ["#{gem_base_dir}/proctor/Proctorfile"]
+      end
+
+      def gem_base_dir
+        File.expand_path('../../../', File.dirname(__FILE__))
+      end
+
+    end
+  end
+end

data/lib/proctor/config/env.rb

@@ -0,0 +1,35 @@
+f1 = %w(./template_files ./app_files)
+f2 = %w(../managers ../services ../nodes)
+
+(f1 + f2).each {|f| require_relative f}
+
+module Proctor
+
+  module Config
+
+    class Env
+
+      attr_reader :template_files
+      attr_reader :app_files, :managers, :services, :nodes
+      attr_reader :app
+
+      def initialize(params)
+        @template_files = TemplateFiles.new(params)
+        @app_files      = AppFiles.new(params)
+        @managers       = Managers.new(@app_files, params)
+        @services       = Services.new(@app_files, params)
+        @nodes          = Nodes.new(@app_files, params)
+        @app            = {}
+        set_app_metadata
+      end
+
+      def set_app_metadata
+        @app['name']   = Dir.pwd.split('/').last
+        @app['gemdir'] = File.expand_path('../../../', File.dirname(__FILE__))
+      end
+
+    end
+
+  end
+
+end

data/lib/proctor/config/template_file.rb

@@ -0,0 +1,41 @@
+require 'yaml'
+require_relative '../util/helpers'
+
+module Proctor
+  module Config
+    class TemplateFile
+
+      include Proctor::Config
+      include Util::Helpers
+
+      attr_reader :name, :path, :handle
+
+      def initialize(file_path)
+        validate_file_existence(file_path)
+        @name   = file_name(file_path)
+        @path   = file_path
+        @handle = md5_handle(file_path)
+      end
+
+      def text
+        File.read(@path)
+      end
+
+      private
+
+      def file_name(file_path)
+        File.basename(file_path, '.erb')
+      end
+
+      def validate_file_existence(file)
+        error_msg = "missing template file (#{file})"
+        raise MissingTemplateFileException, error_msg unless File.exist?(file)
+      end
+
+    end
+
+    class MissingTemplateFileException < RuntimeError
+    end
+
+  end
+end

data/lib/proctor/config/template_files.rb

@@ -0,0 +1,39 @@
+require_relative './template_file'
+
+module Proctor
+  module Config
+    class TemplateFiles
+
+      def initialize (params)
+        @all = path_list(params).map {|f| TemplateFile.new(f)}
+      end
+
+      def all
+        @all.sort {|a,b| a.name <=> b.name}
+      end
+
+      def find_by_name(file_name)
+        @all.find {|template_file| template_file.name == file_name}
+      end
+
+      def find_by_handle(handle)
+        @all.find {|template_file| template_file.handle == handle}
+      end
+
+      def names
+        @all.map {|template| template.name}
+      end
+
+      private
+
+      def path_list(params)
+        Dir.glob("#{gem_base_dir}/proctor/templates/*").map {|f| File.expand_path(f)}
+      end
+
+      def gem_base_dir
+        File.dirname(File.expand_path('../../../', __FILE__))
+      end
+
+    end
+  end
+end

data/lib/proctor/manager.rb

@@ -1,11 +1,44 @@
+require_relative './util/helpers'
+
 module Proctor
   class Manager
-    attr_reader :input_hash, :opts
 
-    def initialize(hash = {})
-      @input_hash = hash
+    include Proctor::Util::Helpers
+
+    attr_reader :name, :data, :handle
+
+    def initialize(name, data)
+      @name = name
+      @data = data
+      @handle = md5_handle(name)
+      validate_required_manager_fields(data)
     end
 
+    private
+
+    def validate_required_manager_fields(manager_hash)
+      required_fields = %w(export_directory)
+      required_fields.each do |field|
+        if manager_hash[field].nil?
+          raise MissingManagerField, "missing required service field (#{field})"
+        end
+      end
+    end
+
+    def validate_export_directory(manager_hash)
+      dir = manager_hash['export_directory']
+      unless Dir.exist?(dir)
+        raise MissingExportDirectory, "missing export directory (#{dir})"
+      end
+    end
+
+  end
+
+  class MissingManagerField < RuntimeError
   end
+
+  class MissingExportDirectory < RuntimeError
+  end
+
 end