// -*- mode: js; js-indent-level: 4; indent-tabs-mode: nil -*-
/* exported ViewSelector */

const { Clutter, GObject, Shell, St } = imports.gi;

const AppDisplay = imports.ui.appDisplay;
const Main = imports.ui.main;
const OverviewControls = imports.ui.overviewControls;
const Search = imports.ui.search;
const ShellEntry = imports.ui.shellEntry;
const Util = imports.misc.util;
const WorkspaceThumbnail = imports.ui.workspaceThumbnail;
const WorkspacesView = imports.ui.workspacesView;

var ViewPage = {
    ACTIVITIES: 1,
    SEARCH: 2,
};

var FocusTrap = GObject.registerClass(
class FocusTrap extends St.Widget {
    vfunc_navigate_focus(from, direction) {
        if (direction == St.DirectionType.TAB_FORWARD ||
            direction == St.DirectionType.TAB_BACKWARD)
            return super.vfunc_navigate_focus(from, direction);
        return false;
    }
});

function getTermsForSearchString(searchString) {
    searchString = searchString.replace(/^\s+/g, '').replace(/\s+$/g, '');
    if (searchString == '')
        return [];

    let terms = searchString.split(/\s+/);
    return terms;
}


var ActivitiesContainer = GObject.registerClass(
class ActivitiesContainer extends St.Widget {
    _init(thumbnailsBox, workspacesDisplay, appDisplay, overviewAdjustment) {
        super._init();

        // 0 for window picker, 1 for app grid
        this._adjustment = new St.Adjustment({
            actor: this,
            value: 0,
            lower: 0,
            upper: 1,
        });
        this._adjustment.connect('notify::value', () => {
            this._update();
            this.queue_relayout();
        });

        overviewAdjustment.connect('notify::value', () => {
            const { ControlsState } = OverviewControls;

            const overviewState = overviewAdjustment.value;

            this._appDisplay.visible =
                overviewState >= ControlsState.WINDOW_PICKER;
            this._adjustment.value = Math.max(0,
                overviewAdjustment.value - ControlsState.WINDOW_PICKER);
        });

        this._thumbnailsBox = thumbnailsBox;
        this.add_child(thumbnailsBox);

        this._appDisplay = appDisplay;
        this.add_child(appDisplay);

        this._workspacesDisplay = workspacesDisplay;
        this.add_child(workspacesDisplay);

        this.connect('notify::mapped', () => {
            workspacesDisplay.setPrimaryWorkspaceVisible(this.mapped);
        });

        this._update();
    }

    _update() {
        const progress = this._adjustment.value;

        this._appDisplay.opacity = progress * 255;

        this._thumbnailsBox.set({
            scale_x: Util.lerp(1, 0.5, progress),
            scale_y: Util.lerp(1, 0.5, progress),
            translation_y: Util.lerp(0, this._thumbnailsBox.height, progress),
            opacity: Util.lerp(0, 255, 1 - progress),
            visible: (1 - progress) !== 0,
        });

        const { fitModeAdjustment } = this._workspacesDisplay;
        fitModeAdjustment.value = Util.lerp(
            WorkspacesView.FitMode.SINGLE,
            WorkspacesView.FitMode.ALL,
            progress);
    }

    _getWorkspacesBoxes(box, thumbnailsHeight) {
        const initialBox = box.copy();
        initialBox.y1 += thumbnailsHeight;

        const finalBox = box.copy();
        finalBox.set_size(
            box.get_width(),
            Math.round(box.get_height() * 0.15));

        return [initialBox, finalBox];
    }

    vfunc_allocate(box) {
        this.set_allocation(box);

        // Workspace Thumbnails
        let thumbnailsHeight = 0;
        if (this._thumbnailsBox.visible) {
            const maxThumbnailScale = WorkspaceThumbnail.MAX_THUMBNAIL_SCALE;
            const primaryMonitor = Main.layoutManager.primaryMonitor;
            const [width, height] = box.get_size();

            [, thumbnailsHeight] =
                this._thumbnailsBox.get_preferred_height(width);
            thumbnailsHeight = Math.min(
                thumbnailsHeight,
                (primaryMonitor ? primaryMonitor.height : height) * maxThumbnailScale);

            const thumbnailsBox = new Clutter.ActorBox();
            thumbnailsBox.set_origin(0, 0);
            thumbnailsBox.set_size(width, thumbnailsHeight);
            this._thumbnailsBox.allocate(thumbnailsBox);
        }

        const progress = this._adjustment.value;
        const [initialBox, finalBox] =
            this._getWorkspacesBoxes(box, thumbnailsHeight);
        const workspacesBox = initialBox.interpolate(finalBox, progress);
        this._workspacesDisplay.allocate(workspacesBox);

        if (this._appDisplay.visible) {
            const appDisplayBox = box.copy();
            appDisplayBox.y1 += Math.ceil(finalBox.get_height());
            this._appDisplay.allocate(appDisplayBox);
        }
    }
});

