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, compress=false
  74     if compress
  75       Zlib::GzipWriter.open(fn) { |f| f.puts object.to_yaml }
  76     else
  77       File.open(fn, "w") { |f| f.puts object.to_yaml }
  78     end
  79   end
  80
  81   def load_yaml_obj fn, compress=false
  82     if File.exists? fn
  83       if compress
  84         Zlib::GzipReader.open(fn) { |f| YAML::load f }
  85       else
  86         YAML::load_file fn
  87       end
  88     end
  89   end
  90
  91   def start
  92     Redwood::PersonManager.new Redwood::PERSON_FN
  93     Redwood::SentManager.new Redwood::SENT_FN
  94     Redwood::ContactManager.new Redwood::CONTACT_FN
  95     Redwood::LabelManager.new Redwood::LABEL_FN
  96     Redwood::AccountManager.new $config[:accounts]
  97     Redwood::DraftManager.new Redwood::DRAFT_DIR
  98     Redwood::UpdateManager.new
  99     Redwood::PollManager.new
 100     Redwood::SuicideManager.new Redwood::SUICIDE_FN
 101   end
 102
 103   def finish
 104     Redwood::LabelManager.save if Redwood::LabelManager.instantiated?
 105     Redwood::ContactManager.save if Redwood::ContactManager.instantiated?
 106     Redwood::PersonManager.save if Redwood::PersonManager.instantiated?
 107     Redwood::BufferManager.deinstantiate! if Redwood::BufferManager.instantiated?
 108   end
 109
 110   ## not really a good place for this, so I'll just dump it here.
 111   def report_broken_sources opts={}
 112     return unless BufferManager.instantiated?
 113
 114     broken_sources = Index.usual_sources.select { |s| s.error.is_a? FatalSourceError }
 115     unless broken_sources.empty?
 116       BufferManager.spawn "Broken source notification", TextMode.new(<<EOM), opts
 117 Source error notification
 118 -------------------------
 119
 120 Hi there. It looks like one or more message sources is reporting
 121 errors. Until this is corrected, messages from these sources cannot
 122 be viewed, and new messages will not be detected.
 123
 124 #{broken_sources.map { |s| "Source: " + s.to_s + "\n Error: " + s.error.message.wrap(70).join("\n        ")}.join("\n\n")}
 125 EOM
 126 #' stupid ruby-mode
 127     end
 128
 129     desynced_sources = Index.usual_sources.select { |s| s.error.is_a? OutOfSyncSourceError }
 130     unless desynced_sources.empty?
 131       BufferManager.spawn "Out-of-sync source notification", TextMode.new(<<EOM), opts
 132 Out-of-sync source notification
 133 -------------------------------
 134
 135 Hi there. It looks like one or more sources has fallen out of sync
 136 with my index. This can happen when you modify these sources with
 137 other email clients. (Sorry, I don't play well with others.)
 138
 139 Until this is corrected, messages from these sources cannot be viewed,
 140 and new messages will not be detected. Luckily, this is easy to correct!
 141
 142 #{desynced_sources.map do |s|
 143   "Source: " + s.to_s +
 144    "\n Error: " + s.error.message.wrap(70).join("\n        ") +
 145    "\n   Fix: sup-sync --changed #{s.to_s}"
 146   end}
 147 EOM
 148 #' stupid ruby-mode
 149     end
 150   end
 151
 152   module_function :save_yaml_obj, :load_yaml_obj, :start, :finish,
 153                   :report_broken_sources
 154 end
 155
 156 ## set up default configuration file
 157 if File.exists? Redwood::CONFIG_FN
 158   $config = Redwood::load_yaml_obj Redwood::CONFIG_FN
 159 else
 160   require 'etc'
 161   require 'socket'
 162   name = Etc.getpwnam(ENV["USER"]).gecos.split(/,/).first
 163   email = ENV["USER"] + "@" +
 164     begin
 165       Socket.gethostbyname(Socket.gethostname).first
 166     rescue SocketError
 167       Socket.gethostname
 168     end
 169
 170   $config = {
 171     :accounts => {
 172       :default => {
 173         :name => name,
 174         :email => email,
 175         :alternates => [],
 176         :sendmail => "/usr/sbin/sendmail -oem -ti",
 177         :signature => File.join(ENV["HOME"], ".signature")
 178       }
 179     },
 180     :editor => ENV["EDITOR"] || "/usr/bin/vim -f -c 'setlocal spell spelllang=en_us' -c 'set filetype=mail'",
 181     :thread_by_subject => false,
 182   }
 183   begin
 184     FileUtils.mkdir_p Redwood::BASE_DIR
 185     Redwood::save_yaml_obj $config, Redwood::CONFIG_FN
 186   rescue StandardError => e
 187     $stderr.puts "warning: #{e.message}"
 188   end
 189 end
 190
 191 require "sup/util"
 192 require "sup/update"
 193 require "sup/suicide"
 194 require "sup/message"
 195 require "sup/source"
 196 require "sup/mbox"
 197 require "sup/maildir"
 198 require "sup/imap"
 199 require "sup/person"
 200 require "sup/account"
 201 require "sup/thread"
 202 require "sup/index"
 203 require "sup/textfield"
 204 require "sup/buffer"
 205 require "sup/keymap"
 206 require "sup/mode"
 207 require "sup/colormap"
 208 require "sup/label"
 209 require "sup/contact"
 210 require "sup/tagger"
 211 require "sup/draft"
 212 require "sup/poll"
 213 require "sup/modes/scroll-mode"
 214 require "sup/modes/text-mode"
 215 require "sup/modes/line-cursor-mode"
 216 require "sup/modes/help-mode"
 217 require "sup/modes/edit-message-mode"
 218 require "sup/modes/compose-mode"
 219 require "sup/modes/resume-mode"
 220 require "sup/modes/forward-mode"
 221 require "sup/modes/reply-mode"
 222 require "sup/modes/label-list-mode"
 223 require "sup/modes/contact-list-mode"
 224 require "sup/modes/thread-view-mode"
 225 require "sup/modes/thread-index-mode"
 226 require "sup/modes/label-search-results-mode"
 227 require "sup/modes/search-results-mode"
 228 require "sup/modes/person-search-results-mode"
 229 require "sup/modes/inbox-mode"
 230 require "sup/modes/buffer-list-mode"
 231 require "sup/modes/log-mode"
 232 require "sup/modes/poll-mode"
 233 require "sup/logger"
 234 require "sup/sent"
 235
 236 module Redwood
 237   def log s; Logger.log s; end
 238   module_function :log
 239 end
 240
 241 $:.each do |base|
 242   d = File.join base, "sup/share/modes/"
 243   Redwood::Mode.load_all_modes d if File.directory? d
 244 end