From 42a81120e510c2c2fb7822ff4d5738db6ba392bf Mon Sep 17 00:00:00 2001
From: keiju <keiju@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Tue, 20 Mar 2007 12:38:58 +0000
Subject: 	* lib/shell.rb, lib/shell: support for ruby 1.9(YARV) thread
 model.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@12110 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 lib/shell/builtin-command.rb    |  31 +++++---
 lib/shell/command-processor.rb  |  87 +++++++++++++---------
 lib/shell/error.rb              |   4 +-
 lib/shell/process-controller.rb | 158 +++++++++++++++++++++++++++-------------
 lib/shell/system-command.rb     |   4 +-
 lib/shell/version.rb            |   8 +-
 6 files changed, 190 insertions(+), 102 deletions(-)

(limited to 'lib/shell')

diff --git a/lib/shell/builtin-command.rb b/lib/shell/builtin-command.rb
index db1adfa48b..cfdbc046aa 100644
--- a/lib/shell/builtin-command.rb
+++ b/lib/shell/builtin-command.rb
@@ -1,9 +1,9 @@
 #
 #   shell/builtin-command.rb - 
-#   	$Release Version: 0.6.0 $
+#   	$Release Version: 0.7 $
 #   	$Revision$
 #   	$Date$
-#   	by Keiju ISHITSUKA(Nihon Rational Software Co.,Ltd)
+#   	by Keiju ISHITSUKA(keiju@ruby-lang.org)
 #
 # --
 #
@@ -22,6 +22,16 @@ class Shell
     end
   end
 
+  class Void < BuiltInCommand
+    def initialize(sh, *opts)
+      super sh
+    end
+    
+    def each(rs = nil)
+      # do nothing
+    end
+  end
+
   class Echo < BuiltInCommand
     def initialize(sh, *strings)
       super sh
@@ -58,20 +68,17 @@ class Shell
       super sh
 
       @pattern = pattern
-      Thread.critical = true
-      back = Dir.pwd
-      begin
-	Dir.chdir @shell.cwd
-	@files = Dir[pattern]
-      ensure
-	Dir.chdir back
-	Thread.critical = false
-      end
     end
 
     def each(rs = nil)
+      if @pattern[0] == ?/ 
+	@files = Dir[@pattern]
+      else
+	prefix = @shell.pwd+"/"
+	@files = Dir[prefix+@pattern].collect{|p| p.sub(prefix, "")}
+      end
       rs =  @shell.record_separator unless rs
-      for f  in @files
+      for f in @files
 	yield f+rs
       end
     end
diff --git a/lib/shell/command-processor.rb b/lib/shell/command-processor.rb
index ecf6c7d5eb..3514feb6ce 100644
--- a/lib/shell/command-processor.rb
+++ b/lib/shell/command-processor.rb
@@ -1,9 +1,9 @@
 #
 #   shell/command-controller.rb - 
-#   	$Release Version: 0.6.0 $
+#   	$Release Version: 0.7 $
 #   	$Revision$
 #   	$Date$
-#   	by Keiju ISHITSUKA(Nippon Rational Inc.)
+#   	by Keiju ISHITSUKA(keiju@ruby-lang.org)
 #
 # --
 #
@@ -26,7 +26,13 @@ class Shell
     #
     # initialize of Shell and related classes.
     #
-    NoDelegateMethods = ["initialize", "expand_path"]
+    m = [:initialize, :expand_path]
+    if Object.methods.first.kind_of?(String)
+      NoDelegateMethods = m.collect{|m| m.id2name}
+    else
+      NoDelegateMethods = m
+    end
+
     def self.initialize
 
       install_builtin_commands
@@ -111,13 +117,20 @@ class Shell
     #	  Dir#open  (when path is directory)
     #	mode has an effect only when path is a file
     #
-    def open(path, mode)
+    def open(path, mode = nil, perm = 0666, &b)
       path = expand_path(path)
       if File.directory?(path)
-	Dir.open(path)
+	Dir.open(path, &b)
       else
-	effect_umask do
-	  File.open(path, mode)
+	if @shell.umask
+	  f = File.open(path, mode, perm)
+	  File.chmod(perm & ~@shell.umask, path)
+	  if block_given?
+	    f.each &b
+	  end
+	  f
+	else
+	  f = File.open(path, mode, perm, &b)
 	end
       end
     end
@@ -130,12 +143,15 @@ class Shell
     #	  File#unlink (when path is file)
     #
     def unlink(path)
+      @shell.check_point
+
       path = expand_path(path)
       if File.directory?(path)
 	Dir.unlink(path)
       else
 	IO.unlink(path)
       end
