joyent-portal/portal-data/lib/index.js

'use strict';

const Hoek = require('hoek');
const Penseur = require('penseur');
const DCClient = require('docker-compose-client');
const awaitify = require('apr-awaitify');

const internals = {
  defaults: {
    name: 'portal'
  }
};

module.exports = class Data {
  constructor(options) {
    const settings = Hoek.applyToDefaults(options || {}, internals.defaults);

    // Penseur will assert that the options are correct
    this._db = new Penseur.Db(settings.name, settings);
    this._docker = new DCClient(settings.dockerHost);

    this._db.establish = awaitify(this._db.establish);
    this._db.deployments = awaitify(this._db.deployments);

    // promisify Penseur
    [
      'activities',
      'datacenters',
      'deployments',
      'manifests',
      'metrics'
    ].forEach(tableName =>
      ['insert', 'get', 'update', 'remove', 'all'].forEach(methodName => {
        this._db[tableName][methodName] = awaitify(
          this._db[tableName][methodName]
        );
      })
    );
  }

  connect(cb) {
    return this._db.establish();
  }

  /*
   * DeploymentGroupUuid
   * Manifest
   *  id: UUID,
   *  created: Date.now(),
   *  type: 'docker-compose',
   *  format: 'yml',
   *  raw: 'original yml file content',
   *  obj: { }
   */
  createDeployment({ deploymentGroupUuid, manifest, deployment }) {
    // trigger deployment
    // create deployment queue (we should think about what is a deployment queue)
    // create the ConvergencePlans
    // create a DeploymentPlan
    // create a Version
    // update the DeploymentGroup

    // TODO
    const updateDb = plan => {
      // deployment.services = [];
      // deployment.state = { current: 'stopped' };

      this._db.deployments
        .insert({
          name:
        })
        .then(key => {
          deployment.id = key;
          return deployment;
        });
    };

    const provision = ({ name }) =>
      this._docker
        .provision({
          projectName: name,
          manifest: manifest.raw
        })
        .then(updateDb);

    this.getDeployment(deploymentGroupUuid).then(provision);
  }

  getDeployment(id) {
    return this._db.deployments.get(id);
  }

  updateDeployment(deployment) {
    return this._db.deployments.update(deployment.id, deployment);
  }

  deleteDeployment(id) {
    return this._db.deployments.remove(id);
  }

  getDeployments() {
    return this._db.deployments.all();
  }

  getDatacenters() {
    return this._db.datacenters.all();
  }

  createManifest(deploymentId, manifest) {
    manifest.deploymentId = deploymentId;
    manifest.created = Date.now();

    return this._db.manifests.insert().then(id => {
      manifest.id = id;
      return manifest;
    });
  }
  getManifest(id) {
    return this._db.manifests.get();
  }

  getActivities(deploymentId) {
    return this._db.activities.query({ deploymentId });
  }

  getMetrics(containerId) {
    return this._db.metrics.get(containerId);
  }

  insertMetrics(containerId, metrics) {
    return this._db.metrics.get(containerId).then(existing => {
      if (existing) {
        return this._db.metrics.update(containerId, {
          metrics: this._db.append(metrics)
        });
      }

      const entry = { id: containerId, metrics };
      return this._db.metrics.insert(entry, { merge: true });
    });
  }

  getServices(deploymentId) {
    this._db.deployments.get(deploymentId, { filter: 'services' });
  }

  updateService(deploymentId, service) {
    this._db.deployments.get(deploymentId, { filter: 'services' }).then(() => {
      const serviceToUpdate = deployment.services.find(currentService => {
        return currentService.name === service.name;
      });

      if (!serviceToUpdate) {
        deployment.services.push(service);
      } else {
        serviceToUpdate.count = service.count;
        serviceToUpdate.containers = service.containers;
      }

      return this._db.deployments.update(deploymentId, {
        services: deployment.services
      });
    });
  }

  deploymentChanges(handler) {
    return this._db.deployments.changes('*', { reconnect: true, handler });
  }
};