modules for each stage of a tube amp

lib/blues/tube_amplifier.rb

@@ -2,49 +2,6 @@
 
 module Blues
   class TubeAmplifier < Amplifier
-    def turn_on
-      if @warm_up.nil?
-        @warm_up = TubeWarmUp.new
-      end
-
-      super
-    end
-
-    def turn_off
-      @warm_up = nil
-
-      super
-    end
-
-    def warm_up_complete?
-      return false unless @warm_up
-
-      @warm_up.complete?
-    end
-
-    def volume
-      return @volume if warm_up_complete?
-      return 0 unless @warm_up
-
-      (@volume * @warm_up.volume_scale).floor
-    end
-
-    def pre_amp_tone
-      if low_volume? || mid_volume?
-        "💡"
-      elsif high_volume?
-        "🔥"
-      else
-        " "
-      end
-    end
-
-    def pre_amp_weight
-      :heavy
-    end
-
-    def power_amp_weight
-      :heavy
-    end
+    include TubePreAmp, TubePowerAmp
   end
 end

lib/blues/tube_power.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Blues
+  module TubePower
+    def turn_on
+      if @warm_up.nil?
+        @warm_up = TubeWarmUp.new
+      end
+
+      super
+    end
+
+    def turn_off
+      @warm_up = nil
+
+      super
+    end
+
+    def warm_up_complete?
+      return false unless @warm_up
+
+      @warm_up.complete?
+    end
+
+    def volume
+      return @volume if warm_up_complete?
+      return 0 unless @warm_up
+
+      (@volume * @warm_up.volume_scale).floor
+    end
+  end
+end

lib/blues/tube_power_amp.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Blues
+  module TubePowerAmp
+    include TubePower
+
+    def power_amp_weight
+      :heavy
+    end
+  end
+end

lib/blues/tube_pre_amp.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Blues
+  module TubePreAmp
+    include TubePower
+
+    def pre_amp_tone
+      if low_volume? || mid_volume?
+        "💡"
+      elsif high_volume?
+        "🔥"
+      else
+        " "
+      end
+    end
+
+    def pre_amp_weight
+      :heavy
+    end
+  end
+end