path: root/lib/puppet/external/event-loop/better-definers.rb

## better-definers.rb --- better attribute and method definers
# Copyright (C) 2005  Daniel Brockman

# This program is free software; you can redistribute it
# and/or modify it under the terms of the GNU General Public
# License as published by the Free Software Foundation;
# either version 2 of the License, or (at your option) any
# later version.

# This file is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty
# of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
# See the GNU General Public License for more details.

# You should have received a copy of the GNU General Public
# License along with this program; if not, write to the Free
# Software Foundation, 51 Franklin Street, Fifth Floor,
# Boston, MA 02110-1301, USA.

class Symbol
    def predicate?
        to_s.include? "?" end
    def imperative?
        to_s.include? "!" end
    def writer?
        to_s.include? "=" end

    def punctuated?
        predicate? or imperative? or writer? end
    def without_punctuation
        to_s.delete("?!=").to_sym end

    def predicate
        without_punctuation.to_s + "?" end
    def imperative
        without_punctuation.to_s + "!" end
    def writer
        without_punctuation.to_s + "=" end
end

class Hash
    def collect! (&block)
        replace Hash[*collect(&block).flatten]
    end

    def flatten
        to_a.flatten
    end
end

module Kernel
    def returning (value)
        yield value ; value
    end
end

