From 09bc08f5660bfeba334e7412a58dfa2bb8a9a7f7 Mon Sep 17 00:00:00 2001
From: Arsenii Kulikov <klkvrr@gmail.com>
Date: Fri, 26 Jul 2024 15:33:21 +0800
Subject: [PATCH] fix: better compatibility with older AST

---
 crates/artifacts/solc/src/ast/mod.rs  | 25 +++++++++++++++++--------
 crates/artifacts/solc/src/ast/util.rs |  1 -
 crates/compilers/src/flatten.rs       |  3 +++
 3 files changed, 20 insertions(+), 9 deletions(-)
 delete mode 100644 crates/artifacts/solc/src/ast/util.rs

diff --git a/crates/artifacts/solc/src/ast/mod.rs b/crates/artifacts/solc/src/ast/mod.rs
index 76b9fd38..31142f7c 100644
--- a/crates/artifacts/solc/src/ast/mod.rs
+++ b/crates/artifacts/solc/src/ast/mod.rs
@@ -15,7 +15,7 @@
 mod macros;
 mod misc;
 pub use misc::*;
-pub mod util;
+pub mod utils;
 pub mod visitor;
 
 /// A low fidelity representation of the AST.
@@ -173,7 +173,7 @@ ast_node!(
         contract_dependencies: Vec<usize>,
         #[serde(rename = "contractKind")]
         kind: ContractKind,
-        documentation: Option<StructuredDocumentation>,
+        documentation: Option<Documentation>,
         fully_implemented: bool,
         linearized_base_contracts: Vec<usize>,
         nodes: Vec<ContractDefinitionPart>,
@@ -526,7 +526,7 @@ ast_node!(
         /// [`VariableDeclaration::mutability()`].
         #[serde(default)]
         state_variable: bool,
-        documentation: Option<StructuredDocumentation>,
+        documentation: Option<Documentation>,
         function_selector: Option<String>, // TODO
         #[serde(default)]
         indexed: bool,
@@ -568,6 +568,13 @@ ast_node!(
     }
 );
 
+#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
+#[serde(untagged)]
+pub enum Documentation {
+    Structured(StructuredDocumentation),
+    Raw(String),
+}
+
 ast_node!(
     /// An override specifier.
     struct OverrideSpecifier {
@@ -670,7 +677,7 @@ ast_node!(
         name: String,
         #[serde(default, with = "serde_helpers::display_from_str_opt")]
         name_location: Option<SourceLocation>,
-        documentation: Option<StructuredDocumentation>,
+        documentation: Option<Documentation>,
         error_selector: Option<String>, // TODO
         parameters: ParameterList,
     }
@@ -684,7 +691,7 @@ ast_node!(
         name_location: Option<SourceLocation>,
         anonymous: bool,
         event_selector: Option<String>, // TODO
-        documentation: Option<StructuredDocumentation>,
+        documentation: Option<Documentation>,
         parameters: ParameterList,
     }
 );
@@ -698,7 +705,7 @@ ast_node!(
         #[serde(default, deserialize_with = "serde_helpers::default_for_null")]
         base_functions: Vec<usize>,
         body: Option<Block>,
-        documentation: Option<StructuredDocumentation>,
+        documentation: Option<Documentation>,
         function_selector: Option<String>, // TODO
         implemented: bool,
         modifiers: Vec<ModifierInvocation>,
@@ -865,9 +872,11 @@ ast_node!(
     struct InlineAssembly {
         documentation: Option<String>,
         #[serde(rename = "AST")]
-        ast: YulBlock,
+        ast: Option<YulBlock>,
+        operations: Option<String>,
         // TODO: We need this camel case for the AST, but pascal case other places in ethers-solc
         //evm_version: EvmVersion,
+        #[serde(deserialize_with = "utils::deserialize_external_assembly_references")]
         external_references: Vec<ExternalInlineAssemblyReference>,
         #[serde(default, deserialize_with = "serde_helpers::default_for_null")]
         flags: Vec<InlineAssemblyFlag>,
@@ -1001,7 +1010,7 @@ ast_node!(
         #[serde(default, deserialize_with = "serde_helpers::default_for_null")]
         base_modifiers: Vec<usize>,
         body: Option<Block>,
-        documentation: Option<StructuredDocumentation>,
+        documentation: Option<Documentation>,
         overrides: Option<OverrideSpecifier>,
         parameters: ParameterList,
         #[serde(default, rename = "virtual")]
diff --git a/crates/artifacts/solc/src/ast/util.rs b/crates/artifacts/solc/src/ast/util.rs
deleted file mode 100644
index 8b137891..00000000
--- a/crates/artifacts/solc/src/ast/util.rs
+++ /dev/null
@@ -1 +0,0 @@
-
diff --git a/crates/compilers/src/flatten.rs b/crates/compilers/src/flatten.rs
index bdc26fdc..b410d861 100644
--- a/crates/compilers/src/flatten.rs
+++ b/crates/compilers/src/flatten.rs
@@ -493,6 +493,9 @@ impl Flattener {
                 });
 
             docs.for_each(|doc| {
+                let Documentation::Structured(doc) = doc else {
+                    return
+                };
                 let src_start = doc.src.start.unwrap();
                 let src_end = src_start + doc.src.length.unwrap();