diff --git a/pkg/docsonnet/load.go b/pkg/docsonnet/load.go index 1bf643d..dd2e3c8 100644 --- a/pkg/docsonnet/load.go +++ b/pkg/docsonnet/load.go @@ -94,9 +94,16 @@ func newImporter(paths []string) (*importer, error) { }, nil } +var docUtilPaths = []string{ + "doc-util/main.libsonnet", + "github.com/sh0rez/docsonnet/doc-util/main.libsonnet", +} + func (i *importer) Import(importedFrom, importedPath string) (contents jsonnet.Contents, foundAt string, err error) { - if importedPath == "doc-util/main.libsonnet" { - return i.util, "", nil + for _, p := range docUtilPaths { + if importedPath == p { + return i.util, "", nil + } } return i.fi.Import(importedFrom, importedPath)