+      Void.new(@shell)
     end
 
     #
@@ -155,6 +171,7 @@ class Shell
     #	  sh[:exists?, "foo"]
     #	  sh["exists?", "foo"]
     #	  
+    alias top_level_test test
     def test(command, file1, file2=nil)
       file1 = expand_path(file1)
       file2 = expand_path(file2) if file2
@@ -162,7 +179,11 @@ class Shell
 
       case command
       when Integer
-	top_level_test(command, file1, file2)
+	if file2
+	  top_level_test(command, file1, file2)
+	else
+	  top_level_test(command, file1)
+	end
       when String
 	if command.size == 1
 	  if file2
@@ -194,9 +215,23 @@ class Shell
     #	same as Dir.mkdir()
     #	  
     def mkdir(*path)
+      @shell.check_point
+      notify("mkdir #{path.join(' ')}")
+      
+      perm = nil
+      if path.last.kind_of?(Integer)
+	perm = path.pop
+      end
       for dir in path
-	Dir.mkdir(expand_path(dir))
+	d = expand_path(dir)
+	if perm
+	  Dir.mkdir(d, perm)
+	else
+	  Dir.mkdir(d)
+	end
+	File.chmod(d, 0666 & ~@shell.umask) if @shell.umask
       end
+      Void.new(@shell)
     end
 
     #
@@ -205,9 +240,13 @@ class Shell
     #	same as Dir.rmdir()
     #	  
     def rmdir(*path)
+      @shell.check_point
+      notify("rmdir #{path.join(' ')}")
+
       for dir in path
 	Dir.rmdir(expand_path(dir))
       end
+      Void.new(@shell)
     end
 
     #
@@ -299,35 +338,17 @@ class Shell
 
     # %pwd, %cwd -> @pwd
     def notify(*opts, &block)
-      Thread.exclusive do
-	Shell.notify(*opts) {|mes|
-	  yield mes if iterator?
+      Shell.notify(*opts) {|mes|
+	yield mes if iterator?
 	
-	  mes.gsub!("%pwd", "#{@cwd}")
-	  mes.gsub!("%cwd", "#{@cwd}")
-	}
-      end
+	mes.gsub!("%pwd", "#{@cwd}")
+	mes.gsub!("%cwd", "#{@cwd}")
+      }
     end
 
     #
     # private functions
     #
-    def effect_umask
-      if @shell.umask
-	Thread.critical = true
-	save = File.umask
-	begin
-	  yield
-	ensure
-	  File.umask save
-	  Thread.critical = false
-	end
-      else
-	yield
-      end
-    end
-    private :effect_umask
-
     def find_system_command(command)
       return command if /^\// =~ command
       case path = @system_commands[command]
@@ -343,7 +364,7 @@ class Shell
 
       for p in @shell.system_path
 	path = join(p, command)
-	if FileTest.exists?(path)
+	if FileTest.exist?(path)
 	  @system_commands[command] = path
 	  return path
 	end
diff --git a/lib/shell/error.rb b/lib/shell/error.rb
index d338b1c5d1..97df30b3c2 100644
--- a/lib/shell/error.rb
+++ b/lib/shell/error.rb
@@ -1,9 +1,9 @@
 #
 #   shell/error.rb - 
-#   	$Release Version: 0.6.0 $
+#   	$Release Version: 0.7 $
 #   	$Revision$
 #   	$Date$
-#   	by Keiju ISHITSUKA(Nihon Rational Software Co.,Ltd)
+#   	by Keiju ISHITSUKA(keiju@ruby-lang.org)
 #
 # --
 #
diff --git a/lib/shell/process-controller.rb b/lib/shell/process-controller.rb
index 8929805506..c38d8bc987 100644
--- a/lib/shell/process-controller.rb
+++ b/lib/shell/process-controller.rb
@@ -1,33 +1,38 @@
 #
 #   shell/process-controller.rb - 
-#   	$Release Version: 0.6.0 $
+#   	$Release Version: 0.7 $
 #   	$Revision$
 #   	$Date$
-#   	by Keiju ISHITSUKA(Nihon Rational Software Co.,Ltd)
+#   	by Keiju ISHITSUKA(keiju@ruby-lang.org)
 #
 # --
 #
 #   
 #
+require "forwardable"
 
-require "mutex_m"
-require "monitor"
+require "thread"
 require "sync"
 
 class Shell
   class ProcessController
 
     @ProcessControllers = {}
-    @ProcessControllers.extend Mutex_m
+    @ProcessControllersMonitor = Mutex.new
+    @ProcessControllersCV = ConditionVariable.new
+
+    @BlockOutputMonitor = Mutex.new
+    @BlockOutputCV = ConditionVariable.new
 
     class<<self
