start-os/core/startos/src/config/mod.rs

use std::collections::BTreeSet;
use std::sync::Arc;
use std::time::Duration;

use clap::Parser;
use color_eyre::eyre::eyre;
use indexmap::{IndexMap, IndexSet};
use itertools::Itertools;
use models::{ErrorKind, OptionExt, PackageId};
use patch_db::value::InternedString;
use patch_db::Value;
use regex::Regex;
use rpc_toolkit::{from_fn_async, Context, Empty, HandlerExt, ParentHandler};
use serde::{Deserialize, Serialize};
use tracing::instrument;
use ts_rs::TS;

use crate::context::{CliContext, RpcContext};
use crate::prelude::*;
use crate::rpc_continuations::Guid;
use crate::util::serde::{HandlerExtSerde, StdinDeserializable};

#[derive(Clone, Debug, Default, Serialize, Deserialize)]
pub struct ConfigSpec(pub IndexMap<InternedString, Value>);

pub mod action;
pub mod util;

use util::NumRange;

use self::action::ConfigRes;

pub type Config = patch_db::value::InOMap<InternedString, Value>;
pub trait TypeOf {
    fn type_of(&self) -> &'static str;
}
impl TypeOf for Value {
    fn type_of(&self) -> &'static str {
        match self {
            Value::Array(_) => "list",
            Value::Bool(_) => "boolean",
            Value::Null => "null",
            Value::Number(_) => "number",
            Value::Object(_) => "object",
            Value::String(_) => "string",
        }
    }
}

#[derive(Debug, thiserror::Error)]
pub enum ConfigurationError {
    #[error("Timeout Error")]
    TimeoutError(#[from] TimeoutError),
    #[error("No Match: {0}")]
    NoMatch(#[from] NoMatchWithPath),
    #[error("System Error: {0}")]
    SystemError(Error),
}
impl From<ConfigurationError> for Error {
    fn from(err: ConfigurationError) -> Self {
        let kind = match &err {
            ConfigurationError::SystemError(e) => e.kind,
            _ => crate::ErrorKind::ConfigGen,
        };
        crate::Error::new(err, kind)
    }
}

#[derive(Clone, Copy, Debug, thiserror::Error)]
#[error("Timeout Error")]
pub struct TimeoutError;

#[derive(Clone, Debug, thiserror::Error)]
pub struct NoMatchWithPath {
    pub path: Vec<InternedString>,
    pub error: MatchError,
}
impl NoMatchWithPath {
    pub fn new(error: MatchError) -> Self {
        NoMatchWithPath {
            path: Vec::new(),
            error,
        }
    }
    pub fn prepend(mut self, seg: InternedString) -> Self {
        self.path.push(seg);
        self
    }
}
impl std::fmt::Display for NoMatchWithPath {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}: {}", self.path.iter().rev().join("."), self.error)
    }
}
impl From<NoMatchWithPath> for Error {
    fn from(e: NoMatchWithPath) -> Self {
        ConfigurationError::from(e).into()
    }
}

#[derive(Clone, Debug, thiserror::Error)]
pub enum MatchError {
    #[error("String {0:?} Does Not Match Pattern {1}")]
    Pattern(Arc<String>, Regex),
    #[error("String {0:?} Is Not In Enum {1:?}")]
    Enum(Arc<String>, IndexSet<String>),
    #[error("Field Is Not Nullable")]
    NotNullable,
    #[error("Length Mismatch: expected {0}, actual: {1}")]
    LengthMismatch(NumRange<usize>, usize),
    #[error("Invalid Type: expected {0}, actual: {1}")]
    InvalidType(&'static str, &'static str),
    #[error("Number Out Of Range: expected {0}, actual: {1}")]
    OutOfRange(NumRange<f64>, f64),
    #[error("Number Is Not Integral: {0}")]
    NonIntegral(f64),
    #[error("Variant {0:?} Is Not In Union {1:?}")]
    Union(Arc<String>, IndexSet<String>),
    #[error("Variant Is Missing Tag {0:?}")]
    MissingTag(InternedString),
    #[error("Property {0:?} Of Variant {1:?} Conflicts With Union Tag")]
    PropertyMatchesUnionTag(InternedString, String),
    #[error("Name of Property {0:?} Conflicts With Map Tag Name")]
    PropertyNameMatchesMapTag(String),
    #[error("Object Key Is Invalid: {0}")]
    InvalidKey(String),
    #[error("Value In List Is Not Unique")]
    ListUniquenessViolation,
}

