state_machines-mermaid 0.1.0

data/test/support/models/mage.rb

@@ -0,0 +1,261 @@
+# frozen_string_literal: true
+
+require_relative 'character'
+
+# Mage with spell casting and specialization states
+class Mage < Character
+  attr_accessor :spell_power, :focus, :school_of_magic, :mana_pool, :spell_book, :affinities, :masteries, :traits,
+                :trials_completed, :current_target, :channel_start_time, :cooldown_end_time
+
+  def initialize
+    super
+    @spell_power = 50
+    @focus = 100
+    @mana_pool = 200
+    @spell_book = []
+    @known_spells = {}
+  end
+
+  # Mental state for concentration
+  state_machine :concentration, initial: :focused do
+    state :focused do
+      def spell_critical_chance
+        0.2
+      end
+    end
+
+    state :distracted do
+      def spell_failure_chance
+        0.3
+      end
+    end
+
+    state :deep_meditation do
+      def mana_regeneration_rate
+        5.0
+      end
+
+      def immune_to_interruption?
+        true
+      end
+    end
+
+    state :interrupted do
+      def can_cast?
+        false
+      end
+    end
+
+    event :meditate do
+      transition focused: :deep_meditation, if: -> { mana < 50 && safe_to_meditate? }
+      transition distracted: :focused, if: :regain_composure?
+    end
+
+    event :disturb do
+      transition %i[focused deep_meditation] => :distracted, unless: :iron_will?
+    end
+
+    event :break_concentration do
+      transition any - :interrupted => :interrupted
+    end
+
+    event :refocus do
+      transition %i[distracted interrupted] => :focused, if: :concentration_check?
+    end
+
+    after_transition to: :deep_meditation, &:begin_mana_regeneration
+  end
+
+  # Spell school specialization with complex hierarchy
+  state_machine :spell_school, initial: :apprentice do
+    # Apprentice level
+    state :apprentice
+
+    # Fire magic tree
+    state :fire do
+      def elemental_affinity
+        :fire
+      end
+    end
+    state :ember
+    state :flame
+    state :inferno
+
+    # Ice magic tree
+    state :ice do
+      def elemental_affinity
+        :ice
+      end
+    end
+    state :frost
+    state :freeze
+    state :blizzard
+
+    # Arcane magic tree
+    state :arcane do
+      def elemental_affinity
+        :arcane
+      end
+    end
+    state :missile
+    state :explosion
+    state :singularity
+
+    # Master of multiple schools
+    state :archmage do
+      def can_dual_cast?
+        true
+      end
+    end
+
+    # Fire progression
+    event :study_fire do
+      transition apprentice: :ember, if: :has_fire_affinity?
+      transition ember: :flame, if: -> { spell_power > 75 && knows_spell?(:fireball) }
+      transition flame: :inferno, if: -> { spell_power > 150 && fire_mastery_complete? }
+    end
+
+    # Ice progression
+    event :study_ice do
+      transition apprentice: :frost, if: :has_ice_affinity?
+      transition frost: :freeze, if: -> { spell_power > 75 && knows_spell?(:ice_lance) }
+      transition freeze: :blizzard, if: -> { spell_power > 150 && ice_mastery_complete? }
+    end
+
+    # Arcane progression
+    event :study_arcane do
+      transition apprentice: :missile, if: :has_arcane_affinity?
+      transition missile: :explosion, if: -> { spell_power > 75 && knows_spell?(:arcane_blast) }
+      transition explosion: :singularity, if: -> { spell_power > 150 && arcane_mastery_complete? }
+    end
+
+    # Become archmage
+    event :transcend do
+      transition %i[inferno blizzard singularity] => :archmage, if: :worthy_of_archmage?
+    end
+
+    before_transition any => any do |mage, transition|
+      mage.update_spell_list(transition.to)
+    end
+  end
+
+  # Spell casting state
+  state_machine :casting_state, initial: :ready do
+    state :ready
+    state :preparing
+    state :channeling
+    state :releasing
+    state :cooldown
+
+    event :begin_cast do
+      transition ready: :preparing, if: :has_target?
+    end
+
+    event :channel do
+      transition preparing: :channeling, if: :channel_started?
+    end
+
+    event :release do
+      transition channeling: :releasing
+    end
+
+    event :complete do
+      transition releasing: :cooldown
+    end
+
+    event :reset do
+      transition cooldown: :ready, if: :cooldown_expired?
+    end
+
+    after_transition to: :releasing, &:cast_spell!
+  end
+
+  # Mage-specific methods
+  def safe_to_meditate?
+    !in_combat? && !enemies_nearby?
+  end
+
+  def regain_composure?
+    focus > 50
+  end
+
+  def iron_will?
+    has_trait?(:iron_will) || focus > 150
+  end
+
+  def concentration_check?
+    rand(100) < focus
+  end
+
+  def begin_mana_regeneration
+    @regenerating_mana = true
+  end
+
+  def has_fire_affinity?
+    @affinities&.include?(:fire)
+  end
+
+  def has_ice_affinity?
+    @affinities&.include?(:ice)
+  end
+
+  def has_arcane_affinity?
+    @affinities&.include?(:arcane)
+  end
+
+  def knows_spell?(spell_name)
+    @known_spells[spell_name] == true
+  end
+
+  def fire_mastery_complete?
+    @masteries && @masteries[:fire] >= 100
+  end
+
+  def ice_mastery_complete?
+    @masteries && @masteries[:ice] >= 100
+  end
+
+  def arcane_mastery_complete?
+    @masteries && @masteries[:arcane] >= 100
+  end
+
+  def worthy_of_archmage?
+    level >= 50 && completed_all_trials?
+  end
+
+  def update_spell_list(school)
+    # Update available spells based on school
+  end
+
+  def has_target?
+    @current_target != nil
+  end
+
+  def channel_started?
+    @channel_start_time != nil
+  end
+
+  def cooldown_expired?
+    @cooldown_end_time && Time.now > @cooldown_end_time
+  end
+
+  def cast_spell!
+    # Execute the spell
+  end
+
+  def in_combat?
+    status == 'combat'
+  end
+
+  def enemies_nearby?
+    false
+  end
+
+  def has_trait?(trait)
+    @traits&.include?(trait)
+  end
+
+  def completed_all_trials?
+    @trials_completed && @trials_completed.size >= 5
+  end
+end