+      extend Forwardable
 
-      def process_controllers_exclusive
-	begin
-	  @ProcessControllers.lock unless Thread.critical 
-	  yield
-	ensure
-	  @ProcessControllers.unlock unless Thread.critical 
+      def_delegator("@ProcessControllersMonitor", 
+		    "synchronize", "process_controllers_exclusive")
+
+      def active_process_controllers
+	process_controllers_exclusive do
+	  @ProcessControllers.dup
 	end
       end
 
@@ -43,6 +48,7 @@ class Shell
 	  if @ProcessControllers[pc]
 	    if (@ProcessControllers[pc] -= 1) == 0
 	      @ProcessControllers.delete(pc)
+	      @ProcessControllersCV.signal
 	    end
 	  end
 	end
@@ -55,6 +61,34 @@ class Shell
 	  end
 	end
       end
+
+      def block_output_synchronize(&b)
+	@BlockOutputMonitor.synchronize &b
+      end
+
+      def wait_to_finish_all_process_controllers
+	process_controllers_exclusive do
+	  while !@ProcessControllers.empty?
+	    Shell::notify("Process finishing, but active shell exists",
+			  "You can use Shell#transact or Shell#check_point for more safe execution.")
+	    if Shell.debug?
+	      for pc in @ProcessControllers.keys
+		Shell::notify(" Not finished jobs in "+pc.shell.to_s)
+ 		for com in pc.jobs
+ 		  com.notify("  Jobs: %id")
+ 		end
+	      end
+	    end
+	    @ProcessControllersCV.wait(@ProcessControllersMonitor)
+	  end
+	end
+      end
+    end
+
+    # for shell-command complete finish at this prosess exit.
+    USING_AT_EXIT_WHEN_PROCESS_EXIT = true
+    at_exit do
+      wait_to_finish_all_process_controllers unless $@
     end
 
     def initialize(shell)
@@ -67,6 +101,8 @@ class Shell
       @job_condition = ConditionVariable.new
     end
 
+    attr_reader :shell
+
     def jobs
       jobs = []
       @jobs_sync.synchronize(:SH) do
@@ -122,15 +158,19 @@ class Shell
 	  @waiting_jobs.delete command
 	else
 	  command = @waiting_jobs.shift
+#	  command.notify "job(%id) pre-start.", @shell.debug?
+	  
 	  return unless command
 	end
 	@active_jobs.push command
 	command.start
+#	command.notify "job(%id) post-start.", @shell.debug?
 
 	# start all jobs that input from the job
-	for job in @waiting_jobs
+	for job in @waiting_jobs.dup
 	  start_job(job) if job.input == command
 	end
+#	command.notify "job(%id) post2-start.", @shell.debug?
       end
     end
 
@@ -152,6 +192,7 @@ class Shell
 	@active_jobs.delete command
 	ProcessController.inactivate(self)
 	if @active_jobs.empty?
+	  command.notify("start_jon in ierminate_jon(%id)", Shell::debug?)
 	  start_job
 	end
       end
@@ -159,7 +200,7 @@ class Shell
 
     # kill a job
     def kill_job(sig, command)
-      @jobs_sync.synchronize(:SH) do
+      @jobs_sync.synchronize(:EX) do
 	if @waiting_jobs.delete command
 	  ProcessController.inactivate(self)
 	  return
@@ -183,6 +224,9 @@ class Shell
 	begin
 	  while !jobs.empty?
 	    @job_condition.wait(@job_monitor)
+	    for job in jobs
+	      job.notify("waiting job(%id)", Shell::debug?)
+	    end
 	  end
 	ensure
 	  redo unless jobs.empty?
@@ -194,66 +238,82 @@ class Shell
     def sfork(command, &block)
       pipe_me_in, pipe_peer_out = IO.pipe
       pipe_peer_in, pipe_me_out = IO.pipe
-      Thread.critical = true
 
