authentik/web/src/elements/forms/ModelForm.ts

import { EVENT_REFRESH } from "@goauthentik/common/constants";
import { Form } from "@goauthentik/elements/forms/Form";

import { TemplateResult } from "lit";
import { property } from "lit/decorators.js";

export abstract class ModelForm<T, PKT extends string | number> extends Form<T> {
    abstract loadInstance(pk: PKT): Promise<T>;

    @property({ attribute: false })
    set instancePk(value: PKT) {
        this._instancePk = value;
        if (this.viewportCheck && !this.isInViewport) {
            return;
        }
        this.loadInstance(value).then((instance) => {
            this.instance = instance;
            this.requestUpdate();
        });
    }

    private _instancePk?: PKT;

    private _initialLoad = false;

    @property({ attribute: false })
    instance?: T = this.defaultInstance;

    get defaultInstance(): T | undefined {
        return undefined;
    }

    constructor() {
        super();
        this.addEventListener(EVENT_REFRESH, () => {
            if (!this._instancePk) return;
            this.loadInstance(this._instancePk).then((instance) => {
                this.instance = instance;
            });
        });
    }

    resetForm(): void {
        this.instance = undefined;
        this._initialLoad = false;
    }

    render(): TemplateResult {
        if (this._instancePk && !this._initialLoad) {
            if (
                // if we're in viewport now and haven't loaded AND have a PK set, load now
                this.isInViewport ||
                // Or if we don't check for viewport in some cases
                !this.viewportCheck
            ) {
                this.instancePk = this._instancePk;
                this._initialLoad = true;
            }
        }
        return super.render();
    }
}