themigrator 0.1.12 → 0.1.14

data/lib/themigrator/script.rb

@@ -5,7 +5,7 @@ module Themigrator
     attr_reader :script, :log_file, :attr
 
     # new creates a new script but doesn't start until called run
-    # Arguments: 
+    # Arguments:
     # *  script : Fullpath of the file to execute
     # * log_file: Fullpath of the file to save the log
     # * attr: Extra attributes that could be handy (not used
@@ -16,7 +16,7 @@ module Themigrator
       @attr = attr
       @script = script
       @log_file = log_file
-      @log_fd = File.open(@log_file, "w", 0600)
+      @log_fd = File.open(@log_file, 'w', 0o600)
       @wd_dir = File.dirname(script)
       @pid = nil
       @thread = nil
@@ -31,36 +31,36 @@ module Themigrator
 
     def run
       @thread = Thread.new do
-	@pid = Process.spawn(@script, 
-			     err: @log_fd, 
-			     out: @log_fd, 
-			     chdir: @wd_dir)
-	pid, status = Process.wait2(@pid)
-	@end_time = Time.now
-	close
-	return_code = if status.signaled?
-	  signal = Signal.list.find{|k,v|
-	    v == status.termsig
-	  }
-	  signal.nil? ? :unknown_signal : signal[0].downcase.to_sym
-	else
-	  status.exitstatus
-	end
-
-	Thread.new { invoke_cbks } if @finish_cbk
-
-	return_code
+        @pid = Process.spawn(@script,
+                             err: @log_fd,
+                             out: @log_fd,
+                             chdir: @wd_dir)
+        pid, status = Process.wait2(@pid)
+        @end_time = Time.now
+        close
+        return_code = if status.signaled?
+                        signal = Signal.list.find do |_k, v|
+                          v == status.termsig
+                        end
+                        signal.nil? ? :unknown_signal : signal[0].downcase.to_sym
+                      else
+                        status.exitstatus
+        end
+
+        Thread.new { invoke_cbks } if @finish_cbk
+
+        return_code
       end
       sleep 0
     end
 
     def invoke_cbks
-      @finish_cbk.call(self)  
+      @finish_cbk.call(self)
     end
 
     # Waits for the process to be over
     def wait(seconds = nil)
-      @thread.join(seconds) 
+      @thread.join(seconds)
     end
 
     def exitcode
@@ -69,26 +69,24 @@ module Themigrator
 
     def stop
       if running?
-	send_int
-	if wait(5) == nil
-	  send_kill
-	end
-	wait
+        send_int
+        send_kill if wait(5).nil?
+        wait
       end
     end
 
     def success?
-      exitcode == 0
+      exitcode.zero?
     end
 
     private
 
     def running?
-      @thread and @thread.alive?
+      @thread && @thread.alive?
     end
 
     def send_signal(signal)
-      Process.kill(signal.to_s.upcase,@pid)
+      Process.kill(signal.to_s.upcase, @pid)
     end
 
     def send_int
@@ -102,6 +100,5 @@ module Themigrator
     def close
       @log_fd.close
     end
-
   end
 end

data/lib/themigrator/script_pool.rb

@@ -1,6 +1,5 @@
 require 'themigrator/script'
 
-
 module Themigrator
   class ScriptPool
     attr_reader :scripts, :run_time
@@ -10,10 +9,10 @@ module Themigrator
       @scripts = []
     end
 
-    def add_script(script,log_file, attr = {},&block)
+    def add_script(script, log_file, attr = {}, &block)
       s = Script.new(script, log_file, attr, &block)
       @lock.synchronize do
-	@scripts << s
+        @scripts << s
       end
       s
     end
@@ -25,8 +24,8 @@ module Themigrator
     def run_and_wait
       start = Time.now
       @lock.synchronize do
-	start_scripts
-	collect_result
+        start_scripts
+        collect_result
       end
     ensure
       @run_time = Time.now - start
@@ -36,33 +35,30 @@ module Themigrator
 
     def start_scripts
       @scripts.each do |s|
-	Thread.new {
-	  s.run
-	  s.wait
-	  @queue << s
-	}
+        Thread.new do
+          s.run
+          s.wait
+          @queue << s
+        end
       end
     end
 
-    def  collect_result
+    def collect_result
       running_scripts = @scripts.size
       success = true
