From 191a15123c055cc8502093444f889ec98724c77e Mon Sep 17 00:00:00 2001
From: Amin Yahyaabadi <aminyahyaabadi74@gmail.com>
Date: Thu, 16 Sep 2021 06:07:52 -0500
Subject: [PATCH] feat: add compiler input

---
 action.yml  |  3 +++
 src/main.ts | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+)

diff --git a/action.yml b/action.yml
index 6ec9256a..bf8e895a 100644
--- a/action.yml
+++ b/action.yml
@@ -3,6 +3,9 @@ description: "Install all the tools required for building and testing C++/C proj
 author: "Amin Yahyaabadi"
 
 inputs:
+  compiler:
+    description: "The compiler to use and its optinal version separated by - e.g. llvm-11"
+    required: false
   architecture:
     description: "The CPU architecture"
     required: false
diff --git a/src/main.ts b/src/main.ts
index 7cccc951..c84d8359 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -13,6 +13,7 @@ import { setupMSVC } from "./msvc/msvc"
 import { setupNinja } from "./ninja/ninja"
 import { setupOpencppcoverage } from "./opencppcoverage/opencppcoverage"
 import { setupPython } from "./python/python"
+import semverValid from "semver/functions/valid"
 
 function maybeGetInput(key: string) {
   const value = core.getInput(key.toLowerCase())
@@ -26,6 +27,43 @@ export async function main(): Promise<number> {
   const arch = core.getInput("architecture") || process.arch
   const setupCppDir = process.env.SETUP_CPP_DIR ?? "~/setup_cpp"
   try {
+    const maybeCompiler = maybeGetInput("compiler")
+    if (maybeCompiler !== undefined) {
+      const compilerAndMaybeVersion = maybeCompiler.split("-")
+      const compiler = compilerAndMaybeVersion[0]
+      let version: string | undefined
+      if (1 in compilerAndMaybeVersion) {
+        const maybeVersion = compilerAndMaybeVersion[1]
+        if (semverValid(maybeVersion) !== null) {
+          version = maybeVersion
+        } else {
+          core.error(`Invalid version ${maybeVersion} used for the compiler. Using the default version...`)
+        }
+      }
+
+      switch (compiler) {
+        case "llvm":
+        case "clang":
+        case "clang++": {
+          await setupLLVM(version ?? "11", setupCppDir)
+          break
+        }
+        case "cl":
+        case "msvc":
+        case "msbuild":
+        case "vs":
+        case "visualstudio":
+        case "visualcpp":
+        case "visualc++": {
+          await setupMSVC(version ?? "2019", setupCppDir)
+          break
+        }
+        default: {
+          core.error(`Unsupported compiler ${compiler}`)
+        }
+      }
+    }
+
     // setup cmake
     const cmakeVersion = maybeGetInput("cmake")
     if (cmakeVersion !== undefined) {