Fixed base table reference in Annotations

src/Entity/Heartbeat.php

@@ -45,7 +45,7 @@ use Drupal\user\UserInterface;
  *   entity_keys = {
  *     "id" = "id",
  *     "revision" = "vid",
- *     "bundle" = "heartbeat_type",
+ *     "bundle" = "type",
  *     "label" = "name",
  *     "uuid" = "uuid",
  *     "uid" = "user_id",
@@ -324,4 +324,19 @@ class Heartbeat extends RevisionableContentEntityBase implements HeartbeatInterf
   }
 
 
+  /**
+   * Updates all heartbeat activities of one type to be of another type.
+   *
+   * @param string $old_id
+   *   The current heartbeat type of the activities.
+   * @param string $new_id
+   *   The new heartbeat type of the activities.
+   *
+   * @return
+   *   The number of activities whose heartbeat type field was modified.
+   */
+  function heartbeat_type_update_nodes($old_id, $new_id) {
+    return \Drupal::entityManager()->getStorage('heartbeat')->updateType($old_id, $new_id);
+  }
+
 }

src/Entity/HeartbeatType.php

@@ -3,6 +3,7 @@
 namespace Drupal\heartbeat8\Entity;
 
 use Drupal\Core\Config\Entity\ConfigEntityBundleBase;
+use Drupal\Core\Entity\EntityStorageInterface;
 use Drupal\heartbeat8\HeartbeatTypeListBuilder;
 
 /**
@@ -71,8 +72,6 @@ class HeartbeatType extends ConfigEntityBundleBase implements HeartbeatTypeInter
 
   public static function getHeartbeatTypeEntity($messageId) {
     $entity_manager = \Drupal::entityTypeManager();
-
-
   }
 
   public function setMessageId($messageId) {
@@ -270,10 +269,56 @@ class HeartbeatType extends ConfigEntityBundleBase implements HeartbeatTypeInter
     return parent::entityTypeManager(); // TODO: Change the autogenerated stub
   }
 
+
+  /**
+   * {@inheritdoc}
+   */
+  public function isLocked() {
+    $locked = \Drupal::state()->get('heartbeat.type.locked');
+    return isset($locked[$this->id()]) ? $locked[$this->id()] : FALSE;
+  }
+
+
+  /**
+   *
+   */
+
   public function loadHeartbeatType() {
     $this->entityTypeManager->getDefinitions();
   }
 
-  public static function getList() {
+  /**
+   * {@inheritdoc}
+   */
+  public function postSave(EntityStorageInterface $storage, $update = TRUE) {
+    parent::postSave($storage, $update);
+
+    if ($update && $this->getOriginalId() != $this->id()) {
+      $update_count = heartbeat_type_update_heartbeats($this->getOriginalId(), $this->id());
+      if ($update_count) {
+        drupal_set_message(\Drupal::translation()->formatPlural($update_count,
+          'Changed the heartbeat type of 1 activity from %old-type to %type.',
+          'Changed the heartbeat type of @count activities from %old-type to %type.',
+          [
+            '%old-type' => $this->getOriginalId(),
+            '%type' => $this->id(),
+          ]));
+      }
+    }
+    if ($update) {
+      // Clear the cached field definitions as some settings affect the field
+      // definitions.
+      $this->entityManager()->clearCachedFieldDefinitions();
+    }
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public static function postDelete(EntityStorageInterface $storage, array $entities) {
+    parent::postDelete($storage, $entities);
+
+    // Clear the heartbeat type cache to reflect the removal.
+    $storage->resetCache(array_keys($entities));
   }
 }