rustup and wasm32-unknown-unknown auto setup

index.js

@@ -10,6 +10,7 @@ const chalk = require('chalk')
 const which = require('which')
 const sh = require('shelljs')
 const path = require('path')
+const rustSetup = require('./utils/rust-setup');
 
 const renameFile = async function(oldPath, newPath) {
   return new Promise((resolve, reject) => {
@@ -119,6 +120,12 @@ const createProject = async function({ contract, frontend, projectDir, veryVerbo
     await replaceInFiles({ files: `${projectDir}/README.md`, from: /npm\b( run)?/g, to: 'yarn' })
   }
 
+  // setup rust
+  let wasRustupInstalled = false;
+  if (contract === 'rust') {
+    wasRustupInstalled = await rustSetup.setupRustAndWasm32Target();
+  }
+
   if (hasNpm || hasYarn) {
     console.log('Installing project dependencies...')
     spawn.sync(hasYarn ? 'yarn' : 'npm', ['install'], { cwd: projectDir, stdio: 'inherit' })
@@ -145,13 +152,21 @@ Inside that directory, you can run several commands:
     Also deploys web frontend using GitHub Pages.
     Consult with {bold README.md} for details on how to deploy and {bold package.json} for full list of commands.
 
-We suggest that you begin by typing:
-
-  {bold cd ${projectDir}}
-  {bold ${runCommand} dev}
+We suggest that you begin by typing:`);
+
+  if (wasRustupInstalled) {
+    console.log(chalk`
+    {bold source $HOME/.cargo/env}
+    {bold cd ${projectDir}}
+    {bold ${runCommand} dev}`);
+  } else {
+    console.log(chalk`
+    {bold cd ${projectDir}}
+    {bold ${runCommand} dev}`);
+  }
 
-Happy hacking!
-`)
+  console.log(chalk`
+Happy hacking!`);
 }
 
 const opts = yargs

package.json

@@ -34,6 +34,7 @@
     "chalk": "^4.0.0",
     "cross-spawn": "^7.0.1",
     "ncp": "^2.0.0",
+    "os": "^0.1.1",
     "replace-in-files": "^3.0.0",
     "shelljs": "^0.8.3",
     "which": "^2.0.2",

utils/rust-setup.js

@@ -0,0 +1,122 @@
+const chalk = require('chalk');
+const os = require('os');
+const readline = require('readline');
+const sh = require('shelljs');
+
+// script from https://rustup.rs/ with auto-accept flag "-y"
+const installRustupScript = "curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y";
+const updatePath = 'source $HOME/.cargo/env';
+const addWasm32TargetScript = "rustup target add wasm32-unknown-unknown";
+// We should update PATH in the same script because every new Bash scripts are executed in a separate shell
+const updatePathAndAddWasm32TargetScript = updatePath + ' && ' + addWasm32TargetScript;
+
+const isWindows = os.platform() === 'win32'
+
+function isRustupInstalled() {
+    console.log(chalk`Checking if {bold rustup} is installed...`);
+    const isInstalled = sh.exec('rustup --version &> /dev/null').code === 0;
+    console.log(chalk`{bold rustup} is`, isInstalled ? 'installed\n' : 'not installed\n');
+    return isInstalled;
+}
+
+function isWasmTargetAdded() {
+    console.log(chalk`Checking installed {bold Rust targets}..`);
+    const addedTargets = sh.exec('rustup target list --installed').stdout;
+    const isWasmTargetAdded = addedTargets.includes('wasm32-unknown-unknown');
+    console.log(chalk`{bold wasm32-unknown-unknown} target `, isWasmTargetAdded ? 'already added' : 'is not added');
+    return isWasmTargetAdded;
+}
+
+function installRustup() {
+    console.log(chalk`Installing {bold rustup}...`);
+    sh.exec(installRustupScript);
+}
+
+function addWasm32Target() {
+    console.log(chalk`Adding {bold wasm32-unknown-unknown} target...`);
+    sh.exec(updatePathAndAddWasm32TargetScript);
+}
+
+async function askYesNoQuestionAndRunFunction(question, functionToRun = null) {
+    const rl = readline.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    try {
+        for (let attempt = 0; attempt < 4; attempt++) {
+            const answer = await new Promise((resolve) => {
+                rl.question(question, (userInput) => {
+                    userInput = userInput.toLowerCase();
+                    if (userInput === 'y' || userInput === '') {
+                        if (functionToRun) functionToRun();
+                        resolve(true);
+                    }
+                    if (userInput === 'n') {
+                        resolve(false);
+                    }
+                    resolve(undefined);
+                });
+            });
+            if (answer !== undefined) {
+                return answer;
+            }
+        }
+    } finally {
+        rl.close();
+    }
+    return false;
+}
+
+const installRustupDisclaimer = chalk`In order to work with {bold rust} smart contracts we recommend you install {bold rustup}, the Rust toolchain installer.`;
+const addWasm32TargetDisclaimer = chalk`To build Rust smart contracts you need to add {bold WebAssembly} compiler target to Rust toolchain.`;
+
+const installRustupQuestion = chalk`
+${installRustupDisclaimer} We can run the following command to do it for you:
+
+    {bold ${installRustupScript}}
+
+Continue with installation (Y/n)?: `;
+
+const addWasm32TargetQuestion = chalk`
+${addWasm32TargetDisclaimer} We can run the following command to do it for you:
+
+    {bold ${addWasm32TargetScript}}
+
+Continue with installation (Y/n)?: `;
+
+const rustupAndWasm32WindowsInstallationInstructions = chalk`
+${installRustupDisclaimer}
+    1. Go to https://rustup.rs
+    2. Download {bold rustup-init.exe}
+    3. Install it on your system
+
+${addWasm32TargetDisclaimer}
+
+Run the following command to do it:
+
+    {bold ${addWasm32TargetScript}}
+
+Press {bold Enter} to continue project creation.`;
+
+async function setupRustAndWasm32Target() {
+    try {
+        if (isWindows) {
+            await askYesNoQuestionAndRunFunction(rustupAndWasm32WindowsInstallationInstructions);
+            return false;
+        }
+        let wasRustupInstalled = false;
+        if (!isRustupInstalled()) {
+            wasRustupInstalled = await askYesNoQuestionAndRunFunction(installRustupQuestion, installRustup);
+        }
+        if (!isWasmTargetAdded()) {
+            await askYesNoQuestionAndRunFunction(addWasm32TargetQuestion, addWasm32Target);
+        }
+        return wasRustupInstalled;
+    } catch (e) {
+        console.log(chalk`Failed to run {bold rust} setup script`, e);
+    }
+}
+
+module.exports = {
+    setupRustAndWasm32Target,
+};

yarn.lock

@@ -3075,6 +3075,11 @@ os-tmpdir@~1.0.2:
   resolved "https://registry.yarnpkg.com/os-tmpdir/-/os-tmpdir-1.0.2.tgz#bbe67406c79aa85c5cfec766fe5734555dfa1274"
   integrity sha1-u+Z0BseaqFxc/sdm/lc0VV36EnQ=
 
+os@^0.1.1:
+  version "0.1.1"
+  resolved "https://registry.yarnpkg.com/os/-/os-0.1.1.tgz#208845e89e193ad4d971474b93947736a56d13f3"
+  integrity sha1-IIhF6J4ZOtTZcUdLk5R3NqVtE/M=
+
 p-cancelable@^1.0.0:
   version "1.1.0"
   resolved "https://registry.yarnpkg.com/p-cancelable/-/p-cancelable-1.1.0.tgz#d078d15a3af409220c886f1d9a0ca2e441ab26cc"