<?php

namespace Knuckles\Scribe\Extracting;

use Illuminate\Database\Eloquent\Model;
use Illuminate\Support\Arr;
use Knuckles\Scribe\Tools\ConsoleOutputUtils as c;
use Knuckles\Scribe\Tools\ErrorHandlingUtils as e;
use Knuckles\Scribe\Tools\Utils;

trait InstantiatesExampleModels
{
    /**
     * @param  string[]  $factoryStates
     * @param  string[]  $relations
     * @param  null|\ReflectionFunctionAbstract  $transformationMethod  A method which has the model as its first parameter. Useful if the `$type` is empty.
     * @return null|Model|object
     */
    protected function instantiateExampleModel(
        ?string $type = null,
        array $factoryStates = [],
        array $relations = [],
        ?\ReflectionFunctionAbstract $transformationMethod = null,
        array $withCount = [],
    ) {
        // If the API Resource uses an empty resource, there won't be an example model
        if ($type === null && $transformationMethod === null) {
            return null;
        }

        if ($type === null) {
            $parameter = Arr::first($transformationMethod->getParameters());
            $parameterType = $parameter->hasType() ? $parameter->getType() : null;
            if ($parameterType instanceof \ReflectionNamedType
                && ! $parameterType->isBuiltin() && class_exists($parameterType->getName())) {
                // Ladies and gentlemen, we have a type!
                $type = $parameterType->getName();
            }
        }
        if ($type === null) {
            throw new \Exception("Couldn't detect a transformer model from your doc block. Did you remember to specify a model using @transformerModel?");
        }

        $configuredStrategies = $this->config->get('examples.models_source', ['factoryCreate', 'factoryMake', 'databaseFirst']);

        $strategies = [
            'factoryCreate' => fn () => $this->getExampleModelFromFactoryCreate($type, $factoryStates, $relations, $withCount),
            'factoryCreateQuietly' => fn () => $this->getExampleModelFromFactoryCreate($type, $factoryStates, $relations, $withCount, true),
            'factoryMake' => fn () => $this->getExampleModelFromFactoryMake($type, $factoryStates, $relations),
            'databaseFirst' => fn () => $this->getExampleModelFromDatabaseFirst($type, $relations),
        ];

        foreach ($configuredStrategies as $strategyName) {
            try {
                $model = $strategies[$strategyName]();
                if ($model) {
                    return $model;
                }
            } catch (\Throwable $e) {
                c::warn("Couldn't get example model for {$type} via {$strategyName}.");
                e::dumpExceptionIfVerbose($e);
            }
        }

        return new $type;
    }

    /**
     * @param  class-string  $type
     * @param  string[]  $factoryStates
     * @param  string[]  $relations
     * @param  string[]  $withCount
     * @return null|Model
     */
    protected function getExampleModelFromFactoryCreate(string $type, array $factoryStates = [], array $relations = [], array $withCount = [], bool $quietly = false)
    {
        // Since $relations and $withCount refer to the same underlying relationships in the model,
        // combining them ensures that all required relationships are initialized when passed to the factory.
        $allRelations = array_unique(array_merge($relations, $withCount));

        $factory = Utils::getModelFactory($type, $factoryStates, $allRelations);

        $factory = $quietly ? Model::withoutEvents(fn () => $factory->create()) : $factory->create();

        return $factory->refresh()->load($relations)->loadCount($withCount);
    }

    /**
     * @param  class-string  $type
     * @param  string[]  $factoryStates
     * @return null|Model
     */
    protected function getExampleModelFromFactoryMake(string $type, array $factoryStates = [], array $relations = [])
    {
        $factory = Utils::getModelFactory($type, $factoryStates, $relations);

        return $factory->make();
    }

    /**
     * @param  class-string  $type
     * @param  string[]  $relations
     * @return null|Model
     */
    protected function getExampleModelFromDatabaseFirst(string $type, array $relations = [])
    {
        return $type::with($relations)->first();
    }
}