--- /dev/null
+<?php
+
+namespace Drupal\media_entity\Form;
+
+use Drupal\Core\Entity\Query\QueryFactory;
+use Drupal\Core\Entity\EntityDeleteForm;
+use Drupal\Core\Form\FormStateInterface;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+/**
+ * Provides a form for media bundle deletion.
+ */
+class MediaBundleDeleteConfirm extends EntityDeleteForm {
+
+ /**
+ * The query factory to create entity queries.
+ *
+ * @var \Drupal\Core\Entity\Query\QueryFactory
+ */
+ protected $queryFactory;
+
+ /**
+ * Constructs a new MediaBundleDeleteConfirm object.
+ *
+ * @param \Drupal\Core\Entity\Query\QueryFactory $query_factory
+ * The entity query object.
+ */
+ public function __construct(QueryFactory $query_factory) {
+ $this->queryFactory = $query_factory;
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public static function create(ContainerInterface $container) {
+ return new static(
+ $container->get('entity.query')
+ );
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public function buildForm(array $form, FormStateInterface $form_state) {
+ $num_entities = $this->queryFactory->get('media')
+ ->condition('bundle', $this->entity->id())
+ ->count()
+ ->execute();
+ if ($num_entities) {
+ $caption = '<p>' . $this->formatPlural($num_entities, '%type is used by 1 piece of content on your site. You can not remove this content type until you have removed all of the %type content.', '%type is used by @count pieces of content on your site. You may not remove %type until you have removed all of the %type content.', ['%type' => $this->entity->label()]) . '</p>';
+ $form['#title'] = $this->getQuestion();
+ $form['description'] = ['#markup' => $caption];
+ return $form;
+ }
+
+ return parent::buildForm($form, $form_state);
+ }
+
+}