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