rsence-pre 2.1.0.1.pre

data/plugins/ticket/lib/favicon.rb

@@ -0,0 +1,39 @@
+##   RSence
+ #   Copyright 2008 Riassence Inc.
+ #   http://riassence.com/
+ #
+ #   You should have received a copy of the GNU General Public License along
+ #   with this software package. If not, contact licensing@riassence.com
+ ##
+
+
+
+# @private Inner workings of Ticket
+module TicketService
+# @private Inner workings of Ticket
+module Favicon
+  
+  def favicon( req, res ) # :nodoc:
+    
+    res.status = 200
+    
+    favicon_data = @raw_uris['favicon.ico']
+    
+    res['Content-Type'] = favicon_data[0]
+    res['Content-Length'] = favicon_data[1]
+    
+    res['Date'] = httime( Time.now )
+    res.body = favicon_data[2]
+    
+  end
+  
+  # Sets favicon. First parameter is favicon data and the second one is content type which defaults to false.
+  def set_favicon( ico_data, content_type=false )
+    @raw_uris['favicon.ico'][0] = content_type if content_type
+    @raw_uris['favicon.ico'][1] = ico_data.size.to_s
+    @raw_uris['favicon.ico'][2] = ico_data
+    
+  end
+  
+end
+end

data/plugins/ticket/lib/file.rb

@@ -0,0 +1,58 @@
+##   RSence
+ #   Copyright 2008 Riassence Inc.
+ #   http://riassence.com/
+ #
+ #   You should have received a copy of the GNU General Public License along
+ #   with this software package. If not, contact licensing@riassence.com
+ ##
+
+
+# @private Inner workings of Ticket
+module TicketService
+# @private Inner workings of Ticket
+module TicketFile
+  
+  # Removes other disposable data and references. File removed by file_id 
+  # which is initially given with the +Common#Serve+ function.
+  def del_file( file_id, ses_id=false )
+    if ses_id and @files[:ses_ids].has_key?( ses_id )
+      @files[:ses_ids][ses_id].delete( file_id ) if @files[:ses_ids][ses_id].include?( file_id )
+    end
+    @files[:by_id].delete( file_id ) if @files[:by_id].has_key?( file_id )
+  end
+  
+  # extends expiration time of disposable data, essentially for keep-alive requests
+  def push_keepalive_file( file_id, keep_alive ) # :nodoc:
+    expiry_time = Time.now.to_i+keep_alive
+    @expire_files[expiry_time] = [] unless @expire_files.has_key?(expiry_time)
+    @expire_files[expiry_time].push( file_id )
+  end
+  
+  # removes all expired files
+  def expire_keepalive_files # :nodoc:
+    curr_time = Time.now.to_i
+    @expire_files.keys.sort.each do |exp_time|
+      if exp_time < curr_time
+        @expire_files[exp_time].size.times do |incr|
+          file_id = @expire_files[exp_time].shift
+          if @files[:by_id].has_key?(file_id)
+            ses_id = @files[:by_id][file_id][3]
+            del_file( file_id, ses_id )
+          end
+        end
+        @expire_files.delete(exp_time) if @expire_files[exp_time].size == 0
+      end
+    end
+  end
+  
+  # Shorthand for +Common#serve+ for serving a file. Expects served file to be
+  # of type :file. Defaults the content_type to 'text/plain' and expects user
+  # to input filename.
+  def serve_file( msg, content='', content_type='text/plain', filename='' )
+    return serve( msg, content, [content_type, filename], :file)
+  end
+  
+end
+end
+
+

data/plugins/ticket/lib/img.rb

