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

   1 module Redwood
   2
   3 class DraftManager
   4   include Singleton
   5
   6   attr_accessor :source
   7   def initialize dir
   8     @dir = dir
   9     @source = nil
  10     self.class.i_am_the_instance self
  11   end
  12
  13   def self.source_name; "sup://drafts"; end
  14   def self.source_id; 9999; end
  15   def new_source; @source = Recoverable.new DraftLoader.new; end
  16
  17   def write_draft
  18     offset = @source.gen_offset
  19     fn = @source.fn_for_offset offset
  20     File.open(fn, "w") { |f| yield f }
  21
  22     my_message = nil
  23     @source.each do |thisoffset, theselabels|
  24       m = Message.build_from_source @source, thisoffset
  25       m.labels = theselabels
  26       Index.sync_message m
  27       UpdateManager.relay self, :added, m
  28       my_message = m if thisoffset == offset
  29     end
  30
  31     my_message
  32   end
  33
  34   def discard m
  35     raise ArgumentError, "not a draft: source id #{m.source.id.inspect}, should be #{DraftManager.source_id.inspect} for #{m.id.inspect}" unless m.source.id.to_i == DraftManager.source_id
  36     Index.delete m.id
  37     File.delete @source.fn_for_offset(m.source_info)
  38     UpdateManager.relay self, :single_message_deleted, m
  39   end
  40 end
  41
  42 class DraftLoader < Source
  43   attr_accessor :dir
  44   yaml_properties :cur_offset
  45
  46   def initialize cur_offset=0
  47     dir = Redwood::DRAFT_DIR
  48     Dir.mkdir dir unless File.exists? dir
  49     super DraftManager.source_name, cur_offset, true, false
  50     @dir = dir
  51   end
  52
  53   def id; DraftManager.source_id; end
  54   def to_s; DraftManager.source_name; end
  55   def uri; DraftManager.source_name; end
  56
  57   def each
  58     ids = get_ids
  59     ids.each do |id|
  60       if id >= cur_offset
  61         self.cur_offset = id + 1
  62         yield [id, [:draft, :inbox]]
  63       end
  64     end
  65   end
  66
  67   def gen_offset
  68     i = cur_offset
  69     while File.exists? fn_for_offset(i)
  70       i += 1
  71     end
  72     i
  73   end
  74
  75   def fn_for_offset o; File.join(@dir, o.to_s); end
  76
  77   def load_header offset
  78     File.open(fn_for_offset(offset)) { |f| parse_raw_email_header f }
  79   end
  80
  81   def load_message offset
  82     File.open fn_for_offset(offset) do |f|
  83       RMail::Mailbox::MBoxReader.new(f).each_message do |input|
  84         return RMail::Parser.read(input)
  85       end
  86     end
  87   end
  88
  89   def raw_header offset
  90     ret = ""
  91     File.open fn_for_offset(offset) do |f|
  92       until f.eof? || (l = f.gets) =~ /^$/
  93         ret += l
  94       end
  95     end
  96     ret
  97   end
  98
  99   def each_raw_message_line offset
 100     File.open(fn_for_offset(offset)) do |f|
 101       yield f.gets until f.eof?
 102     end
 103   end
 104
 105   def raw_message offset
 106     IO.read(fn_for_offset(offset))
 107   end
 108
 109   def start_offset; 0; end
 110   def end_offset
 111     ids = get_ids
 112     ids.empty? ? 0 : (ids.last + 1)
 113   end
 114
 115 private
 116
 117   def get_ids
 118     Dir.entries(@dir).select { |x| x =~ /^\d+$/ }.map { |x| x.to_i }.sort
 119   end
 120 end
 121
 122 end