From 53bb0a38f032d0b414d8fc2b1ed1c9eb6608456c Mon Sep 17 00:00:00 2001
From: timfish <tim@timfish.uk>
Date: Mon, 30 Sep 2019 18:57:21 +0100
Subject: [PATCH] Use std::any::type_name

---
 src/any.rs             |  8 +++++++-
 src/engine.rs          | 38 ++++----------------------------------
 tests/mismatched_op.rs | 29 +++++++++++++++++++++++++++--
 3 files changed, 38 insertions(+), 37 deletions(-)

diff --git a/src/any.rs b/src/any.rs
index ac2d9f68..97f5bf57 100644
--- a/src/any.rs
+++ b/src/any.rs
@@ -1,9 +1,11 @@
-use std::any::{Any as StdAny, TypeId};
+use std::any::{type_name, Any as StdAny, TypeId};
 use std::fmt;
 
 pub trait Any: StdAny {
     fn type_id(&self) -> TypeId;
 
+    fn type_name(&self) -> String;
+
     fn box_clone(&self) -> Box<dyn Any>;
 
     /// This type may only be implemented by `rhai`.
@@ -20,6 +22,10 @@ where
         TypeId::of::<T>()
     }
 
+    fn type_name(&self) -> String {
+        type_name::<T>().to_string()
+    }
+
     #[inline]
     fn box_clone(&self) -> Box<dyn Any> {
         Box::new(self.clone())
diff --git a/src/engine.rs b/src/engine.rs
index 1186afe1..fe8d2a1a 100644
--- a/src/engine.rs
+++ b/src/engine.rs
@@ -127,7 +127,6 @@ pub struct FnSpec {
 pub struct Engine {
     /// A hashmap containing all functions known to the engine
     pub fns: HashMap<FnSpec, Arc<FnIntExt>>,
-    pub type_names: HashMap<TypeId, String>,
 }
 
 pub enum FnIntExt {
@@ -164,7 +163,7 @@ impl Engine {
             .and_then(|b| {
                 b.downcast()
                     .map(|b| *b)
-                    .map_err(|a| EvalAltResult::ErrorMismatchOutputType(self.nice_type_name(a)))
+                    .map_err(|a| EvalAltResult::ErrorMismatchOutputType((*a).type_name()))
             })
     }
 
@@ -202,7 +201,7 @@ impl Engine {
             .ok_or_else(|| {
                 let typenames = args
                     .iter()
-                    .map(|x| self.nice_type_name((&**x).box_clone()))
+                    .map(|x| (*(&**x).box_clone()).type_name())
                     .collect::<Vec<_>>();
                 EvalAltResult::ErrorFunctionNotFound(format!("{} ({})", ident, typenames.join(",")))
             })
@@ -239,13 +238,6 @@ impl Engine {
         // currently a no-op, exists for future extensibility
     }
 
-    /// Register a type, providing a name for nice error messages.
-    pub fn register_type_name<T: Any>(&mut self, name: &str) {
-        self.register_type::<T>();
-        debug_println!("register type {}: {:?}", name, TypeId::of::<T>());
-        self.type_names.insert(TypeId::of::<T>(), name.into());
-    }
-
     /// Register a get function for a member of a registered type
     pub fn register_get<T: Clone + Any, U: Clone + Any, F>(&mut self, name: &str, get_fn: F)
     where
@@ -632,15 +624,6 @@ impl Engine {
         }
     }
 
-    fn nice_type_name(&self, b: Box<dyn Any>) -> String {
-        let tid = <dyn Any as Any>::type_id(&*b);
-        if let Some(name) = self.type_names.get(&tid) {
-            name.to_string()
-        } else {
-            format!("<unknown> {:?}", b.type_id())
-        }
-    }
-
     /// Evaluate a file
     pub fn eval_file<T: Any + Clone>(&mut self, fname: &str) -> Result<T, EvalAltResult> {
         use std::fs::File;
@@ -705,7 +688,7 @@ impl Engine {
                 match x.downcast::<T>() {
                     Ok(out) => Ok(*out),
                     Err(a) => Err(EvalAltResult::ErrorMismatchOutputType(
-                        self.nice_type_name(a),
+                        (*a).type_name(),
                     )),
                 }
             }
@@ -789,18 +772,6 @@ impl Engine {
     /// Register the default library. That means, numberic types, char, bool
     /// String, arithmetics and string concatenations.
     pub fn register_default_lib(engine: &mut Engine) {
-        engine.register_type_name::<i32>("i32");
-        engine.register_type_name::<u32>("u32");
-        engine.register_type_name::<i64>("integer");
-        engine.register_type_name::<u64>("u64");
-        engine.register_type_name::<usize>("usize");
-        engine.register_type_name::<f32>("f32");
-        engine.register_type_name::<f64>("float");
-        engine.register_type_name::<String>("string");
-        engine.register_type_name::<char>("char");
-        engine.register_type_name::<bool>("boolean");
-        engine.register_type_name::<Vec<Box<dyn Any>>>("array");
-
         macro_rules! reg_op {
             ($engine:expr, $x:expr, $op:expr, $( $y:ty ),*) => (
                 $(
@@ -942,8 +913,7 @@ impl Engine {
     /// Make a new engine
     pub fn new() -> Engine {
         let mut engine = Engine {
-            fns: HashMap::new(),
-            type_names: HashMap::new(),
+            fns: HashMap::new()
         };
 
         Engine::register_default_lib(&mut engine);
diff --git a/tests/mismatched_op.rs b/tests/mismatched_op.rs
index 0c455d99..2d6ff5fa 100644
--- a/tests/mismatched_op.rs
+++ b/tests/mismatched_op.rs
@@ -1,4 +1,4 @@
-use rhai::{Engine, EvalAltResult};
+use rhai::{Engine, EvalAltResult, RegisterFn};
 
 #[test]
 fn test_mismatched_op() {
@@ -7,7 +7,32 @@ fn test_mismatched_op() {
     assert_eq!(
         engine.eval::<i64>("60 + \"hello\""),
         Err(EvalAltResult::ErrorFunctionNotFound(
-            "+ (integer,string)".into()
+            "+ (i64,alloc::string::String)".into()
+        ))
+    );
+}
+
+#[test]
+fn test_mismatched_op_custom_type() {
+    #[derive(Clone)]
+    struct TestStruct {
+        x: i64,
+    }
+
+    impl TestStruct {
+        fn new() -> TestStruct {
+            TestStruct { x: 1 }
+        }
+    }
+
+    let mut engine = Engine::new();
+    engine.register_type::<TestStruct>();
+    engine.register_fn("new_ts", TestStruct::new);
+
+    assert_eq!(
+        engine.eval::<i64>("60 + new_ts()"),
+        Err(EvalAltResult::ErrorFunctionNotFound(
+            "+ (i64,mismatched_op::test_mismatched_op_custom_type::TestStruct)".into()
         ))
     );
 }