var ViewSelector = GObject.registerClass({
    Signals: {
        'page-changed': {},
        'page-empty': {},
    },
}, class ViewSelector extends Shell.Stack {
    _init(searchEntry, workspaceAdjustment, showAppsButton, overviewAdjustment) {
        super._init({
            name: 'viewSelector',
            x_expand: true,
            y_expand: true,
            visible: false,
        });

        this._showAppsButton = showAppsButton;
        this._showAppsButton.connect('notify::checked', this._onShowAppsButtonToggled.bind(this));

        this._activePage = null;

        this._searchActive = false;

        this._entry = searchEntry;
        ShellEntry.addContextMenu(this._entry);

        this._text = this._entry.clutter_text;
        this._text.connect('text-changed', this._onTextChanged.bind(this));
        this._text.connect('key-press-event', this._onKeyPress.bind(this));
        this._text.connect('key-focus-in', () => {
            this._searchResults.highlightDefault(true);
        });
        this._text.connect('key-focus-out', () => {
            this._searchResults.highlightDefault(false);
        });
        this._entry.connect('popup-menu', () => {
            if (!this._searchActive)
                return;

            this._entry.menu.close();
            this._searchResults.popupMenuDefault();
        });
        this._entry.connect('notify::mapped', this._onMapped.bind(this));
        global.stage.connect('notify::key-focus', this._onStageKeyFocusChanged.bind(this));

        this._entry.set_primary_icon(new St.Icon({ style_class: 'search-entry-icon',
                                                   icon_name: 'edit-find-symbolic' }));
        this._clearIcon = new St.Icon({ style_class: 'search-entry-icon',
                                        icon_name: 'edit-clear-symbolic' });

        this._iconClickedId = 0;
        this._capturedEventId = 0;

        this._thumbnailsBox =
            new WorkspaceThumbnail.ThumbnailsBox(workspaceAdjustment);
        this._workspacesDisplay =
            new WorkspacesView.WorkspacesDisplay(workspaceAdjustment, overviewAdjustment);
        this.appDisplay = new AppDisplay.AppDisplay();

        const activitiesContainer = new ActivitiesContainer(
            this._thumbnailsBox,
            this._workspacesDisplay,
            this.appDisplay,
            overviewAdjustment);
        this._activitiesPage =
            this._addPage(activitiesContainer, _('Activities'), 'view-app-grid-symbolic');

        Main.ctrlAltTabManager.addGroup(
            this.appDisplay,
            _('Applications'),
            'edit-find-symbolic', {
                proxy: this,
                focusCallback: () => {
                    this._showPage(this._activitiesPage);
                    this._showAppsButton.checked = true;
                    this.appDisplay.navigate_focus(
                        null, St.DirectionType.TAB_FORWARD, false);
                },
            });

        Main.ctrlAltTabManager.addGroup(
            this._workspacesDisplay,
            _('Windows'),
            'focus-windows-symbolic', {
                proxy: this,
                focusCallback: () => {
                    this._showPage(this._activitiesPage);
                    this._showAppsButton.checked = false;
                    this._workspacesDisplay.navigate_focus(
                        null, St.DirectionType.TAB_FORWARD, false);
                },
            });

        this._searchResults = new Search.SearchResultsView();
        this._searchPage = this._addPage(this._searchResults);
        Main.ctrlAltTabManager.addGroup(this._entry, _('Search'), 'edit-find-symbolic');

        // Since the entry isn't inside the results container we install this
        // dummy widget as the last results container child so that we can
        // include the entry in the keynav tab path
        this._focusTrap = new FocusTrap({ can_focus: true });
        this._focusTrap.connect('key-focus-in', () => {
            this._entry.grab_key_focus();
        });
        this._searchResults.add_actor(this._focusTrap);

        global.focus_manager.add_group(this._searchResults);

        this._stageKeyPressId = 0;
        Main.overview.connect('showing', () => {
            this._stageKeyPressId = global.stage.connect('key-press-event',
                                                         this._onStageKeyPress.bind(this));
        });
        Main.overview.connect('hiding', () => {
            if (this._stageKeyPressId != 0) {
                global.stage.disconnect(this._stageKeyPressId);
                this._stageKeyPressId = 0;
            }
        });
    }

    prepareToEnterOverview() {
        this.show();
        this.reset();
        this._workspacesDisplay.prepareToEnterOverview();
        this._activePage = null;
        this._showPage(this._activitiesPage);

        if (!this._workspacesDisplay.activeWorkspaceHasMaximizedWindows())
            Main.overview.fadeOutDesktop();
    }

    prepareToLeaveOverview() {
        this._workspacesDisplay.prepareToLeaveOverview();

        if (!this._workspacesDisplay.activeWorkspaceHasMaximizedWindows())
            Main.overview.fadeInDesktop();
    }

    vfunc_hide() {
        this.reset();
        this._workspacesDisplay.hide();

        super.vfunc_hide();
    }

    _addPage(actor) {
        let page = new St.Bin({ child: actor });
        page.hide();
        this.add_actor(page);
        return page;
    }

    _fadePageIn() {
        this._activePage.ease({
            opacity: 255,
            duration: OverviewControls.SIDE_CONTROLS_ANIMATION_TIME,
            mode: Clutter.AnimationMode.EASE_OUT_QUAD,
        });
    }

    _fadePageOut(page) {
        let oldPage = page;
        page.ease({
            opacity: 0,
            duration: OverviewControls.SIDE_CONTROLS_ANIMATION_TIME,
            mode: Clutter.AnimationMode.EASE_OUT_QUAD,
            onStopped: () => this._animateIn(oldPage),
        });
    }

    _animateIn(oldPage) {
        if (oldPage)
            oldPage.hide();

        this.emit('page-empty');

        if (this._activePage) {
            this._activePage.show();
            this._fadePageIn();
        }
    }

    _animateOut(page) {
        this._fadePageOut(page);
    }

    _showPage(page) {
        if (!Main.overview.visible)
            return;

        if (page == this._activePage)
            return;

        let oldPage = this._activePage;
        this._activePage = page;
        this.emit('page-changed');

        if (oldPage)
            this._animateOut(oldPage);
        else
            this._animateIn();
    }

    _onShowAppsButtonToggled() {
        this._showPage(this._activitiesPage);
    }

    _onStageKeyPress(actor, event) {
        // Ignore events while anything but the overview has
        // pushed a modal (system modals, looking glass, ...)
        if (Main.modalCount > 1)
            return Clutter.EVENT_PROPAGATE;

        let symbol = event.get_key_symbol();

        if (symbol === Clutter.KEY_Escape) {
            if (this._searchActive)
                this.reset();
            else if (this._showAppsButton.checked)
                this._showAppsButton.checked = false;
            else
                Main.overview.hide();
            return Clutter.EVENT_STOP;
        } else if (this._shouldTriggerSearch(symbol)) {
            this.startSearch(event);
        } else if (!this._searchActive && !global.stage.key_focus) {
            if (symbol === Clutter.KEY_Tab || symbol === Clutter.KEY_Down) {
                this._activePage.navigate_focus(null, St.DirectionType.TAB_FORWARD, false);
                return Clutter.EVENT_STOP;
            } else if (symbol === Clutter.KEY_ISO_Left_Tab) {
                this._activePage.navigate_focus(null, St.DirectionType.TAB_BACKWARD, false);
                return Clutter.EVENT_STOP;
            }
        }
        return Clutter.EVENT_PROPAGATE;
    }

    _searchCancelled() {
        this._showPage(this._activitiesPage);

        // Leave the entry focused when it doesn't have any text;
        // when replacing a selected search term, Clutter emits
        // two 'text-changed' signals, one for deleting the previous
        // text and one for the new one - the second one is handled
        // incorrectly when we remove focus
        // (https://bugzilla.gnome.org/show_bug.cgi?id=636341) */
        if (this._text.text != '')
            this.reset();
    }

    reset() {
        // Don't drop the key focus on Clutter's side if anything but the
        // overview has pushed a modal (e.g. system modals when activated using
        // the overview).
        if (Main.modalCount <= 1)
            global.stage.set_key_focus(null);

        this._entry.text = '';

        this._text.set_cursor_visible(true);
        this._text.set_selection(0, 0);
    }

    _onStageKeyFocusChanged() {
        let focus = global.stage.get_key_focus();
        let appearFocused = this._entry.contains(focus) ||
                             this._searchResults.contains(focus);

        this._text.set_cursor_visible(appearFocused);

        if (appearFocused)
            this._entry.add_style_pseudo_class('focus');
        else
            this._entry.remove_style_pseudo_class('focus');
    }

    _onMapped() {
        if (this._entry.mapped) {
            // Enable 'find-as-you-type'
            this._capturedEventId = global.stage.connect('captured-event',
                                                         this._onCapturedEvent.bind(this));
            this._text.set_cursor_visible(true);
            this._text.set_selection(0, 0);
        } else {
            // Disable 'find-as-you-type'
            if (this._capturedEventId > 0)
                global.stage.disconnect(this._capturedEventId);
            this._capturedEventId = 0;
        }
    }

    _shouldTriggerSearch(symbol) {
        if (symbol === Clutter.KEY_Multi_key)
            return true;

        if (symbol === Clutter.KEY_BackSpace && this._searchActive)
            return true;

        let unicode = Clutter.keysym_to_unicode(symbol);
        if (unicode == 0)
            return false;

        if (getTermsForSearchString(String.fromCharCode(unicode)).length > 0)
            return true;

        return false;
    }

    startSearch(event) {
        global.stage.set_key_focus(this._text);

        let synthEvent = event.copy();
        synthEvent.set_source(this._text);
        this._text.event(synthEvent, false);
    }

    // the entry does not show the hint
    _isActivated() {
        return this._text.text == this._entry.get_text();
    }

    _onTextChanged() {
        let terms = getTermsForSearchString(this._entry.get_text());

        this._searchActive = terms.length > 0;
        this._searchResults.setTerms(terms);

        if (this._searchActive) {
            this._showPage(this._searchPage);

            this._entry.set_secondary_icon(this._clearIcon);

            if (this._iconClickedId == 0) {
                this._iconClickedId = this._entry.connect('secondary-icon-clicked',
                                                          this.reset.bind(this));
            }
        } else {
            if (this._iconClickedId > 0) {
                this._entry.disconnect(this._iconClickedId);
                this._iconClickedId = 0;
            }

            this._entry.set_secondary_icon(null);
            this._searchCancelled();
        }
    }

    _onKeyPress(entry, event) {
        let symbol = event.get_key_symbol();
        if (symbol === Clutter.KEY_Escape) {
            if (this._isActivated()) {
                this.reset();
                return Clutter.EVENT_STOP;
            }
        } else if (this._searchActive) {
            let arrowNext, nextDirection;
            if (entry.get_text_direction() == Clutter.TextDirection.RTL) {
                arrowNext = Clutter.KEY_Left;
                nextDirection = St.DirectionType.LEFT;
            } else {
                arrowNext = Clutter.KEY_Right;
                nextDirection = St.DirectionType.RIGHT;
            }

            if (symbol === Clutter.KEY_Tab) {
                this._searchResults.navigateFocus(St.DirectionType.TAB_FORWARD);
                return Clutter.EVENT_STOP;
            } else if (symbol === Clutter.KEY_ISO_Left_Tab) {
                this._focusTrap.can_focus = false;
                this._searchResults.navigateFocus(St.DirectionType.TAB_BACKWARD);
                this._focusTrap.can_focus = true;
                return Clutter.EVENT_STOP;
            } else if (symbol === Clutter.KEY_Down) {
                this._searchResults.navigateFocus(St.DirectionType.DOWN);
                return Clutter.EVENT_STOP;
            } else if (symbol == arrowNext && this._text.position == -1) {
                this._searchResults.navigateFocus(nextDirection);
                return Clutter.EVENT_STOP;
            } else if (symbol === Clutter.KEY_Return || symbol === Clutter.KEY_KP_Enter) {
                this._searchResults.activateDefault();
                return Clutter.EVENT_STOP;
            }
        }
        return Clutter.EVENT_PROPAGATE;
    }

    _onCapturedEvent(actor, event) {
        if (event.type() == Clutter.EventType.BUTTON_PRESS) {
            let source = event.get_source();
            if (source != this._text &&
                this._text.has_key_focus() &&
                this._text.text == '' &&
                !this._text.has_preedit() &&
                !Main.layoutManager.keyboardBox.contains(source)) {
                // the user clicked outside after activating the entry, but
                // with no search term entered and no keyboard button pressed
                // - cancel the search
                this.reset();
            }
        }

        return Clutter.EVENT_PROPAGATE;
    }

    getActivePage() {
        if (this._activePage === this._activitiesPage)
            return ViewPage.ACTIVITIES;
        else
            return ViewPage.SEARCH;
    }
});