// Copyright 2011 The Go Authors.  All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package main

import (
	"go/ast"
)

var netudpgroupFix = fix{
	"netudpgroup",
	netudpgroup,
	`Adapt 1-argument calls of net.(*UDPConn).JoinGroup, LeaveGroup to use 2-argument form.

http://codereview.appspot.com/4815074
`,
}

func init() {
	register(netudpgroupFix)
}

func netudpgroup(f *ast.File) bool {
	if !imports(f, "net") {
		return false
	}

	fixed := false
	for _, d := range f.Decls {
		fd, ok := d.(*ast.FuncDecl)
		if !ok {
			continue
		}
		walk(fd.Body, func(n interface{}) {
			ce, ok := n.(*ast.CallExpr)
			if !ok {
				return
			}
			se, ok := ce.Fun.(*ast.SelectorExpr)
			if !ok || len(ce.Args) != 1 {
				return
			}
			switch se.Sel.String() {
			case "JoinGroup", "LeaveGroup":
				// c.JoinGroup(a) -> c.JoinGroup(nil, a)
				// c.LeaveGroup(a) -> c.LeaveGroup(nil, a)
				arg := ce.Args[0]
				ce.Args = make([]ast.Expr, 2)
				ce.Args[0] = ast.NewIdent("nil")
				ce.Args[1] = arg
				fixed = true
			}
		})
	}
	return fixed
}