-      while running_scripts != 0
-	s = @queue.pop
-	if !s.success?
-	  stop_and_kill
-	  success = false
-	end
-	running_scripts -= 1
+      while running_scripts.nonzero?
+        s = @queue.pop
+        unless s.success?
+          stop_and_kill
+          success = false
+        end
+        running_scripts -= 1
       end
       success
     end
 
     def stop_and_kill
-      @scripts.each {|s|
-	s.stop
-      }
+      @scripts.each(&:stop)
     end
-
   end
 end

data/lib/themigrator/ui/getchar.rb

@@ -0,0 +1,40 @@
+
+# From: http://www.alecjacobson.com/weblog/?p=75
+module Themigrator
+  class UI
+    module Getchar
+      KEY2SYM = Hash.new { |h, k| h[k] = k }
+      KEY2SYM["\e[D"] = :left
+      KEY2SYM["\e[C"] = :right
+      KEY2SYM["\r"] = :enter
+      KEY2SYM["\e"] = :esc
+      KEY2SYM["\e[6~"] = :npage
+      KEY2SYM["\e[5~"] = :ppage
+      KEY2SYM["\u0003"] = :c_c
+      KEY2SYM["\e[B"] = :scroll_down
+      KEY2SYM["\e[A"] = :scroll_up
+
+      def getchar
+        begin
+          c = STDIN.getc.chr
+          # gather next two characters of special keys
+          if c == "\e"
+            extra_thread = Thread.new do
+              c += STDIN.getc.chr
+              c += STDIN.getc.chr
+              c += STDIN.getc.chr if ["\e[6", "\e[5"].include?(c)
+            end
+            # wait just long enough for special keys to get swallowed
+            extra_thread.join(0.00100)
+            # kill thread so not-so-long special keys don't wait on getc
+            extra_thread.kill
+          end
+        rescue => e
+          log_exception(e)
+        ensure
+        end
+        KEY2SYM[c]
+      end
+    end
+  end
+end

data/lib/themigrator/ui/log_area.rb

@@ -0,0 +1,173 @@
+require 'curses'
+require 'thread'
+
+module Themigrator
+  class UI
+    module LogArea
+      include Curses
+
+      class LogWindow < Window
+        attr_accessor :role, :action
+        def initialize(height, width, top, left)
+          @height = height
+          @width = width
+          @top = top
+          @left = left
+          @inner_win = ::Curses::Window.new(height - 2, width - 2, top + 1, left + 1)
+          @inner_win.scrollok(true)
+          @buffer = ''
+          @scroll = -1
+          super
+          idlok(true)
+        end
+
+        def redraw
+          clear
+          @inner_win.clear
+          setpos(0, 0)
+          box('|', '-')
+          addstr(name)
+          log('redraw')
+          refresh
+          render(true)
+        end
+
+        def nextpage
+          @scroll += 5
+          @scroll = -1 if @scroll > -1
+          log("@scroll = #{@scroll}")
+        end
+
+        def prevpage
+          @scroll -= 5
+          buffer_lines = @buffer.split("\n").size
+          offset_lines = buffer_lines - @height + (2 + 5)
+
+          log("@scroll = #{@scroll} and offset_lines - #{offset_lines} | buffer_lines #{buffer_lines} | @height = #{@height}")
+          @height
+          @scroll = -offset_lines if @scroll < -offset_lines
+        end
+
+        def name
+          '%-13s/%-10s' % [action, role]
+        end
+
+        def append(msg)
+          @buffer << msg
+        end
+
+        def render(force = false)
+          # refresh
+          render_inner_win(force)
+        end
+
+        def log(msg)
+          ui_log("#{name}##{msg}")
+        end
+
+        private
+
+        def <<(msg)
+          @inner_win << msg
+        end
+
+        def render_inner_win(force = false)
+          new_content = content_changed?
+          return unless new_content || force
+          log('render')
+          @inner_win.clear
+          @inner_win << @buffer.split("\n")[0..@scroll].join("\n")
+          @inner_win.refresh
+        end
+
+        def get_content_for_win
+          @buffer.split("\n")[0..@scroll].join("\n")
+        end
+
+        def content_changed?
+          new_content = get_content_for_win
+          if @previous == new_content
+            nil
+          else
+            @previous = get_content_for_win
+            new_content
+          end
+        end
+      end
+
+      def initialize_log_area(roles)
+        @la_roles = roles
+        @la_scripts = Themigrator::Migration::SCRIPTS
+        @la_actions = Themigrator::Migration::ACTIONS
+        @la_current_action = nil
+        @la_lines = lines
+        @la_cols = cols
+        @la_mutex = Mutex.new
+        # @windows = Hash.new {|h,a| h[a] = {} }
+        # @windows[action][role]
+        @la_windows
+        @la_windows = Hash.new do |h, action|
+          role_windows = {}
+          @la_roles.each_with_index do |role, n|
+            column_height = @la_lines - 4
+            column_width = @la_cols / @la_roles.size
+            top = 2
+            left = column_width * n
+            w = LogWindow.new(column_height, column_width - 1, top, left)
+            w.role = role
+            w.action = action
+            role_windows[role] = w
+          end
+          h[action] = role_windows
+        end
+      end
+
+      def la_initialize?
+        defined?(@la_roles)
+      end
+
+      def la_npage
+        la_each_window(&:nextpage)
+      end
+
+      def la_ppage
+        la_each_window(&:prevpage)
+      end
+
+      def la_switch_action(action)
+        return if @la_current_action == action || @la_roles.empty?
+        @la_current_action = action.to_s
+        @la_roles.each do |role|
+          la_get_window(action, role).redraw
+        end
+      end
+
+      def la_append(action, role, msg)
+        @la_mutex.synchronize do
+          la_get_window(action, role).append(msg)
+        end
+      end
+
+      def la_render
+        la_each_window do |w, _role|
+          w.render
+        end
+      end
+
+      private
+
+      def la_each_window
+        @la_roles.each do |role|
+          @la_mutex.synchronize do
+            yield la_get_window(@la_current_action, role), role
+          end
+        end
+      end
+
+      def la_get_window(action, role)
+        fuck @la_windows[action][role]
+        @la_windows[action][role]
+      end
+    end
+  end
+end

