YkLib 0.1.0

data/lib/YkLib/Yk/rwhen.rb

@@ -0,0 +1,21 @@
+
+
+def rwhen (expr, diffSec = 0)
+	while true
+		yield
+		waitSec = nil
+		IO.popen "rwhen.rb '#{expr}' #{$PROGRAM_NAME}" do |fr|
+			waitSec = (fr.gets || break).chomp.to_f
+		end
+		if $?.to_i != 0 || waitSec == nil
+			raise ArgumentError.new("failed to calculate the execution time by rwhen\n")
+		end
+		if $DEBUG
+			require 'Yk/debugout'
+			er waitSec + diffSec
+		end
+		sleep waitSec + diffSec
+	end
+end
+
+

data/lib/YkLib/Yk/selector.rb

@@ -0,0 +1,124 @@
+
+
+require 'set'
+
+module SelectorFunc
+	def set_selector s
+		@selectors ||= Set.new
+		@selectors.add s
+	end
+	def selectors
+		@selectors
+	end
+end
+
+
+class Selector
+	def self.select &bl
+		s = new
+		if bl
+			bl.call s
+			s.select
+		else
+			s
+		end
+	end
+	def initialize
+		@readProcs = {}
+		@writeBuffs = Hash.new{|h, k| h[k] = []}
+		@writeFinals = {}
+		@intReader, @intWriter = IO.pipe
+	end
+	def at_read fr, &bl
+		@readProcs[fr] = bl
+		if !fr.respond_to? :set_selector
+			fr.extend SelectorFunc
+		end
+		fr.set_selector self
+	end
+	def int
+		@intWriter.close
+	end
+	def reserve_write fp, buff
+		if buff
+			@writeBuffs[fp].push buff if buff != ""
+		else
+			@writeFinals[fp] = true
+		end
+	end
+	def select
+		selectable = true
+		while selectable
+			selectable = false
+			reads = (@readProcs.keys + [@intReader]).select{|s| !s.closed?}
+			writes = @writeBuffs.keys.select{|s| ((@writeBuffs[s] && @writeBuffs[s].size > 0) || @writeFinals[s]) && !s.closed?}
+			break if reads.size + writes.size == 0
+			begin
+				selected, wselected = IO.select(reads, writes)
+			rescue Errno::EIO
+			rescue Errno::EBADF
+			rescue IOError
+			end
+			selected.each do |fp|
+				if fp != @intReader
+					buff = ""
+					begin
+						buff = fp.readpartial 1024
+					rescue EOFError => e
+					rescue Errno::EIO
+					rescue Errno::EBADF
+					rescue IOError
+					end
+					if buff != ""
+						@readProcs[fp].call buff
+					end
+					if fp.closed?
+						@readProcs[fp].call ""
+						@readProcs.delete fp
+						next
+					end
+					#closed = true if buff == ""
+					#@readProcs[fp].call buff
+					#if closed
+					#	@readProcs.delete fp
+					#end
+				else
+					reads.each do |fz|
+						if fz != fp && fz.closed?
+							@readProcs[fz].call ""
+						end
+					end
+					@intReader.close
+					break
+				end
+			end
+			wselected.each do |fp|
+				buffs = @writeBuffs[fp]
+				if buffs.size > 0
+					buff = buffs[0]
+					sz = fp.write buff
+					fp.flush
+					if sz < buff.size
+						buff.replace buff[sz ... buff.size]
+						break
+					else
+						buff.replace ""
+						buffs.shift
+					end
+				end
+				if (!buffs || buffs.size == 0) && @writeFinals[fp]
+					@writeFinals.delete fp
+					@writeBuffs.delete fp
+					if ![STDERR, STDOUT].include? fp
+						fp.close_write
+					end
+				end	
+			end
+			reads = reads.select{|s| !s.closed?}
+			writes = @writeBuffs.keys.select{|s| !s.closed?}
+			selectable = (reads - [STDIN]).size + (writes - [STDERR, STDOUT]).size > 0
+		end
+	end
+end
+
+

data/lib/YkLib/Yk/set.rb

