@finos/legend-application-studio/lib/stores/editor/editor-state/element-editor-state/ElementFileGenerationState.js

Legend Studio application core

/**
 * Copyright (c) 2020-present, Goldman Sachs
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import { FileGenerationState } from '../../editor-state/FileGenerationState.js';
import { action, flow, flowResult, makeObservable, observable } from 'mobx';
import { ElementEditorState } from './ElementEditorState.js';
import { AssertionError, uuid } from '@finos/legend-shared';
import { FileGenerationSpecification, PackageableElementExplicitReference, } from '@finos/legend-graph';
import { createObservableFileGeneration, fileGeneration_setType, } from '../../../graph-modifier/DSL_Generation_GraphModifierHelper.js';
import { handlePostCreateAction } from '../../NewElementState.js';
export class ElementFileGenerationState {
    uuid = uuid();
    editorStore;
    fileGenerationType;
    fileGenerationState;
    showNewFileGenerationModal = false;
    constructor(editorStore, fileGenerationType) {
        makeObservable(this, {
            fileGenerationState: observable,
            showNewFileGenerationModal: observable,
            setShowNewFileGenerationModal: action,
            promoteToFileGeneration: flow,
            regenerate: flow,
        });
        this.editorStore = editorStore;
        this.fileGenerationType = fileGenerationType;
        const fileGeneration = createObservableFileGeneration();
        fileGeneration_setType(fileGeneration, fileGenerationType);
        this.fileGenerationState = new FileGenerationState(editorStore, fileGeneration);
    }
    setShowNewFileGenerationModal(show) {
        this.showNewFileGenerationModal = show;
    }
    *promoteToFileGeneration(packagePath, name) {
        const fileGeneration = this.fileGenerationState.fileGeneration;
        fileGeneration.name = name;
        yield flowResult(this.editorStore.graphEditorMode.addElement(fileGeneration, packagePath, true));
        yield handlePostCreateAction(fileGeneration, this.editorStore);
        // reset file generation state so since the current file generation is promoted to a packageable element in the graph
        // otherwise if we keep this reference, editing this element generation state will also modify the packageable element
        const newFileGeneration = new FileGenerationSpecification('');
        fileGeneration_setType(newFileGeneration, this.fileGenerationType);
        this.fileGenerationState = new FileGenerationState(this.editorStore, newFileGeneration);
    }
    *regenerate() {
        const currentState = this.editorStore.tabManagerState.currentTab;
        if (currentState instanceof ElementEditorState) {
            this.fileGenerationState.fileGeneration.scopeElements = [
                PackageableElementExplicitReference.create(currentState.element),
            ]; // always set the scope to the current element
            yield flowResult(this.fileGenerationState.generate());
        }
        else {
            throw new AssertionError('Generation state must have at least an element editor opened');
        }
    }
}
//# sourceMappingURL=ElementFileGenerationState.js.map