Move ipfsMultipartWriter to package
All checks were successful
Create and publish a Docker image / build-and-push-image (push) Successful in 59s
All checks were successful
Create and publish a Docker image / build-and-push-image (push) Successful in 59s
This commit is contained in:
parent
ac318177e9
commit
34a681dfd2
51
ipfsnodepin/writer.go
Normal file
51
ipfsnodepin/writer.go
Normal file
@ -0,0 +1,51 @@
|
||||
package ipfsnodepin
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"mime/multipart"
|
||||
"net/textproto"
|
||||
"net/url"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type IpfsMultipartWriter struct {
|
||||
multipart.Writer
|
||||
}
|
||||
|
||||
var quoteEscaper = strings.NewReplacer("\\", "\\\\", `"`, "\\\"")
|
||||
|
||||
func escapeQuotes(s string) string {
|
||||
return quoteEscaper.Replace(s)
|
||||
}
|
||||
|
||||
func NewIpfsMultipartWriter(w io.Writer) *IpfsMultipartWriter {
|
||||
return &IpfsMultipartWriter{
|
||||
Writer: *multipart.NewWriter(w),
|
||||
}
|
||||
}
|
||||
|
||||
func (w *IpfsMultipartWriter) CreateIpfsDirectoryPart(name string) (io.Writer, error) {
|
||||
h := make(textproto.MIMEHeader)
|
||||
encodedName := url.QueryEscape(name)
|
||||
h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, encodedName))
|
||||
h.Set("Content-Type", "application/x-directory")
|
||||
return w.CreatePart(h)
|
||||
}
|
||||
|
||||
func (w *IpfsMultipartWriter) CreateIpfsFilePart(name string) (io.Writer, error) {
|
||||
h := make(textproto.MIMEHeader)
|
||||
encodedName := url.QueryEscape(name)
|
||||
h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, escapeQuotes(encodedName)))
|
||||
h.Set("Content-Type", "application/octet-stream")
|
||||
return w.CreatePart(h)
|
||||
}
|
||||
|
||||
func (w *IpfsMultipartWriter) CreateIpfsAbsFilePart(name, absPath string) (io.Writer, error) {
|
||||
h := make(textproto.MIMEHeader)
|
||||
encodedName := url.QueryEscape(name)
|
||||
h.Set("AbsPath", absPath)
|
||||
h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, escapeQuotes(encodedName)))
|
||||
h.Set("Content-Type", "application/octet-stream")
|
||||
return w.CreatePart(h)
|
||||
}
|
48
main.go
48
main.go
@ -5,14 +5,11 @@ import (
|
||||
"fmt"
|
||||
"io"
|
||||
"io/fs"
|
||||
"mime/multipart"
|
||||
"net/http"
|
||||
"net/textproto"
|
||||
"net/url"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"gitea.bruyant.xyz/alexandre/ipfs-node-pin/ipfsnodepin"
|
||||
"github.com/sethvargo/go-githubactions"
|
||||
)
|
||||
|
||||
@ -23,47 +20,6 @@ type AddResponse struct {
|
||||
Size string
|
||||
}
|
||||
|
||||
type IpfsMultipartWriter struct {
|
||||
multipart.Writer
|
||||
}
|
||||
|
||||
var quoteEscaper = strings.NewReplacer("\\", "\\\\", `"`, "\\\"")
|
||||
|
||||
func escapeQuotes(s string) string {
|
||||
return quoteEscaper.Replace(s)
|
||||
}
|
||||
|
||||
func NewIpfsMultipartWriter(w io.Writer) *IpfsMultipartWriter {
|
||||
return &IpfsMultipartWriter{
|
||||
Writer: *multipart.NewWriter(w),
|
||||
}
|
||||
}
|
||||
|
||||
func (w *IpfsMultipartWriter) CreateIpfsDirectoryPart(name string) (io.Writer, error) {
|
||||
h := make(textproto.MIMEHeader)
|
||||
encodedName := url.QueryEscape(name)
|
||||
h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, encodedName))
|
||||
h.Set("Content-Type", "application/x-directory")
|
||||
return w.CreatePart(h)
|
||||
}
|
||||
|
||||
func (w *IpfsMultipartWriter) CreateIpfsFilePart(name string) (io.Writer, error) {
|
||||
h := make(textproto.MIMEHeader)
|
||||
encodedName := url.QueryEscape(name)
|
||||
h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, escapeQuotes(encodedName)))
|
||||
h.Set("Content-Type", "application/octet-stream")
|
||||
return w.CreatePart(h)
|
||||
}
|
||||
|
||||
func (w *IpfsMultipartWriter) CreateIpfsAbsFilePart(name, absPath string) (io.Writer, error) {
|
||||
h := make(textproto.MIMEHeader)
|
||||
encodedName := url.QueryEscape(name)
|
||||
h.Set("AbsPath", absPath)
|
||||
h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, escapeQuotes(encodedName)))
|
||||
h.Set("Content-Type", "application/octet-stream")
|
||||
return w.CreatePart(h)
|
||||
}
|
||||
|
||||
func main() {
|
||||
githubactions.Debugf("Checking inputs")
|
||||
|
||||
@ -112,7 +68,7 @@ func main() {
|
||||
q.Add("stream-true", "false")
|
||||
req.URL.RawQuery = q.Encode()
|
||||
|
||||
mwriter := NewIpfsMultipartWriter(writer)
|
||||
mwriter := ipfsnodepin.NewIpfsMultipartWriter(writer)
|
||||
req.Header.Add("Content-Type", mwriter.FormDataContentType())
|
||||
|
||||
go func() {
|
||||
|
Loading…
Reference in New Issue
Block a user