generator.rb in watir-6.10.1

- old
+ new
@@ -1,115 +1,115 @@
-module Watir
-  module Generator
-    class Base
-
-      def generate(spec_url, io = StringIO.new)
-        @spec_url, @io = spec_url, io
-
-        extract_spec
-        cleanup_spec
-
-        write_header
-        write_class_defs
-        write_container_methods
-        write_footer
-
-        io
-      end
-
-      private
-
-      def generator
-        @generator ||= WebIDL::Generator.new(visitor)
-      end
-
-      def visitor
-        @visitor ||= visitor_class.new
-      end
-
-      def extractor
-        @extractor ||= extractor_class.new(@spec_url)
-      end
-
-      def extract_spec
-        @tag2interfaces    = extractor.process
-        @sorted_interfaces = extractor.sorted_interfaces
-
-        if extractor.errors.any?
-          raise "error extracting spec: #{extractor.errors.join("\n")}"
-        end
-      end
-
-      def cleanup_spec
-        ignored_tags.each do |tag|
-          @tag2interfaces.delete(tag)
-        end
-
-        ignored_interfaces.each do |interface|
-          @sorted_interfaces.reject! { |intf| intf.name == interface }
-        end
-
-        @sorted_interfaces.each do |intf|
-          intf.members.delete_if { |member| ignored_attributes.include?(member.name) }
-        end
-      end
-
-      def write_header
-        @io.puts "# Autogenerated from #{generator_implementation} specification. Edits may be lost."
-        @io.puts "module Watir"
-      end
-
-      def write_class_defs
-        @sorted_interfaces.each do |interface|
-          interface = generator.generate(interface)
-          unless interface.empty?
-            interface.gsub!(/^\s+\n/, '') # remove empty lines
-            @io.puts indent(interface)
-            @io.puts "\n"
-          end
-        end
-      end
-
-      def write_container_methods
-        @io.puts "\n"
-        @io.puts indent("module Container")
-
-        @tag2interfaces.sort.each do |tag, interfaces|
-          raise "multiple interfaces for tag #{tag.inspect}" if interfaces.map(&:name).uniq.size != 1
-
-          tag_string       = tag.inspect
-          singular         = Util.paramify(visitor.classify_regexp, tag)
-          plural           = singular.pluralize
-          element_class    = Util.classify(visitor.classify_regexp, interfaces.first.name)
-          collection_class = "#{element_class}Collection"
-
-          # visitor.visit_tag(tag, interfaces.first.name) !?
-          @io.puts indent(<<-CODE, 2)
-
-# @return [#{element_class}]
-def #{singular}(*args)
-  #{element_class}.new(self, extract_selector(args).merge(tag_name: #{tag_string}))
-end
-# @return [#{collection_class}]
-def #{plural}(*args)
-  #{collection_class}.new(self, extract_selector(args).merge(tag_name: #{tag_string}))
-end
-Watir.tag_to_class[#{tag.to_sym.inspect}] = #{element_class}
-
-CODE
-        end
-
-        @io.puts indent("end # Container")
-      end
-
-      def write_footer
-        @io.puts "end # Watir"
-      end
-
-      def indent(code, indent = 1)
-        indent_string = "  " * indent
-        code.split("\n").map { |line| line.empty? ? line : indent_string + line }.join("\n")
-      end
-
-    end # Base
-  end # Generator
-end # Watir
+module Watir
+  module Generator
+    class Base
+
+      def generate(spec_url, io = StringIO.new)
+        @spec_url, @io = spec_url, io
+
+        extract_spec
+        cleanup_spec
+
+        write_header
+        write_class_defs
+        write_container_methods
+        write_footer
+
+        io
+      end
+
+      private
+
+      def generator
+        @generator ||= WebIDL::Generator.new(visitor)
+      end
+
+      def visitor
+        @visitor ||= visitor_class.new
+      end
+
+      def extractor
+        @extractor ||= extractor_class.new(@spec_url)
+      end
+
+      def extract_spec
+        @tag2interfaces    = extractor.process
+        @sorted_interfaces = extractor.sorted_interfaces
+
+        if extractor.errors.any?
+          raise "error extracting spec: #{extractor.errors.join("\n")}"
+        end
+      end
+
+      def cleanup_spec
+        ignored_tags.each do |tag|
+          @tag2interfaces.delete(tag)
+        end
+
+        ignored_interfaces.each do |interface|
+          @sorted_interfaces.reject! { |intf| intf.name == interface }
+        end
+
+        @sorted_interfaces.each do |intf|
+          intf.members.delete_if { |member| ignored_attributes.include?(member.name) }
+        end
+      end
+
+      def write_header
+        @io.puts "# Autogenerated from #{generator_implementation} specification. Edits may be lost."
+        @io.puts "module Watir"
+      end
+
+      def write_class_defs
+        @sorted_interfaces.each do |interface|
+          interface = generator.generate(interface)
+          unless interface.empty?
+            interface.gsub!(/^\s+\n/, '') # remove empty lines
+            @io.puts indent(interface)
+            @io.puts "\n"
+          end
+        end
+      end
+
+      def write_container_methods
+        @io.puts "\n"
+        @io.puts indent("module Container")
+
+        @tag2interfaces.sort.each do |tag, interfaces|
+          raise "multiple interfaces for tag #{tag.inspect}" if interfaces.map(&:name).uniq.size != 1
+
+          tag_string       = tag.inspect
+          singular         = Util.paramify(visitor.classify_regexp, tag)
+          plural           = singular.pluralize
+          element_class    = Util.classify(visitor.classify_regexp, interfaces.first.name)
+          collection_class = "#{element_class}Collection"
+
+          # visitor.visit_tag(tag, interfaces.first.name) !?
+          @io.puts indent(<<-CODE, 2)
+
+# @return [#{element_class}]
+def #{singular}(*args)
+  #{element_class}.new(self, extract_selector(args).merge(tag_name: #{tag_string}))
+end
+# @return [#{collection_class}]
+def #{plural}(*args)
+  #{collection_class}.new(self, extract_selector(args).merge(tag_name: #{tag_string}))
+end
+Watir.tag_to_class[#{tag.to_sym.inspect}] = #{element_class}
+
+CODE
+        end
+
+        @io.puts indent("end # Container")
+      end
+
+      def write_footer
+        @io.puts "end # Watir"
+      end
+
+      def indent(code, indent = 1)
+        indent_string = "  " * indent
+        code.split("\n").map { |line| line.empty? ? line : indent_string + line }.join("\n")
+      end
+
+    end # Base
+  end # Generator
+end # Watir