Fix bug about generate native simulator

Author: joii2020

native-simulator/GlobalMakefile

@@ -6,12 +6,12 @@ MODE := release
 BUILD_DIR := ../build/$(MODE)
 
 build:
-	mkdir -p ../$(BUILD_DIR)
+	mkdir -p $(BUILD_DIR)
 	@set -eu; \
 	if [ "x$(CONTRACT)" = "x" ]; then \
 		for contract in $(wildcard ./*); do \
 			if [ -f $$contract/Cargo.toml ]; then \
-				cd $$contract && cargo build && cp target/debug/*_dbg.* ../$(BUILD_DIR); \
+				cd $$contract && cargo build && cp target/debug/*_sim.* ../$(BUILD_DIR)/; \
 			fi; \
 		done; \
 	else \

workspace/scripts/generate-native-simulator

@@ -14,7 +14,7 @@ if [ "$CONTRACT_NAME" = "_" ]; then
     contracts=()
     for i in "${!folders[@]}"; do
         contract_name=$(echo "${folders[$i]}" | awk -F'/' '{print $(NF-1)}')
-        if [ ! -d "${folders[$i]}/${contract_name}-dbg" ]; then
+        if [ ! -d "${folders[$i]}/${contract_name}-sim" ]; then
             contracts+=("${contract_name}")
         fi
     done
@@ -54,15 +54,22 @@ SIMULATOR_NAME=${CONTRACT_NAME}-sim
 cargo generate $2 $3 native-simulator -n $SIMULATOR_NAME --destination native-simulators -d contract_name=$CONTRACT_NAME -d contract_crate_name=$CONTRACT_CRATE
 mv native-simulators/$SIMULATOR_NAME/src/contract-lib.rs contracts/$CONTRACT_NAME/src/lib.rs
 
+if [ ! -d 'native-simulators/Makefile' ]; then
+    mv native-simulators/$SIMULATOR_NAME/GlobalMakefile native-simulators/Makefile
+else
+    rm -f native-simulators/$SIMULATOR_NAME/GlobalMakefile
+fi
+
+# Insert feature: simulator into Cargo.toml
 FILE=contracts/$CONTRACT_NAME/Cargo.toml
 if grep -q "\[features\]" "$FILE"; then
     sed -i '/\[features\]/a\simulator = ["ckb-std/simulator"]' "$FILE"
 else
     echo -e "\n[features]\nsimulator = [\"ckb-std/simulator\"]" >> "$FILE"
 fi
 
-if [ ! -d 'native-simulators/Makefile' ]; then
-    mv native-simulators/$SIMULATOR_NAME/GlobalMakefile native-simulators/Makefile
-else
-    rm -f native-simulators/$SIMULATOR_NAME/GlobalMakefile
-fi
+FILE=contracts/$CONTRACT_NAME/src/main.rs
+sed -i 's/#!\[no_std\]/#!\[cfg_attr(not(feature = "simulator"), no_std)\]/' $FILE
+sed -i 's/#!\[cfg_attr(not(test), no_main)\]/#!\[cfg_attr(not(any(feature = "simulator", test)), no_main)\]/' $FILE
+sed -i 's/#\[cfg(test)\]/#\[cfg(any(feature = "simulator", test))\]/' $FILE
+sed -i 's/#\[cfg(not(test))\]/#\[cfg(not(any(feature = "simulator", test)))\]/' $FILE