1 files changed, 184 insertions, 0 deletions
diff --git a/extensions/vertical-workspaces/lib/workspaceAnimation.js b/extensions/vertical-workspaces/lib/workspaceAnimation.js
new file mode 100644
index 0000000..07008c6
--- /dev/null
+++ b/extensions/vertical-workspaces/lib/workspaceAnimation.js
@@ -0,0 +1,184 @@
+/**
+ * V-Shell (Vertical Workspaces)
+ * workspacesAnimation.js
+ *
+ * @author     GdH <G-dH@github.com>
+ * @copyright  2022 - 2023
+ * @license    GPL-3.0
+ *
+ */
+
+'use strict';
+const Main = imports.ui.main;
+const WorkspaceSwitcherPopup = imports.ui.workspaceSwitcherPopup;
+const WorkspaceAnimation = imports.ui.workspaceAnimation;
+const Me = imports.misc.extensionUtils.getCurrentExtension();
+const _Util = Me.imports.lib.util;
+
+// first reference to constant defined using const in other module returns undefined, the MonitorGroup const will remain empty and unused
+let MonitorGroupDummy = WorkspaceAnimation.MonitorGroup;
+MonitorGroupDummy = null;
+
+let _origBaseDistance;
+let _wsAnimationSwipeBeginId;
+let _wsAnimationSwipeUpdateId;
+let _wsAnimationSwipeEndId;
+
+let _overrides;
+let opt;
+let _firstRun = true;
+
+function update(reset = false) {
+    opt = Me.imports.lib.settings.opt;
+    const moduleEnabled = opt.get('workspaceAnimationModule', true);
+    reset = reset || !moduleEnabled;
+
+    // don't even touch this module if disabled
+    if (_firstRun && reset)
+        return;
+
+    _firstRun = false;
+
+    if (_overrides)
+        _overrides.removeAll();
+
+    if (reset || !moduleEnabled) {
+        _connectWsAnimationSwipeTracker(true);
+        _overrideMonitorGroupProperty(true);
+        _overrides = null;
+        opt = null;
+        return;
+    }
+
+    if (opt.STATIC_WS_SWITCHER_BG) {
+        _overrides = new _Util.Overrides();
+        _overrideMonitorGroupProperty();
+        _overrides.addOverride('WorkspaceAnimationMonitorGroup', WorkspaceAnimation.MonitorGroup.prototype, MonitorGroup);
+    }
+
+    _connectWsAnimationSwipeTracker();
+}
+
+// remove spacing between workspaces during transition to remove flashing wallpaper between workspaces with maximized windows
+function _overrideMonitorGroupProperty(reset = false) {
+    if (!_origBaseDistance)
+        _origBaseDistance = Object.getOwnPropertyDescriptor(WorkspaceAnimation.MonitorGroup.prototype, 'baseDistance').get;
+
+    let getter;
+    if (reset) {
+        if (_origBaseDistance)
+            getter = { get: _origBaseDistance };
+    } else {
+        getter = {
+            get() {
+                // const spacing = 100 * imports.gi.St.ThemeContext.get_for_stage(global.stage).scale_factor;
+                const spacing = 0;
+                if (global.workspace_manager.layout_rows === -1)
+                    return this._monitor.height + spacing + (opt.PANEL_MODE ? Main.panel.height : 0); // compensation for hidden panel
+                else
+                    return this._monitor.width + spacing;
+            },
+        };
+    }
+
+    if (getter)
+        Object.defineProperty(WorkspaceAnimation.MonitorGroup.prototype, 'baseDistance', getter);
+}
+
+const MonitorGroup = {
+    // injection to _init()
+    after__init() {
+        // we have two options to implement static bg feature
+        // one is adding background to monitorGroup
+        // but this one has disadvantage - sticky windows will be always on top of animated windows
+        // which is bad for conky, for example, that window should be always below
+        /* this._bgManager = new Background.BackgroundManager({
+            container: this,
+            monitorIndex: this._monitor.index,
+            controlPosition: false,
+        });*/
+
+        // the second option is to make background of the monitorGroup transparent so the real desktop content will stay visible,
+        // hide windows that should be animated and keep only sticky windows
+        // we can keep certain sticky windows bellow and also extensions like DING (icons on desktop) will stay visible
+        this.set_style('background-color: transparent;');
+        // stickyGroup holds the Always on Visible Workspace windows to keep them static and above other windows during animation
+        const stickyGroup = this.get_children()[1];
+        stickyGroup._windowRecords.forEach(r => {
+            const metaWin = r.windowActor.metaWindow;
+            // conky is sticky but should never get above other windows during ws animation
+            // so we hide it from the overlay group, we will see the original if not covered by other windows
+            if (metaWin.wm_class === 'conky')
+                r.clone.opacity = 0;
+        });
+        this._hiddenWindows = [];
+        // remove (hide) background wallpaper from the animation, we will see the original one
+        this._workspaceGroups.forEach(w => {
+            w._background.opacity = 0;
+        });
+        // hide (scale to 0) all non-sticky windows, their clones will be animated
+        global.get_window_actors().forEach(actor => {
+            const metaWin = actor.metaWindow;
+            if (metaWin?.get_monitor() === this._monitor.index && !(metaWin?.wm_class === 'conky' && metaWin?.is_on_all_workspaces())) { //* && !w.is_on_all_workspaces()*/) {
+                // hide original window. we cannot use opacity since it also affects clones.
+                // scaling them to 0 works well
+                actor.scale_x = 0;
+                this._hiddenWindows.push(actor);
+            }
+        });
+
+        // restore all hidden windows at the end of animation
+        // todo - actors removed during transition need to be removed from the list  to avoid access to destroyed actor
+        this.connect('destroy', () => {
+            this._hiddenWindows.forEach(actor => {
+                actor.scale_x = 1;
+            });
+        });
+    },
+};
+
+function _connectWsAnimationSwipeTracker(reset = false) {
+    if (reset) {
+        if (_wsAnimationSwipeBeginId) {
+            Main.wm._workspaceAnimation._swipeTracker.disconnect(_wsAnimationSwipeBeginId);
+            _wsAnimationSwipeBeginId = 0;
+        }
+        if (_wsAnimationSwipeEndId) {
+            Main.wm._workspaceAnimation._swipeTracker.disconnect(_wsAnimationSwipeEndId);
+            _wsAnimationSwipeEndId = 0;
+        }
+    } else if (!_wsAnimationSwipeBeginId) {
+        // display ws switcher popup when gesture begins and connect progress
+        _wsAnimationSwipeBeginId = Main.wm._workspaceAnimation._swipeTracker.connect('begin', () => _connectWsAnimationProgress(true));
+        // we want to be sure that popup with the final ws index show up when gesture ends
+        _wsAnimationSwipeEndId = Main.wm._workspaceAnimation._swipeTracker.connect('end', (tracker, duration, endProgress) => _connectWsAnimationProgress(false, endProgress));
+    }
+}
+
+function _connectWsAnimationProgress(connect, endProgress = null) {
+    if (Main.overview.visible)
+        return;
+
+    if (connect && !_wsAnimationSwipeUpdateId) {
+        _wsAnimationSwipeUpdateId = Main.wm._workspaceAnimation._swipeTracker.connect('update', (tracker, progress) => _showWsSwitcherPopup(progress));
+    } else if (!connect && _wsAnimationSwipeUpdateId) {
+        Main.wm._workspaceAnimation._swipeTracker.disconnect(_wsAnimationSwipeUpdateId);
+        _wsAnimationSwipeUpdateId = 0;
+        _showWsSwitcherPopup(Math.round(endProgress));
+    }
+}
+
+function _showWsSwitcherPopup(progress) {
+    if (Main.overview.visible)
+        return;
+
+    const wsIndex = Math.round(progress);
+    if (Main.wm._workspaceSwitcherPopup === null) {
+        Main.wm._workspaceSwitcherPopup = new WorkspaceSwitcherPopup.WorkspaceSwitcherPopup();
+        Main.wm._workspaceSwitcherPopup.connect('destroy', () => {
+            Main.wm._workspaceSwitcherPopup = null;
+        });
+    }
+
+    Main.wm._workspaceSwitcherPopup.display(wsIndex);
+}