summaryrefslogtreecommitdiff
path: root/fake-service/lib/pixelated_service/mail_scope_filter.rb
blob: dae6d99e368e7ff192be46dd4e504b65933dd08c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
module PixelatedService
  module MailScopeFilter
    include Enumerable

    def initialize(c)
      @c = c
    end

    def each
      @c.each do |m|
        yield m if retain?(m)
      end
    end

    class Default
      include MailScopeFilter

      def initialize(c)
        super
        @tags = [Tags.get('sent'), Tags.get('trash'), Tags.get('drafts')]
      end

      def retain?(m)
        !(@tags.any? { |t| m.is_tagged?(t) })
      end

      class << self
        def +(o)
          o
        end
      end
    end

    class All
      include MailScopeFilter

      def initialize(c)
        super
        @t = Tags.get('trash')
      end

      def retain?(m)
        !m.is_tagged?(@t)
      end

      class << self
        def +(o)
          All
        end
      end
    end

    def self.tagged_with(n)
      t = Tags.get(n)
      c = Class.new
      c.send :include, MailScopeFilter
      c.send :define_method, :retain? do |m|
        m.is_tagged?(t)
      end
      c.class.send :define_method, :+ do |o|
        All === o ? All : self
      end
      c
    end

    Trash  = tagged_with('trash')
    Sent   = tagged_with('sent')
    Drafts = tagged_with('drafts')
  end
end