reapply modifications as per patch

Closes #3158196

src/DraggableViews.php

@@ -3,7 +3,6 @@
 namespace Drupal\draggableviews;
 
 use Drupal\Component\Utility\Html;
-use Drupal\media\Entity\Media;
 use Drupal\views\ViewExecutable;
 
 /**
@@ -33,15 +32,7 @@ class DraggableViews {
    */
   public function getIndex($id) {
     foreach ($this->view->result as $item) {
-      if ($item->_entity instanceof Media) {
-        // Media index name.
-        $name = 'mid';
-      }
-      else {
-        // Node index name. The default one.
-        $name = 'nid';
-      }
-      if ($item->$name == $id) {
+      if ($item->_entity->id() == $id) {
         return $item->index;
       }
     }