shields/services/codeclimate/codeclimate-analysis.service.js

import Joi from 'joi'
import { colorScale, letterScore } from '../color-formatters.js'
import { nonNegativeInteger } from '../validators.js'
import { BaseJsonService, NotFound, pathParams } from '../index.js'
import { isLetterGrade, fetchRepo } from './codeclimate-common.js'

const schema = Joi.object({
  data: Joi.object({
    meta: Joi.object({
      issues_count: nonNegativeInteger,
    }).required(),
    attributes: Joi.object({
      ratings: Joi.array()
        .items(
          Joi.object({
            letter: isLetterGrade,
            measure: Joi.object({
              value: Joi.number().required(),
            }).required(),
          }),
        )
        .length(1)
        .required(),
    }).required(),
  }).required(),
}).required()

const maintainabilityColorScale = colorScale(
  [50, 80, 90, 95],
  ['red', 'yellow', 'yellowgreen', 'green', 'brightgreen'],
)
const techDebtColorScale = colorScale(
  [5, 10, 20, 50],
  ['brightgreen', 'green', 'yellowgreen', 'yellow', 'red'],
)
const issueColorScale = colorScale(
  [1, 5, 10, 20],
  ['brightgreen', 'green', 'yellowgreen', 'yellow', 'red'],
)

const variantMap = {
  maintainability: {
    transform: data => ({
      maintainabilityLetter: data.attributes.ratings[0].letter,
    }),
    render: ({ maintainabilityLetter }) => ({
      label: 'maintainability',
      message: maintainabilityLetter,
      color: letterScore(maintainabilityLetter),
    }),
  },
  'maintainability-percentage': {
    transform: data => ({
      techDebtPercentage: data.attributes.ratings[0].measure.value,
    }),
    render: ({ techDebtPercentage }) => {
      // maintainability = 100 - technical debt.
      const maintainabilityPercentage = 100 - techDebtPercentage
      return {
        label: 'maintainability',
        message: `${maintainabilityPercentage.toFixed(0)}%`,
        color: maintainabilityColorScale(maintainabilityPercentage),
      }
    },
  },
  'tech-debt': {
    transform: data => ({
      techDebtPercentage: data.attributes.ratings[0].measure.value,
    }),
    render: ({ techDebtPercentage }) => ({
      label: 'technical debt',
      message: `${techDebtPercentage.toFixed(0)}%`,
      color: techDebtColorScale(techDebtPercentage),
    }),
  },
  issues: {
    transform: data => ({
      issueCount: data.meta.issues_count,
    }),
    render: ({ issueCount }) => ({
      label: 'issues',
      message: `${issueCount}`,
      color: issueColorScale(issueCount),
    }),
  },
}

export default class CodeclimateAnalysis extends BaseJsonService {
  static category = 'analysis'
  static route = {
    base: 'codeclimate',
    pattern:
      ':variant(maintainability|maintainability-percentage|tech-debt|issues)/:user/:repo',
  }

  static openApi = {
    '/codeclimate/{variant}/{user}/{repo}': {
      get: {
        summary: 'Code Climate maintainability',
        parameters: pathParams(
          {
            name: 'variant',
            example: 'maintainability',
            schema: {
              type: 'string',
              enum: ['maintainability', 'maintainability-percentage'],
            },
          },
          { name: 'user', example: 'tensorflow' },
          { name: 'repo', example: 'models' },
        ),
      },
    },
    '/codeclimate/tech-debt/{user}/{repo}': {
      get: {
        summary: 'Code Climate technical debt',
        parameters: pathParams(
          { name: 'user', example: 'tensorflow' },
          { name: 'repo', example: 'models' },
        ),
      },
    },
    '/codeclimate/issues/{user}/{repo}': {
      get: {
        summary: 'Code Climate issues',
        parameters: pathParams(
          { name: 'user', example: 'tensorflow' },
          { name: 'repo', example: 'models' },
        ),
      },
    },
  }

  static render({ variant, ...props }) {
    const { render } = variantMap[variant]

    return render(props)
  }

  async fetch({ user, repo }) {
    const repoInfos = await fetchRepo(this, { user, repo })
    const repoInfosWithSnapshot = repoInfos.filter(
      repoInfo => repoInfo.relationships.latest_default_branch_snapshot.data,
    )
    if (repoInfosWithSnapshot.length === 0) {
      throw new NotFound({ prettyMessage: 'snapshot not found' })
    }
    const {
      id: repoId,
      relationships: {
        latest_default_branch_snapshot: { data: snapshotInfo },
      },
    } = repoInfosWithSnapshot[0]
    const { data } = await this._requestJson({
      schema,
      url: `https://api.codeclimate.com/v1/repos/${repoId}/snapshots/${snapshotInfo.id}`,
    })
    return data
  }

  async handle({ variant, user, repo }) {
    const { transform } = variantMap[variant]

    const data = await this.fetch({ user, repo })
    const props = transform(data)
    return this.constructor.render({
      variant,
      ...props,
    })
  }
}