plugin_run.rs

use crate::{
    api_model::{
        CreateItem, PluginCallApiReq, PluginCallApiRes, PluginGetApiReq, PluginGetApiRes,
        PluginIdentifier, PluginStatus, PluginStatusMetadata, PluginStatusReq, Search,
    },
    command_line_interface::CliOptions,
    database_api::{self, in_read_transaction},
    database_pool::PooledConnection,
    db_model::{ItemWithBase, PluginRunItem, CONTAINER_ID, CONTAINER_IMAGE},
    error::{Error, ErrorContext, Result},
    internal_api,
    internal_api::{new_random_string, search},
    plugin_auth_crypto::DatabaseKey,
    schema::Schema,
};
use futures::future;
use md5;
use rusqlite::Transaction;
use serde::Deserialize;
use serde_json::{json, Value};
use std::{
    collections::HashMap,
    ffi::OsStr,
    fmt::Debug,
    process::{Child, Command},
    sync::atomic::AtomicU32,
    time::Duration,
};
use tracing::{debug, info, warn};
use warp::http::status::StatusCode;

use duct;

const UNIQUE_ID: &str = "unique_id";
// TODO: this is barely maintainable, split to strategy pattern: Docker, K8S, sharing probably one trait

/// Run a plugin, making sure that the correct ENV variables and settings are passed
/// to the containerization / deployment processes.
///
/// Internally passes to docker / kubernetes / scripts
/// depending on how Pod is configured by the user.
#[allow(clippy::too_many_arguments)]
pub fn run_plugin_container(
    tx: &Transaction,
    schema: &Schema,
    triggered_by_item_id: &str,
    plugin: &PluginRunItem,
    pod_owner: &str,
    database_key: &DatabaseKey,
    cli_options: &CliOptions,
) -> Result<()> {
    info!(
        "Trying to run plugin container for target_item_id {}",
        plugin.target_item_id
    );
    let target_item = internal_api::get_item_tx(tx, schema, plugin.target_item_id.as_str())?;
    let target_item = target_item.into_iter().next().ok_or_else(|| Error {
        code: StatusCode::BAD_REQUEST,
        msg: format!(
            "Failed to find target item {} to run a plugin against",
            plugin.target_item_id
        ),
    })?;

    // TODO: that should be not passed to the plugin, plugin should request that data from the DB
    // passing plugin run id is enough.
    let target_item_json = serde_json::to_string(&target_item)?;
    let auth = database_key.create_plugin_auth()?;
    let auth = serde_json::to_string(&auth)?;