package archive

import (
	"context"

	"git.kmsign.ru/royalcat/tstor/src/vfs"
	"github.com/bodgit/sevenzip"
	"github.com/royalcat/ctxio"
)

var _ archiveLoader = SevenZipLoader

func SevenZipLoader(ctx context.Context, archivePath string, ctxreader vfs.File, size int64) (map[string]fileEntry, error) {
	hash, err := vfs.FileHash(ctx, ctxreader)
	if err != nil {
		return nil, err
	}

	reader := ctxio.IoReaderAt(ctx, ctxreader)
	r, err := sevenzip.NewReader(reader, size)
	if err != nil {
		return nil, err
	}

	out := make(map[string]fileEntry)
	for i, f := range r.File {
		if f.FileInfo().IsDir() {
			continue
		}

		af := func(ctx context.Context) (ctxio.ReadCloser, error) {
			reader := ctxio.IoReaderAt(ctx, ctxreader)
			zr, err := sevenzip.NewReader(reader, size)
			if err != nil {
				return nil, err
			}

			rc, err := zr.File[i].Open()
			if err != nil {
				return nil, err
			}

			return ctxio.WrapIoReadCloser(rc), nil
		}

		info := f.FileInfo()

		rr := newRandomReaderFromLinear(archiveFileIndex{archiveHash: hash, filename: f.Name}, info.Size(), af)

		out[vfs.AbsPath(f.Name)] = fileEntry{
			FileInfo: f.FileInfo(),
			open: func(ctx context.Context) (vfs.File, error) {
				return newArchiveFile(info.Name(), info.Size(), rr), nil
			},
		}
	}

	return out, nil
}