AtomicGameEngine/Script/AtomicEditor/hostExtensions/HostExtensionServices.ts

//
// Copyright (c) 2014-2016 THUNDERBEAST GAMES LLC
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//

import * as EditorEvents from "../editor/EditorEvents";
import * as EditorUI from "../ui/EditorUI";
import MainFramMenu = require("../ui/frames/menus/MainFrameMenu");
import HierarchyFrameMenu = require("../ui/frames/menus/HierarchyFrameMenu");
import ProjectFrameMenu = require("../ui/frames/menus/ProjectFrameMenu");
import ModalOps = require("../ui/modal/ModalOps");
/**
 * Generic registry for storing Editor Extension Services
 */
export class ServicesProvider<T extends Editor.Extensions.ServiceEventListener> implements Editor.Extensions.ServicesProvider<T> {
    registeredServices: T[] = [];

    /**
     * Adds a service to the registered services list for this type of service
     * @param  {T}      service the service to register
     */
    register(service: T) {
        this.registeredServices.push(service);
    }

    unregister(service: T) {
        var index = this.registeredServices.indexOf(service, 0);
        if (index > -1) {
            this.registeredServices.splice(index, 1);
        }
    }
}

export interface ServiceEventSubscriber {
    /**
     * Allow this service registry to subscribe to events that it is interested in
     * @param  {Atomic.UIWidget} topLevelWindow The top level window that will be receiving these events
     */
    subscribeToEvents(topLevelWindow: Atomic.UIWidget);
}

/**
 * Registry for service extensions that are concerned about project events
 */
export class ProjectServicesProvider extends ServicesProvider<Editor.HostExtensions.ProjectServicesEventListener> implements Editor.HostExtensions.ProjectServicesProvider {
    constructor() {
        super();
    }

    /**
     * Allow this service registry to subscribe to events that it is interested in
     * @param  {Atomic.UIWidget} topLevelWindow The top level window that will be receiving these events
     */
    subscribeToEvents(eventDispatcher: Editor.Extensions.EventDispatcher) {
        eventDispatcher.subscribeToEvent(EditorEvents.LoadProjectNotification, (ev) => this.projectLoaded(ev));
        eventDispatcher.subscribeToEvent(EditorEvents.CloseProject, (ev) => this.projectUnloaded(ev));
        eventDispatcher.subscribeToEvent(EditorEvents.PlayerStartRequest, () => this.playerStarted());
    }

