imagevenue 0.0.2 → 0.0.3

data/lib/image_venue/directory.rb

@@ -1,28 +1,23 @@
 module ImageVenue
   class Directory
     class << self
-      def list_uri
-        @list_uri ||= URI.parse(ImageVenue.base_url + '/view_dir.php')
-      end
+      attr_accessor :debug
 
-      def files_uri
-        @files_uri ||= URI.parse(ImageVenue.base_url + '/process_get_dir_codes.php')
+      def debug?
+        (ImageVenue.debug? or self.debug) ? true : false
       end
 
-      def files_format_key
-        @files_format_key ||= 'poster'
-      end
-
-      def files_format_forum
-        @files_format_forum ||= 'forum'
-      end
-
-      def files_format_website
-        @files_format_website ||= 'webpage'
+      def puts_debug(*args)
+        if self.debug?
+          puts(*args)
+          return true
+        else
+          return false
+        end
       end
 
-      def files_directory_key
-        @files_directory_key ||= 'dir'
+      def list_uri
+        @list_uri ||= URI.parse(ImageVenue.base_url + '/view_dir.php')
       end
 
       def select_uri
@@ -49,218 +44,160 @@ module ImageVenue
         @delete_directory_key ||= 'delete_dir[]'
       end
 
-      def cookie_directory_key
-        @cookie_directory_key ||= 'cur_upload_dir'
+      def create_params(directory_name)
+        {
+          self.create_directory_key => directory_name
+        }
       end
-    end
-
-    attr_accessor :directories
-    attr_accessor :directories_files
-    attr_accessor :selected_directory
-    attr_accessor :debug
 
-    alias :selected :selected_directory
-
-    def initialize
-      self.directories = nil
-      self.directories_files = {}
-      self.selected_directory = nil
-      return nil
-    end
-
-    def debug?
-      (ImageVenue.debug or self.debug) ? true : false
-    end
+      def delete_params(directory_name)
+        {
+          self.delete_directory_key => directory_name
+        }
+      end
 
-    def puts_debug(*args)
-      if self.debug?
-        puts(*args)
-        return true
-      else
-        return false
+      def select_params(directory_name)
+        {
+          self.select_directory_key => directory_name
+        }
       end
-    end
 
-    def list(reload=false)
-      if self.directories.nil? or reload
-        puts_debug "Trying to fetch directories ..."
-        request = Net::HTTP::Get.new(self.class.list_uri.path)
-        request.header['cookie'] = [ImageVenue.authenticate.cookie_as_string]
-        response = Net::HTTP.start(self.class.list_uri.host, self.class.list_uri.port) do |http|
+      def list(connection=nil)
+        self.puts_debug "Trying to fetch directories ..."
+        request = Net::HTTP::Get.new(self.list_uri.path)
+        request.header['cookie'] = [connection.cookie_as_string]
+        response = Net::HTTP.start(self.list_uri.host, self.list_uri.port) do |http|
           http.request(request)
         end
         if response.is_a?(Net::HTTPSuccess)
+          directories = []
           hpricot = Hpricot(response.body)
-          self.directories = hpricot.search('form:first').first.search('input[@type="radio"]').collect {|input| input['value']}
-          puts_debug "Directories successfully fetched!"
-          return self.directories
+          form = hpricot.search('form:first').first || Hpricot::Elem.new('')
+          form.search('input[@type="radio"]').each do |input|
+            directories << self.new(connection, input['value'], false)
+          end
+          self.puts_debug "Directories successfully fetched!"
+          return directories
         else
-          puts_debug "Directories fetch failed!"
+          self.puts_debug "Directories fetch failed!"
           return false
         end
-      else
-        puts_debug "Directories from cache!"
-        return self.directories
       end
     end
 