@@ -0,0 +1,170 @@
+
+
+class Set
+	include Enumerable
+	def initialize (*a)
+		if a.size == 1 && a[0].is_a?(Set)
+			@hash = a[0].hash.clone
+			return
+		end
+		@hash = Hash.new
+		insert(*a)
+	end
+	def insert (*a)
+		failed = true
+		a.each do |e|
+			if !@hash.key?(e)
+				failed = false
+				@hash[e] = true
+			end
+		end
+		!failed
+	end
+	attr :hash, true
+	def each
+		@hash.each do |e, v|
+			yield e
+		end
+	end
+	def begin
+		ret = nil
+		each do |e|
+			ret = e
+			break
+		end
+		ret
+	end
+	def clone
+		self.class.new(self)
+	end
+	def include? (a)
+		@hash.key? a
+	end
+	def + (arg)
+		clone.union! arg
+	end
+	def - (arg)
+		clone.except! arg
+	end
+	def union! (arg)
+		arg.each do |e|
+			insert e
+		end
+		self
+	end
+	def except! (arg)
+		arg.each do |e|
+			delete e
+		end
+		self
+	end
+	def clear
+		@hash.clear
+	end
+	def delete (k)
+		@hash.delete k
+	end
+	def inspect
+		res = "("
+		s = true
+		each do |e|
+			res +=  (s ? "" : ",") + e.inspect
+			s &&= false
+		end
+		res += ")"
+		res
+	end
+	def to_a
+		@hash.keys
+	end
+	def index a
+		@hash.keys.index a
+	end
+	def to_set
+		self
+	end
+	def == (arg)
+        if arg.is_a? Array
+            arg = arg.to_set
+            @hash == arg._hash
+        elsif arg.is_a? Set
+            @hash == arg._hash
+        end
+    end
+    def _hash
+        @hash
+	end
+	def size
+		@hash.size
+	end
+	def join (*args)
+		@hash.keys.join(*args)
+	end
+end
+
+
+class Array
+	def to_set
+		ret = Set.new
+		each do |e|
+			ret.insert e
+		end
+		ret
+	end
+end
+
+
+class KeyedSet < Set
+	include Enumerable
+	def initialize (keyMethod)
+		if keyMethod.is_a? KeyedSet
+			super(self)
+			@keyMethod = keyMethod.keyMethod
+		else
+			super()
+			@keyMethod = keyMethod
+		end
+	end
+	def callKey e
+		if @keyMethod.is_a? Symbol
+			e.method(@keyMethod).call
+		elsif @keyMethod.is_a?(String) && @keyMethod[0..0] == "."
+			eval("e" + @keyMethod)
+		else
+			@keyMethod.to_proc.call e
+		end
+	end
+	def insert (*a)
+		failed = true
+		a.each do |e|
+			if !@hash.key?(k = callKey(e))
+				failed = false
+				@hash[k] = e
+			end
+		end
+		!failed
+	end
+	def [] (i)
+		@hash[i]
+	end
+	def []= (i, a)
+		if callKey(a) != i
+			raise ArgumentError.new("key value is different")
+		end
+		@hash[i] = a
+	end
+	attr :hash, true
+	def include? (a)
+		@hash.key? callKey(a)
+	end
+	def delete (a)
+		@hash.delete callKey(a)
+	end
+	def each
+		@hash.values.each do |e|
+			yield e
+		end
+	end
+end
+
+

data/lib/YkLib/Yk/shellquote.rb

