Доработка

cmd/mtdown/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"os"
 
 	"github.com/google/uuid"
 	"gitstore.ru/tolikproh/mchstest/internal/cnst"
@@ -10,24 +11,35 @@ import (
 )
 
 func main() {
-	fmt.Println("Загрузка вопросов для тестирования с сайта:", cnst.Url)
+	fmt.Println("Загрузка вопросов для тестирования с сайта: https://digital.mchs.gov.ru/fgpn/simulator")
 	uidMchs, err := uuid.Parse(util.InputTerminal("Необходимо ввести UUID сессии для начала загрузки"))
-	if err != nil {
-		fmt.Println("ОШИБКА: формат UUID не верный. Зазрузка отменена.")
-	} else {
-		var rN model.ReqNull
-		var rA model.ReqAnswer
-		body, _ := util.ReqestToSiteJSON(rN, cnst.Url, uidMchs)
-		fmt.Println("[")
-		for i := 0; i < 520; i++ {
-			d := model.UnmarshalJSONToMTType(body)
-			fmt.Print(model.MTTypeToJSON(d), ",\n")
-			rA.Answer = d.ValidAnswer
-			body, _ = util.ReqestToSiteJSON(rA, cnst.Url, uidMchs)
-		}
+	util.Check(err, "ОШИБКА: формат UUID не верный. Зазрузка отменена.")
+	filename := util.InputTerminal("Введите имя файла для сохранения")
+	util.Check(err, "")
+
+	var rN model.ReqNull
+	var rA model.ReqAnswer
+	f, err := os.Create(filename + ".json")
+	util.Check(err, "")
+	defer f.Close()
+
+	body, err := util.ReqestToSiteJSON(rN, cnst.Url, uidMchs)
+	util.Check(err, "")
+	_, err = f.WriteString("[\n")
+	util.Check(err, "")
+	fmt.Print("Выполняется сохранение")
+	for i := 0; i < 520; i++ {
 		d := model.UnmarshalJSONToMTType(body)
-		fmt.Println(model.MTTypeToJSON(d))
-		fmt.Println("]")
+		_, err = f.WriteString(model.MTTypeToJSON(d) + ",\n")
+		util.Check(err, "")
+		rA.Answer = d.ValidAnswer
+		body, err = util.ReqestToSiteJSON(rA, cnst.Url, uidMchs)
+		util.Check(err, "")
+		fmt.Print(".")
 	}
+	d := model.UnmarshalJSONToMTType(body)
+	_, err = f.WriteString(model.MTTypeToJSON(d) + "\n]")
+	util.Check(err, "")
+	fmt.Println("\nСохранение списка вопросов завершено в файле", filename+".json")
 
 }