init

internal/service/main_service.go

@@ -0,0 +1,168 @@
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"equipment-analyzer/internal/capture"
+	"equipment-analyzer/internal/config"
+	"equipment-analyzer/internal/model"
+	"equipment-analyzer/internal/utils"
+)
+
+type App struct {
+	config         *config.Config
+	logger         *utils.Logger
+	captureService *CaptureService
+	parserService  *ParserService
+}
+
+func NewApp(cfg *config.Config, logger *utils.Logger) *App {
+	return &App{
+		config:         cfg,
+		logger:         logger,
+		captureService: NewCaptureService(cfg, logger),
+		parserService:  NewParserService(cfg, logger),
+	}
+}
+
+func (a *App) Startup(ctx context.Context) {
+	a.logger.Info("应用启动")
+}
+
+func (a *App) DomReady(ctx context.Context) {
+	a.logger.Info("DOM准备就绪")
+}
+
+func (a *App) BeforeClose(ctx context.Context) (prevent bool) {
+	a.logger.Info("应用即将关闭")
+	return false
+}
+
+func (a *App) Shutdown(ctx context.Context) {
+	a.logger.Info("应用关闭")
+}
+
+// GetNetworkInterfaces 获取网络接口列表
+func (a *App) GetNetworkInterfaces() ([]model.NetworkInterface, error) {
+	interfaces, err := capture.GetNetworkInterfaces()
+	if err != nil {
+		a.logger.Error("获取网络接口失败", "error", err)
+		return nil, err
+	}
+	return interfaces, nil
+}
+
+// StartCapture 开始抓包
+func (a *App) StartCapture(interfaceName string) error {
+	if a.captureService.IsCapturing() {
+		return fmt.Errorf("抓包已在进行中")
+	}
+
+	config := capture.Config{
+		InterfaceName: interfaceName,
+		Filter:        a.config.Capture.DefaultFilter,
+		Timeout:       time.Duration(a.config.Capture.DefaultTimeout) * time.Millisecond,
+		BufferSize:    a.config.Capture.BufferSize,
+	}
+
+	err := a.captureService.StartCapture(context.Background(), config)
+	if err != nil {
+		a.logger.Error("开始抓包失败", "error", err)
+		return err
+	}
+
+	a.logger.Info("抓包开始", "interface", interfaceName)
+	return nil
+}
+
+// StopCapture 停止抓包
+func (a *App) StopCapture() error {
+	if !a.captureService.IsCapturing() {
+		return fmt.Errorf("没有正在进行的抓包")
+	}
+
+	err := a.captureService.StopCapture()
+	if err != nil {
+		a.logger.Error("停止抓包失败", "error", err)
+		return err
+	}
+
+	// 处理所有收集的数据
+	a.captureService.ProcessAllData()
+
+	a.logger.Info("抓包停止")
+	return nil
+}
+
+// GetCapturedData 获取抓包数据
+func (a *App) GetCapturedData() ([]string, error) {
+	return a.captureService.GetCapturedData(), nil
+}
+
+// ParseData 解析数据为JSON
+func (a *App) ParseData(hexDataList []string) (string, error) {
+	result, err := a.parserService.ParseHexData(hexDataList)
+	if err != nil {
+		a.logger.Error("解析数据失败", "error", err)
+		return "", err
+	}
+
+	jsonData, err := json.MarshalIndent(result, "", "  ")
+	if err != nil {
+		a.logger.Error("JSON序列化失败", "error", err)
+		return "", err
+	}
+
+	a.logger.Info("数据解析完成", "count", len(result.Data))
+	return string(jsonData), nil
+}
+
+// ExportData 导出数据到文件
+func (a *App) ExportData(hexDataList []string, filename string) error {
+	result, err := a.parserService.ParseHexData(hexDataList)
+	if err != nil {
+		a.logger.Error("解析数据失败", "error", err)
+		return err
+	}
+
+	jsonData, err := json.MarshalIndent(result, "", "  ")
+	if err != nil {
+		a.logger.Error("JSON序列化失败", "error", err)
+		return err
+	}
+
+	// 这里可以添加文件写入逻辑
+	a.logger.Info("导出数据", "filename", filename, "count", len(result.Data))
+
+	// 简单示例：写入到当前目录
+	err = utils.WriteFile(filename, jsonData)
+	if err != nil {
+		a.logger.Error("写入文件失败", "error", err)
+		return err
+	}
+
+	return nil
+}
+
+// GetCaptureStatus 获取抓包状态
+func (a *App) GetCaptureStatus() model.CaptureStatus {
+	return model.CaptureStatus{
+		IsCapturing: a.captureService.IsCapturing(),
+		Status:      a.getStatusMessage(),
+	}
+}
+
+func (a *App) getStatusMessage() string {
+	if a.captureService.IsCapturing() {
+		return "正在抓包..."
+	}
+	return "准备就绪"
+}
+
+// ReadRawJsonFile 供前端调用，读取output_raw.json内容
+func (a *App) ReadRawJsonFile() (string, error) {
+	return a.parserService.ReadRawJsonFile()
+}