@logicflow/core 2.0.0-beta.2 → 2.0.0-beta.4

package/src/view/text/BaseText.tsx

@@ -1,63 +1,102 @@
-import { Component, createElement as h } from 'preact/compat'
-import { Text } from '../shape'
-import { StepDrag } from '../../util'
+import classNames from 'classnames'
+import { createElement as h, Component } from 'preact/compat'
 import { ElementState } from '../../constant'
-import { GraphModel, Model } from '../../model'
+import { GraphModel, BaseNodeModel, BaseEdgeModel } from '../../model'
+import { Text } from '../shape'
+import { IDragParams, StepDrag } from '../../util'
 
-type IProps = {
-  model: Model.BaseModel
+export type IBaseTextProps = {
+  model: BaseNodeModel | BaseEdgeModel
   graphModel: GraphModel
   draggable: boolean
   editable: boolean
 }
-type IState = {
+
+export type IBaseTextState = {
   isHovered: boolean
 }
 
-export class BaseText extends Component<IProps, IState> {
-  stepDrag: StepDrag
+export class BaseText<
+  P extends IBaseTextProps,
+  S extends IBaseTextState,
+> extends Component<P, S> {
+  stepperDrag: StepDrag
 
-  constructor(config) {
+  constructor(props: P) {
     super()
-    const { draggable } = config
-    this.stepDrag = new StepDrag({
+    const { draggable } = props
+    // TODO: 确认为什么不在 new 的时候传入 model，而在下面使用的时候赋值
+    this.stepperDrag = new StepDrag({
       onDragging: this.onDragging,
       step: 1,
+      // model,
       isStopPropagation: draggable,
     })
   }
 
   getShape(): h.JSX.Element | null {
     const { model, graphModel } = this.props
-    const { text } = model
     const { editConfigModel } = graphModel
-    const { value, x, y, editable, draggable } = text
+    const {
+      text: { value, x, y, editable, draggable },
+    } = model
     const attr = {
       x,
       y,
       className: '',
       value,
     }
-    if (editable) {
-      attr.className = 'lf-element-text'
-    } else if (draggable || editConfigModel.nodeTextDraggable) {
-      attr.className = 'lf-text-draggable'
-    } else {
-      attr.className = 'lf-text-disabled'
-    }
+    // DONE: 代码优化，看是否可以引入 classnames
+    // TODO: 确认下面逻辑是否正确，确认正确后删除下面注释
+    // if (editable) {
+    //   attr.className = 'lf-element-text';
+    // } else if (draggable || editConfigModel.nodeTextDraggable) {
+    //   attr.className = 'lf-text-draggable';
+    // } else {
+    //   attr.className = 'lf-text-disabled';
+    // }
     const style = model.getTextStyle()
-    return <Text {...attr} {...style} model={model} />
+    const isDraggable = editConfigModel.nodeTextDraggable || draggable
+
+    return (
+      <Text
+        {...attr}
+        {...style}
+        className={classNames({
+          'lf-element-text': editable,
+          'lf-text-draggable': !editable && isDraggable,
+          'lf-text-disabled': !editable && !isDraggable,
+        })}
+        model={model}
+      />
+    )
+  }
+
+  mouseDownHandler = (e: MouseEvent) => {
+    const { draggable, model, graphModel } = this.props
+    const {
+      editConfigModel: { nodeTextDraggable },
+    } = graphModel
+
+    if (draggable || nodeTextDraggable) {
+      this.stepperDrag.model = model
+      this.stepperDrag.handleMouseDown(e)
+    }
   }
 
-  onDragging = ({ deltaX, deltaY }) => {
+  onDragging = ({ deltaX, deltaY }: IDragParams) => {
     const {
       model,
       graphModel: { transformModel },
     } = this.props
-    const [curDeltaX, curDeltaY] = transformModel.fixDeltaXY(deltaX, deltaY)
-    model.moveText(curDeltaX, curDeltaY)
+
+    if (deltaX && deltaY) {
+      const [curDeltaX, curDeltaY] = transformModel.fixDeltaXY(deltaX, deltaY)
+      model.moveText(curDeltaX, curDeltaY)
+    }
   }
-  dblClickHandler = () => {
+
+  dbClickHandler = () => {
     // 静默模式下，双击不更改状态，不可编辑
     const { editable } = this.props
     if (editable) {
@@ -65,27 +104,14 @@ export class BaseText extends Component<IProps, IState> {
       model.setElementState(ElementState.TEXT_EDIT)
     }
   }
-  mouseDownHandle = (ev: MouseEvent) => {
-    const {
-      draggable,
-      model,
-      graphModel: {
-        editConfigModel: { nodeTextDraggable },
-      },
-    } = this.props
-    if (draggable || nodeTextDraggable) {
-      this.stepDrag.model = model
-      this.stepDrag.handleMouseDown(ev)
-    }
-  }
 
-  render() {
+  render(): h.JSX.Element | undefined {
     const {
       model: { text },
     } = this.props
     if (text) {
       return (
-        <g onMouseDown={this.mouseDownHandle} onDblClick={this.dblClickHandler}>
+        <g onMouseDown={this.mouseDownHandler} onDblClick={this.dbClickHandler}>
           {this.getShape()}
         </g>
       )

package/src/view/text/LineText.tsx

@@ -1,40 +1,62 @@
-import Text from '../shape/Text'
-import Rect from '../shape/Rect'
-import BaseText from './BaseText'
-import { BaseEdgeModel } from '../../model'
-import {
-  getBytesLength,
-  getHtmlTextHeight,
-  getSvgTextWidthHeight,
-} from '../../util'
-
-export class LineText extends BaseText {
-  constructor(config) {
-    super(config)
+import { createElement as h } from 'preact/compat'
+import { Text, ITextProps, Rect, IRectProps } from '..'
+import { BaseText } from '.'
+import { BaseEdgeModel, GraphModel } from '../../model'
+import { getHtmlTextHeight, getSvgTextSize } from '../../util'
+
+export type ILineTextProps = {
+  model: BaseEdgeModel
+  graphModel: GraphModel
+  draggable: boolean
+  editable: boolean
+  [key: string]: unknown
+}
+
+export type ILineTextState = {
+  isHovered: boolean
+}
+
+export class LineText extends BaseText<ILineTextProps, ILineTextState> {
+  constructor(props: ILineTextProps) {
+    super(props)
     this.state = {
       isHovered: false,
     }
   }
 
-  getBackground() {
-    const model = this.props.model as BaseEdgeModel
+  // Hover 状态相关
+  setHoverOn = () => {
+    this.setState({
+      isHovered: true,
+    })
+  }
+  setHoverOff = () => {
+    this.setState({
+      isHovered: false,
+    })
+  }
+
+  getBackground(): h.JSX.Element | null {
+    const { isHovered } = this.state
+    const { model } = this.props
+    const { text } = model
     const style = model.getTextStyle()
-    const { text, width: modelWidth } = model
+
     let backgroundStyle = style.background || {}
-    const { isHovered } = this.state
-    if (isHovered && style?.hover && style?.hover.background) {
-      backgroundStyle = { ...backgroundStyle, ...style?.hover.background }
+    if (isHovered && style.hover && style.hover.background) {
+      backgroundStyle = { ...backgroundStyle, ...style.hover.background }
     }
-    // 存在文本并且文本背景不为透明时计算背景框
-    if (text && text.value && backgroundStyle.fill !== 'transparent') {
-      const { fontSize, overflowMode, lineHeight, wrapPadding, textWidth } =
-        style ?? {}
-      const { value } = text
-      let { x, y } = text
-      const rows = String(value).split(/[\r\n]/g)
-      // 计算行数
+
+    // 当存在文本并且文本背景不为透明时，计算背景框
+    if (text?.value && backgroundStyle?.fill !== 'transparent') {
+      const { fontSize, textWidth, lineHeight, overflowMode } = style
+      const { wrapPadding } = backgroundStyle
+      const rows = text?.value.split(/[\r\n]/g)
       const rowsLength = rows.length
-      let rectAttr
+
+      let { x, y } = text
+      let rectAttr: unknown = {}
+
       if (overflowMode === 'autoWrap' && textWidth) {
         const textHeight = getHtmlTextHeight({
           rows,
@@ -47,30 +69,18 @@ export class LineText extends BaseText {
           rowsLength,
           className: 'lf-get-text-height',
         })
+
         rectAttr = {
           ...backgroundStyle,
-          x: x - 1,
-          y: y - 1,
+          x,
+          y,
           width: textWidth,
           height: textHeight,
         }
       } else {
-        // 计算文本中最长的一行的字节数
-        let longestBytes = 0
-        rows &&
-          rows.forEach((item) => {
-            const rowByteLength = getBytesLength(item)
-            longestBytes =
-              rowByteLength > longestBytes ? rowByteLength : longestBytes
-          })
-        // 背景框宽度，最长一行字节数/2 * fontsize + 2
-        // 背景框宽度， 行数 * fontsize + 2
-        let { width, height } = getSvgTextWidthHeight({
-          rows,
-          fontSize,
-          rowsLength,
-        })
-
+        // 背景框宽度，最长一行字节数 / 2 * fontSize + 2
+        // 背景框宽度，行数 * fontSize + 2
+        let { width, height } = getSvgTextSize({ rows, rowsLength, fontSize })
         if (overflowMode === 'ellipsis') {
           // https://github.com/didi/LogicFlow/issues/1151
           // 边上的文字过长（使用"ellipsis"模式）出现省略号，背景也需要进行宽度的重新计算
@@ -78,30 +88,38 @@ export class LineText extends BaseText {
           // 跟Text.tsx保持同样的计算逻辑(overflowMode === 'ellipsis')
           // Text.tsx使用textRealWidth=textWidth || width
           // Text.tsx使用foreignObjectHeight = fontSize + 2;
-          width = textWidth || (modelWidth as number)
+          width = textWidth
           height = fontSize + 2
         }
 
-        // 根据设置的padding调整width, height, x, y的值
+        // 根据设置的 padding 调整 width, height, x, y 的值
+        // TODO: 下面方法感觉可以提取成工具方法
         if (typeof backgroundStyle.wrapPadding === 'string') {
-          let paddings = backgroundStyle.wrapPadding
+          let padding = backgroundStyle.wrapPadding
             .split(',')
             .filter((padding) => padding.trim())
             .map((padding) => parseFloat(padding.trim()))
-          if (paddings.length > 0 && paddings.length <= 4) {
-            if (paddings.length === 1) {
-              paddings = [paddings[0], paddings[0], paddings[0], paddings[0]]
-            } else if (paddings.length === 2) {
-              paddings = [paddings[0], paddings[1], paddings[0], paddings[1]]
-            } else if (paddings.length === 3) {
-              paddings = [paddings[0], paddings[1], paddings[2], paddings[1]]
+
+          if (padding.length > 0 && padding.length <= 4) {
+            if (padding.length === 1) {
+              const [allSides] = padding
+              padding = [allSides, allSides, allSides, allSides]
+            } else if (padding.length === 2) {
+              const [vertical, horizontal] = padding
+              padding = [vertical, horizontal, vertical, horizontal]
+            } else if (padding.length === 3) {
+              const [top, horizontal, bottom] = padding
+              padding = [top, horizontal, bottom, horizontal]
             }
-            width += paddings[1] + paddings[3]
-            height += paddings[0] + paddings[2]
-            x = x + (paddings[1] - paddings[3]) / 2
-            y = y + (paddings[2] - paddings[0]) / 2
+
+            const [top, right, bottom, left] = padding
+            width += right + left
+            height += top + bottom
+            x = x + (right - left) / 2
+            y = y + (bottom - top) / 2
           }
         }
+
         rectAttr = {
           ...backgroundStyle,
           x: x - 1,
@@ -110,42 +128,38 @@ export class LineText extends BaseText {
           height,
         }
       }
-      return <Rect {...rectAttr} />
+
+      return <Rect {...(rectAttr as IRectProps)} />
     }
-  }
 
-  setHoverON = () => {
-    this.setState({
-      isHovered: true,
-    })
-  }
-  setHoverOFF = () => {
-    this.setState({
-      isHovered: false,
-    })
+    return null
   }
 
-  getShape() {
+  getShape(): h.JSX.Element | null {
     const { model } = this.props
-    const { text } = model
-    const { value, x, y } = text
+    const {
+      text: { x, y, value },
+    } = model
     if (!value) return null
+
     const style = model.getTextStyle()
-    const attr = {
+    const attrs: ITextProps = {
       x,
       y,
-      className: 'lf-element-text',
       value,
-      ...style, // 透传 edgeText 属性, 如 color fontSize fontWeight fontFamily textAnchor 等
+      model,
+      className: 'lf-element-text',
+      ...style, // 透传 edgeText 属性，如：color, fontSize, fontWeight, fontFamily, textAnchor 等
     }
+
     return (
       <g
         className="lf-line-text"
-        onMouseEnter={this.setHoverON}
-        onMouseLeave={this.setHoverOFF}
+        onMouseEnter={this.setHoverOn}
+        onMouseLeave={this.setHoverOff}
       >
         {this.getBackground()}
-        <Text {...attr} model={model} />
+        <Text {...attrs} />
       </g>
     )
   }