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