knjrbfw 0.0.8 → 0.0.9

data/lib/knj/sysuser.rb

@@ -1,27 +1,27 @@
 class Knj::Sysuser
-	def initialize(data)
-		@data = data
-	end
-	
-	def self.list(paras = {})
-		cont = File.read("/etc/passwd")
-		
-		list = []
-		cont.split("\n").each do |line|
-			linearr = line.split(":")
-			
-			list << Knj::Sysuser.new(
-				"nick" => linearr[0],
-				"home" => linearr[5],
-				"shell" => linearr[6]
-			)
-		end
-		
-		return list
-	end
-	
-	def [](key)
-		raise "No such key: " + key if !@data.key?(key)
-		return @data[key]
-	end
+  def initialize(data)
+    @data = data
+  end
+  
+  def self.list(paras = {})
+    cont = File.read("/etc/passwd")
+    
+    list = []
+    cont.split("\n").each do |line|
+      linearr = line.split(":")
+      
+      list << Knj::Sysuser.new(
+        "nick" => linearr[0],
+        "home" => linearr[5],
+        "shell" => linearr[6]
+      )
+    end
+    
+    return list
+  end
+  
+  def [](key)
+    raise "No such key: " + key if !@data.key?(key)
+    return @data[key]
+  end
 end

data/lib/knj/table_writer.rb

@@ -0,0 +1,97 @@
+class Knj::Table_writer
+  def initialize(args = {})
+    @args = args
+    
+    if !@args["filepath"]
+      raise "No filepath was given."
+    end
+    
+    if @args["format"] == "csv"
+      @fp = File.open(@args["filepath"], "w")
+    elsif @args["format"] == "excel5"
+      require "spreadsheet"
+      
+      @wb = Spreadsheet::Workbook.new
+      @ws = @wb.create_worksheet
+      @row = 0
+    else
+      raise "Unsupported format: '#{@args["format"]}'."
+    end
+  end
+  
+  def write_row(arr)
+    if @args["format"] == "csv"
+      arr.each_index do |key|
+        val = arr[key]
+        
+        if val.is_a?(Hash) and val["type"] == "decimal"
+          arr[key] = Knj::Php.number_format(val["value"], @args["amount_decimals"], @args["amount_dsep"], @args["amount_tsep"])
+        elsif val.is_a?(Hash) and val["type"] == "date"
+          arr[key] = Knj::Php.date(@args["date_format"], val["value"])
+        end
+      end
+      
+      line_str = Knj::Csv.arr_to_csv(arr, @args["expl"], @args["surr"])
+      
+      #line_str = line_str.encode("iso8859-1") if @args["encoding"] == "iso8859-1"
+      
+      @fp.write(line_str)
+    elsif @args["format"] == "excel5"
+      col_count = 0
+      arr.each do |val|
+        if val.is_a?(Hash) and val["type"] == "decimal"
+          @ws[@row, col_count] = Knj::Php.number_format(val["value"], @args["amount_decimals"], @args["amount_dsep"], @args["amount_tsep"])
+        elsif val.is_a?(Hash) and val["type"] == "date"
+          @ws[@row, col_count] = Knj::Php.date(@args["date_format"], val["value"])
+        else
+          @ws[@row, col_count] = val
+        end
+        
+        col_count += 1
+      end
+      
+      @row += 1
+    else
+      raise "Unsupported format: '#{@args["format"]}'."
+    end
+  end
+  
+  def close
+    if @args["format"] == "csv"
+      @fp.close
+    elsif @args["format"] == "excel5"
+      dirname = File.dirname(@args["filepath"])
+      basename = File.basename(@args["filepath"], File.extname(@args["filepath"]))
+      
+      temp_path = "#{dirname}/#{basename}.xls"
+      
+      @wb.write(temp_path)
+      @wb = nil
+      @ws = nil
+      
+      FileUtils.mv(temp_path, @args["filepath"])
+    else
+      raise "Unsupported format: '#{@args["format"]}'."
+    end
+  end
+  
+  def ext
+    if @args["format"] == "csv"
+      return "csv"
+    elsif @args["format"] == "excel5"
+      return "xls"
+    else
+      raise "Unsupported format: '#{@args["format"]}'."
+    end
+  end
+  
+  def ftype
+    if @args["format"] == "csv"
+      return "text/csv"
+    elsif @args["format"] == "excel5"
+      return "application/ms-excel"
+    else
+      raise "Unsupported format: '#{@args["format"]}'."
+    end
+  end
+end

data/lib/knj/tests/compiler/compiler_test.rb

@@ -1,12 +1,12 @@
 require "knj/knj"