@@ -0,0 +1,50 @@
+##   RSence
+ #   Copyright 2008 Riassence Inc.
+ #   http://riassence.com/
+ #
+ #   You should have received a copy of the GNU General Public License along
+ #   with this software package. If not, contact licensing@riassence.com
+ ##
+
+
+# @private Inner workings of Ticket
+module TicketService
+# @private Inner workings of Ticket
+module Img
+  
+  # Removes image data and references. Image is removed with
+  # its ID which is the same as its URI.
+  def del_img( img_id, ses_id=false )
+    if ses_id and @imgs[:ses_ids].has_key?( ses_id )
+      @imgs[:ses_ids][ses_id].delete( img_id ) if @imgs[:ses_ids][ses_id].include?( img_id )
+    end
+    @imgs[:by_id].delete( img_id ) if @imgs[:by_id].has_key?( img_id )
+  end
+  
+  # Extends expiration time of disposable files, essentially for keep-alive
+  # requests. Image URI is used as an ID and extension is given as seconds.
+  def push_keepalive( img_id, keep_alive )
+    expiry_time = Time.now.to_i+keep_alive
+    @expires[expiry_time] = [] unless @expires.has_key?(expiry_time)
+    @expires[expiry_time].push( img_id )
+  end
+  
+  # Removes all expired images.
+  def expire_keepalives
+    curr_time = Time.now.to_i
+    @expires.keys.sort.each do |exp_time|
+      if exp_time < curr_time
+        @expires[exp_time].size.times do |incr|
+          img_id = @expires[exp_time].shift
+          if @imgs[:by_id].has_key?(img_id)
+            ses_id = @imgs[:by_id][img_id][3]
+            del_img( img_id, ses_id )
+          end
+        end
+        @expires.delete(exp_time) if @expires[exp_time].size == 0
+      end
+    end
+  end
+  
+end
+end

data/plugins/ticket/lib/objblob.rb

@@ -0,0 +1,66 @@
+##   RSence
+ #   Copyright 2008 Riassence Inc.
+ #   http://riassence.com/
+ #
+ #   You should have received a copy of the GNU General Public License along
+ #   with this software package. If not, contact licensing@riassence.com
+ ##
+
+
+# @private Inner workings of Ticket
+module TicketService
+# @private Inner workings of Ticket
+module ObjBlob
+  
+  # Serves a blob object. Will expire by default. Returns ID as URI.
+  def serve_blobobj( msg, blob_obj, no_expire=false )
+    # gets a new, unique identifier
+    ticket_id = @randgen.gen
+    if no_expire
+      @raw_uris[ticket_id] = blob_obj
+    else
+      @blob_objs[:ses_ids][msg.ses_id] = [] unless @imgs[:ses_ids].has_key?(msg.ses_id)
+      @blob_objs[:ses_ids][msg.ses_id].push( ticket_id )
+      @blob_objs[:by_id][ticket_id] = [msg.ses_id,blob_obj]
+    end
+    uri = File.join(::RSence.config[:broker_urls][:b],ticket_id)
+    return uri
+  end
+  
+  # Deletes blob object by ID. 
+  def del_blobobj( ticket_id, ses_id=false )
+    if @raw_uris.has_key?( ticket_id )
+      @raw_uris.delete( rsrc_id )
+    else
+      if ses_id and @blob_objs[:ses_ids].has_key?( ses_id )
+        @blob_objs[:ses_ids][ses_id].delete( ticket_id ) if @blob_objs[:ses_ids][ses_id].include?( ticket_id )
+      end
+      @blob_objs[:by_id].delete( ticket_id ) if @blob_objs[:by_id].has_key?( ticket_id )
+    end
+  end
+  
+  # Increases expiration time by keep_alive in seconds.
+  def push_keepalive_blobobj( ticket_id, keep_alive )
+    expiry_time = Time.now.to_i+keep_alive
+    @expire_blobobj[expiry_time] = [] unless @expire_blobobj.has_key?(expiry_time)
+    @expire_blobobj[expiry_time].push( ticket_id )
+  end
+  
+  # Removes all expired blob objects.
+  def expire_keepalive_blobobjs
+    curr_time = Time.now.to_i
+    @expire_blobobj.keys.sort.each do |exp_time|
+      if exp_time < curr_time
+        @expire_blobobj[exp_time].size.times do |incr|
+          ticket_id = @expire_blobobj[exp_time].shift
+          if @blob_objs[:by_id].has_key?(ticket_id)
+            ses_id = @blob_objs[:by_id][ticket_id][3]
+            del_blobobj( ticket_id, ses_id )
+          end
+        end
+        @expire_blobobj.delete(exp_time) if @expire_blobobj[exp_time].size == 0
+      end
+    end
+  end
+end
+end

data/plugins/ticket/lib/rsrc.rb

