feat: support profile path cmd options

This commit is contained in:
ᴍᴏᴏɴD4ʀᴋ
2022-04-14 23:36:11 +08:00
parent da2beb9453
commit f728d3201b
12 changed files with 170 additions and 126 deletions
-44
View File
@@ -1,20 +1,14 @@
package utils
import (
"archive/zip"
"bytes"
"fmt"
"io/ioutil"
"os"
"path"
"strings"
"time"
"hack-browser-data/internal/log"
)
const Prefix = "[x]: "
func IntToBool(a int) bool {
switch a {
case 0, -1:
@@ -73,41 +67,3 @@ func MakeDir(dirName string) error {
}
return nil
}
func Compress(exportDir string) error {
files, err := ioutil.ReadDir(exportDir)
if err != nil {
log.Error(err)
}
var b = new(bytes.Buffer)
zw := zip.NewWriter(b)
for _, f := range files {
fw, _ := zw.Create(f.Name())
fileName := path.Join(exportDir, f.Name())
fileContent, err := ioutil.ReadFile(fileName)
if err != nil {
zw.Close()
return err
}
_, err = fw.Write(fileContent)
if err != nil {
zw.Close()
return err
}
err = os.Remove(fileName)
if err != nil {
log.Error(err)
}
}
if err := zw.Close(); err != nil {
return err
}
zipName := exportDir + `/archive.zip`
outFile, _ := os.Create(zipName)
_, err = b.WriteTo(outFile)
if err != nil {
return err
}
fmt.Printf("%s Compress success, zip filename is %s \n", Prefix, zipName)
return nil
}