package controller

import (
	"encoding/json"
	"fmt"
	"github.com/gin-gonic/gin"
	"hy2xs-admin/model/constant"
	"hy2xs-admin/model/dto"
	"hy2xs-admin/model/vo"
	"hy2xs-admin/service"
	"hy2xs-admin/util"
	"time"
)

func LogSystem(c *gin.Context) {
	logSystemDto, err := validateField(c, dto.LogDto{})
	if err != nil {
		return
	}
	logSystemVos := make([]vo.LogSystemVo, 0)

	// Форма ответа одна для ВСЕХ веток: {records, total}.
	//
	// Раньше ветка «файла ещё нет» отвечала голым массивом:
	//
	//	vo.Success(logSystemVos, c)
	//
	// Панель читает `data.records`, поэтому получала undefined и передавала
	// его в `:data` таблицы. То есть на свежепоставленном хосте — до первой
	// записи в журнал — страница системных логов была сломана целиком, и
	// именно этот сценарий воспроизводится на каждой приёмке чистой установки.
	if !util.Exists(constant.SystemLogPath) {
		vo.Success(vo.LogSystemPage[vo.LogSystemVo]{LogSystemVos: logSystemVos, Total: 0}, c)
		return
	}
	numLine := 0
	if logSystemDto.NumLine != nil && *logSystemDto.NumLine > 0 {
		numLine = *logSystemDto.NumLine
	}
	logLines, total, err := util.ReadLinesFromBottom(constant.SystemLogPath, numLine)
	if err != nil {
		vo.Fail("Unable to read log file", c)
		return
	}

	for _, line := range logLines {
		if line == "" {
			continue
		}
		logSystemVo := vo.LogSystemVo{}
		// Неразбираемая строка ПРОПУСКАЕТСЯ, а не отказывает запросу.
		//
		// Было так:
		//
		//	if err != nil {
		//	    vo.Fail("Unable to unmarshal log data", c)
		//	    continue
		//	}
		//
		// — ответ записывался в поток и цикл шёл дальше, а в конце
		// безусловно выполнялся vo.Success. Клиент получал два JSON-документа
		// подряд, то есть невалидный ответ, и gin ругался на повторную запись
		// заголовков. Достаточно было ОДНОЙ битой строки, чтобы страница
		// перестала открываться целиком.
		//
		// Строка при этом бьётся штатно: lumberjack ротирует файл, и обрыв
		// последней записи на границе ротации — обычное событие, а не признак
		// повреждения журнала. Отказывать из-за него всей выдаче нечем
		// оправдать: остальные записи прочитаны и полезны.
		if err := json.Unmarshal([]byte(line), &logSystemVo); err != nil {
			continue
		}
		// Собственный журнал санитизируется так же, как чужой.
		//
		// Раньше через SanitizeLogText проходил только журнал Hysteria: он
		// «чужой», а свой мы якобы контролируем. Контроль этот держался на
		// внимательности — ровно до logrus.Warnf с паролем администратора в
		// dao/sqlite.go. Санитайз здесь стоит не вместо аккуратности в местах
		// записи, а на случай следующего такого места.
		logSystemVo.Msg = service.SanitizeLogText(logSystemVo.Msg)
		logSystemVos = append(logSystemVos, logSystemVo)
	}
	vo.Success(vo.LogSystemPage[vo.LogSystemVo]{
		LogSystemVos: logSystemVos,
		Total:        int64(total),
	}, c)
}

func LogHysteria2(c *gin.Context) {
	logSystemDto, err := validateField(c, dto.LogDto{})
	if err != nil {
		return
	}
	logHysteria2Vos := make([]vo.LogHysteria2Vo, 0)
	numLine := 0
	if logSystemDto.NumLine != nil && *logSystemDto.NumLine > 0 {
		numLine = *logSystemDto.NumLine
	}
	logHysteria2Vos, total, err := service.ReadHysteriaJournalLogs(numLine)
	if err != nil {
		vo.Fail("Unable to read hysteria journal logs", c)
		return
	}

	vo.Success(vo.LogSystemPage[vo.LogHysteria2Vo]{
		LogSystemVos: logHysteria2Vos,
		Total:        int64(total),
	}, c)
}

func ExportLog(c *gin.Context) {
	logExportDto, err := validateField(c, dto.LogExportDto{})
	if err != nil {
		return
	}

	var fileName string
	if *logExportDto.Option == 0 {
		fileName = fmt.Sprintf("hy2xs-admin-%s.log", time.Now().Format("20060102150405"))

		// Журнал админки отдаётся санитизированным, а не файлом «как есть».
		//
		// Раньше здесь стоял c.File(constant.SystemLogPath): оператор скачивал
		// сырой /var/log/hy2xs/hy2xs-admin.log. Журнал Hysteria при этом
		// проходил через SanitizeLogText — то есть чужому журналу продукт не
		// доверял, а своему доверял. Асимметрия ничем не обоснована: файл в
		// обоих случаях покидает сервер и пересылается в переписке.
		output, exportErr := service.ExportAdminLog()
		if exportErr != nil {
			vo.Fail("log file not exist", c)
			return
		}
		c.Header("Content-Type", "text/plain; charset=utf-8")
		c.Header("Content-Transfer-Encoding", "binary")
		c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=%s", fileName))
		c.Data(200, "text/plain; charset=utf-8", []byte(output))
		return
	} else if *logExportDto.Option == 1 {
		fileName = fmt.Sprintf("hysteria2-%s.log", time.Now().Format("20060102150405"))
		output, exportErr := service.ExportHysteriaJournalLogs(5000)
		if exportErr != nil {
			vo.Fail("failed to export hysteria journal logs", c)
			return
		}
		c.Header("Content-Type", "text/plain; charset=utf-8")
		c.Header("Content-Transfer-Encoding", "binary")
		c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=%s", fileName))
		c.Data(200, "text/plain; charset=utf-8", []byte(output))
		return
	}

	// Обе ветки выше завершаются return. Сюда попадает только неизвестное
	// значение option, и это отказ, а не отдача пустого файла: раньше здесь
	// оставался путь с пустым filePath, который сводился к тому же сообщению
	// окольной дорогой.
	vo.Fail("unsupported log export option", c)
}