-    def files(directory, reload=false)
-      if self.directories_files[directory].nil? or reload
-        puts_debug "Trying to fetch files from directory: '#{directory}' ..."
-        request = Net::HTTP::Post.new(self.class.files_uri.path)
-        request.form_data = self.files_params(directory)
-        request.header['cookie'] = [ImageVenue.authenticate.cookie_as_string]
-        response = Net::HTTP.start(self.class.files_uri.host, self.class.files_uri.port) do |http|
-          http.request(request)
-        end
-        if response.is_a?(Net::HTTPSuccess)
-          self.directories_files[directory] = ImageVenue::File.parse_forum_codes(response.body, directory)
-          puts_debug "Files successfully fetched!"
-          return self.directories_files[directory]
-        else
-          puts_debug "Files fetch failed!"
-          return false
-        end
-      else
-        puts_debug "Files from cache for directory: '#{directory}'"
-        return self.directories_files[directory]
-      end
+    attr_accessor :connection
+    attr_accessor :debug
+    attr_accessor :name
+
+    def initialize(connection, name, is_new=true)
+      self.connection = connection
+      self.name = name
+      @is_new = is_new
+      return nil
     end
 
-    def files_params(directory)
-      {
-        self.class.files_format_key => self.class.files_format_forum,
-        self.class.files_directory_key => directory
-      }
+    def debug?
+      (self.class.debug? or self.debug) ? true : false
     end
 
-    def select(directory, reselect=false)
-      if directory != self.selected_directory or reselect
-        puts_debug "Trying to select directory: '#{directory}'"
-        request = Net::HTTP::Post.new(self.class.select_uri.path)
-        request.form_data = self.select_params(directory)
-        request.header['cookie'] = [ImageVenue.authenticate.cookie_as_string]
-        response = Net::HTTP.start(self.class.select_uri.host, self.class.select_uri.port) do |http|
-          http.request(request)
-        end
-        if response.is_a?(Net::HTTPFound) and not (response.header['Set-Cookie'].nil? or response.header['Set-Cookie'].empty?)
-          self.selected_directory = directory
-          ImageVenue.authenticate.cookie[self.class.cookie_directory_key] = directory
-          puts_debug "Directory successfully selected!"
-          return true
-        else
-          puts_debug "Directory select failed!"
-          return false
-        end
+    def puts_debug(*args)
+      if self.debug?
+        puts(*args)
+        return true
       else
-        puts_debug "Directory already selected."
-        return nil
+        return false
       end
     end
 
-    def select_params(directory)
-      {
-        self.class.select_directory_key => directory
-      }
+    def is_new?
+      @is_new
     end
 
-    def create(directory, reload=false)
-      unless self.list(reload).include?(directory)
-        puts_debug "Trying to create directory: '#{directory}'"
+    def save(force=false, reload=false)
+      if force or self.connection.directories(reload).find {|dir| dir.name == self.name }.nil?
+        self.puts_debug "Trying to create directory: '#{self.name}'"
         request = Net::HTTP::Post.new(self.class.create_uri.path)
-        request.form_data = self.create_params(directory)
-        request.header['cookie'] = [ImageVenue.authenticate.cookie_as_string]
+        request.form_data = self.class.create_params(self.name)
+        request.header['cookie'] = [self.connection.cookie_as_string]
         response = Net::HTTP.start(self.class.create_uri.host, self.class.create_uri.port) do |http|
           http.request(request)
         end
         if response.is_a?(Net::HTTPSuccess) and not (response.header['Set-Cookie'].nil? or response.header['Set-Cookie'].empty?)
-          self.selected_directory = directory
-          self.directories.insert(0, directory)
-          ImageVenue.authenticate.cookie[self.class.cookie_directory_key] = directory
-          puts_debug "Directory successfully created!"
+          self.connection.selected_directory = self.name
+          self.puts_debug "Directory successfully created and selected!"
+          @is_new = false
           return true
         else
-          puts_debug "Directory create failed!"
+          self.puts_debug "Directory create failed!"
           return false
         end
       else
-        puts_debug "Directory already exists: '#{directory}'"
+        self.puts_debug "Directory already exists: '#{self.name}'"
         return nil
       end
     end
 
-    def create_params(directory)
-      {
-        self.class.create_directory_key => directory
-      }
-    end
-
-    def select_or_create(directory, reload=false, reselect=false)
-      if self.list(reload).include?(directory)
-        return self.select(directory, reselect)
-      else
-        return self.create(directory, false) # reload)
-      end
-    end
-
-    def delete(directory, reload=false)
-      if self.list(reload).include?(directory)
-        puts_debug "Trying to delete directory: '#{directory}'"
+    def destroy(force=false, reload=false)
+      if force or not self.connection.directories(reload).find {|dir| dir.name == self.name }.nil?
+        self.puts_debug "Trying to delete directory: '#{self.name}'"
         request = Net::HTTP::Post.new(self.class.delete_uri.path)
