使用Go语言开发一个智能exe文件重命名工具

文章目录

在日常开发和软件管理中，我们经常会遇到需要整理大量EXE文件的情况。这些文件往往有着不规范的命名，如setup.exe、installer.exe等，难以直接了解其具体内容和版本。本文将介绍如何使用Go语言开发一个智能EXE文件重命名工具，它能自动提取文件的版本信息，并利用大模型API生成规范的命名建议。

这个工具主要实现了以下功能：解析EXE文件的版本信息资源利用大模型API生成智能命名建议提供友好的命令行交互界面安全执行文件重命名操作

Windows系统通过version.dll提供了访问文件版本信息的API。我们的工具使用了三个关键函数： GetFileVersionInfoSizeW – 获取版本信息数据的大小 GetFileVersionInfoW – 获取完整的版本信息数据 VerQueryValueW – 查询特定的版本信息字段在Go中调用这些API需要使用syscall包和unsafe指针操作： var ( versionDLL = syscall.NewLazyDLL(“version.dll”) procGetFileVersionInfoSizeW = versionDLL.NewProc(“GetFileVersionInfoSizeW”) procGetFileVersionInfoW = versionDLL.NewProc(“GetFileVersionInfoW”) procVerQueryValueW = versionDLL.NewProc(“VerQueryValueW”) )
工具集成了bigmodel API来生成智能命名建议(该模型免费，应付这种场景绰绰有余)： cfg := openai.DefaultConfig(apiKey) cfg.BaseURL = “https://open.bigmodel.cn/api/paas/v4/” client := openai.NewClientWithConfig(cfg) resp, err := client.CreateChatCompletion( context.Background(), openai.ChatCompletionRequest{ Model: “GLM-4-Flash-250414”, Messages: []openai.ChatCompletionMessage{ { Role: openai.ChatMessageRoleUser, Content: prompt, }, }, MaxTokens: 100, Temperature: 0.3, }, ) 我们设计了提示词(prompt)，确保大模型返回规范的文件名：包含产品名称和版本号只使用字母、数字、中文、下划线、连字符格式统一，如产品名_版本号.exe或产品名-版本号.exe
使用promptui库创建了友好的交互界面： // 确认对话框 prompt := promptui.Select{ Label: “是否根据建议重命名文件？”, Items: []string{“是”, “否”}, } // 文本输入框 prompt := promptui.Prompt{ Label: “请输入新的文件名”, Default: suggestedName, AllowEdit: true, }

在日常开发和软件管理中，我们经常会遇到需要整理大量EXE文件的情况。这些文件往往有着不规范的命名，如`setup.exe`、`installer.exe`等，难以直接了解其具体内容和版本。本文将介绍如何使用Go语言开发一个智能EXE文件重命名工具，它能自动提取文件的版本信息，并利用大模型API生成规范的命名建议。

这个工具主要实现了以下功能：

解析EXE文件的版本信息资源

利用大模型API生成智能命名建议

提供友好的命令行交互界面

安全执行文件重命名操作

Windows系统通过version.dll提供了访问文件版本信息的API。我们的工具使用了三个关键函数：

GetFileVersionInfoSizeW – 获取版本信息数据的大小
GetFileVersionInfoW – 获取完整的版本信息数据
VerQueryValueW – 查询特定的版本信息字段

在Go中调用这些API需要使用syscall包和unsafe指针操作：

var (
    versionDLL                  = syscall.NewLazyDLL("version.dll")
    procGetFileVersionInfoSizeW = versionDLL.NewProc("GetFileVersionInfoSizeW")
    procGetFileVersionInfoW     = versionDLL.NewProc("GetFileVersionInfoW")
    procVerQueryValueW          = versionDLL.NewProc("VerQueryValueW")
)

工具集成了bigmodel API来生成智能命名建议(该模型免费，应付这种场景绰绰有余)：

cfg := openai.DefaultConfig(apiKey)
cfg.BaseURL = "https://open.bigmodel.cn/api/paas/v4/"
client := openai.NewClientWithConfig(cfg)

