From 965071bcf302473c78637f14f7e574c13be83b52 Mon Sep 17 00:00:00 2001
From: Evann Regnault <evann.regnault@pm.me>
Date: Mon, 12 Dec 2022 14:06:35 +0100
Subject: [PATCH] Hotfix - class artifacts

---
 Cargo.toml            |  2 +-
 src/data/artifacts.rs | 29 +++++++++++++++++++++++++++++
 2 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/Cargo.toml b/Cargo.toml
index 6d24de4..ce723d4 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "drunk-venti-rust"
-version = "1.0.4"
+version = "1.1.0"
 edition = "2021"
 authors = ["Evann Regnault"]
 license = "MIT"
diff --git a/src/data/artifacts.rs b/src/data/artifacts.rs
index 19defcb..17e1d20 100644
--- a/src/data/artifacts.rs
+++ b/src/data/artifacts.rs
@@ -25,9 +25,23 @@ pub struct Artifact {
     pub domain: Option<Box<str>>
 }
 
+pub fn get_real_artifact_name(artifact: &str) -> &str {
+    match artifact {
+        "+18%_atk_set" => "gladiators_finale",
+        "+20%_energy_recharge" => "emblem_of_severed_fate",
+        "+25%_physical_dmg" => "bloodstained_chivalry",
+        "+80_em" => "gilded_dreams",
+        "+15%_healing_bonus_set" => "ocean-hued_clam",
+        _ => artifact
+    }
+}
+
 impl Artifact{
+
+
     #[allow(dead_code)]
     pub async fn get(artifact: &str) -> Artifact {
+        let artifact = get_real_artifact_name(artifact);
         let host = env::var("API_HOST").unwrap();
         let port = env::var("API_PORT").unwrap();
         let url = format!("http://{}:{}/api/artifacts/{}", host, port, artifact);
@@ -46,6 +60,7 @@ impl Artifact{
 
     #[allow(dead_code)]
     pub(crate) async fn search(artifact: &str) -> Vec<Artifact> {
+        let artifact = get_real_artifact_name(artifact);
         let host = env::var("API_HOST").unwrap();
         let port = env::var("API_PORT").unwrap();
         let url = format!("http://{}:{}/api/artifacts/search/{}", host, port, artifact);
@@ -94,4 +109,18 @@ impl Artifact{
 fn test_artifact() {
     let data = std::fs::read_to_string("test/artifact.json").expect("No Artifact test file");
     serde_json::from_str::<Artifact>(&data).expect("Didn't work");
+}
+
+#[test]
+fn test_real_artfifact_name() {
+    match get_real_artifact_name("+18%_atk_set") {
+        "gladiators_finale" => {}
+        _ => panic!()
+    };
+
+    match get_real_artifact_name("lasfhkalkfhafsk") {
+        "lasfhkalkfhafsk" => {}
+        _ => panic!()
+    };
+
 }
\ No newline at end of file