node-plantuml-2 1.0.0

package/scripts/download.js

@@ -0,0 +1,95 @@
+'use strict'
+
+var url = require('url')
+var fs = require('fs')
+
+var trace = []
+var MAX_RETRIES = 5
+var INITIAL_RETRY_DELAY = 2000 // 2 seconds
+
+function sleep (ms) {
+  return new Promise(function (resolve) {
+    setTimeout(resolve, ms)
+  })
+}
+
+function downloadWithRetry (uri, filename, changeExitCodeOnError, callback, retryCount) {
+  retryCount = retryCount || 0
+  trace.push('GET ' + uri + (retryCount > 0 ? ' (retry ' + retryCount + ')' : ''))
+  var protocol = url.parse(uri).protocol.slice(0, -1)
+
+  require(protocol).get(uri, function (res) {
+    trace.push('Reponse: ' + res.statusCode)
+    if (res.statusCode === 200) {
+      // Success, pipe to file
+      var fileStream = fs.createWriteStream(filename)
+      res.pipe(fileStream)
+      if (callback) {
+        res.on('end', function () {
+          callback()
+        })
+      }
+      fileStream.on('error', function (err) {
+        if (callback) {
+          callback(err)
+        }
+      })
+    } else if (res.headers.location) {
+      // Follow redirect
+      downloadWithRetry(res.headers.location, filename, changeExitCodeOnError, callback, retryCount)
+    } else if ((res.statusCode === 429 || res.statusCode === 503 || res.statusCode === 502) && retryCount < MAX_RETRIES) {
+      // Rate limited or server error - retry with exponential backoff
+      var retryDelay = INITIAL_RETRY_DELAY * Math.pow(2, retryCount)
+      var retryAfter = res.headers['retry-after']
+      if (retryAfter) {
+        retryDelay = parseInt(retryAfter) * 1000
+      }
+      console.log('Rate limited or server error (' + res.statusCode + '). Retrying in ' + (retryDelay / 1000) + ' seconds...')
+      sleep(retryDelay).then(function () {
+        downloadWithRetry(uri, filename, changeExitCodeOnError, callback, retryCount + 1)
+      })
+    } else {
+      // Error
+      trace.forEach(function (line) {
+        console.error(line)
+      })
+      var error = 'Failed to download ' + filename
+      console.error(error)
+
+      if (changeExitCodeOnError) {
+        process.exitCode = 1
+      }
+
+      if (callback) {
+        callback(error)
+      }
+    }
+  }).on('error', function (err) {
+    if (retryCount < MAX_RETRIES) {
+      // Network error - retry with exponential backoff
+      var retryDelay = INITIAL_RETRY_DELAY * Math.pow(2, retryCount)
+      console.log('Network error. Retrying in ' + (retryDelay / 1000) + ' seconds...')
+      sleep(retryDelay).then(function () {
+        downloadWithRetry(uri, filename, changeExitCodeOnError, callback, retryCount + 1)
+      })
+    } else {
+      trace.forEach(function (line) {
+        console.error(line)
+      })
+      var error = 'Failed to download ' + filename + ': ' + err.message
+      console.error(error)
+
+      if (changeExitCodeOnError) {
+        process.exitCode = 1
+      }
+
+      if (callback) {
+        callback(error)
+      }
+    }
+  })
+}
+
+module.exports = function download (uri, filename, changeExitCodeOnError, callback) {
+  downloadWithRetry(uri, filename, changeExitCodeOnError, callback, 0)
+}

package/scripts/get-vizjs.js

@@ -0,0 +1,51 @@
+#!/usr/bin/env node
+'use strict'
+
+var fs = require('fs')
+var path = require('path')
+var download = require('./download')
+
+var JAR_DIR_PATH = path.join(__dirname, '../vendor')
+var VIZJS_JAR = path.join(JAR_DIR_PATH, 'vizjs.jar')
+var J2V8_WIN_JAR = path.join(JAR_DIR_PATH, 'j2v8_win32_x86_64-3.1.6.jar')
+var J2V8_LINUX_JAR = path.join(JAR_DIR_PATH, 'j2v8_linux_x86_64-3.1.6.jar')
+var J2V8_MAC_JAR = path.join(JAR_DIR_PATH, 'j2v8_macosx_x86_64-3.1.6.jar')
+
+var VIZJS_URL = 'http://beta.plantuml.net/vizjs.jar'
+var J2V8_WIN_URL = 'http://beta.plantuml.net/j2v8_win32_x86_64-3.1.6.jar'
+var J2V8_LINUX_URL = 'http://beta.plantuml.net/j2v8_linux_x86_64-3.1.6.jar'
+var J2V8_MAC_URL = 'http://beta.plantuml.net/j2v8_macosx_x86_64-3.1.6.jar'
+
+var plantuml = require('../lib/node-plantuml')
+
+if (!fs.existsSync(JAR_DIR_PATH)) {
+  fs.mkdirSync(JAR_DIR_PATH)
+}
+
+plantuml.testdot(function (isOk) {
+  if (isOk) {
+    console.info('graphviz was found on the system. Skipping download of vizjs.')
+  } else {
+    console.info('graphviz was not found on the system. Downloading vizjs instead. See http://plantuml.com/vizjs. This may take a few minutes.')
+
+    // download additional libraries for working without dot installed.
+    download(VIZJS_URL, VIZJS_JAR, false, function (err) {
+      if (!err) {
+        // also install the V8 engine just in case the currently installed Java does not have Nashorn
+        switch (process.platform) {
+          case 'win32':
+            download(J2V8_WIN_URL, J2V8_WIN_JAR, false)
+            break
+          case 'linux':
+            download(J2V8_LINUX_URL, J2V8_LINUX_JAR, false)
+            break
+          case 'darwin':
+            download(J2V8_MAC_URL, J2V8_MAC_JAR, false)
+            break
+          default:
+            console.error('Unsupported operating system for V8 jars.')
+        }
+      }
+    })
+  }
+})