Go语言实现证书下载和配置工具:使用弹窗交互
这段代码实现了一个简单的证书下载和配置的工具,使用了第三方库 dlgs 和 walk 来实现弹窗和 GUI 界面。用户可以在编辑框中输入编号,然后点击按钮下载证书并写入配置文件。如果下载或写入失败,会弹出相应的错误提示窗口。最后,会弹出提示设置完成的信息窗口,用户需要手动安装证书。程序没有使用 GUI 界面,只有弹窗闪过,需要进一步完善。
package main
import (
"fmt"
"github.com/gen2brain/dlgs"
"github.com/lxn/walk"
. "github.com/lxn/walk/declarative"
"io"
"net/http"
"os"
"path/filepath"
)
func downloadFile(url string, filepath string) error {
// 创建文件
out, err := os.Create(filepath)
if err != nil {
return err
}
defer out.Close()
// 发送请求并下载文件
resp, err := http.Get(url)
if err != nil {
return err
}
defer resp.Body.Close()
// 写入文件
_, err = io.Copy(out, resp.Body)
if err != nil {
return err
}
return nil
}
func writeIni(name string) error {
// 获取用户目录
userDir, err := os.UserHomeDir()
if err != nil {
return err
}
// 获取配置文件路径
iniPath := filepath.Join(userDir, "HC.ini")
// 创建配置文件
f, err := os.Create(iniPath)
if err != nil {
return err
}
defer f.Close()
// 写入配置
_, err = f.WriteString("[general]
")
if err != nil {
return err
}
return nil
}
func main() {
var (
entry *walk.LineEdit
)
// 创建GUI界面
MainWindow{
Title: "证书下载和配置",
MinSize: Size{Width: 300, Height: 100},
Layout: VBox{},
Children: []Widget{
LineEdit{
AssignTo: &entry,
},
PushButton{
Text: "下载证书并写入配置",
OnClicked: func() {
// 获取编辑框内容
entryText := entry.Text()
// 如果编辑框内容为空,则提示错误信息
if entryText == "" {
dlgs.Error("错误", "请输入编号")
return
}
// 下载证书
filepath := filepath.Join(os.Getenv("USERPROFILE"), "Desktop", "mitmproxy-ca-cert.p12")
err := downloadFile("http://mitm.it/cert/p12", filepath)
if err != nil {
dlgs.Error("错误", fmt.Sprintf("下载证书失败: %v", err))
return
}
// 写入配置文件
err = writeIni(entryText)
if err != nil {
dlgs.Error("错误", fmt.Sprintf("写入配置文件失败: %v", err))
return
}
// 提示设置完成信息
dlgs.Info("信息", "设置完成,证书已下载到桌面请手动安装")
// 关闭窗口
walk.App().Exit(0)
},
},
},
}.Run()
}
进一步完善
- 使用 GUI 界面代替弹窗,可以提高用户体验。
- 增加错误处理机制,例如在下载证书失败时可以提供重试选项。
- 添加进度条显示下载进度。
- 完善证书安装步骤的说明,帮助用户更轻松地完成证书安装。
原文地址: https://www.cveoy.top/t/topic/nRPa 著作权归作者所有。请勿转载和采集!