module Dodge.Humanoid (
    chaseCritInternal,
    crabCritInternal,
    hoverCritInternal,
) where

import Data.Foldable
import Dodge.Creature
import Dodge.Data.World
import LensHelp

chaseCritInternal :: World -> Creature -> Creature
chaseCritInternal w cr =
    foldl'
        (\c f -> f w c)
        cr
        [ const doStrategyActions
        , overrideMeleeCloseTarget
        , setViewPos
        , setMvPosToTargetCr
        , chaseCritMv
        , perceptionUpdate [0]
        , targetYouWhenCognizant
        , const searchIfDamaged
        , const (crType . meleeCooldown %~ max 0 . subtract 1)
        , const (crVocalization %~ updateVocTimer)
        ]

crabCritInternal :: Int -> World -> World
crabCritInternal cid w =
--    (crVocalization %~ updateVocTimer)
        (tocr %~ ((crType . meleeCooldownL %~ max 0 . subtract 1)))
        . (tocr %~ ((crType . meleeCooldownR %~ max 0 . subtract 1)))
        . (tocr %~ ((crType . dodgeCooldown %~ max 0 . subtract 1)))
        . (tocr %~ (searchIfDamaged))
        . (tocr %~ (targetYouWhenCognizant w))
        . (tocr %~ (perceptionUpdate [0] w))
        . crabActionUpdate cid
        . (tocr %~ (setMvPosToTargetCr w))
        . (tocr %~ (setViewPos w))
--        . overrideMeleeCloseTarget w
        $ (tocr %~ doStrategyActions) w
  where
    tocr = cWorld . lWorld . creatures . ix cid

hoverCritInternal :: World -> Creature -> Creature
hoverCritInternal w cr =
    foldl'
        (\c f -> f w c)
        cr
        [ const doStrategyActions
        , overrideMeleeCloseTarget
        , setViewPos
        , setMvPosToTargetCr
        , hoverCritMv
        , perceptionUpdate [0]
        , targetYouWhenCognizant
        , const searchIfDamaged
        , const (crType . meleeCooldown %~ max 0 . subtract 1)
        , const (crVocalization %~ updateVocTimer)
        ]

updateVocTimer :: Vocalization -> Vocalization
updateVocTimer VocTimer{_vcTime = x, _vcMaxTime = y} | x < y = VocTimer (x + 1) y
updateVocTimer _ = VocReady

