dify-plugin-daemon/internal/core/bundle_packager/zip.go

package bundle_packager

import (
	"archive/zip"
	"io"
	"os"
)

type ZipBundlePackager struct {
	*MemoryZipBundlePackager

	path string
}

func NewZipBundlePackager(path string) (BundlePackager, error) {
	zipFile, err := os.Open(path)
	if err != nil {
		return nil, err
	}
	defer zipFile.Close()

	zipBytes, err := io.ReadAll(zipFile)
	if err != nil {
		return nil, err
	}

	memoryPackager, err := NewMemoryZipBundlePackager(zipBytes)
	if err != nil {
		return nil, err
	}

	zipBundlePackager := &ZipBundlePackager{
		MemoryZipBundlePackager: memoryPackager,
		path:                    path,
	}

	return zipBundlePackager, nil
}

func (p *ZipBundlePackager) Save() error {
	// export the bundle to a zip file
	zipBytes, err := p.Export()
	if err != nil {
		return err
	}

	// save the zip file
	err = os.WriteFile(p.path, zipBytes, 0644)
	if err != nil {
		return err
	}

	// reload zip reader
	zipFile, err := os.Open(p.path)
	if err != nil {
		return err
	}
	defer zipFile.Close()

	zipFileInfo, err := zipFile.Stat()
	if err != nil {
		return err
	}

	p.zipReader, err = zip.NewReader(zipFile, zipFileInfo.Size())
	if err != nil {
		return err
	}

	return nil
}