San 6078

lib/models/mongo/context-version.js

@@ -721,7 +721,7 @@ ContextVersionSchema.statics.updateAndGetFailedBuild = (buildId, errorMessage) =
  * @param {string}   buildId - build id associated with context version
  * @return {Promise}
  */
-ContextVersionSchema.statics.updateAndGetSuccessfulBuild = (buildId) => {
+ContextVersionSchema.statics.updateAndGetSuccessfulBuild = (buildId, imageData) => {
   var log = logger.child({
     buildId: buildId,
     method: 'updateAndGetSuccessfulBuild'
@@ -736,6 +736,10 @@ ContextVersionSchema.statics.updateAndGetSuccessfulBuild = (buildId) => {
     }
   }
 
+  if (imageData) {
+    update.$set.imageData = imageData
+  }
+
   return ContextVersion._updateByBuildIdAndEmit(buildId, update)
 }
 

lib/models/mongo/schemas/context-version.js

@@ -210,6 +210,34 @@ var ContextVersionSchema = module.exports = new Schema({
     },
     log: Schema.Types.Mixed,
     failed: Boolean
+  },
+  imageData: {
+    type: {
+      ports: {
+        type: [{
+          protocol: {
+            type: String
+          },
+          port: {
+            type: String
+          }
+        }],
+        'default': []
+      },
+      cmd: [{
+        type: String,
+        'default': []
+      }],
+      entryPoint: [{
+        type: String,
+        'default': []
+      }]
+    },
+    'default': {
+      ports: [],
+      cmd: [],
+      entryPoint: []
+    }
   }
 })
 

lib/models/services/build-service.js

