add methods to allow supplying your own custom http.Client

rtorrent/rtorrent.go

@@ -2,13 +2,15 @@ package rtorrent
 
 import (
 	"fmt"
+	"net/http"
 
 	"github.com/mrobinsn/go-rtorrent/xmlrpc"
 	"github.com/pkg/errors"
 )
 
 // RTorrent is used to communicate with a remote rTorrent instance
 type RTorrent struct {
+	addr         string
 	xmlrpcClient *xmlrpc.Client
 }
 
@@ -59,10 +61,17 @@ func (f *File) Pretty() string {
 // Pass in a true value for `insecure` to turn off certificate verification
 func New(addr string, insecure bool) *RTorrent {
 	return &RTorrent{
+		addr:         addr,
 		xmlrpcClient: xmlrpc.NewClient(addr, insecure),
 	}
 }
 
+// WithHTTPClient allows you to a provide a custom http.Client.
+func (r *RTorrent) WithHTTPClient(client *http.Client) *RTorrent {
+	r.xmlrpcClient = xmlrpc.NewClientWithHTTPClient(r.addr, client)
+	return r
+}
+
 // Add adds a new torrent by URL
 func (r *RTorrent) Add(url string) error {
 	_, err := r.xmlrpcClient.Call("load_start", url)

xmlrpc/client.go

@@ -32,6 +32,15 @@ func NewClient(addr string, insecure bool) *Client {
 	}
 }
 
+// NewClientWithHTTPClient returns a new instance of Client.
+// This allows you to use a custom http.Client setup for your needs.
+func NewClientWithHTTPClient(addr string, client *http.Client) *Client {
+	return &Client{
+		addr:       addr,
+		httpClient: client,
+	}
+}
+
 // Call calls the method with "name" with the given args
 // Returns the result, and an error for communication errors
 func (c *Client) Call(name string, args ...interface{}) (interface{}, error) {