internal/fetch/unit.go

// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package fetch

import (
	"path"
	"sort"

	"golang.org/x/pkgsite/internal"
	"golang.org/x/pkgsite/internal/licenses"
	"golang.org/x/pkgsite/internal/stdlib"
)

// moduleUnit returns the requested unit in a given module, along
// with the contents for the unit.
func moduleUnit(modulePath string, unitMeta *internal.UnitMeta,
	pkg *goPackage,
	readme *internal.Readme,
	d *licenses.Detector) *internal.Unit {

	suffix := internal.Suffix(unitMeta.Path, modulePath)
	if modulePath == stdlib.ModulePath {
		suffix = unitMeta.Path
	}
	isRedist, lics := d.PackageInfo(suffix)
	var meta []*licenses.Metadata
	for _, l := range lics {
		meta = append(meta, l.Metadata)
	}
	unit := &internal.Unit{
		UnitMeta:          *unitMeta,
		Licenses:          meta,
		IsRedistributable: isRedist,
	}
	if readme != nil {
		unit.Readme = readme
	}
	if pkg != nil {
		unit.Name = pkg.name
		unit.Imports = pkg.imports
		unit.Documentation = pkg.docs
		var bcs []internal.BuildContext
		for _, d := range unit.Documentation {
			bcs = append(bcs, internal.BuildContext{GOOS: d.GOOS, GOARCH: d.GOARCH})
		}
		sort.Slice(bcs, func(i, j int) bool {
			return internal.CompareBuildContexts(bcs[i], bcs[j]) < 0
		})
		unit.BuildContexts = bcs
	}
	return unit
}

// moduleUnitMetas returns UnitMetas for all the units in a given module.
func moduleUnitMetas(minfo internal.ModuleInfo, pkgs []*packageMeta) []*internal.UnitMeta {
	pkgLookup := map[string]*packageMeta{}
	for _, pkg := range pkgs {
		pkgLookup[pkg.path] = pkg
	}
	dirPaths := unitPaths(minfo.ModulePath, pkgs)

	var ums []*internal.UnitMeta
	for _, dirPath := range dirPaths {
		um := &internal.UnitMeta{
			ModuleInfo: minfo,
			Path:       dirPath,
		}
		if pkg, ok := pkgLookup[dirPath]; ok {
			um.Name = pkg.name
		}
		ums = append(ums, um)
	}
	return ums
}

// unitPaths returns the paths for all the units in a module.
func unitPaths(modulePath string, packageMetas []*packageMeta) []string {
	shouldContinue := func(p string) bool {
		if modulePath == stdlib.ModulePath {
			return p != "."
		}
		return len(p) > len(modulePath)
	}

	pathSet := map[string]bool{modulePath: true}
	for _, p := range packageMetas {
		for p := p.path; shouldContinue(p); p = path.Dir(p) {
			pathSet[p] = true
		}
	}

	var dirPaths []string
	for d := range pathSet {
		dirPaths = append(dirPaths, d)
	}
	return dirPaths
}