rbhtml 0.1.0 → 0.2.0

data/lib/rbhtml.rb

@@ -19,6 +19,7 @@ class HTMLDocument
     @full_doc = @open_doc + @close_doc
     @index_dict = {}
     @index_id = 0
+    @count = {}
   end
   def create_element tag, inner, attribute=false, value=false
     if attribute and value
@@ -27,18 +28,31 @@ class HTMLDocument
       @open_doc += "<#{tag}>#{inner}</#{tag}>"
     end
     @index_id += 1
+    if @count.keys.include? tag
+      @count[tag] += 1
+    else
+      @count[tag] = 1
+    end
     refresh_full
     index_element tag, inner
   end
   def remove_element tag, inner
-    @open_doc.gsub!("<#{tag}>#{inner}", "")
+    if @open_doc.gsub!("<#{tag}>#{inner}</#{tag}>", "") == nil
+      return
+    else
+      @open_doc.gsub!("<#{tag}>#{inner}</#{tag}>", "")
+    end
     @close_doc.gsub!("</#{tag}>", "")
+    @count[tag] -= 1
     refresh_full
   end
   def remove_char char
     @open_doc.gsub!(">#{char}", ">")
     refresh_full
   end
+  def count tag
+    @count[tag]
+  end
   def to_s
     @full_doc
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rbhtml
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -11,8 +11,8 @@ bindir: bin
 cert_chain: []
 date: 2012-06-03 00:00:00.000000000 Z
 dependencies: []
-description: An HTML generator written in Ruby. Added attributes to create_element
-  method. Removed bug where if you create two elements, the elements overlap.
+description: An HTML generator written in Ruby. Added count method, which counts the
+  number of the specified elements in the HTMLDoc. Removed bug in remove_element.
 email: slmnwise@gmail.com
 executables: []
 extensions: []