这段代码实现了一个简单的证书下载和配置的工具,使用了第三方库 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 界面代替弹窗,可以提高用户体验。
  • 增加错误处理机制,例如在下载证书失败时可以提供重试选项。
  • 添加进度条显示下载进度。
  • 完善证书安装步骤的说明,帮助用户更轻松地完成证书安装。
Go语言实现证书下载和配置工具:使用弹窗交互

原文地址: https://www.cveoy.top/t/topic/nRPa 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录