tstor/node/zip.go

package node

import (
	"archive/zip"
	"context"
	"io"
	"log"
	"syscall"

	"github.com/ajnavarro/distribyted/iio"
	"github.com/hanwen/go-fuse/v2/fs"
	"github.com/hanwen/go-fuse/v2/fuse"
)

var _ fs.NodeGetattrer = &Zip{}
var _ fs.NodeOpendirer = &Zip{}

type Zip struct {
	fs.Inode

	reader ReaderFunc
	size   int64
	files  []*zip.File
}

func NewZip(reader ReaderFunc, size int64) *Zip {
	return &Zip{
		reader: reader,
		size:   size,
	}
}

func (z *Zip) Opendir(ctx context.Context) syscall.Errno {
	if z.files == nil {
		r, err := z.reader()
		if err != nil {
			log.Println("error opening reader for zip", err)
			return syscall.EIO
		}
		zr, err := zip.NewReader(r, z.size)
		if err != nil {
			log.Println("error getting zip reader from reader", err)
			return syscall.EIO
		}

		for _, f := range zr.File {
			f := f
			if f.FileInfo().IsDir() {
				continue
			}
			LoadNodeByPath(
				ctx,
				f.Name,
				func() (io.ReaderAt, error) {
					zfr, err := f.Open()
					if err != nil {
						log.Println("ERROR OPENING ZIP", err)
						return nil, err
					}

					return iio.NewDiskTeeReader(zfr)
				},
				&z.Inode,
				int64(f.UncompressedSize64),
				0,
				0,
			)
		}

		z.files = zr.File
	}

	return fs.OK
}

func (z *Zip) Getattr(ctx context.Context, f fs.FileHandle, out *fuse.AttrOut) syscall.Errno {
	out.Mode = syscall.S_IFDIR & 0555
	out.Size = uint64(z.size)

	return fs.OK
}