From fc76a28abf506e14f65cdbea7298025b01fffaf8 Mon Sep 17 00:00:00 2001
From: Johannes Zellner <johannes@cloudron.io>
Date: Thu, 2 Feb 2017 17:41:28 -0800
Subject: [PATCH] Add missing file

---
 digitalocean.js | 122 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 122 insertions(+)
 create mode 100644 digitalocean.js

diff --git a/digitalocean.js b/digitalocean.js
new file mode 100644
index 0000000..87c0171
--- /dev/null
+++ b/digitalocean.js
@@ -0,0 +1,122 @@
+/* jslint node:true */
+
+'use strict';
+
+var assert = require('assert'),
+    debug = require('debug')('e2e:digitalocean'),
+    safe = require('safetydance'),
+    superagent = require('superagent'),
+    util = require('util');
+
+module.exports = exports = {
+    setCredentials: setCredentials,
+
+    create: createDroplet,
+    get: getDroplet,
+    destroy: destroyDroplet
+};
+
+var DIGITALOCEAN = 'https://api.digitalocean.com/v2';
+
+var gAuth = null;
+
+function setCredentials(token) {
+    gAuth = 'Bearer ' + token;
+}
+
+function getSSHKeyId(keyName, callback) {
+    assert.strictEqual(typeof keyName, 'string');
+    assert.strictEqual(typeof callback, 'function');
+
+    superagent.get(DIGITALOCEAN + '/account/keys').set('Authorization', gAuth).end(function (error, result) {
+        if (error) return callback(error);
+        if (result.statusCode === 403) return callback('not authorized');
+        if (result.statusCode !== 200) return callback(util.format('failed with %s - %j', result.statusCode, result.text));
+
+        var sshKeys = result.body.ssh_keys || [];
+
+        var sshKeyId = null;
+        sshKeys.forEach(function (key) { if (key.name === keyName) sshKeyId = key.id; });
+
+        if (!sshKeyId) return callback('no ssh key found');
+
+        callback(null, sshKeyId);
+    });
+}
+
+function getDroplet(id, callback) {
+    assert.strictEqual(typeof id, 'number');
+    assert.strictEqual(typeof callback, 'function');
+
+    superagent.get(DIGITALOCEAN + '/droplets/' + id).set('Authorization', gAuth).end(function (error, result) {
+        if (error) return callback(error);
+        if (result.statusCode === 403) return callback('not authorized');
+        if (result.statusCode !== 200) return callback('droplet not found');
+
+        var droplet = result.body.droplet;
+        if (!droplet || typeof droplet !== 'object') return callback('Missing droplet in body, this is a DO issue');
+
+        var ip = safe.query(droplet, 'networks.v4[0].ip_address');
+
+        callback(null, { name: droplet.name, id: droplet.id, ip: ip, status: droplet.status });
+    });
+}
+
+function createDroplet(name, image, sshKeyName, region, size, userData, callback) {
+    assert.strictEqual(typeof name, 'string');
+    assert.strictEqual(typeof image, 'string');
+    assert.strictEqual(typeof sshKeyName, 'string');
+    assert.strictEqual(typeof region, 'string');
+    assert.strictEqual(typeof size, 'string');
+    assert.strictEqual(typeof userData, 'string');
+    assert.strictEqual(typeof callback, 'function');
+
+    debug('addDroplet: with name "' + name + '".');
+
+    getSSHKeyId(sshKeyName, function (error, sshKeyId) {
+        if (error) return callback(error);
+
+        var options = {
+            name: name,
+            size: size,
+            image: image,
+            region: region,
+            backups: false,
+            user_data: userData,
+            ssh_keys: [ sshKeyId ]
+        };
+
+        debug('addDroplet: Creating droplet with name:%s size: %s image: %s', options.name, options.size, options.image);
+
+        superagent.post(DIGITALOCEAN + '/droplets').set('Authorization', gAuth).send(options).end(function (error, result) {
+            if (error) return callback(error);
+            if (result.statusCode === 403) return callback('not authorized');
+            if (result.statusCode !== 202) return callback(util.format('Failed with status %s - %j', result.statusCode, result.text));
+
+            var droplet = result.body.droplet;
+            if (!droplet || typeof droplet !== 'object') return callback('Missing droplet in body, this is a DO issue');
+
+            debug('addDroplet: new droplet with id %s and name %s created', droplet.id, droplet.name);
+
+            callback(null, { id: droplet.id, name: droplet.name });
+        });
+    });
+}
+
+function destroyDroplet(id, callback) {
+    assert.strictEqual(typeof id, 'number');
+    assert.strictEqual(typeof callback, 'function');
+
+    debug('destroyDroplet: ' + id);
+
+    superagent.del(DIGITALOCEAN + '/droplets/' + id).set('Authorization', gAuth).end(function (error, result) {
+        if (error) return callback(error);
+        if (result.statusCode === 403) return callback('not authorized');
+        if (result.statusCode === 404) return callback(null);
+        if (result.statusCode !== 204) debug('destroyDroplet: unable to delete droplet. Maybe it was already deleted.');
+
+        debug('destroyDroplet: success');
+
+        callback(null);
+    });
+}