git.cworth.org Git - sup/blob - lib/sup.rb

   1 require 'rubygems'
   2 require 'yaml'
   3 require 'zlib'
   4 require 'thread'
   5 require 'fileutils'
   6 require 'gettext'
   7 require 'curses'
   8
   9 ## the following magic enables wide characters when used with a ruby
  10 ## ncurses.so that's been compiled against libncursesw. (note the w.) why
  11 ## this works, i have no idea. much like pretty much every aspect of
  12 ## dealing with curses.  cargo cult programming at its best.
  13
  14 require 'dl/import'
  15 module LibC
  16   extend DL::Importable
  17   dlload Config::CONFIG['arch'] =~ /darwin/ ? "libc.dylib" : "libc.so.6"
  18   extern "void setlocale(int, const char *)"
  19 end
  20 LibC.setlocale(6, "")  # LC_ALL == 6
  21
  22 class Object
  23   ## this is for debugging purposes because i keep calling #id on the
  24   ## wrong object and i want it to throw an exception
  25   def id
  26     raise "wrong id called on #{self.inspect}"
  27   end
  28 end
  29
  30 class Module
  31   def yaml_properties *props
  32     props = props.map { |p| p.to_s }
  33     vars = props.map { |p| "@#{p}" }
  34     klass = self
  35     path = klass.name.gsub(/::/, "/")
  36
  37     klass.instance_eval do
  38       define_method(:to_yaml_properties) { vars }
  39       define_method(:to_yaml_type) { "!#{Redwood::YAML_DOMAIN},#{Redwood::YAML_DATE}/#{path}" }
  40     end
  41
  42     YAML.add_domain_type("#{Redwood::YAML_DOMAIN},#{Redwood::YAML_DATE}", path) do |type, val|
  43       klass.new(*props.map { |p| val[p] })
  44     end
  45   end
  46 end
  47
  48 module Redwood
  49   VERSION = "git"
  50
  51   BASE_DIR   = ENV["SUP_BASE"] || File.join(ENV["HOME"], ".sup")
  52   CONFIG_FN  = File.join(BASE_DIR, "config.yaml")
  53   COLOR_FN   = File.join(BASE_DIR, "colors.yaml")
  54   SOURCE_FN  = File.join(BASE_DIR, "sources.yaml")
  55   LABEL_FN   = File.join(BASE_DIR, "labels.txt")
  56   PERSON_FN  = File.join(BASE_DIR, "people.txt")
  57   CONTACT_FN = File.join(BASE_DIR, "contacts.txt")
  58   DRAFT_DIR  = File.join(BASE_DIR, "drafts")
  59   SENT_FN    = File.join(BASE_DIR, "sent.mbox")
  60   LOCK_FN    = File.join(BASE_DIR, "lock")
  61   SUICIDE_FN = File.join(BASE_DIR, "please-kill-yourself")
  62   HOOK_DIR   = File.join(BASE_DIR, "hooks")
  63
  64   YAML_DOMAIN = "masanjin.net"
  65   YAML_DATE = "2006-10-01"
  66
  67   ## record exceptions thrown in threads nicely
  68   @exceptions = []
  69   @exception_mutex = Mutex.new
  70
  71   attr_reader :exceptions
  72   def record_exception e, name
  73     @exception_mutex.synchronize do
  74       @exceptions ||= []
  75       @exceptions << [e, name]
  76     end
  77   end
  78
  79   def reporting_thread name
  80     if $opts[:no_threads]
  81       yield
  82     else
  83       ::Thread.new do
  84         begin
  85           yield
  86         rescue Exception => e
  87           record_exception e, name
  88         end
  89       end
  90     end
  91   end
  92
  93   module_function :reporting_thread, :record_exception, :exceptions
  94
  95 ## one-stop shop for yamliciousness
  96   def save_yaml_obj object, fn, safe=false
  97     if safe
  98       safe_fn = "#{File.dirname fn}/safe_#{File.basename fn}"
  99       mode = File.stat(fn).mode if File.exists? fn
 100       File.open(safe_fn, "w", mode) { |f| f.puts object.to_yaml }
 101       FileUtils.mv safe_fn, fn
 102     else
 103       File.open(fn, "w") { |f| f.puts object.to_yaml }
 104     end
 105   end
 106
 107   def load_yaml_obj fn, compress=false
 108     if File.exists? fn
 109       if compress
 110         Zlib::GzipReader.open(fn) { |f| YAML::load f }
 111       else
 112         YAML::load_file fn
 113       end
 114     end
 115   end
 116
 117   def start
 118     Redwood::PersonManager.new Redwood::PERSON_FN
 119     Redwood::SentManager.new Redwood::SENT_FN
 120     Redwood::ContactManager.new Redwood::CONTACT_FN
 121     Redwood::LabelManager.new Redwood::LABEL_FN
 122     Redwood::AccountManager.new $config[:accounts]
 123     Redwood::DraftManager.new Redwood::DRAFT_DIR
 124     Redwood::UpdateManager.new
 125     Redwood::PollManager.new
 126     Redwood::SuicideManager.new Redwood::SUICIDE_FN
 127     Redwood::CryptoManager.new
 128   end
 129
 130   def finish
 131     Redwood::LabelManager.save if Redwood::LabelManager.instantiated?
 132     Redwood::ContactManager.save if Redwood::ContactManager.instantiated?
 133     Redwood::PersonManager.save if Redwood::PersonManager.instantiated?
 134     Redwood::BufferManager.deinstantiate! if Redwood::BufferManager.instantiated?
 135   end
 136
 137   ## not really a good place for this, so I'll just dump it here.
 138   ##
 139   ## a source error is either a FatalSourceError or an OutOfSyncSourceError.
 140   ## the superclass SourceError is just a generic.
 141   def report_broken_sources opts={}
 142     return unless BufferManager.instantiated?
 143
 144     broken_sources = Index.sources.select { |s| s.error.is_a? FatalSourceError }
 145     unless broken_sources.empty?
 146       BufferManager.spawn_unless_exists("Broken source notification for #{broken_sources.join(',')}", opts) do
 147         TextMode.new(<<EOM)
 148 Source error notification
 149 -------------------------
 150
 151 Hi there. It looks like one or more message sources is reporting
 152 errors. Until this is corrected, messages from these sources cannot
 153 be viewed, and new messages will not be detected.
 154
 155 #{broken_sources.map { |s| "Source: " + s.to_s + "\n Error: " + s.error.message.wrap(70).join("\n        ")}.join("\n\n")}
 156 EOM
 157 #' stupid ruby-mode
 158       end
 159     end
 160
 161     desynced_sources = Index.sources.select { |s| s.error.is_a? OutOfSyncSourceError }
 162     unless desynced_sources.empty?
 163       BufferManager.spawn_unless_exists("Out-of-sync source notification for #{broken_sources.join(',')}", opts) do
 164         TextMode.new(<<EOM)
 165 Out-of-sync source notification
 166 -------------------------------
 167
 168 Hi there. It looks like one or more sources has fallen out of sync
 169 with my index. This can happen when you modify these sources with
 170 other email clients. (Sorry, I don't play well with others.)
 171
 172 Until this is corrected, messages from these sources cannot be viewed,
 173 and new messages will not be detected. Luckily, this is easy to correct!
 174
 175 #{desynced_sources.map do |s|
 176   "Source: " + s.to_s +
 177    "\n Error: " + s.error.message.wrap(70).join("\n        ") +
 178    "\n   Fix: sup-sync --changed #{s.to_s}"
 179   end}
 180 EOM
 181 #' stupid ruby-mode
 182       end
 183     end
 184   end
 185
 186   module_function :save_yaml_obj, :load_yaml_obj, :start, :finish,
 187                   :report_broken_sources
 188 end
 189
 190 ## set up default configuration file
 191 if File.exists? Redwood::CONFIG_FN
 192   $config = Redwood::load_yaml_obj Redwood::CONFIG_FN
 193 else
 194   require 'etc'
 195   require 'socket'
 196   name = Etc.getpwnam(ENV["USER"]).gecos.split(/,/).first rescue nil
 197   name ||= ENV["USER"]
 198   email = ENV["USER"] + "@" +
 199     begin
 200       Socket.gethostbyname(Socket.gethostname).first
 201     rescue SocketError
 202       Socket.gethostname
 203     end
 204
 205   $config = {
 206     :accounts => {
 207       :default => {
 208         :name => name,
 209         :email => email,
 210         :alternates => [],
 211         :sendmail => "/usr/sbin/sendmail -oem -ti",
 212         :signature => File.join(ENV["HOME"], ".signature")
 213       }
 214     },
 215     :editor => ENV["EDITOR"] || "/usr/bin/vim -f -c 'setlocal spell spelllang=en_us' -c 'set filetype=mail'",
 216     :thread_by_subject => false,
 217     :edit_signature => false,
 218     :ask_for_cc => true,
 219     :ask_for_bcc => false,
 220     :ask_for_subject => true,
 221     :confirm_no_attachments => true,
 222     :confirm_top_posting => true,
 223     :discard_snippets_from_encrypted_messages => false,
 224   }
 225   begin
 226     FileUtils.mkdir_p Redwood::BASE_DIR
 227     Redwood::save_yaml_obj $config, Redwood::CONFIG_FN
 228   rescue StandardError => e
 229     $stderr.puts "warning: #{e.message}"
 230   end
 231 end
 232
 233 require "sup/util"
 234 require "sup/hook"
 235
 236 ## we have to initialize this guy first, because other classes must
 237 ## reference it in order to register hooks, and they do that at parse
 238 ## time.
 239 Redwood::HookManager.new Redwood::HOOK_DIR
 240
 241 ## everything we need to get logging working
 242 require "sup/buffer"
 243 require "sup/keymap"
 244 require "sup/mode"
 245 require "sup/modes/scroll-mode"
 246 require "sup/modes/text-mode"
 247 require "sup/modes/log-mode"
 248 require "sup/logger"
 249 module Redwood
 250   def log s; Logger.log s; end
 251   module_function :log
 252 end
 253
 254 ## determine encoding and character set
 255   $encoding = Locale.current.charset
 256   if $encoding
 257     Redwood::log "using character set encoding #{$encoding.inspect}"
 258   else
 259     Redwood::log "warning: can't find character set by using locale, defaulting to utf-8"
 260     $encoding = "utf-8"
 261   end
 262
 263 ## now everything else (which can feel free to call Redwood::log at load time)
 264 require "sup/update"
 265 require "sup/suicide"
 266 require "sup/message-chunks"
 267 require "sup/message"
 268 require "sup/source"
 269 require "sup/mbox"
 270 require "sup/maildir"
 271 require "sup/imap"
 272 require "sup/person"
 273 require "sup/account"
 274 require "sup/thread"
 275 require "sup/index"
 276 require "sup/textfield"
 277 require "sup/colormap"
 278 require "sup/label"
 279 require "sup/contact"
 280 require "sup/tagger"
 281 require "sup/draft"
 282 require "sup/poll"
 283 require "sup/crypto"
 284 require "sup/horizontal-selector"
 285 require "sup/modes/line-cursor-mode"
 286 require "sup/modes/help-mode"
 287 require "sup/modes/edit-message-mode"
 288 require "sup/modes/compose-mode"
 289 require "sup/modes/resume-mode"
 290 require "sup/modes/forward-mode"
 291 require "sup/modes/reply-mode"
 292 require "sup/modes/label-list-mode"
 293 require "sup/modes/contact-list-mode"
 294 require "sup/modes/thread-view-mode"
 295 require "sup/modes/thread-index-mode"
 296 require "sup/modes/label-search-results-mode"
 297 require "sup/modes/search-results-mode"
 298 require "sup/modes/person-search-results-mode"
 299 require "sup/modes/inbox-mode"
 300 require "sup/modes/buffer-list-mode"
 301 require "sup/modes/poll-mode"
 302 require "sup/modes/file-browser-mode"
 303 require "sup/modes/completion-mode"
 304 require "sup/sent"
 305
 306 $:.each do |base|
 307   d = File.join base, "sup/share/modes/"
 308   Redwood::Mode.load_all_modes d if File.directory? d
 309 end