From c312eae3e3cfa28d8442b9e691ff063a85b3e3e5 Mon Sep 17 00:00:00 2001
From: Caleb Cartwright <calebcartwright@users.noreply.github.com>
Date: Sat, 22 Jun 2019 09:13:34 -0500
Subject: [PATCH] feat: add CodeFactor grade badge (#3588)

---
 .../codefactor/codefactor-grade.service.js    | 60 +++++++++++++++++++
 services/codefactor/codefactor-grade.spec.js  | 57 ++++++++++++++++++
 .../codefactor/codefactor-grade.tester.js     | 27 +++++++++
 services/codefactor/codefactor-helpers.js     | 43 +++++++++++++
 4 files changed, 187 insertions(+)
 create mode 100644 services/codefactor/codefactor-grade.service.js
 create mode 100644 services/codefactor/codefactor-grade.spec.js
 create mode 100644 services/codefactor/codefactor-grade.tester.js
 create mode 100644 services/codefactor/codefactor-helpers.js

diff --git a/services/codefactor/codefactor-grade.service.js b/services/codefactor/codefactor-grade.service.js
new file mode 100644
index 0000000000..150320f4e9
--- /dev/null
+++ b/services/codefactor/codefactor-grade.service.js
@@ -0,0 +1,60 @@
+'use strict'
+
+const Joi = require('@hapi/joi')
+const { BaseSvgScrapingService } = require('..')
+const { letterGrades } = require('./codefactor-helpers')
+
+const schema = Joi.object({
+  message: Joi.allow(...Object.keys(letterGrades)),
+}).required()
+
+module.exports = class CodeFactorGrade extends BaseSvgScrapingService {
+  static get category() {
+    return 'analysis'
+  }
+
+  static get route() {
+    return {
+      base: 'codefactor/grade',
+      pattern: ':vcsType(github|bitbucket)/:user/:repo/:branch*',
+    }
+  }
+
+  static get examples() {
+    return [
+      {
+        title: 'CodeFactor Grade',
+        namedParams: {
+          vcsType: 'github',
+          user: 'pallets',
+          repo: 'flask',
+          branch: 'master',
+        },
+        staticPreview: this.render({ grade: 'B+' }),
+      },
+    ]
+  }
+
+  static get defaultBadgeData() {
+    return {
+      label: 'code quality',
+    }
+  }
+
+  static render({ grade }) {
+    return {
+      message: grade,
+      color: letterGrades[grade].color,
+    }
+  }
+
+  async handle({ vcsType, user, repo, branch }) {
+    const { message } = await this._requestSvg({
+      schema,
+      url: `https://codefactor.io/repository/${vcsType}/${user}/${repo}/badge/${branch ||
+        ''}`,
+      errorMessages: { 404: 'repo or branch not found' },
+    })
+    return this.constructor.render({ grade: message })
+  }
+}
diff --git a/services/codefactor/codefactor-grade.spec.js b/services/codefactor/codefactor-grade.spec.js
new file mode 100644
index 0000000000..a35b08f61c
--- /dev/null
+++ b/services/codefactor/codefactor-grade.spec.js
@@ -0,0 +1,57 @@
+'use strict'
+
+const { test, given } = require('sazerac')
+const CodeFactorGrade = require('./codefactor-grade.service')
+
+describe('CodeFactorGrade', function() {
+  test(CodeFactorGrade.render, () => {
+    given({ grade: 'A' }).expect({
+      message: 'A',
+      color: 'brightgreen',
+    })
+    given({ grade: 'A-' }).expect({
+      message: 'A-',
+      color: 'green',
+    })
+    given({ grade: 'B+' }).expect({
+      message: 'B+',
+      color: 'yellowgreen',
+    })
+    given({ grade: 'B' }).expect({
+      message: 'B',
+      color: 'yellowgreen',
+    })
+    given({ grade: 'B-' }).expect({
+      message: 'B-',
+      color: 'yellowgreen',
+    })
+    given({ grade: 'C+' }).expect({
+      message: 'C+',
+      color: 'yellow',
+    })
+    given({ grade: 'C' }).expect({
+      message: 'C',
+      color: 'yellow',
+    })
+    given({ grade: 'C-' }).expect({
+      message: 'C-',
+      color: 'yellow',
+    })
+    given({ grade: 'D+' }).expect({
+      message: 'D+',
+      color: 'orange',
+    })
+    given({ grade: 'D' }).expect({
+      message: 'D',
+      color: 'orange',
+    })
+    given({ grade: 'D-' }).expect({
+      message: 'D-',
+      color: 'orange',
+    })
+    given({ grade: 'F' }).expect({
+      message: 'F',
+      color: 'red',
+    })
+  })
+})
diff --git a/services/codefactor/codefactor-grade.tester.js b/services/codefactor/codefactor-grade.tester.js
new file mode 100644
index 0000000000..b578ff1299
--- /dev/null
+++ b/services/codefactor/codefactor-grade.tester.js
@@ -0,0 +1,27 @@
+'use strict'
+
+const Joi = require('@hapi/joi')
+const t = (module.exports = require('../tester').createServiceTester())
+const { letterGrades } = require('./codefactor-helpers')
+const codeFactorGrade = Joi.allow(...Object.keys(letterGrades))
+
+t.create('Grade')
+  .get('/github/google/guava.json')
+  .expectBadge({
+    label: 'code quality',
+    message: codeFactorGrade,
+  })
+
+t.create('Grade (branch)')
+  .get('/github/pallets/flask/master.json')
+  .expectBadge({
+    label: 'code quality',
+    message: codeFactorGrade,
+  })
+
+t.create('Grade (nonexistent repo)')
+  .get('/github/badges/asdfasdfasdfasdfasfae.json')
+  .expectBadge({
+    label: 'code quality',
+    message: 'repo or branch not found',
+  })
diff --git a/services/codefactor/codefactor-helpers.js b/services/codefactor/codefactor-helpers.js
new file mode 100644
index 0000000000..98ef5382cd
--- /dev/null
+++ b/services/codefactor/codefactor-helpers.js
@@ -0,0 +1,43 @@
+'use strict'
+
+// https://support.codefactor.io/i14-glossary
+const letterGrades = {
+  A: {
+    color: 'brightgreen',
+  },
+  'A-': {
+    color: 'green',
+  },
+  'B+': {
+    color: 'yellowgreen',
+  },
+  B: {
+    color: 'yellowgreen',
+  },
+  'B-': {
+    color: 'yellowgreen',
+  },
+  'C+': {
+    color: 'yellow',
+  },
+  C: {
+    color: 'yellow',
+  },
+  'C-': {
+    color: 'yellow',
+  },
+  'D+': {
+    color: 'orange',
+  },
+  D: {
+    color: 'orange',
+  },
+  'D-': {
+    color: 'orange',
+  },
+  F: {
+    color: 'red',
+  },
+}
+
+module.exports = { letterGrades }