lib/rabbit/slideshare.rb in rabbit-2.0.5 vs lib/rabbit/slideshare.rb in rabbit-2.0.6

@@ -24,22 +24,26 @@
 require "rabbit/command/rabbit"
 
 module Rabbit
   module Task
     class SlideShare
+      class Error < StandardError
+      end
+
       include GetText
 
       BASE_URL = "https://www.slideshare.net"
       API_PATH_PREFIX = "/api/2"
       API_KEY = "NB1B0IzS"
       SHARED_SECRET = "iuTFlPzU"
 
-      attr_accessor :user, :pdf_path, :title, :description, :tags
+      attr_accessor :user, :pdf_path, :id, :title, :description, :tags
       def initialize(logger)
         @logger = logger
         @user = nil
         @pdf_path = nil
+        @id = nil
         @title = nil
         @description = nil
         @tags = []
         @connection = Faraday.new(:url => BASE_URL) do |builder|
           builder.request  :multipart
@@ -48,28 +52,93 @@
           builder.adapter  :net_http
         end
       end
 
       def upload
+        slideshow_id = nil
+        begin
+          slideshow_id = upload_slide
+        rescue Error
+          @logger.error(_("Feailed to upload: %s") % $!.message)
+          return nil
+        end
+
+        begin
+          edit_title(slideshow_id)
+        rescue Error
+          @logger.error(_("Feailed to edit title: %s") % $!.message)
+          return nil
+        end
+
+        url = nil
+        begin
+          url = slide_url(slideshow_id)
+        rescue Error
+          @logger.error(_("Feailed to get slide URL: %s") % $!.message)
+          return nil
+        end
+        url
+      end
+
+      private
+      def upload_slide
         payload = {
           :username              => @user,
           :password              => password,
-          :slideshow_title       => @title,
+          :slideshow_title       => upload_title,
           :slideshow_srcfile     => Faraday::UploadIO.new(@pdf_path,
                                                           "application/pdf"),
           :slideshow_description => @description,
           :tags                  => @tags.join(","),
         }
+        response = post("upload_slideshow", payload)
+        parse_upload_slideshow_response(response)
+      end
+
+      def edit_title(slideshow_id)
+        payload = {
+          :username              => @user,
+          :password              => password,
+          :slideshow_id          => slideshow_id,
+          :slideshow_title       => @title,
+        }
+        response = get("edit_slideshow", payload)
+        parse_edit_slideshow_response(response)
+      end
+
+      def slide_url(slideshow_id)
+        payload = {
+          :slideshow_id => slideshow_id,
+        }
+        response = get("get_slideshow", payload)
+        parse_get_slideshow_response(response)
+      end
+
+      def prepare_payload(payload)
         payload = common_payload.merge(payload)
         payload.keys.each do |key|
           payload.delete(key) if payload[key].nil?
         end
-        parse_response(@connection.post("#{API_PATH_PREFIX}/upload_slideshow",
-                                        payload))
+        payload
       end
 
-      private
+      def get(command, payload)
+        @connection.get(api_url(command), prepare_payload(payload))
+      end
+
+      def post(command, payload)
+        @connection.post(api_url(command), prepare_payload(payload))
+      end
+
+      def api_url(command)
+        "#{API_PATH_PREFIX}/#{command}"
+      end
+
+      def upload_title
+        @id.gsub(/-/, " ")
+      end
+
       def password
         @password ||= read_password(_("Enter password on SlideShare"))
       end
 
       def read_password(prompt)
@@ -94,23 +163,34 @@
 
       def parse_response(http_response)
         @logger.debug(http_response.body)
 
         unless http_response.success?
-          detail = "#{http_response.status}\n#{http_response.body}"
-          @logger.error(_("Failed to upload: %s") % detail)
-          return nil
+          raise Error, "#{http_response.status}\n#{http_response.body}"
         end
 
-        puts http_response.body
         response = Nokogiri::XML(http_response.body)
         if response.root.name == "SlideShareServiceError"
           message = response.root.elements[0]
-          @logger.error("Failed to upload: %s" % message.text)
-          nil
-        else
-          response.xpath("/SlideShowUploaded/SlideShowID").text.to_i
+          raise Error, message
         end
+
+        response
+      end
+
+      def parse_upload_slideshow_response(http_response)
+        response = parse_response(http_response)
+        response.xpath("/SlideShowUploaded/SlideShowID").text.to_i
+      end
+
+      def parse_edit_slideshow_response(http_response)
+        response = parse_response(http_response)
+        response.xpath("/SlideShowEdited/SlideShowID").text.to_i
+      end
+
+      def parse_get_slideshow_response(http_response)
+        response = parse_response(http_response)
+        response.xpath("/Slideshow/URL").text
       end
     end
   end
 end