Imported Upstream version 2012.02.22upstream-weekly/2012.02.22

1 files changed, 63 insertions, 0 deletions

diff --git a/src/cmd/fix/httputil.go b/src/cmd/fix/httputil.go
new file mode 100644
index 000000000..86c42e160
--- /dev/null
+++ b/src/cmd/fix/httputil.go
@@ -0,0 +1,63 @@
+// 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"
+
+func init() {
+	register(httputilFix)
+}
+
+var httputilFix = fix{
+	"httputil",
+	"2011-11-18",
+	httputil,
+	`Move some functions in http package into httputil package.
+
+http://codereview.appspot.com/5336049
+`,
+}
+
+var httputilFuncs = []string{
+	"DumpRequest",
+	"DumpRequestOut",
+	"DumpResponse",
+	"NewChunkedReader",
+	"NewChunkedWriter",
+	"NewClientConn",
+	"NewProxyClientConn",
+	"NewServerConn",
+	"NewSingleHostReverseProxy",
+}
+
+func httputil(f *ast.File) bool {
+	if imports(f, "net/http/httputil") {
+		return false
+	}
+
+	fixed := false
+
+	walk(f, func(n interface{}) {
+		// Rename package name.
+		if expr, ok := n.(ast.Expr); ok {
+			for _, s := range httputilFuncs {
+				if isPkgDot(expr, "http", s) {
+					if !fixed {
+						addImport(f, "net/http/httputil")
+						fixed = true
+					}
+					expr.(*ast.SelectorExpr).X.(*ast.Ident).Name = "httputil"
+				}
+			}
+		}
+	})
+
+	// Remove the net/http import if no longer needed.
+	if fixed && !usesImport(f, "net/http") {
+		deleteImport(f, "net/http")
+	}
+
+	return fixed
+}