data/test/support/models/regiment.rb

@@ -0,0 +1,244 @@
+# frozen_string_literal: true
+
+# Regiment with formation and morale states
+class Regiment
+  attr_accessor :morale, :soldiers, :commander, :supplies, :casualties, :battle_experience, :threats, :enemy_positions,
+                :immediate_threats, :enemy_activity_level, :defending_critical_position, :battle_start_time, :pike_wall_ready
+
+  def initialize
+    @morale = 100
+    @soldiers = 100
+    @supplies = 100
+    @casualties = 0
+    @battle_experience = 0
+  end
+
+  state_machine :formation, initial: :column do
+    state :column do
+      def movement_speed
+        2.0
+      end
+
+      def defense_rating
+        1.0
+      end
+    end
+
+    state :line do
+      def firepower
+        3.0
+      end
+
+      def defense_rating
+        1.5
+      end
+    end
+
+    state :square do
+      def cavalry_defense
+        5.0
+      end
+
+      def movement_speed
+        0.5
+      end
+    end
+
+    state :wedge do
+      def charge_bonus
+        4.0
+      end
+
+      def breakthrough_chance
+        0.7
+      end
+    end
+
+    state :scattered do
+      def cohesion
+        0.1
+      end
+
+      def rally_difficulty
+        10
+      end
+    end
+
+    state :circle do
+      def all_around_defense
+        3.0
+      end
+
+      def surrounded_bonus
+        2.0
+      end
+    end
+
+    event :deploy do
+      transition column: :line, if: :sufficient_space?
+    end
+
+    event :form_square do
+      transition %i[column line] => :square, if: :cavalry_threat?
+      transition wedge: :square, if: :emergency_defense?
+    end
+
+    event :form_wedge do
+      transition %i[column line] => :wedge, if: -> { morale > 70 && commander_present? }
+    end
+
+    event :form_circle do
+      transition %i[line square] => :circle, if: :surrounded?
+    end
+
+    event :break_formation do
+      transition any - :scattered => :scattered, if: -> { morale < 20 || casualties > 50 }
+    end
+
+    event :rally do
+      transition scattered: :column, if: -> { commander_alive? && morale > 40 }
+    end
+
+    event :reform do
+      transition %i[square wedge circle] => :line, unless: :under_immediate_threat?
+    end
+
+    before_transition any => :scattered, &:apply_panic_casualties
+
+    after_transition to: :square, &:prepare_pikes
+  end
+
+  # Supply state
+  state_machine :supply_state, initial: :supplied do
+    state :supplied do
+      def combat_effectiveness
+        1.0
+      end
+    end
+
+    state :low_supplies do
+      def combat_effectiveness
+        0.8
+      end
+    end
+
+    state :foraging do
+      def vulnerable_to_ambush?
+        true
+      end
+    end
+
+    state :starving do
+      def combat_effectiveness
+        0.4
+      end
+
+      def desertion_rate
+        0.1
+      end
+    end
+
+    event :consume_supplies do
+      transition supplied: :low_supplies, if: -> { supplies < 30 }
+      transition low_supplies: :starving, if: -> { supplies <= 0 }
+    end
+
+    event :send_foragers do
+      transition %i[low_supplies starving] => :foraging, if: :safe_to_forage?
+    end
+
+    event :resupply do
+      transition %i[low_supplies foraging starving] => :supplied
+    end
+
+    after_transition on: :resupply, to: :supplied do |regiment|
+      regiment.supplies = 100
+    end
+
+    event :foraging_success do
+      transition foraging: :low_supplies
+    end
+
+    after_transition on: :foraging_success, to: :low_supplies do |regiment|
+      regiment.supplies += 20
+    end
+  end
+
+  # Battle readiness
+  state_machine :readiness, initial: :fresh do
+    state :fresh
+    state :engaged
+    state :exhausted
+    state :victorious
+    state :routed
+
+    event :enter_battle do
+      transition fresh: :engaged
+      transition exhausted: :engaged, if: :desperate_defense?
+    end
+
+    event :win_engagement do
+      transition engaged: :victorious, if: -> { morale > 60 && casualties < 30 }
+    end
+
+    event :lose_engagement do
+      transition engaged: :routed, if: -> { morale < 30 || casualties > 60 }
+    end
+
+    event :exhaust do
+      transition engaged: :exhausted, if: -> { battle_duration > 480 }
+    end
+
+    event :rest do
+      transition %i[exhausted victorious] => :fresh
+    end
+  end
+
+  def sufficient_space?
+    true # Check battlefield conditions
+  end
+
+  def cavalry_threat?
+    @threats&.include?(:cavalry)
+  end
+
+  def emergency_defense?
+    morale < 40 && surrounded?
+  end
+
+  def surrounded?
+    @enemy_positions && @enemy_positions.count >= 3
+  end
+
+  def commander_present?
+    commander&.alive? && commander.with_regiment?
+  end
+
+  def commander_alive?
+    commander&.alive?
+  end
+
+  def under_immediate_threat?
+    @immediate_threats && !@immediate_threats.empty?
+  end
+
+  def apply_panic_casualties
+    @casualties += soldiers * 0.1
+  end
+
+  def prepare_pikes
+    @pike_wall_ready = true
+  end
+
+  def safe_to_forage?
+    !surrounded? && @enemy_activity_level < 3
+  end
+
+  def desperate_defense?
+    @defending_critical_position
+  end
+
+  def battle_duration
+    @battle_start_time ? Time.now - @battle_start_time : 0
+  end
+end

