updated

Author: yisu

api_encryption.go

@@ -0,0 +1,34 @@
+package godingtalk
+
+//DataMessage 服务端加密、解密消息
+type DataMessage struct {
+	OAPIResponse
+	Data string
+}
+
+
+//Encrypt is 服务端加密 
+func (c *DingTalkClient) Encrypt(str string) (string, error) {
+	var data DataMessage
+	request := map[string]interface{}{
+		"data":  str,
+	}
+	err := c.httpRPC("encryption/encrypt", nil, request, &data)
+    if err!=nil {
+        return "", err
+    }
+	return data.Data, nil
+}
+
+//Decrypt is 服务端解密
+func (c *DingTalkClient) Decrypt(str string) (string, error) {
+	var data DataMessage
+	request := map[string]interface{}{
+		"data":  str,
+	}
+	err := c.httpRPC("encryption/decrypt", nil, request, &data)
+    if err!=nil {
+        return "", err
+    }
+	return data.Data, nil    
+}

api_encryption_test.go

@@ -0,0 +1,14 @@
+package godingtalk
+
+import (
+	"testing"
+)
+
+func TestEncryption(t *testing.T) {
+	str, err := c.Encrypt("Hello")
+    if err!=nil {
+        t.Error(err)
+    } else {
+        t.Log(str)
+    }
+}

api_file.go

@@ -0,0 +1,33 @@
+package godingtalk
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"net/url"
+)
+
+//FileResponse is
+type FileResponse struct {
+	OAPIResponse
+	Code     int
+	Msg      string
+	UploadID string `json:"uploadid"`
+	Writer   io.Writer
+}
+
+func (f *FileResponse) getWriter() io.Writer {
+	return f.Writer
+}
+
+//CreateFile is to create a new file in Ding Space
+func (c *DingTalkClient) CreateFile(size int64) (file FileResponse, err error) {
+	buf := bytes.Buffer{}
+	file = FileResponse{
+		Writer: &buf,
+	}
+	params := url.Values{}
+	params.Add("size", fmt.Sprintf("%d", size))
+	err = c.httpRPC("file/upload/create", params, nil, &file)
+	return file, err
+}

api_file_test.go

@@ -0,0 +1,10 @@
+package godingtalk
+
+import (
+	"testing"
+)
+
+func TestCreateFile(t *testing.T) {
+	file, err := c.CreateFile(1024)
+	t.Log(file, err)
+}

transport.go

@@ -6,6 +6,7 @@ import (
 	"errors"
 	"io"
 	"io/ioutil"
+	"log"
 	"mime/multipart"
 	"net/http"
 	"net/url"
@@ -32,7 +33,7 @@ func (c *DingTalkClient) httpRPC(path string, params url.Values, requestData int
 		if params == nil {
 			params = url.Values{}
 		}
-		if (params.Get("access_token")=="") {
+		if params.Get("access_token") == "" {
 			params.Set("access_token", c.AccessToken)
 		}
 	}
@@ -67,24 +68,30 @@ func (c *DingTalkClient) httpRequest(path string, params url.Values, requestData
 			request.Header.Set("Content-Type", w.FormDataContentType())
 		default:
 			d, _ := json.Marshal(requestData)
-			request, _ = http.NewRequest("POST", url2, bytes.NewReader(d))
 			// log.Printf("url: %s request: %s", url2, string(d))
+			request, _ = http.NewRequest("POST", url2, bytes.NewReader(d))
 			request.Header.Set("Content-Type", typeJSON)
 		}
 	} else {
+		// log.Printf("url: %s", url2)
 		request, _ = http.NewRequest("GET", url2, nil)
 	}
 	resp, err := client.Do(request)
 	if err != nil {
 		return err
 	}
+
+	if resp.StatusCode != 200 {
+		return errors.New("Server error: " + resp.Status)
+	}
+
 	defer resp.Body.Close()
 	contentType := resp.Header.Get("Content-Type")
+	log.Printf("url: %s response content type: %s", url2, contentType)
 	pos := len(typeJSON)
 	if len(contentType) >= pos && contentType[0:pos] == typeJSON {
 		content, err := ioutil.ReadAll(resp.Body)
 		if err == nil {
-			// log.Printf("url: %s response: %s", url2, string(content))
 			json.Unmarshal(content, responseData)
 			return responseData.checkError()
 		}