/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * license agreements; and to You under the Apache License, version 2.0:
 *
 *   https://www.apache.org/licenses/LICENSE-2.0
 *
 * This file is part of the Apache Pekko project, which was derived from Akka.
 */

/*
 * Copyright (C) 2009-2022 Lightbend Inc. <https://www.lightbend.com>
 */

package org.apache.pekko.actor

import java.lang.IllegalStateException

import scala.concurrent.Await
import scala.concurrent.Promise
import scala.concurrent.duration._

import org.apache.pekko
import pekko.pattern.ask
import pekko.serialization.JavaSerializer
import pekko.testkit._
import pekko.util.Timeout

object ActorRefSpec {

  final case class ReplyTo(sender: ActorRef)

  class ReplyActor extends Actor {
    var replyTo: ActorRef = null

    def receive = {
      case "complexRequest" => {
        replyTo = sender()
        val worker = context.actorOf(Props[WorkerActor]())
        worker ! "work"
      }
      case "complexRequest2" =>
        val worker = context.actorOf(Props[WorkerActor]())
        worker ! ReplyTo(sender())
      case "workDone"      => replyTo ! "complexReply"
      case "simpleRequest" => sender() ! "simpleReply"
    }
  }

  class WorkerActor() extends Actor {
    import context.system
    def receive = {
      case "work" => {
        work()
        sender() ! "workDone"
        context.stop(self)
      }
      case ReplyTo(replyTo) => {
        work()
        replyTo ! "complexReply"
      }
    }

    private def work(): Unit = Thread.sleep(1.second.dilated.toMillis)
  }

  class SenderActor(replyActor: ActorRef, latch: TestLatch) extends Actor {

    def receive = {
      case "complex"      => replyActor ! "complexRequest"
      case "complex2"     => replyActor ! "complexRequest2"
      case "simple"       => replyActor ! "simpleRequest"
      case "complexReply" => {
        latch.countDown()
      }
      case "simpleReply" => {
        latch.countDown()
      }
    }
  }

  class OuterActor(val inner: ActorRef) extends Actor {
    def receive = {
      case "self" => sender() ! self
      case x      => inner.forward(x)
    }
  }

  class FailingOuterActor(val inner: ActorRef) extends Actor {
    val fail = new InnerActor

    def receive = {
      case "self" => sender() ! self
      case x      => inner.forward(x)
    }
  }

  class FailingInheritingOuterActor(_inner: ActorRef) extends OuterActor(_inner) {
    val fail = new InnerActor
  }

  class InnerActor extends Actor {
    def receive = {
      case "innerself" => sender() ! self
      case other       => sender() ! other
    }
  }

  class FailingInnerActor extends Actor {
    val fail = new InnerActor

    def receive = {
      case "innerself" => sender() ! self
      case other       => sender() ! other
    }
  }

  class FailingInheritingInnerActor extends InnerActor {
    val fail = new InnerActor
  }
}

