-- name: \\#833ab4\\DGoki \\#c02c69\\Moveset \\#fcb045\\v1.0
-- incompatible: movesets
-- description: This is a moveset created by me (\\#833ab4\\D\\#c02c69\\G\\#fd1d1d\\o\\#fd6731\\k\\#fcb045\\i\\#FFFFFF\\)\nIt tries to be advanced while also not removing things like BLJs.\nI was going to keep this private, however many people asked me to release it.\n\nHere a list what the moveset does:\n\\#FFFF00\\\n+ Ground pound jump. (A HOLD OR PRESS)\n+ Slide kick on dive. (A PRESS)\n+ Allows to go up on slopes easier.\n+ Allows you to dive on a ground pound to cancel it. (B PRESS)\n+ Allows you to slide kick on a long jump. (A PRESS WHILE NOT HOLDING Z)\n+ Allows you to dive on a long jump. (B PRESS)\n+ Adds a quadruple jump that causes you to twirl. (A PRESS)\n+ Modified speed (PRESS Y TO SWITCH MODES)\n+ Allows for Mid-air rotation (PRESS X TO TOGGLE)\n+ Other small changes.
local dg_moveset = true
local dg_moveset_hud_on = true
local dg_directional = true

-- Used for the death / reset command
local dg_reset_char = false
local dg_moveset_speed = 0

-- gPlayerSyncTable bullshit | i don't like using this in general because desync exists. :(
for player = 0, MAX_PLAYERS - 1 do gPlayerSyncTable[player].dg_moveset_toggled = true end
for player = 0, MAX_PLAYERS - 1 do gPlayerSyncTable[player].dg_moveset_speed = 0 end
for player = 0, MAX_PLAYERS - 1 do gPlayerSyncTable[player].dg_directional = true end

local function DG_Die() -- The reset / death command
    dg_reset_char = true
    return true
end

local function death_check(m)
    -- This is the most stupid crash fix that i ever used.
    -- but it works for rare crashes related to deaths.
    -- so i keep this in the code.
    
    -- Deaths.
    if m.action == ACT_DEATH_ON_BACK then return true end
    if m.action == ACT_DEATH_ON_STOMACH then return true end
    if m.action == ACT_WATER_DEATH then return true end
    if m.action == ACT_STANDING_DEATH then return true end
    if m.action == ACT_QUICKSAND_DEATH then return true end
    -- Death exits.
    if m.action == ACT_DEATH_EXIT then return true end
    if m.action == ACT_FALLING_DEATH_EXIT then return true end
    if m.action == ACT_DEATH_EXIT_LAND then return true end
    if m.action == ACT_SPECIAL_DEATH_EXIT then return true end
    -- Bubble because yes.
    if m.action == ACT_BUBBLED then return true end
    return false
end
local function knockback_check(m)
    -- Used for the speed increase.
    -- Since not having this would send you to the moon if pvp is active.
    -- Also cannons and other things like being thrown is also checked here. 
    
    -- Ground KB
    if m.action == ACT_SOFT_BACKWARD_GROUND_KB then return true end
    if m.action == ACT_SOFT_FORWARD_GROUND_KB then return true end
    if m.action == ACT_HARD_FORWARD_GROUND_KB then return true end
    if m.action == ACT_HARD_BACKWARD_GROUND_KB then return true end
    if m.action == ACT_FORWARD_GROUND_KB then return true end
    if m.action == ACT_BACKWARD_GROUND_KB then return true end
    -- Cannon
    if m.action == ACT_SHOT_FROM_CANNON then return true end
    if m.action == ACT_IN_CANNON then return true end
    -- Air KB
    if m.action == ACT_HARD_FORWARD_AIR_KB then return true end
    if m.action == ACT_HARD_BACKWARD_AIR_KB then return true end
    if m.action == ACT_FORWARD_AIR_KB then return true end
    if m.action == ACT_BACKWARD_AIR_KB then return true end
    -- Water KB
    if m.action == ACT_FORWARD_WATER_KB then return true end
    if m.action == ACT_BACKWARD_WATER_KB then return true end
    -- Throws
    if m.action == ACT_THROWN_BACKWARD then return true end
    if m.action == ACT_THROWN_FORWARD then return true end
    return false
end

local A_PRESS = false
local B_PRESS = false
local X_PRESS = false
local Y_PRESS = false
local Z_TRIG_PRESS = false
---@param m MarioState
local function dg_moveset_mario_update(m)
    ---------------------------------------------------------------------------------------
    -- This is the reset / death command
    -- Used for if the moveset get's somehow desynced or you get OOB trapped in a location.
    ---------------------------------------------------------------------------------------
    if m.playerIndex == 0 then
        gPlayerSyncTable[m.playerIndex].dg_moveset_toggled = dg_moveset
    end
    if dg_reset_char == true and m.playerIndex == 0 then
        m.health = 0
        m.healCounter = 0
        m.numLives = m.numLives + 1
        dg_reset_char = false
    end
    -- this fixes some rare crashes related to the death animation.
    if death_check(m) or gPlayerSyncTable[m.playerIndex].dg_moveset_toggled == false then return end
    ---------------------------------------------------------------------------------------
    -- Getting binds because is important.
    ---------------------------------------------------------------------------------------
    A_PRESS = ((m.controller.buttonPressed & A_BUTTON) ~= 0)
    B_PRESS = ((m.controller.buttonPressed & B_BUTTON) ~= 0)
    X_PRESS = ((m.controller.buttonPressed & X_BUTTON) ~= 0)
    Y_PRESS = ((m.controller.buttonPressed & Y_BUTTON) ~= 0)
    Z_TRIG_PRESS = ((m.controller.buttonPressed & Z_TRIG) ~= 0)
    ---------------------------------------------------------------------------------------
    -- There is a simpler way to do this but i am lazy, who will read this code anyways?
    ---------------------------------------------------------------------------------------
    if X_PRESS and m.playerIndex == 0 then
        if gPlayerSyncTable[m.playerIndex].dg_directional == true then
           gPlayerSyncTable[m.playerIndex].dg_directional = false
            dg_directional = false
        else 
            gPlayerSyncTable[m.playerIndex].dg_directional = true
            dg_directional = true
        end
    end
    if Y_PRESS and m.playerIndex == 0 then
        gPlayerSyncTable[m.playerIndex].dg_moveset_speed = gPlayerSyncTable[m.playerIndex].dg_moveset_speed + 1
        dg_moveset_speed = dg_moveset_speed + 1
        if gPlayerSyncTable[m.playerIndex].dg_moveset_speed == 3 then
            gPlayerSyncTable[m.playerIndex].dg_moveset_speed = -1
            dg_moveset_speed = -1
        end
    end
    ---------------------------------------------------------------------------------------
    -- This code is horrible to read and edit, i know, i know what i did here.
    -- but also i am kinda too lazy to fix it right now, so uh, enjoy this mess. :P
    -- Edit: I Made it a bit easier to read.
    ---------------------------------------------------------------------------------------
    -- Ground pound jump. (some people call it: "ass rocket" for some reason, idk.)
    if (m.controller.buttonDown & A_BUTTON) ~= 0 and m.action == ACT_GROUND_POUND_LAND then
	    set_mario_action(m, ACT_LAVA_BOOST, 0)
            m.vel.y = 70
    -- Allows you to dive from a top pole jump or a lava boost. (which also includes the ground pound jump)
    elseif B_PRESS and (m.action == ACT_TOP_OF_POLE_JUMP or m.action == ACT_LAVA_BOOST) then
	   set_mario_action(m, ACT_DIVE, 0)
    -- Allws you to ground pound from a top pole jump or a lava boost. (which also includes the ground pound jump)
    elseif Z_TRIG_PRESS and (m.action == ACT_TOP_OF_POLE_JUMP or m.action == ACT_LAVA_BOOST) then
	   set_mario_action(m, ACT_GROUND_POUND, 0)
    -- Allows you to dive from a long jump (really useful not gonna lie.)
    elseif B_PRESS and m.action == ACT_LONG_JUMP then
	   set_mario_action(m, ACT_DIVE, 0)
    -- Allows you to slide kick from a long jump (this checks also if Z_TRIG is not being hold, so you can BLJ with it.)
    elseif A_PRESS and (m.controller.buttonDown & Z_TRIG) == 0 and m.action == ACT_LONG_JUMP then
	    set_mario_action(m, ACT_SLIDE_KICK, 0)
    -- Allows you to dive faster (i added it for the funny.)
    elseif B_PRESS and m.action == ACT_DIVE_PICKING_UP then
	    set_mario_action(m, ACT_DIVE, 0)
    -- Allows you to dive out of a ground pound. (a guy called "sonk" or something, said i should add this instead of a kick.)
    elseif B_PRESS and m.action == ACT_GROUND_POUND then
	    set_mario_action(m, ACT_DIVE, 0)
            m.vel.y = 50
    -- Allows you to Slide kick from a dive. (this makes like 90% of movement tech possible from this moveset.)
    elseif A_PRESS and (m.action == ACT_DIVE or m.action == ACT_DIVE_SLIDE) then
	    set_mario_action(m, ACT_SLIDE_KICK, 0)
    -- Allows you to cancel a slide kick if you are sliding with it. (not in air so it prevents, you getting infinite height.)
    elseif B_PRESS and (m.action == ACT_SLIDE_KICK_SLIDE) then
	    set_mario_action(m, ACT_DIVE, 0)
    -- Adds a twirl jump after a triple jump. (not gonna lie, this is i guess the most pointless feature.)
    elseif A_PRESS and m.action == ACT_TRIPLE_JUMP_LAND then
            m.vel.y = 150
	    set_mario_action(m,ACT_TWIRLING, 0)
    -- Allows you to dive from a SPECIAL triple jump.
    elseif B_PRESS and (m.action == ACT_SPECIAL_TRIPLE_JUMP) then
            set_mario_action(m,ACT_DIVE, 0)
    -- Allows you to ground pound from a SPECIAL triple jump.
    elseif Z_TRIG_PRESS and (m.action == ACT_SPECIAL_TRIPLE_JUMP) then
            set_mario_action(m,ACT_GROUND_POUND, 0)
    -- Allows you to dive while twirling.
    elseif B_PRESS and m.action == ACT_TWIRLING then
            set_mario_action(m,ACT_DIVE, 0)
    -- Allows you to ground pound while twirling.
    elseif Z_TRIG_PRESS and m.action == ACT_TWIRLING then
            set_mario_action(m,ACT_GROUND_POUND, 0)
    -- Allows you to jump while sliding on your stomach.
    elseif A_PRESS and m.action == ACT_STOMACH_SLIDE then
            set_mario_action(m,ACT_JUMP, 0)
    -- Allows you to climp slopes way easier.
    elseif (m.action == ACT_STEEP_JUMP) then
            set_mario_action(m,ACT_JUMP, 0)
    -- Makes the water jump actually useful.
    elseif (m.action == ACT_WATER_JUMP) then
            m.vel.y = 35
            set_mario_action(m,ACT_TRIPLE_JUMP, 0)
    -- Allows you to dive from a backflip.
    elseif B_PRESS and m.action == ACT_BACKFLIP then
            set_mario_action(m,ACT_DIVE, 0)
    -- Allows you to triple jump from a backflip.
    elseif A_PRESS and m.action == ACT_BACKFLIP and (m.controller.buttonDown & Z_TRIG) == 0 then
            set_mario_action(m,ACT_TRIPLE_JUMP, 0)
    -- Literally bonk immunity, like is absurd.
    elseif m.action == ACT_SOFT_BONK or m.action == ACT_GROUND_BONK or m.action == ACT_HARD_BACKWARD_AIR_KB or m.action == ACT_BACKWARD_AIR_KB then
            set_mario_action(m,ACT_JUMP,0)
    end
    -- Allows you to better control mario. (kind of, it's still goofy but i have no idea to fix this, so it has to do.)
    if m.action == ACT_DIVE or m.action == ACT_LONG_JUMP or m.action == ACT_DIVE_SLIDE or m.action == ACT_JUMP_KICK or m.action == ACT_SLIDE_KICK or m.action == ACT_SLIDE_KICK_SLIDE or m.action == ACT_JUMP or m.action == ACT_DOUBLE_JUMP or m.action == ACT_TRIPLE_JUMP or m.action == ACT_GROUND_POUND then
        if gPlayerSyncTable[m.playerIndex].dg_directional == true then
            m.faceAngle.y = m.faceAngle.y - (m.controller.rawStickX*10)
        end
    end
end

---@param m MarioState
local function dg_moveset_before_phys_step(m)
    if (gPlayerSyncTable[m.playerIndex].dg_moveset_toggled == false) then return end
    if death_check(m) or knockback_check(m) then return end
    -- Increases movement speed since it was too slow for my taste.
    if gPlayerSyncTable[m.playerIndex].dg_moveset_speed == -1 then
        m.vel.x = m.vel.x * 0.50
        m.vel.z = m.vel.z * 0.50
    end
    if gPlayerSyncTable[m.playerIndex].dg_moveset_speed == 1 then
        m.vel.x = m.vel.x * 1.35
        m.vel.z = m.vel.z * 1.35
    end
    if gPlayerSyncTable[m.playerIndex].dg_moveset_speed == 2 then
        m.vel.x = m.vel.x * 1.50
        m.vel.z = m.vel.z * 1.50
    end
end

-- i am too lazy to use an better alternative.
local dg_x = 69
local dg_y = 69
local function dg_moveset_hud()
    if dg_moveset_hud_on == false or dg_moveset == false then return end
    dg_x = djui_hud_get_screen_width()/2
    dg_y = (djui_hud_get_screen_height()/2)+250
    djui_hud_set_color(0,0,0,155)
    djui_hud_render_rect(dg_x-10,dg_y-55,635,240)
    djui_hud_render_rect(dg_x,dg_y-45,615,220)
    djui_hud_set_font(FONT_MENU)
    djui_hud_set_color(255,255,0,255)
    djui_hud_print_text("---DGoki-Moveset---",dg_x,dg_y-45,1)
    if dg_directional == true then  
        djui_hud_set_color(0,255,0,255)
        djui_hud_print_text("AIR-ROTATION: ON",dg_x+20,dg_y,1)
        djui_hud_set_color(0,255,0,155)
        djui_hud_print_text("PRESS X TO TOGGLE",dg_x+20,dg_y+40,1)
    end
    if dg_directional == false then  
        djui_hud_set_color(255,0,0,255)
        djui_hud_print_text("AIR-ROTATION: OFF",dg_x+20,dg_y,1)
        djui_hud_set_color(255,0,0,155)
        djui_hud_print_text("PRESS X TO TOGGLE",dg_x+20,dg_y+40,1)
    end
    if dg_moveset_speed == -1 then
        djui_hud_set_color(55,55,255,255)
        djui_hud_print_text("SPEED: 0.5X",dg_x+20,dg_y+80,1)
        djui_hud_set_color(55,55,255,155)
        djui_hud_print_text("PRESS Y TO SWITCH",dg_x+20,dg_y+120,1)
    end
    if dg_moveset_speed == 0 then
        djui_hud_set_color(255,255,255,255)
        djui_hud_print_text("SPEED: 1.0X",dg_x+20,dg_y+80,1)
        djui_hud_set_color(255,255,255,155)
        djui_hud_print_text("PRESS Y TO SWITCH",dg_x+20,dg_y+120,1)
    end
    if dg_moveset_speed == 1 then
        djui_hud_set_color(255,155,155,255)
        djui_hud_print_text("SPEED: 1.3X",dg_x+20,dg_y+80,1)
        djui_hud_set_color(255,155,155,155)
        djui_hud_print_text("PRESS Y TO SWITCH",dg_x+20,dg_y+120,1)
    end
    if dg_moveset_speed == 2 then
        djui_hud_set_color(255,55,55,255)
        djui_hud_print_text("SPEED: 1.5X",dg_x+20,dg_y+80,1)
        djui_hud_set_color(255,55,55,155)
        djui_hud_print_text("PRESS Y TO SWITCH",dg_x+20,dg_y+120,1)
    end
end

hook_event(HOOK_BEFORE_PHYS_STEP, dg_moveset_before_phys_step)
hook_event(HOOK_ON_HUD_RENDER, dg_moveset_hud)
hook_event(HOOK_MARIO_UPDATE, dg_moveset_mario_update)
local function DG_Info()
    djui_chat_message_create("\\#FFFF00\\+-----------------------------------------+")
    djui_chat_message_create("\\#FFFF00\\Here a list what the moveset does:\n(With some extra information.)")
    djui_chat_message_create("\\#FFFF00\\Ground pound jump. (A HOLD)\n(Also it's immune to quicksand.)")
    djui_chat_message_create("\\#FFFF00\\Slide kick on dive. (A PRESS)\n(Allowing you to get absurdly far.)")
    djui_chat_message_create("\\#FFFF00\\Allows to go up on slopes easier.\n(Also you cannot bonk anymore.)")
    djui_chat_message_create("\\#FFFF00\\Allows you to dive on a ground pound to cancel it. (B PRESS)\n(Allowing for funky tech.)")
    djui_chat_message_create("\\#FFFF00\\Allows you to slide kick on a long jump. (A PRESS)\n(Rarely useful but it has it's moments.)")
    djui_chat_message_create("\\#FFFF00\\Allows you to dive on a long jump. (B PRESS)\n(Allows for Longjump>dive>slidekick.)")
    djui_chat_message_create("\\#FFFF00\\Adds a quadruple jump that causes you to twirl. (A PRESS)\n(Good for getting into tricky spots.)")
    djui_chat_message_create("\\#FFFF00\\Modified speed. (PRESS Y TO SWITCH MODES)\n(Switches between: 0.5x,1.0x,1.3x,1.5x)")
    djui_chat_message_create("\\#FFFF00\\Allows for Mid-air rotation. (PRESS X TO TOGGLE)\n(It's hard to control but is useful.)")
    djui_chat_message_create("\\#FFFF00\\Allows you to triple jump on a backflip, mid-air. (A PRESS)\n(Allows for quadruple jump after land.)")
    djui_chat_message_create("\\#FFFF00\\Also other small changes.")
    djui_chat_message_create("\\#FFFF00\\+-----------------------------------------+")
    djui_chat_message_create("\\#FFFF00\\+ Scroll up to see all of the information +")
    djui_chat_message_create("\\#FFFF00\\+-----------------------------------------+")
    return true
end

hook_chat_command("dg_reset", "- kills your local player to reset it.", DG_Die)
hook_chat_command("dg_help","- gives information about the moveset.", DG_Info)

local function dg_moveset_toggle()
    if dg_moveset == true then dg_moveset = false else dg_moveset = true end
end
local function dg_moveset_hud_toggle()
    if dg_moveset_hud_on == true then dg_moveset_hud_on = false else dg_moveset_hud_on = true end
end

hook_mod_menu_checkbox("Enable Moveset", true, dg_moveset_toggle)
hook_mod_menu_checkbox("Enable HUD", true, dg_moveset_hud_toggle)