shields/lib/badge-data.js

'use strict';

const isCSSColor = require('is-css-color');
const logos = require('./load-logos')();
const simpleIcons = require('./load-simple-icons')();
const {
  svg2base64,
  isDataUri,
} = require('./logo-helper');
const colorschemes = require('./colorscheme.json');

function toArray(val) {
  if (val === undefined) {
    return [];
  } else if (Object(val) instanceof Array) {
    return val;
  } else {
    return [val];
  }
}

function prependPrefix(s, prefix) {
  if (s === undefined) {
    return undefined;
  }

  s = '' + s;

  if (s.startsWith(prefix)) {
    return s;
  } else {
    return prefix + s;
  }
}

function isHexColor (s = ''){
  return /^([\da-f]{3}){1,2}$/i.test(s);
}

function makeColor(color) {
  if (isHexColor(color)) {
    return '#' + color;
  } else if (colorschemes[color] !== undefined){
    return colorschemes[color].colorB;
  } else if (isCSSColor(color)){
    return color;
  } else {
    return undefined;
  }
}

function makeColorB(defaultColor, overrides) {
  return makeColor(overrides.colorB || defaultColor);
}

function setBadgeColor(badgeData, color) {
  if (isHexColor(color)) {
    badgeData.colorB = '#' + color;
    delete badgeData.colorscheme;
  } else if (colorschemes[color] !== undefined){
    badgeData.colorscheme = color;
    delete badgeData.colorB;
  } else if (isCSSColor(color)){
    badgeData.colorB = color;
    delete badgeData.colorscheme;
  } else {
    badgeData.colorscheme = 'red';
    delete badgeData.colorB;
  }
  return badgeData;
}

function makeLabel(defaultLabel, overrides) {
  return '' + (overrides.label === undefined ? defaultLabel || '' : overrides.label);
}

function getShieldsIcon(icon = '', color = ''){
  icon = typeof icon === 'string' ? icon.toLowerCase() : '';
  if (!logos[icon]){
    return undefined;
  }
  color = makeColor(color);
  return color ? logos[icon].svg.replace(/fill="(.+?)"/g, `fill="${color}"`) : logos[icon].base64;
}

function getSimpleIcon(icon = '', color = null){
  icon = typeof icon === 'string' ? icon.toLowerCase().replace(/ /g, '-') : '';
  if (!simpleIcons[icon]){
    return undefined;
  }
  color = makeColor(color);
  return color ? simpleIcons[icon].svg.replace('<svg', `<svg fill="${color}"`) : simpleIcons[icon].base64;
}

function makeLogo(defaultNamedLogo, overrides) {
  if (overrides.logo === undefined){
    return svg2base64(getShieldsIcon(defaultNamedLogo, overrides.logoColor) || getSimpleIcon(defaultNamedLogo, overrides.logoColor));
  }

  // +'s are replaced with spaces when used in query params, this returns them to +'s, then removes remaining whitespace - #1546
  const maybeDataUri = prependPrefix(overrides.logo, 'data:').replace(/ /g, '+').replace(/\s/g, '');

  if (isDataUri(maybeDataUri)) {
    return maybeDataUri;
  } else {
    return svg2base64(getShieldsIcon(overrides.logo, overrides.logoColor) || getSimpleIcon(overrides.logo, overrides.logoColor));
  }
}

// Generate the initial badge data. Pass the URL query parameters, which
// override the default label.
//
// The following parameters are supported:
//
//   - label
//   - style
//   - logo
//   - logoWidth
//   - link
//   - colorA
//   - colorB
//   - maxAge
//
// Note: maxAge is handled by cache(), not this function.
function makeBadgeData(defaultLabel, overrides) {
  return {
    text: [makeLabel(defaultLabel, overrides), 'n/a'],
    colorscheme: 'lightgrey',
    template: overrides.style,
    logo: makeLogo(undefined, overrides),
    logoPosition: +overrides.logoPosition,
    logoWidth: +overrides.logoWidth,
    links: toArray(overrides.link),
    colorA: makeColor(overrides.colorA),
    colorB: makeColor(overrides.colorB),
  };
}

module.exports = {
  toArray,
  prependPrefix,
  isHexColor,
  makeLabel,
  makeLogo,
  makeBadgeData,
  makeColor,
  makeColorB,
  setBadgeColor
};