diff options
author | Patrick J Cherry <patrick@bytemark.co.uk> | 2011-11-10 16:58:02 +0000 |
---|---|---|
committer | Patrick J Cherry <patrick@bytemark.co.uk> | 2011-11-10 16:58:02 +0000 |
commit | 139fc8ae74c5019003e1b4fa350240f8e9fedebc (patch) | |
tree | 105f82ae90082dfa16b94a4b200e37570e712453 | |
parent | e272804ec875ab1562a5ecc3f58bcaf135fe6fbb (diff) |
Changed mauve config to allow including directory and filenames.
-rw-r--r-- | bin/mauveconsole | 5 | ||||
-rw-r--r-- | bin/mauveserver | 9 | ||||
-rw-r--r-- | lib/object_builder.rb | 139 |
3 files changed, 107 insertions, 46 deletions
diff --git a/bin/mauveconsole b/bin/mauveconsole index 169ca54..211eaab 100644 --- a/bin/mauveconsole +++ b/bin/mauveconsole @@ -89,7 +89,6 @@ require 'pp' def error(msg) logger = Log4r::Logger['Mauve'] - logger.error logger.error "*** Error: #{msg}" logger.error "*** For help, type: #{$0} -h" @@ -139,7 +138,9 @@ module Mauve end begin - Mauve::Configuration.current = Mauve::ConfigurationBuilder.load(configuration_file) + builder = Mauve::ConfigurationBuilder.new + builder.include_file(configuration_file) + Mauve::Configuration.current = builder.result Mauve::Server.instance.setup rescue StandardError => ex error ex diff --git a/bin/mauveserver b/bin/mauveserver index 77b372d..b0c3573 100644 --- a/bin/mauveserver +++ b/bin/mauveserver @@ -130,7 +130,9 @@ require 'mauve/configuration_builder' require 'mauve/configuration_builders' begin - Mauve::Configuration.current = Mauve::ConfigurationBuilder.load(configuration_file) + builder = Mauve::ConfigurationBuilder.new + builder.include_file(configuration_file) + Mauve::Configuration.current = builder.result rescue StandardError => ex error ex end @@ -149,8 +151,9 @@ end # Reload configuration # begin - new_config = Mauve::ConfigurationBuilder.load(configuration_file) - Mauve::Configuration.current = new_config + builder = Mauve::ConfigurationBuilder.new + builder.include_file(configuration_file) + Mauve::Configuration.current = builder.result rescue BuildException => ex Mauve::Server.instance.logger.error "Reconfiguration failed: #{ex.to_s}. Sticking with old one." end diff --git a/lib/object_builder.rb b/lib/object_builder.rb index 8f338ff..e012b5a 100644 --- a/lib/object_builder.rb +++ b/lib/object_builder.rb @@ -65,9 +65,103 @@ class ObjectBuilder @@sequence += 1 "anon.#{Time.now.to_i}.#{@@sequence}" end + + # + # Merge a file into self. If the file is a relative path, it is relative to + # the file from which it has been included. + # + # @param [String] file The filename to include + # + # @raise [BuildException] When an expected exception is raised + # @raise [NameError] + # @raise [SyntaxError] + # @raise [ArgumentError] + # + # @returns [ObjectBuilder] self + def include_file(file) + # + # Share the current directory with all instances of this class. + # + @@directory ||= nil + + # + # Do we have an absolute path? + # + file = File.join(@@directory, file) if !@@directory.nil? and '/' != file[0,1] + + # + # Resolve the filename. + # + file = File.expand_path(file) + + # + # Save the current wd + # + oldwd = @@directory + + # + # Set the new one + # + @@directory = File.dirname(file) + + # + # Read the file and eval it. + # + instance_eval(File.read(file), file) + + # + # Reset back to where we were. + # + @@directory = oldwd + + self + rescue NameError, NoMethodError => ex + # + # Ugh. Catch NameError and re-raise as a BuildException + # + if ex.backtrace.find{|l| l =~ /^#{file}:(\d+):/} + build_ex = BuildException.new "Unknown word `#{ex.name}' in #{file} at line #{$1}" + build_ex.set_backtrace ex.backtrace + raise build_ex + else + raise ex + end + rescue SyntaxError, ArgumentError => ex + if ex.backtrace.find{|l| l =~ /^#{file}:(\d+):/} + build_ex = BuildException.new "#{ex.message} in #{file} at line #{$1}" + build_ex.set_backtrace ex.backtrace + raise build_ex + else + raise ex + end + end + # + # Loads a stack of files in a directory, and merges them into the current object + # + # @params [String] dir Directory in which to search for files. + # @params [Regexp] regexp Regular expression for filename to include. + # + # @returns [ObjectBuilder] self + def include_directory(dir, regex = /^[a-zA-Z0-9_-]+\.conf$/) + files = [] + # + # Exceptions are caught by #include_file + # + Dir.entries(dir).sort.each do |entry| + next unless entry =~ regex + + file = File.join(dir,entry) + next unless File.file?(file) + + include_file(file) + end + + self + end + class << self - + # Defines a new builder # # @param [String] word The builder's name @@ -125,48 +219,11 @@ class ObjectBuilder @result.__send__("#{word}=".to_sym, true) end end - - # Loads a new file - # - # @param [String] file - def load(file) - parse(File.read(file), file) - end - - # Parses a string - # - # @param [String] string The string to parse - # @param [String] file The filename that the string was read from, for helpful error messages. - # - # @raise [BuildException] When an expected exception is raised - # @raise [NameError] - # @raise [SyntaxError] - # @raise [ArgumentError] - # - # @return [ObjectBuidler] The - def parse(string, file="string") + + def parse(s) builder = self.new - builder.instance_eval(string, file) + builder.instance_eval(s) builder.result - rescue NameError, NoMethodError => ex - # - # Ugh. Catch NameError and re-raise as a BuildException - # - if ex.backtrace.find{|l| l =~ /^#{file}:(\d+):/} - build_ex = BuildException.new "Unknown word `#{ex.name}' in #{file} at line #{$1}" - build_ex.set_backtrace ex.backtrace - raise build_ex - else - raise ex - end - rescue SyntaxError, ArgumentError => ex - if ex.backtrace.find{|l| l =~ /^#{file}:(\d+):/} - build_ex = BuildException.new "#{ex.message} in #{file} at line #{$1}" - build_ex.set_backtrace ex.backtrace - raise build_ex - else - raise ex - end end def inherited(*args) |