From 5778bf947538c3c0b42372e80037afc5904b30cc Mon Sep 17 00:00:00 2001
From: yuvrajjsingh0 <yuvraj.singh@juspay.in>
Date: Thu, 30 Jan 2025 23:44:19 +0530
Subject: [PATCH] feat: Create Experiments From Overrides

---
 .../frontend/src/components/context_card.rs   | 13 ++++
 .../src/components/experiment_form.rs         | 19 +++--
 crates/frontend/src/pages/context_override.rs | 71 ++++++++++++++++++-
 crates/frontend/src/pages/experiment.rs       |  1 +
 crates/frontend/src/pages/experiment_list.rs  |  1 +
 5 files changed, 98 insertions(+), 7 deletions(-)

diff --git a/crates/frontend/src/components/context_card.rs b/crates/frontend/src/components/context_card.rs
index 7094e1e12..a37e51426 100644
--- a/crates/frontend/src/components/context_card.rs
+++ b/crates/frontend/src/components/context_card.rs
@@ -15,6 +15,10 @@ pub fn context_card(
     context: Context,
     overrides: Map<String, Value>,
     #[prop(default = true)] show_actions: bool,
+    #[prop(default=Callback::new(|_| {}))] handle_create_experiment: Callback<
+        (Context, Map<String, Value>),
+        (),
+    >,
     #[prop(default=Callback::new(|_| {}))] handle_edit: Callback<
         (Context, Map<String, Value>),
         (),
@@ -65,6 +69,15 @@ pub fn context_card(
                 <Show when=move || actions_supported>
                     <div class="h-fit text-right space-x-4">
                         <Show when=move || !edit_unsupported>
+                            <button
+                                class="btn-purple font-medium rounded-lg text-xs px-5 py-2 text-center me-1.5 mb-1.5 drawer-button"
+                                on:click=move |_| {
+                                    handle_create_experiment.call((context.get_value(), overrides.get_value()));
+                                }
+                            >
+                                Create Experiment
+                                <i class="ri-edit-2-line ml-2"></i>
+                            </button>
                             <i
                                 class="ri-pencil-line ri-lg text-blue-500 cursor-pointer"
                                 on:click=move |_| {
diff --git a/crates/frontend/src/components/experiment_form.rs b/crates/frontend/src/components/experiment_form.rs
index c4744ce9f..87a1429c1 100644
--- a/crates/frontend/src/components/experiment_form.rs
+++ b/crates/frontend/src/components/experiment_form.rs
@@ -2,6 +2,7 @@ pub mod types;
 pub mod utils;
 
 use leptos::*;
+use serde_json::Value;
 use superposition_types::{
     database::models::experimentation::VariantType,
     database::{models::cac::DefaultConfig, types::DimensionWithMandatory},
@@ -20,14 +21,16 @@ use crate::{
 
 use crate::logic::Conditions;
 
-fn default_variants_for_form() -> Vec<(String, VariantFormT)> {
+fn default_variants_for_form(
+    overrides: Vec<(String, Value)>,
+) -> Vec<(String, VariantFormT)> {
     vec![
         (
             "control-variant".to_string(),
             VariantFormT {
                 id: "control".to_string(),
                 variant_type: VariantType::CONTROL,
-                overrides: vec![],
+                overrides: overrides.clone(),
             },
         ),
         (
@@ -35,15 +38,18 @@ fn default_variants_for_form() -> Vec<(String, VariantFormT)> {
             VariantFormT {
                 id: "experimental".to_string(),
                 variant_type: VariantType::EXPERIMENTAL,
-                overrides: vec![],
+                overrides: overrides.clone(),
             },
         ),
     ]
 }
 
-fn get_init_state(variants: &VariantFormTs) -> Vec<(String, VariantFormT)> {
+fn get_init_state(
+    variants: &[VariantFormT],
+    overrides: Vec<(String, Value)>,
+) -> Vec<(String, VariantFormT)> {
     let init_variants = if variants.is_empty() {
-        default_variants_for_form()
+        default_variants_for_form(overrides)
     } else {
         variants
             .iter()
@@ -61,6 +67,7 @@ pub fn experiment_form<NF>(
     name: String,
     context: Conditions,
     variants: VariantFormTs,
+    overrides: Vec<(String, Value)>,
     handle_submit: NF,
     default_config: Vec<DefaultConfig>,
     dimensions: Vec<DimensionWithMandatory>,
@@ -70,7 +77,7 @@ pub fn experiment_form<NF>(
 where
     NF: Fn() + 'static + Clone,
 {
-    let init_variants = get_init_state(&variants);
+    let init_variants = get_init_state(&variants, overrides);
     let default_config = StoredValue::new(default_config);
     let tenant_rws = use_context::<RwSignal<Tenant>>().unwrap();
     let org_rws = use_context::<RwSignal<OrganisationId>>().unwrap();
diff --git a/crates/frontend/src/pages/context_override.rs b/crates/frontend/src/pages/context_override.rs
index f44f8acba..a4764c8a0 100644
--- a/crates/frontend/src/pages/context_override.rs
+++ b/crates/frontend/src/pages/context_override.rs
@@ -8,12 +8,12 @@ use superposition_types::{
     Config, Context,
 };
 
-use crate::components::button::Button;
 use crate::components::context_card::ContextCard;
 use crate::components::context_form::utils::{create_context, update_context};
 use crate::components::context_form::ContextForm;
 use crate::components::delete_modal::DeleteModal;
 use crate::components::drawer::{close_drawer, open_drawer, Drawer, DrawerBtn};
+use crate::components::experiment_form::ExperimentForm;
 use crate::components::override_form::OverrideForm;
 use crate::components::skeleton::{Skeleton, SkeletonVariant};
 use crate::logic::{Condition, Conditions};
@@ -25,6 +25,7 @@ use crate::{
     types::{OrganisationId, Tenant},
 };
 use crate::{components::alert::AlertType, schema::SchemaType};
+use crate::{components::button::Button, types::VariantFormTs};
 
 #[derive(Clone, Debug, Default)]
 pub struct Data {
@@ -277,6 +278,32 @@ pub fn context_override() -> impl IntoView {
             }
         };
     });
+    let handle_submit_experiment_form = move || {
+        page_resource.refetch();
+        close_drawer("create_exp_drawer");
+    };
+
+    let handle_create_experiment =
+        Callback::new(move |data: (Context, Map<String, Value>)| {
+            let (context, overrides) = data;
+
+            match Conditions::from_context_json(&context.condition.into()) {
+                Ok(conditions) => {
+                    selected_context_ws.set(Some(Data {
+                        context: conditions,
+                        overrides: overrides
+                            .into_iter()
+                            .collect::<Vec<(String, Value)>>(),
+                    }));
+
+                    open_drawer("create_exp_drawer");
+                }
+                Err(e) => {
+                    logging::error!("Error parsing context: {}", e);
+                    enqueue_alert(e.to_string(), AlertType::Error, 5000);
+                }
+            };
+        });
 
     let on_context_clone = Callback::new(move |data: (Context, Map<String, Value>)| {
         let (context, overrides) = data;
@@ -342,6 +369,47 @@ pub fn context_override() -> impl IntoView {
             }>
                 <div class="space-y-6">
 
+                    {move || {
+                        let PageResource { config: _, dimensions, default_config } = page_resource
+                            .get()
+                            .unwrap_or_default();
+                        let data = selected_context_rs.get();
+                        view! {
+                            <Drawer
+                                id="create_exp_drawer".to_string()
+                                header="Create Experiment"
+                                handle_close=move || {
+                                    close_drawer("create_exp_drawer");
+                                    selected_context_ws.set(None);
+                                }
+                            >
+
+                                <EditorProvider>
+                                    {match data {
+                                        Some(data) => {
+                                            view! {
+                                                <ExperimentForm
+                                                    name="".to_string()
+                                                    context=data.context
+                                                    variants=VariantFormTs::default()
+                                                    overrides=data.overrides
+                                                    dimensions=dimensions
+                                                    default_config=default_config
+                                                    handle_submit=handle_submit_experiment_form
+                                                />
+                                            }
+                                                .into_view()
+                                        }
+                                        None => {
+                                            view! {}.into_view()
+                                        }
+                                    }}
+
+                                </EditorProvider>
+                            </Drawer>
+                        }
+                    }}
+
                     {move || {
                         let PageResource { config: _, dimensions, default_config } = page_resource
                             .get()
@@ -450,6 +518,7 @@ pub fn context_override() -> impl IntoView {
                                             <ContextCard
                                                 context=context
                                                 overrides=overrides
+                                                handle_create_experiment=handle_create_experiment
                                                 handle_edit=on_context_edit
                                                 handle_clone=on_context_clone
                                                 handle_delete=on_context_delete
diff --git a/crates/frontend/src/pages/experiment.rs b/crates/frontend/src/pages/experiment.rs
index 085cfb4b1..84ae6f128 100644
--- a/crates/frontend/src/pages/experiment.rs
+++ b/crates/frontend/src/pages/experiment.rs
@@ -163,6 +163,7 @@ pub fn experiment_page() -> impl IntoView {
                                         id=experiment.id
                                         name=experiment_ef.name
                                         context=experiment_ef.context.clone()
+                                        overrides=vec![]
                                         variants=FromIterator::from_iter(experiment_ef.variants)
                                         default_config=default_config
                                         dimensions=dimensions
diff --git a/crates/frontend/src/pages/experiment_list.rs b/crates/frontend/src/pages/experiment_list.rs
index bb1faccf5..ae90beead 100644
--- a/crates/frontend/src/pages/experiment_list.rs
+++ b/crates/frontend/src/pages/experiment_list.rs
@@ -233,6 +233,7 @@ pub fn experiment_list() -> impl IntoView {
                                     name="".to_string()
                                     context=Conditions::default()
                                     variants=VariantFormTs::default()
+                                    overrides=vec![]
                                     dimensions=dim.clone()
                                     default_config=def_conf.clone()
                                     handle_submit=handle_submit_experiment_form