import type { ModelType } from '../ModelType';
import type { GenericModelSchema } from '../ModelSchema';
import type { ModelRelationshipField, ModelRelationshipTypes, RelationTypeFunctionOmitMapping } from '../ModelRelationshipField';
import type { BaseModelField } from '../ModelField';
import type { CustomType, CustomTypeParamShape } from '../CustomType';
import type { EnumType } from '../EnumType';
import type { RefType, RefTypeParamShape } from '../RefType';
import type { CustomOperation, CustomOperationParamShape } from '../CustomOperation';
export type ResolveSchema<Schema> = FieldTypes<ModelTypes<SchemaTypes<Schema>>>;
export type SchemaTypes<T> = T extends GenericModelSchema<any> ? T['data']['types'] : never;
/**
 * Resolves model types
 *
 * Removes CustomTypes and Enums from resolved schema.
 * They are extracted separately in ExtractNonModelTypes.ts and
 * added to ModelMeta in ClientSchema.ts
 */
export type ModelTypes<Schema> = {
    [Model in keyof Schema as Schema[Model] extends EnumType | CustomType<CustomTypeParamShape> | CustomOperation<CustomOperationParamShape, any> ? never : Model]: Schema[Model] extends ModelType<infer R, any> ? R['fields'] : never;
};
/**
 * Gets the collection of all ModelTypes and CustomTypes which are explicitly
 * defined in the schema.
 */
export type ModelAndCustomTypes<Schema> = {
    [Model in keyof Schema as Schema[Model] extends EnumType | CustomOperation<CustomOperationParamShape, any> ? never : Model]: Schema[Model] extends ModelType<any, any> | CustomType<CustomTypeParamShape> ? Schema[Model] : never;
};
/**
 * Resolves field types
 *
 * Non-model types are replaced with Refs. Refs remain and are resolved in ResolveFieldProperties.ts
 */
export type FieldTypes<T> = {
    [ModelProp in keyof T]: {
        [FieldProp in keyof T[ModelProp]]: T[ModelProp][FieldProp] extends BaseModelField<infer R> ? R : T[ModelProp][FieldProp] extends RefType<infer R extends RefTypeParamShape, any, any> ? R['valueRequired'] extends true ? T[ModelProp][FieldProp] : T[ModelProp][FieldProp] | null : T[ModelProp][FieldProp] extends EnumType | CustomType<CustomTypeParamShape> ? RefType<{
            link: `${Capitalize<ModelProp & string>}${Capitalize<FieldProp & string>}`;
            type: 'ref';
            valueRequired: false;
            array: false;
            arrayRequired: false;
            authorization: [];
        }> | null : T[ModelProp][FieldProp] extends ModelRelationshipField<infer R, string, RelationTypeFunctionOmitMapping<ModelRelationshipTypes>, any> ? R : never;
    };
};
/**
 * Resolves field types for a CustomType.
 *
 * This utility type is needed in addition to the `FieldTypes` utility type as
 * without checking `ModelRelationshipField` can improve ~5% on resolving performance.
 *
 * Non-model types are replaced with Refs. Refs remain and are resolved in ResolveFieldProperties.ts
 */
export type FieldTypesOfCustomType<T> = {
    [CustomTypeName in keyof T]: {
        [FieldProp in keyof T[CustomTypeName]]: T[CustomTypeName][FieldProp] extends BaseModelField<infer R> ? R : T[CustomTypeName][FieldProp] extends RefType<infer R extends RefTypeParamShape, any, any> ? R['valueRequired'] extends true ? T[CustomTypeName][FieldProp] : T[CustomTypeName][FieldProp] | null : T[CustomTypeName][FieldProp] extends EnumType | CustomType<CustomTypeParamShape> ? RefType<{
            link: `${Capitalize<CustomTypeName & string>}${Capitalize<FieldProp & string>}`;
            type: 'ref';
            valueRequired: false;
            array: false;
            arrayRequired: false;
            authorization: [];
        }> | null : never;
    };
};