#[derive(Deserialize, Serialize, Parser, TS)]
#[serde(rename_all = "camelCase")]
#[command(rename_all = "kebab-case")]
pub struct ConfigParams {
    pub id: PackageId,
}

// #[command(subcommands(get, set))]
pub fn config<C: Context>() -> ParentHandler<C, ConfigParams> {
    ParentHandler::new()
        .subcommand(
            "get",
            from_fn_async(get)
                .with_inherited(|ConfigParams { id }, _| id)
                .with_display_serializable()
                .with_call_remote::<CliContext>(),
        )
        .subcommand(
            "set",
            set::<C>().with_inherited(|ConfigParams { id }, _| id),
        )
}

#[instrument(skip_all)]
pub async fn get(ctx: RpcContext, _: Empty, id: PackageId) -> Result<ConfigRes, Error> {
    ctx.services
        .get(&id)
        .await
        .as_ref()
        .or_not_found(lazy_format!("Manager for {id}"))?
        .get_config(Guid::new())
        .await
}

#[derive(Deserialize, Serialize, Parser, TS)]
#[serde(rename_all = "camelCase")]
pub struct SetParams {
    #[arg(long = "timeout")]
    pub timeout: Option<crate::util::serde::Duration>,
    #[command(flatten)]
    #[ts(type = "{ [key: string]: any } | null")]
    pub config: StdinDeserializable<Option<Config>>,
}

// #[command(
//     subcommands(self(set_impl(async, context(RpcContext))), set_dry),
//     display(display_none),
//     metadata(sync_db = true)
// )]
#[instrument(skip_all)]
pub fn set<C: Context>() -> ParentHandler<C, SetParams, PackageId> {
    ParentHandler::new()
        .root_handler(
            from_fn_async(set_impl)
                .with_metadata("sync_db", Value::Bool(true))
                .with_inherited(|set_params, id| (id, set_params))
                .no_display()
                .with_call_remote::<CliContext>(),
        )
        .subcommand(
            "dry",
            from_fn_async(set_dry)
                .with_inherited(|set_params, id| (id, set_params))
                .no_display()
                .with_call_remote::<CliContext>(),
        )
}

pub async fn set_dry(
    ctx: RpcContext,
    _: Empty,
    (
        id,
        SetParams {
            timeout,
            config: StdinDeserializable(config),
        },
    ): (PackageId, SetParams),
) -> Result<BTreeSet<PackageId>, Error> {
    let mut breakages = BTreeSet::new();

    let procedure_id = Guid::new();

    let db = ctx.db.peek().await;
    for dep in db
        .as_public()
        .as_package_data()
        .as_entries()?
        .into_iter()
        .filter_map(
            |(k, v)| match v.as_current_dependencies().contains_key(&id) {
                Ok(true) => Some(Ok(k)),
                Ok(false) => None,
                Err(e) => Some(Err(e)),
            },
        )
    {
        let dep_id = dep?;

        let Some(dependent) = &*ctx.services.get(&dep_id).await else {
            continue;
        };

        if dependent
            .dependency_config(procedure_id.clone(), id.clone(), config.clone())
            .await?
            .is_some()
        {
            breakages.insert(dep_id);
        }
    }

    Ok(breakages)
}

#[derive(Default)]
pub struct ConfigureContext {
    pub timeout: Option<Duration>,
    pub config: Option<Config>,
}

#[instrument(skip_all)]
pub async fn set_impl(
    ctx: RpcContext,
    _: Empty,
    (
        id,
        SetParams {
            timeout,
            config: StdinDeserializable(config),
        },
    ): (PackageId, SetParams),
) -> Result<(), Error> {
    let configure_context = ConfigureContext {
        timeout: timeout.map(|t| *t),
        config,
    };
    ctx.services
        .get(&id)
        .await
        .as_ref()
        .ok_or_else(|| {
            Error::new(
                eyre!("There is no manager running for {id}"),
                ErrorKind::Unknown,
            )
        })?
        .configure(Guid::new(), configure_context)
        .await?;
    Ok(())
}