/*
 * This file belongs to Hoist, an application development toolkit
 * developed by Extremely Heavy Industries (www.xh.io | info@xh.io)
 *
 * Copyright © 2026 Extremely Heavy Industries Inc.
 */
import {elementFactory, ModelSelector, HoistModel, ModelPublishMode} from './..';
import {forOwn} from 'lodash';
import {createContext} from 'react';

/**
 * Support for making models available to components via React context.
 *
 * Not created directly by applications. Components specify how/if they publish/source their
 * models from context via the `model` config option passed into the `hoistCmp()` factory.
 * Hoist will in turn create and manage instances of this class to power those links.
 *
 * @internal
 */
export class ModelLookup {
    model: HoistModel;
    parent: ModelLookup;
    publishMode: ModelPublishMode;

    constructor(model: HoistModel, parent: ModelLookup, publishMode: ModelPublishMode) {
        this.model = model;
        this.parent = parent;
        this.publishMode = publishMode;
    }

    /**
     * Lookup a model in the context hierarchy
     * @returns model, or null if no matching model found.
     */
    lookupModel(selector: ModelSelector): HoistModel {
        const {model, publishMode, parent} = this,
            modeIsDefault = publishMode === 'default';

        if (model.matchesSelector(selector, modeIsDefault)) {
            return model;
        }

        // Try model's direct children. Wildcard not accepted (but would capture model itself above)
        if (modeIsDefault) {
            let ret = null;

            forOwn(model, (value, key) => {
                if (
                    !key.startsWith('_') &&
                    value?.isHoistModel &&
                    value.matchesSelector(selector)
                ) {
                    ret = value;
                    return false;
                }
            });
            if (ret) return ret;
        }

        // Try parent
        return parent?.lookupModel(selector) ?? null;
    }
}

/**
 * Context used to publish a ModelLookup
 */
export const ModelLookupContext = createContext(null);

export const modelLookupContextProvider = elementFactory(ModelLookupContext.Provider);