[lang] Add check for no env params in message signatures

use-ink · Mar 7, 2019 · 6f0ab0f · 6f0ab0f
1 parent 5dc9607
commit 6f0ab0f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lang/src/hir.rs b/lang/src/hir.rs
@@ -111,6 +111,18 @@ impl Contract {
 					_ => ()
 				}
 			}
+			for fn_arg in inputs.iter().skip(1) {
+				if let ast::FnArg::Captured(arg_captured) = fn_arg {
+					if let syn::Pat::Ident(pat_ident) = &arg_captured.pat {
+						if pat_ident.ident == "env" {
+							bail!(
+								pat_ident.ident,
+								"contract messages must not contain an env argument"
+							)
+						}
+					}
+				}
+			}
 		}
 		Ok((deploy_handler, messages, methods))
 	}