package main
 
 import (
 	"fmt"
 	"net/http"
 	"regexp"
 
 	"github.com/go-check/check"
 )
 
 // unescapeBackslashSemicolonParens unescapes \;()
 func unescapeBackslashSemicolonParens(s string) string {

 	re := regexp.MustCompile(`\\;`)

 	ret := re.ReplaceAll([]byte(s), []byte(";"))
 

 	re = regexp.MustCompile(`\\\(`)

 	ret = re.ReplaceAll([]byte(ret), []byte("("))
 

 	re = regexp.MustCompile(`\\\)`)

 	ret = re.ReplaceAll([]byte(ret), []byte(")"))
 

 	re = regexp.MustCompile(`\\\\`)
 	ret = re.ReplaceAll([]byte(ret), []byte(`\`))

 
 	return string(ret)
 }
 
 func regexpCheckUA(c *check.C, ua string) {
 	re := regexp.MustCompile("(?P<dockerUA>.+) UpstreamClient(?P<upstreamUA>.+)")
 	substrArr := re.FindStringSubmatch(ua)
 
 	c.Assert(substrArr, check.HasLen, 3, check.Commentf("Expected 'UpstreamClient()' with upstream client UA"))
 	dockerUA := substrArr[1]
 	upstreamUAEscaped := substrArr[2]
 
 	// check dockerUA looks correct
 	reDockerUA := regexp.MustCompile("^docker/[0-9A-Za-z+]")
 	bMatchDockerUA := reDockerUA.MatchString(dockerUA)
 	c.Assert(bMatchDockerUA, check.Equals, true, check.Commentf("Docker Engine User-Agent malformed"))
 
 	// check upstreamUA looks correct
 	// Expecting something like:  Docker-Client/1.11.0-dev (linux)
 	upstreamUA := unescapeBackslashSemicolonParens(upstreamUAEscaped)
 	reUpstreamUA := regexp.MustCompile("^\\(Docker-Client/[0-9A-Za-z+]")
 	bMatchUpstreamUA := reUpstreamUA.MatchString(upstreamUA)
 	c.Assert(bMatchUpstreamUA, check.Equals, true, check.Commentf("(Upstream) Docker Client User-Agent malformed"))
 }
 

 func registerUserAgentHandler(reg *testRegistry, result *string) {

 	reg.registerHandler("/v2/", func(w http.ResponseWriter, r *http.Request) {
 		w.WriteHeader(404)
 		var ua string
 		for k, v := range r.Header {
 			if k == "User-Agent" {
 				ua = v[0]
 			}
 		}

 		*result = ua

 	})

 }
 
 // TestUserAgentPassThroughOnPull verifies that when an image is pulled from
 // a registry, the registry should see a User-Agent string of the form
 //   [docker engine UA] UptreamClientSTREAM-CLIENT([client UA])
 func (s *DockerRegistrySuite) TestUserAgentPassThrough(c *check.C) {
 	var (
 		buildUA string
 		pullUA  string
 		pushUA  string
 		loginUA string
 	)
 
 	buildReg, err := newTestRegistry(c)
 	c.Assert(err, check.IsNil)
 	registerUserAgentHandler(buildReg, &buildUA)
 	buildRepoName := fmt.Sprintf("%s/busybox", buildReg.hostport)

 	pullReg, err := newTestRegistry(c)

 	c.Assert(err, check.IsNil)

 	registerUserAgentHandler(pullReg, &pullUA)
 	pullRepoName := fmt.Sprintf("%s/busybox", pullReg.hostport)

 	pushReg, err := newTestRegistry(c)
 	c.Assert(err, check.IsNil)
 	registerUserAgentHandler(pushReg, &pushUA)
 	pushRepoName := fmt.Sprintf("%s/busybox", pushReg.hostport)

 	loginReg, err := newTestRegistry(c)
 	c.Assert(err, check.IsNil)
 	registerUserAgentHandler(loginReg, &loginUA)
 
 	err = s.d.Start(
 		"--insecure-registry", buildReg.hostport,
 		"--insecure-registry", pullReg.hostport,
 		"--insecure-registry", pushReg.hostport,
 		"--insecure-registry", loginReg.hostport,
 		"--disable-legacy-registry=true")
 	c.Assert(err, check.IsNil)
 
 	dockerfileName, cleanup1, err := makefile(fmt.Sprintf("FROM %s", buildRepoName))
 	c.Assert(err, check.IsNil, check.Commentf("Unable to create test dockerfile"))
 	defer cleanup1()

 	s.d.Cmd("build", "--file", dockerfileName, ".")

 	regexpCheckUA(c, buildUA)
 
 	s.d.Cmd("login", "-u", "richard", "-p", "testtest", "-e", "testuser@testdomain.com", loginReg.hostport)
 	regexpCheckUA(c, loginUA)

 	s.d.Cmd("pull", pullRepoName)
 	regexpCheckUA(c, pullUA)
 
 	dockerfileName, cleanup2, err := makefile(`FROM scratch
 	ENV foo bar`)
 	c.Assert(err, check.IsNil, check.Commentf("Unable to create test dockerfile"))
 	defer cleanup2()
 	s.d.Cmd("build", "-t", pushRepoName, "--file", dockerfileName, ".")

 	s.d.Cmd("push", pushRepoName)
 	regexpCheckUA(c, pushUA)

 }