github.com/taylorchu/generic@v0.0.0-20171113184323-cd81575befa2/rewrite/rewrite_package_name.go (about)

     1  package rewrite
     2  
     3  import (
     4  	"errors"
     5  	"os"
     6  	"path/filepath"
     7  )
     8  
     9  // rewritePackageName sets current package name.
    10  func (s *Spec) rewritePackageName(pkg *Package) error {
    11  	pkgName := filepath.Base(s.Name)
    12  	if s.Local {
    13  		pkgName = os.Getenv("GOPACKAGE")
    14  		if pkgName == "" {
    15  			return errors.New("GOPACKAGE cannot be empty")
    16  		}
    17  	}
    18  	for _, node := range pkg.Files {
    19  		node.Name.Name = pkgName
    20  	}
    21  	return nil
    22  }