--    SpreadGunAI ->
--        defaultImpulsive
--            [ performActions
--            , watchUpdateStrat
--                [
--                    ( crHasTargetLOS
--                    , \_ _ ->
--                        StrategyActions
--                            (ShootAt 0)
--                            [ DoActionIf (WdCrNegate $ WdCrBlfromCrBl CrIsAiming) drawWeapon
--                            , DoActionThen
--                                (DoActionWhile WdCrLOSTarget $ ArbitraryAction ChooseMovementSpreadGun)
--                                (DoImpulses [ChangeStrategy WatchAndWait])
--                            ]
--                    )
--                , (const crAwayFromPost, const goToPostStrat)
--                ]
--            , perceptionUpdate [0]
--            , const doStrategyActions
--            , targetYouWhenCognizant
--            , const $
--                overrideInternal
--                    (\cr' -> crHasTarget cr' && crStratConMatches (GetTo (V2 0 0)) cr')
--                    (\cr' -> cr' & crActionPlan . apStrategy .~ WatchAndWait)
--            ]
--            cr
--    PistolAI ->
--        defaultImpulsive
--            [ performActions
--            , watchUpdateStrat
--                [
--                    ( crHasTargetLOS
--                    , \w cr' ->
--                        StrategyActions
--                            (ShootAt 0)
--                            [DoActionIf (WdCrNegate $ WdCrBlfromCrBl CrIsAiming) drawWeapon, chooseMovementPistol cr' w]
--                    )
--                , (const crAwayFromPost, const goToPostStrat)
--                ]
--            , perceptionUpdate [0]
--            , const doStrategyActions
--            , targetYouWhenCognizant
--            , const $
--                overrideInternal
--                    (\cr' -> crHasTarget cr' && crStratConMatches (GetTo (V2 0 0)) cr')
--                    (crActionPlan . apStrategy .~ WatchAndWait)
--            ]
--            cr
--    LtAutoAI ->
--        defaultImpulsive
--            [ performActions
--            , watchUpdateStrat
--                [
--                    ( crHasTargetLOS
--                    , \_ _ ->
--                        StrategyActions
--                            (ShootAt 0)
--                            [ DoActionIf (WdCrNegate $ WdCrBlfromCrBl CrIsAiming) drawWeapon
--                            , DoActionThen
--                                (DoActionWhile WdCrLOSTarget $ ArbitraryAction ChooseMovementLtAuto)
--                                (DoImpulses [ChangeStrategy WatchAndWait])
--                            ]
--                    )
--                , (const crAwayFromPost, const goToPostStrat)
--                ]
--            , perceptionUpdate [0]
--            , const doStrategyActions
--            , targetYouWhenCognizant
--            , const $
--                overrideInternal
--                    (\cr' -> crHasTarget cr' && crStratConMatches (GetTo (V2 0 0)) cr')
--                    (crActionPlan . apStrategy .~ WatchAndWait)
--            ]
--            cr
--    LauncherAI ->
--        defaultImpulsive
--            [ performActions
--            , watchUpdateStrat
--                [ (crHasTargetLOS, \_ _ -> StrategyActions (ShootAt 0) [retreatFireLauncher])
--                , (const crAwayFromPost, const goToPostStrat)
--                ]
--            , perceptionUpdate [0]
--            , const doStrategyActions
--            , targetYouWhenCognizant
--            , const $
--                overrideInternal
--                    (\cr' -> crHasTarget cr' && crStratConMatches (GetTo (V2 0 0)) cr')
--                    (crActionPlan . apStrategy .~ WatchAndWait)
--            ]
--            cr
--    SwarmAI ->
--        defaultImpulsive
--            [ flockToPointUsing' (encircleDistP 100) (meleeHeadingMove 0.1 0.1 (pi / 4) 3.9)
--            , const (crMeleeCooldown %~ max 0 . subtract 1)
--            , perceptionUpdate [0]
--            , const doStrategyActions
--            , targetYouWhenCognizant
--            ]
--            cr
--    AutoAI -> defaultImpulsive [sentinelAI] cr
--    FlockArmourChaseAI ->
--        defaultImpulsive
--            [ watchUpdateStrat []
--            , const doStrategyActions
--            , performActions
--            , targetYouWhenCognizant
--            , setTargetMv (\_ -> _targetCr . _crIntention)
--            , flockACC
--            , perceptionUpdate [0]
--            , goToTarget
--            , const overrideMeleeCloseTarget
--            , const (crMeleeCooldown %~ max 0 . subtract 1)
--            ]
--            cr
--    MiniGunAI -> defaultImpulsive [sentinelFireType (const shootTillEmpty)] cr
--    LongAI -> defaultImpulsive [sentinelFireType $ const shootTillEmpty] cr
--    MultGunAI ->
--        defaultImpulsive
--            [ sentinelExtraWatchUpdate
--                [
----                    ( const $ not . crWeaponReady
----                    , \_ _ -> StrategyActions Reload reloadActions
----                    )
----                ,
--                    ( const $ not . crSafeDistFromTarg 150
--                    , \_ _ ->
--                        StrategyActions
--                            Flee
--                            [ WdCrNegate (WdCrSafeDistFromTarget 150) `DoActionWhile` UseSelf CrFleeFromTarget -- (fleeFrom cr')
--                                `DoActionThen` DoImpulses [ChangeStrategy WatchAndWait]
--                            ]
--                    )
--                ,
--                    ( crHasTargetLOS
--                    , \_ _ ->
--                        StrategyActions
--                            (ShootAt 0)
--                            [drawwp `DoActionThen` shootFirstMiss `DoActionThen` DoImpulses [ChangeStrategy WatchAndWait]]
--                    )
--                ]
--            ]
--            cr
--    InanimateAI -> id
--  where
--    drawwp = DoActionIfElse NoAction (WdCrBlfromCrBl CrIsAiming) (DoActionThen drawWeapon (WaitThen 50 NoAction))
--    reloadActions =
--        [ holsterWeapon
--        --, WaitThen 1 $ DoActionWhileInterrupt NoAction (WdCrBlfromCrBl CrIsReloading) (DoImpulses [ChangeStrategy WatchAndWait])
--        , WaitThen 1 $  (DoImpulses [ChangeStrategy WatchAndWait])
--        ]

-- chooseMovementPistol :: Creature -> World -> Action
-- chooseMovementPistol cr w =
--    chooseMovementPistol' cr w
--        `DoActionThen` DoImpulses [ChangeStrategy WatchAndWait]
--
-- chooseMovementPistol' :: Creature -> World -> Action
-- chooseMovementPistol' cr w =
--    takeOneWeighted
--        [chargeProb, retreatProb, strafeProb, strafeProb]
--        [ chargeActions
--        , retreatActionsPistol ycr cr
--        , strafeLeftActions
--        , strafeRightActions
--        ]
--        & evalState
--        $ g
--  where
--    g = _randGen w
--    cpos = cr ^. crPos . _xy
--    ycr = w ^?! cWorld . lWorld . creatures . ix 0
--    ypos = ycr ^. crPos . _xy
--    chargeProb
--        | dist cpos ypos > 300 = 5
--        | dist cpos ypos > 150 = 1
--        | otherwise = 0
--    strafeProb
--        | dist cpos ypos > 150 = 1
--        | otherwise = 0
--    retreatProb
--        | dist cpos ypos < 200 = 1 :: Float
--        | otherwise = 0
--    chargeActions =
--        [TurnToward ypos 0.1]
--            `DoImpulsesAlongside` 3
--            `DoReplicate` ImpulsesList (replicate 9 [Move (V2 3 0)] ++ [[Move (V2 3 0), UseItem]])
--    strafeLeftActions =
--        DoImpulses [TurnToward yposr (2 * pi)]
--            `DoActionThen` 3
--            `DoReplicate` ImpulsesList (replicate 9 [Move (V2 0 3)] ++ [[Move (V2 0 3), UseItem]])
--    strafeRightActions =
--        DoImpulses [TurnToward yposl (2 * pi)]
--            `DoActionThen` 3
--            `DoReplicate` ImpulsesList (replicate 9 [Move (V2 0 (-3))] ++ [[Move (V2 0 (-3)), UseItem]])
--    yposl = ypos -.- 100 *.* vNormal (normalizeV $ ypos -.- cpos)
--    yposr = ypos +.+ 100 *.* vNormal (normalizeV $ ypos -.- cpos)
--
-- retreatActionsPistol :: Creature -> Creature -> Action
-- retreatActionsPistol tcr cr =
--    [TurnToward retreatOffset 0.2]
--        `DoImpulsesAlongside` 3
--        `DoReplicate` ImpulsesList (replicate 9 [Move (V2 (-3) 0)] ++ [[UseItem]])
--  where
--    cpos = cr ^. crPos . _xy
--    tpos = tcr ^. crPos . _xy
--    retreatOffset =
--        let a
--                | dist cpos tpos < 50 = 0
--                | isLeftOfA (_crDir cr) (argV $ tpos -.- cpos) =
--                    -0.7
--                | otherwise = 0.7
--         in fromMaybe tpos $
--                intersectLineLine
--                    cpos
--                    (cpos +.+ rotateV a (tpos -.- cpos))
--                    tpos
--                    (tpos +.+ vNormal (cpos -.- tpos))
--
-- retreatFireLauncher :: Action
-- retreatFireLauncher =
--    ImpulsesList ([UseItem] : replicate 20 [Turn 0.16])
--        `DoActionThen` holsterWeapon
--        `DoActionThen` ImpulsesList (replicate 30 [MoveForward 3])
--        `DoActionThen` drawWeapon
--        `DoActionThen` ImpulsesList ([UseItem] : replicate 20 [Turn $ negate 0.16])
--        `DoActionThen` holsterWeapon
--        `DoActionThen` ImpulsesList (replicate 15 [MoveForward 3])
--        `DoActionThen` drawWeapon
--        `DoActionThen` ImpulsesList
--            ( replicate
--                100
--                [ UseItem
--                , ImpulseUseTarget (TurnTowardCr (pi / 16))
--                ]
--            )
--        `DoActionThen` 20
--        `WaitThen` holsterWeapon
--        `DoActionThen` ImpulsesList (replicate 15 [MoveForward 3])
--        `DoActionThen` DoImpulses [ChangeStrategy WatchAndWait]