Breaking News: Grepper is joining You.com. Read the official announcement!

go zip file

Restu Wahyu Saputra answered on April 29, 2024 Popularity 5/10 Helpfulness 1/10

answer go zip file

go zip file

Comment

package main

import (
	"bytes"
	"errors"
	"fmt"
	"io"
	"io/fs"
	"os"
	"path"
	"strings"

	"github.com/yeka/zip"
)

type (
	IZipWriter interface {
		WriteFile(fileName, password, content string) error
		RemoveFile(fileName string) error
	}

	zipWriter struct{}
)

func NewZip() IZipWriter {
	return &zipWriter{}
}

func (h *zipWriter) WriteFile(fileName, password, content string) error {
	dir := path.Join(os.Getenv("STORAGE_DIR"))
	filedir := fmt.Sprintf("%s/%s", dir, fileName)

	if !path.IsAbs(dir) {
		return errors.New("Path must be absolute value")
	}

	if err := os.WriteFile(filedir+".txt", []byte{}, 644); err != nil {
		return err
	}

	file, err := os.Create(filedir + ".zip")
	if err != nil {
		return err
	}

	zipw := zip.NewWriter(file)

	zw, err := zipw.Encrypt(fileName+".txt", password, zip.AES256Encryption)
	if err != nil {
		return err
	}

	if _, err = io.Copy(zw, bytes.NewReader([]byte(content))); err != nil {
		return err
	}

	defer func() {
		zipw.Close()
		zipw.Flush()
	}()

	return nil
}

func (h *zipWriter) RemoveFile(fileName string) error {
	dir := path.Join(os.Getenv("STORAGE_DIR"))

	files, err := os.ReadDir(dir)
	if err != nil {
		return err
	}

	if len(files) > 0 {
		for _, file := range files {
			fn := file.Name()

			if _, err := fs.Stat(os.DirFS(dir), fn); err != nil {
				return err
			}

			if strings.Contains(fn, fileName) {
				if err := os.Remove(fmt.Sprintf("%s/%s", dir, fn)); err != nil {
					return err
				}
			} else {
				return errors.New("File name is not exists")
			}
		}
	}

	return nil
}

Popularity 5/10 Helpfulness 1/10 Language go

Source: Grepper

Tags: file go zip

Link to this answer
Share Copy Link

Contributed on Apr 29 2024

Restu Wahyu Saputra

0 Answers Avg Quality 2/10

go zip file

Contents

More Related Answers

go zip file

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.