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

   1 module Redwood
   2
   3 class LabelManager
   4   include Singleton
   5
   6   ## labels that have special semantics. user will be unable to
   7   ## add/remove these via normal label mechanisms.
   8   RESERVED_LABELS = [ :starred, :spam, :draft, :unread, :killed, :sent, :deleted, :inbox, :attachment ]
   9
  10   ## labels that will typically be hidden from the user
  11   HIDDEN_RESERVED_LABELS = [ :starred, :unread, :attachment ]
  12
  13   def initialize fn
  14     @fn = fn
  15     labels =
  16       if File.exists? fn
  17         IO.readlines(fn).map { |x| x.chomp.intern }
  18       else
  19         []
  20       end
  21     @labels = {}
  22     @new_labels = {}
  23     @modified = false
  24     labels.each { |t| @labels[t] = true }
  25   end
  26
  27   def new_label? l; @new_labels.include?(l) end
  28
  29   ## all labels user-defined and system, ordered
  30   ## nicely and converted to pretty strings. use #label_for to recover
  31   ## the original label.
  32   def all_labels
  33     ## uniq's only necessary here because of certain upgrade issues
  34     (RESERVED_LABELS + @labels.keys).uniq
  35   end
  36
  37   ## all user-defined labels, ordered
  38   ## nicely and converted to pretty strings. use #label_for to recover
  39   ## the original label.
  40   def user_defined_labels
  41     @labels.keys
  42   end
  43
  44   ## reverse the label->string mapping, for convenience!
  45   def string_for l
  46     if RESERVED_LABELS.include? l
  47       l.to_s.capitalize
  48     else
  49       l.to_s
  50     end
  51   end
  52
  53   def label_for s
  54     l = s.intern
  55     l2 = s.downcase.intern
  56     if RESERVED_LABELS.include? l2
  57       l2
  58     else
  59       l
  60     end
  61   end
  62
  63   def << t
  64     raise ArgumentError, "expecting a symbol" unless t.is_a? Symbol
  65     unless @labels.member?(t) || RESERVED_LABELS.member?(t)
  66       @labels[t] = true
  67       @new_labels[t] = true
  68       @modified = true
  69     end
  70   end
  71
  72   def delete t
  73     if @labels.delete(t)
  74       @modified = true
  75     end
  76   end
  77
  78   def save
  79     return unless @modified
  80     File.open(@fn, "w") { |f| f.puts @labels.keys.sort_by { |l| l.to_s } }
  81     @new_labels = {}
  82   end
  83 end
  84
  85 end