main.go

package main

import (
	"flag"
	"fmt"
	"os"
	"path/filepath"
	"strings"
)

// 配置项
type Config struct {
	ApiKey        string
	ApiUrl        string
	Model         string
	Language      string
	SkeletonLevel int // 0-100，代码骨架的完整度
	OutputDir     string
}

// 支持的编程语言
var supportedLanguages = []string{
	"go", "python", "java", "cpp", "javascript", "typescript",
	"rust", "c", "csharp", "php", "ruby", "swift", "kotlin",
}

// 算法分类
var algoCategories = []string{
	"数组", "链表", "栈", "队列", "哈希表",
	"字符串", "二分查找", "排序", "贪心", "动态规划",
	"深度优先搜索", "广度优先搜索", "回溯", "树", "图",
	"数学", "位操作", "并查集", "前缀和", "滑动窗口",
}

// 获取API URL，优先使用环境变量，否则使用默认值
func getApiUrl() string {
	if url := os.Getenv("ARK_API_URL"); url != "" {
		return url
	}
	return "https://maas-cn-southwest-2.modelarts-maas.com/v1/infers/8a062fd4-7367-4ab4-a936-5eeb8fb821c4/v1/chat/completions"
}

func main() {
	// 解析命令行参数
	problemID := flag.Int("id", 0, "LeetCode题目ID")
	language := flag.String("lang", "go", "编程语言")
	level := flag.Int("level", 30, "代码骨架完整度(0-100)")
	category := flag.String("category", "", "算法分类")
	model := flag.String("model", "deepseek-v3-1-terminus", "LLM模型 (可选: DeepSeek-V3, DeepSeek-R1, deepseek-v3-250324, deepseek-r1-250120)")
	output := flag.String("output", "leetcode_practice", "输出目录")

	flag.Parse()

	// 验证参数
	if *problemID <= 0 {
		fmt.Println("❌ 错误: 必须提供有效的LeetCode题目ID")
		flag.Usage()
		os.Exit(1)
	}

	if *level < 0 || *level > 100 {
		fmt.Println("❌ 错误: 代码骨架完整度必须在0-100之间")
		flag.Usage()
		os.Exit(1)
	}

	// 验证语言
	langValid := false
	for _, lang := range supportedLanguages {
		if lang == *language {
			langValid = true
			break
		}
	}
	if !langValid {
		fmt.Printf("❌ 错误: 不支持的语言: %s\n", *language)
		fmt.Printf("支持的语言: %s\n", strings.Join(supportedLanguages, ", "))
		os.Exit(1)
	}

	// 加载配置
	cfg := &Config{
		ApiKey:        os.Getenv("ARK_API_KEY"),
		ApiUrl:        getApiUrl(),
		Model:         *model,
		Language:      *language,
		SkeletonLevel: *level,
		OutputDir:     *output,
	}

	// 验证API密钥
	if cfg.ApiKey == "" {
		fmt.Println("❌ 错误: 未设置ARK_API_KEY环境变量")
		fmt.Println("请设置环境变量: export ARK_API_KEY='您的API密钥'")
		os.Exit(1)
	}

	// 确保输出目录存在
	if err := os.MkdirAll(cfg.OutputDir, 0755); err != nil {
		fmt.Printf("❌ 错误: 无法创建输出目录: %v\n", err)
		os.Exit(1)
	}

	// 生成代码文件
	err := generateCodeFile(*problemID, cfg, *category)
	if err != nil {
		fmt.Printf("❌ 错误: %s\n", err)
		os.Exit(1)
	}
}

