v1.8.1

zeplin · Dec 21, 2021 · e549074 · e549074
1 parent a82d8a9
commit e549074
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/models/code.ts b/src/models/code.ts
@@ -12,19 +12,24 @@
  */
 
 
+import {
+    LanguageEnum,
+    transformLanguageEnumToJSON,
+    transformJSONToLanguageEnum
+} from './language-enum';
 
 
 export const transformCodeToJSON = function (value: Code): any {
     return {
         snippet: value.snippet,
-        language: value.language
+        language: value.language && transformLanguageEnumToJSON(value.language)
     }
 }
 
 export const transformJSONToCode = function (value: any): Code {
     return {
         snippet: value.snippet,
-        language: value.language
+        language: value.language && transformJSONToLanguageEnum(value.language)
     }
 }
 
@@ -41,11 +46,11 @@ export interface Code {
      */
     snippet: string;
     /**
-     * Language of the code snippet
-     * @type {string}
+     * 
+     * @type {LanguageEnum}
      * @memberof Code
      */
-    language?: string;
+    language?: LanguageEnum;
 }