-require "knj/compiler"
+require "#{$knjpath}compiler"
 
 compiler = Knj::Compiler.new
 
 time_start = Time.new.to_f
 
 #0.upto(10000) do
-	compiler.eval_file("compiler_test_file.rb")
+  compiler.eval_file("compiler_test_file.rb")
 #end
 
 time_spent = Time.new.to_f - time_start

data/lib/knj/tests/test_degulesider.rb

@@ -5,7 +5,7 @@ include Knj
 
 dgs = Degulesider.new
 results = dgs.search(
-	:where => "Engvej 3, 4970 Rødby"
+  :where => "Engvej 3, 4970 Rødby"
 )
 
 Php.print_r(results)

data/lib/knj/tests/test_http2_proxy.rb

@@ -0,0 +1,26 @@
+#!/usr/bin/env ruby1.9.1
+
+require "#{File.dirname(__FILE__)}/../../knjrbfw.rb"
+require "knj/autoload"
+
+proxy_settings = Marshal.load(File.read("#{File.dirname(__FILE__)}/test_http2_proxy_settings.marshal"))
+
+http = Knj::Http2.new(
+  :host => "www.partyworm.dk",
+  :proxy => proxy_settings
+)
+
+urls = [
+  "?show=users_search",
+  "?show=users_online",
+  "?show=drinksdb",
+  "?show=forum&fid=9&tid=1917&page=0"
+]
+urls = ["robots.txt"]
+
+0.upto(105) do |count|
+  url = urls[rand(urls.size)]
+  print "Doing request #{count} of 200 (#{url}).\n"
+  res = http.get(url)
+  raise "Body was empty." if res.body.to_s.length <= 0
+end

data/lib/knj/tests/test_mount.rb

@@ -6,25 +6,25 @@ require "../autoload"
 print "Listing all mounts.\n"
 mounts = Knj::Mount.list
 mounts.each do |mount|
-	Knj::Php.print_r(mount.data)
-	print "Access: #{mount.access?}\n\n"
+  Knj::Php.print_r(mount.data)
+  print "Access: #{mount.access?}\n\n"
 end
 
 mounts = Knj::Mount.list("from_search" => "test_dir")
 mounts.each do |mount|
-	print "Unmounting test_dir.\n"
-	mount.umount
+  print "Unmounting test_dir.\n"
+  mount.umount
 end
 
 print "Mount binding test_dir to test_dir_to.\n"
 Knj::Mount.mount(
-	"from" => "test_dir",
-	"to" => "test_dir_to",
-	"bind" => true
+  "from" => "test_dir",
+  "to" => "test_dir_to",
+  "bind" => true
 )
 
 mounts = Knj::Mount.list("from_search" => "test_dir")
 mounts.each do |mount|
-	print "Unmounting test_dir.\n"
-	mount.umount
+  print "Unmounting test_dir.\n"
+  mount.umount
 end

data/lib/knj/tests/test_retry.rb

@@ -5,24 +5,24 @@ require "../autoload"
 
 count = 0
 result = Knj::Retry.try(
-	:exit => false,
-	:interrupt => true,
-	:errors => [Knj::Errors::NotFound, RuntimeError],
-	:timeout => 1,
-	:tries => 3,
-	:return_error => true
+  :exit => false,
+  :interrupt => true,
+  :errors => [Knj::Errors::NotFound, RuntimeError],
+  :timeout => 1,
+  :tries => 3,
+  :return_error => true
 ) do
-	count += 1
-	print "Count: #{count.to_s}\n"
-	
-	if count <= 2
-		exit
-	end
-	
-	if count <= 3
-		
-		#raise "Test"
-	end
+  count += 1
+  print "Count: #{count.to_s}\n"
+  
+  if count <= 2
+    exit
+  end
+  
+  if count <= 3
+    
+    #raise "Test"
+  end
 end
 
 #print "Error 1 was a #{result[0][:error].class.to_s} with the message: #{result[0][:error].message.to_s}\n"

data/lib/knj/thread.rb

@@ -1,28 +1,27 @@
 #This class behaves like a normal thread - but it shows error-messages and tracebacks. Normal threads dont do that.
 class Knj::Thread < Thread
-	attr_accessor :data
-	
-	def initialize(*args, &block)
-		@data = {}
-		raise "No block was given." if !block_given?
-		
-		abort_on_exception = true
-		super(*args) do
-			begin
-				block.call(*args)
-			rescue SystemExit
-				exit
-			rescue Exception => e
+  attr_accessor :data
+  
+  def initialize(*args)
+    @data = {}
+    raise "No block was given." if !block_given?
+    
+    super(*args) do
+      begin
+        yield(*args)
+      rescue SystemExit
+        exit
+      rescue Exception => e
         print Knj::Errors.error_str(e)