    /**
     * Called when the project is unloaded
     * @param  {[type]} data Event info from the project unloaded event
     */
    projectUnloaded(data) {
        // Need to use a for loop for length down to 0 because extensions *could* delete themselves from the list on projectUnloaded
        for (let i = this.registeredServices.length - 1; i >= 0; i--) {
            let service = this.registeredServices[i];
            // Notify services that the project has been unloaded
            try {
                if (service.projectUnloaded) {
                    service.projectUnloaded();
                }
            } catch (e) {
                EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}:\n${e} \n\n ${e.stack}`);
            }
        };
    }

    /**
     * Called when the project is loaded
     * @param  {[type]} data Event info from the project unloaded event
     */
    projectLoaded(ev: Editor.EditorEvents.LoadProjectEvent) {
        // Need to use a for loop and don't cache the length because the list of services *may* change while processing.  Extensions could be appended to the end
        for (let i = 0; i < this.registeredServices.length; i++) {
            let service = this.registeredServices[i];
            try {
                // Notify services that the project has just been loaded
                if (service.projectLoaded) {
                    service.projectLoaded(ev);
                }
            } catch (e) {
                EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}:\n${e}\n\n ${e.stack}`);
            }
        };
    }

    playerStarted() {
        this.registeredServices.forEach((service) => {
            try {
                // Notify services that the project has just been loaded
                if (service.playerStarted) {
                    service.playerStarted();
                }
            } catch (e) {
                EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}\n \n ${e.stack}`);
            }
        });
    }
}

/**
 * Registry for service extensions that are concerned about Resources
 */
export class ResourceServicesProvider extends ServicesProvider<Editor.HostExtensions.ResourceServicesEventListener> implements Editor.HostExtensions.ResourceServicesProvider {
    constructor() {
        super();
    }

    /**
     * Allow this service registry to subscribe to events that it is interested in
     * @param  {Atomic.UIWidget} topLevelWindow The top level window that will be receiving these events
     */
    subscribeToEvents(eventDispatcher: Editor.Extensions.EventDispatcher) {
        eventDispatcher.subscribeToEvent(EditorEvents.SaveResourceNotification, (ev) => this.saveResource(ev));
        eventDispatcher.subscribeToEvent(EditorEvents.DeleteResourceNotification, (ev) => this.deleteResource(ev));
        eventDispatcher.subscribeToEvent(EditorEvents.RenameResourceNotification, (ev) => this.renameResource(ev));
    }

    /**
     * Called after a resource has been saved
     * @param  {Editor.EditorEvents.SaveResourceEvent} ev
     */
    saveResource(ev: Editor.EditorEvents.SaveResourceEvent) {
        // run through and find any services that can handle this.
        this.registeredServices.forEach((service) => {
            try {
                // Verify that the service contains the appropriate methods and that it can save
                if (service.save) {
                    service.save(ev);
                }
            } catch (e) {
                EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}:\n${e}\n\n ${e.stack}`);
            }
        });
    }

    /**
     * Called when a resource has been deleted
     */
    deleteResource(ev: Editor.EditorEvents.DeleteResourceEvent) {
        this.registeredServices.forEach((service) => {
            try {
                // Verify that the service contains the appropriate methods and that it can delete
                if (service.delete) {
                    service.delete(ev);
                }
            } catch (e) {
                EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}:\n${e}\n\n ${e.stack}`);
            }
        });
    }

    /**
     * Called when a resource has been renamed
     * @param  {Editor.EditorEvents.RenameResourceEvent} ev
     */
    renameResource(ev: Editor.EditorEvents.RenameResourceEvent) {
        this.registeredServices.forEach((service) => {
            try {
                // Verify that the service contains the appropriate methods and that it can handle the rename
                if (service.rename) {
                    service.rename(ev);
                }
            } catch (e) {
                EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}:\n${e}\n\n ${e.stack}`);
            }
        });
    }

}

/**
 * Registry for service extensions that are concerned about and need access to parts of the editor user interface
 * Note: we may want to move this out into it's own file since it has a bunch of editor dependencies
 */
export class UIServicesProvider extends ServicesProvider<Editor.HostExtensions.UIServicesEventListener> implements Editor.HostExtensions.UIServicesProvider {
    constructor() {
        super();
    }

    private mainFrameMenu: MainFramMenu = null;
    private hierarchyFrameMenu: HierarchyFrameMenu = null;
    private projectFrameMenu: ProjectFrameMenu = null;
    private modalOps: ModalOps;

    init(mainFrameMenu: MainFramMenu, hierarchyFrameMenu: HierarchyFrameMenu, projectFrameMenu: ProjectFrameMenu, modalOps: ModalOps) {
        // Only set these once
        if (this.mainFrameMenu == null) {
            this.mainFrameMenu = mainFrameMenu;
        }
        if (this.hierarchyFrameMenu == null) {
            this.hierarchyFrameMenu = hierarchyFrameMenu;
        }
        if (this.projectFrameMenu == null) {
            this.projectFrameMenu = projectFrameMenu;
        }
        if (this.modalOps == null) {
            this.modalOps = modalOps;
        }
    }

    /**
     * Adds a new menu to the plugin menu
     * @param  {string} id
     * @param  {any} items
     * @return {Atomic.UIMenuItemSource}
     */
    createPluginMenuItemSource(id: string, items: any): Atomic.UIMenuItemSource {
        return this.mainFrameMenu.createPluginMenuItemSource(id, items);
    }

    /**
     * Removes a previously added menu from the plugin menu
     * @param  {string} id
     */
    removePluginMenuItemSource(id: string) {
        this.mainFrameMenu.removePluginMenuItemSource(id);
    }

