feat: improve labels

snyk · Apr 28, 2020 · 1c80dfb · 1c80dfb
1 parent f85070d
commit 1c80dfb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 14 deletions.
diff --git a/src/SnykAction.js b/src/SnykAction.js
@@ -49,19 +49,28 @@ export class SnykVulnInfo {
       .map(diagnostic => {
         const pkg = getPackageFromMessage(diagnostic.message);
         const vulns = getPackageFromCache(pkg);
-
+        const count = vulns.count;
         const res = [];
 
         if (isAuthed()) {
           res.push(
             createShowOutputAction({
               diagnostic,
-              actionTitle: 'Show vulnerability details',
+              actionTitle: `Fix vuln${count === 1 ? '' : 's'}`,
               title: 'Show vulnerability details',
               args: [vulns],
               isPreferred: true,
             })
           );
+        } else {
+          res.push(
+            createAuthAction({
+              diagnostic,
+              actionTitle: `Fix vuln${count === 1 ? '' : 's'}`,
+              title: 'Connect to Snyk',
+              isPreferred: true,
+            })
+          );
         }
 
         res.push(
@@ -84,17 +93,6 @@ export class SnykVulnInfo {
         //   );
         // }
 
-        if (!isAuthed()) {
-          res.push(
-            createAuthAction({
-              diagnostic,
-              actionTitle: 'Connect Snyk to fix vulnerabilities',
-              title: 'Connect to Snyk',
-              isPreferred: true,
-            })
-          );
-        }
-
         return res;
       })
       .flat();

diff --git a/src/decorator.js b/src/decorator.js
@@ -48,7 +48,7 @@ function getDecorationMessage(packageInfo) {
   }
 
   const { count } = packageInfo.vulns;
-  let decorationMessage = `${count} vuln${count === 1 ? '' : 's'}`;
+  let decorationMessage = `${count} vuln${count === 1 ? '' : 's'} (click to fix)`;
 
   return decorationMessage;
 }