simple_extract.go

package boar

import (
	"github.com/itchio/headway/state"
	"github.com/itchio/httpkit/eos"
	"github.com/itchio/httpkit/eos/option"
	"github.com/itchio/savior"
	"github.com/pkg/errors"
)

type SimpleExtractParams struct {
	ArchivePath       string
	DestinationFolder string
	Consumer          *state.Consumer
}

func SimpleExtract(params *SimpleExtractParams) (*savior.ExtractorResult, error) {
	f, err := eos.Open(params.ArchivePath, option.WithConsumer(params.Consumer))
	if err != nil {
		return nil, errors.WithStack(err)
	}
	defer f.Close()

	ai, err := Probe(ProbeParams{
		Consumer: params.Consumer,
		File:     f,
	})
	if err != nil {
		return nil, errors.WithStack(err)
	}

	ex, err := ai.GetExtractor(f, params.Consumer)
	if err != nil {
		return nil, errors.WithStack(err)
	}
	ex.SetConsumer(params.Consumer)

	sink := &savior.FolderSink{
		Directory: params.DestinationFolder,
		Consumer:  params.Consumer,
	}
	defer sink.Close()

	res, err := ex.Resume(nil, sink)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return res, nil
}