+<?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');
+ }
+ }
+
+}