hlockey 2 → 4

data/lib/hlockey/game.rb

@@ -1,159 +1,321 @@
-# frozen_string_literal: true
-
-require('hlockey/messages')
-
-module Hlockey
-  class Game
-    attr_reader(:home, :away, :stream, :score, :in_progress)
-
-    def initialize(home, away, prng)
-      @home = home
-      @away = away
-      @prng = prng
-      @stream = [Messages.StartOfGame]
-      @score = { home: 0, away: 0 }
-      @in_progress = true
-      @actions = 0
-      @period = 1
-      @face_off = true
-      @team_with_puck = nil
-      @puckless_team = nil
-      @puck_holder = nil
-      @shooting_chance = 0
-    end
-
-    def to_s
-      "#{@home} vs #{@away}"
-    end
-
-    def update
-      return unless @in_progress
-
-      @stream << Messages.StartOfPeriod(@period) if @actions.zero?
-      @actions += 1
-
-      if @actions == 60
-        @stream << Messages.EndOfPeriod(@period, @home, @away, *@score.values)
-        @actions = 0
-        @period += 1
-        start_faceoff
-
-        if @period > 3 && !(@score[:home] == @score[:away] && @period < 12)
-          # Game is over
-          @in_progress = false
-
-          winner, loser = @score[:away] > @score[:home] ?
-            [@away, @home] : [@home, @away]
-
-          @stream << Messages.EndOfGame(winner)
-          winner.wins += 1
-          loser.losses += 1
-        end
-        return
-      end
-
-      if @face_off
-        if @puck_holder.nil?
-          # Pass opposite team to who wins the puck to switch_team_with_puck,
-          # so @team_with_puck & @puckless_team are set to the correct values.
-          switch_team_with_puck(
-            action_succeeds?(@home.roster[:center].stats[:offense],
-                             @away.roster[:center].stats[:offense]) ? @away : @home
-          )
-
-          @puck_holder = @team_with_puck.roster[:center]
-          @stream << Messages.FaceOff(@puck_holder, @team_with_puck)
-        else
-          pass
-          @face_off = false
-        end
-        return
-      end
-
-      case @prng.rand 5 + @shooting_chance
-      when 0..4
-        pass
-      when 5..6 # Check
-        check
-      else      # Shoot
-        unless @shooting_chance < 5 &&
-            try_block_shot(@puckless_team.roster[@prng.rand(2).zero? ?
-                                                 :ldef : :rdef]) ||
-            try_block_shot(@puckless_team.roster[:goalie])
-          @score[@team_with_puck == @home ? :home : :away] += 1
-
-          @stream << Messages.ShootScore(@puck_holder, @home, @away, *@score.values)
-
-          start_faceoff
-          @actions = 59 if @period > 3 # Sudden death overtime
-        end
-      end
-    end
-
-    private
-
-    def action_succeeds?(helping_stat, hindering_stat)
-      @prng.rand(10) + helping_stat - hindering_stat > 4
-    end
-
-    def start_faceoff
-      @shooting_chance = 0
-      @face_off = true
-      @puck_holder = nil
-    end
-
-    def switch_team_with_puck(team_with_puck = @team_with_puck)
-      @team_with_puck, @puckless_team = team_with_puck == @home ?
-                                        [@away, @home] : [@home, @away]
-      @shooting_chance = 0
-    end
-
-    def pass
-      sender = @puck_holder
-      receiver = puckless_non_goalie(@team_with_puck)
-      interceptor = puckless_non_goalie
-
-      if !@face_off && try_take_puck(interceptor, 4) # Pass intercepted
-        @stream << Messages.Pass(sender, receiver, interceptor, @team_with_puck)
-        return
-      end
-
-      @stream << Messages.Pass(sender, receiver)
-      @puck_holder = receiver
-      @shooting_chance += 1
-    end
-
-    def check
-      defender = puckless_non_goalie
-      @stream << Messages.Hit(@puck_holder, defender,
-                              try_take_puck(defender, 0, :defense), @team_with_puck)
-    end
-
-    def try_take_puck(player, dis = 0, stat = :agility)
-      return false unless action_succeeds?(player.stats[stat] - dis,
-                                           @puck_holder.stats[stat])
-
-      switch_team_with_puck
-      @puck_holder = player
-
-      true
-    end
-
-    def try_block_shot(blocker)
-      return false unless action_succeeds?(blocker.stats[:defense],
-                                           @puck_holder.stats[:offense])
-
-      @shooting_chance += 1
-      @stream << Messages.ShootBlock(@puck_holder, blocker,
-                                     try_take_puck(blocker), @team_with_puck)
-
-      true
-    end
-
-    def puckless_non_goalie(team = @puckless_team)
-      team.roster.values.select do |p|
-        p != team.roster[:goalie] and p != @puck_holder
-      end.sample(random: @prng)
-    end
-  end
-end
+require("hlockey/game/actions")
+require("hlockey/game/fight")
+require("hlockey/constants")
+require("hlockey/message")
+require("hlockey/utils")
+
+module Hlockey
+  ##
+  # A single game of Hlockey
+  class Game
+    include(Actions)
+
+    ACTIONS_PER_PERIOD = 60
+    NON_OT_PERIODS = 4
+    NON_OT_ACTIONS = ACTIONS_PER_PERIOD * NON_OT_PERIODS
+    TOTAL_ACTIONS = UPDATES_PER_HOUR - ACTIONS_PER_PERIOD
+    TOTAL_PERIODS = TOTAL_ACTIONS / ACTIONS_PER_PERIOD
+
+    # @return [Weather]
+    attr_reader(:weather)
+
+    # @return [Array<Message>]
+    attr_reader(:stream)
+
+    # @return [Integer]
+    attr_reader(:actions, :shooting_chance)
+
+    # @return [Fight, nil]
+    attr_reader(:fight)
+
+    # @param home [Team]
+    # @param away [Team]
+    # @param prng [Random] should be League#prng
+    # @param weather [Class<Weather>]
+    def initialize(home, away, prng, weather)
+      @home = home
+      @away = away
+      @prng = prng
+      @weather = weather.new(self)
+      @stream = [Message.StartOfGame]
+      @score = { home: 0, away: 0 }
+      @in_progress = true
+      @actions = 0
+      @period = 1
+      @faceoff = true
+      @faceoff_won = false
+      @team_with_puck = nil
+      @puckless_team = nil
+      @puck_holder_pos = nil
+      @shooting_chance = 0
+      @fight = nil
+      @fight_chance = 0
+      @pre_morale_change_stats = {}
+      @partying_teams = [@home, @away].select { |t| t.status == :partying }
+
+      @weather.on_game_start
+    end
+
+    # @return [String]
+    def to_s
+      "#{Message.color(@home)} vs #{Message.color(@away)}"
+    end
+
+    # Does an action in the game
+    def update
+      return unless @in_progress
+
+      @stream << Message.StartOfPeriod(@period) if @actions.zero?
+      @actions += 1
+
+      @weather.on_action
+
+      unless @fight.nil?
+        @stream << @fight.next_action
+        if @stream.last.event == :FightEnded
+          morale_change_amount = (@fight.score[:home] - @fight.score[:away]) / 10.0
+          change_morale(@home, morale_change_amount)
+          change_morale(@away, -morale_change_amount)
+
+          @fight = nil
+        end
+        return
+      end
+
+      if @actions >= ACTIONS_PER_PERIOD
+        end_period
+        return
+      end
+      if @faceoff
+        do_faceoff
+        return
+      end
+
+      case @prng.rand(5 + @shooting_chance)
+      when 0..4
+        pass
+      when 5, 6
+        check
+      else
+        shoot
+      end
+
+      return unless @partying_teams.any? && @prng.rand(NON_OT_ACTIONS / 5).zero?
+
+      @partying_teams.each do |team|
+        player = team.roster.values.sample(random: @prng)
+        stat = player.stats.keys.sample(random: @prng)
+        player.stats[stat] += 0.1
+        @stream << Message.Partying(player)
+        next if @pre_morale_change_stats[player].nil?
+
+        @pre_morale_change_stats[player][stat] += 0.1
+      end
+    end
+
+    # Makes a player shoot in the game, regardless of if it would normally occur
+    # @param audacity [Boolean] if the function was called by Audacity weather
+    def shoot(audacity: false)
+      return if @puck_holder_pos.nil?
+
+      if @shooting_chance < 5 &&
+         try_block_shot(@prng.rand(2).zero? ? :ldef : :rdef, audacity: audacity)
+        return
+      end
+      return if try_block_shot(:goalie, audacity: audacity)
+
+      # Goal scored
+
+      scoring_team = @team_with_puck == @home ? :home : :away
+      @score[scoring_team] += 1
+      weather.on_goal(scoring_team)
+
+      @stream << Message.ShootScore(puck_holder, @home, @away, *@score.values, audacity)
+
+      start_fight if @prng.rand(3 + @fight_chance) > 3
+
+      start_faceoff
+      @actions = ACTIONS_PER_PERIOD - 1 if @period >= NON_OT_PERIODS # Sudden death OT
+    end
+
+    private
+
+    # Makes a player pass in the game, regardless of if it would normally occur
+    def pass
+      sender = puck_holder
+      receiver = pass_reciever
+      interceptor = defensive_pos
+
+      if !@faceoff && try_take_puck(interceptor, 7 - @shooting_chance, :agility)
+        @stream << Message.Pass(sender,
+                                @puckless_team.roster[receiver],
+                                @shooting_chance,
+                                @team_with_puck.roster[interceptor],
+                                @team_with_puck)
+        return
+      end
+
+      @puck_holder_pos = receiver
+      @shooting_chance += 1
+      @stream << Message.Pass(sender, @team_with_puck.roster[receiver], @shooting_chance)
+    end
+
+    # Makes a player check in the game, regardless of if it would normally occur
+    def check
+      defender_pos = defensive_pos
+      defender = @puckless_team.roster[defender_pos]
+      @stream << Message.Hit(puck_holder, defender,
+                             try_take_puck(defender_pos), @team_with_puck,
+                             @shooting_chance)
+      @fight_chance += 0.1
+    end
+
+    def end_period
+      @weather.on_period_end
+      @stream << Message.EndOfPeriod(@period, @home, @away, *@score.values)
+      @actions = 0
+      @period += 1
+      start_faceoff
+
+      if @period >= NON_OT_PERIODS &&
+         !(@score[:home] == @score[:away] && @period <= TOTAL_PERIODS)
+        # Game is over
+        @pre_morale_change_stats.each { |player, stats| player.stats = stats }
+        @weather.on_game_end
+        @in_progress = false
+        winner, loser = @score[:away] > @score[:home] ? [@away, @home] : [@home, @away]
+        @stream << Message.EndOfGame(winner)
+        winner.wins += 1
+        loser.losses += 1
+      end
+    end
+
+    def start_faceoff
+      @shooting_chance = 0
+      @faceoff = true
+      @faceoff_won = false
+    end
+
+    def do_faceoff
+      if @faceoff_won
+        pass
+        @faceoff = false
+        return
+      end
+
+      # Pass opposite team to who wins the puck to switch_team_with_puck,
+      # so @team_with_puck & @puckless_team are set to the correct values.
+      switch_team_with_puck(
+        if action_succeeds?(@home.roster[:center].stats[:offense],
+                            @away.roster[:center].stats[:offense])
+          @away
+        else
+          @home
+        end
+      )
+      @shooting_chance = 2
+      @puck_holder_pos = :center
+      @faceoff_won = true
+      @stream << Message.FaceOff(puck_holder, @team_with_puck)
+    end
+
+    def switch_team_with_puck(team_with_puck = @team_with_puck)
+      @team_with_puck, @puckless_team = if team_with_puck == @home
+                                          [@away, @home]
+                                        else
+                                          [@home, @away]
+                                        end
+      @shooting_chance = 3 - @shooting_chance
+      @shooting_chance = 0 if @shooting_chance.negative?
+    end
+
+    # @param pos [Symbol] position on puckless team that tries to take the puck
+    # @param dis [Integer] disadvantage against taking puck
+    # @param stat [Symbol] stat compared to change the odds of success/failure
+    # @return [Boolean] if taking the puck succeeded
+    def try_take_puck(pos, dis = 0, stat = :defense)
+      return false unless action_succeeds?(@puckless_team.roster[pos].stats[stat] - dis,
+                                           puck_holder.stats[stat])
+
+      switch_team_with_puck
+      @puck_holder_pos = pos
+
+      true
+    end
+
+    # @return [Boolean] if blocking the shot succeeded
+    def try_block_shot(pos, audacity: false)
+      blocker = @puckless_team.roster[pos]
+
+      return false unless action_succeeds?(blocker.stats[:defense],
+                                           puck_holder.stats[:offense])
+
+      @shooting_chance += 1
+      @stream << Message.ShootBlock(puck_holder, blocker,
+                                    try_take_puck(pos), @team_with_puck,
+                                    @shooting_chance, audacity)
+
+      true
+    end
+
+    def start_fight
+      @fight = Fight.new(self)
+      @stream << Message.FightStarted(@fight.players[:home].first,
+                                      @fight.players[:away].first)
+    end
+
+    # @param team [Team]
+    # @param amount [Numeric]
+    def change_morale(team, amount)
+      return if amount.zero?
+
+      team.roster.each_value do |player|
+        if @pre_morale_change_stats[player].nil?
+          @pre_morale_change_stats[player] = player.stats.clone
+        end
+
+        player.stats.transform_values! { |stat| stat + amount }
+      end
+
+      @stream << Message.MoraleChange(team, amount)
+    end
+
+    # @return [Player]
+    def puck_holder
+      @team_with_puck.roster[@puck_holder_pos]
+    end
+
+    # @return [Symbol]
+    def pass_reciever
+      w = weights(@shooting_chance > 3)
+      w.delete(@puck_holder_pos)
+      Utils.weighted_random(w, @prng)
+    end
+
+    # @return [Symbol]
+    def defensive_pos
+      Utils.weighted_random(weights(@shooting_chance < 3), @prng)
+    end
+
+    # @return [Hash<Symbol => Integer>]
+    def weights(offensive)
+      if offensive
+        {
+          lwing: 2,
+          center: 2,
+          rwing: 2,
+          ldef: 1,
+          rdef: 1
+        }
+      else
+        {
+          lwing: 1,
+          center: 1,
+          rwing: 1,
+          ldef: 3,
+          rdef: 3
+        }
+      end
+    end
+  end
+end