Further modules included.

[yaffs-website] / web / modules / contrib / drupalmoduleupgrader / src / Plugin / DMU / Converter / Functions / DrupalWriteRecord.php

diff --git a/web/modules/contrib/drupalmoduleupgrader/src/Plugin/DMU/Converter/Functions/DrupalWriteRecord.php b/web/modules/contrib/drupalmoduleupgrader/src/Plugin/DMU/Converter/Functions/DrupalWriteRecord.php
new file mode 100644 (file)
index 0000000..c836a89
--- /dev/null
+++ b/web/modules/contrib/drupalmoduleupgrader/src/Plugin/DMU/Converter/Functions/DrupalWriteRecord.php
@@ -0,0 +1,48 @@
+<?php
+
+namespace Drupal\drupalmoduleupgrader\Plugin\DMU\Converter\Functions;
+
+use Drupal\drupalmoduleupgrader\TargetInterface;
+use Pharborist\Functions\FunctionCallNode;
+use Pharborist\Objects\ClassMethodCallNode;
+use Pharborist\Types\ArrayNode;
+use Pharborist\Types\StringNode;
+
+/**
+ * @Converter(
+ *  id = "drupal_write_record",
+ *  description = @Translation("Rewrites calls to drupal_write_record().")
+ * )
+ */
+class DrupalWriteRecord extends FunctionCallModifier {
+
+  /**
+   * {@inheritdoc}
+   */
+  public function rewrite(FunctionCallNode $call, TargetInterface $target) {
+    $rewritten = ClassMethodCallNode::create('\Drupal', 'database');
+
+    $arguments = $call->getArguments();
+    if (sizeof($arguments) == 3) {
+      $key = $arguments[2] instanceof StringNode ? ArrayNode::create([ clone $arguments[2] ]) : clone $arguments[2];
+
+      return $rewritten
+        ->appendMethodCall('merge')
+        ->appendArgument(clone $arguments[0])
+        ->appendMethodCall('fields')
+        ->appendArgument(clone $arguments[1])
+        ->appendMethodCall('key')
+        ->appendArgument($key)
+        ->appendMethodCall('execute');
+    }
+    else {
+      return $rewritten
+        ->appendMethodCall('insert')
+        ->appendArgument(clone $arguments[0])
+        ->appendMethodCall('fields')
+        ->appendArgument(clone $arguments[1])
+        ->appendMethodCall('execute');
+    }
+  }
+
+}