resp, err := client.CreateChatCompletion(
    context.Background(),
    openai.ChatCompletionRequest{
        Model: "GLM-4-Flash-250414",
        Messages: []openai.ChatCompletionMessage{
            {
                Role:    openai.ChatMessageRoleUser,
                Content: prompt,
            },
        },
        MaxTokens:   100,
        Temperature: 0.3,
    },
)

我们设计了提示词(prompt)，确保大模型返回规范的文件名：

包含产品名称和版本号
只使用字母、数字、中文、下划线、连字符
格式统一，如产品名_版本号.exe或产品名-版本号.exe

使用promptui库创建了友好的交互界面：

// 确认对话框
prompt := promptui.Select{
    Label: "是否根据建议重命名文件？",
    Items: []string{"是", "否"},
}

// 文本输入框
prompt := promptui.Prompt{
    Label:     "请输入新的文件名",
    Default:   suggestedName,
    AllowEdit: true,
}

package main

import (
	"context"
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"syscall"
	"unicode/utf16"
	"unsafe"

	"github.com/manifoldco/promptui"
	openai "github.com/sashabaranov/go-openai"
)

var (
	versionDLL                  = syscall.NewLazyDLL("version.dll")
	procGetFileVersionInfoSizeW = versionDLL.NewProc("GetFileVersionInfoSizeW")
	procGetFileVersionInfoW     = versionDLL.NewProc("GetFileVersionInfoW")
	procVerQueryValueW          = versionDLL.NewProc("VerQueryValueW")
)

// 检查BIG_MODEL_KEY环境变量
func checkBigModelKey() error {
	key := os.Getenv("BIG_MODEL_KEY")
	if key != "" {
		return nil // 已经设置了
	}

	fmt.Println("未找到BIG_MODEL_KEY环境变量，请手动设置。例如：set BIG_MODEL_KEY=你的API密钥")
	return fmt.Errorf("未设置BIG_MODEL_KEY环境变量")
}

func utf16PtrFromString(s string) *uint16 {
	u := utf16.Encode([]rune(s + "x00"))
	return &u[0]
}

// Windows API方式获取版本信息
func GetFileVersionInfoAPI(path string) (map[string]string, error) {
	info := make(map[string]string)
	pPath := utf16PtrFromString(path)
	var handle uint32
	size, _, _ := procGetFileVersionInfoSizeW.Call(
		uintptr(unsafe.Pointer(pPath)),
		uintptr(unsafe.Pointer(&handle)),
	)
	if size == 0 {
		return nil, fmt.Errorf("GetFileVersionInfoSizeW failed")
	}
	buf := make([]byte, size)
	ret, _, _ := procGetFileVersionInfoW.Call(
		uintptr(unsafe.Pointer(pPath)),
		0,
		uintptr(size),
		uintptr(unsafe.Pointer(&buf[0])),
	)
	if ret == 0 {
		return nil, fmt.Errorf("GetFileVersionInfoW failed")
	}

	// 查询语言和代码页
	var transPtr uintptr
	var transLen uint32
	subBlock := utf16PtrFromString(`VarFileInfoTranslation`)
	ret, _, _ = procVerQueryValueW.Call(
		uintptr(unsafe.Pointer(&buf[0])),
		uintptr(unsafe.Pointer(subBlock)),
		uintptr(unsafe.Pointer(&transPtr)),
		uintptr(unsafe.Pointer(&transLen)),
	)
	if ret == 0 || transLen < 4 {
		return nil, fmt.Errorf("VerQueryValueW Translation failed")
	}
	lang := *(*uint16)(unsafe.Pointer(transPtr))
	codepage := *(*uint16)(unsafe.Pointer(transPtr + 2))
	langCode := fmt.Sprintf("%04x%04x", lang, codepage)

	fields := []string{
		"FileDescription",
		"FileVersion",
		"ProductName",
		"ProductVersion",
		"LegalCopyright",
		"OriginalFilename",
		"InternalName",
		"CompanyName",
		"Comments",
	}

	for _, field := range fields {
		block := fmt.Sprintf(`StringFileInfo%s%s`, langCode, field)
		blockPtr := utf16PtrFromString(block)
		var valuePtr uintptr
		var valueLen uint32
		ret, _, _ := procVerQueryValueW.Call(
			uintptr(unsafe.Pointer(&buf[0])),
			uintptr(unsafe.Pointer(blockPtr)),
			uintptr(unsafe.Pointer(&valuePtr)),
			uintptr(unsafe.Pointer(&valueLen)),
		)
		if ret != 0 && valueLen > 0 {
			val := syscall.UTF16ToString((*[1 << 16]uint16)(unsafe.Pointer(valuePtr))[:valueLen])
			info[field] = val
		}
	}
	return info, nil
}

