From aac4769bcd18ee2ab19fcd7a5c96b3e9f67a8a26 Mon Sep 17 00:00:00 2001
From: Volodymyr Kupriienko <vldmr.kuprienko@gmail.com>
Date: Sun, 28 Feb 2021 16:55:49 +0200
Subject: [PATCH] Add support of i16 data type

---
 src/renderer/thrift/thriftrw.d.ts         | 2 +-
 src/renderer/utils/defaultRequest.ts      | 1 +
 src/renderer/utils/getMethodJsonSchema.ts | 1 +
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/renderer/thrift/thriftrw.d.ts b/src/renderer/thrift/thriftrw.d.ts
index e7654a9..f089669 100644
--- a/src/renderer/thrift/thriftrw.d.ts
+++ b/src/renderer/thrift/thriftrw.d.ts
@@ -47,7 +47,7 @@ declare module 'thriftrw' {
 
     export interface BaseType {
         type: 'BaseType';
-        baseType: 'string' | 'i32' | 'i64' | 'double' | 'float' | 'bool' | 'void' | 'binary';
+        baseType: 'string' | 'i16' | 'i32' | 'i64' | 'double' | 'float' | 'bool' | 'void' | 'binary';
         annotations?: TypeAnnotation[] | {};
     }
 
diff --git a/src/renderer/utils/defaultRequest.ts b/src/renderer/utils/defaultRequest.ts
index 45b70d8..cbd4c2b 100644
--- a/src/renderer/utils/defaultRequest.ts
+++ b/src/renderer/utils/defaultRequest.ts
@@ -40,6 +40,7 @@ function getDefaultValue(fieldAst: FieldAst, thriftDefinitions: Record<string, P
             switch (fieldAst.valueType.baseType) {
                 case 'string':
                     return '';
+                case 'i16':
                 case 'i32':
                 case 'i64':
                 case 'double':
diff --git a/src/renderer/utils/getMethodJsonSchema.ts b/src/renderer/utils/getMethodJsonSchema.ts
index b265112..a78c84f 100644
--- a/src/renderer/utils/getMethodJsonSchema.ts
+++ b/src/renderer/utils/getMethodJsonSchema.ts
@@ -44,6 +44,7 @@ function getFieldSchema(fieldAst: FieldAst, thriftDefinitions: Record<string, Pr
             switch (fieldAst.valueType.baseType) {
                 case 'string':
                     return { type: 'string' };
+                case 'i16':
                 case 'i32':
                 case 'i64':
                 case 'double':