-			end
-		end
-	end
-	
-	def [](key)
-		return @data[key]
-	end
-	
-	def []=(key, value)
-		return @data[key] = value
-	end
+      end
+    end
+  end
+  
+  def [](key)
+    return @data[key]
+  end
+  
+  def []=(key, value)
+    return @data[key] = value
+  end
 end

data/lib/knj/thread2.rb

@@ -1,47 +1,47 @@
 class Knj::Thread2
-	def initialize(args = {}, &block)
-		@args = args
-		@block = block if block_given?
-		@callbacks = {}
-	end
-	
-	def connect(signal, &block)
-		@callbacks[signal] = [] if !@callbacks.key?(signal)
-		@callbacks[signal] << block
-	end
-	
-	def call(signal, *args)
-		return false if !@callbacks.key?(signal)
-		@callbacks[signal].each do |block|
-			block.call(*args)
-		end
-		
-		return {:count => count}
-	end
-	
-	def run
-		Thread.new do
-			abort_on_exception = true
-			call(:on_run)
-			
-			begin
-				@block.call
-			rescue SystemExit
-				call(:on_exit)
-				exit
-			rescue Exception => e
-				call(:on_error, e)
-				
-				if !@args.key?(:print_error) or @args[:print_error]
-					print "Error: "
-					puts e.inspect
-					print "\n"
-					puts e.backtrace
-					print "\n\n"
-				end
-			end
-			
-			call(:on_done)
-		end
-	end
+  def initialize(args = {}, &block)
+    @args = args
+    @block = block if block_given?
+    @callbacks = {}
+  end
+  
+  def connect(signal, &block)
+    @callbacks[signal] = [] if !@callbacks.key?(signal)
+    @callbacks[signal] << block
+  end
+  
+  def call(signal, *args)
+    return false if !@callbacks.key?(signal)
+    @callbacks[signal].each do |block|
+      block.call(*args)
+    end
+    
+    return {:count => count}
+  end
+  
+  def run
+    Thread.new do
+      abort_on_exception = true
+      call(:on_run)
+      
+      begin
+        @block.call
+      rescue SystemExit
+        call(:on_exit)
+        exit
+      rescue Exception => e
+        call(:on_error, e)
+        
+        if !@args.key?(:print_error) or @args[:print_error]
+          print "Error: "
+          puts e.inspect
+          print "\n"
+          puts e.backtrace
+          print "\n\n"
+        end
+      end
+      
+      call(:on_done)
+    end
+  end
 end

data/lib/knj/threadhandler.rb

@@ -1,39 +1,46 @@
 class Knj::Threadhandler
-	attr_reader :inactive_blocks, :args, :activate_blocks, :mutex
-	
-	def initialize(args = {})
-		@args = args
-		@objects = []
-		@args[:timeout] = 5 if !@args[:timeout]
-		@inactive_blocks = []
-		@activate_blocks = []
-		@mutex = Mutex.new
-		
-		@thread_timeout = Knj::Thread.new do
-			loop do
-				sleep @args[:timeout]
-				break if !@mutex
-				check_inactive
-			end
-		end
-	end
-	
-	def on_spawn_new(&block)
+  attr_reader :inactive_blocks, :args, :activate_blocks, :mutex, :objects
+  
+  def initialize(args = {})
+    require "#{$knjpath}errors"
+    
+    @args = args
+    @objects = []
+    @args[:timeout] = 5 if !@args[:timeout]
+    @args[:max] = 50 if !@args[:max]
+    @inactive_blocks = []
+    @activate_blocks = []
+    @mutex = Mutex.new
+    
+    @thread_timeout = Thread.new do
+      begin
+        loop do
+          sleep @args[:timeout]
+          break if !@mutex
+          check_inactive
+        end
+      rescue => e
+        STDOUT.print Knj::Errors.error_str(e)
+      end
+    end
+  end
+  
+  def on_spawn_new(&block)
     raise "Destroyed Knj::Threadhandler." if !@mutex
-		@spawn_new_block = block
-	end
-	
-	def on_inactive(&block)
+    @spawn_new_block = block
+  end
+  
+  def on_inactive(&block)
     raise "Destroyed Knj::Threadhandler." if !@mutex
-		@inactive_blocks << block
-	end
-	
-	def on_activate(&block)
+    @inactive_blocks << block
+  end
+  
+  def on_activate(&block)
     raise "Destroyed Knj::Threadhandler." if !@mutex