@@ -0,0 +1,34 @@
+##   RSence
+ #   Copyright 2008 Riassence Inc.
+ #   http://riassence.com/
+ #
+ #   You should have received a copy of the GNU General Public License along
+ #   with this software package. If not, contact licensing@riassence.com
+ ##
+
+
+# @private Inner workings of Ticket
+module TicketService
+# @private Inner workings of Ticket
+module Rsrc
+  
+  # Removes static data by ID.
+  def del_rsrc( rsrc_id )
+    @raw_uris.delete( rsrc_id )
+  end
+  
+  # Serves static resources. ID returned by function.
+  def serve_rsrc( content, content_type )
+    
+    rsrc_id = @randgen.gen
+    #puts "rsrc_id: #{rsrc_id.inspect}"
+    
+    content_size = content.size.to_s
+    
+    @raw_uris[rsrc_id] = [content_type,content_size,content]
+    
+    uri = File.join(::RSence.config[:broker_urls][:d],rsrc_id)
+  end
+  
+end
+end

data/plugins/ticket/lib/upload.rb

@@ -0,0 +1,236 @@
+##   RSence
+ #   Copyright 2008 Riassence Inc.
+ #   http://riassence.com/
+ #
+ #   You should have received a copy of the GNU General Public License along
+ #   with this software package. If not, contact licensing@riassence.com
+ ##
+
+
+
+require 'sequel'
+
+# @private Inner workings of Ticket
+module TicketService
+
+# @private Inner workings of Ticket
+module Upload
+  
+=begin
+
+Handles uploads by using tickets and other filters.
+
+Upload success states:
+ 0: Idle, default, ready to upload
+ 1: Upload started
+ 2: Upload completed
+ 3: Upload processed
+ 4: Client requests new key
+
+Upload failure states:
+ -1: Invalid request (missing query keys)
+ -2: Invalid or missing key
+ -3: Invalid mime-type (invalid data format)
+ -4: File too big
+ -5: Key request forbidden (when calling upload_key)
+ -6: Post-processing failed
+
+
+Default upload table:
+
+create table rsence_uploads (
+  id int primary key auto_increment,
+  ses_id int not null,
+  upload_date int not null,
+  upload_done tinyint not null default 0,
+  file_size int not null default 0,
+  file_mime varchar(255) not null default 'text/plain',
+  file_data mediumblob
+)
+=end
+  def upload(request,response)
+    
+    ticket_id = req.unparsed_uri.match(/^#{::RSence.config[:broker_urls][:u]}(.*)$/)[1]
+    value_id  = request.query['value_id']
+    
+    if not @upload_slots[:by_id].has_key?(ticket_id)
+      done_value = '-2:::Invalid or missing key'
+    else
+      ## Get data stored in the upload slot
+      (mime_allow,max_size,ses_id,value_key,allow_multi) = @upload_slots[:by_id][ticket_id]
+      if not allow_multi
+        @upload_slots[:by_id].delete( ticket_id )
+      end
+      ## The upload form field
+      file_data = request.query['file_data']
+      file_mimetype = file_data[:type]
+      file_filename = file_data[:filename]
+      ## Get the size from the temporary file
+      file_size = file_data[:tempfile].stat.size
+      ## Check for errors
+      if not mime_allow.match(file_mimetype)
+        done_value = "-3:::#{ticket_id}"
+      elsif not file_size < max_size
+        done_value = "-4:::#{ticket_id}"
+      else
+        done_value = "2:::#{ticket_id}"
+        ## Insert basic data about the upload and get its key
+        insert_hash = {
+          :ses_id      => ses_id,
+          :ticket_id   => ticket_id,
+          :upload_date => Time.now.to_i,
+          :upload_done => false,
+          :file_name   => file_filename,
+          :file_size   => file_size,
+          :file_mime   => file_mimetype,
+          :file_data   => ''
+        }
+        if @db
+          upload_id = @db[:rsence_uploads].insert( insert_hash )
+        else
+          @upload_id += 1
+          upload_id = @upload_id
+          @upload_slots[:rsence_uploads] = {} unless @upload_slots.has_key?(:rsence_uploads)
+          @upload_slots[:rsence_uploads][upload_id] = insert_hash
+        end
+        if not @upload_slots[:uploaded].has_key?(ticket_id)
+          @upload_slots[:uploaded][ticket_id] = []
+        end
+        @upload_slots[:uploaded][ticket_id].push( upload_id )
+        
+        update_hash = {
+          :file_data   => file_data[:tempfile].read.to_sequel_blob,
+          :upload_done => true
+        }
+        if @db
+          @db[:rsence_uploads].filter(:id => upload_id).update( update_hash )
+        else
+          @upload_slots[:rsence_uploads][upload_id].merge!( update_hash )
+        end
+      end
+    end
+    
+    response_body = %{
+      <html>
+        <head>
+          <script type="text/javascript">
+            with(parent){
+              HVM.values[#{value_id.to_json}].set(#{done_value.to_json});
+            }
+          </script>
+        </head>
+        <body></body>
+      </html>
+    }
+    
+    response.status = 200
+    response['content-type'] = 'text/html; charset=UTF-8'
+    response['content-length'] = response_body.size.to_s
+    response.body = response_body
+  end
+  
+  def get_uploads( ticket_id, with_data=false )
+    uploads = []
+    if @upload_slots[:uploaded].has_key?(ticket_id)
+      @upload_slots[:uploaded][ticket_id].each do |row_id|
+        if with_data
+          if @db
+            row_datas = @db[:rsence_uploads].select(
+              :upload_date, :upload_done, :file_name,
+              :file_size, :file_mime, :file_data
+            ).filter(
+              :id => row_id
+            )
+          else
+            row_datas = @upload_slots[:rsence_uploads][row_id]
+          end
+          if row_datas.size == 1
+            row_data = row_datas.first
+            row_hash = {
+              :date => Time.at(row_data[:upload_date]),
+              :done => row_data[:upload_done],
+              :size => row_data[:file_size],
+              :mime => row_data[:file_mime],
+              :name => row_data[:file_name],
+              :data => row_data[:file_data]
+            }
+            uploads.push(row_hash)
+          end
+        else
+          if @db
+            row_datas = @db[:rsence_uploads].select(
+              :upload_date, :upload_done, :file_name,
+              :file_size, :file_mime
+            ).filter(
+              :id => row_id
+            )
+          else
+            row_datas = @upload_slots[:rsence_uploads][row_id]
+          end
+          if row_datas.size == 1
+            row_data = row_datas.first
+            row_hash = {
+              :date => Time.at(row_data[:upload_date]),
+              :done => row_data[:upload_done],
+              :size => row_data[:file_size],
+              :mime => row_data[:file_mime],
+              :name => row_data[:file_name],
+              :data => nil
+            }
+            uploads.push(row_hash)
+          end
+        end
+      end
+    end
+    return uploads
+  end
+  
+  def del_upload( ticket_id, row_id )
+    @upload_slots[:uploaded][ticket_id].delete(row_id)
+    if @db
+      @db[:rsence_uploads].filter( :id => row_id ).delete
+    else
+      @upload_slots[:rsence_uploads].delete(row_id)
+    end
+  end
+  
+  # removes uploaded files
+  def del_uploads( ticket_id, ses_id=false )
+    if ses_id and @upload_slots[:ses_ids].has_key?( ses_id )
+      if @upload_slots[:ses_ids][ses_id].include?( ticket_id )
+        @upload_slots[:ses_ids][ses_id].delete( ticket_id )
+      end
+    end
+    if @upload_slots[:uploaded].has_key?( ticket_id )
+      @upload_slots[:uploaded][ticket_id].each do |row_id|
+        @upload_slots[:uploaded][ticket_id].delete( row_id )
+        if @db
+          @db[:rsence_uploads].filter( :id => row_id ).delete
+        else
+          @upload_slots[:rsence_uploads].delete(row_id)
+        end
+      end
+      @upload_slots[:uploaded].delete( ticket_id )
+    end
+    if @upload_slots[:by_id].has_key?( ticket_id )
+      @upload_slots[:by_id].delete( ticket_id )
+    end
+    if @db
+      @db[:rsence_uploads].filter( :ticket_id => ticket_id ).delete
+      @db[:rsence_uploads].filter( :ses_id => ses_id ).delete
+    end
+  end
+  
+  def upload_key(msg,value_key,max_size=1000000,mime_allow=/(.*?)\/(.*?)/,allow_multi=true)
+    key = @randgen.gen
+    while @upload_slots[:by_id].has_key?(key)
+      key = @randgen.gen
+    end 
+    @upload_slots[:by_id][key] = [mime_allow,max_size,msg.ses_id,value_key,allow_multi]
+    @upload_slots[:uploaded][key]  = []
+    @upload_slots[:ses_ids][msg.ses_id] = [] unless @upload_slots[:ses_ids].has_key?(msg.ses_id)
+    @upload_slots[:ses_ids][msg.ses_id].push( key )
+    return "0:::#{key}"
+  end
+end
+end