From 758e1ad96274cd9359d770d5526493a89c70b7e5 Mon Sep 17 00:00:00 2001
From: Elena Makarova <el-makarova@yandex-team.ru>
Date: Fri, 13 Sep 2024 11:28:42 +0300
Subject: [PATCH] fix(autocomplete): do not lose folder if its name is equals
 to database

---
 src/utils/monaco/yql/generateSuggestions.ts | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/utils/monaco/yql/generateSuggestions.ts b/src/utils/monaco/yql/generateSuggestions.ts
index 836a058e50..ef24722b58 100644
--- a/src/utils/monaco/yql/generateSuggestions.ts
+++ b/src/utils/monaco/yql/generateSuggestions.ts
@@ -65,7 +65,9 @@ const suggestionEntityToAutocomplete: Partial<Record<YQLEntity, AutocompleteEnti
     tableIndex: ['table_index', 'index'],
 };
 
-const commonSuggestionEntities: AutocompleteEntityType[] = ['dir', 'unknown'];
+const commonSuggestionEntities: AutocompleteEntityType[] = ['dir', 'unknown', 'ext_sub_domain'];
+
+const directoryTypes: AutocompleteEntityType[] = ['dir', 'ext_sub_domain'];
 
 function filterAutocompleteEntities(
     autocompleteEntities: TAutocompleteEntity[],
@@ -109,7 +111,11 @@ function removeStartSlash(value: string) {
 }
 
 function normalizeEntityPrefix(value = '', database: string) {
-    let cleanedValue = removeStartSlash(removeBackticks(value));
+    const valueWithoutBackticks = removeBackticks(value);
+    if (!valueWithoutBackticks.startsWith('/')) {
+        return valueWithoutBackticks;
+    }
+    let cleanedValue = removeStartSlash(valueWithoutBackticks);
     const cleanedDatabase = removeStartSlash(database);
     if (cleanedValue.startsWith(cleanedDatabase)) {
         cleanedValue = cleanedValue.slice(cleanedDatabase.length);
@@ -297,7 +303,7 @@ export async function generateEntitiesSuggestion(
     if (data.Success) {
         const filteredEntities = filterAutocompleteEntities(data.Result.Entities, suggestEntities);
         return filteredEntities.reduce((acc, {Name, Type}) => {
-            const isDir = Type === 'dir';
+            const isDir = directoryTypes.includes(Type);
             const label = isDir ? `${Name}/` : Name;
             let labelAsSnippet;
             if (isDir && !withBackticks) {