YkLib 0.1.0

data/lib/YkLib/Yk/rlprompt.rb

@@ -0,0 +1,73 @@
+
+require 'readline.so'
+
+class String
+	def prompt (echo = true)
+		if !STDIN.tty?
+			raise Exception.new("cannot use #{String@prompt} when STDIN is redirected\n")
+		end
+		prpt = self
+		aList = []
+		if self.rstrip =~ /\<(.*?)\>$/
+			pre = $`
+			aarr = $1.split(/\|/)
+			aarr.each_index do |i|
+				e = aarr[i]
+				case e
+				when "y"
+					aList.push ["y", "Yes"]
+					aarr[i] = "Yes"
+				when "n"
+					aList.push ["n", "No"]
+					aarr[i] = "No"
+				when "ya"
+					aList.push ["ya", "Yes to all"]
+					aarr[i] = "Yes to all(ya)"
+				when "na"
+					aList.push ["na", "No to all"]
+					aarr[i] = "No to all(na)"
+				when "a"
+					aList.push ["a", "All"]
+					aarr[i] = "All"
+				else
+					if e =~ /\[(.+?)\]/
+						c = $1[0].chr
+						e = $` + $1 + $'
+						aList.push [c, e]
+					elsif e =~ /\((.+?)\)/
+						c = $1[0].chr
+						e = $`
+						aList.push [c, e]
+					else
+						c = e[0].chr
+						aList.push [c, e]
+					end
+				end
+			end
+			prpt = "#{pre}<#{aarr.join("|")}>"
+		end
+		system "stty -echo"
+		begin
+			res = Readline.readline(prpt)
+		ensure
+			system "stty echo"
+		end
+		if aList && aList.size >= 2
+			if res == nil || (res = res.strip) == ""
+				return aList[0][0]
+			else
+				aList.each do |e|
+					e.each do |f|
+						if res =~ /^#{Regexp.escape f}$/i
+							return e[0]
+						end
+					end
+				end
+				return aList[0][0]
+			end
+		else
+			return res
+		end
+	end
+end
+

data/lib/YkLib/Yk/rootexec.rb

@@ -0,0 +1,48 @@
+
+require 'Yk/shellquote'
+
+
+if !defined? CYGWIN
+	CYGWIN = (`uname` =~ /CYGWIN/)
+end
+if CYGWIN
+	if !defined?(CYGADMIN)
+		begin
+			testFName = "/var/tmp/__test_admin__#{rand(10000000000)}"
+			File.open testFName, "w" do |fw|
+				File.chmod 0666, testFName
+			end
+			isAdmin = false
+			begin
+				require 'etc'
+				File.chown Etc.getpwnam("SYSTEM").uid, Etc.getgrnam("Administrators").gid, testFName
+				isAdmin = true
+			rescue
+			end
+			CYGADMIN = isAdmin
+		ensure
+			File.delete testFName
+		end
+	end
+end
+
+
+if !CYGWIN
+	if Process.euid != 0
+		if (File.executable?(tmp = "/usr/sbin/cansudo") && system(tmp) && $? == 0 && STDIN.tty?) or "/etc/group".read =~ /\nwheel|sudo:.*\b(#{Regexp.escape Etc.getpwuid(Process.euid).name})\b/
+				exec "sudo", $0, *ARGV
+		else
+			exec "su",  "-c",  "#{$0} #{ARGV.condSQuote}"
+		end
+	end
+else
+	if !CYGADMIN
+		if File.executable?(tmp = "/usr/bin/cygsu")
+			exec "cygsu #{$0} #{ARGV.condSQuote}"
+		else
+			raise Exception.new("cannot execute as an administrator")
+		end
+	end
+end
+
+