func getAPIKey() string {
	key := os.Getenv("BIG_MODEL_KEY")
	if key != "" {
		return key
	}
	fmt.Print("请输入百川大模型 API KEY（BIG_MODEL_KEY）：")
	var input string
	fmt.Scanln(&input)
	return strings.TrimSpace(input)
}

func GetRenameSuggestion(info map[string]string, originalName string) (string, error) {
	apiKey := getAPIKey()
	if apiKey == "" {
		return "", fmt.Errorf("未提供 BIG_MODEL_KEY")
	}

	cfg := openai.DefaultConfig(apiKey)
	cfg.BaseURL = "https://open.bigmodel.cn/api/paas/v4/"
	client := openai.NewClientWithConfig(cfg)

	prompt := fmt.Sprintf(`请根据以下EXE文件的版本信息，给出一个简洁、规范的文件重命名建议。
原始文件名: %s

版本信息:
- 文件描述: %s
- 文件版本: %s
- 产品名称: %s
- 产品版本: %s
- 版权信息: %s
- 原始文件名: %s
- 内部名称: %s
- 公司名称: %s
- 注释: %s

重命名要求:
1. 使用中文或英文，简洁明了
2. 包含产品名称和版本号
3. 避免特殊字符，只使用字母、数字、中文、下划线、连字符
4. 格式建议: 产品名_版本号.exe 或 产品名-版本号.exe
5. 如果产品名包含特殊字符，请适当简化

请只返回重命名后的文件名（包含.exe扩展名），不要其他解释。`,
		originalName,
		getValueOrDefault(info, "FileDescription"),
		getValueOrDefault(info, "FileVersion"),
		getValueOrDefault(info, "ProductName"),
		getValueOrDefault(info, "ProductVersion"),
		getValueOrDefault(info, "LegalCopyright"),
		getValueOrDefault(info, "OriginalFilename"),
		getValueOrDefault(info, "InternalName"),
		getValueOrDefault(info, "CompanyName"),
		getValueOrDefault(info, "Comments"))

	resp, err := client.CreateChatCompletion(
		context.Background(),
		openai.ChatCompletionRequest{
			Model: "GLM-4-Flash-250414",
			Messages: []openai.ChatCompletionMessage{
				{
					Role:    openai.ChatMessageRoleUser,
					Content: prompt,
				},
			},
			MaxTokens:   100,
			Temperature: 0.3,
		},
	)
	if err != nil {
		return "", fmt.Errorf("API调用失败: %w", err)
	}
	if len(resp.Choices) == 0 {
		return "", fmt.Errorf("未返回有效响应")
	}
	suggestion := strings.TrimSpace(resp.Choices[0].Message.Content)
	return suggestion, nil
}

func getValueOrDefault(info map[string]string, key string) string {
	if value, exists := info[key]; exists && value != "" {
		return value
	}
	return "(无)"
}

// 确认是否重命名
func confirmRename() bool {
	prompt := promptui.Select{
		Label: "是否根据建议重命名文件？",
		Items: []string{"是", "否"},
	}

	_, result, err := prompt.Run()
	if err != nil {
		fmt.Printf("选择失败: %vn", err)
		return false
	}

	return result == "是"
}

// 输入新文件名
func inputNewFileName(suggestedName string) string {
	prompt := promptui.Prompt{
		Label:     "请输入新的文件名",
		Default:   suggestedName,
		AllowEdit: true,
	}

	result, err := prompt.Run()
	if err != nil {
		fmt.Printf("输入失败: %vn", err)
		return ""
	}

	return result
}

