package coverage

import "strings"

// Files contains coverage files indexed by file path
type Files struct {
	packagePath string
	files       map[string]File
}

// File represents every coverage chunk for a file
type File []Coverage

// Coverage represents a covered chunk
type Coverage struct {
	StartLine   uint64
	StartColumn uint64
	EndLine     uint64
	EndColumn   uint64
}

// GetFile returns the formatted coverage for the file located at @path
func (f *Files) GetFile(path string) []File {
	var isDir bool = !strings.HasSuffix(path, ".go")

	files := make([]File, 0)

	// search in coverage files
	for fPath, fileCoverage := range f.files {

		if !isDir && path == fPath {
			files = append(files, fileCoverage)
			// if not directory search, stop at first result
			break
		}

		if isDir && (path == "." || strings.HasPrefix(fPath, path)) {
			files = append(files, fileCoverage)
		}
	}

	return files
}