feat(katana): rollup and dev chain spec

bin/katana/src/cli/init/mod.rs

@@ -6,11 +6,13 @@ use std::sync::Arc;
 use anyhow::{Context, Result};
 use clap::Args;
 use inquire::{Confirm, CustomType, Select};
-use katana_chain_spec::{SettlementLayer, DEV_UNALLOCATED};
+use katana_chain_spec::rollup::FeeContract;
+use katana_chain_spec::{rollup, SettlementLayer};
 use katana_primitives::chain::ChainId;
 use katana_primitives::genesis::allocation::DevAllocationsGenerator;
+use katana_primitives::genesis::constant::DEFAULT_PREFUNDED_ACCOUNT_BALANCE;
 use katana_primitives::genesis::Genesis;
-use katana_primitives::{ContractAddress, Felt};
+use katana_primitives::{felt, ContractAddress, Felt, U256};
 use lazy_static::lazy_static;
 use starknet::accounts::{ExecutionEncoding, SingleOwnerAccount};
 use starknet::core::types::{BlockId, BlockTag};
@@ -40,12 +42,12 @@ impl InitArgs {
             core_contract: input.settlement_contract,
         };
 
-        let mut chain_spec = DEV_UNALLOCATED.clone();
-        chain_spec.genesis = GENESIS.clone();
-        chain_spec.id = ChainId::parse(&input.id)?;
-        chain_spec.settlement = Some(settlement);
+        let id = ChainId::parse(&input.id)?;
+        let genesis = GENESIS.clone();
+        let fee_contract = FeeContract { strk: DEFAULT_APPCHAIN_FEE_TOKEN_ADDRESS.into() };
 
-        katana_chain_spec::file::write(&chain_spec).context("failed to write chain spec file")?;
+        let chain_spec = rollup::ChainSpec { id, genesis, settlement, fee_contract };
+        rollup::file::write(&chain_spec).context("failed to write chain spec file")?;
 
         Ok(())
     }
@@ -181,9 +183,12 @@ struct PromptOutcome {
 lazy_static! {
     static ref GENESIS: Genesis = {
         // master account
-        let accounts = DevAllocationsGenerator::new(1).generate();
+        let accounts = DevAllocationsGenerator::new(1).with_balance(U256::from(DEFAULT_PREFUNDED_ACCOUNT_BALANCE)).generate();
         let mut genesis = Genesis::default();
         genesis.extend_allocations(accounts.into_iter().map(|(k, v)| (k, v.into())));
         genesis
     };
 }
+
+const DEFAULT_APPCHAIN_FEE_TOKEN_ADDRESS: Felt =
+    felt!("0x2e7442625bab778683501c0eadbc1ea17b3535da040a12ac7d281066e915eea");

crates/dojo/test-utils/src/sequencer.rs

@@ -45,7 +45,7 @@ impl TestSequencer {
         let url =
             Url::parse(&format!("http://{}", handle.rpc.addr())).expect("Failed to parse URL");
 
-        let account = handle.node.backend.chain_spec.genesis.accounts().next().unwrap();
+        let account = handle.node.backend.chain_spec.genesis().accounts().next().unwrap();
         let account = TestAccount {
             private_key: Felt::from_bytes_be(&account.1.private_key().unwrap().to_bytes_be()),
             account_address: Felt::from_bytes_be(&account.0.to_bytes_be()),
@@ -81,7 +81,7 @@ impl TestSequencer {
         index: usize,
     ) -> SingleOwnerAccount<JsonRpcClient<HttpTransport>, LocalWallet> {
         let accounts: Vec<_> =
-            self.handle.node.backend.chain_spec.genesis.accounts().collect::<_>();
+            self.handle.node.backend.chain_spec.genesis().accounts().collect::<_>();
 
         let account = accounts[index];
         let private_key = Felt::from_bytes_be(&account.1.private_key().unwrap().to_bytes_be());
@@ -115,7 +115,8 @@ impl TestSequencer {
 
 pub fn get_default_test_config(sequencing: SequencingConfig) -> Config {
     let dev = DevConfig { fee: false, account_validation: true, fixed_gas_prices: None };
-    let mut chain = ChainSpec { id: ChainId::SEPOLIA, ..Default::default() };
+    let mut chain =
+        katana_chain_spec::dev::ChainSpec { id: ChainId::SEPOLIA, ..Default::default() };
     chain.genesis.sequencer_address = *DEFAULT_SEQUENCER_ADDRESS;
 
     let rpc = RpcConfig {
@@ -128,5 +129,5 @@ pub fn get_default_test_config(sequencing: SequencingConfig) -> Config {
         max_proof_keys: Some(100),
     };
 
-    Config { sequencing, rpc, dev, chain: chain.into(), ..Default::default() }
+    Config { sequencing, rpc, dev, chain: ChainSpec::Dev(chain).into(), ..Default::default() }
 }

crates/katana/chain-spec/Cargo.toml

@@ -10,14 +10,15 @@ katana-primitives.workspace = true
 
 alloy-primitives.workspace = true
 anyhow.workspace = true
+dirs = "6.0.0"
 lazy_static.workspace = true
+num-traits.workspace = true
 serde.workspace = true
 serde_json.workspace = true
 starknet.workspace = true
 thiserror.workspace = true
-url.workspace = true
-dirs = "6.0.0"
 toml.workspace = true
+url.workspace = true
 
 [dev-dependencies]
 similar-asserts.workspace = true