models + container management

Author: siloenix

app.js

@@ -1,22 +1,21 @@
 const express = require('express');
+const config = require('./config');
+const mongoose = require('mongoose');
 const fileUpload = require('express-fileupload');
 const createError = require('http-errors');
-const logger = require('morgan');
 const requestLogger = require('./middleware/requestLogger');
 const errorHandler = require('./middleware/errorHandler');
-
-const rootRouter = require('./routes/root');
-const dockerRouter = require('./routes/docker');
+const registerRoutes = require('./routes');
 
 const app = express();
 
-// app.use(logger('dev'));
+mongoose.connect(config.mongo.url, {useNewUrlParser: true});
+
 app.use(express.json());
 app.use(fileUpload());
 app.use(requestLogger);
 
-app.use('/', rootRouter);
-app.use('/docker', dockerRouter);
+registerRoutes(app);
 
 // catch 404 and forward to error handler
 app.use(function (req, res, next) {

config/default.json

@@ -1,5 +1,8 @@
 {
   "docker": {
     "socketPath": "/var/run/docker.sock"
+  },
+  "mongo": {
+    "url": "mongodb://localhost/docker-composer"
   }
 }

controllers/ContainerController.js

@@ -0,0 +1,43 @@
+const ContainerService = require('../services/ContainerService');
+const handler = require('../helpers/handler');
+const ValidationError = require('../errors/ValidationError');
+
+class ContainerController {
+  static async create(req, res) {
+    const result = await ContainerService.dao().create(req.body);
+    return res.json(result);
+  }
+
+  static async update(req, res) {
+    const id = retrieveId(req);
+    const result = await ContainerService.dao().update(id, req.body);
+    return res.json(result);
+  }
+
+  static async get(req, res) {
+    const id = retrieveId(req);
+    const result = await ContainerService.dao().findOne(id);
+    return res.json(result);
+  }
+
+  static async getAll(req, res) {
+    const result = await ContainerService.dao().findAll(req.body);
+    return res.json(result);
+  }
+
+  static async delete(req, res) {
+    const id = retrieveId(req);
+    const result = await ContainerService.dao().delete(id);
+    return res.json(result);
+  }
+}
+
+function retrieveId(req) {
+  const id = req.params.id;
+  if (!id) {
+    throw new ValidationError('id must be provided')
+  }
+  return id;
+}
+
+module.exports = handler(ContainerController);

dao/ContainerDao.js

@@ -0,0 +1,29 @@
+const Container = require('../models/Container');
+
+class ContainerDao {
+  static model() {
+    return Container;
+  }
+
+  static create(data) {
+    return Container.create(data);
+  }
+
+  static update(id, data) {
+    return Container.findByIdAndUpdate(id, data, {new: true}).exec();
+  }
+
+  static findOne(id) {
+    return Container.findById(id).exec()
+  }
+
+  static findAll(query = {}) {
+    return Container.find(query).exec()
+  }
+
+  static delete(id) {
+    return Container.findByIdAndDelete(id).exec()
+  }
+}
+
+module.exports = ContainerDao;

models/Container.js

@@ -0,0 +1,11 @@
+const {BaseSchema, mongoose, referenceField} = require('./index');
+
+const ContainerSchema = new BaseSchema({
+  name: String,
+  image: String,
+  ports: [Number],
+  volumes: [referenceField('Volume')],
+  networks: [referenceField('Network')]
+});
+
+module.exports = mongoose.model('Container', ContainerSchema);

models/Network.js

@@ -0,0 +1,15 @@
+const {BaseSchema, mongoose} = require('./index');
+
+
+const DriverOption = new BaseSchema({
+  name: String,
+  value: String
+});
+
+const NetworkSchema = new BaseSchema({
+  name: String,
+  driver: String,
+  driverOptions: [DriverOption],
+});
+
+module.exports = mongoose.model('Network', NetworkSchema);

models/Volume.js

@@ -0,0 +1,14 @@
+const {BaseSchema, mongoose} = require('./index');
+
+const DriverOption = new BaseSchema({
+  name: String,
+  value: String
+});
+
+const VolumeSchema = new BaseSchema({
+  name: String,
+  driver: String,
+  driverOptions: [DriverOption],
+});
+
+module.exports = mongoose.model('Volume', VolumeSchema);

models/index.js

@@ -0,0 +1,27 @@
+const mongoose = require('mongoose');
+const Schema = mongoose.Schema;
+
+class BaseSchema extends Schema {
+  constructor(def, opts = {}) {
+    super(...arguments);
+    this.set('toJSON', {
+      transform: function (doc, ret, options) {
+        // remove the _id of every document before returning the result
+        ret.id = ret._id;
+        delete ret._id;
+        delete ret.__v;
+      },
+      virtuals: true
+    });
+  }
+}
+
+function referenceField(model) {
+  return {type: Schema.Types.ObjectId, ref: model}
+}
+
+module.exports.BaseSchema = BaseSchema;
+
+module.exports.mongoose = mongoose;
+
+module.exports.referenceField = referenceField;

package.json

@@ -16,6 +16,7 @@
     "express-fileupload": "^1.0.0",
     "http-errors": "~1.6.2",
     "mocha": "^5.2.0",
+    "mongoose": "^5.3.9",
     "morgan": "~1.9.0",
     "nconf": "^0.10.0",
     "parser-yaml": "^0.1.1",

routes/container_route.js

@@ -0,0 +1,13 @@
+const router = require('express').Router();
+const ContainerController = require('../controllers/ContainerController');
+
+router.route('/')
+  .get(ContainerController.getAll)
+  .post(ContainerController.create);
+
+router.route('/:id')
+  .get(ContainerController.get)
+  .put(ContainerController.update)
+  .delete(ContainerController.delete);
+
+module.exports = router;