// 重命名文件
func renameFile(oldPath, newName string) error {
	dir := filepath.Dir(oldPath)
	newPath := filepath.Join(dir, newName)

	// 检查新文件名是否已存在
	if _, err := os.Stat(newPath); err == nil {
		return fmt.Errorf("文件 %s 已存在", newName)
	}

	// 执行重命名
	err := os.Rename(oldPath, newPath)
	if err != nil {
		return fmt.Errorf("重命名失败: %w", err)
	}

	fmt.Printf("文件已重命名为: %sn", newName)
	return nil
}

func main() {
	// 检查BIG_MODEL_KEY
	if err := checkBigModelKey(); err != nil {
		return
	}

	if len(os.Args) < 2 {
		fmt.Println("请指定EXE文件路径")
		fmt.Println("示例: .\exeRename.exe your_file.exe")
		return
	}

	exePath := os.Args[1]
	info, err := GetFileVersionInfoAPI(exePath)
	if err != nil {
		fmt.Printf("错误: %vn", err)
		return
	}

	// 显示版本信息
	fmt.Println("=== 版本信息 ===")
	fields := []string{"FileDescription", "FileVersion", "ProductName", "ProductVersion", "LegalCopyright", "OriginalFilename", "InternalName", "CompanyName", "Comments"}
	for _, field := range fields {
		value := info[field]
		if value == "" {
			value = "(无)"
		}
		fmt.Printf("%-20s: %sn", field, value)
	}

	// 获取原始文件名（不含路径）
	originalName := exePath
	if lastSlash := strings.LastIndex(exePath, "\"); lastSlash != -1 {
		originalName = exePath[lastSlash+1:]
	}
	if lastSlash := strings.LastIndex(originalName, "/"); lastSlash != -1 {
		originalName = originalName[lastSlash+1:]
	}

	// 获取重命名建议
	fmt.Println("n=== 智能重命名建议 ===")
	suggestion, err := GetRenameSuggestion(info, originalName)
	if err != nil {
		fmt.Printf("获取重命名建议失败: %vn", err)
		fmt.Println("请确保已设置环境变量 OPENAI_API_KEY")
		return
	}

	fmt.Printf("原始文件名: %sn", originalName)
	fmt.Printf("建议重命名: %sn", suggestion)

	// 询问是否重命名
	if confirmRename() {
		// 输入新文件名
		newName := inputNewFileName(suggestion)
		if newName != "" {
			// 执行重命名
			if err := renameFile(exePath, newName); err != nil {
				fmt.Printf("重命名失败: %vn", err)
			}
		}
	}
}

到此这篇关于使用Go语言开发一个智能exe文件重命名工具的文章就介绍到这了,更多相关Go文件重命名内容请搜索风君子博客以前的文章或继续浏览下面的相关文章希望大家以后多多支持风君子博客！

您可能感兴趣的文章:

利用go语言实现Git 重命名远程分支
使用Go语言实现xmind文件转换为markdown
Go+Gin实现安全多文件上传功能
利用Go语言开发文件操作工具轻松处理所有文件
使用Go语言开发一个命令行文件管理工具
详解如何使用Go语言进行文件监控和通知

华夏福临门盛世加强版怎么样？产品责任是重中之重

国寿福庆典版值得买吗？附投保建议

理财中低风险是几级？中低风险理财会亏吗

瑞泰泰安心防癌险优缺点是什么？这份分析不容错过

2020最新自助免费申请Office365教育版，免费5TOneDrive云盘详细图文教程

MacBook怎么绕过BootCamp安装Win10双系统？

国内知名的前端博客

域名为什么会被墙？如何检测域名是否被墙？域名被墙如何处理？

最新的windows xp sp3产品密匙 xp sp3序列号 xp sp3激活码

使用Go语言开发一个智能exe文件重命名工具

目录

这个工具主要实现了以下功能： 解析EXE文件的版本信息资源 利用大模型API生成智能命名建议 提供友好的命令行交互界面 安全执行文件重命名操作

More posts

站内搜索

标签云

热门文章

友情链接

这个工具主要实现了以下功能：

解析EXE文件的版本信息资源

利用大模型API生成智能命名建议

提供友好的命令行交互界面

安全执行文件重命名操作