-      STDOUT.flush
-      ProcessController.each_active_object do |pc|
-	for jobs in pc.active_jobs
-	  jobs.flush
-	end
-      end
-      
-      pid = fork {
-	Thread.critical = true
-
-	Thread.list.each do |th| 
-	  th.kill unless [Thread.main, Thread.current].include?(th)
-	end
 
-	STDIN.reopen(pipe_peer_in)
-	STDOUT.reopen(pipe_peer_out)
+      pid = nil
+      pid_mutex = Mutex.new
+      pid_cv = ConditionVariable.new
 
-	ObjectSpace.each_object(IO) do |io| 
-	  if ![STDIN, STDOUT, STDERR].include?(io)
-	    io.close unless io.closed?
+      Thread.start do
+	ProcessController.block_output_synchronize do
+	  STDOUT.flush
+	  ProcessController.each_active_object do |pc|
+	    for jobs in pc.active_jobs
+	      jobs.flush
+	    end
 	  end
+
+	  pid = fork {
+	    Thread.list.each do |th| 
+#	      th.kill unless [Thread.main, Thread.current].include?(th)
+	      th.kill unless Thread.current == th
+	    end
+
+	    STDIN.reopen(pipe_peer_in)
+	    STDOUT.reopen(pipe_peer_out)
+
+	    ObjectSpace.each_object(IO) do |io| 
+	      if ![STDIN, STDOUT, STDERR].include?(io)
+		io.close unless io.closed?
+	      end
+	    end
+
+	    yield
+	  }
 	end
-	yield
-      }
+	pid_cv.signal
 
-      pipe_peer_in.close
-      pipe_peer_out.close
-      command.notify "job(%name:##{pid}) start", @shell.debug?
-      Thread.critical = false
+	pipe_peer_in.close
+	pipe_peer_out.close
+	command.notify "job(%name:##{pid}) start", @shell.debug?
 
-      th = Thread.start {
-	Thread.critical = true
 	begin
 	  _pid = nil
 	  command.notify("job(%id) start to waiting finish.", @shell.debug?)
-	  Thread.critical = false
 	  _pid = Process.waitpid(pid, nil)
 	rescue Errno::ECHILD
 	  command.notify "warn: job(%id) was done already waitipd."
 	  _pid = true
+	  #	rescue
+	  #	  STDERR.puts $!
 	ensure
+	  command.notify("Job(%id): Wait to finish when Process finished.", @shell.debug?)
 	  # when the process ends, wait until the command termintes
-	  if _pid
+	  if USING_AT_EXIT_WHEN_PROCESS_EXIT or _pid
 	  else
 	    command.notify("notice: Process finishing...",
 			   "wait for Job[%id] to finish.",
 			   "You can use Shell#transact or Shell#check_point for more safe execution.")
 	    redo
 	  end
-	  Thread.exclusive do
-	    @job_monitor.synchronize do 
-	      terminate_job(command)
-	      @job_condition.signal
-	      command.notify "job(%id) finish.", @shell.debug?
-	    end
+	  
+#	  command.notify "job(%id) pre-pre-finish.", @shell.debug?
+	  @job_monitor.synchronize do 
+#	    command.notify "job(%id) pre-finish.", @shell.debug?
+	    terminate_job(command)
+#	    command.notify "job(%id) pre-finish2.", @shell.debug?
+	    @job_condition.signal
+	    command.notify "job(%id) finish.", @shell.debug?
 	  end
 	end
-      }
+      end
+
+      pid_mutex.synchronize do
+	while !pid
+	  pid_cv.wait(pid_mutex)
+	end
+      end
+
       return pid, pipe_me_in, pipe_me_out
     end
   end
diff --git a/lib/shell/system-command.rb b/lib/shell/system-command.rb
index f87ba890ea..cefa5d6e5a 100644
--- a/lib/shell/system-command.rb
+++ b/lib/shell/system-command.rb
@@ -1,9 +1,9 @@
 #
 #   shell/system-command.rb - 
-#   	$Release Version: 0.6.0 $
+#   	$Release Version: 0.7 $
 #   	$Revision$
 #   	$Date$
-#   	by Keiju ISHITSUKA(Nihon Rational Software Co.,Ltd)
+#   	by Keiju ISHITSUKA(keiju@ruby-lang.org)
 #
 # --
 #
diff --git a/lib/shell/version.rb b/lib/shell/version.rb
index 6694c804d8..e40aa03abb 100644
--- a/lib/shell/version.rb
+++ b/lib/shell/version.rb
@@ -1,9 +1,9 @@
 #
 #   version.rb - shell version definition file
-#   	$Release Version: 0.6.0$
+#   	$Release Version: 0.7$
 #   	$Revision$
 #   	$Date$
-#   	by Keiju ISHITSUKA(Nihon Rational Software Co.,Ltd)
+#   	by Keiju ISHITSUKA(keiju@ruby-lang.org)
 #
 # --
 #
@@ -11,6 +11,6 @@
 #
 
 class Shell
-  @RELEASE_VERSION = "0.6.0"
-  @LAST_UPDATE_DATE = "01/03/19"
+  @RELEASE_VERSION = "0.7"
+  @LAST_UPDATE_DATE = "07/03/20"
 end
-- 
cgit v1.2.3