capcut_test.go

package main

import (
	"comic/pkg/capcut"
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"sort"
	"strings"
	"testing"

	"go.uber.org/zap"
)

func TestCapcut(t *testing.T) {
	// TODO: test Capcut
	// 初始化日志记录器
	logger, _ := zap.NewProduction()
	defer logger.Sync()

	// 初始化CapCut生成器
	fmt.Println("初始化CapCut项目生成器...")
	capcutGen := capcut.NewCapcutGenerator(logger)

	// 指定下载图片目录
	outputDir := "./downloaded_images"

	// 检查目录是否存在
	if _, err := os.Stat(outputDir); os.IsNotExist(err) {
		fmt.Printf("目录 %s 不存在\n", outputDir)
		return
	}

	// 获取所有图片和音频文件
	files, err := ioutil.ReadDir(outputDir)
	if err != nil {
		fmt.Printf("读取下载目录失败: %v\n", err)
		return
	}

	// 分离图片和音频文件
	var imageFiles []string
	var audioFiles []string
	for _, file := range files {
		filename := file.Name()
		filePath := filepath.Join(outputDir, filename)
		if strings.HasSuffix(strings.ToLower(filename), ".jpg") ||
			strings.HasSuffix(strings.ToLower(filename), ".png") ||
			strings.HasSuffix(strings.ToLower(filename), ".jpeg") {
			imageFiles = append(imageFiles, filePath)
		} else if strings.HasSuffix(strings.ToLower(filename), ".wav") ||
			strings.HasSuffix(strings.ToLower(filename), ".mp3") {
			audioFiles = append(audioFiles, filePath)
		}
	}

	fmt.Printf("找到 %d 张图片和 %d 个音频文件\n", len(imageFiles), len(audioFiles))

	// 按文件名排序以确保图片和音频文件的顺序匹配
	sort.Strings(imageFiles)
	sort.Strings(audioFiles)

	// 生成剪映项目
	fmt.Println("开始生成剪映项目...")
	err = capcutGen.GenerateProject(outputDir)
	if err != nil {
		fmt.Printf("生成剪映项目失败: %v\n", err)
	} else {
		fmt.Printf("成功生成剪映项目: %s\n", outputDir)
		fmt.Printf("项目已保存到: %s/capcut_project\n", outputDir)
		fmt.Println("请在剪映中导入项目文件夹以查看内容")
	}
}