Add 'type' member variable to BaseService and integrate object STIX t…

…ype validation check to 'create' method

app/controllers/groups-controller.js

@@ -100,7 +100,7 @@ exports.create = async function(req, res) {
     // Create the group
     try {
 
-        const group = await groupsService.createGroup(groupData, options);
+        const group = await groupsService.create(groupData, options);
         logger.debug('Success: Created group with id ' + group.stix.id);
         return res.status(201).send(group);
     }

app/services/_base.service.js

@@ -8,13 +8,15 @@ const config = require('../config/config');
 const { DatabaseError,
     IdentityServiceError,
     MissingParameterError,
-    InvalidQueryStringParameterError } = require('../exceptions');
+    InvalidQueryStringParameterError,
+    InvalidTypeError } = require('../exceptions');
 const AbstractService = require('./_abstract.service');
 
 class BaseService extends AbstractService {
 
-    constructor(repository) {
+    constructor(type, repository) {
         super();
+        this.type = type;
         this.repository = repository;
     }
 
@@ -216,6 +218,11 @@ class BaseService extends AbstractService {
         if (BaseService.isCallback(arguments[arguments.length - 1])) {
             callback = arguments[arguments.length - 1];
         }
+
+        if (data?.stix?.type !== this.type) {
+            throw new InvalidTypeError();
+        }
+
         // eslint-disable-next-line no-useless-catch
         try {
             // This function handles two use cases:

app/services/assets-service.js

@@ -5,7 +5,6 @@ const assetsRepository = require('../repository/assets-repository');
 
 const BaseService = require('./_base.service');
 
-
 class AssetsService extends BaseService { }
 
-module.exports = new AssetsService(assetsRepository);
+module.exports = new AssetsService('x-mitre-asset', assetsRepository);

app/services/groups-service.js

@@ -2,21 +2,7 @@
 
 const BaseService = require('./_base.service');
 const groupsRepository = require('../repository/groups-repository');
-const { InvalidTypeError } = require('../exceptions');
 
-class GroupsService extends BaseService {
+class GroupsService extends BaseService { }
 
-    createGroup(data, options) {
-
-        // Overrides the base method for groups to inject an additional 
-        // logic check to verify that the creation request is not for an
-        // intrusion-set, which is an unsupported/invalid use case.  
-
-        if (data.stix.type !== 'intrusion-set') {
-            throw new InvalidTypeError();
-        }
-        return this.create(data, options);
-    }
-}
-
-module.exports = new GroupsService(groupsRepository);
+module.exports = new GroupsService('intrusion-set', groupsRepository);

app/services/matrices-service.js

@@ -9,8 +9,8 @@ const BaseService = require('./_base.service');
 
 class MatrixService extends BaseService {
 
-    constructor(repository) {
-        super(repository);
+    constructor(type, repository) {
+        super(type, repository);
 
         this.retrieveTacticById = null;
         this.retrieveTechniquesForTactic = null;
@@ -115,4 +115,4 @@ class MatrixService extends BaseService {
     }
 }
 
-module.exports = new MatrixService(matrixRepository);
+module.exports = new MatrixService('x-mitre-matrix', matrixRepository);

app/tests/api/groups/groups.query.spec.js

@@ -127,7 +127,7 @@ async function loadGroups(groups) {
         }
 
         // eslint-disable-next-line no-await-in-loop
-        await groupsService.createGroup(group, { import: false, userAccountId: group.userAccountId });
+        await groupsService.create(group, { import: false, userAccountId: group.userAccountId });
     }
 }