add.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package ipfs_api
  2. import (
  3. "encoding/json"
  4. "io"
  5. "io/ioutil"
  6. "context"
  7. "github.com/ipfs/go-ipfs-api"
  8. "github.com/whyrusleeping/go-multipart-files"
  9. )
  10. type object struct {
  11. Hash string
  12. }
  13. // Add a file to ipfs from the given reader, returns the hash of the added file
  14. func (s *MyShell) Add(r io.Reader) (string, error) {
  15. var rc io.ReadCloser
  16. if rclose, ok := r.(io.ReadCloser); ok {
  17. rc = rclose
  18. } else {
  19. rc = ioutil.NopCloser(r)
  20. }
  21. // handler expects an array of files
  22. fr := files.NewReaderFile("", "", rc, nil)
  23. slf := files.NewSliceFile("", "", []files.File{fr})
  24. fileReader := files.NewMultiFileReader(slf, true)
  25. ctx := context.Background()
  26. req := shell.NewRequest(ctx, s.Url, "add")
  27. req.Body = fileReader
  28. req.Opts["progress"] = "false"
  29. req.Opts["chunker"] = "size-1048576"
  30. resp, err := req.Send(s.Client)
  31. if err != nil {
  32. return "", err
  33. }
  34. defer resp.Close()
  35. if resp.Error != nil {
  36. return "", resp.Error
  37. }
  38. var out object
  39. err = json.NewDecoder(resp.Output).Decode(&out)
  40. if err != nil {
  41. return "", err
  42. }
  43. return out.Hash, nil
  44. }