78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
|
package imaginary
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"mime/multipart"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Client struct {
|
||
|
|
baseURL string
|
||
|
|
http *http.Client
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(baseURL string) *Client {
|
||
|
|
return &Client{
|
||
|
|
baseURL: baseURL,
|
||
|
|
http: &http.Client{},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type ProcessOptions struct {
|
||
|
|
Width int
|
||
|
|
Height int
|
||
|
|
Quality int
|
||
|
|
Type string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) Process(src io.Reader, filename string, opts ProcessOptions) ([]byte, error) {
|
||
|
|
var body bytes.Buffer
|
||
|
|
writer := multipart.NewWriter(&body)
|
||
|
|
|
||
|
|
part, err := writer.CreateFormFile("file", filename)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if _, err := io.Copy(part, src); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
writer.Close()
|
||
|
|
|
||
|
|
if opts.Width == 0 {
|
||
|
|
opts.Width = 2000
|
||
|
|
}
|
||
|
|
if opts.Height == 0 {
|
||
|
|
opts.Height = 2000
|
||
|
|
}
|
||
|
|
if opts.Quality == 0 {
|
||
|
|
opts.Quality = 80
|
||
|
|
}
|
||
|
|
if opts.Type == "" {
|
||
|
|
opts.Type = "webp"
|
||
|
|
}
|
||
|
|
|
||
|
|
url := fmt.Sprintf("%s/resize?width=%d&height=%d&quality=%d&type=%s",
|
||
|
|
c.baseURL, opts.Width, opts.Height, opts.Quality, opts.Type)
|
||
|
|
|
||
|
|
req, err := http.NewRequest("POST", url, &body)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
req.Header.Set("Content-Type", writer.FormDataContentType())
|
||
|
|
|
||
|
|
resp, err := c.http.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
if resp.StatusCode != http.StatusOK {
|
||
|
|
errBody, _ := io.ReadAll(resp.Body)
|
||
|
|
return nil, fmt.Errorf("imaginary error %d: %s", resp.StatusCode, string(errBody))
|
||
|
|
}
|
||
|
|
|
||
|
|
return io.ReadAll(resp.Body)
|
||
|
|
}
|