@@ -0,0 +1,300 @@
+class String
+	def shellSQuote (mode = true)
+		if self[-1] == ?\\
+			self =~ /\\+$/
+			post = '"' + $& + $& + '"'
+			tmp = $'
+		else
+			post = ""
+			tmp = self
+		end
+		tmp = tmp.gsub('\'', '\'"\'"\'')
+		if mode
+			tmp = "'" + tmp + "'"
+		end
+		tmp + post
+	end
+	def shellSQuote! (mode = true)
+		self.replace(shellSQuote(mode))
+	end
+	def condSQuote (mode = true)
+		if self =~ /\s/ || self =~ /\'/
+			tmp = strip
+			if tmp.size >= 2 && ((tmp[0] == ?\' && tmp[-1] == ?\') || (tmp[0] == ?\" && tmp[-1] == ?\"))
+				tmp
+			else
+				self.shellSQuote
+			end
+		else
+			self
+		end
+	end
+	def condSQuote! (mode = true)
+		self.replace(condSQuote(mode))
+	end
+	def shellDQuote (mode = true)
+		tmp = self.gsub(?\\.chr, "\\\\")
+		tmp = self.gsub("\"", "\""'"'"\"")
+		if mode
+			tmp = "\"" + tmp + "\""
+		end
+		tmp
+	end
+	def shellDQuote! (mode = true)
+ 		self.replace(shellDQuote(mode))
+	end
+	def condDQuote (mode = true)
+		if self =~ /\s/ || self =~ /\"/
+			tmp = strip
+			if tmp.size >= 2 && ((tmp[0] == ?\' && tmp[-1] == ?\') || (tmp[0] == ?\" && tmp[-1] == ?\"))
+				tmp
+			else
+				self.shellDQuote
+			end
+		else
+			self
+		end
+	end
+	def condDQuote! (mode = true)
+		self.replate(condDQuote(mode))
+	end
+end
+
+
+class Array
+	def shellSQuote
+		arr = []
+		each do |e|
+			arr.push e.shellSQuote
+		end
+		arr.join(" ")
+	end
+	def condSQuote
+		arr = []
+		each do |e|
+			arr.push e.condSQuote
+		end
+		arr.join(" ")
+	end
+	def shellDQuote
+        arr = []
+        each do |e|
+            arr.push e.shellDQuote
+        end
+        arr.join(" ")
+	end
+	def condDQuote
+        arr = []
+        each do |e|
+            arr.push e.condDQuote
+        end
+        arr.join(" ")
+	end
+end
+
+
+
+require 'Yk/generator__.rb'
+require 'shellwords'
+
+
+class String
+	def dequote (env = nil)
+		ret = ""
+		g = nil
+		procEnv = Proc.new do
+			if env && g.current == ?$
+				if !g.next?
+					ret += "$"
+				else
+					g.inc
+					vName = g.current.chr
+					if g.current.chr =~ /^[A-Za-z_]$/
+						while g.next? && g.next.chr =~ /^\w$/
+							g.inc
+							vName += g.current.chr
+						end
+						if env[vName]
+							ret += env[vName]
+						end
+					elsif g.current.chr =~ /^\s$/
+						ret += "$#{g.current.chr}"
+					else
+						if env[vName]
+							ret += env[vName]
+						end
+					end
+				end
+				true
+			else
+				false
+			end
+		end
+		self.each__ :each_byte do |g|
+			case g.current.chr
+			when '"'
+				while true
+					g.inc
+					if g.current == ?\"
+						break
+					else
+						if g.current == ?\\
+							g.inc
+							case g.current
+							when ?0
+								ret += "\x00"
+							when ?n
+								ret += "\n"
+							when ?t
+								ret += "\t"
+							when ?r
+								ret += "\r"
+							when ?a
+								ret += "\a"
+							when ?x
+								tmp = ""
+								g.inc
+								tmp += g.current.chr
+								g.inc
+								tmp += g.current.chr
+								ret += tmp.to_x.chr
+							when ?\\
+								ret += "\\"
+							when ?\"
+								ret += '"'
+							when ?\'
+								ret += "'"
+							end
+						elsif !procEnv.call
+							ret += g.current.chr
+						end
+					end
+				end
+			when "'"
+				while true
+					g.inc
+					if g.current == ?\'
+						break
+					else
+						if g.current == ?\\ && g.next? && g.next == ?\'
+							g.inc
+						end
+						ret += g.current.chr
+					end
+				end
+			else
+				if !procEnv.call
+					ret += g.current.chr
+				end
+			end
+		end
+		ret
+	end
+	def shell_split (*envOrComtOrLim)
+		if envOrComtOrLim.size == 0
+			return Shellwords.shellwords(self)	
+		end
+		s = self
+		env = nil
+		comt = "#"
+		spPos = []
+		arr = []
+		lim = nil
+		com = ""
+		envOrComtOrLim.each do |e|
+			if e.is_a?(Hash) || e == ENV
+				env = e
+			elsif e.is_a? String
+				comt = e
+			elsif e.is_a? Integer
+				lim = e
+			end
+		end
+		s.each__ :each_byte do |g|
+			case g.current.chr
+			when '"'
+				begin
+					g.current == ?\\ && g.inc
+					g.inc
+				end while g.current != ?\"
+			when "'"
+				begin
+					g.current == ?\\ && g.next == ?\' && g.inc
+					g.inc
+				end while g.current != ?\'
+			when /^\s$/
+				start = g.index
+				while g.next? && g.next.chr =~ /^\s$/
+					g.inc
+				end
+				stop = g.index
+				spPos.push start..stop
+			when comt
+				spPos.push(tmp = g.index .. s.size - 1)
+				com = s[tmp]
+				break
+			end
+		end
+		prevLast1 = 0
+		spPos.each do |sr|
+			if sr.first != 0
+				if prevLast1 <= sr.first - 1
+					arr.push s[prevLast1 .. sr.first - 1]
+					if arr.size == lim
+						arr[-1] += s[sr.first ... s.size - com.size - 1]
+						arr[-1].rstrip!
+						prevLast1 = s.size
+						break
+					end
+				end
+			end
+			prevLast1 = sr.last + 1
+		end
+		if prevLast1 < s.size
+			arr.push s[prevLast1 .. s.size - 1]
+		end
+		arr.map!{ |e| e.dequote(env) }
+		def arr.com= (arg)
+			@com = arg
+		end
+		def arr.com
+			@com
+		end
+		arr.com = com
+		arr
+	end
+end
+
+
+class Array
+	def shell_join (*splOrQuots)
+		qMode = :dQuot
+		spl = " "
+		splOrQuots.each do |e|
+			case e
+			when ?\', "'"
+				qMode = :sQuot
+			when ?\", '"'
+				qMode = :dQuot
+			else
+				if e.is_a? Integer
+					spl = e.chr
+				else
+					spl = e
+				end
+			end
+		end
+		case qMode
+		when :sQuot
+			toJoin = map{ |e| e.condSQuote }
+		when :dQuot
+			toJoin = map{ |e| e.condDQuote }
+		end
+		if self.respond_to?(:com) && com != nil
+			toJoin += [com]
+		end
+		toJoin.join(spl)
+	end
+end
+
+