data/lib/themigrator/ui/progress_window.rb

@@ -0,0 +1,52 @@
+
+module Themigrator
+  class UI
+    module ProgressWindow
+      include Curses
+      def render_progress
+        progress_window.clear
+        progress_window.setpos(0, 0)
+        progress_window.addstr(build_status_line)
+        progress_window.setpos(1, 0)
+        progress_window.addstr(@pw_status) if defined?(@pw_status)
+        progress_window.refresh
+      end
+
+      def pg_set_action(action)
+        @pg_action = action
+      end
+
+      def progress_window
+        @progress_widnow ||= Window.new(@lines - 2, @cols, @lines - 2, 0).tap do |w|
+          w.attrset(A_DIM)
+        end
+      end
+
+      def say(*msg)
+        @pw_status = msg.map(&:to_s).join(' ')
+      end
+
+      def build_status_line
+        scripts = Themigrator::Migration::SCRIPTS
+        width = scripts.map(&:size).inject(&:+)
+        free_space = @cols - width
+        sep_space = scripts.size - 1
+        inline_space = ((free_space - sep_space) / scripts.size - 1) / 2
+
+        action_msg = []
+        scripts.map do |script|
+          a = ''
+          inline_space.times { a << ' ' }
+          a << if defined?(@pg_action) && @pg_action.to_s == script.to_s
+                 script.upcase
+               else
+                 script.downcase
+               end
+          inline_space.times { a << ' ' }
+          action_msg << a
+        end
+        action_msg.join('>')
+      end
+    end
+  end
+end

data/lib/themigrator/ui/title_bar.rb

@@ -0,0 +1,31 @@
+
+module Themigrator
+  class UI
+    module TitleBar
+      include Curses
+
+      protected
+
+      def render_title
+        now_str = Time.now.strftime('%d %b - %H:%M.%S')
+        time_window.setpos(0, 0)
+        time_window.addstr(now_str)
+        time_window.refresh
+      end
+
+      def title_window
+        @title_window ||= Window.new(1, @cols - 18, 0, 0).tap do |w|
+          w.attrset(A_BOLD)
+          w.addstr('The migrator')
+        end
+      end
+
+      def time_window
+        width = 18
+        @time_window ||= Window.new(1, 18, 0, @cols - 18).tap do |w|
+          w.attrset(A_DIM)
+        end
+        end
+    end
+  end
+end