package packages

import (
	"go/build"
	"path/filepath"
)

type Package struct {
	bp *build.Package

	isFake bool
	dir    string // dir != bp.dir only if isFake == true
}

func (pkg *Package) Files(includeTest bool) []string {
	pkgFiles := append([]string{}, pkg.bp.GoFiles...)

	// TODO: add cgo files
	if includeTest {
		pkgFiles = append(pkgFiles, pkg.TestFiles()...)
	}

	for i, f := range pkgFiles {
		pkgFiles[i] = filepath.Join(pkg.bp.Dir, f)
	}

	return pkgFiles
}

func (pkg *Package) Dir() string {
	if pkg.dir != "" { // for fake packages
		return pkg.dir
	}

	return pkg.bp.Dir
}

func (pkg *Package) IsTestOnly() bool {
	return len(pkg.bp.GoFiles) == 0
}

func (pkg *Package) TestFiles() []string {
	var pkgFiles []string
	pkgFiles = append(pkgFiles, pkg.bp.TestGoFiles...)
	pkgFiles = append(pkgFiles, pkg.bp.XTestGoFiles...)
	return pkgFiles
}