summaryrefslogtreecommitdiff
path: root/fake-service/lib/pixelated_service/mailset.rb
blob: 595571e8d66f35a011558c764523f0ee1d499189 (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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
require 'set'

module PixelatedService
  class Mailset
    DIR = File.expand_path File.join(File.dirname(__FILE__), "..", "..", "data", "mail-sets")

    class << self
      def create(name, number, tagging)
        ms = new name, number, tagging, nil
        ms.generate!
        ms
      end

      def load(name, observers)
        ms = new(name, -1, nil, observers)
        if ms.load!
          ms
        else
          nil
        end
      end
    end

    attr_reader :mails
    attr_reader :tags

    def initialize(name, number, tagging, observers)
      @name, @number, @tagging, @observers = name, number, tagging, observers
    end

    def generate!
      @persona = Generator.random_persona
      @tags = Generator.tags(Generator.ladder_distribution(4, 40))

      @mails = {}
      @tags = Set.new
      (0...(@number)).each do |i|
        res = if @tagging
                Generator.random_tagged_mail(@tags)
              else
                Generator.random_mail
              end
        @observers.mail_added res
        @tags.merge res.tags
        res.ident = i
        @mails[res.ident] = res
      end
    end

    def save!
      dir = File.join(DIR, @name)
      Dir.mkdir(dir) unless Dir.exists?(dir)
      File.open(File.join(dir, "persona.yml"), "w") do |f|
        f.write @persona.to_yaml
      end

      @mails.each do |(k, m)|
        nm = "mbox%08x" % m.ident
        File.open(File.join(dir, nm), "w") do |f|
          f.write m.to_s
        end
      end
    end

    def load!
      dir = File.join(DIR, @name)
      return false unless Dir.exists?(dir)
      @persona = YAML.load_file(File.join(dir, "persona.yml"))
      @mails = {}
      @ix = 0
      Dir["#{dir}/mbox*"].each do |f|
        File.open(f) do |fio|
          res = PixelatedService::Mail.read fio, @ix
          res.read = true if (res.tag_names.include?('sent') || res.tag_names.include?('drafts'))
          @mails[res.ident] = res
          @observers.mail_added res
          res.security_casing = SecurityCasingExamples::Case.case_from(res.ident.to_i)
          @ix += 1
        end
      end
      true
    end
  end
end