cmd/abigen: parse contract name as abi identifier

Vandal38 · Feb 6, 2017 · d0eeb3e · d0eeb3e
1 parent 2964504
commit d0eeb3e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cmd/abigen/main.go b/cmd/abigen/main.go
@@ -94,7 +94,9 @@ func main() {
 			abi, _ := json.Marshal(contract.Info.AbiDefinition) // Flatten the compiler parse
 			abis = append(abis, string(abi))
 			bins = append(bins, contract.Code)
-			types = append(types, name)
+
+			nameParts := strings.Split(name, ":")
+			types = append(types, nameParts[len(nameParts)-1])
 		}
 	} else {
 		// Otherwise load up the ABI, optional bytecode and type name from the parameters