foundry_cheatcodes_spec/
lib.rs

Help
1//! Cheatcode specification for Foundry.
2
3#![cfg_attr(not(test), warn(unused_crate_dependencies))]
4#![cfg_attr(docsrs, feature(doc_cfg, doc_auto_cfg))]
5
6use serde::{Deserialize, Serialize};
7use std::{borrow::Cow, fmt};
8
9mod cheatcode;
10pub use cheatcode::{Cheatcode, CheatcodeDef, Group, Safety, Status};
11
12mod function;
13pub use function::{Function, Mutability, Visibility};
14
15mod items;
16pub use items::{Enum, EnumVariant, Error, Event, Struct, StructField};
17
18mod vm;
19pub use vm::Vm;
20
21// The `cheatcodes.json` schema.
22/// Foundry cheatcodes. Learn more: <https://book.getfoundry.sh/cheatcodes/>
23#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Serialize, Deserialize)]
24#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
25#[serde(rename_all = "camelCase")]
26pub struct Cheatcodes<'a> {
27    /// Cheatcode errors.
28    #[serde(borrow)]
29    pub errors: Cow<'a, [Error<'a>]>,
30    /// Cheatcode events.
31    #[serde(borrow)]
32    pub events: Cow<'a, [Event<'a>]>,
33    /// Cheatcode enums.
34    #[serde(borrow)]
35    pub enums: Cow<'a, [Enum<'a>]>,
36    /// Cheatcode structs.
37    #[serde(borrow)]
38    pub structs: Cow<'a, [Struct<'a>]>,
39    /// All the cheatcodes.
40    #[serde(borrow)]
41    pub cheatcodes: Cow<'a, [Cheatcode<'a>]>,
42}
43
44impl fmt::Display for Cheatcodes<'_> {
45    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
46        for error in self.errors.iter() {
47            writeln!(f, "{error}")?;
48        }
49        for event in self.events.iter() {
50            writeln!(f, "{event}")?;
51        }
52        for enumm in self.enums.iter() {
53            writeln!(f, "{enumm}")?;
54        }
55        for strukt in self.structs.iter() {
56            writeln!(f, "{strukt}")?;
57        }
58        for cheatcode in self.cheatcodes.iter() {
59            writeln!(f, "{}", cheatcode.func)?;
60        }
61        Ok(())
62    }
63}
64
65impl Default for Cheatcodes<'static> {
66    fn default() -> Self {
67        Self::new()
68    }
69}
70
71impl Cheatcodes<'static> {
72    /// Returns the default cheatcodes.
73    pub fn new() -> Self {
74        Self {
75            // unfortunately technology has not yet advanced to the point where we can get all
76            // items of a certain type in a module, so we have to hardcode them here
77            structs: Cow::Owned(vec![
78                Vm::Log::STRUCT.clone(),
79                Vm::Rpc::STRUCT.clone(),
80                Vm::EthGetLogs::STRUCT.clone(),
81                Vm::DirEntry::STRUCT.clone(),
82                Vm::FsMetadata::STRUCT.clone(),
83                Vm::Wallet::STRUCT.clone(),
84                Vm::FfiResult::STRUCT.clone(),
85                Vm::ChainInfo::STRUCT.clone(),
86                Vm::AccountAccess::STRUCT.clone(),
87                Vm::StorageAccess::STRUCT.clone(),
88                Vm::Gas::STRUCT.clone(),
89                Vm::DebugStep::STRUCT.clone(),
90                Vm::BroadcastTxSummary::STRUCT.clone(),
91                Vm::SignedDelegation::STRUCT.clone(),
92                Vm::PotentialRevert::STRUCT.clone(),
93                Vm::AccessListItem::STRUCT.clone(),
94            ]),
95            enums: Cow::Owned(vec![
96                Vm::CallerMode::ENUM.clone(),
97                Vm::AccountAccessKind::ENUM.clone(),
98                Vm::ForgeContext::ENUM.clone(),
99                Vm::BroadcastTxType::ENUM.clone(),
100            ]),
101            errors: Vm::VM_ERRORS.iter().copied().cloned().collect(),
102            events: Cow::Borrowed(&[]),
103            // events: Vm::VM_EVENTS.iter().copied().cloned().collect(),
104            cheatcodes: Vm::CHEATCODES.iter().copied().cloned().collect(),
105        }
106    }
107}
108
109#[cfg(test)]
110#[expect(clippy::disallowed_macros)]
111mod tests {
112    use super::*;
113    use std::{fs, path::Path};
114
115    const JSON_PATH: &str = concat!(env!("CARGO_MANIFEST_DIR"), "/../assets/cheatcodes.json");
116    #[cfg(feature = "schema")]
117    const SCHEMA_PATH: &str =
118        concat!(env!("CARGO_MANIFEST_DIR"), "/../assets/cheatcodes.schema.json");
119    const IFACE_PATH: &str =
120        concat!(env!("CARGO_MANIFEST_DIR"), "/../../../testdata/cheats/Vm.sol");
121
122    /// Generates the `cheatcodes.json` file contents.
123    fn json_cheatcodes() -> String {
124        serde_json::to_string_pretty(&Cheatcodes::new()).unwrap()
125    }
126
127    /// Generates the [cheatcodes](json_cheatcodes) JSON schema.
128    #[cfg(feature = "schema")]
129    fn json_schema() -> String {
130        serde_json::to_string_pretty(&schemars::schema_for!(Cheatcodes<'_>)).unwrap()
131    }
132
133    fn sol_iface() -> String {
134        let mut cheats = Cheatcodes::new();
135        cheats.errors = Default::default(); // Skip errors to allow <0.8.4.
136        let cheats = cheats.to_string().trim().replace('\n', "\n    ");
137        format!(
138            "\
139// Automatically generated from `foundry-cheatcodes` Vm definitions. Do not modify manually.
140// This interface is just for internal testing purposes. Use `forge-std` instead.
141
142// SPDX-License-Identifier: MIT OR Apache-2.0
143pragma solidity >=0.6.2 <0.9.0;
144pragma experimental ABIEncoderV2;
145
146interface Vm {{
147    {cheats}
148}}
149"
150        )
151    }
152
153    #[test]
154    fn spec_up_to_date() {
155        ensure_file_contents(Path::new(JSON_PATH), &json_cheatcodes());
156    }
157
158    #[test]
159    #[cfg(feature = "schema")]
160    fn schema_up_to_date() {
161        ensure_file_contents(Path::new(SCHEMA_PATH), &json_schema());
162    }
163
164    #[test]
165    fn iface_up_to_date() {
166        ensure_file_contents(Path::new(IFACE_PATH), &sol_iface());
167    }
168
169    /// Checks that the `file` has the specified `contents`. If that is not the
170    /// case, updates the file and then fails the test.
171    fn ensure_file_contents(file: &Path, contents: &str) {
172        if let Ok(old_contents) = fs::read_to_string(file) {
173            if normalize_newlines(&old_contents) == normalize_newlines(contents) {
174                // File is already up to date.
175                return
176            }
177        }
178
179        eprintln!("\n\x1b[31;1merror\x1b[0m: {} was not up-to-date, updating\n", file.display());
180        if std::env::var("CI").is_ok() {
181            eprintln!("    NOTE: run `cargo cheats` locally and commit the updated files\n");
182        }
183        if let Some(parent) = file.parent() {
184            let _ = fs::create_dir_all(parent);
185        }
186        fs::write(file, contents).unwrap();
187        panic!("some file was not up to date and has been updated, simply re-run the tests");
188    }
189
190    fn normalize_newlines(s: &str) -> String {
191        s.replace("\r\n", "\n")
192    }
193}
foundry_cheatcodes_spec/lib.rs

foundry_cheatcodes_spec/
lib.rs