-        request.form_data = self.delete_params(directory)
-        request.header['cookie'] = [ImageVenue.authenticate.cookie_as_string]
+        request.form_data = self.class.delete_params(self.name)
+        request.header['cookie'] = [self.connection.cookie_as_string]
         response = Net::HTTP.start(self.class.delete_uri.host, self.class.delete_uri.port) do |http|
           http.request(request)
         end
         if response.is_a?(Net::HTTPSuccess)
-          if directory == self.selected_directory
-            self.selected_directory = nil
-            ImageVenue.authenticate.cookie.delete(self.class.cookie_directory_key)
+          if self.connection.selected_directory == self.name
+            self.connection.selected_directory = nil
           end
-          self.directories.delete(directory)
-          self.directories_files.delete(directory)
-          puts_debug "Directory successfully deleted!"
+          self.puts_debug "Directory successfully deleted!"
+          @is_new = true
           return true
         else
-          puts_debug "Directory delete failed!"
+          self.puts_debug "Directory delete failed!"
           return false
         end
       else
-        puts_debug "Directory does not exist: '#{directory}'"
+        self.puts_debug "Directory does not exist: '#{self.name}'"
         return nil
       end
     end
 
-    def delete_params(directory)
-      {
-        self.class.delete_directory_key => directory
-      }
-    end
-
-    def delete_file(file, reload=false)
-      directory = file.directory
-      files = self.files(directory, reload)
-      puts_debug "Trying to delete file '#{file.basename}' from directory '#{file.directory}' ..."
-      if file.delete
-        files.delete(file)
-        puts_debug "File successfully deleted!"
-        return true
+    def select(reselect=false)
+      if self.connection.selected_directory != self.name or reselect
+        self.puts_debug "Trying to select directory: '#{self.name}'"
+        request = Net::HTTP::Post.new(self.class.select_uri.path)
+        request.form_data = self.class.select_params(self.name)
+        request.header['cookie'] = [self.connection.cookie_as_string]
+        response = Net::HTTP.start(self.class.select_uri.host, self.class.select_uri.port) do |http|
+          http.request(request)
+        end
+        if response.is_a?(Net::HTTPFound) and not (response.header['Set-Cookie'].nil? or response.header['Set-Cookie'].empty?)
+          self.connection.selected_directory = self.name
+          self.puts_debug "Directory successfully selected!"
+          return true
+        else
+          self.puts_debug "Directory select failed!"
+          return false
+        end
       else
-        puts_debug "File delete failed!"
-        return false
+        self.puts_debug "Directory already selected."
+        return nil
       end
     end
 
-    def clear(directory=nil)
-      if directory.nil?
-        self.directories = []
-        self.directories_files = {}
-        puts_debug "Directories cache cleared!"
+    def files(reload=false)
+      if reload or @files.nil?
+        @files = ImageVenue::File.list(self)
       else
-        self.directories_files.delete(directory)
-        puts_debug "Directory's `#{directory}` cache cleared!"
+        self.puts_debug "Files from cache for directory: '#{self.name}'"
       end
-      return true
+      return @files
+    end
+
+    def clear_cache
+      @files = nil
+      self.puts_debug "Directory's cache cleared."
     end
   end
 end

data/lib/image_venue/file.rb

@@ -1,25 +1,115 @@
 module ImageVenue
   class File
     class << self
-      def parse_upload(response_body)
-        hpricot = Hpricot(response_body)
-        textareas = hpricot.search('textarea')
-        image_url = textareas[0].inner_text.strip
-        file = self.new(image_url, ImageVenue.directory.selected)
-        return file
+      attr_accessor :debug
+
+      def debug?
+        (ImageVenue.debug? or self.debug) ? true : false
       end
 
-      def parse_forum_codes(response_body, directory)
+      def puts_debug(*args)
+        if self.debug?
+          puts(*args)
+          return true
+        else
+          return false
+        end
+      end
+
+      def parse_forum_codes(directory, response_body)
         files = []
+        self.puts_debug "Parsing Forum-Codes ..."
         hpricot = Hpricot(response_body)
