fix: issue 707

Taskfile.yml

@@ -13,12 +13,12 @@ tasks:
       - "**/*.go"
     cmds:
       - go fmt ./...
-  
+
   mocks.remove:
     desc: remove all mock files
     cmds:
-    - find . -name '*_mock.go' | xargs rm
-    - rm -rf mocks/
+      - find . -name '*_mock.go' | xargs rm
+      - rm -rf mocks/
 
   mocks.generate:
     desc: generate mockery mocks
@@ -75,6 +75,5 @@ tasks:
       - task: test
       - task: test.e2e
 
-
   default:
     deps: [test.ci]

pkg/fixtures/variadic.go

@@ -0,0 +1,11 @@
+package test
+
+type VariadicFunction = func(args1 string, args2 ...interface{}) interface{}
+
+type Variadic interface {
+	VariadicFunction(str string, vFunc VariadicFunction) error
+}
+
+type VariadicExtra interface {
+	SampleMethod(str string) func(str string, arr []int, a ...interface{})
+}

pkg/generator.go

@@ -481,19 +481,19 @@ func (g *Generator) renderType(ctx context.Context, typ types.Type) string {
 		case 0:
 			return fmt.Sprintf(
 				"func(%s)",
-				g.renderTypeTuple(ctx, t.Params()),
+				g.renderTypeTuple(ctx, t.Params(), t.Variadic()),
 			)
 		case 1:
 			return fmt.Sprintf(
 				"func(%s) %s",
-				g.renderTypeTuple(ctx, t.Params()),
+				g.renderTypeTuple(ctx, t.Params(), t.Variadic()),
 				g.renderType(ctx, t.Results().At(0).Type()),
 			)
 		default:
 			return fmt.Sprintf(
 				"func(%s)(%s)",
-				g.renderTypeTuple(ctx, t.Params()),
-				g.renderTypeTuple(ctx, t.Results()),
+				g.renderTypeTuple(ctx, t.Params(), t.Variadic()),
+				g.renderTypeTuple(ctx, t.Results(), t.Variadic()),
 			)
 		}
 	case *types.Map:
@@ -562,13 +562,20 @@ func (g *Generator) renderType(ctx context.Context, typ types.Type) string {
 	}
 }
 
-func (g *Generator) renderTypeTuple(ctx context.Context, tup *types.Tuple) string {
+func (g *Generator) renderTypeTuple(ctx context.Context, tup *types.Tuple, variadic bool) string {
 	var parts []string
 
 	for i := 0; i < tup.Len(); i++ {
 		v := tup.At(i)
 
-		parts = append(parts, g.renderType(ctx, v.Type()))
+		if variadic && i == tup.Len()-1 {
+			t := v.Type()
+			elem := t.(*types.Slice).Elem()
+
+			parts = append(parts, "..."+g.renderType(ctx, elem))
+		} else {
+			parts = append(parts, g.renderType(ctx, v.Type()))
+		}
 	}
 
 	return strings.Join(parts, " , ")