fix(links): build links when change view type

worktile · Jun 20, 2020 · 3f5655f · 3f5655f
1 parent 5ef8441
commit 3f5655f
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 2 deletions.
diff --git a/packages/gantt/src/components/links/links.component.ts b/packages/gantt/src/components/links/links.component.ts
@@ -153,7 +153,6 @@ export class GanttLinksComponent implements OnInit, OnDestroy {
     }
 
     buildLinks() {
-        console.log(1);
         this.computeItemPosition();
         this.links = [];
         this.linkItems.forEach((source) => {

diff --git a/packages/gantt/src/components/main/gantt-main.component.ts b/packages/gantt/src/components/main/gantt-main.component.ts
@@ -18,6 +18,8 @@ export class GanttMainComponent implements OnInit {
 
     @Output() lineClick = new EventEmitter<GanttLineClickEvent>();
 
+    @ViewChild(GanttLinksComponent, { static: false, }) links: GanttLinksComponent;
+
     @HostBinding('class.gantt-main-container') ganttMainClass = true;
 
     constructor(@Inject(GANTT_REF_TOKEN) public ganttRef: GanttRef) {}

diff --git a/packages/gantt/src/gantt-upper.ts b/packages/gantt/src/gantt-upper.ts
@@ -32,6 +32,7 @@ import { Subject } from 'rxjs';
 import { GanttCalendarComponent } from './components/calendar/calendar.component';
 import { uniqBy } from './utils/helpers';
 
+
 export abstract class GanttUpper {
     @Input('items') originItems: GanttItem[] = [];
 

diff --git a/packages/gantt/src/gantt.component.ts b/packages/gantt/src/gantt.component.ts
@@ -15,7 +15,8 @@ import {
     TemplateRef,
     ContentChildren,
     QueryList,
-    AfterViewInit
+    AfterViewInit,
+    ViewChild
 } from '@angular/core';
 import { startWith, takeUntil } from 'rxjs/operators';
 import { Subject } from 'rxjs';
@@ -27,6 +28,7 @@ import { GanttDragContainer } from './gantt-drag-container';
 import { NgxGanttTableColumnComponent } from './table/gantt-column.component';
 import { sideWidth } from './gantt.styles';
 import { getColumnWidthConfig } from './utils/column-compute';
+import { GanttMainComponent } from './components/main/gantt-main.component';
 
 @Component({
     selector: 'ngx-gantt',
@@ -60,6 +62,8 @@ export class NgxGanttComponent extends GanttUpper implements GanttRef, OnInit, A
 
     @ContentChildren(NgxGanttTableColumnComponent, { descendants: true }) columns: QueryList<NgxGanttTableColumnComponent>;
 
+    @ViewChild(GanttMainComponent, { static: false }) mainContent: GanttMainComponent;
+
     constructor(
         elementRef: ElementRef<HTMLElement>,
         cdr: ChangeDetectorRef,
@@ -111,6 +115,11 @@ export class NgxGanttComponent extends GanttUpper implements GanttRef, OnInit, A
 
     ngOnChanges(changes: SimpleChanges) {
         super.onChanges(changes);
+        if (!this.firstChange) {
+            if (changes.viewType && changes.viewType.currentValue) {
+                this.mainContent.links.buildLinks();
+            }
+        }
     }
 
     computeRefs() {