gistfile1.rb

module Compiler::Plugins
  class StaticConstantSet < Compiler::Plugin
    register Compiler::Node::Call

    def self.run(compiler, call)
      if !call.object and
           call.method == :static and
           call.arguments.size == 1 and
           call.arguments.first.is? Compiler::Node::ConstSet
        cf = call.arguments.first
        begin
          value = constant_value(cf.value)
        rescue RuntimeError
          return call
        end
        compiler.plugin_data[:constants][cf.name] = value
        return cf
      end
      call
    end

    def self.constant_value(node)
      case node
      when Compiler::Node::False
        false
      when Compiler::Node::True
        true
      when Compiler::Node::Nil
        nil
      when Compiler::Node::NumberLiteral
        return node.value
      else
        raise "Not a constant value"
      end
    end
  end

  class StaticConstantRead < Compiler::Plugin
    register Compiler::Node::ConstFind

    def self.run(compiler, const_find)
      if val = compiler.plugin_data[:constants][const_find.name]
        new(val)
      else
        const_find
      end
    end

    def initialize(val)
      @value = val
    end

    def bytecode(g)
      g.push @value
    end
  end
end