update: 修复输出

imiphp · Oct 31, 2023 · ffa629c · ffa629c
1 parent ca0a48a
commit ffa629c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/Command/AnnotationMigrationCommand.php b/src/Command/AnnotationMigrationCommand.php
@@ -119,7 +119,12 @@ protected function execute(InputInterface $input, OutputInterface $output)
             if ($handle->isModified()) {
                 $output->writeln("Rewrite\t{$item->getRealPath()}");
                 if (!$isDryRun) {
-                    file_put_contents($item->getRealPath(), $handle->rewriteCode());
+                    if ($isAnnotationRewrite) {
+                        $code = $handle->execPrintStmt();
+                    } else {
+                        $code = $handle->rewriteCode();
+                    }
+                    file_put_contents($item->getRealPath(), $code);
                 }
             } else {
                 $output->writeln("Skip\t{$item->getRealPath()}");