From b22f0b45418b76ba008a2f6200bf8210f3c6f0dd Mon Sep 17 00:00:00 2001
From: Hufe921 <huangyunfeihufe@hotmail.com>
Date: Mon, 31 Jul 2023 20:02:27 +0800
Subject: [PATCH] fix: set header and footer data error #224

---
 src/editor/core/draw/Draw.ts         | 20 ++++++++------------
 src/editor/core/draw/frame/Footer.ts |  4 ++--
 src/editor/core/draw/frame/Header.ts |  4 ++--
 3 files changed, 12 insertions(+), 16 deletions(-)

diff --git a/src/editor/core/draw/Draw.ts b/src/editor/core/draw/Draw.ts
index 6189cad..9e02c24 100644
--- a/src/editor/core/draw/Draw.ts
+++ b/src/editor/core/draw/Draw.ts
@@ -93,9 +93,7 @@ export class Draw {
   private options: DeepRequired<IEditorOption>
   private position: Position
   private zone: Zone
-  private headerElementList: IElement[]
   private elementList: IElement[]
-  private footerElementList: IElement[]
   private listener: Listener
   private eventBus: EventBus<EventBusMap>
 
@@ -159,9 +157,7 @@ export class Draw {
     this.pagePixelRatio = null
     this.mode = options.mode
     this.options = options
-    this.headerElementList = data.header || []
     this.elementList = data.main
-    this.footerElementList = data.footer || []
     this.listener = listener
     this.eventBus = eventBus
 
@@ -189,8 +185,8 @@ export class Draw {
     this.pageNumber = new PageNumber(this)
     this.waterMark = new Watermark(this)
     this.placeholder = new Placeholder(this)
-    this.header = new Header(this)
-    this.footer = new Footer(this)
+    this.header = new Header(this, data.header)
+    this.footer = new Footer(this, data.footer)
     this.hyperlinkParticle = new HyperlinkParticle(this)
     this.dateParticle = new DateParticle(this)
     this.separatorParticle = new SeparatorParticle()
@@ -440,7 +436,7 @@ export class Draw {
   }
 
   public getHeaderElementList(): IElement[] {
-    return this.headerElementList
+    return this.header.getElementList()
   }
 
   public getTableElementList(sourceElementList: IElement[]): IElement[] {
@@ -467,10 +463,10 @@ export class Draw {
   public getOriginalElementList() {
     const zoneManager = this.getZone()
     if (zoneManager.isHeaderActive()) {
-      return this.header.getElementList()
+      return this.getHeaderElementList()
     }
     if (zoneManager.isFooterActive()) {
-      return this.footer.getElementList()
+      return this.getFooterElementList()
     }
     return this.elementList
   }
@@ -480,7 +476,7 @@ export class Draw {
   }
 
   public getFooterElementList(): IElement[] {
-    return this.footerElementList
+    return this.footer.getElementList()
   }
 
   public insertElementList(payload: IElement[]) {
@@ -849,9 +845,9 @@ export class Draw {
       )
     }
     const data: IEditorData = {
-      header: zipElementList(this.headerElementList),
+      header: zipElementList(this.getHeaderElementList()),
       main: zipElementList(mainElementList),
-      footer: zipElementList(this.footerElementList)
+      footer: zipElementList(this.getFooterElementList())
     }
     return {
       version,
diff --git a/src/editor/core/draw/frame/Footer.ts b/src/editor/core/draw/frame/Footer.ts
index 50b046b..e5fbb7a 100644
--- a/src/editor/core/draw/frame/Footer.ts
+++ b/src/editor/core/draw/frame/Footer.ts
@@ -16,12 +16,12 @@ export class Footer {
   private rowList: IRow[]
   private positionList: IElementPosition[]
 
-  constructor(draw: Draw) {
+  constructor(draw: Draw, data?: IElement[]) {
     this.draw = draw
     this.position = draw.getPosition()
     this.options = draw.getOptions()
 
-    this.elementList = draw.getFooterElementList()
+    this.elementList = data || []
     this.rowList = []
     this.positionList = []
   }
diff --git a/src/editor/core/draw/frame/Header.ts b/src/editor/core/draw/frame/Header.ts
index 2b63122..3dd5deb 100644
--- a/src/editor/core/draw/frame/Header.ts
+++ b/src/editor/core/draw/frame/Header.ts
@@ -16,12 +16,12 @@ export class Header {
   private rowList: IRow[]
   private positionList: IElementPosition[]
 
-  constructor(draw: Draw) {
+  constructor(draw: Draw, data?: IElement[]) {
     this.draw = draw
     this.position = draw.getPosition()
     this.options = draw.getOptions()
 
-    this.elementList = draw.getHeaderElementList()
+    this.elementList = data || []
     this.rowList = []
     this.positionList = []
   }