-        image_urls = hpricot.search('p:last').last.inner_text.scan(/\[URL=(.+?)\]/i).flatten
-        image_urls.each do |image_url|
-          file = self.new(image_url, directory)
-          files.push(file)
+        urls = hpricot.search('p:last').last.inner_text.scan(/\[URL=(.+?)\]/i).flatten
+        urls.each do |url|
+          self.puts_debug "Found URL: `#{url}`"
+          files << self.new(directory, nil, nil, url, false)
         end
         return files
       end
 
+      def list_uri
+        @list_uri ||= URI.parse(ImageVenue.base_url + '/process_get_dir_codes.php')
+      end
+
+      def list_format_key
+        @list_format_key ||= 'poster'
+      end
+
+      def list_format_forum
+        @list_format_forum ||= 'forum'
+      end
+
+      def list_format_website
+        @list_format_website ||= 'webpage'
+      end
+
+      def list_directory_key
+        @list_directory_key ||= 'dir'
+      end
+
+      def upload_uri
+        @upload_uri ||= URI.parse(ImageVenue.base_url + '/upload.php')
+      end
+
+      def upload_action_key
+        @upload_action_key ||= 'action'
+      end
+
+      def upload_user_key
+        @upload_user_key ||= 'user_id'
+      end
+
+      def upload_file_key
+        @upload_file_key ||= 'userfile[]'
+      end
+
+      def upload_directory_key
+        @upload_directory_key ||= 'upload_dir'
+      end
+
+      def upload_content_key
+        @upload_content_key ||= 'imgcontent'
+      end
+
+      def upload_content_safe
+        @upload_content_safe ||= 'safe'
+      end
+      alias :upload_content_family :upload_content_safe
+
+      def upload_content_notsafe
+        @upload_content_notsafe ||= 'notsafe'
+      end
+      alias :upload_content_adult :upload_content_notsafe
+
+      def upload_interlink_key
+        @upload_interlink_key ||= 'interlinkimage'
+      end
+
+      def upload_interlink_yes
+        @upload__interlink_yes ||= 'interlinkyes'
+      end
+
+      def upload_interlink_no
+        @upload__interlink_no ||= 'interlinkno'
+      end
+
+      def upload_resize_key
+        @upload_resize_key ||= 'img_resize'
+      end
+
+      def upload_maximum_files
+        @upload_maximum_files ||= 10
+      end
+
+      def upload_maximum_size
+        @upload_maximum_size ||= 3 * 1024 * 1024 # 3 MByte
+      end
+
+      def upload_image_types
+        @upload_image_types ||= ['jpeg', 'jpg']
+      end
+
       def delete_uri
         @delete_uri ||= URI.parse(ImageVenue.base_url + '/manage_files.php')
       end
@@ -36,23 +126,75 @@ module ImageVenue
         @delete_submit_value ||= 'Delete'
       end
 
-      def cookie_view_directory
-        @cookie_view_directory ||= 'view_dir'
+      def upload_file_params(file)
+        {
+          self.upload_file_key => file
+        }
+      end
+
+      def upload_form_params(directory, content_safe=false, interlink=false, resize=nil)
+        return {
+          self.upload_action_key => '1',
+          self.upload_directory_key => directory.name,
+          self.upload_user_key => directory.connection.cookie[directory.connection.class.cookie_user_key],
+          self.upload_content_key => (content_safe) ? self.upload_content_safe : self.upload_content_notsafe,
+          self.upload_interlink_key => (interlink) ? self.upload_interlink_yes : self.upload_interlink_no,
+          self.upload_resize_key => resize || ''
+        }
+      end
+
+      def delete_params(file_name)
+        {
+          self.delete_submit_key => self.delete_submit_value,
+          self.delete_file_key => file_name
+        }
+      end
+
+      def list(directory)
+        self.puts_debug "Trying to fetch files from directory: '#{directory.name}' ..."
+        request = Net::HTTP::Post.new(self.list_uri.path)
+        request.form_data = self.find_params(directory.name)
+        request.header['cookie'] = [directory.connection.cookie_as_string]
+        response = Net::HTTP.start(self.list_uri.host, self.list_uri.port) do |http|
+          http.request(request)
+        end
+        if response.is_a?(Net::HTTPSuccess)
+          files = self.parse_forum_codes(directory, response.body)
+          self.puts_debug "Files successfully fetched!"
+          return files
+        else
+          self.puts_debug "Files fetch failed!"
+          return false
+        end
+      end
+
+      def find_params(directory_name)
+        {
+          self.list_format_key => self.list_format_forum,
+          self.list_directory_key => directory_name
+        }
       end
     end
 
