include subcommands in bash completions via nested case statements

Agent-Logs-Url: https://github.com/oscript-library/ovm/sessions/bd06fa86-ee67-41b7-8808-5df53c01d281

Co-authored-by: nixel2007 <1132840+nixel2007@users.noreply.github.com>

Author: Copilot

src/cmd/Классы/КомандаCompletions.os

@@ -29,22 +29,56 @@
 
 Функция СкриптАвтодополненияBash()
 
-	СписокКоманд = ПолучитьСписокКоманд();
+	ДеревоКоманд = ПолучитьДеревоКоманд();
 
 	МассивИменВсехКоманд = Новый Массив();
 	СтрокиCase = Новый Массив();
 
-	Для Каждого ОписаниеКоманды Из СписокКоманд Цикл
+	Для Каждого ОписаниеКоманды Из ДеревоКоманд Цикл
 		Для Каждого ИмяКоманды Из ОписаниеКоманды.Имена Цикл
 			МассивИменВсехКоманд.Добавить(ИмяКоманды);
 		КонецЦикла;
 
 		ПаттернCase = СтрСоединить(ОписаниеКоманды.Имена, "|");
-		СтрокаОпций = СтрСоединить(ОписаниеКоманды.Опции, " ");
 
-		СтрокиCase.Добавить("        " + ПаттернCase + ")");
-		СтрокиCase.Добавить("            COMPREPLY=($(compgen -W """ + СтрокаОпций + """ -- ""$cur""))");
-		СтрокиCase.Добавить("            ;;");
+		Если ОписаниеКоманды.Подкоманды.Количество() > 0 Тогда
+			МассивИменПодкоманд = Новый Массив();
+			МассивПодкомандCase = Новый Массив();
+
+			Для Каждого ОписаниеПодкоманды Из ОписаниеКоманды.Подкоманды Цикл
+				Для Каждого ИмяПодкоманды Из ОписаниеПодкоманды.Имена Цикл
+					МассивИменПодкоманд.Добавить(ИмяПодкоманды);
+				КонецЦикла;
+
+				ПаттернПодкомандCase = СтрСоединить(ОписаниеПодкоманды.Имена, "|");
+				СтрокаОпцийПодкоманды = СтрСоединить(ОписаниеПодкоманды.Опции, " ");
+
+				МассивПодкомандCase.Добавить("                " + ПаттернПодкомандCase + ")");
+				МассивПодкомандCase.Добавить("                    COMPREPLY=($(compgen -W """ + СтрокаОпцийПодкоманды + """ -- ""$cur""))");
+				МассивПодкомандCase.Добавить("                    ;;");
+			КонецЦикла;
+
+			ИменаПодкоманд = СтрСоединить(МассивИменПодкоманд, " ");
+			СтрокаОпцийРодителя = СтрСоединить(ОписаниеКоманды.Опции, " ");
+			ТелоПодкомандCase = СтрСоединить(МассивПодкомандCase, Символы.ПС);
+
+			СтрокиCase.Добавить("        " + ПаттернCase + ")");
+			СтрокиCase.Добавить("            if [ $cword -eq 2 ]; then");
+			СтрокиCase.Добавить("                COMPREPLY=($(compgen -W """ + ИменаПодкоманд + " " + СтрокаОпцийРодителя + """ -- ""$cur""))");
+			СтрокиCase.Добавить("            else");
+			СтрокиCase.Добавить("                local subcommand=""${words[2]}""");
+			СтрокиCase.Добавить("                case ""$subcommand"" in");
+			СтрокиCase.Добавить(ТелоПодкомандCase);
+			СтрокиCase.Добавить("                esac");
+			СтрокиCase.Добавить("            fi");
+			СтрокиCase.Добавить("            ;;");
+		Иначе
+			СтрокаОпций = СтрСоединить(ОписаниеКоманды.Опции, " ");
+
+			СтрокиCase.Добавить("        " + ПаттернCase + ")");
+			СтрокиCase.Добавить("            COMPREPLY=($(compgen -W """ + СтрокаОпций + """ -- ""$cur""))");
+			СтрокиCase.Добавить("            ;;");
+		КонецЕсли;
 	КонецЦикла;
 
 	ВсеКоманды = СтрСоединить(МассивИменВсехКоманд, " ");
@@ -89,21 +123,18 @@ complete -F _ovm_completions ovm";
 
 КонецФункции
 
-Функция ПолучитьСписокКоманд()
+Функция ПолучитьДеревоКоманд()
 
 	ОпределенияКоманд = Поделка.ПолучитьОпределенияЖелудей("КомандаПриложения");
 
-	Результат = Новый Массив();
+	// Первый проход: собрать все команды с их именами желудей и родителями
+	ВсеИнфо = Новый Соответствие();
 
 	Для Каждого КлючЗначение Из ОпределенияКоманд Цикл
+		ИмяЖелудя = КлючЗначение.Ключ;
 		ОпределениеЖелудя = КлючЗначение.Значение;
 		Конструктор = ОпределениеЖелудя.Завязь().ДанныеМетода();
 
-		АннотацияПодкоманды = РаботаСАннотациями.НайтиАннотацию(Конструктор.Аннотации, "ПодкомандаПриложения");
-		Если АннотацияПодкоманды <> Неопределено Тогда
-			Продолжить;
-		КонецЕсли;
-
 		АннотацияКоманды = РаботаСАннотациями.НайтиАннотацию(Конструктор.Аннотации, "КомандаПриложения");
 		Если АннотацияКоманды = Неопределено Тогда
 			Продолжить;
@@ -114,12 +145,33 @@ complete -F _ovm_completions ovm";
 			Продолжить;
 		КонецЕсли;
 
-		ЧастиИмени = СтрРазделить(ИмяКоманды, " ");
-		Опции = ПолучитьОпцииКоманды(ОпределениеЖелудя);
+		Родитель = РаботаСАннотациями.ПолучитьЗначениеПараметраАннотации(АннотацияКоманды, "Родитель", "");
+
+		ИнфоКоманды = Новый Структура("ИмяЖелудя, Родитель, Имена, Опции");
+		ИнфоКоманды.ИмяЖелудя = ИмяЖелудя;
+		ИнфоКоманды.Родитель = Родитель;
+		ИнфоКоманды.Имена = СтрРазделить(ИмяКоманды, " ");
+		ИнфоКоманды.Опции = ПолучитьОпцииКоманды(ОпределениеЖелудя);
+
+		ВсеИнфо.Вставить(ИмяЖелудя, ИнфоКоманды);
+	КонецЦикла;
+
+	// Второй проход: собрать дерево — верхнеуровневые команды со своими подкомандами
+	Результат = Новый Массив();
+
+	Для Каждого КлючЗначение Из ВсеИнфо Цикл
+		ИнфоКоманды = КлючЗначение.Значение;
+
+		Если ЗначениеЗаполнено(ИнфоКоманды.Родитель) Тогда
+			Продолжить;
+		КонецЕсли;
+
+		Подкоманды = НайтиПодкоманды(ИнфоКоманды.ИмяЖелудя, ВсеИнфо);
 
-		ОписаниеКоманды = Новый Структура("Имена, Опции");
-		ОписаниеКоманды.Имена = ЧастиИмени;
-		ОписаниеКоманды.Опции = Опции;
+		ОписаниеКоманды = Новый Структура("Имена, Опции, Подкоманды");
+		ОписаниеКоманды.Имена = ИнфоКоманды.Имена;
+		ОписаниеКоманды.Опции = ИнфоКоманды.Опции;
+		ОписаниеКоманды.Подкоманды = Подкоманды;
 
 		Результат.Добавить(ОписаниеКоманды);
 	КонецЦикла;
@@ -128,6 +180,25 @@ complete -F _ovm_completions ovm";
 
 КонецФункции
 
+Функция НайтиПодкоманды(ИмяРодителя, ВсеИнфо)
+
+	Результат = Новый Массив();
+
+	Для Каждого КлючЗначение Из ВсеИнфо Цикл
+		ИнфоКоманды = КлючЗначение.Значение;
+
+		Если НРег(ИнфоКоманды.Родитель) = НРег(ИмяРодителя) Тогда
+			ОписаниеПодкоманды = Новый Структура("Имена, Опции");
+			ОписаниеПодкоманды.Имена = ИнфоКоманды.Имена;
+			ОписаниеПодкоманды.Опции = ИнфоКоманды.Опции;
+			Результат.Добавить(ОписаниеПодкоманды);
+		КонецЕсли;
+	КонецЦикла;
+
+	Возврат Результат;
+
+КонецФункции
+
 Функция ПолучитьОпцииКоманды(ОпределениеЖелудя)
 
 	Опции = Новый Массив();