@stackbit/sdk 0.3.3-alpha.1 → 0.3.3-alpha.2

package/src/utils/model-utils.ts

@@ -7,6 +7,12 @@ import {
     DataModel,
     PageModel,
     ConfigModel,
+    ImageModel,
+    YamlModel,
+    YamlObjectModel,
+    YamlPageModel,
+    YamlDataModel,
+    YamlDataModelList,
     Field,
     FieldEnum,
     FieldList,
@@ -19,7 +25,8 @@ import {
     FieldObject,
     FieldObjectProps,
     FieldReference,
-    FieldReferenceProps
+    FieldReferenceProps,
+    DataModelList
 } from '../config/config-types';
 
 export function getModelByName(models: Model[], modelName: string): Model | undefined {
@@ -30,23 +37,27 @@ export function isConfigModel(model: Model): model is ConfigModel {
     return model.type === 'config';
 }
 
-export function isDataModel(model: Model): model is DataModel {
+export function isDataModel(model: Model | YamlModel): model is DataModel | YamlDataModel {
     return model.type === 'data';
 }
 
-export function isListDataModel(model: Model): model is DataModel & { isList: true } {
+export function isListDataModel(model: Model | YamlModel): model is DataModelList | YamlDataModelList {
     return isDataModel(model) && Boolean(model.isList);
 }
 
-export function isPageModel(model: Model): model is PageModel {
+export function isPageModel(model: Model | YamlModel): model is PageModel | YamlPageModel {
     return model.type === 'page';
 }
 
-export function isObjectModel(model: Model): model is ObjectModel {
+export function isObjectModel(model: Model | YamlModel): model is ObjectModel | YamlObjectModel {
     return model.type === 'object';
 }
 
-export function isSingleInstanceModel(model: Model): boolean {
+export function isImageModel(model: Model): model is ImageModel {
+    return model.type === 'image';
+}
+
+export function isSingleInstanceModel(model: Model | YamlModel): boolean {
     if (model.type === 'config') {
         return true;
     } else if (model.type === 'data') {
@@ -198,8 +209,8 @@ export function resolveLabelFieldForModel(modelOrField: Model | FieldObjectProps
     return labelField || null;
 }
 
-export function getModelFieldForModelKeyPath(model: Model, modelKeyPath: string[]) {
-    function _getField(field: Field, modelKeyPath: string[]): Model | Field | FieldListItems | FieldObjectProps | null {
+export function getModelFieldForModelKeyPath(model: Model | YamlModel, modelKeyPath: string[]) {
+    function _getField(field: Field, modelKeyPath: string[]): Model | YamlModel | Field | FieldListItems | FieldObjectProps | null {
         if (modelKeyPath.length === 0) {
             return field;
         } else if (isObjectField(field)) {
@@ -211,7 +222,10 @@ export function getModelFieldForModelKeyPath(model: Model, modelKeyPath: string[
         }
     }
 
-    function _getObjectFields<T extends FieldObjectProps | Model>(field: T, modelKeyPath: string[]): Model | Field | FieldListItems | FieldObjectProps | null {
+    function _getObjectFields<T extends FieldObjectProps | Model | YamlModel>(
+        field: T,
+        modelKeyPath: string[]
+    ): Model | YamlModel | Field | FieldListItems | FieldObjectProps | null {
         if (modelKeyPath.length === 0) {
             return field;
         }
@@ -230,10 +244,10 @@ export function getModelFieldForModelKeyPath(model: Model, modelKeyPath: string[
         return _getField(childField, modelKeyPath);
     }
 
-    function _getListItems<T extends FieldList | (DataModel & { isList: true })>(
+    function _getListItems<T extends FieldList | DataModelList | YamlDataModelList>(
         field: T,
         modelKeyPath: string[]
-    ): Model | Field | FieldListItems | FieldObjectProps | null {
+    ): Model | YamlModel | Field | FieldListItems | FieldObjectProps | null {
         if (modelKeyPath.length === 0) {
             return field;
         }