package file

import (
	"github.com/anchore/syft/internal/log"
	"github.com/anchore/syft/syft/source"
)

type ClassificationCataloger struct {
	classifiers []Classifier
}

func NewClassificationCataloger(classifiers []Classifier) (*ClassificationCataloger, error) {
	return &ClassificationCataloger{
		classifiers: classifiers,
	}, nil
}

func (i *ClassificationCataloger) Catalog(resolver source.FileResolver) (map[source.Location][]Classification, error) {
	results := make(map[source.Location][]Classification)

	numResults := 0
	for location := range resolver.AllLocations() {
		for _, classifier := range i.classifiers {
			result, err := classifier.Classify(resolver, location)
			if err != nil {
				return nil, err
			}
			if result != nil {
				results[location] = append(results[location], *result)
				numResults++
			}
		}
	}
	log.Debugf("file classifier discovered %d results", numResults)

	return results, nil
}