upload.go

package smmssdk

import (
	"bytes"
	"context"
	"errors"
	"fmt"
	"io"
	"log/slog"
	"mime/multipart"
	"net/http"
	"os"
	"time"
)

type UploadRequest struct {
	Smfile string `json:"smfile"`
	Format string `json:"format"` // Return Type: json or xml, the default value is json

	SmfileStream io.Reader `json:"-"`
}

func (r *UploadRequest) Request(ctx context.Context) (*http.Request, error) {
	if r == nil {
		return nil, errors.New("invalid request is nil")
	}
	if r.Smfile == "" && r.SmfileStream == nil {
		return nil, errors.New("invalid request source is empty")
	}

	var fileReader io.Reader
	var filename string
	if r.SmfileStream != nil {
		fileReader = r.SmfileStream
		filename = fmt.Sprintf("%d.jpg", time.Now().UnixNano())
	} else {
		slog.InfoContext(ctx, "open file...", "smfile", r.Smfile)
		file, err := os.Open(r.Smfile)
		if err != nil {
			return nil, err
		}
		defer file.Close()

		fileReader = file
		filename = file.Name()
	}

	body := &bytes.Buffer{}
	writer := multipart.NewWriter(body)
	part, err := writer.CreateFormFile("smfile", filename)
	if err != nil {
		return nil, err
	}
	_, err = io.Copy(part, fileReader)
	if err != nil {
		return nil, err
	}
	err = writer.Close()
	if err != nil {
		return nil, err
	}

	method := http.MethodPost
	url := fmt.Sprintf("%s/upload", baseURL)
	req, err := http.NewRequestWithContext(ctx, method, url, body)
	if err != nil {
		return nil, err
	}
	req.Header.Set("Content-Type", writer.FormDataContentType())
	return req, nil
}

func NewUploadRequest() *UploadRequest {
	return &UploadRequest{}
}

type UploadResponse struct {
	baseResponse
	Data *struct {
		Width     int    `json:"width"`
		Height    int    `json:"height"`
		Filename  string `json:"filename"`
		Storename string `json:"storename"`
		Size      int64  `json:"size"`
		Path      string `json:"path"`
		Hash      string `json:"hash"`
		Url       string `json:"url"`
		Delete    string `json:"delete"`
		Page      string `json:"page"`
	} `json:"data"`
	CurrentPage int64 `json:"CurrentPage"`
	TotalPages  int64 `json:"TotalPages"`
	PerPage     int64 `json:"PerPage"`
	Count       int64 `json:"Count"`
}

func (c *Client) Upload(ctx context.Context, req *UploadRequest) (*UploadResponse, error) {
	return c.upload(ctx, req)
}

func (c *Client) upload(ctx context.Context, req *UploadRequest) (*UploadResponse, error) {
	var rsp UploadResponse
	err := c.Do(ctx, req, &rsp)
	if err != nil {
		return nil, err
	}

	return &rsp, nil
}