node build fixed

seanime-2.9.10/internal/torrent_clients/qbittorrent/util/util.go

@@ -0,0 +1,228 @@
+package qbittorrent_util
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/goccy/go-json"
+	"io"
+	"mime/multipart"
+	"net/http"
+	"net/textproto"
+	"seanime/internal/torrent_clients/qbittorrent/model"
+	"strings"
+)
+
+func GetInto(client *http.Client, target interface{}, url string, body interface{}) (err error) {
+	var buffer bytes.Buffer
+	if body != nil {
+		if err := json.NewEncoder(&buffer).Encode(body); err != nil {
+			return err
+		}
+	}
+	r, err := http.NewRequest("GET", url, &buffer)
+	if err != nil {
+		return err
+	}
+	resp, err := client.Do(r)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err2 := resp.Body.Close(); err2 != nil {
+			err = err2
+		}
+	}()
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("invalid response status %s", resp.Status)
+	}
+	buf, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return err
+	}
+	if err := json.NewDecoder(bytes.NewReader(buf)).Decode(target); err != nil {
+		if err2 := json.NewDecoder(strings.NewReader(`"` + string(buf) + `"`)).Decode(target); err2 != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func Post(client *http.Client, url string, body interface{}) (err error) {
+	var buffer bytes.Buffer
+	if err := json.NewEncoder(&buffer).Encode(body); err != nil {
+		return err
+	}
+	r, err := http.NewRequest("POST", url, &buffer)
+	if err != nil {
+		return err
+	}
+	resp, err := client.Do(r)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err2 := resp.Body.Close(); err2 != nil {
+			err = err2
+		}
+	}()
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("invalid status %s", resp.Status)
+	}
+	return nil
+}
+
+func createFormFileWithHeader(writer *multipart.Writer, name, filename string, headers map[string]string) (io.Writer, error) {
+	header := textproto.MIMEHeader{}
+	header.Add("Content-Disposition", fmt.Sprintf(`form-data; name="%s"; filename="%s"`, name, filename))
+	for key, value := range headers {
+		header.Add(key, value)
+	}
+	return writer.CreatePart(header)
+}
+
+func PostMultipartLinks(client *http.Client, url string, options *qbittorrent_model.AddTorrentsOptions, links []string) (err error) {
+	var o map[string]interface{}
+	if options != nil {
+		b, err := json.Marshal(options)
+		if err != nil {
+			return err
+		}
+		if err := json.Unmarshal(b, &o); err != nil {
+			return err
+		}
+	}
+	buf := bytes.Buffer{}
+	form := multipart.NewWriter(&buf)
+	if err := form.WriteField("urls", strings.Join(links, "\n")); err != nil {
+		return err
+	}
+	for key, value := range o {
+		if err := form.WriteField(key, fmt.Sprintf("%v", value)); err != nil {
+			return err
+		}
+	}
+	if err := form.Close(); err != nil {
+		return err
+	}
+	req, err := http.NewRequest("POST", url, &buf)
+	if err != nil {
+		return err
+	}
+	req.Header.Add("content-type", "multipart/form-data; boundary="+form.Boundary())
+	resp, err := client.Do(req)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err2 := resp.Body.Close(); err2 != nil {
+			err = err2
+		}
+	}()
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("invalid status %s", resp.Status)
+	}
+	return nil
+}
+
+func PostMultipartFiles(client *http.Client, url string, options *qbittorrent_model.AddTorrentsOptions, files map[string][]byte) (err error) {
+	var o map[string]interface{}
+	if options != nil {
+		b, err := json.Marshal(options)
+		if err != nil {
+			return err
+		}
+		if err := json.Unmarshal(b, &o); err != nil {
+			return err
+		}
+	}
+	buf := bytes.Buffer{}
+	form := multipart.NewWriter(&buf)
+	for filename, file := range files {
+		writer, err := createFormFileWithHeader(form, "torrents", filename, map[string]string{
+			"content-type": "application/x-bittorrent",
+		})
+		if err != nil {
+			return err
+		}
+		if _, err := writer.Write(file); err != nil {
+			return err
+		}
+	}
+	for key, value := range o {
+		if err := form.WriteField(key, fmt.Sprintf("%v", value)); err != nil {
+			return err
+		}
+	}
+	if err := form.Close(); err != nil {
+		return err
+	}
+	req, err := http.NewRequest("POST", url, &buf)
+	if err != nil {
+		return err
+	}
+	req.Header.Add("content-type", "multipart/form-data; boundary="+form.Boundary())
+	resp, err := client.Do(req)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err2 := resp.Body.Close(); err2 != nil {
+			err = err2
+		}
+	}()
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("invalid status %s", resp.Status)
+	}
+	return nil
+}
+
+func PostWithContentType(client *http.Client, url string, body io.Reader, contentType string) (err error) {
+	r, err := http.NewRequest("POST", url, body)
+	if err != nil {
+		return err
+	}
+	r.Header.Add("content-type", contentType)
+	resp, err := client.Do(r)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err2 := resp.Body.Close(); err2 != nil {
+			err = err2
+		}
+	}()
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("invalid status %s", resp.Status)
+	}
+	return nil
+}
+
+func GetIntoWithContentType(client *http.Client, target interface{}, url string, body io.Reader, contentType string) (err error) {
+	r, err := http.NewRequest("GET", url, body)
+	if err != nil {
+		return err
+	}
+	r.Header.Add("content-type", contentType)
+	resp, err := client.Do(r)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err2 := resp.Body.Close(); err2 != nil {
+			err = err2
+		}
+	}()
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("invalid response status %s", resp.Status)
+	}
+	buf, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return err
+	}
+	if err := json.NewDecoder(bytes.NewReader(buf)).Decode(target); err != nil {
+		if err2 := json.NewDecoder(strings.NewReader(`"` + string(buf) + `"`)).Decode(target); err2 != nil {
+			return err
+		}
+	}
+	return nil
+}