git.cworth.org Git - sup/blob - lib/sup/modes/console-mode.rb

   1 require 'pp'
   2
   3 module Redwood
   4
   5 class Console
   6   def initialize mode
   7     @mode = mode
   8   end
   9
  10   def query(query)
  11     Enumerable::Enumerator.new(Index, :each_message, Index.parse_query(query))
  12   end
  13
  14   def add_labels(query, *labels)
  15     query(query).each { |m| m.labels += labels; m.save Index }
  16   end
  17
  18   def remove_labels(query, *labels)
  19     query(query).each { |m| m.labels -= labels; m.save Index }
  20   end
  21
  22   def xapian; Index.instance.instance_variable_get :@xapian; end
  23   def ferret; Index.instance.instance_variable_get :@index; end
  24
  25   ## files that won't cause problems when reloaded
  26   ## TODO expand this list / convert to blacklist
  27   RELOAD_WHITELIST = %w(sup/xapian_index.rb sup/modes/console-mode.rb)
  28
  29   def reload
  30     old_verbose = $VERBOSE
  31     $VERBOSE = nil
  32     old_features = $".dup
  33     begin
  34       fs = $".grep(/^sup\//)
  35       fs.reject! { |f| not RELOAD_WHITELIST.member? f }
  36       fs.each { |f| $".delete f }
  37       fs.each do |f|
  38         @mode << "reloading #{f}\n"
  39         begin
  40           require f
  41         rescue LoadError => e
  42           raise unless e.message =~ /no such file to load/
  43         end
  44       end
  45     rescue Exception
  46       $".clear
  47       $".concat old_features
  48       raise
  49     ensure
  50       $VERBOSE = old_verbose
  51     end
  52     true
  53   end
  54 end
  55
  56 class ConsoleMode < LogMode
  57   def initialize
  58     super
  59     @binding = Console.new(self).instance_eval { binding }
  60   end
  61
  62   def execute cmd
  63     begin
  64       self << ">> #{cmd}\n"
  65       ret = eval cmd, @binding
  66       self << "=> #{ret.pretty_inspect}\n"
  67     rescue Exception
  68       self << "#{$!.class}: #{$!.message}\n"
  69       clean_backtrace = []
  70       $!.backtrace.each { |l| break if l =~ /console-mode/; clean_backtrace << l }
  71       clean_backtrace.each { |l| self << "#{l}\n" }
  72     end
  73   end
  74
  75   def prompt
  76     BufferManager.ask :console, "eval: "
  77   end
  78
  79   def run
  80     while true
  81       cmd = prompt or return
  82       execute cmd
  83     end
  84   end
  85 end
  86
  87 end