From 16f0e0462d080d6edbbbb101fd138e56bb7d3d21 Mon Sep 17 00:00:00 2001
From: Arsenii Kulikov <klkvrr@gmail.com>
Date: Wed, 31 Jul 2024 19:12:19 +0800
Subject: [PATCH] feat: sanitize EVM version for vyper (#181)

ref https://t.me/foundry_support/55572

Vyper is a bit weird about EVM versions support, e.g. 0.4 does not
support versions older than London and 0.3 never supported London at
all, but this should be enough to cover basic scenarios for 0.3 and 0.4
users
---
 crates/artifacts/vyper/src/settings.rs | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/crates/artifacts/vyper/src/settings.rs b/crates/artifacts/vyper/src/settings.rs
index 6d789f9d..52a2bb08 100644
--- a/crates/artifacts/vyper/src/settings.rs
+++ b/crates/artifacts/vyper/src/settings.rs
@@ -9,6 +9,10 @@ use std::{
 };
 
 pub const VYPER_SEARCH_PATHS: Version = Version::new(0, 4, 0);
+pub const VYPER_BERLIN: Version = Version::new(0, 3, 0);
+pub const VYPER_PARIS: Version = Version::new(0, 3, 7);
+pub const VYPER_SHANGHAI: Version = Version::new(0, 3, 8);
+pub const VYPER_CANCUN: Version = Version::new(0, 3, 8);
 
 #[derive(Clone, Copy, Debug, PartialEq, Eq, Serialize, Deserialize)]
 #[serde(rename_all = "lowercase")]
@@ -79,6 +83,7 @@ impl VyperSettings {
         }
 
         self.sanitize_output_selection();
+        self.normalize_evm_version(version);
     }
 
     /// Sanitize the settings based on the compiler version.
@@ -86,4 +91,21 @@ impl VyperSettings {
         self.sanitize(version);
         self
     }
+
+    /// Adjusts the EVM version based on the compiler version.
+    pub fn normalize_evm_version(&mut self, version: &Version) {
+        if let Some(evm_version) = &mut self.evm_version {
+            *evm_version = if *evm_version >= EvmVersion::Cancun && *version >= VYPER_CANCUN {
+                EvmVersion::Cancun
+            } else if *evm_version >= EvmVersion::Shanghai && *version >= VYPER_SHANGHAI {
+                EvmVersion::Shanghai
+            } else if *evm_version >= EvmVersion::Paris && *version >= VYPER_PARIS {
+                EvmVersion::Paris
+            } else if *evm_version >= EvmVersion::Berlin && *version >= VYPER_BERLIN {
+                EvmVersion::Berlin
+            } else {
+                *evm_version
+            };
+        }
+    }
 }