lib/redd/models/wiki_page.rb in redd-0.8.4 vs lib/redd/models/wiki_page.rb in redd-0.8.5

@@ -4,18 +4,32 @@
 
 module Redd
   module Models
     # A reddit user.
     class WikiPage < LazyModel
+      # Edit the wiki page.
+      # @param content [String] the new wiki page contents
+      # @param reason [String, nil] an optional reason for editing the page
+      def edit(content, reason: nil)
+        params = { page: @attributes.fetch(:title), content: content }
+        params[:reason] = reason if reason
+        @client.post("/r/#{@attributes.fetch(:subreddit).display_name}/api/wiki/edit", params)
+      end
+
       private
 
       def default_loader
         title = @attributes.fetch(:title)
         if @attributes.key?(:subreddit)
-          sr_name = attributes[:subreddit].display_name
+          sr_name = @attributes[:subreddit].display_name
           return @client.get("/r/#{sr_name}/wiki/#{title}").body[:data]
         end
         @client.get("/wiki/#{title}").body[:data]
+      end
+
+      def after_initialize
+        return unless @attributes[:revision_by]
+        @attributes[:revision_by] = @client.unmarshal(@attributes[:revision_by])
       end
     end
   end
 end