Version 1
[yaffs-website] / web / core / modules / node / src / Plugin / migrate / source / d6 / ViewModeBase.php
diff --git a/web/core/modules/node/src/Plugin/migrate/source/d6/ViewModeBase.php b/web/core/modules/node/src/Plugin/migrate/source/d6/ViewModeBase.php
new file mode 100644 (file)
index 0000000..315f069
--- /dev/null
@@ -0,0 +1,48 @@
+<?php
+
+namespace Drupal\node\Plugin\migrate\source\d6;
+
+use Drupal\migrate_drupal\Plugin\migrate\source\DrupalSqlBase;
+
+/**
+ * A base class for migrations that require view mode info.
+ */
+abstract class ViewModeBase extends DrupalSqlBase {
+
+  /**
+   * {@inheritdoc}
+   */
+  public function count() {
+    return count($this->initializeIterator());
+  }
+
+  /**
+   * Get a list of D6 view modes.
+   *
+   * Drupal 6 supported the following view modes.
+   * NODE_BUILD_NORMAL = 0
+   * NODE_BUILD_PREVIEW = 1
+   * NODE_BUILD_SEARCH_INDEX = 2
+   * NODE_BUILD_SEARCH_RESULT = 3
+   * NODE_BUILD_RSS = 4
+   * NODE_BUILD_PRINT = 5
+   * teaser
+   * full
+   *
+   * @return array
+   *   The view mode names.
+   */
+  public function getViewModes() {
+    return [
+      0,
+      1,
+      2,
+      3,
+      4,
+      5,
+      'teaser',
+      'full',
+    ];
+  }
+
+}