assemblyline-models 0.8.1

Data models for the Assemblyline malware analysis platform.
Documentation
use serde::{Deserialize, Serialize};

use crate::types::Sid;

pub mod dispatching;
pub mod changes;
pub mod task;
pub mod submission;
pub mod ingest_heartbeat;
pub mod service_heartbeat;
pub mod dispatcher_heartbeat;


#[derive(Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all="lowercase")]
pub enum ArchiveAction {
    Archive,
    Resubmit,
}

#[derive(Serialize, Deserialize)]
pub struct ArchivedMessage {
    pub action: ArchiveAction,
    pub sid: Option<Sid>,
}

impl ArchivedMessage {
    pub fn archive() -> Self {
        Self {action: ArchiveAction::Archive, sid: None}
    }
    pub fn resubmit(sid: Sid) -> Self {
        Self {action: ArchiveAction::Resubmit, sid: Some(sid)}
    }
}

#[derive(Serialize, Deserialize)]
pub struct KillContainerCommand {
    pub service: String,
    pub container: String,
}