@prefecthq/prefect-ui-library/src/maps/schema.ts

This library is the Vue and Typescript component library for [Prefect 2](https://github.com/PrefectHQ/prefect) and [Prefect Cloud 2](https://www.prefect.io/cloud/). _The components and utilities in this project are not meant to be used independently_.

import { SchemaDefinitionsResponse, SchemaPropertiesResponse, SchemaPropertyResponse, SchemaResponse } from '@/models/api/SchemaResponse'
import { MapFunction } from '@/services/Mapper'
import { resolveSchema } from '@/services/schemas/resolvers/schemas'
import { Schema, SchemaDefinitions, schemaHas, SchemaProperties, SchemaProperty } from '@/types/schemas'
import { mapValues } from '@/utilities'

export const mapSchemaResponseToSchema: MapFunction<SchemaResponse, Schema> = function(source) {
  const { definitions, properties, $ref, secret_fields, block_type_slug, ...rest } = source

  const mapped: Schema = {
    secretFields: secret_fields,
    blockTypeSlug: block_type_slug,
    properties: this.map('SchemaPropertiesResponse', properties, 'SchemaProperties'),
    definitions: this.map('SchemaDefinitionsResponse', definitions, 'SchemaDefinitions'),
    ...rest,
    // todo: this map isn't working (ts error). Not using this anywhere so commenting it out for now
    // blockSchemaReferences: this.map('BlockSchemaReferencesResponse', block_schema_references, 'BlockSchemaReferences'),
  }

  if ($ref) {
    mapped.$ref = $ref
  }

  return resolveSchema(mapped)
}

export const mapSchemaDefinitionsResponseToSchemaDefinitions: MapFunction<SchemaDefinitionsResponse, SchemaDefinitions> = function(source) {
  return mapValues(source, (key, value) => this.map('SchemaResponse', value, 'Schema'))
}

export const mapSchemaPropertiesResponseToSchemaProperties: MapFunction<SchemaPropertiesResponse, SchemaProperties> = function(source) {
  return mapValues(source, (key, value) => this.map('SchemaPropertyResponse', value, 'SchemaProperty'))
}

export const mapSchemaPropertyResponseToSchemaProperty: MapFunction<SchemaPropertyResponse, SchemaProperty> = function(source) {
  const { properties, $ref, block_type_slug, ...rest } = source

  const mapped: SchemaProperty = {
    blockTypeSlug: block_type_slug,
    ...rest,
  }

  // its important that if mapped.properties doesn't exist at all that we don't add it as undefined
  if (schemaHas(source, 'properties')) {
    mapped.properties = this.map('SchemaPropertiesResponse', properties, 'SchemaProperties')
  }

  if ($ref) {
    mapped.$ref = $ref
  }

  return mapped
}