System Initiative

use serde::{ Deserialize, Serialize, }; use telemetry::prelude::*; use thiserror::Error; use crate::{ DalContext, Prop, PropId, SchemaVariant, SchemaVariantError, SchemaVariantId, prop::{ PropError, PropPath, }, property_editor::schema::PropertyEditorPropWidgetKind, }; #[allow(missing_docs)] #[remain::sorted] #[derive(Error, Debug)] pub enum SecretDefinitionViewError { #[error("prop error: {0}")] Prop(#[from] PropError), #[error("schema variant error: {0}")] SchemaVariant(#[from] SchemaVariantError), } type SecretDefinitionViewResult<T> = Result<T, SecretDefinitionViewError>; #[derive(Deserialize, Serialize, Debug)] #[serde(rename_all(serialize = "camelCase"))] pub struct SecretDefinitionView { pub secret_definition: String, pub form_data: Vec<SecretFormDataView>, } #[derive(Deserialize, Serialize, Debug)] #[serde(rename_all(serialize = "camelCase"))] pub struct SecretFormDataView { pub name: String, pub kind: String, pub widget_kind: PropertyEditorPropWidgetKind, } impl SecretDefinitionView { /// Assembles [`views`](SecretDefinitionView) for all secret definitions in the /// [`snapshot`](crate::WorkspaceSnapshot). #[instrument(level = "debug", name = "list_secret_definition_views", skip_all)] pub async fn list(ctx: &DalContext) -> SecretDefinitionViewResult<Vec<Self>> { let schema_variant_ids = SchemaVariant::list_default_ids(ctx).await?; let secret_definition_path = PropPath::new(["root", "secret_definition"]); let mut views = Vec::new(); for schema_variant_id in schema_variant_ids { let maybe_secret_definition_prop_id = Prop::find_prop_id_by_path_opt(ctx, schema_variant_id, &secret_definition_path) .await?; // We have found a schema variant with a secret definition! if let Some(secret_definition_prop_id) = maybe_secret_definition_prop_id { let view = Self::assemble(ctx, schema_variant_id, secret_definition_prop_id).await?; views.push(view); } } Ok(views) } async fn assemble( ctx: &DalContext, schema_variant_id: SchemaVariantId, secret_definition_prop_id: PropId, ) -> SecretDefinitionViewResult<Self> { // Now, find all the fields of the definition. let field_props = Prop::direct_child_props_ordered(ctx, secret_definition_prop_id).await?; // Assemble the form data views. let mut form_data_views = Vec::new(); for field_prop in field_props { form_data_views.push(SecretFormDataView { name: field_prop.name, kind: field_prop.kind.to_string(), widget_kind: PropertyEditorPropWidgetKind::new( field_prop.widget_kind, field_prop.widget_options, ), }); } // Get the secret output socket corresponding to the definition. There should only be one // output socket as secret defining schema variants are required to have one and only one. let secret_output_socket = SchemaVariant::find_output_socket_for_secret_defining_id(ctx, schema_variant_id) .await?; Ok(Self { secret_definition: secret_output_socket.name().to_owned(), form_data: form_data_views, }) } }