@@ -337,11 +337,11 @@ BuildService._buildBuild = function (build, buildInfo, sessionUser) {
  * @param  {String}  buildId
  * @return {Promise}
  */
-BuildService.updateSuccessfulBuild = (buildId) => {
+BuildService.updateSuccessfulBuild = (buildId, imageData) => {
   var log = logger.child({ method: 'updateSuccessfulBuild' })
   log.info({ buildId }, 'updateSuccessfulBuild called')
 
-  return ContextVersion.updateAndGetSuccessfulBuild(buildId)
+  return ContextVersion.updateAndGetSuccessfulBuild(buildId, imageData)
     .tap((SuccessfullContextVersions) => {
       var contextVersionIds = SuccessfullContextVersions.map(pluck('_id'))
 

lib/models/services/image-service.js

@@ -0,0 +1,56 @@
+/**
+ * @module lib/models/services/image-service
+ */
+'use strict'
+
+require('loadenv')('models/services/infracode-version-service')
+
+const keypather = require('keypather')()
+const logger = require('logger')
+const ImageService = module.exports = {}
+
+ImageService.logger = logger.child({
+  module: 'ImageService'
+})
+
+ImageService.parsePorts = function (ports) {
+  let portList = []
+  let splitPort
+
+  if (ports) {
+    for (let key of Object.keys(ports)) {
+      splitPort = key.split('/')
+
+      if (splitPort.length === 2) {
+        portList.push({
+          protocol: splitPort[1],
+          port: splitPort[0]
+        })
+      }
+    }
+  }
+
+  return portList
+}
+
+ImageService.getImageDataFromJob = function (job) {
+  const log = ImageService.logger.child({
+    method: 'getImageDataFromJob'
+  })
+  log.info('called')
+
+  let rawImageData = keypather.get(job, 'inspectImageData.Config')
+  if (rawImageData) {
+    return {
+      port: ImageService.parsePorts(keypather.get(rawImageData, 'ExposedPorts')),
+      cmd: keypather.get(rawImageData, 'Cmd') || [],
+      entryPoint: keypather.get(rawImageData, 'Entrypoint') || []
+    }
+  }
+
+  return {
+    ports: [],
+    cmd: [],
+    entryPoint: []
+  }
+}

lib/workers/build.container.died.js

@@ -19,44 +19,27 @@ const Promise = require('bluebird')
 const error = require('error')
 const InstanceService = require('models/services/instance-service')
 const BuildService = require('models/services/build-service')
+const ImageService = require('models/services/image-service')
+
 const Isolation = require('models/mongo/isolation')
 const logger = require('logger')
 const rabbitMQ = require('models/rabbitmq')
 
-const ContainerImageBuilderDied = {}
-
-module.exports = ContainerImageBuilderDied
-
-ContainerImageBuilderDied.jobSchema = joi.object({
-  from: joi.string().required(),
-  host: joi.string().uri({ scheme: 'http' }).required(),
-  id: joi.string().required(),
-  time: joi.number().required(),
-  inspectData: joi.object({
-    Config: joi.object({
-      Labels: joi.object({
-        'contextVersion.build._id': joi.string().required(),
-        'ownerUsername': joi.string().required(),
-        'sessionUserGithubId': joi.number().required()
-      }).unknown().required()
-    }).unknown().required()
-  }).unknown().required()
-}).unknown().required()
-
 class Worker {
   constructor (job) {
+    this.log = logger.child({
+      job: job,
+      module: 'ContainerImageBuilderDied'
+    })
+
     this.host = job.host
     this.id = job.id
+    this.inspectImageData = ImageService.getImageDataFromJob(job)
     this.inspectData = job.inspectData
     this.contextVersionBuildId = job.inspectData.Config.Labels['contextVersion.build._id']
     this.ownerUsername = job.inspectData.Config.Labels.ownerUsername
     this.sessionUserGithubId = job.inspectData.Config.Labels.sessionUserGithubId
     this.dockerImageTag = job.inspectData.Config.Labels.dockerTag
-
-    this.log = logger.child({
-      job: job,
-      module: 'ContainerImageBuilderDied'
-    })
   }
 
   /**
@@ -103,7 +86,7 @@ class Worker {
       imageTag: this.dockerImageTag
     })
 
-    return BuildService.updateSuccessfulBuild(this.contextVersionBuildId)
+    return BuildService.updateSuccessfulBuild(this.contextVersionBuildId, this.inspectImageData)
   }
 
   /**
@@ -323,6 +306,16 @@ module.exports = {
     from: joi.string().required(),
     host: joi.string().uri({ scheme: 'http' }).required(),
     id: joi.string().required(),
+    time: joi.number().required(),
+    inspectImageData: joi.object({
+      exposedPorts: joi.array(joi.object({}).unknown()).default([]),
+      cmd: joi.array(joi.string()).default([]),
+      entrypoint: joi.array(joi.string()).default([])
+    }).default({
+      exposedPorts: [],
+      cmd: [],
+      entrypoint: []
+    }).unknown().required(),
     inspectData: joi.object({
       Config: joi.object({
         Labels: joi.object({

unit/models/mongo/context-version.js

@@ -309,7 +309,7 @@ describe('Context Version Unit Test', function () {
       done()
     })
 
-    it('should save a successful build', (done) => {
+    it('should save a successful build with no image data', (done) => {
       const buildId = 12341
 
       ContextVersion.updateAndGetSuccessfulBuild(buildId).asCallback(() => {
@@ -329,6 +329,36 @@ describe('Context Version Unit Test', function () {
         done()
       })
     })
+
+    it('should save a successful build with image data', (done) => {
+      const buildId = 12341
+      const imageData = {
+        ports: [{
+          protocol: 'Omega',
+          port: '13'
+        }],
+        cmd: ['cmd 1'],
+        entryPoint: ['entry 1']
+      }
+
+      ContextVersion.updateAndGetSuccessfulBuild(buildId, imageData).asCallback(() => {
+        sinon.assert.calledOnce(ContextVersion.updateByAsync)
+        sinon.assert.calledWith(ContextVersion.updateByAsync,
+          'build._id',
+          buildId, {
+            $set: {
+              'build.failed': false,
+              'build.completed': sinon.match.number,
+              'state': ContextVersion.states.buildSucceeded,
+              'imageData': imageData
+            }
+          })
+
+        sinon.assert.calledOnce(ContextVersion.findByAsync)
+        sinon.assert.calledWith(ContextVersion.findByAsync, 'build._id', buildId)
+        done()
+      })
+    })
   }) // end updateAndGetSuccessfulBuild
 
   describe('findOneCreating', function () {