Skip to content

Commit

Permalink
feat: add encoding capability to forc-client for variable encoding (F…
Browse files Browse the repository at this point in the history
…uelLabs#4680)

## Description
closes FuelLabs#4679.

This PR lays down the foundation for variable encoding in forc-client.
Unblocks things like:

1. FuelLabs#3439 
2. FuelLabs#3440 
3. FuelLabs#3446
  • Loading branch information
kayagokalp authored Jul 3, 2023
1 parent aa87c8e commit 2e7c20e
Show file tree
Hide file tree
Showing 4 changed files with 172 additions and 0 deletions.
1 change: 1 addition & 0 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions forc-plugins/forc-client/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ forc-pkg = { version = "0.41.0", path = "../../forc-pkg" }
forc-tracing = { version = "0.41.0", path = "../../forc-tracing" }
forc-tx = { version = "0.41.0", path = "../forc-tx" }
forc-util = { version = "0.41.0", path = "../../forc-util" }
fuel-abi-types = "0.3"
fuel-core-client = { workspace = true }
fuel-crypto = { workspace = true }
fuel-tx = { workspace = true, features = ["builder"] }
Expand Down
169 changes: 169 additions & 0 deletions forc-plugins/forc-client/src/util/encode.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,169 @@
use std::str::FromStr;

use fuel_abi_types::abi::full_program::FullTypeApplication;
use serde::{Deserialize, Deserializer, Serialize};

/// A wrapper around fuels_core::types::Token, which enables serde de/serialization.
#[derive(Debug, PartialEq)]
pub(crate) struct Token(fuels_core::types::Token);

impl FromStr for Token {
type Err = anyhow::Error;

fn from_str(s: &str) -> Result<Self, Self::Err> {
let token = fuels_core::types::Token::from_str(s)?;
Ok(Self(token))
}
}

impl AsRef<fuels_core::types::Token> for Token {
fn as_ref(&self) -> &fuels_core::types::Token {
&self.0
}
}

impl Serialize for Token {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
let token_str = self.as_ref().to_string();
serializer.serialize_str(&token_str)
}
}

impl<'de> Deserialize<'de> for Token {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
let s = String::deserialize(deserializer)?;
FromStr::from_str(&s).map_err(serde::de::Error::custom)
}
}

#[derive(Debug, PartialEq, Eq)]
pub(crate) enum Type {
Unit,
U8,
U16,
U32,
U64,
Bool,
}

impl TryFrom<&FullTypeApplication> for Type {
type Error = anyhow::Error;

fn try_from(value: &FullTypeApplication) -> Result<Self, Self::Error> {
let type_field_string = &value.type_decl.type_field;
Type::from_str(type_field_string)
}
}

impl Token {
/// Generate a new token using provided type information and the value for the argument.
///
/// Generates an error if there is a mismatch between the type information and the provided
/// value for that type.
#[allow(dead_code)]
pub(crate) fn from_type_and_value(arg_type: &Type, value: &str) -> anyhow::Result<Self> {
match arg_type {
Type::Unit => Ok(Token(fuels_core::types::Token::Unit)),
Type::U8 => {
let u8_val = value.parse::<u8>()?;
Ok(Token(fuels_core::types::Token::U8(u8_val)))
}
Type::U16 => {
let u16_val = value.parse::<u16>()?;
Ok(Token(fuels_core::types::Token::U16(u16_val)))
}
Type::U32 => {
let u32_val = value.parse::<u32>()?;
Ok(Token(fuels_core::types::Token::U32(u32_val)))
}
Type::U64 => {
let u64_val = value.parse::<u64>()?;
Ok(Token(fuels_core::types::Token::U64(u64_val)))
}
Type::Bool => {
let bool_val = value.parse::<bool>()?;
Ok(Token(fuels_core::types::Token::Bool(bool_val)))
}
}
}
}

impl FromStr for Type {
type Err = anyhow::Error;

fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"()" => Ok(Type::Unit),
"u8" => Ok(Type::U8),
"u16" => Ok(Type::U16),
"u32" => Ok(Type::U32),
"u64" => Ok(Type::U64),
"bool" => Ok(Type::Bool),
other => anyhow::bail!("{other} type is not supported."),
}
}
}

#[cfg(test)]
mod tests {
use super::*;

#[test]
fn test_token_generation_success() {
let u8_token = Token::from_type_and_value(&Type::U8, "1").unwrap();
let u16_token = Token::from_type_and_value(&Type::U16, "1").unwrap();
let u32_token = Token::from_type_and_value(&Type::U32, "1").unwrap();
let u64_token = Token::from_type_and_value(&Type::U64, "1").unwrap();
let bool_token = Token::from_type_and_value(&Type::Bool, "true").unwrap();

let generated_tokens = [u8_token, u16_token, u32_token, u64_token, bool_token];
let expected_tokens = [
Token(fuels_core::types::Token::U8(1)),
Token(fuels_core::types::Token::U16(1)),
Token(fuels_core::types::Token::U32(1)),
Token(fuels_core::types::Token::U64(1)),
Token(fuels_core::types::Token::Bool(true)),
];

assert_eq!(generated_tokens, expected_tokens)
}

#[test]
#[should_panic]
fn test_token_generation_fail_type_mismatch() {
Token::from_type_and_value(&Type::U8, "false").unwrap();
}

#[test]
fn test_type_generation_success() {
let possible_type_list = ["()", "u8", "u16", "u32", "u64", "bool"];
let types = possible_type_list
.iter()
.map(|type_str| Type::from_str(type_str))
.collect::<anyhow::Result<Vec<_>>>()
.unwrap();

let expected_types = vec![
Type::Unit,
Type::U8,
Type::U16,
Type::U32,
Type::U64,
Type::Bool,
];
assert_eq!(types, expected_types)
}

#[test]
#[should_panic(expected = "u2 type is not supported.")]
fn test_type_generation_fail_invalid_type() {
let invalid_type_str = "u2";
Type::from_str(invalid_type_str).unwrap();
}
}
1 change: 1 addition & 0 deletions forc-plugins/forc-client/src/util/mod.rs
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
pub(crate) mod encode;
pub(crate) mod pkg;
pub(crate) mod tx;

0 comments on commit 2e7c20e

Please sign in to comment.