// 1
import { File } from '@asyncapi/generator-react-sdk';
// 2
import { PythonGenerator, PYTHON_PYDANTIC_PRESET } from '@asyncapi/modelina';

/**
 * @typedef RenderArgument
 * @type {object}
 * @property {AsyncAPIDocument} asyncapi document object received from the generator.
 */

function generateModels(model) {
  const properties = model.model?.properties ?? {};
  const ownIncludes = Object.values(properties)
      .filter(p => p.property.constructor.name === 'ConstrainedReferenceModel')
      .map(p => `from ${p.property.ref.name} import ${p.property.ref.name}`).join('\n')
  const imports = model.dependencies
      .join('\n');

  return (
      <File name={`${model.modelName}.py`}>
        {imports}
        {'\n\n'}
        {ownIncludes}
        {'\n\n'}
        {model.result}
      </File>
  );
}

/**
 * Render all schema models
 * @param {RenderArgument} param0
 * @returns
 */

export default async function schemaRender({ asyncapi }) {
  const pythonGenerator = new PythonGenerator({
    presets: [PYTHON_PYDANTIC_PRESET],
    processorOptions: {}
  });
  const models = await pythonGenerator.generate(asyncapi);
  return models.map(model => generateModels(model))
}