lib/atom/entry.rb in atom-tools-0.9.2 vs lib/atom/entry.rb in atom-tools-0.9.3

@@ -3,10 +3,11 @@
 require "atom/element"
 require "atom/text"
 
 module Atom
   NS = "http://www.w3.org/2005/Atom"
+  PP_NS = "http://purl.org/atom/app#"
 
   # An individual entry in a feed. As an Atom::Element, it can be
   # manipulated using accessors for each of its child elements. You
   # should be able to set them using an instance of any class that
   # makes sense
@@ -31,35 +32,35 @@
   class Entry < Atom::Element
     # the master list of standard children and the types they map to
     element :id, String, true
     element :title, Atom::Text, true
     element :content, Atom::Content, true
-    
+
     element :rights, Atom::Text
     # element :source, Atom::Feed  # complicated, eg. serialization
-    
+
     element :authors, Atom::Multiple(Atom::Author)
     element :contributors, Atom::Multiple(Atom::Contributor)
-    
+
     element :categories, Atom::Multiple(Atom::Category)
     element :links, Atom::Multiple(Atom::Link)
-    
+
     element :published, Atom::Time
     element :updated, Atom::Time, true
-    
+
     element :summary, Atom::Text
 
     def initialize # :nodoc:
       super "entry"
-     
+
       # XXX I don't think I've ever actually used this
       yield self if block_given?
     end
 
     # parses XML into an Atom::Entry
-    # 
-    # +base+ is the absolute URI the document was fetched from 
+    #
+    # +base+ is the absolute URI the document was fetched from
     # (if there is one)
     def self.parse xml, base = ""
       if xml.respond_to? :to_atom_entry
         xml.to_atom_entry(base)
       elsif xml.respond_to? :read
@@ -78,15 +79,16 @@
     # (note that this is different from Atom::Feed#update!)
     def updated!
       self.updated = Time.now
     end
 
-    # categorize the entry based on a space-separated string
-    def tag_with string
-      return if string.nil?
+    # categorize the entry with each of an array or a space-separated
+    #   string
+    def tag_with tags
+      return unless tags
 
-      string.split.each do |tag|
+      (tags.is_a?(String) ? tags.split : tags).each do |tag|
         categories.new["term"] = tag
       end
     end
 
     # the @href of an entry's link[@rel="edit"]
@@ -100,10 +102,43 @@
       rescue
         nil
       end
     end
 
+    def draft
+      elem = REXML::XPath.first(extensions, "app:control/app:draft", {"app" => PP_NS})
+
+      elem and elem.text == "yes"
+    end
+
+    def draft= is_draft
+      nses = {"app" => PP_NS}
+      draft_e = REXML::XPath.first(extensions, "app:control/app:draft", nses)
+      control_e = REXML::XPath.first(extensions, "app:control", nses)
+
+      if is_draft and not draft
+        unless draft_e
+          unless control_e
+            control_e = REXML::Element.new("control")
+            control_e.add_namespace PP_NS
+
+            extensions << control_e
+          end
+
+          draft_e = REXML::Element.new("draft")
+          control_e << draft_e
+        end
+
+        draft_e.text = "yes"
+      elsif not is_draft and draft
+        draft_e.remove
+        control_e.remove if control_e.elements.empty?
+      end
+
+      is_draft
+    end
+
 # XXX this needs a test suite before it can be trusted.
 =begin
     # tests the entry's validity
     def valid?
       self.class.required.each do |element|
@@ -124,21 +159,21 @@
           return [ false, "no atom:content or atom:link[rel='alternate']" ]
       end
 
       alternates.each do |link|
         if alternates.find do |x|
-          not x == link and 
-            x["type"] == link["type"] and 
+          not x == link and
+            x["type"] == link["type"] and
             x["hreflang"] == link["hreflang"]
           end
-         
+
           return [ false, 'more than one atom:link with a rel attribute value of "alternate" that has the same combination of type and hreflang attribute values.' ]
         end
       end
 
       type = @content["type"]
 
-      base64ed = (not ["", "text", "html", "xhtml"].member? type) and 
+      base64ed = (not ["", "text", "html", "xhtml"].member? type) and
         type.match(/^text\/.*/).nil? and  # not text
         type.match(/.*[\+\/]xml$/).nil?   # not XML
 
       if (@content["src"] or base64ed) and not summary
         return [ false, "out-of-line or base64ed atom:content and no atom:summary" ]