"""Wavedash by TheMikirog
    
    This is an early version of the plugin. Feedback appreciated!

"""

# ba_meta require api 6

from __future__ import annotations

from typing import TYPE_CHECKING

import ba
import math
import bastd
from bastd.actor.spaz import Spaz

if TYPE_CHECKING:
    pass

# ba_meta export plugin
class MikiWavedashTest(ba.Plugin):
    

    class FootConnectMessage:
        """Spaz started touching the ground"""
    
    class FootDisconnectMessage:
        """Spaz stopped touching the ground"""
        
    def wavedash(self) -> None:
        if not self.node:
            return

        isMoving = abs(self.node.move_up_down) >= 0.5 or abs(self.node.move_left_right) >= 0.5
        
        if self._dead or not self.grounded or not isMoving:
            return
            
        if self.node.knockout > 0.0 or self.frozen or self.node.hold_node:
            return
            
        t_ms = ba.time(timeformat=ba.TimeFormat.MILLISECONDS)
        assert isinstance(t_ms, int)
        
        if t_ms - self.last_wavedash_time_ms >= self._wavedash_cooldown:
            
            move = [self.node.move_left_right, -self.node.move_up_down]
            vel = [self.node.velocity[0], self.node.velocity[2]]
            
            move_length = math.hypot(move[0], move[1])
            vel_length = math.hypot(vel[0], vel[1])
            if vel_length < 1.25: return
            move_norm = [m/move_length for m in move]
            vel_norm = [v/vel_length for v in vel]
            dot = sum(x*y for x,y in zip(move_norm,vel_norm))
            turn_power = min(round(math.acos(dot) / math.pi,2)*1.3,1)
            if turn_power < 0.2: return
            
            boost_power = math.sqrt(math.pow(vel[0],2) + math.pow(vel[1],2)) * 1.2
            boost_power = min(pow(boost_power,4),160)
            #print(boost_power * turn_power)
            
            self.last_wavedash_time_ms = t_ms
            
            # FX
            ba.emitfx(position=self.node.position,
                      velocity=(vel[0]*0.5,-1,vel[1]*0.5),
                      chunk_type='sweat',
                      count=8,
                      scale=boost_power / 160 * turn_power,
                      spread=0.25);
            
            # Boost itself
            pos = self.node.position
            for i in range(6):
                self.node.handlemessage('impulse',pos[0],-0.1+pos[1]+i*0.1,pos[2],
                                                0,0,0,
                                                boost_power * turn_power,
                                                boost_power * turn_power,0,0,
                                                move[0],0,move[1])
        
    def new_spaz_init(func):
        def wrapper(*args, **kwargs):

            func(*args, **kwargs)
                
            # args[0] = self
            args[0]._wavedash_cooldown = 30
            args[0].last_wavedash_time_ms = -9999
            args[0].grounded = 0
            
        return wrapper
    bastd.actor.spaz.Spaz.__init__ = new_spaz_init(bastd.actor.spaz.Spaz.__init__)
    
    def new_factory(func):
        def wrapper(*args, **kwargs):
            func(*args, **kwargs)

            args[0].roller_material.add_actions(
            conditions=('they_have_material', bastd.gameutils.SharedObjects.get().footing_material),
            actions=(('message', 'our_node', 'at_connect', MikiWavedashTest.FootConnectMessage),
                     ('message', 'our_node', 'at_disconnect', MikiWavedashTest.FootDisconnectMessage)))
        return wrapper
    bastd.actor.spazfactory.SpazFactory.__init__ = new_factory(bastd.actor.spazfactory.SpazFactory.__init__)
    
    def new_handlemessage(func):
        def wrapper(*args, **kwargs):
            if args[1] == MikiWavedashTest.FootConnectMessage:
                args[0].grounded += 1
            elif args[1] == MikiWavedashTest.FootDisconnectMessage:
                if args[0].grounded > 0: args[0].grounded -= 1
        
            func(*args, **kwargs)
        return wrapper
    bastd.actor.spaz.Spaz.handlemessage = new_handlemessage(bastd.actor.spaz.Spaz.handlemessage)
    
    def new_on_run(func):
        def wrapper(*args, **kwargs):
            if args[0]._last_run_value < args[1] and args[1] > 0.8:
                MikiWavedashTest.wavedash(args[0])
            func(*args, **kwargs)
        return wrapper
    bastd.actor.spaz.Spaz.on_run = new_on_run(bastd.actor.spaz.Spaz.on_run)