From db898a40d0cc57b25b4e46c74b2369002f4a2129 Mon Sep 17 00:00:00 2001
From: Yusuke Sasaki <yusuke.sasaki.nuem@gmail.com>
Date: Tue, 15 Oct 2019 18:11:12 +0900
Subject: [PATCH] add an input argument to specify whether to run 'rustup self
 update'

---
 action.yml  | 3 +++
 src/args.ts | 6 ++++--
 src/main.ts | 4 ++++
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/action.yml b/action.yml
index 6c43026..9eed804 100644
--- a/action.yml
+++ b/action.yml
@@ -19,6 +19,9 @@ inputs:
   override:
     description: Set installed toolchain as an override for a directory
     default: false
+  self_update:
+    description: Update rustup itself at first
+    default: false
 
 runs:
   using: 'node12'
diff --git a/src/args.ts b/src/args.ts
index e3fe0b9..9606c5f 100644
--- a/src/args.ts
+++ b/src/args.ts
@@ -30,7 +30,8 @@ export interface ToolchainOptions {
     name: string,
     target?: string,
     default: boolean,
-    override: boolean
+    override: boolean,
+    self_update: boolean,
 }
 
 export function toolchain_args(): ToolchainOptions {
@@ -38,6 +39,7 @@ export function toolchain_args(): ToolchainOptions {
         name: getInput('toolchain', {required: true}),
         target: getInput('target') || undefined,
         default: inputBoolean('default'),
-        override: inputBoolean('override')
+        override: inputBoolean('override'),
+        self_update: inputBoolean('self_update'),
     };
 }
diff --git a/src/main.ts b/src/main.ts
index c6129eb..43b77c8 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -74,6 +74,10 @@ async function run() {
     const opts = args.toolchain_args();
     const rustup = await get_rustup(opts.name);
 
+    if (opts.self_update) {
+        await exec.exec(rustup, ['self', 'update']);
+    }
+
     await exec.exec(rustup, ['toolchain', 'install', opts.name]);
 
     if (opts.default) {