--- /dev/null
+package shell
+
+import (
+ "context"
+ "encoding/json"
+ "errors"
+ "fmt"
+ "io"
+ "io/ioutil"
+ "net/http"
+ "net/url"
+ "os"
+ "strings"
+
+ files "github.com/ipfs/go-ipfs-files"
+)
+
+type Request struct {
+ ApiBase string
+ Command string
+ Args []string
+ Opts map[string]string
+ Body io.Reader
+ Headers map[string]string
+}
+
+func NewRequest(ctx context.Context, url, command string, args ...string) *Request {
+ if !strings.HasPrefix(url, "http") {
+ url = "http://" + url
+ }
+
+ opts := map[string]string{
+ "encoding": "json",
+ "stream-channels": "true",
+ }
+ return &Request{
+ ApiBase: url + "/api/v0",
+ Command: command,
+ Args: args,
+ Opts: opts,
+ Headers: make(map[string]string),
+ }
+}
+
+type trailerReader struct {
+ resp *http.Response
+}
+
+func (r *trailerReader) Read(b []byte) (int, error) {
+ n, err := r.resp.Body.Read(b)
+ if err != nil {
+ if e := r.resp.Trailer.Get("X-Stream-Error"); e != "" {
+ err = errors.New(e)
+ }
+ }
+ return n, err
+}
+
+func (r *trailerReader) Close() error {
+ return r.resp.Body.Close()
+}
+
+type Response struct {
+ Output io.ReadCloser
+ Error *Error
+}
+
+func (r *Response) Close() error {
+ if r.Output != nil {
+ // always drain output (response body)
+ _, err1 := io.Copy(ioutil.Discard, r.Output)
+ err2 := r.Output.Close()
+ if err1 != nil {
+ return err1
+ }
+ if err2 != nil {
+ return err2
+ }
+ }
+ return nil
+}
+
+func (r *Response) Decode(dec interface{}) error {
+ defer r.Close()
+ if r.Error != nil {
+ return r.Error
+ }
+
+ return json.NewDecoder(r.Output).Decode(dec)
+}
+
+type Error struct {
+ Command string
+ Message string
+ Code int
+}
+
+func (e *Error) Error() string {
+ var out string
+ if e.Command != "" {
+ out = e.Command + ": "
+ }
+ if e.Code != 0 {
+ out = fmt.Sprintf("%s%d: ", out, e.Code)
+ }
+ return out + e.Message
+}
+
+func (r *Request) Send(c *http.Client) (*Response, error) {
+ url := r.getURL()
+ req, err := http.NewRequest("POST", url, r.Body)
+ if err != nil {
+ return nil, err
+ }
+
+ // Add any headers that were supplied via the RequestBuilder.
+ for k, v := range r.Headers {
+ req.Header.Add(k, v)
+ }
+
+ if fr, ok := r.Body.(*files.MultiFileReader); ok {
+ req.Header.Set("Content-Type", "multipart/form-data; boundary="+fr.Boundary())
+ req.Header.Set("Content-Disposition", "form-data; name=\"files\"")
+ }
+
+ resp, err := c.Do(req)
+ if err != nil {
+ return nil, err
+ }
+
+ contentType := resp.Header.Get("Content-Type")
+ parts := strings.Split(contentType, ";")
+ contentType = parts[0]
+
+ nresp := new(Response)
+
+ nresp.Output = &trailerReader{resp}
+ if resp.StatusCode >= http.StatusBadRequest {
+ e := &Error{
+ Command: r.Command,
+ }
+ switch {
+ case resp.StatusCode == http.StatusNotFound:
+ e.Message = "command not found"
+ case contentType == "text/plain":
+ out, err := ioutil.ReadAll(resp.Body)
+ if err != nil {
+ fmt.Fprintf(os.Stderr, "ipfs-shell: warning! response (%d) read error: %s\n", resp.StatusCode, err)
+ }
+ e.Message = string(out)
+ case contentType == "application/json":
+ if err = json.NewDecoder(resp.Body).Decode(e); err != nil {
+ fmt.Fprintf(os.Stderr, "ipfs-shell: warning! response (%d) unmarshall error: %s\n", resp.StatusCode, err)
+ }
+ default:
+ fmt.Fprintf(os.Stderr, "ipfs-shell: warning! unhandled response (%d) encoding: %s", resp.StatusCode, contentType)
+ out, err := ioutil.ReadAll(resp.Body)
+ if err != nil {
+ fmt.Fprintf(os.Stderr, "ipfs-shell: response (%d) read error: %s\n", resp.StatusCode, err)
+ }
+ e.Message = fmt.Sprintf("unknown ipfs-shell error encoding: %q - %q", contentType, out)
+ }
+ nresp.Error = e
+ nresp.Output = nil
+
+ // drain body and close
+ io.Copy(ioutil.Discard, resp.Body)
+ resp.Body.Close()
+ }
+
+ return nresp, nil
+}
+
+func (r *Request) getURL() string {
+
+ values := make(url.Values)
+ for _, arg := range r.Args {
+ values.Add("arg", arg)
+ }
+ for k, v := range r.Opts {
+ values.Add(k, v)
+ }
+
+ return fmt.Sprintf("%s/%s?%s", r.ApiBase, r.Command, values.Encode())
+}
OSDN Git Service
