base_file.go

package cmd

import (
	"github.com/dpb587/metalink"
	"github.com/pkg/errors"
)

type Meta4File struct {
	Meta4

	File string `long:"file" description:"The file name"`
}

func (f Meta4File) Get() (metalink.File, error) {
	meta4, err := f.Meta4.Get()
	if err != nil {
		return metalink.File{}, errors.Wrap(err, "Preparing storage")
	}

	fileName := f.File

	if fileName == "" && len(meta4.Files) == 1 {
		fileName = meta4.Files[0].Name
	}

	for _, file := range meta4.Files {
		if file.Name != fileName {
			continue
		}

		return file, nil
	}

	return metalink.File{}, errors.New("File does not exist")
}

func (f Meta4File) Put(put metalink.File) error {
	meta4, err := f.Meta4.Get()
	if err != nil {
		return errors.Wrap(err, "Preparing storage")
	}

	fileName := f.File

	if fileName == "" && len(meta4.Files) == 1 {
		fileName = meta4.Files[0].Name
	}

	for fileIdx, file := range meta4.Files {
		if file.Name != fileName {
			continue
		}

		meta4.Files[fileIdx] = put

		return f.Meta4.Put(meta4)
	}

	return errors.New("File does not exist")
}