addui

2025-12-19 17:05:09 +08:00
parent cc3472ff40
commit 39c46937ea
307 changed files with 87686 additions and 129 deletions
--- a/internal/application/product/ui_component_file_service.go
+++ b/internal/application/product/ui_component_file_service.go
@@ -0,0 +1,341 @@
+package product
+
+import (
+	"archive/zip"
+	"context"
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"go.uber.org/zap"
+)
+
+// UIComponentFileService UI组件文件服务接口
+type UIComponentFileService interface {
+	// 上传并解压UI组件文件
+	UploadAndExtract(ctx context.Context, componentID, componentCode string, file io.Reader, filename string) error
+
+	// 批量上传UI组件文件（支持文件夹结构）
+	UploadMultipleFiles(ctx context.Context, componentID, componentCode string, files []io.Reader, filenames []string, paths []string) error
+
+	// 根据组件编码创建文件夹
+	CreateFolderByCode(componentCode string) (string, error)
+
+	// 删除组件文件夹
+	DeleteFolder(folderPath string) error
+
+	// 检查文件夹是否存在
+	FolderExists(folderPath string) bool
+
+	// 获取文件夹内容
+	GetFolderContent(folderPath string) ([]FileInfo, error)
+}
+
+// FileInfo 文件信息
+type FileInfo struct {
+	Name     string    `json:"name"`
+	Path     string    `json:"path"`
+	Size     int64     `json:"size"`
+	Type     string    `json:"type"` // "file" or "folder"
+	Modified time.Time `json:"modified"`
+}
+
+// UIComponentFileServiceImpl UI组件文件服务实现
+type UIComponentFileServiceImpl struct {
+	basePath string
+	logger   *zap.Logger
+}
+
+// NewUIComponentFileService 创建UI组件文件服务
+func NewUIComponentFileService(basePath string, logger *zap.Logger) UIComponentFileService {
+	// 确保基础路径存在
+	if err := os.MkdirAll(basePath, 0755); err != nil {
+		logger.Error("创建基础存储目录失败", zap.Error(err), zap.String("path", basePath))
+	}
+
+	return &UIComponentFileServiceImpl{
+		basePath: basePath,
+		logger:   logger,
+	}
+}
+
+// UploadAndExtract 上传并解压UI组件文件
+func (s *UIComponentFileServiceImpl) UploadAndExtract(ctx context.Context, componentID, componentCode string, file io.Reader, filename string) error {
+	// 直接使用基础路径作为文件夹路径，不再创建组件编码子文件夹
+	folderPath := s.basePath
+
+	// 确保基础目录存在
+	if err := os.MkdirAll(folderPath, 0755); err != nil {
+		return fmt.Errorf("创建基础目录失败: %w", err)
+	}
+
+	// 保存上传的文件
+	filePath := filepath.Join(folderPath, filename)
+	savedFile, err := os.Create(filePath)
+	if err != nil {
+		return fmt.Errorf("创建文件失败: %w", err)
+	}
+	defer savedFile.Close()
+
+	// 复制文件内容
+	if _, err := io.Copy(savedFile, file); err != nil {
+		// 删除部分写入的文件
+		_ = os.Remove(filePath)
+		return fmt.Errorf("保存文件失败: %w", err)
+	}
+
+	// 仅对ZIP文件执行解压逻辑
+	if strings.HasSuffix(strings.ToLower(filename), ".zip") {
+		// 解压文件到基础目录
+		if err := s.extractZipFile(filePath, folderPath); err != nil {
+			// 删除ZIP文件
+			_ = os.Remove(filePath)
+			return fmt.Errorf("解压文件失败: %w", err)
+		}
+
+		// 删除ZIP文件
+		_ = os.Remove(filePath)
+
+		s.logger.Info("UI组件文件上传并解压成功",
+			zap.String("componentID", componentID),
+			zap.String("componentCode", componentCode),
+			zap.String("folderPath", folderPath))
+	} else {
+		s.logger.Info("UI组件文件上传成功（未解压）",
+			zap.String("componentID", componentID),
+			zap.String("componentCode", componentCode),
+			zap.String("filePath", filePath))
+	}
+
+	return nil
+}
+
+// UploadMultipleFiles 批量上传UI组件文件（支持文件夹结构）
+func (s *UIComponentFileServiceImpl) UploadMultipleFiles(ctx context.Context, componentID, componentCode string, files []io.Reader, filenames []string, paths []string) error {
+	// 直接使用基础路径作为文件夹路径，不再创建组件编码子文件夹
+	folderPath := s.basePath
+
+	// 确保基础目录存在
+	if err := os.MkdirAll(folderPath, 0755); err != nil {
+		return fmt.Errorf("创建基础目录失败: %w", err)
+	}
+
+	// 处理每个文件
+	for i, file := range files {
+		filename := filenames[i]
+		path := paths[i]
+
+		// 如果有路径信息，创建对应的子文件夹
+		if path != "" && path != filename {
+			// 获取文件所在目录
+			dir := filepath.Dir(path)
+			if dir != "." {
+				// 创建子文件夹
+				subDirPath := filepath.Join(folderPath, dir)
+				if err := os.MkdirAll(subDirPath, 0755); err != nil {
+					return fmt.Errorf("创建子文件夹失败: %w", err)
+				}
+			}
+		}
+
+		// 确定文件保存路径
+		var filePath string
+		if path != "" && path != filename {
+			// 有路径信息，使用完整路径
+			filePath = filepath.Join(folderPath, path)
+		} else {
+			// 没有路径信息，直接保存在根目录
+			filePath = filepath.Join(folderPath, filename)
+		}
+
+		// 保存上传的文件
+		savedFile, err := os.Create(filePath)
+		if err != nil {
+			return fmt.Errorf("创建文件失败: %w", err)
+		}
+		defer savedFile.Close()
+
+		// 复制文件内容
+		if _, err := io.Copy(savedFile, file); err != nil {
+			// 删除部分写入的文件
+			_ = os.Remove(filePath)
+			return fmt.Errorf("保存文件失败: %w", err)
+		}
+
+		// 对ZIP文件执行解压逻辑
+		if strings.HasSuffix(strings.ToLower(filename), ".zip") {
+			// 确定解压目录
+			var extractDir string
+			if path != "" && path != filename {
+				// 有路径信息，解压到对应目录
+				dir := filepath.Dir(path)
+				if dir != "." {
+					extractDir = filepath.Join(folderPath, dir)
+				} else {
+					extractDir = folderPath
+				}
+			} else {
+				// 没有路径信息，解压到根目录
+				extractDir = folderPath
+			}
+
+			// 解压文件
+			if err := s.extractZipFile(filePath, extractDir); err != nil {
+				// 删除ZIP文件
+				_ = os.Remove(filePath)
+				return fmt.Errorf("解压文件失败: %w", err)
+			}
+
+			// 删除ZIP文件
+			_ = os.Remove(filePath)
+
+			s.logger.Info("UI组件文件上传并解压成功",
+				zap.String("componentID", componentID),
+				zap.String("componentCode", componentCode),
+				zap.String("filePath", filePath),
+				zap.String("extractDir", extractDir))
+		} else {
+			s.logger.Info("UI组件文件上传成功（未解压）",
+				zap.String("componentID", componentID),
+				zap.String("componentCode", componentCode),
+				zap.String("filePath", filePath))
+		}
+	}
+
+	return nil
+}
+
+// CreateFolderByCode 根据组件编码创建文件夹
+func (s *UIComponentFileServiceImpl) CreateFolderByCode(componentCode string) (string, error) {
+	folderPath := filepath.Join(s.basePath, componentCode)
+
+	// 创建文件夹（如果不存在）
+	if err := os.MkdirAll(folderPath, 0755); err != nil {
+		return "", fmt.Errorf("创建文件夹失败: %w", err)
+	}
+
+	return folderPath, nil
+}
+
+// DeleteFolder 删除组件文件夹
+func (s *UIComponentFileServiceImpl) DeleteFolder(folderPath string) error {
+	if !s.FolderExists(folderPath) {
+		return nil // 文件夹不存在，不视为错误
+	}
+
+	if err := os.RemoveAll(folderPath); err != nil {
+		return fmt.Errorf("删除文件夹失败: %w", err)
+	}
+
+	s.logger.Info("删除组件文件夹成功", zap.String("folderPath", folderPath))
+	return nil
+}
+
+// FolderExists 检查文件夹是否存在
+func (s *UIComponentFileServiceImpl) FolderExists(folderPath string) bool {
+	info, err := os.Stat(folderPath)
+	if err != nil {
+		return false
+	}
+	return info.IsDir()
+}
+
+// GetFolderContent 获取文件夹内容
+func (s *UIComponentFileServiceImpl) GetFolderContent(folderPath string) ([]FileInfo, error) {
+	var files []FileInfo
+
+	err := filepath.Walk(folderPath, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		// 跳过根目录
+		if path == folderPath {
+			return nil
+		}
+
+		// 获取相对路径
+		relPath, err := filepath.Rel(folderPath, path)
+		if err != nil {
+			return err
+		}
+
+		fileType := "file"
+		if info.IsDir() {
+			fileType = "folder"
+		}
+
+		files = append(files, FileInfo{
+			Name:     info.Name(),
+			Path:     relPath,
+			Size:     info.Size(),
+			Type:     fileType,
+			Modified: info.ModTime(),
+		})
+
+		return nil
+	})
+
+	if err != nil {
+		return nil, fmt.Errorf("扫描文件夹失败: %w", err)
+	}
+
+	return files, nil
+}
+
+// extractZipFile 解压ZIP文件
+func (s *UIComponentFileServiceImpl) extractZipFile(zipPath, destPath string) error {
+	reader, err := zip.OpenReader(zipPath)
+	if err != nil {
+		return fmt.Errorf("打开ZIP文件失败: %w", err)
+	}
+	defer reader.Close()
+
+	for _, file := range reader.File {
+		path := filepath.Join(destPath, file.Name)
+
+		// 防止路径遍历攻击
+		if !strings.HasPrefix(filepath.Clean(path), filepath.Clean(destPath)+string(os.PathSeparator)) {
+			return fmt.Errorf("无效的文件路径: %s", file.Name)
+		}
+
+		if file.FileInfo().IsDir() {
+			// 创建目录
+			if err := os.MkdirAll(path, file.Mode()); err != nil {
+				return fmt.Errorf("创建目录失败: %w", err)
+			}
+			continue
+		}
+
+		// 创建文件
+		fileReader, err := file.Open()
+		if err != nil {
+			return fmt.Errorf("打开ZIP内文件失败: %w", err)
+		}
+
+		// 确保父目录存在
+		if err := os.MkdirAll(filepath.Dir(path), 0755); err != nil {
+			fileReader.Close()
+			return fmt.Errorf("创建父目录失败: %w", err)
+		}
+
+		destFile, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, file.Mode())
+		if err != nil {
+			fileReader.Close()
+			return fmt.Errorf("创建目标文件失败: %w", err)
+		}
+
+		_, err = io.Copy(destFile, fileReader)
+		fileReader.Close()
+		destFile.Close()
+
+		if err != nil {
+			return fmt.Errorf("写入文件失败: %w", err)
+		}
+	}
+
+	return nil
+}