From e828ef572df90662473f68b8ab6074dc2b3e322f Mon Sep 17 00:00:00 2001
From: Dmitry Zakharov <dzakh.dev@gmail.com>
Date: Fri, 25 Oct 2024 11:51:11 +0400
Subject: [PATCH] Add Mainnet selection in Fuel contract import (#291)

---
 codegenerator/cli/CommandLineHelp.md          | 34 ++++----
 .../cli/src/cli_args/clap_definitions.rs      |  4 +-
 codegenerator/cli/src/cli_args/init_config.rs | 81 +++++++++++++------
 .../cli_args/interactive_init/evm_prompts.rs  |  2 +-
 .../cli_args/interactive_init/fuel_prompts.rs | 16 +++-
 5 files changed, 89 insertions(+), 48 deletions(-)

diff --git a/codegenerator/cli/CommandLineHelp.md b/codegenerator/cli/CommandLineHelp.md
index 0e79a2a50..eac40be45 100644
--- a/codegenerator/cli/CommandLineHelp.md
+++ b/codegenerator/cli/CommandLineHelp.md
@@ -11,9 +11,9 @@ This document contains the help content for the `envio` command-line program.
 * [`envio init contract-import local`↴](#envio-init-contract-import-local)
 * [`envio init template`↴](#envio-init-template)
 * [`envio init fuel`↴](#envio-init-fuel)
-* [`envio init fuel template`↴](#envio-init-fuel-template)
 * [`envio init fuel contract-import`↴](#envio-init-fuel-contract-import)
 * [`envio init fuel contract-import local`↴](#envio-init-fuel-contract-import-local)
+* [`envio init fuel template`↴](#envio-init-fuel-template)
 * [`envio dev`↴](#envio-dev)
 * [`envio stop`↴](#envio-stop)
 * [`envio codegen`↴](#envio-codegen)
@@ -150,23 +150,8 @@ Initialization option for creating Fuel indexer
 
 ###### **Subcommands:**
 
-* `template` — Initialize Fuel indexer from an example template
 * `contract-import` — Initialize Fuel indexer by importing config from a contract for a given chain
-
-
-
-## `envio init fuel template`
-
-Initialize Fuel indexer from an example template
-
-**Usage:** `envio init fuel template [OPTIONS]`
-
-###### **Options:**
-
-* `-t`, `--template <TEMPLATE>` — Name of the template to be used in initialization
-
-  Possible values: `greeter`
-
+* `template` — Initialize Fuel indexer from an example template
 
 
 
@@ -201,6 +186,21 @@ Initialize from a local json ABI file
 
 
 
+## `envio init fuel template`
+
+Initialize Fuel indexer from an example template
+
+**Usage:** `envio init fuel template [OPTIONS]`
+
+###### **Options:**
+
+* `-t`, `--template <TEMPLATE>` — Name of the template to be used in initialization
+
+  Possible values: `greeter`
+
+
+
+
 ## `envio dev`
 
 Development commands for starting, stopping, and restarting the indexer with automatic codegen for any changed files
diff --git a/codegenerator/cli/src/cli_args/clap_definitions.rs b/codegenerator/cli/src/cli_args/clap_definitions.rs
index 789df813b..9407824de 100644
--- a/codegenerator/cli/src/cli_args/clap_definitions.rs
+++ b/codegenerator/cli/src/cli_args/clap_definitions.rs
@@ -293,11 +293,11 @@ pub mod fuel {
 
     #[derive(Subcommand, Debug, EnumIter, Display, EnumString, Clone)]
     pub enum InitFlow {
-        ///Initialize Fuel indexer from an example template
-        Template(TemplateArgs),
         ///Initialize Fuel indexer by importing config from a contract for a given chain
         #[strum(serialize = "Contract Import")]
         ContractImport(ContractImportArgs),
+        ///Initialize Fuel indexer from an example template
+        Template(TemplateArgs),
     }
 
     #[derive(Args, Debug, Default, Clone)]
diff --git a/codegenerator/cli/src/cli_args/init_config.rs b/codegenerator/cli/src/cli_args/init_config.rs
index 1ab38e5ae..7857c2e2d 100644
--- a/codegenerator/cli/src/cli_args/init_config.rs
+++ b/codegenerator/cli/src/cli_args/init_config.rs
@@ -200,13 +200,17 @@ pub mod evm {
 }
 
 pub mod fuel {
+    use std::collections::HashMap;
+
     use clap::ValueEnum;
     use serde::{Deserialize, Serialize};
-    use strum::{Display, EnumIter, EnumString};
+    use strum::{Display, EnumIter, EnumString, IntoEnumIterator};
 
     use crate::{
         config_parsing::human_config::{
-            fuel::{ContractConfig, EcosystemTag, EventConfig, HumanConfig, Network},
+            fuel::{
+                ContractConfig, EcosystemTag, EventConfig, HumanConfig, Network as NetworkConfig,
+            },
             NetworkContract,
         },
         fuel::{abi::FuelAbi, address::Address},
@@ -219,12 +223,19 @@ pub mod fuel {
         Greeter,
     }
 
+    #[derive(Clone, Debug, Display, Eq, Hash, PartialEq, EnumIter)]
+    pub enum Network {
+        Mainnet = 9889,
+        Testnet = 0,
+    }
+
     #[derive(Clone, Debug)]
     pub struct SelectedContract {
         pub name: String,
         pub addresses: Vec<Address>,
         pub abi: FuelAbi,
         pub selected_events: Vec<EventConfig>,
+        pub network: Network,
     }
 
     impl SelectedContract {
@@ -240,6 +251,47 @@ pub mod fuel {
 
     impl ContractImportSelection {
         pub fn to_human_config(self: &Self, init_config: &InitConfig) -> HumanConfig {
+            let mut contracts_by_network: HashMap<Network, Vec<SelectedContract>> = HashMap::new();
+
+            for contract in self.contracts.clone() {
+                match contracts_by_network.get_mut(&contract.network) {
+                    None => {
+                        contracts_by_network.insert(contract.network.clone(), vec![contract]);
+                    }
+                    Some(contracts) => contracts.push(contract),
+                }
+            }
+
+            let mut network_configs = vec![];
+            for network in Network::iter() {
+                match contracts_by_network.get(&network) {
+                    None => (),
+                    Some(contracts) => network_configs.push(NetworkConfig {
+                        id: network as u64,
+                        start_block: 0,
+                        end_block: None,
+                        hyperfuel_config: None,
+                        contracts: contracts
+                            .iter()
+                            .map(|selected_contract| NetworkContract {
+                                name: selected_contract.name.clone(),
+                                address: selected_contract
+                                    .addresses
+                                    .iter()
+                                    .map(|a| a.to_string())
+                                    .collect::<Vec<String>>()
+                                    .into(),
+                                config: Some(ContractConfig {
+                                    abi_file_path: selected_contract.get_vendored_abi_file_path(),
+                                    handler: init_config.language.get_event_handler_directory(),
+                                    events: selected_contract.selected_events.clone(),
+                                }),
+                            })
+                            .collect(),
+                    }),
+                }
+            }
+
             HumanConfig {
                 name: init_config.name.clone(),
                 description: None,
@@ -247,30 +299,7 @@ pub mod fuel {
                 schema: None,
                 contracts: None,
                 raw_events: None,
-                networks: vec![Network {
-                    id: 0,
-                    start_block: 0,
-                    end_block: None,
-                    hyperfuel_config: None,
-                    contracts: self
-                        .contracts
-                        .iter()
-                        .map(|selected_contract| NetworkContract {
-                            name: selected_contract.name.clone(),
-                            address: selected_contract
-                                .addresses
-                                .iter()
-                                .map(|a| a.to_string())
-                                .collect::<Vec<String>>()
-                                .into(),
-                            config: Some(ContractConfig {
-                                abi_file_path: selected_contract.get_vendored_abi_file_path(),
-                                handler: init_config.language.get_event_handler_directory(),
-                                events: selected_contract.selected_events.clone(),
-                            }),
-                        })
-                        .collect(),
-                }],
+                networks: network_configs,
             }
         }
     }
diff --git a/codegenerator/cli/src/cli_args/interactive_init/evm_prompts.rs b/codegenerator/cli/src/cli_args/interactive_init/evm_prompts.rs
index 78b61f09e..3d7d9f5e6 100644
--- a/codegenerator/cli/src/cli_args/interactive_init/evm_prompts.rs
+++ b/codegenerator/cli/src/cli_args/interactive_init/evm_prompts.rs
@@ -214,7 +214,7 @@ fn prompt_for_network_id(
     //Action prompt
     let choose_from_networks = Select::new("Choose network:", options)
         .prompt()
-        .context("Failed during prompt for abi file path")?;
+        .context("Failed during prompt for network")?;
 
     let selected = match choose_from_networks.as_str() {
         //If the user's choice evaluates to the enter network id option, prompt them for
diff --git a/codegenerator/cli/src/cli_args/interactive_init/fuel_prompts.rs b/codegenerator/cli/src/cli_args/interactive_init/fuel_prompts.rs
index 7e0994c59..9f9eadc53 100644
--- a/codegenerator/cli/src/cli_args/interactive_init/fuel_prompts.rs
+++ b/codegenerator/cli/src/cli_args/interactive_init/fuel_prompts.rs
@@ -7,7 +7,7 @@ use crate::{
     },
     config_parsing::human_config::fuel::EventConfig,
     fuel::abi::{FuelAbi, BURN_EVENT_NAME, CALL_EVENT_NAME, MINT_EVENT_NAME, TRANSFER_EVENT_NAME},
-    init_config::fuel::{ContractImportSelection, InitFlow, SelectedContract, Template},
+    init_config::fuel::{ContractImportSelection, InitFlow, Network, SelectedContract, Template},
 };
 use anyhow::{Context, Result};
 use inquire::{validator::Validation, Select};
@@ -92,7 +92,13 @@ fn prompt_event_selection(events: Vec<EventConfig>) -> Result<Vec<EventConfig>>
 
 impl Contract for SelectedContract {
     fn get_network_name(&self) -> Result<String> {
-        Ok("Fuel".to_string())
+        Ok(format!(
+            "Fuel {}",
+            match self.network {
+                Network::Mainnet => "Mainnet",
+                Network::Testnet => "Testnet",
+            }
+        ))
     }
 
     fn get_name(&self) -> String {
@@ -150,6 +156,11 @@ async fn get_contract_import_selection(args: ContractImportArgs) -> Result<Selec
 
     let name = get_contract_name(&local_import_args).context("Failed getting contract name")?;
 
+    let choose_from_networks =
+        Select::new("Choose network:", vec![Network::Mainnet, Network::Testnet])
+            .prompt()
+            .context("Failed during prompt for network")?;
+
     let addresses = vec![prompt_contract_address(None)?];
 
     Ok(SelectedContract {
@@ -157,6 +168,7 @@ async fn get_contract_import_selection(args: ContractImportArgs) -> Result<Selec
         addresses,
         abi,
         selected_events,
+        network: choose_from_networks,
     })
 }