import magicbot
from phoenix6.hardware import TalonFX
from phoenix6.configs.talon_fx_configs import TalonFXConfiguration
from phoenix6.signals import NeutralModeValue
from phoenix6.controls import VoltageOut


class Agitator:
    motor: TalonFX
    voltage = magicbot.will_reset_to(0.0)

    def execute(self):
        self.motor.set_control(VoltageOut(self.voltage))

    def setup(self):
        config = TalonFXConfiguration()
        config.motor_output.neutral_mode = NeutralModeValue.BRAKE
        self.motor.configurator.apply(config)

    def spin(self):
        self.voltage = 1.0

    def stop(self):
        self.voltage = 0.0