feat: expose a Datetime EthereumSqlTypeWrapper

Author: joshstevens19

core/Cargo.toml

@@ -15,7 +15,7 @@ mockito = "0.30"
 ethers = { version = "2.0", features = ["rustls", "openssl"] }
 ethers-solc = "2.0.14"
 tokio = { version = "1", features = ["full"] }
-tokio-postgres = { version="0.7", features=["with-uuid-1"] }
+tokio-postgres = { version="0.7", features=["with-uuid-1", "with-chrono-0_4"] }
 bb8 = "0.8.3"
 bb8-postgres = "0.8.1"
 serde = "1.0"
@@ -40,7 +40,7 @@ ctrlc = "3.4.4"
 percent-encoding = "2.3.1"
 tracing = "0.1"
 tracing-subscriber = { version = "0.3", features = ["env-filter", "fmt", "time"] }
-chrono = "0.4.38"
+chrono = { version = "0.4", features = ["serde"] }
 log = "0.4.20"
 colored = "2.0"
 hex = "0.4.3"

core/src/database/postgres/client.rs

@@ -317,7 +317,7 @@ impl PostgresClient {
     }
 
     /// This will use COPY to insert the data into the database
-    /// or use the normal bulk inserts if the data is not large enough to 
+    /// or use the normal bulk inserts if the data is not large enough to
     /// need a COPY. This uses `bulk_insert` and `bulk_insert_via_copy` under the hood
     pub async fn insert_bulk(
         &self,

core/src/database/postgres/sql_type_wrapper.rs

@@ -1,6 +1,7 @@
 use std::str::FromStr;
 
 use bytes::BytesMut;
+use chrono::{DateTime, Utc};
 use ethers::{
     abi::{Int, LogParam, Token},
     addressbook::Address,
@@ -79,6 +80,8 @@ pub enum EthereumSqlTypeWrapper {
     VecString(Vec<String>),
     Bytes(Bytes),
     VecBytes(Vec<Bytes>),
+
+    DateTime(DateTime<Utc>),
 }
 
 impl EthereumSqlTypeWrapper {
@@ -147,6 +150,8 @@ impl EthereumSqlTypeWrapper {
             EthereumSqlTypeWrapper::VecString(_) => "VecString",
             EthereumSqlTypeWrapper::Bytes(_) => "Bytes",
             EthereumSqlTypeWrapper::VecBytes(_) => "VecBytes",
+
+            EthereumSqlTypeWrapper::DateTime(_) => "DateTime",
         }
     }
 
@@ -215,6 +220,9 @@ impl EthereumSqlTypeWrapper {
             EthereumSqlTypeWrapper::VecString(_) => PgType::TEXT_ARRAY,
             EthereumSqlTypeWrapper::Bytes(_) => PgType::BYTEA,
             EthereumSqlTypeWrapper::VecBytes(_) => PgType::BYTEA_ARRAY,
+
+            // DateTime
+            EthereumSqlTypeWrapper::DateTime(_) => PgType::TIMESTAMPTZ,
         }
     }
 }
@@ -472,6 +480,7 @@ impl ToSql for EthereumSqlTypeWrapper {
                     int_values.to_sql(ty, out)
                 }
             }
+            EthereumSqlTypeWrapper::DateTime(value) => value.to_sql(ty, out),
         }
     }
 
@@ -1133,6 +1142,9 @@ pub fn map_ethereum_wrapper_to_json(
                     EthereumSqlTypeWrapper::VecBytes(bytes) => {
                         json!(bytes.iter().map(hex::encode).collect::<Vec<_>>())
                     }
+                    EthereumSqlTypeWrapper::DateTime(date_time) => {
+                        json!(date_time.to_rfc3339())
+                    }
                 };
                 result.insert(abi_input.name.clone(), value);
                 wrappers_index_processed.push(current_wrapper_index);

core/src/manifest/yaml.rs

@@ -266,7 +266,7 @@ pub fn read_manifest(file_path: &PathBuf) -> Result<Manifest, ReadManifestError>
     let manifest_before_transform: Manifest = serde_yaml::from_str(&contents)?;
 
     contents = substitute_env_variables(&contents)?;
-
+    
     let mut manifest_after_transform: Manifest = serde_yaml::from_str(&contents)?;
 
     // as we don't want to inject the RPC URL in rust projects in clear text we should change