Add experimental -M support for go import path overrides

Author: cludden

docs/docs/configuration/plugin.mdx

@@ -265,51 +265,50 @@ protoc --go_temporal_opt=ignore-acronyms=AWS;SQL
 </TabItem>
 </Tabs>
 
-### nexus
+### module-map [EXPERIMENTAL]
 
-`bool`
+`[]string`
 
-Enables experimental [Nexus generation](/docs/guides/nexus).
+Repeated set of `<proto_file>=<go_import_path>` (e.g. -Mgoogle/protobuf/timestamp.proto=google.golang.org/protobuf/types/known/timestamppb)
 
 <Tabs>
-<TabItem value="nexus-buf" label="Buf">
+<TabItem value="module-map-buf" label="Buf">
 ```yaml title="buf.gen.yaml"
 plugins:
   - plugin: go_temporal
     out: gen
     opt: 
-      - nexus=true
+      - Mgoogle/protobuf/timestamp.proto=google.golang.org/protobuf/types/known/timestamppb
     strategy: all
 ```
 </TabItem>
-<TabItem value="nexus-protoc" label="Protoc">
+<TabItem value="module-map-protoc" label="Protoc">
 ```sh
-protoc --go_temporal_opt=nexus=trues
+protoc --go_temporal_opt=Mgoogle/protobuf/timestamp.proto=google.golang.org/protobuf/types/known/timestamppb
 ```
 </TabItem>
 </Tabs>
 
-### nexus-exclude-service-tags
+### nexus
 
-`string`
+`bool`
 
-Semicolon-delimited list of Nexus service tags to exclude from generation.
+Enables experimental [Nexus generation](/docs/guides/nexus).
 
 <Tabs>
-<TabItem value="nexus-exclude-service-tags-buf" label="Buf">
+<TabItem value="nexus-buf" label="Buf">
 ```yaml title="buf.gen.yaml"
 plugins:
   - plugin: go_temporal
     out: gen
-    opt:
+    opt: 
       - nexus=true
-      - nexus-exclude-service-tags=disabled;internal
     strategy: all
 ```
 </TabItem>
-<TabItem value="nexus-exclude-service-tags-protoc" label="Protoc">
+<TabItem value="nexus-protoc" label="Protoc">
 ```sh
-protoc --go_temporal_opt=nexus=true,nexus-exclude-service-tags=disabled;internal
+protoc --go_temporal_opt=nexus=trues
 ```
 </TabItem>
 </Tabs>
@@ -339,27 +338,27 @@ protoc --go_temporal_opt=nexus=true,nexus-exclude-operation-tags=disabled;intern
 </TabItem>
 </Tabs>
 
-### nexus-include-service-tags
+### nexus-exclude-service-tags
 
 `string`
 
-Semicolon-delimited list of Nexus service tags to include in generation.
+Semicolon-delimited list of Nexus service tags to exclude from generation.
 
 <Tabs>
-<TabItem value="nexus-include-service-tags-buf" label="Buf">
+<TabItem value="nexus-exclude-service-tags-buf" label="Buf">
 ```yaml title="buf.gen.yaml"
 plugins:
   - plugin: go_temporal
     out: gen
     opt:
       - nexus=true
-      - nexus-include-service-tags=enabled;internal
+      - nexus-exclude-service-tags=disabled;internal
     strategy: all
 ```
 </TabItem>
-<TabItem value="nexus-include-service-tags-protoc" label="Protoc">
+<TabItem value="nexus-exclude-service-tags-protoc" label="Protoc">
 ```sh
-protoc --go_temporal_opt=nexus=true,nexus-include-service-tags=enabled;internal
+protoc --go_temporal_opt=nexus=true,nexus-exclude-service-tags=disabled;internal
 ```
 </TabItem>
 </Tabs>
@@ -389,6 +388,31 @@ protoc --go_temporal_opt=nexus=true,nexus-include-operation-tags=enabled;interna
 </TabItem>
 </Tabs>
 
+### nexus-include-service-tags
+
+`string`
+
+Semicolon-delimited list of Nexus service tags to include in generation.
+
+<Tabs>
+<TabItem value="nexus-include-service-tags-buf" label="Buf">
+```yaml title="buf.gen.yaml"
+plugins:
+  - plugin: go_temporal
+    out: gen
+    opt:
+      - nexus=true
+      - nexus-include-service-tags=enabled;internal
+    strategy: all
+```
+</TabItem>
+<TabItem value="nexus-include-service-tags-protoc" label="Protoc">
+```sh
+protoc --go_temporal_opt=nexus=true,nexus-include-service-tags=enabled;internal
+```
+</TabItem>
+</Tabs>
+
 ### patches
 
 `string`

internal/plugin/parse.go

@@ -130,6 +130,9 @@ func parse(p *Plugin) (*Manifest, error) {
 	}
 
 	for _, file := range p.Files {
+		if goImportPath, ok := p.moduleMap[string(file.Desc.FullName())]; ok {
+			file.GoImportPath = protogen.GoImportPath(goImportPath)
+		}
 		if !file.Generate {
 			continue
 		}

internal/plugin/plugin.go

@@ -43,6 +43,7 @@ type Config struct {
 	EnableDebugLogging         bool
 	EnablePatchSupport         bool
 	EnableXNS                  bool
+	ModuleMap                  []string
 	NexusEnabled               bool
 	NexusExcludeOperationTags  string
 	NexusIncludeOperationTags  string
@@ -66,6 +67,7 @@ type Plugin struct {
 	excludeServiceTags   map[string]struct{}
 	includeOperationTags map[string]struct{}
 	includeServiceTags   map[string]struct{}
+	moduleMap            map[string]string
 }
 
 func Run(commit, version string) {
@@ -88,6 +90,7 @@ func New(commit, version string) *Plugin {
 	flags.BoolVar(&cfg.EnablePatchSupport, "enable-patch-support", false, "enables support for alta/protopatch renaming")
 	flags.BoolVar(&cfg.EnableXNS, "enable-xns", false, "enable experimental cross-namespace workflow client")
 	flags.StringVar(&cfg.IgnoreAcronyms, "ignore-acronyms", "", "semicolon-delimited string of acronyms to ignore when converting generated output to camel case")
+	flags.StringArrayVarP(&cfg.ModuleMap, "module-map", "M", []string{}, "[EXPERIMENTAL] list of <proto_file>=<go_import_path> (e.g. -Mgoogle/protobuf/timestamp.proto=google.golang.org/protobuf/types/known/timestamppb)")
 	flags.BoolVar(&cfg.NexusEnabled, "nexus", false, "enable nexus handler generation")
 	flags.StringVar(&cfg.NexusExcludeOperationTags, "nexus-exclude-operation-tags", "", "semicolon-delimited list of operation tags to exclude from nexus generation")
 	flags.StringVar(&cfg.NexusIncludeOperationTags, "nexus-include-operation-tags", "", "semicolon-delimited list of operation tags to include in nexus generation")
@@ -116,6 +119,14 @@ func (p *Plugin) Run(plugin *protogen.Plugin) (err error) {
 	plugin.SupportedEditionsMinimum = descriptorpb.Edition_EDITION_PROTO3
 	plugin.SupportedEditionsMaximum = descriptorpb.Edition_EDITION_2023
 
+	for _, m := range p.cfg.ModuleMap {
+		fields := strings.SplitN(m, "=", 2)
+		if len(fields) != 2 {
+			return fmt.Errorf("invalid module map: %s", m)
+		}
+		p.moduleMap[fields[0]] = fields[1]
+	}
+
 	p.Plugin = plugin
 	services, err := parse(p)
 	if err != nil {