// 生成代码文件
func generateCodeFile(problemID int, cfg *Config, category string) error {
	fmt.Printf("🔍 处理题目 #%d...\n", problemID)

	// 使用一步到位的方法
	codeContent, problemInfo, err := generateDirectCode(problemID, cfg, category)
	if err != nil {
		return fmt.Errorf("生成代码骨架失败: %v", err)
	}

	// 创建文件名
	title, ok := problemInfo["title"].(string)
	if !ok {
		title = fmt.Sprintf("题目%d", problemID)
	}

	// 使用符合测试规范的文件名
	filename := fmt.Sprintf("leetcode_%d_test.%s",
		problemID,
		getFileExtension(cfg.Language))

	// 根据语言创建子目录
	langDir := filepath.Join(cfg.OutputDir, cfg.Language)
	if err := os.MkdirAll(langDir, 0755); err != nil {
		return fmt.Errorf("无法创建语言目录: %v", err)
	}

	filePath := filepath.Join(langDir, filename)

	// 添加题目信息注释
	difficulty, ok := problemInfo["difficulty"].(string)
	if !ok {
		difficulty = "未知"
	}

	description, ok := problemInfo["description"].(string)
	if !ok {
		description = "由大模型直接生成"
	}

	header := fmt.Sprintf(`/*
 * LeetCode #%d: %s
 * 难度: %s
 * 
 * 题目描述:
 * %s
 * 
 * 代码骨架完整度: %d%%
 */

`, problemID, title, difficulty, description, cfg.SkeletonLevel)

	// 写入文件
	err = os.WriteFile(filePath, []byte(header+codeContent), 0644)
	if err != nil {
		return fmt.Errorf("写入文件失败: %v", err)
	}

	fmt.Printf("\n✅ 成功生成练习文件: %s\n", filePath)
	fmt.Printf("🔍 题目难度: %s\n", difficulty)
	if category != "" {
		fmt.Printf("📚 算法分类: %s\n", category)
	}
	fmt.Printf("⚙️ 代码完整度: %d%%\n", cfg.SkeletonLevel)
	fmt.Printf("🧪 调试命令: %s\n", getDebugCommand(cfg.Language, cfg.OutputDir, filename))

	return nil
}

// 文件名清理
func sanitizeFilename(name string) string {
	// 移除非法字符
	name = strings.Map(func(r rune) rune {
		if r < 32 || strings.ContainsRune(`<>:"/\|?*`, r) {
			return '_'
		}
		return r
	}, name)

	// 截断过长的文件名
	if len(name) > 50 {
		name = name[:50]
	}

	return name
}

// 获取文件扩展名
func getFileExtension(language string) string {
	switch language {
	case "python":
		return "py"
	case "java":
		return "java"
	case "cpp":
		return "cpp"
	case "javascript":
		return "js"
	case "typescript":
		return "ts"
	case "rust":
		return "rs"
	case "c":
		return "c"
	case "csharp":
		return "cs"
	case "php":
		return "php"
	case "ruby":
		return "rb"
	case "swift":
		return "swift"
	case "kotlin":
		return "kt"
	default:
		return "go"
	}
}

// 获取调试命令（仅用于打印提示，不影响生成逻辑）
func getDebugCommand(language, outputDir, filename string) string {
	langDir := filepath.Join(outputDir, language)
	switch language {
	case "go":
		return fmt.Sprintf("cd %s && go test -v %s", langDir, filename)
	case "c":
		exe := strings.TrimSuffix(filename, ".c")
		return fmt.Sprintf("cd %s && cc -std=c11 -O2 -Wall -Wextra -o %s %s && ./%s", langDir, exe, filename, exe)
	case "cpp":
		exe := strings.TrimSuffix(filename, ".cpp")
		return fmt.Sprintf("cd %s && c++ -std=c++17 -O2 -Wall -Wextra -o %s %s && ./%s", langDir, exe, filename, exe)
	case "python":
		return fmt.Sprintf("cd %s && python3 %s", langDir, filename)
	case "java":
		className := strings.TrimSuffix(filename, ".java")
		return fmt.Sprintf("cd %s && javac %s && java %s", langDir, filename, className)
	case "javascript":
		return fmt.Sprintf("cd %s && node %s", langDir, filename)
	case "typescript":
		jsName := strings.TrimSuffix(filename, ".ts") + ".js"
		return fmt.Sprintf("cd %s && tsc %s && node %s", langDir, filename, jsName)
	case "rust":
		exe := strings.TrimSuffix(filename, ".rs")
		return fmt.Sprintf("cd %s && rustc %s && ./%s", langDir, filename, exe)
	case "php":
		return fmt.Sprintf("cd %s && php %s", langDir, filename)
	case "ruby":
		return fmt.Sprintf("cd %s && ruby %s", langDir, filename)
	case "swift":
		exe := strings.TrimSuffix(filename, ".swift")
		return fmt.Sprintf("cd %s && swiftc %s -o %s && ./%s", langDir, filename, exe, exe)
	case "kotlin":
		jar := strings.TrimSuffix(filename, ".kt") + ".jar"
		return fmt.Sprintf("cd %s && kotlinc %s -include-runtime -d %s && java -jar %s", langDir, filename, jar, jar)
	case "csharp":
		return fmt.Sprintf("cd %s && # C# 单文件运行方式取决于环境（csc/dotnet-script/dotnet），请自行选择执行 %s", langDir, filename)
	default:
		return fmt.Sprintf("cd %s && # 请按%s语言方式运行: %s", langDir, language, filename)
	}
}