Auto-create molecule/projects dataset

Author: sergiimk

src/adapter/graphql/src/mutations/molecule_mut.rs

@@ -68,7 +68,7 @@ impl MoleculeMut {
         }
 
         // Resolve projects dataset
-        let projects_dataset = Molecule::get_projects_dataset(ctx).await?;
+        let projects_dataset = Molecule::get_projects_dataset(ctx, true).await?;
 
         // Check for conflicts
         let query_res = query_svc
@@ -236,7 +236,7 @@ impl MoleculeMut {
         let query_svc = from_catalog_n!(ctx, dyn domain::QueryService);
 
         // Resolve projects dataset
-        let projects_dataset = Molecule::get_projects_dataset(ctx).await?;
+        let projects_dataset = Molecule::get_projects_dataset(ctx, false).await?;
 
         // Query data
         let query_res = query_svc

src/adapter/graphql/src/queries/molecule.rs

@@ -169,15 +169,42 @@ impl Molecule {
         }
     }
 
-    pub async fn get_projects_dataset(ctx: &Context<'_>) -> Result<domain::ResolvedDataset> {
+    pub async fn get_projects_dataset(
+        ctx: &Context<'_>,
+        create_if_not_exist: bool,
+    ) -> Result<domain::ResolvedDataset> {
         let dataset_reg = from_catalog_n!(ctx, dyn domain::DatasetRegistry);
 
-        let projects_dataset = dataset_reg
+        match dataset_reg
             .get_dataset_by_ref(&"molecule/projects".parse().unwrap())
             .await
-            .int_err()?;
-
-        Ok(projects_dataset)
+        {
+            Ok(ds) => Ok(ds),
+            Err(odf::DatasetRefUnresolvedError::NotFound(_)) if create_if_not_exist => {
+                let create_dataset_use_case =
+                    from_catalog_n!(ctx, dyn kamu_datasets::CreateDatasetFromSnapshotUseCase);
+
+                let snapshot = Self::dataset_snapshot_projects(odf::DatasetAlias::new(
+                    None,
+                    odf::DatasetName::new_unchecked("projects"),
+                ));
+
+                let create_res = create_dataset_use_case
+                    .execute(
+                        snapshot,
+                        kamu_datasets::CreateDatasetUseCaseOptions {
+                            dataset_visibility: odf::DatasetVisibility::Private,
+                        },
+                    )
+                    .await
+                    .int_err()?;
+
+                Ok(dataset_reg
+                    .get_dataset_by_handle(&create_res.dataset_handle)
+                    .await)
+            }
+            Err(err) => Err(err.int_err().into()),
+        }
     }
 }
 
@@ -200,7 +227,7 @@ impl Molecule {
         let query_svc = from_catalog_n!(ctx, dyn domain::QueryService);
 
         // Resolve projects dataset
-        let projects_dataset = Self::get_projects_dataset(ctx).await?;
+        let projects_dataset = Self::get_projects_dataset(ctx, false).await?;
 
         // Query data
         let query_res = query_svc
@@ -251,7 +278,7 @@ impl Molecule {
         let query_svc = from_catalog_n!(ctx, dyn domain::QueryService);
 
         // Resolve projects dataset
-        let projects_dataset = Self::get_projects_dataset(ctx).await?;
+        let projects_dataset = Self::get_projects_dataset(ctx, false).await?;
 
         // Query data
         let query_res = query_svc

src/adapter/graphql/tests/tests/test_gql_custom_molecule.rs

@@ -335,10 +335,7 @@ async fn test_molecule_data_room_operations() {
         .build()
         .await;
 
-    // Setup `projects` dataset
-    harness.create_projects_dataset().await;
-
-    // Create project
+    // Create project (projects dataset is auto-created)
     let res = harness
         .execute_authorized_query(async_graphql::Request::new(CREATE_PROJECT).variables(
             async_graphql::Variables::from_json(json!({
@@ -519,10 +516,7 @@ async fn test_molecule_announcements_operations() {
         .build()
         .await;
 
-    // Setup `projects` dataset
-    harness.create_projects_dataset().await;
-
-    // Create project
+    // Create project (projects dataset is auto-created)
     let res = harness
         .execute_authorized_query(async_graphql::Request::new(CREATE_PROJECT).variables(
             async_graphql::Variables::from_json(json!({