class ActorRefSpec extends PekkoSpec("""
  # testing Java serialization of ActorRef
  pekko.actor.allow-java-serialization = on
  """) with DefaultTimeout {
  import org.apache.pekko.actor.ActorRefSpec._

  def promiseIntercept(f: => Actor)(to: Promise[Actor]): Actor =
    try {
      val r = f
      to.success(r)
      r
    } catch {
      case e: Throwable =>
        to.failure(e)
        throw e
    }

  def wrap[T](f: Promise[Actor] => T): T = {
    val result = Promise[Actor]()
    val r = f(result)
    Await.result(result.future, 1.minute)
    r
  }

  "An ActorRef" must {

    "not allow Actors to be created outside of an actorOf" in {
      import system.actorOf
      intercept[pekko.actor.ActorInitializationException] {
        new Actor { def receive = { case _ => } }
      }

      def contextStackMustBeEmpty(): Unit = ActorCell.contextStack.get.headOption should ===(None)

      EventFilter[ActorInitializationException](occurrences = 1).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(Props(new Actor {
              val nested = promiseIntercept(new Actor { def receive = { case _ => } })(result)
              def receive = { case _ => }
            })))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 1).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(Props(promiseIntercept(new FailingOuterActor(actorOf(Props(new InnerActor))))(result))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 1).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(Props(new OuterActor(actorOf(Props(promiseIntercept(new FailingInnerActor)(result)))))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 1).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(Props(promiseIntercept(new FailingInheritingOuterActor(actorOf(Props(new InnerActor))))(result))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 2).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(
              Props(new FailingOuterActor(actorOf(Props(promiseIntercept(new FailingInheritingInnerActor)(result)))))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 2).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(Props(new FailingInheritingOuterActor(
              actorOf(Props(promiseIntercept(new FailingInheritingInnerActor)(result)))))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 2).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(
              Props(new FailingInheritingOuterActor(actorOf(Props(promiseIntercept(new FailingInnerActor)(result)))))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 1).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(Props(new OuterActor(actorOf(Props(new InnerActor {
              val a = promiseIntercept(new InnerActor)(result)
            }))))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 2).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(
              Props(new FailingOuterActor(actorOf(Props(promiseIntercept(new FailingInheritingInnerActor)(result)))))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 1).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(Props(new OuterActor(actorOf(Props(promiseIntercept(new FailingInheritingInnerActor)(result)))))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 1).intercept {
        intercept[pekko.actor.ActorInitializationException] {
          wrap(result =>
            actorOf(
              Props(new OuterActor(actorOf(Props(promiseIntercept { new InnerActor; new InnerActor }(result)))))))
        }

        contextStackMustBeEmpty()
      }

      EventFilter[ActorInitializationException](occurrences = 1).intercept {
        intercept[java.lang.IllegalStateException] {
          wrap(result =>
            actorOf(Props(new OuterActor(actorOf(Props(promiseIntercept {
              throw new IllegalStateException("Ur state be b0rked")
            }(result)))))))
        }.getMessage should ===("Ur state be b0rked")

        contextStackMustBeEmpty()
      }
    }

    "insert its path in a ActorInitializationException" in {
      EventFilter[ActorInitializationException](occurrences = 1, pattern = "/user/failingActor:").intercept {
        intercept[java.lang.IllegalStateException] {
          wrap(result =>
            system.actorOf(Props(promiseIntercept {
                throw new IllegalStateException
              }(result)), "failingActor"))
        }
      }
    }

    "be serializable using Java Serialization on local node" in {
      val a = system.actorOf(Props[InnerActor]())
      val esys = system.asInstanceOf[ExtendedActorSystem]

      import java.io._

      val baos = new ByteArrayOutputStream(8192 * 32)
      val out = new ObjectOutputStream(baos)

      out.writeObject(a)

      out.flush
      out.close

      val bytes = baos.toByteArray

      JavaSerializer.currentSystem.withValue(esys) {
        val in = new ObjectInputStream(new ByteArrayInputStream(bytes))
        val readA = in.readObject

        a.isInstanceOf[ActorRefWithCell] should ===(true)
        readA.isInstanceOf[ActorRefWithCell] should ===(true)
        (readA eq a) should ===(true)
      }

      val ser = new JavaSerializer(esys)
      val readA = ser.fromBinary(bytes, None)
      readA.isInstanceOf[ActorRefWithCell] should ===(true)
      (readA eq a) should ===(true)
    }

    "throw an exception on deserialize if no system in scope" in {
      val a = system.actorOf(Props[InnerActor]())

      import java.io._

      val baos = new ByteArrayOutputStream(8192 * 32)
      val out = new ObjectOutputStream(baos)

      out.writeObject(a)

      out.flush
      out.close

      val in = new ObjectInputStream(new ByteArrayInputStream(baos.toByteArray))

      intercept[java.lang.IllegalStateException] {
        in.readObject
      }.getMessage should ===(
        "Trying to deserialize a serialized ActorRef without an ActorSystem in scope." +
        " Use 'org.apache.pekko.serialization.JavaSerializer.currentSystem.withValue(system) { ... }'")
    }

    "return EmptyLocalActorRef on deserialize if not present in actor hierarchy (and remoting is not enabled)" in {
      import java.io._

      val baos = new ByteArrayOutputStream(8192 * 32)
      val out = new ObjectOutputStream(baos)

      val sysImpl = system.asInstanceOf[ActorSystemImpl]
      val ref = system.actorOf(Props[ReplyActor](), "non-existing")
      val serialized = SerializedActorRef(ref)

      out.writeObject(serialized)

      out.flush
      out.close

      ref ! PoisonPill

      watch(ref)
      expectTerminated(ref)

      JavaSerializer.currentSystem.withValue(sysImpl) {
        val in = new ObjectInputStream(new ByteArrayInputStream(baos.toByteArray))
        in.readObject should ===(new EmptyLocalActorRef(sysImpl.provider, ref.path, system.eventStream))
      }
    }

    "support nested actorOfs" in {
      val a = system.actorOf(Props(new Actor {
        val nested = system.actorOf(Props(new Actor { def receive = { case _ => } }))
        def receive = { case _ => sender() ! nested }
      }))

      val nested = Await.result((a ? "any").mapTo[ActorRef], timeout.duration)
      a should not be null
      nested should not be null
      (a ne nested) should ===(true)
    }

    "support advanced nested actorOfs" in {
      val a = system.actorOf(Props(new OuterActor(system.actorOf(Props(new InnerActor)))))
      val inner = Await.result(a ? "innerself", timeout.duration)

      Await.result(a ? a, timeout.duration) should ===(a)
      Await.result(a ? "self", timeout.duration) should ===(a)
      inner should not be a

      Await.result(a ? "msg", timeout.duration) should ===("msg")
    }

    "support reply via sender" in {
      val latch = new TestLatch(4)
      val serverRef = system.actorOf(Props[ReplyActor]())
      val clientRef = system.actorOf(Props(new SenderActor(serverRef, latch)))

      clientRef ! "complex"
      clientRef ! "simple"
      clientRef ! "simple"
      clientRef ! "simple"

      Await.ready(latch, timeout.duration)

      latch.reset()

      clientRef ! "complex2"
      clientRef ! "simple"
      clientRef ! "simple"
      clientRef ! "simple"

      Await.ready(latch, timeout.duration)

      system.stop(clientRef)
      system.stop(serverRef)
    }

    "support actorOfs where the class of the actor isn't public" in {
      val a = system.actorOf(NonPublicClass.createProps())
      a.tell("pigdog", testActor)
      expectMsg("pigdog")
      system.stop(a)
    }

    "stop when sent a poison pill" in {
      val timeout = Timeout(20.seconds)
      val ref = system.actorOf(Props(new Actor {
        def receive = {
          case 5 => sender() ! "five"
          case 0 => sender() ! "null"
        }
      }))

      val ffive = ref.ask(5)(timeout).mapTo[String]
      val fnull = ref.ask(0)(timeout).mapTo[String]
      ref ! PoisonPill

      Await.result(ffive, timeout.duration) should ===("five")
      Await.result(fnull, timeout.duration) should ===("null")

      watch(ref)
      expectTerminated(ref)
    }

    "restart when Kill:ed" in {
      filterException[ActorKilledException] {
        val latch = TestLatch(2)

        val boss = system.actorOf(Props(new Actor {

          override val supervisorStrategy =
            OneForOneStrategy(maxNrOfRetries = 2, withinTimeRange = 1.second)(List(classOf[Throwable]))

          val ref = context.actorOf(Props(new Actor {
            def receive = { case _ => }
            override def preRestart(reason: Throwable, msg: Option[Any]) = latch.countDown()
            override def postRestart(reason: Throwable) = latch.countDown()
          }))

          def receive = { case "sendKill" => ref ! Kill }
        }))

        boss ! "sendKill"
        Await.ready(latch, 5.seconds)
      }
    }

    "be able to check for existence of children" in {
      val parent = system.actorOf(Props(new Actor {

          val child = context.actorOf(Props(new Actor {
              def receive = { case _ => }
            }), "child")

          def receive = { case name: String => sender() ! context.child(name).isDefined }
        }), "parent")

      assert(Await.result(parent ? "child", timeout.duration) === true)
      assert(Await.result(parent ? "whatnot", timeout.duration) === false)
    }
  }
}
