Use GitHub Actions cache for installed binaries

action.yml

@@ -16,10 +16,12 @@ inputs:
     description: Use tool cache to speed up installation
     required: false
     default: false
-  use-cache:
-    description: Store installed binary in the GitHub Actions cache (not used yet)
+  key:
+    description: Store installed binary in the GitHub Actions cache with the specified key
+    required: false
+  restore-keys:
+    description: Restore keys as used in the GitHub cache action
     required: false
-    default: true
 
 runs:
   using: 'node12'

src/download.ts

@@ -2,10 +2,10 @@ import os from "os";
 import { promises as fs } from "fs";
 import path from "path";
 
+import { resolveVersion } from "@actions-rs/core";
 import * as core from "@actions/core";
 import * as exec from "@actions/exec";
 import * as tc from "@actions/tool-cache";
-import * as http from "@actions/http-client";
 
 const CLOUDFRONT_ROOT = "https://d1ad61wkrfbmp3.cloudfront.net";
 // Path to the public key of the sign certificate.
@@ -32,20 +32,6 @@ function getRunner(): string {
     }
 }
 
-async function resolveVersion(crate: string): Promise<string> {
-    const url = `https://crates.io/api/v1/crates/${crate}`;
-    const client = new http.HttpClient(
-        "@actions-rs (https://github.com/actions-rs/)"
-    );
-
-    const resp: any = await client.getJson(url);
-    if (resp.result == null) {
-        throw new Error("Unable to fetch latest crate version");
-    }
-
-    return resp.result["crate"]["newest_version"];
-}
-
 function buildUrl(crate: string, version: string): string {
     const runner = getRunner();
 

src/input.ts

@@ -9,19 +9,22 @@ export interface Input {
     crate: string;
     version: string;
     useToolCache: boolean;
-    useCache: boolean;
+    primaryKey?: string;
+    restoreKeys?: string[];
 }
 
 export function get(): Input {
     const crate = input.getInput("crate", { required: true });
     const version = input.getInput("version", { required: true });
     const useToolCache = input.getInputBool("use-tool-cache") || false;
-    const useCache = input.getInputBool("use-cache") || true;
+    const primaryKey = input.getInput("key") || undefined;
+    const restoreKeys = input.getInputAsArray("restore-keys") || undefined;
 
     return {
         crate: crate,
         version: version,
         useToolCache: useToolCache,
-        useCache: useCache,
+        primaryKey: primaryKey,
+        restoreKeys: restoreKeys,
     };
 }

src/main.ts

@@ -6,7 +6,8 @@ import * as download from "./download";
 
 interface Options {
     useToolCache: boolean;
-    useCache: boolean;
+    primaryKey?: string;
+    restoreKeys?: string[];
 }
 
 async function downloadFromToolCache(
@@ -52,7 +53,12 @@ export async function run(
     } catch (error) {
         core.info("Falling back to the `cargo install` command");
         const cargo = await Cargo.get();
-        await cargo.installCached(crate, version);
+        await cargo.installCached(
+            crate,
+            version,
+            options.primaryKey,
+            options.restoreKeys
+        );
     }
 }
 
@@ -62,7 +68,8 @@ async function main(): Promise<void> {
 
         await run(actionInput.crate, actionInput.version, {
             useToolCache: actionInput.useToolCache,
-            useCache: actionInput.useCache,
+            primaryKey: actionInput.primaryKey,
+            restoreKeys: actionInput.restoreKeys,
         });
     } catch (error) {
         core.setFailed(error.message);