-    attr_accessor :directory
-    attr_accessor :image_url
+    attr_accessor :content_type
     attr_accessor :debug
+    attr_accessor :directory
+    attr_accessor :io
+    attr_accessor :name
+    attr_accessor :url
 
-    def initialize(image_url, directory=nil)
+    def initialize(directory, name=nil, io=nil, url=nil, is_new=true)
+      self.content_type = 'image/jpeg'
       self.directory = directory
-      self.image_url = image_url
+      self.name = name
+      self.io = io
+      self.url = url
+      @is_new = is_new
       return nil
     end
 
     def debug?
-      (ImageVenue.debug or self.debug) ? true : false
+      (self.class.debug? or self.debug) ? true : false
     end
 
     def puts_debug(*args)
@@ -64,90 +206,130 @@ module ImageVenue
       end
     end
 
-    def image_io
-      if @image_io.nil?
-        puts_debug "Trying to fetch image data ..."
-        image_html = open(self.image_url)
-        image_uri = URI.parse(self.image_url)
-        image_uri.path = '/' + Hpricot(image_html).search('img[@id="thepic"]:first').first['src']
-        @image_io = open(image_uri)
-        puts_debug "Image data successfully fetched!"
+    def is_new?
+      @is_new
+    end
+
+    def save(reselect=false, content_safe=false, interlink=false, resize=nil)
+      self.directory.select(reselect)
+      self.puts_debug "Trying to upload file `#{self.name}` to directory `#{self.directory.name}` ..."
+      request = Net::HTTP::Post.new(self.class.upload_uri.path)
+      request.set_multipart_data(self.class.upload_file_params(self), self.class.upload_form_params(self.directory, content_safe, interlink, resize))
+      request.header['cookie'] = [self.directory.connection.cookie_as_string]
+      response = Net::HTTP.start(self.class.upload_uri.host, self.class.upload_uri.port) do |http|
+        http.request(request)
+      end
+      if response.is_a?(Net::HTTPSuccess)
+        self.url = Hpricot(response.body).search('textarea').first.inner_text.strip
+        self.puts_debug "File successfully uploaded!"
+        @is_new = false
+        return true
+      else
+        self.puts_debug "File upload failed!"
+        return false
+      end
+    end
+
+    def destroy()
+      self.puts_debug "Trying to delete file `#{self.name}` from directory `#{self.directory.name}` ..."
+      request = Net::HTTP::Post.new(self.class.delete_uri.path)
+      request.form_data = self.class.delete_params(self.name)
+      self.directory.connection.view_directory = self.directory.name
+      request.header['cookie'] = [self.directory.connection.cookie_as_string]
+      response = Net::HTTP.start(self.class.delete_uri.host, self.class.delete_uri.port) do |http|
+        http.request(request)
+      end
+      self.directory.connection.view_directory = nil
+      if response.is_a?(Net::HTTPSuccess)
+        self.puts_debug "File `#{self.name}` successfully deleted!"
+        @is_new = true
+        return true
+      else
+        self.puts_debug "File `#{self.name}` delete failed!"
+        return false
+      end
+    end
+
+    def name
+      @name ||= $1 if self.url =~ /image=(.+)$/
+      @name
+    end
+
+    def base_name
+      @base_name ||= Kernel::File.basename(self.name)
+    end
+
+    def real_name
+      @real_name ||= $1+$2 if self.name =~ /^\d+_(.*)_\d+_\d+\w{2}(\.\w{2,3})$/
+    end
+
+    def large_io(reload=false)
+      if @large_io.nil? or reload
+        self.puts_debug "Trying to fetch large image data from `#{self.url}` ..."
+        self.large_url(false)
+        @large_io = open(@large_url)
+        self.puts_debug "Image data successfully fetched!"
       else
-        puts_debug "Image data from cache!"
+        @large_io.rewind
+        self.puts_debug "Image data from cache!"
       end
