fix(entproto): allow specifying import path

Basically a merge of: ent#616

Author: kdevo

entproto/extension.go

@@ -58,6 +58,7 @@ type Extension struct {
 	entc.DefaultExtension
 	protoDir    string
 	skipGenFile bool
+	goPkg       string // inspired from: https://github.com/ent/contrib/pull/616
 }
 
 // WithProtoDir sets the directory where the generated .proto files will be written.
@@ -67,6 +68,13 @@ func WithProtoDir(dir string) ExtensionOption {
 	}
 }
 
+// WithProtoDir sets the directory where the generated .proto files will be written.
+func WithGoPkg(pkg string) ExtensionOption {
+	return func(e *Extension) {
+		e.protoDir = pkg
+	}
+}
+
 // SkipGenFile skips the generation of a generate.go file next to each .proto file.
 func SkipGenFile() ExtensionOption {
 	return func(e *Extension) {
@@ -197,7 +205,7 @@ func (e *Extension) generate(g *gen.Graph) error {
 				if err != nil {
 					return err
 				}
-				contents := protocGenerateGo(fd, toSchema, toEnt, g.Config.Package)
+				contents := e.protocGenerateGo(fd, toSchema, toEnt, g.Config.Package)
 				if err := os.WriteFile(genGoPath, []byte(contents), 0600); err != nil {
 					return fmt.Errorf("entproto: failed generating generate.go file for %q: %w", protoFilePath, err)
 				}
@@ -216,7 +224,7 @@ func fileExists(fpath string) bool {
 	return true
 }
 
-func protocGenerateGo(fd *desc.FileDescriptor, toSchemaDir, entPath, entPackage string) string {
+func (e *Extension) protocGenerateGo(fd *desc.FileDescriptor, toSchemaDir, entPath, entPackage string) string {
 	levelsUp := len(strings.Split(fd.GetPackage(), "."))
 	toProtoBase := ""
 	for i := 0; i < levelsUp; i++ {
@@ -235,5 +243,8 @@ func protocGenerateGo(fd *desc.FileDescriptor, toSchemaDir, entPath, entPackage
 	}
 	goGen := fmt.Sprintf("//go:generate %s", strings.Join(protocCmd, " "))
 	goPkgName := extractLastFqnPart(fd.GetPackage())
+	if e.goPkg != "" {
+		goPkgName = e.goPkg
+	}
 	return fmt.Sprintf("package %s\n%s\n", goPkgName, goGen)
 }