copilot/packages/portal-api/lib/index.js

'use strict';

const Schema = require('joyent-cp-gql-schema');
const Graphi = require('graphi');
const Hoek = require('hoek');
const Piloted = require('piloted');
const Data = require('./data');
const Pack = require('../package.json');
const Resolvers = require('./resolvers');
const ContainerPilotWatcher = require('./watch/container-pilot');
const MachinesWatcher = require('./watch/machines');


const internals = {
  namespace: process.env.NAMESPACE ? `/${process.env.NAMESPACE}` : '',
  defaults: {
    data: {
      db: {
        host: 'localhost'
      },
      dockerComposeHost: 'tcp://0.0.0.0:4242'
    },
    watch: {
      url: process.env.SDC_URL,
      account: process.env.SDC_ACCOUNT,
      keyId: process.env.SDC_KEY_ID
    }
  }
};

module.exports = function (server, options, next) {
  const settings = Hoek.applyToDefaults(internals.defaults, options || {});

  try {
    const docker = Piloted.service('compose-api');
    if (docker) {
      settings.data.dockerComposeHost = `tcp://${docker.address}:${docker.port}`;
    }

    const rethinkdb = Piloted.service('rethinkdb');
    if (rethinkdb) {
      settings.data.db.host = rethinkdb.address;
    }
  } catch (ex) {
    server.log(['error'], ex);
  }

  settings.watch.server = server;
  settings.data.server = server;

  const data = new Data(settings.data);
  const watcherOptions = settings.watch;
  watcherOptions.data = data;
  const cpWatcher = new ContainerPilotWatcher(watcherOptions);
  const machinesWatcher = new MachinesWatcher(watcherOptions);

  // watcher <-> watcher
  // portal depends on watcher and vice-versa
  // I'm sure there is a better way to organize this domains
  // but this works for now
  data.setMachinesWatcher(machinesWatcher);

  cpWatcher.on('error', (err) => {
    server.log(['error'], err);
  });

  data.on('error', (err) => {
    server.log(['error'], err);
  });

  data.connect((err) => {
    if (err) {
      return next(err);
    }

    server.bind(data);

    Piloted.on('refresh', internals.refresh(data));

    machinesWatcher.poll();
    cpWatcher.poll();

    server.register([
      {
        register: Graphi,
        options: {
          graphqlPath: `${internals.namespace}/graphql`,
          graphiqlPath: `${internals.namespace}/graphiql`,
          schema: Schema,
          resolvers: Resolvers(data)
        }
      }
    ]);

    next();
  });
};

module.exports.attributes = {
  name: Pack.name,
  version: Pack.version,
  once: true,
  multiple: false
};


internals.refresh = function (data) {
  return () => {
    const docker = Piloted.service('compose-api');
    if (docker) {
      data.reconnectCompose(`tcp://${docker.address}:${docker.port}`);
    }

    const rethinkdb = Piloted.service('rethinkdb');
    if (rethinkdb) {
      data.reconnectDb({ host: rethinkdb.address });
    }
  };
};