-		@activate_blocks << block
-	end
-	
-	def destroy
+    @activate_blocks << block
+  end
+  
+  def destroy
     return false if !@mutex
     
     @thread_timeout.kill if @thread_timeout and @thread_timeout.alive?
@@ -53,78 +60,104 @@ class Knj::Threadhandler
     @inactive_blocks = nil
     @activate_blocks = nil
     @mutex = nil
-	end
-	
-	def check_inactive
+  end
+  
+  def check_inactive
     raise "Destroyed Knj::Threadhandler." if !@mutex
-		@mutex.synchronize do
-			cur_time = Time.new.to_i - @args[:timeout]
-			@objects.each do |data|
-				if data[:free] and !data[:inactive] and data[:free] < cur_time
-					@inactive_blocks.each do |block|
-						data[:inactive] = true
-						block.call(:obj => data[:object])
-					end
-				end
-			end
-		end
-	end
-	
-	def get_and_lock
+    @mutex.synchronize do
+      cur_time = Time.now.to_i - @args[:timeout]
+      @objects.each do |data|
+        if data[:free] and !data[:inactive] and data[:free] < cur_time
+          @inactive_blocks.each do |block|
+            data[:inactive] = true
+            block.call(:obj => data[:object])
+          end
+        end
+      end
+    end
+  end
+  
+  def get_and_lock
     raise "Destroyed Knj::Threadhandler." if !@mutex
-		newobj = nil
-		
-		@mutex.synchronize do
-			retdata = false
-			@objects.each do |data|
-				if data[:free]
-					retdata = data
-					break
-				end
-			end
-			
-			if retdata
-				#Test if object is still free - if not, try again - knj.
-				return get_and_lock if !retdata[:free]
-				retdata[:free] = false
-				
-				if retdata[:inactive]
-					@activate_blocks.each do |block|
-						block.call(:obj => retdata[:object])
-					end
-					
-					retdata.delete(:inactive)
-				end
-				
-				return retdata[:object]
-			end
-			
-			newobj = @spawn_new_block.call
-			@objects << {
-				:free => false,
-				:object => newobj
-			}
-			STDOUT.print "Spawned db and locked new.\n" if @args[:debug]
-		end
-		
-		return newobj
-	end
-	
-	def free(obj)
-		@mutex.synchronize do
+    newobj = nil
+    sleep_do = false
+    
+    begin
+      @mutex.synchronize do
+        retdata = false
+        @objects.each do |data|
+          if data[:free]
+            retdata = data
+            break
+          end
+        end
+        
+        if retdata
+          #Test if object is still free - if not, try again - knj.
+          return get_and_lock if !retdata[:free]
+          retdata[:free] = false
+          
+          if retdata[:inactive]
+            @activate_blocks.each do |block|
+              block.call(:obj => retdata[:object])
+            end
+            
+            retdata.delete(:inactive)
+          end
+          
+          return retdata[:object]
+        end
+        
+        if @objects.length >= @args[:max]
+          #The maximum amount of objects has already been spawned... Sleep 0.1 sec and try to lock an object again...
+          raise Knj::Errors::Retry
+        else
+          #No free objects, but we can spawn a new one and use that...
+          newobj = @spawn_new_block.call
+          @objects << {
+            :free => false,
+            :object => newobj
+          }
+          STDOUT.print "Spawned db and locked new.\n" if @args[:debug]
+        end
+      end
+      
+      return newobj
+    rescue Knj::Errors::Retry
+      STDOUT.print "All objects was taken - sleeping 0.1 sec and tries again.\n" #if @args[:debug]
+      sleep 0.1
+      retry
+    end
+  end
+  
+  def free(obj)
+    @mutex.synchronize do
       return false if !@mutex or !@objects #something is trying to free and object, but the handler is destroyed. Dont crash but return false.
       
-			freedata = false
-			@objects.each do |data|
-				if data[:object] == obj
-					freedata = data
-					break
-				end
-			end
-			
-			raise "Could not find that object in list." if !freedata
-			STDOUT.print "Freed one.\n" if @args[:debug]
-			freedata[:free] = Time.new.to_i
-		end
-	end
+      freedata = false
+      @objects.each do |data|
+        if data[:object] == obj
+          freedata = data
+          break
+        end
+      end
+      
+      raise "Could not find that object in list." if !freedata
+      STDOUT.print "Freed one.\n" if @args[:debug]
+      freedata[:free] = Time.now.to_i
+    end
+  end
+  
+  #Executes the given block with an element and then frees it.
+  def use
+    raise "No block was given." if !block_given?
+    
+    obj = self.get_and_lock
+    
+    begin
+      yield(obj)
+    ensure
+      self.free(obj)
+    end
+  end
 end