fix: init should add only the required file according to the selection (

#3041)

Co-authored-by: Tushar Mathur <[email protected]>

src/cli/tc/helpers.rs

@@ -6,9 +6,9 @@ use crate::core::http::API_URL_PREFIX;
 use crate::core::print_schema;
 use crate::core::rest::{EndpointSet, Unchecked};
 
-pub const FILE_NAME: &str = ".tailcallrc.graphql";
-pub const YML_FILE_NAME: &str = ".graphqlrc.yml";
-pub const JSON_FILE_NAME: &str = ".tailcallrc.schema.json";
+pub const TAILCALL_RC: &str = ".tailcallrc.graphql";
+pub const GRAPHQL_RC: &str = ".graphqlrc.yml";
+pub const TAILCALL_RC_SCHEMA: &str = ".tailcallrc.schema.json";
 
 lazy_static! {
     pub static ref TRACKER: tailcall_tracker::Tracker = tailcall_tracker::Tracker::default();

src/cli/tc/init.rs

@@ -3,7 +3,7 @@ use std::path::Path;
 
 use anyhow::Result;
 
-use super::helpers::{FILE_NAME, JSON_FILE_NAME, YML_FILE_NAME};
+use super::helpers::{GRAPHQL_RC, TAILCALL_RC, TAILCALL_RC_SCHEMA};
 use crate::cli::runtime::{confirm_and_write, create_directory, select_prompt};
 use crate::core::config::{Config, Expr, Field, Resolver, RootSchema, Source};
 use crate::core::merge_right::MergeRight;
@@ -21,23 +21,35 @@ pub(super) async fn init_command(runtime: TargetRuntime, folder_path: &str) -> R
     let tailcallrc = include_str!("../../../generated/.tailcallrc.graphql");
     let tailcallrc_json: &str = include_str!("../../../generated/.tailcallrc.schema.json");
 
-    let file_path = Path::new(folder_path).join(FILE_NAME);
-    let json_file_path = Path::new(folder_path).join(JSON_FILE_NAME);
-    let yml_file_path = Path::new(folder_path).join(YML_FILE_NAME);
-
-    confirm_and_write(
-        runtime.clone(),
-        &file_path.display().to_string(),
-        tailcallrc.as_bytes(),
-    )
-    .await?;
-    confirm_and_write(
-        runtime.clone(),
-        &json_file_path.display().to_string(),
-        tailcallrc_json.as_bytes(),
-    )
-    .await?;
-    confirm_and_write_yml(runtime.clone(), &yml_file_path).await?;
+    let tailcall_rc = Path::new(folder_path).join(TAILCALL_RC);
+    let tailcall_rc_schema = Path::new(folder_path).join(TAILCALL_RC_SCHEMA);
+    let graphql_rc = Path::new(folder_path).join(GRAPHQL_RC);
+
+    match selection {
+        Source::GraphQL => {
+            // .tailcallrc.graphql
+            confirm_and_write(
+                runtime.clone(),
+                &tailcall_rc.display().to_string(),
+                tailcallrc.as_bytes(),
+            )
+            .await?;
+
+            // .graphqlrc.yml
+            confirm_and_write_yml(runtime.clone(), &graphql_rc).await?;
+        }
+
+        Source::Json | Source::Yml => {
+            // .tailcallrc.schema.json
+            confirm_and_write(
+                runtime.clone(),
+                &tailcall_rc_schema.display().to_string(),
+                tailcallrc_json.as_bytes(),
+            )
+            .await?;
+        }
+    }
+
     create_main(runtime.clone(), folder_path, selection).await?;
 
     Ok(())