forked from mirror/pkger
97 lines
1.9 KiB
Go
97 lines
1.9 KiB
Go
package here
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/markbates/pkger/internal/takeon/github.com/markbates/hepa"
|
|
"github.com/markbates/pkger/internal/takeon/github.com/markbates/hepa/filters"
|
|
)
|
|
|
|
// Info represents details about the directory/package
|
|
type Info struct {
|
|
Dir string
|
|
ImportPath string
|
|
Name string
|
|
// Imports []string
|
|
Module Module
|
|
}
|
|
|
|
func (fi Info) MarshalJSON() ([]byte, error) {
|
|
mm := map[string]interface{}{
|
|
"ImportPath": fi.ImportPath,
|
|
"Name": fi.Name,
|
|
// "Imports": fi.Imports,
|
|
"Module": fi.Module,
|
|
}
|
|
|
|
hep := hepa.New()
|
|
hep = hepa.With(hep, filters.Home())
|
|
hep = hepa.With(hep, filters.Golang())
|
|
|
|
cm := map[string]string{
|
|
"Dir": fi.Dir,
|
|
}
|
|
|
|
for k, v := range cm {
|
|
b, err := hep.Filter([]byte(v))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
mm[k] = string(b)
|
|
}
|
|
|
|
return json.Marshal(mm)
|
|
}
|
|
|
|
func (i Info) FilePath(paths ...string) string {
|
|
res := []string{i.Dir}
|
|
for _, p := range paths {
|
|
p = strings.TrimPrefix(p, i.Dir)
|
|
p = strings.TrimPrefix(p, "/")
|
|
if runtime.GOOS == "windows" {
|
|
p = strings.Replace(p, "/", "\\", -1)
|
|
}
|
|
res = append(res, p)
|
|
}
|
|
return filepath.Join(res...)
|
|
}
|
|
|
|
func (i Info) Open(p string) (*os.File, error) {
|
|
return os.Open(i.FilePath(p))
|
|
}
|
|
|
|
// ModuleName returns the name of the current
|
|
// module, or if not using modules, the current
|
|
// package. These *might* not match.
|
|
func (i Info) ModuleName() string {
|
|
if i.Mods() {
|
|
return i.Module.Path
|
|
}
|
|
return i.ImportPath
|
|
}
|
|
|
|
// IsZero checks if the type has been filled
|
|
// with rich chocolately data goodness
|
|
func (i Info) IsZero() bool {
|
|
return i.String() == Info{}.String()
|
|
}
|
|
|
|
// Mods returns whether Go modules are used
|
|
// in this directory/package.
|
|
func (i Info) Mods() bool {
|
|
return !i.Module.IsZero()
|
|
}
|
|
|
|
func (i Info) String() string {
|
|
b, err := json.MarshalIndent(i, "", " ")
|
|
if err != nil {
|
|
return err.Error()
|
|
}
|
|
s := string(b)
|
|
return s
|
|
}
|