gear-core 2.0.0-pre.1

// Copyright (C) Gear Technologies Inc.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

use crate::{
    ids::{ActorId, MessageId, prelude::*},
    message::{Dispatch, DispatchKind, Message, SignalDetails},
};
use gear_core_errors::SignalCode;

/// Message for signal entry point.
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct SignalMessage {
    /// Message id.
    id: MessageId,
    /// Reply status code.
    code: SignalCode,
}

impl SignalMessage {
    /// Creates a new [`SignalMessage`].
    pub fn new(origin_msg_id: MessageId, code: SignalCode) -> Self {
        let id = MessageId::generate_signal(origin_msg_id);

        Self { id, code }
    }

    /// Convert [`SignalMessage`] into [`Message`].
    pub fn into_message(self, origin_msg_id: MessageId, destination: ActorId) -> Message {
        Message::new(
            self.id,
            ActorId::SYSTEM,
            destination,
            Default::default(),
            None,
            0,
            Some(SignalDetails::new(origin_msg_id, self.code).into()),
        )
    }

    /// Convert [`SignalMessage`] into [`Dispatch`].
    pub fn into_dispatch(self, origin_msg_id: MessageId, destination: ActorId) -> Dispatch {
        Dispatch::new(
            DispatchKind::Signal,
            self.into_message(origin_msg_id, destination),
        )
    }

    /// Message id.
    pub fn id(&self) -> MessageId {
        self.id
    }

    /// Status code of the reply message.
    pub fn code(&self) -> SignalCode {
        self.code
    }
}