implement upload richmenu content api

Author: musou1500

lib/client.ts

@@ -1,7 +1,8 @@
-import { get, post, stream, delete as deleteRequest } from "./http";
+import { Readable } from "stream";
+import { get, post, stream, delete as deleteRequest, postBinary } from "./http";
 import * as Types from "./types";
 import * as URL from "./urls";
-import { toArray } from "./util";
+import { toArray, detectContentType } from "./util";
 
 export default class Client {
   public config: Types.ClientConfig;
@@ -136,7 +137,13 @@ export default class Client {
     return this.stream(URL.richMenuContent(richMenuId));
   }
 
-  // TODO: implement method for uploading richmenu image
+  public uploadRichMenuContent(
+    richMenuId: string,
+    data: Buffer | Readable,
+    contentType?: string,
+  ): Promise<any> {
+    return this.postBinary(URL.richMenuContent(richMenuId), data, contentType);
+  }
 
   public getRichMenuList(): Promise<any> {
     return this.get(URL.richMenuList());
@@ -158,6 +165,14 @@ export default class Client {
     return post(url, this.authHeader(), body);
   }
 
+  private postBinary(
+    url: string,
+    data: Buffer | Readable,
+    contentType?: string,
+  ) {
+    return postBinary(url, this.authHeader(), data, contentType);
+  }
+
   private stream(url: string): Promise<NodeJS.ReadableStream> {
     return stream(url, this.authHeader());
   }

lib/http.ts

@@ -1,4 +1,6 @@
 import axios, { AxiosError } from "axios";
+import { detectContentType } from "./util";
+import { Readable } from "stream";
 import {
   HTTPError,
   JSONParseError,
@@ -65,6 +67,24 @@ export function post(url: string, headers: any, data?: any): Promise<any> {
     .catch(wrapError);
 }
 
+export function postBinary(
+  url: string,
+  headers: any,
+  data: Buffer | Readable,
+  contentType?: string,
+): Promise<Readable> {
+  return new Promise(resolve => {
+    return resolve(contentType ? contentType : detectContentType(data));
+  }).then((contentType: string) => {
+    headers["Content-Type"] = contentType;
+    headers["User-Agent"] = userAgent;
+    const responseType = "stream";
+    return axios
+      .post(url, data, { headers, responseType })
+      .then(res => res.data);
+  });
+}
+
 function del(url: string, headers: any): Promise<any> {
   headers["User-Agent"] = userAgent;
 

test/client.spec.ts

@@ -1,3 +1,5 @@
+import { readFileSync } from "fs";
+import { join } from "path";
 import { deepEqual, equal } from "assert";
 import Client from "../lib/client";
 import * as Types from "../lib/types";
@@ -219,6 +221,32 @@ describe("client", () => {
       });
   });
 
+  it("uploadRichMenuContent", () => {
+    const filepath = join(__dirname, "/helpers/LINE_Icon.png");
+    const buffer = readFileSync(filepath);
+    return client
+      .uploadRichMenuContent("test_rich_menu_id", buffer)
+      .then(s => getStreamData(s))
+      .then((data: string) => {
+        const res = JSON.parse(data);
+        equal(res.headers.authorization, "Bearer test_channel_access_token");
+        equal(res.path, "/richmenu/test_rich_menu_id/content");
+        equal(res.method, "POST");
+      });
+  });
+
+  it("getRichMenuContent", () => {
+    return client
+      .getRichMenuContent("test_rich_menu_id")
+      .then((s: NodeJS.ReadableStream) => getStreamData(s))
+      .then((data: string) => {
+        const res = JSON.parse(data);
+        equal(res.headers.authorization, "Bearer test_channel_access_token");
+        equal(res.path, "/richmenu/test_rich_menu_id/content");
+        equal(res.method, "GET");
+      });
+  });
+
   it("getRichMenuList", () => {
     return client.getRichMenuList().then((res: any) => {
       equal(res.headers.authorization, "Bearer test_channel_access_token");

test/http.spec.ts

@@ -1,8 +1,16 @@
 import { deepEqual, equal, ok } from "assert";
 import { HTTPError, JSONParseError, RequestError } from "../lib/exceptions";
-import { get, post, stream, delete as deleteRequest } from "../lib/http";
+import {
+  get,
+  post,
+  stream,
+  delete as deleteRequest,
+  postBinary,
+} from "../lib/http";
 import { getStreamData } from "./helpers/stream";
 import { close, listen } from "./helpers/test-server";
+import { readFileSync } from "fs";
+import { join } from "path";
 
 const pkg = require("../package.json");
 
@@ -88,6 +96,25 @@ describe("http", () => {
     });
   });
 
+  it("postBinary", () => {
+    const testHeaders = {
+      "test-header-key": "Test-Header-Value",
+    };
+
+    const filepath = join(__dirname, "/helpers/LINE_Icon.png");
+    const buffer = readFileSync(filepath);
+    return postBinary(`${TEST_URL}/post`, testHeaders, buffer)
+      .then(s => getStreamData(s))
+      .then((data: string) => {
+        const res = JSON.parse(data);
+        equal(res.method, "POST");
+        equal(res.path, "/post");
+        equal(res.headers["test-header-key"], testHeaders["test-header-key"]);
+        equal(res.headers["user-agent"], `${pkg.name}/${pkg.version}`);
+        equal(res.headers["content-type"], "image/png");
+      });
+  });
+
   it("fail to parse json", () => {
     return get(`${TEST_URL}/text`, {})
       .then(() => ok(false))