about summary refs log tree commit diff stats
path: root/app/models/pokeviewer/trainer.rb
blob: 950dac06940a132611c3438b5bc80c9378f21d7e (plain) (blame)
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
85
86
87
88
89
90
91
92
93
94
95
module Pokeviewer
  class Trainer < ApplicationRecord
    extend Enumerize

    has_many :pokemon, dependent: :nullify

    has_many :pokedex_entries, dependent: :destroy

    validates :number, presence: true,
      numericality: { greater_than_or_equal_to: 0, only_integer: true }

    validates :name, presence: true, uniqueness: {
      scope: :number,
      message: "and number should be pairwise unique" }

    validates :game, presence: true
    enumerize :game, in: [:ruby, :sapphire, :firered, :leafgreen, :emerald],
      predicates: true

    belongs_to :marine_ribbon, class_name: "GiftRibbon", optional: true
    belongs_to :land_ribbon, class_name: "GiftRibbon", optional: true
    belongs_to :sky_ribbon, class_name: "GiftRibbon", optional: true
    belongs_to :country_ribbon, class_name: "GiftRibbon", optional: true
    belongs_to :national_ribbon, class_name: "GiftRibbon", optional: true
    belongs_to :earth_ribbon, class_name: "GiftRibbon", optional: true
    belongs_to :world_ribbon, class_name: "GiftRibbon", optional: true

    validates :box_1_name, presence: true
    validates :box_2_name, presence: true
    validates :box_3_name, presence: true
    validates :box_4_name, presence: true
    validates :box_5_name, presence: true
    validates :box_6_name, presence: true
    validates :box_7_name, presence: true
    validates :box_8_name, presence: true
    validates :box_9_name, presence: true
    validates :box_10_name, presence: true
    validates :box_11_name, presence: true
    validates :box_12_name, presence: true
    validates :box_13_name, presence: true
    validates :box_14_name, presence: true

    def party
      pokemon.party.includes(current: [:species])
    end

    def box(n)
      pokemon.box(n).includes(current: [:species])
    end

    def box_name(n)
      if n > 0 and n <= 14
        send "box_#{n}_name".intern
      else
        nil
      end
    end

    def box_contents(n)
      pokes = box(n).to_a

      result = []
      (0..29).each do |i|
        if pokes.empty? or (pokes.first.slot == i)
          result << pokes.shift
        else
          result << nil
        end
      end

      result
    end

    def boxes
      (1..14).map { |n| {
        name: box_name(n),
        pokemon: box_contents(n)
      }}
    end

    def display_number
      number.to_s.rjust(5, '0')
    end

    def gift_ribbon_description(ribbon)
      gift_ribbon = send ribbon

      if gift_ribbon.nil?
        ""
      else
        gift_ribbon.description
      end
    end
  end
end