-      return @image_io
+      return @large_io
+    end
+
+    def large_url(reload=false)
+      if reload or @large_url.nil?
+        self.puts_debug "Trying to fetch large image URL from `#{self.url}` ..."
+        html = open(self.url)
+        img = Hpricot(html).search('img[@id="thepic"]:first').first
+        uri = URI.parse(self.url)
+        uri.path = '/' + img.attributes.find {|key, value| key =~ /^src$/i}.last
+        @large_url = uri.to_s
+        self.puts_debug "Large image URL successfully fetched!"
+      else
+        self.puts_debug "Large image URL from cache!"
+      end
+      return @large_url
     end
 
     def thumbnail_io
       if @thumbnail_io.nil?
-        puts_debug "Trying to fetch thumbnail data ..."
+        self.puts_debug "Trying to fetch thumbnail data ..."
         @thumbnail_io = open(self.thumbnail_url)
-        puts_debug "Thumbnail data successfully fetched!"
+        self.puts_debug "Thumbnail data successfully fetched!"
       else
-        puts_debug "Thumbnail data from cache!"
+        self.puts_debug "Thumbnail data from cache!"
       end
       return @thumbnail_io
     end
 
     def thumbnail_url
-      if @thumbnail_url.nil? and self.image_url =~ /_(\d+)lo/
-        @thumbnail_url = self.image_url.gsub('img.php?image=', "loc#{$1}/th_")
+      if @thumbnail_url.nil? and self.url =~ /_(\d+)lo/
+        @thumbnail_url = self.url.gsub('img.php?image=', "loc#{$1}/th_")
       end
       return @thumbnail_url
     end
 
     def thumbnail_for_board_upper
-      @thumbnail_for_board_upper ||= "[URL=#{self.image_url}][IMG]#{self.thumbnail_url}[/IMG][/URL]"
+      @thumbnail_for_board_upper ||= "[URL=#{self.url}][IMG]#{self.thumbnail_url}[/IMG][/URL]"
     end
 
     def thumbnail_for_board_lower
-      @thumbnail_for_board_lower ||= "[url=#{self.image_url}][img]#{self.thumbnail_url}[/img][/url]"
+      @thumbnail_for_board_lower ||= "[url=#{self.url}][img]#{self.thumbnail_url}[/img][/url]"
     end
 
     def thumbnail_for_website
-      @thumbnail_for_website ||= "<a href=\"#{self.image_url}\" target=_blank><img src=\"#{self.thumbnail_url}\" border=\"0\"></a>"
-    end
-
-    def basename
-      @basename ||= $1 if self.image_url =~ /image=(.+)$/
+      @thumbnail_for_website ||= "<a href=\"#{self.url}\" target=_blank><img src=\"#{self.thumbnail_url}\" border=\"0\"></a>"
     end
 
-    def clear
-      @image_io = nil
+    def clear_cache
+      @large_io = nil
+      @large_url = nil
       @thumbnail_io = nil
       @thumbnail_url = nil
       @thumbnail_for_board_upper = nil
       @thumbnail_for_board_lower = nil
       @thumbnail_for_website = nil
-      @basename = nil
-      puts_debug "File's `#{self.basename}` cache cleared!"
-      return true
-    end
-
-    def delete
-      puts_debug "Trying to delete file `#{self.basename}` from directory `#{self.directory}` ..."
-      request = Net::HTTP::Post.new(self.class.delete_uri.path)
-      request.form_data = self.delete_params
-      ImageVenue.authenticate.cookie[self.class.cookie_view_directory] = self.directory
-      request.header['cookie'] = [ImageVenue.authenticate.cookie_as_string]
-      response = Net::HTTP.start(self.class.delete_uri.host, self.class.delete_uri.port) do |http|
-        http.request(request)
-      end
-      ImageVenue.authenticate.cookie.delete(self.class.cookie_view_directory)
-      if response.is_a?(Net::HTTPSuccess)
-        puts_debug "File `#{self.basename}` successfully deleted!"
-        return true
-      else
-        puts_debug "File `#{self.basename}` delete failed!"
-        return false
-      end
-    end
-
-    def delete_params
-      {
-        self.class.delete_submit_key => self.class.delete_submit_value,
-        self.class.delete_file_key => self.basename
-      }
+      self.puts_debug "File's cache cleared."
     end
   end
 end