feat(v2): api upload

Author: crispgm

v2/api_upload.go

@@ -0,0 +1,150 @@
+package lark
+
+// Import from Lark API Go demo
+// with adaption to go-lark frame
+
+import (
+	"bytes"
+	"context"
+	"fmt"
+	"image"
+	"image/jpeg"
+	"io"
+	"mime/multipart"
+	"net/http"
+	"os"
+)
+
+const (
+	uploadImageURL = "/open-apis/im/v1/images"
+	uploadFileURL  = "/open-apis/im/v1/files"
+)
+
+// UploadImageResponse .
+type UploadImageResponse struct {
+	BaseResponse
+	Data struct {
+		ImageKey string `json:"image_key"`
+	} `json:"data"`
+}
+
+// UploadFileRequest .
+type UploadFileRequest struct {
+	FileType string    `json:"-"`
+	FileName string    `json:"-"`
+	Duration int       `json:"-"`
+	Path     string    `json:"-"`
+	Reader   io.Reader `json:"-"`
+}
+
+// UploadFileResponse .
+type UploadFileResponse struct {
+	BaseResponse
+	Data struct {
+		FileKey string `json:"file_key"`
+	} `json:"data"`
+}
+
+// UploadImage uploads image file
+func (bot Bot) UploadImage(ctx context.Context, path string) (*UploadImageResponse, error) {
+	file, err := os.Open(path)
+	if err != nil {
+		return nil, err
+	}
+	defer file.Close()
+
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+	writer.WriteField("image_type", "message")
+	part, err := writer.CreateFormFile("image", path)
+	if err != nil {
+		return nil, err
+	}
+	_, err = io.Copy(part, file)
+	if err != nil {
+		return nil, err
+	}
+	err = writer.Close()
+	if err != nil {
+		return nil, err
+	}
+	var respData UploadImageResponse
+	header := make(http.Header)
+	header.Set("Content-Type", writer.FormDataContentType())
+	err = bot.PerformAPIRequest(ctx, "POST", "UploadImage", uploadImageURL, header, true, body, &respData)
+	if err != nil {
+		return nil, err
+	}
+	return &respData, err
+}
+
+// UploadImageObject uploads image object
+func (bot Bot) UploadImageObject(ctx context.Context, img image.Image) (*UploadImageResponse, error) {
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+	writer.WriteField("image_type", "message")
+	part, err := writer.CreateFormFile("image", "temp_image")
+	if err != nil {
+		return nil, err
+	}
+	err = jpeg.Encode(part, img, nil)
+	if err != nil {
+		return nil, err
+	}
+	err = writer.Close()
+	if err != nil {
+		return nil, err
+	}
+	var respData UploadImageResponse
+	header := make(http.Header)
+	header.Set("Content-Type", writer.FormDataContentType())
+	err = bot.PerformAPIRequest(ctx, "POST", "UploadImage", uploadImageURL, header, true, body, &respData)
+	if err != nil {
+		return nil, err
+	}
+	return &respData, err
+}
+
+// UploadFile uploads file to Lark server
+func (bot Bot) UploadFile(ctx context.Context, req UploadFileRequest) (*UploadFileResponse, error) {
+	var content io.Reader
+	if req.Reader == nil {
+		file, err := os.Open(req.Path)
+		if err != nil {
+			return nil, err
+		}
+		defer file.Close()
+		content = file
+	} else {
+		content = req.Reader
+		req.Path = req.FileName
+	}
+
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+	writer.WriteField("file_type", req.FileType)
+	writer.WriteField("file_name", req.FileName)
+	if req.FileType == "mp4" && req.Duration > 0 {
+		writer.WriteField("duration", fmt.Sprintf("%d", req.Duration))
+	}
+	part, err := writer.CreateFormFile("file", req.Path)
+	if err != nil {
+		return nil, err
+	}
+	_, err = io.Copy(part, content)
+	if err != nil {
+		return nil, err
+	}
+	err = writer.Close()
+	if err != nil {
+		return nil, err
+	}
+	var respData UploadFileResponse
+	header := make(http.Header)
+	header.Set("Content-Type", writer.FormDataContentType())
+	err = bot.PerformAPIRequest(ctx, "POST", "UploadFile", uploadFileURL, header, true, body, &respData)
+	if err != nil {
+		return nil, err
+	}
+	return &respData, err
+}

v2/api_upload_test.go

@@ -0,0 +1,64 @@
+package lark
+
+import (
+	"image/jpeg"
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestUploadImage(t *testing.T) {
+	resp, err := bot.UploadImage(t.Context(), "./fixtures/test.jpg")
+	if assert.NoError(t, err) {
+		assert.Zero(t, resp.Code)
+		t.Log(resp.Data.ImageKey)
+		assert.NotEmpty(t, resp.Data.ImageKey)
+	}
+}
+
+func TestUploadImageObject(t *testing.T) {
+	file, _ := os.Open("./fixtures/test.jpg")
+	img, _ := jpeg.Decode(file)
+
+	resp, err := bot.UploadImageObject(t.Context(), img)
+	if assert.NoError(t, err) {
+		assert.Zero(t, resp.Code)
+		t.Log(resp.Data.ImageKey)
+		assert.NotEmpty(t, resp.Data.ImageKey)
+	}
+}
+
+func TestUploadFile(t *testing.T) {
+	resp, err := bot.UploadFile(
+		t.Context(),
+		UploadFileRequest{
+			FileType: "pdf",
+			FileName: "hello.pdf",
+			Path:     "./fixtures/test.pdf",
+		})
+	if assert.NoError(t, err) {
+		assert.Zero(t, resp.Code)
+		t.Log(resp.Data.FileKey)
+		assert.NotEmpty(t, resp.Data.FileKey)
+	}
+}
+
+func TestUploadFile_Binary(t *testing.T) {
+	resp, err := bot.UploadFile(
+		t.Context(),
+		UploadFileRequest{
+			FileType: "stream",
+			FileName: "test-data.csv",
+			Reader: strings.NewReader(`Name,Age,Location
+Foo,25,Sleman
+Bar,23,Sidoarjo
+Baz,27,Bantul`),
+		})
+	if assert.NoError(t, err) {
+		assert.Zero(t, resp.Code)
+		t.Log(resp.Data.FileKey)
+		assert.NotEmpty(t, resp.Data.FileKey)
+	}
+}

v2/event_test.go

@@ -35,7 +35,7 @@ func TestPostEvent(t *testing.T) {
 	}, "POST", "/", message)
 	assert.Equal(t, "2.0", string(w.Body.Bytes()))
 
-	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
 		m, _ := json.Marshal(message)
 		w.Write([]byte(m))
 	}))