From 19c1b98f076587a7f8452a6babe22b3abd78cca6 Mon Sep 17 00:00:00 2001
From: Artur Arseniev <artur.catch@hotmail.it>
Date: Sat, 5 Aug 2023 13:44:09 +0400
Subject: [PATCH] Up fitViewport with el option

---
 src/canvas/view/CanvasView.ts            | 35 +++++++++++++++++++-----
 src/dom_components/view/ComponentView.ts | 23 ++++++++++------
 2 files changed, 42 insertions(+), 16 deletions(-)

diff --git a/src/canvas/view/CanvasView.ts b/src/canvas/view/CanvasView.ts
index ef5306aec..e3dcd6c1b 100644
--- a/src/canvas/view/CanvasView.ts
+++ b/src/canvas/view/CanvasView.ts
@@ -4,7 +4,7 @@ import { BoxRect, ElementRect } from '../../common';
 import Component from '../../dom_components/model/Component';
 import ComponentView from '../../dom_components/view/ComponentView';
 import { createEl, getDocumentScroll } from '../../utils/dom';
-import { getElRect, getElement, getKeyChar, getUiClass, isTextNode, off, on } from '../../utils/mixins';
+import { getElRect, getElement, getKeyChar, getUiClass, getViewEl, isTextNode, off, on } from '../../utils/mixins';
 import Canvas from '../model/Canvas';
 import Frame from '../model/Frame';
 import FrameView from './FrameView';
@@ -31,6 +31,7 @@ export interface FitViewportOptions {
   frame?: Frame;
   gap?: number | { x: number; y: number };
   ignoreHeight?: boolean;
+  el?: HTMLElement;
 }
 
 export default class CanvasView extends ModuleView<Canvas> {
@@ -193,9 +194,11 @@ export default class CanvasView extends ModuleView<Canvas> {
   }
 
   fitViewport(opts: FitViewportOptions = {}) {
-    const { em, module } = this;
+    const { em, module, model } = this;
     const canvasRect = this.getCanvasOffset();
-    const frame = opts.frame || em.getCurrentFrameModel() || this.model.frames.at(0);
+    const { el } = opts;
+    const elFrame = el && (getViewEl(el).frameView as FrameView);
+    const frame = elFrame ? elFrame.model : opts.frame || em.getCurrentFrameModel() || model.frames.at(0);
     const { x, y } = frame.attributes;
     const boxRect: BoxRect = {
       x: x ?? 0,
@@ -204,6 +207,14 @@ export default class CanvasView extends ModuleView<Canvas> {
       height: frame.height,
     };
 
+    if (el) {
+      const elRect = this.getElBoxRect(el);
+      boxRect.x = boxRect.x + elRect.x;
+      boxRect.y = boxRect.y + elRect.y;
+      boxRect.width = elRect.width;
+      boxRect.height = elRect.height;
+    }
+
     const noHeight = opts.ignoreHeight;
     const gap = opts.gap ?? 0;
     const gapIsNum = isNumber(gap);
@@ -220,9 +231,9 @@ export default class CanvasView extends ModuleView<Canvas> {
     const zoom = zoomRatio * 100;
     module.setZoom(zoom);
 
-    // check for the boxWidth is necessary as we're centering the frame via CSS
-    const coordX = boxRect.x + (boxWidth > canvasWidth ? canvasWidth / 2 - boxWidth / 2 : -gapX);
-    const coordY = boxRect.y + canvasHeight / 2 - boxHeight / 2;
+    // check for the frame witdh is necessary as we're centering the frame via CSS
+    const coordX = -boxRect.x + (frame.width >= canvasWidth ? canvasWidth / 2 - boxWidth / 2 : -gapX);
+    const coordY = -boxRect.y + canvasHeight / 2 - boxHeight / 2;
 
     const coords = {
       x: (coordX + gapX) * zoomRatio,
@@ -234,7 +245,7 @@ export default class CanvasView extends ModuleView<Canvas> {
       const canvasWorldHeight = canvasHeight * zoomMltp;
       const canvasHeightDiff = canvasWorldHeight - canvasHeight;
       const yDelta = canvasHeightDiff / 2;
-      coords.y = (boxRect.y + gapY) * zoomRatio - yDelta / zoomMltp;
+      coords.y = (-boxRect.y + gapY) * zoomRatio - yDelta / zoomMltp;
     }
 
     module.setCoords(coords.x, coords.y);
@@ -272,6 +283,16 @@ export default class CanvasView extends ModuleView<Canvas> {
     };
   }
 
+  getElBoxRect(el: HTMLElement): BoxRect {
+    const { width, height, left, top } = getElRect(el);
+    return {
+      x: left,
+      y: top,
+      width,
+      height,
+    };
+  }
+
   /**
    * Cleare cached offsets
    * @private
diff --git a/src/dom_components/view/ComponentView.ts b/src/dom_components/view/ComponentView.ts
index 9d87d5c34..9174e14cd 100644
--- a/src/dom_components/view/ComponentView.ts
+++ b/src/dom_components/view/ComponentView.ts
@@ -1,16 +1,17 @@
-import { isEmpty, each, keys, result } from 'underscore';
-import Component, { avoidInline } from '../model/Component';
-import Components from '../model/Components';
-import ComponentsView from './ComponentsView';
+import { each, isEmpty, keys, result } from 'underscore';
+import FrameView from '../../canvas/view/FrameView';
+import { ExtractMethods, ObjectAny, View } from '../../common';
+import { GetSetRuleOptions } from '../../css_composer';
+import Editor from '../../editor';
+import EditorModel from '../../editor/model/Editor';
 import Selectors from '../../selector_manager/model/Selectors';
 import { replaceWith } from '../../utils/dom';
 import { setViewEl } from '../../utils/mixins';
-import { ExtractMethods, ObjectAny, View } from '../../common';
-import { ComponentOptions } from '../model/types';
-import EditorModel from '../../editor/model/Editor';
 import { DomComponentsConfig } from '../config/config';
-import Editor from '../../editor';
-import { GetSetRuleOptions } from '../../css_composer';
+import Component, { avoidInline } from '../model/Component';
+import Components from '../model/Components';
+import { ComponentOptions } from '../model/types';
+import ComponentsView from './ComponentsView';
 
 type ClbObj = ReturnType<ComponentView['_clbObj']>;
 
@@ -94,6 +95,10 @@ Component> {
     return { state: '', mediaText: '' };
   }
 
+  get frameView(): FrameView {
+    return this.opts.config.frameView;
+  }
+
   __isDraggable() {
     const { model, config } = this;
     const { draggable } = model.attributes;