Merge pull request #3 from xinsnake/roundtripper

Implement http.RoundTripper

Author: xinsnake

README.md

@@ -22,6 +22,26 @@ response2, err := dr.Execute()
 
 // check error, get response
 ```
+
+Or you can use it with `http.Request`
+
+```go
+t := dac.NewTransport(username, password)
+req, err := http.NewRequest(method, uri, payload)
+
+if err != nil {
+    log.Fatalln(err)
+}
+
+resp, err := t.RoundTrip(req)
+if err != nil {
+    log.Fatalln(err)
+}
+
+fmt.Println(resp)
+```
+
+
 # Todos
 
 * Unit testing

digest_auth_client.go

@@ -17,13 +17,24 @@ type DigestRequest struct {
 	Wa       *wwwAuthenticate
 }
 
-func NewRequest(username string, password string, method string, uri string, body string) DigestRequest {
+type DigestTransport struct {
+	Password string
+	Username string
+}
 
+func NewRequest(username string, password string, method string, uri string, body string) DigestRequest {
 	dr := DigestRequest{}
 	dr.UpdateRequest(username, password, method, uri, body)
 	return dr
 }
 
+func NewTransport(username string, password string) DigestTransport {
+	dt := DigestTransport{}
+	dt.Password = password
+	dt.Username = username
+	return dt
+}
+
 func (dr *DigestRequest) UpdateRequest(username string,
 	password string, method string, uri string, body string) *DigestRequest {
 
@@ -35,6 +46,23 @@ func (dr *DigestRequest) UpdateRequest(username string,
 	return dr
 }
 
+func (dt *DigestTransport) RoundTrip(req *http.Request) (resp *http.Response, err error) {
+	username := dt.Username
+	password := dt.Password
+	method := req.Method
+	uri := req.URL.String()
+
+	var body string
+	if req.Body != nil {
+		buf := new(bytes.Buffer)
+		buf.ReadFrom(req.Body)
+		body = buf.String()
+	}
+
+	dr := NewRequest(username, password, method, uri, body)
+	return dr.Execute()
+}
+
 func (dr *DigestRequest) Execute() (resp *http.Response, err error) {
 
 	if dr.Auth == nil {