class Module
    def define_hard_aliases (name_pairs)
        for new_aliases, existing_name in name_pairs do
            new_aliases.kind_of? Array or new_aliases = [new_aliases]
            for new_alias in new_aliases do
                alias_method(new_alias, existing_name)
            end
        end
    end

    def define_soft_aliases (name_pairs)
        for new_aliases, existing_name in name_pairs do
            new_aliases.kind_of? Array or new_aliases = [new_aliases]
            for new_alias in new_aliases do
                class_eval %{def #{new_alias}(*args, &block)
                            #{existing_name}(*args, &block) end}
            end
        end
    end

    define_soft_aliases \
        :define_hard_alias => :define_hard_aliases,
        :define_soft_alias => :define_soft_aliases

    # This method lets you define predicates like :foo?,
    # which will be defined to return the value of @foo.
    def define_readers (*names)
        for name in names.map { |x| x.to_sym } do
            if name.punctuated?
                # There's no way to define an efficient reader whose
                # name is different from the instance variable.
                class_eval %{def #{name} ; @#{name.without_punctuation} end}
            else
                # Use `attr_reader' to define an efficient method.
                attr_reader(name)
            end
        end
    end

    def writer_defined? (name)
        method_defined? name.to_sym.writer
    end

    # If you pass a predicate symbol :foo? to this method, it'll first
    # define a regular writer method :foo, without a question mark.
    # Then it'll define an imperative writer method :foo! as a shorthand
    # for setting the property to true.
    def define_writers (*names, &body)
        for name in names.map { |x| x.to_sym } do
            if block_given?
                define_method(name.writer, &body)
            else
                attr_writer(name.without_punctuation)
            end
            if name.predicate?
                class_eval %{def #{name.imperative}
                           self.#{name.writer} true end}
            end
        end
    end

    define_soft_aliases \
        :define_reader => :define_readers,
        :define_writer => :define_writers

    # We don't need a singular alias for `define_accessors',
    # because it always defines at least two methods.

    def define_accessors (*names)
        define_readers(*names)
        define_writers(*names)
    end

    def define_opposite_readers (name_pairs)
        name_pairs.collect! { |k, v| [k.to_sym, v.to_sym] }
        for opposite_name, name in name_pairs do
            define_reader(name) unless method_defined? name
            class_eval %{def #{opposite_name} ; not #{name} end}
        end
    end

    def define_opposite_writers (name_pairs)
        name_pairs.collect! { |k, v| [k.to_sym, v.to_sym] }
        for opposite_name, name in name_pairs do
            define_writer(name) unless writer_defined? name
            class_eval %{def #{opposite_name.writer} x
                         self.#{name.writer} !x end}
            class_eval %{def #{opposite_name.imperative}
                         self.#{name.writer} false end}
        end
    end

    define_soft_aliases \
        :define_opposite_reader => :define_opposite_readers,
        :define_opposite_writer => :define_opposite_writers

    def define_opposite_accessors (name_pairs)
        define_opposite_readers name_pairs
        define_opposite_writers name_pairs
    end

    def define_reader_with_opposite (name_pair, &body)
        name, opposite_name = name_pair.flatten.collect { |x| x.to_sym }
        define_method(name, &body)
        define_opposite_reader(opposite_name => name)
    end

    def define_writer_with_opposite (name_pair, &body)
        name, opposite_name = name_pair.flatten.collect { |x| x.to_sym }
        define_writer(name, &body)
        define_opposite_writer(opposite_name => name)
    end

  public :define_method

    def define_methods (*names, &body)
        names.each { |name| define_method(name, &body) }
    end

    def define_private_methods (*names, &body)
        define_methods(*names, &body)
        names.each { |name| private name }
    end

    def define_protected_methods (*names, &body)
        define_methods(*names, &body)
        names.each { |name| protected name }
    end

    def define_private_method (name, &body)
        define_method(name, &body)
        private name
    end

    def define_protected_method (name, &body)
        define_method(name, &body)
        protected name
    end
end

class ImmutableAttributeError < StandardError
    def initialize (attribute=nil, message=nil)
        super message
        @attribute = attribute
    end

    define_accessors :attribute

    def to_s
        if @attribute and @message
            "cannot change the value of `#@attribute': #@message"
        elsif @attribute
            "cannot change the value of `#@attribute'"
        elsif @message
            "cannot change the value of attribute: #@message"
        else
            "cannot change the value of attribute"
        end
    end
end

class Module
    # Guard each of the specified attributes by replacing the writer
    # method with a proxy that asks the supplied block before proceeding
    # with the change.
    #
    # If it's okay to change the attribute, the block should return
    # either nil or the symbol :mutable.  If it isn't okay, the block
    # should return a string saying why the attribute can't be changed.
    # If you don't want to provide a reason, you can have the block
    # return just the symbol :immutable.
    def guard_writers(*names, &predicate)
        for name in names.map { |x| x.to_sym } do
            define_hard_alias("__unguarded_#{name.writer}" => name.writer)
            define_method(name.writer) do |new_value|
                case result = predicate.call
                when :mutable, nil
                    __send__("__unguarded_#{name.writer}", new_value)
                when :immutable
                    raise ImmutableAttributeError.new(name)
                else
                    raise ImmutableAttributeError.new(name, result)
                end
            end
        end
    end

    def define_guarded_writers (*names, &block)
        define_writers(*names)
        guard_writers(*names, &block)
    end

    define_soft_alias :guard_writer => :guard_writers
    define_soft_alias :define_guarded_writer => :define_guarded_writers
end

if __FILE__ == $0
    require "test/unit"

    class DefineAccessorsTest < Test::Unit::TestCase
        def setup
            @X = Class.new
            @Y = Class.new @X
            @x = @X.new
            @y = @Y.new
        end

        def test_define_hard_aliases
            @X.define_method(:foo) { 123 }
            @X.define_method(:baz) { 321 }
            @X.define_hard_aliases :bar => :foo, :quux => :baz
            assert_equal @x.foo, 123
            assert_equal @x.bar, 123
            assert_equal @y.foo, 123
            assert_equal @y.bar, 123
            assert_equal @x.baz, 321
            assert_equal @x.quux, 321
            assert_equal @y.baz, 321
            assert_equal @y.quux, 321
            @Y.define_method(:foo) { 456 }
            assert_equal @y.foo, 456
            assert_equal @y.bar, 123
            @Y.define_method(:quux) { 654 }
            assert_equal @y.baz, 321
            assert_equal @y.quux, 654
        end

        def test_define_soft_aliases
            @X.define_method(:foo) { 123 }
            @X.define_method(:baz) { 321 }
            @X.define_soft_aliases :bar => :foo, :quux => :baz
            assert_equal @x.foo, 123
            assert_equal @x.bar, 123
            assert_equal @y.foo, 123
            assert_equal @y.bar, 123
            assert_equal @x.baz, 321
            assert_equal @x.quux, 321
            assert_equal @y.baz, 321
            assert_equal @y.quux, 321
            @Y.define_method(:foo) { 456 }
            assert_equal @y.foo, @y.bar, 456
            @Y.define_method(:quux) { 654 }
            assert_equal @y.baz, 321
            assert_equal @y.quux, 654
        end

        def test_define_readers
            @X.define_readers :foo, :bar
            assert !@x.respond_to?(:foo=)
            assert !@x.respond_to?(:bar=)
            @x.instance_eval { @foo = 123 ; @bar = 456 }
            assert_equal @x.foo, 123
            assert_equal @x.bar, 456
            @X.define_readers :baz?, :quux?
            assert !@x.respond_to?(:baz=)
            assert !@x.respond_to?(:quux=)
            @x.instance_eval { @baz = false ; @quux = true }
            assert !@x.baz?
            assert @x.quux?
        end

        def test_define_writers
            assert !@X.writer_defined?(:foo)
            assert !@X.writer_defined?(:bar)
            @X.define_writers :foo, :bar
            assert @X.writer_defined?(:foo)
            assert @X.writer_defined?(:bar)
            assert @X.writer_defined?(:foo=)
            assert @X.writer_defined?(:bar=)
            assert @X.writer_defined?(:foo?)
            assert @X.writer_defined?(:bar?)
            assert !@x.respond_to?(:foo)
            assert !@x.respond_to?(:bar)
            @x.foo = 123
            @x.bar = 456
            assert_equal @x.instance_eval { @foo }, 123
            assert_equal @x.instance_eval { @bar }, 456
            @X.define_writers :baz?, :quux?
            assert !@x.respond_to?(:baz?)
            assert !@x.respond_to?(:quux?)
            @x.baz = true
            @x.quux = false
            assert_equal @x.instance_eval { @baz }, true
            assert_equal @x.instance_eval { @quux }, false
        end

        def test_define_accessors
            @X.define_accessors :foo, :bar
            @x.foo = 123 ; @x.bar = 456
            assert_equal @x.foo, 123
            assert_equal @x.bar, 456
        end

        def test_define_opposite_readers
            @X.define_opposite_readers :foo? => :bar?, :baz? => :quux?
            assert !@x.respond_to?(:foo=)
            assert !@x.respond_to?(:bar=)
            assert !@x.respond_to?(:baz=)
            assert !@x.respond_to?(:quux=)
            @x.instance_eval { @bar = true ; @quux = false }
            assert !@x.foo?
            assert @x.bar?
            assert @x.baz?
            assert !@x.quux?
        end

        def test_define_opposite_writers
            @X.define_opposite_writers :foo? => :bar?, :baz => :quux
        end
    end
end