client.go

package magopie

import (
	"bytes"
	"fmt"
	"log"
	"net/http"
)

// Client provides methods for talking to a magopie server.
type Client struct {
	ServerAddr string
	ServerKey  string
	verbose    bool
}

// NewClient creates a Client
func NewClient(server, key string) *Client {
	return &Client{ServerAddr: server, ServerKey: key}
}

// Search asks the Magopie server for a list of results
func (c *Client) Search(s string) *TorrentCollection {
	if c.verbose {
		log.Println("Sending Search Request: " + c.ServerAddr + " - " + c.ServerKey)
	}
	ret := &TorrentCollection{}

	req, err := http.NewRequest("GET", c.ServerAddr+"/torrents", nil)
	if err != nil {
		// TODO can we return an error?
		if c.verbose {
			log.Print(err)
		}
		return ret
	}
	vals := req.URL.Query()
	vals.Add("q", s)
	req.URL.RawQuery = vals.Encode()

	signRequest(req, c.ServerKey)

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		// TODO can we return an error?
		if c.verbose {
			log.Print(err)
		}
		return ret
	}
	defer res.Body.Close()

	var buf bytes.Buffer
	if _, err := buf.ReadFrom(res.Body); err != nil {
		// TODO can we return an error?
		if c.verbose {
			log.Print(err)
		}
		return ret
	}

	if err := ret.UnmarshalJSON(buf.Bytes()); err != nil {
		// TODO can we return an error?
		if c.verbose {
			log.Print(err)
		}
		return ret
	}

	return ret
}

// Download triggers the Magopie server to download a particular torrent by ID.
func (c *Client) Download(t *Torrent) bool {
	if c.verbose {
		log.Println("Sending Download Request: " + c.ServerAddr + " - " + c.ServerKey)
	}
	url := fmt.Sprintf("%s/download/%s", c.ServerAddr, t.ID)
	req, err := http.NewRequest("POST", url, nil)
	if err != nil {
		if c.verbose {
			log.Print(err)
		}
		return false
	}

	signRequest(req, c.ServerKey)

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		if c.verbose {
			log.Print(err)
		}
		return false
	}

	if res.StatusCode != http.StatusCreated {
		if c.verbose {
			log.Print("Download request responded with status", res.StatusCode)
		}
		return false
	}

	return true
}

// ListSites returns a collection of the sites that the server knows about
func (c *Client) ListSites() *SiteCollection {
	if c.verbose {
		log.Println("Sending SiteList Request: " + c.ServerAddr + " - " + c.ServerKey)
	}
	ret := &SiteCollection{}

	req, err := http.NewRequest("GET", c.ServerAddr+"/sites", nil)
	if err != nil {
		// TODO can we return an error?
		if c.verbose {
			log.Print(err)
		}
		return ret
	}

	signRequest(req, c.ServerKey)

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		// TODO can we return an error?
		if c.verbose {
			log.Print(err)
		}
		return ret
	}
	defer res.Body.Close()
	var buf bytes.Buffer
	if _, err := buf.ReadFrom(res.Body); err != nil {
		// TODO can we return an error?
		if c.verbose {
			log.Print(err)
		}
		return ret
	}

	if err := ret.UnmarshalJSON(buf.Bytes()); err != nil {
		// TODO can we return an error?
		if c.verbose {
			log.Print(err)
		}
		return ret
	}

	return ret
}

// SetVerbose tells whether the client chatters about what's going on
func (c *Client) SetVerbose(b bool) {
	c.verbose = b
}