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"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"mime/multipart"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/textproto"
|
|
||||||
"net/url"
|
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
|
||||||
|
|
||||||
|
"gitea.bruyant.xyz/alexandre/ipfs-node-pin/ipfsnodepin"
|
||||||
"github.com/sethvargo/go-githubactions"
|
"github.com/sethvargo/go-githubactions"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -23,47 +20,6 @@ type AddResponse struct {
|
|||||||
Size string
|
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() {
|
func main() {
|
||||||
githubactions.Debugf("Checking inputs")
|
githubactions.Debugf("Checking inputs")
|
||||||
|
|
||||||
@ -112,7 +68,7 @@ func main() {
|
|||||||
q.Add("stream-true", "false")
|
q.Add("stream-true", "false")
|
||||||
req.URL.RawQuery = q.Encode()
|
req.URL.RawQuery = q.Encode()
|
||||||
|
|
||||||
mwriter := NewIpfsMultipartWriter(writer)
|
mwriter := ipfsnodepin.NewIpfsMultipartWriter(writer)
|
||||||
req.Header.Add("Content-Type", mwriter.FormDataContentType())
|
req.Header.Add("Content-Type", mwriter.FormDataContentType())
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
|
Loading…
Reference in New Issue
Block a user