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
   7 class Object
   8   ## this is for debugging purposes because i keep calling #id on the
   9   ## wrong object and i want it to throw an exception
  10   def id
  11     raise "wrong id called on #{self.inspect}"
  12   end
  13 end
  14
  15 class Module
  16   def yaml_properties *props
  17     props = props.map { |p| p.to_s }
  18     vars = props.map { |p| "@#{p}" }
  19     klass = self
  20     path = klass.name.gsub(/::/, "/")
  21
  22     klass.instance_eval do
  23       define_method(:to_yaml_properties) { vars }
  24       define_method(:to_yaml_type) { "!#{Redwood::YAML_DOMAIN},#{Redwood::YAML_DATE}/#{path}" }
  25     end
  26
  27     YAML.add_domain_type("#{Redwood::YAML_DOMAIN},#{Redwood::YAML_DATE}", path) do |type, val|
  28       klass.new(*props.map { |p| val[p] })
  29     end
  30   end
  31 end
  32
  33 module Redwood
  34   VERSION = "0.0.8"
  35
  36   BASE_DIR   = ENV["SUP_BASE"] || File.join(ENV["HOME"], ".sup")
  37   CONFIG_FN  = File.join(BASE_DIR, "config.yaml")
  38   SOURCE_FN  = File.join(BASE_DIR, "sources.yaml")
  39   LABEL_FN   = File.join(BASE_DIR, "labels.txt")
  40   PERSON_FN  = File.join(BASE_DIR, "people.txt")
  41   CONTACT_FN = File.join(BASE_DIR, "contacts.txt")
  42   DRAFT_DIR  = File.join(BASE_DIR, "drafts")
  43   SENT_FN    = File.join(BASE_DIR, "sent.mbox")
  44   LOCK_FN    = File.join(BASE_DIR, "lock")
  45   SUICIDE_FN = File.join(BASE_DIR, "please-kill-yourself")
  46
  47   YAML_DOMAIN = "masanjin.net"
  48   YAML_DATE = "2006-10-01"
  49
  50 ## record exceptions thrown in threads nicely
  51   $exception = nil
  52   def reporting_thread
  53     if $opts[:no_threads]
  54       yield
  55     else
  56       ::Thread.new do
  57         begin
  58           yield
  59         rescue Exception => e
  60           File.open("sup-exception-log.txt", "w") do |f|
  61             f.puts "--- #{e.class.name} at #{Time.now}"
  62             f.puts e.message, e.backtrace
  63           end
  64           $exception ||= e
  65           raise
  66         end
  67       end
  68     end
  69   end
  70   module_function :reporting_thread
  71
  72 ## one-stop shop for yamliciousness
  73   def save_yaml_obj object, fn, safe=false
  74     if safe
  75       safe_fn = "#{File.dirname fn}/safe_#{File.basename fn}"
  76       mode = File.stat(fn) if File.exists? fn
  77       File.open(safe_fn, "w", mode) { |f| f.puts object.to_yaml }
  78       FileUtils.mv safe_fn, fn
  79     else
  80       File.open(fn, "w") { |f| f.puts object.to_yaml }
  81     end
  82   end
  83
  84   def load_yaml_obj fn, compress=false
  85     if File.exists? fn
  86       if compress
  87         Zlib::GzipReader.open(fn) { |f| YAML::load f }
  88       else
  89         YAML::load_file fn
  90       end
  91     end
  92   end
  93
  94   def start
  95     Redwood::PersonManager.new Redwood::PERSON_FN
  96     Redwood::SentManager.new Redwood::SENT_FN
  97     Redwood::ContactManager.new Redwood::CONTACT_FN
  98     Redwood::LabelManager.new Redwood::LABEL_FN
  99     Redwood::AccountManager.new $config[:accounts]
 100     Redwood::DraftManager.new Redwood::DRAFT_DIR
 101     Redwood::UpdateManager.new
 102     Redwood::PollManager.new
 103     Redwood::SuicideManager.new Redwood::SUICIDE_FN
 104   end
 105
 106   def finish
 107     Redwood::PollManager.stop
 108     Redwood::LabelManager.save if Redwood::LabelManager.instantiated?
 109     Redwood::ContactManager.save if Redwood::ContactManager.instantiated?
 110     Redwood::PersonManager.save if Redwood::PersonManager.instantiated?
 111     Redwood::BufferManager.deinstantiate! if Redwood::BufferManager.instantiated?
 112   end
 113
 114   ## not really a good place for this, so I'll just dump it here.
 115   def report_broken_sources opts={}
 116     return unless BufferManager.instantiated?
 117
 118     broken_sources = Index.usual_sources.select { |s| s.error.is_a? FatalSourceError }
 119     unless broken_sources.empty?
 120       BufferManager.spawn "Broken source notification", TextMode.new(<<EOM), opts
 121 Source error notification
 122 -------------------------
 123
 124 Hi there. It looks like one or more message sources is reporting
 125 errors. Until this is corrected, messages from these sources cannot
 126 be viewed, and new messages will not be detected.
 127
 128 #{broken_sources.map { |s| "Source: " + s.to_s + "\n Error: " + s.error.message.wrap(70).join("\n        ")}.join("\n\n")}
 129 EOM
 130 #' stupid ruby-mode
 131     end
 132
 133     desynced_sources = Index.usual_sources.select { |s| s.error.is_a? OutOfSyncSourceError }
 134     unless desynced_sources.empty?
 135       BufferManager.spawn "Out-of-sync source notification", TextMode.new(<<EOM), opts
 136 Out-of-sync source notification
 137 -------------------------------
 138
 139 Hi there. It looks like one or more sources has fallen out of sync
 140 with my index. This can happen when you modify these sources with
 141 other email clients. (Sorry, I don't play well with others.)
 142
 143 Until this is corrected, messages from these sources cannot be viewed,
 144 and new messages will not be detected. Luckily, this is easy to correct!
 145
 146 #{desynced_sources.map do |s|
 147   "Source: " + s.to_s +
 148    "\n Error: " + s.error.message.wrap(70).join("\n        ") +
 149    "\n   Fix: sup-sync --changed #{s.to_s}"
 150   end}
 151 EOM
 152 #' stupid ruby-mode
 153     end
 154   end
 155
 156   module_function :save_yaml_obj, :load_yaml_obj, :start, :finish,
 157                   :report_broken_sources
 158 end
 159
 160 ## set up default configuration file
 161 if File.exists? Redwood::CONFIG_FN
 162   $config = Redwood::load_yaml_obj Redwood::CONFIG_FN
 163 else
 164   require 'etc'
 165   require 'socket'
 166   name = Etc.getpwnam(ENV["USER"]).gecos.split(/,/).first
 167   email = ENV["USER"] + "@" +
 168     begin
 169       Socket.gethostbyname(Socket.gethostname).first
 170     rescue SocketError
 171       Socket.gethostname
 172     end
 173
 174   $config = {
 175     :accounts => {
 176       :default => {
 177         :name => name,
 178         :email => email,
 179         :alternates => [],
 180         :sendmail => "/usr/sbin/sendmail -oem -ti",
 181         :signature => File.join(ENV["HOME"], ".signature")
 182       }
 183     },
 184     :editor => ENV["EDITOR"] || "/usr/bin/vim -f -c 'setlocal spell spelllang=en_us' -c 'set filetype=mail'",
 185     :thread_by_subject => false,
 186   }
 187   begin
 188     FileUtils.mkdir_p Redwood::BASE_DIR
 189     Redwood::save_yaml_obj $config, Redwood::CONFIG_FN
 190   rescue StandardError => e
 191     $stderr.puts "warning: #{e.message}"
 192   end
 193 end
 194
 195 require "sup/util"
 196 require "sup/update"
 197 require "sup/suicide"
 198 require "sup/message"
 199 require "sup/source"
 200 require "sup/mbox"
 201 require "sup/maildir"
 202 require "sup/imap"
 203 require "sup/person"
 204 require "sup/account"
 205 require "sup/thread"
 206 require "sup/index"
 207 require "sup/textfield"
 208 require "sup/buffer"
 209 require "sup/keymap"
 210 require "sup/mode"
 211 require "sup/colormap"
 212 require "sup/label"
 213 require "sup/contact"
 214 require "sup/tagger"
 215 require "sup/draft"
 216 require "sup/poll"
 217 require "sup/modes/scroll-mode"
 218 require "sup/modes/text-mode"
 219 require "sup/modes/line-cursor-mode"
 220 require "sup/modes/help-mode"
 221 require "sup/modes/edit-message-mode"
 222 require "sup/modes/compose-mode"
 223 require "sup/modes/resume-mode"
 224 require "sup/modes/forward-mode"
 225 require "sup/modes/reply-mode"
 226 require "sup/modes/label-list-mode"
 227 require "sup/modes/contact-list-mode"
 228 require "sup/modes/thread-view-mode"
 229 require "sup/modes/thread-index-mode"
 230 require "sup/modes/label-search-results-mode"
 231 require "sup/modes/search-results-mode"
 232 require "sup/modes/person-search-results-mode"
 233 require "sup/modes/inbox-mode"
 234 require "sup/modes/buffer-list-mode"
 235 require "sup/modes/log-mode"
 236 require "sup/modes/poll-mode"
 237 require "sup/logger"
 238 require "sup/sent"
 239
 240 module Redwood
 241   def log s; Logger.log s; end
 242   module_function :log
 243 end
 244
 245 $:.each do |base|
 246   d = File.join base, "sup/share/modes/"
 247   Redwood::Mode.load_all_modes d if File.directory? d
 248 end