data/test/support/models/troll.rb

@@ -0,0 +1,167 @@
+# frozen_string_literal: true
+
+require_relative 'character'
+
+# Troll with regeneration mechanics
+class Troll < Character
+  attr_accessor :regeneration_rate, :rage_level, :last_damage_type, :berserker_stacks, :last_fire_damage_time,
+                :recovery_started, :health_changes, :enemies_nearby
+
+  def initialize
+    super
+    @regeneration_rate = 5
+    @rage_level = 0
+    @berserker_stacks = 0
+  end
+
+  state_machine :regeneration, initial: :normal do
+    state :normal do
+      def regen_multiplier
+        1.0
+      end
+    end
+
+    state :accelerated do
+      def regen_multiplier
+        2.0
+      end
+    end
+
+    state :berserk do
+      def regen_multiplier
+        3.0
+      end
+
+      def damage_taken_reduction
+        0.5
+      end
+    end
+
+    state :suppressed do
+      def regen_multiplier
+        0
+      end
+    end
+
+    state :dormant do
+      def regen_multiplier
+        0.1
+      end
+
+      def appears_dead?
+        true
+      end
+    end
+
+    event :enrage do
+      transition normal: :accelerated, if: -> { health < 75 }
+      transition accelerated: :berserk, if: -> { rage_level > 75 && health < 50 }
+    end
+
+    event :take_fire_damage do
+      transition %i[normal accelerated berserk] => :suppressed
+    end
+
+    event :take_acid_damage do
+      transition %i[normal accelerated] => :suppressed
+      transition berserk: :accelerated # Berserk provides some resistance
+    end
+
+    event :cool_down do
+      transition suppressed: :normal, if: -> { time_since_fire_damage > 60 }
+      transition %i[accelerated berserk] => :normal, unless: -> { health < 50 }
+    end
+
+    event :play_dead do
+      transition %i[normal suppressed] => :dormant, if: -> { health < 10 }
+    end
+
+    event :rise_again do
+      transition dormant: :normal, if: -> { health > 25 }
+      transition dormant: :berserk, if: -> { rage_level > 90 }
+    end
+
+    after_transition any => :berserk do |troll|
+      troll.regeneration_rate = 20
+      troll.berserker_stacks = 5
+    end
+
+    after_transition any => :suppressed do |troll|
+      troll.regeneration_rate = 0
+    end
+
+    after_transition to: :dormant do |troll|
+      troll.regeneration_rate = 1
+      troll.begin_slow_recovery
+    end
+
+    around_transition any => any do |troll, _transition, block|
+      health_before = troll.health
+      block.call
+      health_after = troll.health
+      troll.track_health_change(health_before - health_after)
+    end
+  end
+
+  # Combat stance
+  state_machine :combat_stance, initial: :defensive do
+    state :defensive do
+      def armor_bonus
+        5
+      end
+    end
+
+    state :aggressive do
+      def attack_bonus
+        3
+      end
+    end
+
+    state :reckless do
+      def attack_bonus
+        6
+      end
+
+      def armor_penalty
+        3
+      end
+    end
+
+    event :change_stance do
+      transition defensive: :aggressive, if: :confident?
+      transition aggressive: :reckless, if: :bloodthirsty?
+      transition %i[aggressive reckless] => :defensive, if: :threatened?
+    end
+  end
+
+  def time_since_fire_damage
+    return Float::INFINITY unless @last_fire_damage_time
+
+    Time.now - @last_fire_damage_time
+  end
+
+  def begin_slow_recovery
+    @recovery_started = Time.now
+  end
+
+  def track_health_change(amount)
+    @health_changes ||= []
+    @health_changes << { amount: amount, time: Time.now }
+  end
+
+  def confident?
+    health > 75 && berserker_stacks.positive?
+  end
+
+  def bloodthirsty?
+    rage_level > 80 && health > 50
+  end
+
+  def threatened?
+    health < 30 || surrounded?
+  end
+
+  def surrounded?
+    @enemies_nearby && @enemies_nearby.count > 3
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require 'minitest/autorun'
+require 'minitest/reporters'
+
+require 'state_machines/test_helper'
+require 'state_machines-mermaid'
+
+StateMachines::Machine.ignore_method_conflicts = true
+
+Dir[File.expand_path('support/models/**/*.rb', __dir__)].sort.each { |file| require file }
+
+class Minitest::Test
+  include StateMachines::TestHelper
+end
+
+Minitest::Reporters.use!(Minitest::Reporters::ProgressReporter.new)