Added support for sending relative folder

schollz · Jul 6, 2022 · b50fe88 · b50fe88
1 parent 37ae453
commit b50fe88
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/croc/croc.go b/src/croc/croc.go
@@ -286,7 +286,8 @@ func GetFilesInfo(fnames []string, zipfolder bool) (filesInfo []FileInfo, emptyF
 		}
 
 		if stat.IsDir() && zipfolder {
-			dest := path + ".zip"
+			path := filepath.Dir(path)
+			dest := filepath.Base(path) + ".zip"
 			utils.ZipDirectory(dest, path)
 			stat, errStat = os.Lstat(dest)
 			if errStat != nil {

diff --git a/src/utils/utils.go b/src/utils/utils.go
@@ -394,15 +394,16 @@ func ZipDirectory(destination string, source string) (err error) {
 				log.Fatalln(err)
 			}
 			defer f1.Close()
-			w1, err := writer.Create(path)
+			zip_path := strings.ReplaceAll(path, source, strings.TrimSuffix(destination, ".zip"))
+			w1, err := writer.Create(zip_path)
 			if err != nil {
 				log.Fatalln(err)
 			}
 			if _, err := io.Copy(w1, f1); err != nil {
 				log.Fatalln(err)
 			}
 			fmt.Fprintf(os.Stderr, "\r\033[2K")
-			fmt.Fprintf(os.Stderr, "\rAdding %s", path)
+			fmt.Fprintf(os.Stderr, "\rAdding %s", zip_path)
 		}
 		return nil
 	})