From 0e5f4898b5689173f241b7fc1f29381d31438f3b Mon Sep 17 00:00:00 2001
From: clearloop <26088946+clearloop@users.noreply.github.com>
Date: Thu, 22 Feb 2024 12:07:32 +0100
Subject: [PATCH] feat(alpha): add alpha program (#14)

* feat(alpha): alpha program

* fix(alpha): build problems

* feat(prog): state queries
---
 Cargo.lock             | 28 +++++++++++++++++
 Cargo.toml             |  3 +-
 alpha/Cargo.toml       | 16 ++++++++++
 alpha/build.rs         |  5 ++++
 alpha/io/Cargo.toml    |  9 ++++++
 alpha/io/src/lib.rs    | 68 ++++++++++++++++++++++++++++++++++++++++++
 alpha/src/lib.rs       | 30 +++++++++++++++++++
 alpha/state/Cargo.toml | 13 ++++++++
 alpha/state/build.rs   |  3 ++
 alpha/state/src/lib.rs | 22 ++++++++++++++
 alpha/tests/main.rs    | 37 +++++++++++++++++++++++
 tests/tests.rs         |  1 -
 12 files changed, 233 insertions(+), 2 deletions(-)
 create mode 100644 alpha/Cargo.toml
 create mode 100644 alpha/build.rs
 create mode 100644 alpha/io/Cargo.toml
 create mode 100644 alpha/io/src/lib.rs
 create mode 100644 alpha/src/lib.rs
 create mode 100644 alpha/state/Cargo.toml
 create mode 100644 alpha/state/build.rs
 create mode 100644 alpha/state/src/lib.rs
 create mode 100644 alpha/tests/main.rs

diff --git a/Cargo.lock b/Cargo.lock
index 2b65df7..4aca9b9 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -119,6 +119,24 @@ version = "0.2.16"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "0942ffc6dcaadf03badf6e6a2d0228460359d5e34b57ccdc720b7382dfbd5ec5"
 
+[[package]]
+name = "alpha-io"
+version = "0.1.0"
+dependencies = [
+ "gmeta",
+ "gstd",
+]
+
+[[package]]
+name = "alpha-state"
+version = "0.1.0"
+dependencies = [
+ "alpha-io",
+ "gear-wasm-builder",
+ "gmeta",
+ "gstd",
+]
+
 [[package]]
 name = "android-tzdata"
 version = "0.1.1"
@@ -6262,6 +6280,16 @@ version = "0.1.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "830b7e5d4d90034032940e4ace0d9a9a057e7a45cd94e6c007832e39edb82f6d"
 
+[[package]]
+name = "vara-go-alpha"
+version = "0.1.0"
+dependencies = [
+ "alpha-io",
+ "gear-wasm-builder",
+ "gstd",
+ "gtest",
+]
+
 [[package]]
 name = "version_check"
 version = "0.9.4"
diff --git a/Cargo.toml b/Cargo.toml
index 6ec346b..93e5095 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -28,7 +28,7 @@ tokio.workspace = true
 # It's necessary to include all metawasm crates in the workspace section, otherwise they'll be
 # ignored by Cargo and won't be built.
 [workspace]
-members = ["state", "xtask", "domain", "domain/state"]
+members = ["state", "xtask", "domain", "domain/state", "alpha", "alpha/state", "alpha/io"]
 
 [workspace.dependencies]
 gstd = "1.1.1"
@@ -37,6 +37,7 @@ gear-wasm-builder = "1.1.1"
 gtest = "1.1.1"
 gclient = "1.1.1"
 template-io.path = "io"
+alpha-io.path = "alpha/io"
 tokio = "1"
 xshell = "0.2"
 anyhow = "1"
diff --git a/alpha/Cargo.toml b/alpha/Cargo.toml
new file mode 100644
index 0000000..b022e96
--- /dev/null
+++ b/alpha/Cargo.toml
@@ -0,0 +1,16 @@
+[package]
+name = "vara-go-alpha"
+version.workspace = true
+edition.workspace = true
+publish.workspace = true
+
+[dependencies]
+gstd.workspace = true
+alpha-io.workspace = true
+
+[dev-dependencies]
+gtest.workspace = true
+
+[build-dependencies]
+alpha-io.workspace = true
+gear-wasm-builder.workspace = true
diff --git a/alpha/build.rs b/alpha/build.rs
new file mode 100644
index 0000000..00e834f
--- /dev/null
+++ b/alpha/build.rs
@@ -0,0 +1,5 @@
+use alpha_io::ContractMetadata;
+
+fn main() {
+    gear_wasm_builder::build_with_metadata::<ContractMetadata>();
+}
diff --git a/alpha/io/Cargo.toml b/alpha/io/Cargo.toml
new file mode 100644
index 0000000..c57b0a2
--- /dev/null
+++ b/alpha/io/Cargo.toml
@@ -0,0 +1,9 @@
+[package]
+name = "alpha-io"
+version.workspace = true
+edition.workspace = true
+publish.workspace = true
+
+[dependencies]
+gmeta.workspace = true
+gstd.workspace = true
diff --git a/alpha/io/src/lib.rs b/alpha/io/src/lib.rs
new file mode 100644
index 0000000..08ef60c
--- /dev/null
+++ b/alpha/io/src/lib.rs
@@ -0,0 +1,68 @@
+#![no_std]
+
+use gmeta::{In, Metadata, Out};
+use gstd::*;
+
+/// The contract metadata. Used by frontend apps & for describing the types of messages that can be
+/// sent in contract's entry points. See also [`Metadata`].
+pub struct ContractMetadata;
+
+pub type State = Vec<Resource>;
+
+#[derive(Encode, Decode, TypeInfo, PartialEq, Eq, Debug, Clone)]
+#[codec(crate = gstd::codec)]
+#[scale_info(crate = gstd::scale_info)]
+pub enum Label {
+    Apple,
+    Banana,
+    Football,
+    Vara,
+}
+
+impl AsRef<str> for Label {
+    fn as_ref(&self) -> &str {
+        match self {
+            Label::Apple => "apple",
+            Label::Banana => "banana",
+            Label::Football => "football",
+            Label::Vara => "vara",
+        }
+    }
+}
+
+impl Label {
+    pub fn list() -> Vec<Label> {
+        vec![Label::Apple, Label::Banana, Label::Football, Label::Vara]
+    }
+}
+
+#[derive(Encode, Decode, TypeInfo, PartialEq, Eq, Debug, Clone)]
+#[codec(crate = gstd::codec)]
+#[scale_info(crate = gstd::scale_info)]
+pub struct Resource {
+    pub labels: Vec<Label>,
+    pub domain: String,
+    pub description: String,
+    pub html: String,
+    pub styles: String,
+}
+
+/// `()` means the contract doesn't process & reply messages at the above written entry point or
+/// doesn't implement it.
+impl Metadata for ContractMetadata {
+    /// I/O types for the `init()` entry point.
+    type Init = In<()>;
+    /// I/O types for the `handle()` entry point.
+    type Handle = In<Resource>;
+    /// Types for miscellaneous scenarios.
+    type Others = ();
+    /// The input type for the `handle_reply()` entry point.
+    type Reply = ();
+    /// The output type for the `handle_signal()` entry point.
+    type Signal = ();
+    /// I/O types for the `state()` entry point.
+    ///
+    /// You can also specify just an output ([`Out`]) or input ([`In`]) type, if both
+    /// ([`In`]) are expected like here.
+    type State = Out<State>;
+}
diff --git a/alpha/src/lib.rs b/alpha/src/lib.rs
new file mode 100644
index 0000000..10b00e7
--- /dev/null
+++ b/alpha/src/lib.rs
@@ -0,0 +1,30 @@
+#![no_std]
+
+use alpha_io::{Resource, State};
+use gstd::{msg, prelude::*};
+
+/// The resource of the domain.
+static mut STATE: Option<State> = None;
+
+// do nothing atm.
+#[no_mangle]
+extern fn init() {
+    unsafe {
+        STATE = Some(vec![]);
+    }
+}
+
+// The `handle()` entry point.
+#[no_mangle]
+extern fn handle() {
+    let res = msg::load::<Resource>().expect("Failed to load handle input");
+    let state = unsafe { STATE.as_mut().expect("State isn't initialized") };
+    state.push(res);
+}
+
+// The `state()` entry point.
+#[no_mangle]
+extern fn state() {
+    let state = unsafe { STATE.take().expect("State isn't initialized") };
+    msg::reply(state, 0).expect("Failed to reply from `state()`");
+}
diff --git a/alpha/state/Cargo.toml b/alpha/state/Cargo.toml
new file mode 100644
index 0000000..259446b
--- /dev/null
+++ b/alpha/state/Cargo.toml
@@ -0,0 +1,13 @@
+[package]
+name = "alpha-state"
+version.workspace = true
+edition.workspace = true
+publish.workspace = true
+
+[dependencies]
+gstd.workspace = true
+gmeta = { workspace = true, features = ["codegen"] }
+alpha-io.workspace = true
+
+[build-dependencies]
+gear-wasm-builder = { workspace = true, features = ["metawasm"] }
diff --git a/alpha/state/build.rs b/alpha/state/build.rs
new file mode 100644
index 0000000..b4c4a44
--- /dev/null
+++ b/alpha/state/build.rs
@@ -0,0 +1,3 @@
+fn main() {
+    gear_wasm_builder::build_metawasm();
+}
diff --git a/alpha/state/src/lib.rs b/alpha/state/src/lib.rs
new file mode 100644
index 0000000..2144fd4
--- /dev/null
+++ b/alpha/state/src/lib.rs
@@ -0,0 +1,22 @@
+#![no_std]
+
+use alpha_io::{Label, Resource};
+use gstd::prelude::*;
+
+#[gmeta::metawasm]
+pub mod metafns {
+    pub type State = alpha_io::State;
+
+    /// Returns all domains (pages) that matches the search input.
+    pub fn labels(_state: State) -> Vec<Label> {
+        Label::list()
+    }
+
+    /// Search all resouces matching the input label.
+    pub fn search(state: State, label: Label) -> Vec<Resource> {
+        state
+            .into_iter()
+            .filter(|s| s.labels.contains(&label))
+            .collect()
+    }
+}
diff --git a/alpha/tests/main.rs b/alpha/tests/main.rs
new file mode 100644
index 0000000..734ff7b
--- /dev/null
+++ b/alpha/tests/main.rs
@@ -0,0 +1,37 @@
+use alpha_io::{Label, Resource};
+use gtest::{state_args, Program, System};
+use std::{fs, path::PathBuf};
+
+fn get_state_binary() -> Vec<u8> {
+    let bin = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
+        .join("../target/wasm32-unknown-unknown/debug/alpha_state.meta.wasm");
+    fs::read(bin).unwrap()
+}
+
+#[test]
+fn test() {
+    let system = System::new();
+    system.init_logger();
+
+    let program = Program::current(&system);
+    let mut result = program.send_bytes(2, []);
+    assert!(!result.main_failed());
+
+    let input = Resource {
+        labels: Label::list(),
+        domain: "vara-go".into(),
+        description: "test test".into(),
+        html: "<div>hello, world<div>".into(),
+        styles: "".into(),
+    };
+    result = program.send(2, input);
+    assert!(!result.main_failed());
+
+    let state_bin = get_state_binary();
+    let search = Label::Vara;
+    let output: Vec<Resource> = program
+        .read_state_using_wasm(b"", "search", state_bin, state_args!(search))
+        .expect("Failed to search source");
+    // assert_eq!(output.get(&input.domain), Some(input.src).as_ref());
+    println!("{output:?}");
+}
diff --git a/tests/tests.rs b/tests/tests.rs
index bc55bc9..7dc1f11 100644
--- a/tests/tests.rs
+++ b/tests/tests.rs
@@ -60,7 +60,6 @@ fn test() {
     assert_eq!(output.get(&input.domain), Some(input.src).as_ref());
 }
 
-#[allow(unused)]
 fn get_state_binary() -> Vec<u8> {
     fs::read("target/wasm32-unknown-unknown/debug/template_state.meta.wasm").unwrap()
 }