mirror of https://gitee.com/godoos/godoos.git
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
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
|
|
}
|
|
|