You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

84 lines
2.2 KiB

package deps
import (
"archive/zip"
"bytes"
"fmt"
"godo/libs"
"io"
"os"
"path/filepath"
"strings"
)
func InitDir() error {
// 获取当前用户主目录
runDir := libs.GetAppExeDir()
if !libs.PathExists(runDir) {
if err := os.MkdirAll(runDir, 0o755); err != nil {
return fmt.Errorf("failed to create user directory: %v", err)
}
err := ExtractEmbeddedZip(runDir)
if err != nil {
return fmt.Errorf("failed to extract embedded zip: %v", err)
}
}
return nil
}
// ExtractEmbeddedZip 解压嵌入的ZIP文件到指定目录
func ExtractEmbeddedZip(exeDir string) error {
// 使用内存缓冲区来读取嵌入的ZIP数据
reader := bytes.NewReader(embeddedZip)
zipReader, err := zip.NewReader(reader, int64(len(embeddedZip)))
if err != nil {
return fmt.Errorf("failed to create zip reader: %v", err)
}
// 遍历ZIP文件中的每个条目并解压
for _, zipEntry := range zipReader.File {
// 检查条目名称是否以"."开头,如果是,则跳过
if strings.HasPrefix(zipEntry.Name, ".") {
fmt.Printf("Skipping hidden entry: %s\n", zipEntry.Name)
continue
}
// 构建解压后的文件或目录路径
entryPath := filepath.Join(exeDir, zipEntry.Name)
// 如果是目录,则创建目录
if zipEntry.FileInfo().IsDir() {
if err := os.MkdirAll(entryPath, zipEntry.Mode()); err != nil {
return fmt.Errorf("failed to create directory: %v", err)
}
continue
}
// 如果是文件,则解压文件
zipFile, err := zipEntry.Open()
if err != nil {
return fmt.Errorf("failed to open zip file entry: %v", err)
}
defer zipFile.Close()
// 确保目标文件的父目录存在
if err := os.MkdirAll(filepath.Dir(entryPath), 0755); err != nil {
return fmt.Errorf("failed to create parent directory: %v", err)
}
dstFile, err := os.OpenFile(entryPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0755)
if err != nil {
return fmt.Errorf("failed to create destination file: %v", err)
}
defer dstFile.Close()
if _, err := io.Copy(dstFile, zipFile); err != nil {
return fmt.Errorf("failed to copy content to destination file: %v", err)
}
}
fmt.Println("Embedded ZIP extracted to", exeDir)
return nil
}