    /**
     * Adds a new menu to the hierarchy context menu
     * @param  {string} id
     * @param  {any} items
     * @return {Atomic.UIMenuItemSource}
     */
    createHierarchyContextMenuItemSource(id: string, items: any): Atomic.UIMenuItemSource {
        return this.hierarchyFrameMenu.createPluginItemSource(id, items);
    }

    /**
     * Removes a previously added menu from the hierarchy context menu
     * @param  {string} id
     */
    removeHierarchyContextMenuItemSource(id: string) {
        this.hierarchyFrameMenu.removePluginItemSource(id);
    }

    /**
     * Adds a new menu to the project context menu
     * @param  {string} id
     * @param  {any} items
     * @return {Atomic.UIMenuItemSource}
     */
    createProjectContextMenuItemSource(id: string, items: any): Atomic.UIMenuItemSource {
        return this.projectFrameMenu.createPluginItemSource(id, items);
    }

    /**
     * Removes a previously added menu from the project context menu
     * @param  {string} id
     */
    removeProjectContextMenuItemSource(id: string) {
        this.projectFrameMenu.removePluginItemSource(id);
    }

    /**
     * Disaplays a modal window
     * @param  {Editor.Modal.ModalWindow} window
     */
    showModalWindow(windowText: string, uifilename: string, handleWidgetEventCB: (ev: Atomic.UIWidgetEvent) => void): Editor.Modal.ExtensionWindow {
        return this.modalOps.showExtensionWindow(windowText, uifilename, handleWidgetEventCB);
    }

    /**
     * Called when a menu item has been clicked
     * @param  {string} refId
     * @type {boolean} return true if handled
     */
    menuItemClicked(refid: string): boolean {
        // run through and find any services that can handle this.
        return this.registeredServices.some((service) => {
            try {
                // Verify that the service contains the appropriate methods and that it can handle it
                if (service.menuItemClicked) {
                    if (service.menuItemClicked(refid)) {
                        return true;
                    }
                }
            } catch (e) {
               EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}:\n${e}\n\n ${e.stack}`);
            }
        });
    }

    /**
     * Called when a context menu item in the hierarchy pane has been clicked
     * @param  {Atomic.Node} node
     * @param  {string} refId
     * @type {boolean} return true if handled
     */
    hierarchyContextItemClicked(node: Atomic.Node, refid: string): boolean {
        if (!node) 
            return false;

        // run through and find any services that can handle this.
        return this.registeredServices.some((service) => {
            try {
                // Verify that the service contains the appropriate methods and that it can handle it
                if (service.hierarchyContextItemClicked) {
                    if (service.hierarchyContextItemClicked(node, refid)) {
                        return true;
                    }
                }
            } catch (e) {
               EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}:\n${e}\n\n ${e.stack}`);
            }
        });
    }


    /**
     * Called when a context menu item in the hierarchy pane has been clicked
     * @param  {ToolCore.Asset} asset
     * @param  {string} refId
     * @type {boolean} return true if handled
     */
    projectContextItemClicked(asset: ToolCore.Asset, refid: string): boolean {
        if (!asset)
            return false;

        // run through and find any services that can handle this.
        return this.registeredServices.some((service) => {
            try {
                // Verify that the service contains the appropriate methods and that it can handle it
                if (service.projectContextItemClicked) {
                    if (service.projectContextItemClicked(asset, refid)) {
                        return true;
                    }
                }
            } catch (e) {
               EditorUI.showModalError("Extension Error", `Error detected in extension ${service.name}:\n${e}\n\n ${e.stack}`);
            }
        });
    }

    /**
     * Allow this service registry to subscribe to events that it is interested in
     * @param  {Atomic.UIWidget} topLevelWindow The top level window that will be receiving these events
     */
    subscribeToEvents(eventDispatcher: Editor.Extensions.EventDispatcher) {
        // Placeholder for when UI events published by the editor need to be listened for
        //eventDispatcher.subscribeToEvent(EditorEvents.SaveResourceNotification, (ev) => this.doSomeUiMessage(ev));
    }
}