feat(rest): allow setting the "Accept" type

OpenNMS · Jun 2, 2017 · 5bd300c · 5bd300c
1 parent 0d6d39f
commit 5bd300c
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/src/api/OnmsHTTPOptions.ts b/src/api/OnmsHTTPOptions.ts
@@ -11,6 +11,9 @@ export class OnmsHTTPOptions {
   /** how long to wait for ReST calls to time out */
   public timeout = 10000;
 
+  /** the type of response to accept */
+  public accept = 'application/json';
+
   /**
    * Construct a new OnmsHTTPOptions object.
    * @constructor

diff --git a/src/rest/AxiosHTTP.ts b/src/rest/AxiosHTTP.ts
@@ -67,6 +67,14 @@ export class AxiosHTTP extends AbstractHTTP implements IOnmsHTTP {
       if (options.timeout) {
         ret.timeout = options.timeout;
       }
+
+      if (options.accept === 'application/json') {
+        ret.responseType = 'json';
+      } else if (options.accept === 'text/plain') {
+        ret.responseType = 'text/plain';
+      } else {
+        throw new OnmsError('Unhandled response type: ' + options.accept);
+      }
     }
 
     return {};

diff --git a/src/rest/SuperAgentHTTP.ts b/src/rest/SuperAgentHTTP.ts
@@ -51,6 +51,7 @@ export class SuperAgentHTTP extends AbstractHTTP implements IOnmsHTTP {
     return request[method](url)
       .withCredentials()
       .timeout(options.timeout || super.timeout || super.options.timeout)
+      .set('Accept', options.accept)
       .auth(options.auth.username || super.server.auth.username || super.options.auth.username,
         options.auth.password || super.server.auth.password || super.options.auth.password);
   }