open-consul/ui/packages/consul-ui/app/services/repository/dc.js

import Error from '@ember/error';
import { inject as service } from '@ember/service';
import RepositoryService from 'consul-ui/services/repository';
import dataSource from 'consul-ui/decorators/data-source';
import { HEADERS_DEFAULT_ACL_POLICY as DEFAULT_ACL_POLICY } from 'consul-ui/utils/http/consul';

const SECONDS = 1000;
const MODEL_NAME = 'dc';

const zero = {
  Total: 0,
  Passing: 0,
  Warning: 0,
  Critical: 0,
};
const aggregate = (prev, body, type) => {
  return body[type].reduce((prev, item) => {
    // for each Partitions, Namespaces
    ['Partition', 'Namespace'].forEach((bucket) => {
      // lazily initialize
      let obj = prev[bucket][item[bucket]];
      if (typeof obj === 'undefined') {
        obj = prev[bucket][item[bucket]] = {
          Name: item[bucket],
        };
      }
      if (typeof obj[type] === 'undefined') {
        obj[type] = {
          ...zero,
        };
      }
      //

      // accumulate
      obj[type].Total += item.Total;
      obj[type].Passing += item.Passing;
      obj[type].Warning += item.Warning;
      obj[type].Critical += item.Critical;
    });

    // also aggregate the Datacenter, without doubling up
    // for Partitions/Namespaces
    prev.Datacenter[type].Total += item.Total;
    prev.Datacenter[type].Passing += item.Passing;
    prev.Datacenter[type].Warning += item.Warning;
    prev.Datacenter[type].Critical += item.Critical;
    return prev;
  }, prev);
};

export default class DcService extends RepositoryService {
  @service('env') env;

  getModelName() {
    return MODEL_NAME;
  }

  @dataSource('/:partition/:ns/:dc/datacenters')
  async fetchAll({ partition, ns, dc }, { uri }, request) {
    const Local = this.env.var('CONSUL_DATACENTER_LOCAL');
    const Primary = this.env.var('CONSUL_DATACENTER_PRIMARY');
    return (
      await request`
      GET /v1/catalog/datacenters
      X-Request-ID: ${uri}
    `
    )((headers, body, cache) => {
      // TODO: Not sure nowadays whether we need to keep lowercasing everything
      // I vaguely remember when I last looked it was not needed for browsers anymore
      // but I also vaguely remember something about Pretender lowercasing things still
      // so if we can work around Pretender I think we can remove all the header lowercasing
      // For the moment we lowercase here so as to not effect the ember-data-flavoured-v1 fork
      const entry = Object.entries(headers).find(
        ([key, value]) => key.toLowerCase() === DEFAULT_ACL_POLICY.toLowerCase()
      );
      //
      const DefaultACLPolicy = entry[1] || 'allow';
      return {
        meta: {
          version: 2,
          uri: uri,
        },
        body: body.map((dc) => {
          return cache(
            {
              Name: dc,
              Datacenter: '',
              Local: dc === Local,
              Primary: dc === Primary,
              DefaultACLPolicy: DefaultACLPolicy,
            },
            (uri) => uri`${MODEL_NAME}:///${''}/${''}/${dc}/datacenter`
          );
        }),
      };
    });
  }

  @dataSource('/:partition/:ns/:dc/datacenter')
  async fetch({ partition, ns, dc }, { uri }, request) {
    return (
      await request`
      GET /v1/operator/autopilot/state?${{ dc }}
      X-Request-ID: ${uri}
    `
    )((headers, body, cache) => {
      // turn servers into an array instead of a map/object
      const servers = Object.values(body.Servers);
      const grouped = [];
      return {
        meta: {
          version: 2,
          uri: uri,
        },
        body: cache(
          {
            ...body,
            // all servers
            Servers: servers,
            RedundancyZones: Object.entries(body.RedundancyZones || {}).map(([key, value]) => {
              const zone = {
                ...value,
                Name: key,
                Healthy: true,
                // convert the string[] to Server[]
                Servers: value.Servers.reduce((prev, item) => {
                  const server = body.Servers[item];
                  // keep a record of things
                  grouped.push(server.ID);
                  prev.push(server);
                  return prev;
                }, []),
              };
              return zone;
            }),
            ReadReplicas: (body.ReadReplicas || []).map((item) => {
              // keep a record of things
              grouped.push(item);
              return body.Servers[item];
            }),
            Default: {
              Servers: servers.filter((item) => !grouped.includes(item.ID)),
            },
          },
          (uri) => uri`${MODEL_NAME}:///${''}/${''}/${dc}/datacenter`
        ),
      };
    });
  }

  @dataSource('/:partition/:ns/:dc/catalog/health')
  async fetchCatalogHealth({ partition, ns, dc }, { uri }, request) {
    return (
      await request`
      GET /v1/internal/ui/catalog-overview?${{ dc, stale: null }}
      X-Request-ID: ${uri}
    `
    )((headers, body, cache) => {
      // for each Services/Nodes/Checks aggregate
      const agg = ['Nodes', 'Services', 'Checks'].reduce(
        (prev, item) => aggregate(prev, body, item),
        {
          Datacenter: {
            Name: dc,
            Nodes: {
              ...zero,
            },
            Services: {
              ...zero,
            },
            Checks: {
              ...zero,
            },
          },
          Partition: {},
          Namespace: {},
        }
      );

      return {
        meta: {
          version: 2,
          uri: uri,
          interval: 30 * SECONDS,
        },
        body: {
          Datacenter: agg.Datacenter,
          Partitions: Object.values(agg.Partition),
          Namespaces: Object.values(agg.Namespace),
          ...body,
        },
      };
    });
  }

  @dataSource('/:partition/:ns/:dc/datacenter-cache/:name')
  async find(params) {
    const items = this.store.peekAll('dc');
    const item = items.findBy('Name', params.name);
    if (typeof item === 'undefined') {
      // TODO: We should use a HTTPError error here and remove all occurances of
      // the custom shaped ember-data error throughout the app
      const e = new Error('Page not found');
      e.status = '404';
      throw { errors: [e] };
    }
    return item;
  }
}