Fixed bug which caused all non "node" related heartbeatTypes to save

heartbeat.module

@@ -96,41 +96,43 @@ function heartbeat_entity_insert(EntityInterface $entity) {
         foreach ($heartbeatTypeService->getTypes() as $type) {
 
           $heartbeatTypeEntity = $heartbeatTypeService->load($type);
-          $arguments = json_decode($heartbeatTypeEntity->getArguments());
 
-          foreach ($arguments as $key => $argument) {
-            $variables[$key] = $argument;
-          }
+          if ($heartbeatTypeEntity->getMainEntity() == 'node') {
 
-          $preparsedMessageString = strtr($heartbeatTypeEntity->getMessage(), $variables);
+            $arguments = json_decode($heartbeatTypeEntity->getArguments());
 
-          $entities = array(
-            'node' => $entity,
-            'user' => $user,
-          );
+            foreach ($arguments as $key => $argument) {
+              $variables[$key] = $argument;
+            }
 
-          if (array_key_exists($heartbeatTypeEntity->getMainEntity(), $heartbeatTypeService->getEntityBundles($entity->getEntityType()))) {
-$stophere = null;
-          }
+            $preparsedMessageString = strtr($heartbeatTypeEntity->getMessage(), $variables);
+
+            $entities = array(
+              'node' => $entity,
+              'user' => $user,
+            );
 
+            if (array_key_exists($heartbeatTypeEntity->getMainEntity(), $heartbeatTypeService->getEntityBundles($entity->getEntityType()))) {
+    //TODO handle logic for bundle-specific Heartbeat types based on the Node entity
+            }
 
-          $heartbeatMessage = Heartbeat::buildMessage($tokenService, $preparsedMessageString, $entities, $entity->getEntityTypeId(), $media);
+            $heartbeatMessage = Heartbeat::buildMessage($tokenService, $preparsedMessageString, $entities, $entity->getEntityTypeId(), $media);
 
-          //      $translatedMessage = t($messageTemplate);
+            //      $translatedMessage = t($messageTemplate);
 
-          $heartbeatActivity = Heartbeat::create([
-            'type' => $heartbeatTypeEntity->id(),
-            'uid' => $user->id(),
-            'nid' => $entity->id(),
-            'name' => 'Dev Test',
-          ]);
+            $heartbeatActivity = Heartbeat::create([
+              'type' => $heartbeatTypeEntity->id(),
+              'uid' => $user->id(),
+              'nid' => $entity->id(),
+              'name' => 'Dev Test',
+            ]);
 
-          $heartbeatActivity->setMessage($heartbeatMessage);
-          $heartbeatActivity->save();
+            $heartbeatActivity->setMessage($heartbeatMessage);
+            $heartbeatActivity->save();
 
+          }
         }
       }
-
       break;
     case $entity instanceof \Drupal\user\Entity\User:
 //      $userService = \Drupal\User\Entity\user