From ffb705fe2fc9dcc325f072d47d46cc13a94d5eb5 Mon Sep 17 00:00:00 2001
From: Dmitrii Gridin <dmitry.gridin@jetbrains.com>
Date: Thu, 13 Apr 2023 15:16:46 +0200
Subject: [PATCH] [AA FIR] add missing new line to addImportToFile

^KTIJ-24928
---
 .../kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt  | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt b/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt
index 1c0174deeee..28b0a9262da 100644
--- a/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt
+++ b/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt
@@ -59,7 +59,10 @@ fun addImportToFile(
     val newDirective = psiFactory.createImportDirective(importPath)
     val imports = importList.imports
     if (imports.isEmpty()) {
-        importList.add(psiFactory.createNewLine(2))
+        file.packageDirective?.takeIf { it.packageKeyword != null }?.let {
+            file.addAfter(psiFactory.createNewLine(2), it)
+        }
+
         importList.add(newDirective)
     } else {
         val insertAfter = imports.lastOrNull {