namespace NightLight.Core.Tests

open NightLight.Core.Core
open NightLight.Core.Tests.GenHelpers
open NightLight.Core.Tests.TimeChangedGenerators
open NightLight.Core.Tests.InteractionListGenerators
open NightLight.Core.Models
open FsCheck.Xunit
open FsCheck.FSharp

type NightLightTests() =
    let createFakeHomeWithNightLightAndInteract (interactions: Interaction list) =
        let mutable nightLightStateMachine = NightLightStateMachine()

        let fakeHome = FakeHome()

        fakeHome.OnEventPublished.Add(fun event ->
            match event |> nightLightStateMachine.OnEventReceived with
            | Ok(newState, commands) ->
                commands |> Seq.iter fakeHome.ProcessCommand
                nightLightStateMachine <- newState
            | Error error -> failwith $"Unexpected error {error}")

        fakeHome.Interact interactions

        fakeHome

    let doesLightHavePowerAfter light interactions =
        // `FakeHome` intentionally doesn't expose this information, since there is no
        // (easy) way for a person in the real world either to distinguish between
        // a lamp that *physically* does not have power, and a lamp that has simply
        // been turned off programmatically (but which still has power and can thus
        // receive new commands).
        //
        // They can, however, deduce it by remembering the last time that they flicked
        // the lamp's switch, just like this function does.
        interactions
        |> Seq.choose (fun interaction ->
            match interaction with
            | HumanInteraction(LightPoweredOff l) when l = light -> Some false
            | HumanInteraction(LightPoweredOn l) when l = light -> Some true
            | _ -> None)
        |> Seq.tryLast
        |> Option.defaultValue false

    [<Property(Arbitrary = [| typeof<ArbitraryLight> |])>]
    let ``All lights should be either off, white or yellow during the day`` (light: Light) =
        concatGens
            [ genInitialInteractions light
              genTimeChangedToRandomDayTime |> Gen.map List.singleton
              genInteractionsExcept light isTimeChangedToAnyNightTime ]
        |> Arb.fromGen
        |> Prop.forAll
        <| fun interactions ->
            let fakeHome = createFakeHomeWithNightLightAndInteract interactions

            fakeHome.LightShouldHaveState light (function
                | Off -> true
                | On(_, color) -> color = White || color = Yellow)

    [<Property(Arbitrary = [| typeof<ArbitraryLight> |])>]
    let ``All lights should be either off or red during the night`` (light: Light) =
        concatGens
            [ genInitialInteractions light
              genTimeChangedToRandomNightTime |> Gen.map List.singleton
              genInteractionsExcept light isTimeChangedToAnyDayTime ]
        |> Arb.fromGen
        |> Prop.forAll
        <| fun interactions ->
            let fakeHome = createFakeHomeWithNightLightAndInteract interactions

            fakeHome.LightShouldHaveState light (function
                | Off -> true
                | On(_, color) -> color = Red)

    [<Property(Arbitrary = [| typeof<ArbitraryNonRemotelyControlledLight> |])>]
    let ``All non-remotely controlled lights should be on iff they have power`` (light: Light) =
        genInitialInteractions light |> Arb.fromGen |> Prop.forAll
        <| fun interactions ->
            let fakeHome = createFakeHomeWithNightLightAndInteract interactions

            doesLightHavePowerAfter light interactions = fakeHome.LightShouldHaveState light _.IsOn

    [<Property(Arbitrary = [| typeof<ArbitraryRemotelyControlledLight> |])>]
    let ``All remote controlled lights with power should be on if the 'Off' button on the remote was never pressed``
        (light: Light)
        =
        genInitialInteractionsExcept light ((=) (HumanInteraction RemotePressedOffButton))
        |> Arb.fromGen
        |> Prop.forAll
        <| fun interactions ->
            let fakeHome = createFakeHomeWithNightLightAndInteract interactions

            doesLightHavePowerAfter light interactions
            ==> fakeHome.LightShouldHaveState light _.IsOn

    [<Property(Arbitrary = [| typeof<ArbitraryRemotelyControlledLight> |])>]
    let ``After pressing 'On' on the remote, if the 'Off' button isn't pressed, all remotely controlled lights with power should be on``
        (light: Light)
        =
        concatGens
            [ genInitialInteractions light
              HumanInteraction RemotePressedOnButton |> List.singleton |> Gen.constant
              genInteractionsExcept light ((=) (HumanInteraction RemotePressedOffButton)) ]
        |> Arb.fromGen
        |> Prop.forAll
        <| fun interactions ->
            let fakeHome = createFakeHomeWithNightLightAndInteract interactions

            doesLightHavePowerAfter light interactions
            ==> fakeHome.LightShouldHaveState light _.IsOn

    [<Property(Arbitrary = [| typeof<ArbitraryRemotelyControlledLight> |])>]
    let ``After a new day starts, if the 'Off' button isn't pressed, all remotely controlled lights with power should be on``
        (light: Light)
        =
        concatGens
            [ genInitialInteractions light
              genTimeChangedToRandomNightTime |> Gen.map List.singleton
              genInteractionsExcept light isTimeChangedToAnyDayTime
              genTimeChangedToRandomDayTime |> Gen.map List.singleton
              genInteractionsExcept light ((=) (HumanInteraction RemotePressedOffButton)) ]
        |> Arb.fromGen
        |> Prop.forAll
        <| fun interactions ->
            let fakeHome = createFakeHomeWithNightLightAndInteract interactions

            doesLightHavePowerAfter light interactions
            ==> fakeHome.LightShouldHaveState light _.IsOn

    [<Property(Arbitrary = [| typeof<ArbitraryRemotelyControlledLight> |])>]
    let ``After pressing 'Off' on the remote, if the 'On' button isn't pressed and a new day doesn't start, all remotely controlled lights should be off``
        (light: Light)
        =
        concatGens
            [ genInitialInteractions light
              HumanInteraction RemotePressedOffButton |> List.singleton |> Gen.constant
              genInteractionsExcept light (fun interaction ->
                  interaction = HumanInteraction RemotePressedOnButton
                  || interaction |> isTimeChangedToAnyDayTime) ]
        |> Arb.fromGen
        |> Prop.forAll
        <| fun interactions ->
            let fakeHome = createFakeHomeWithNightLightAndInteract interactions

            fakeHome.LightShouldHaveState light _.IsOff