fix(dependencies): remove dependencies fromTask model on remove event.

angular-gantt · Oct 6, 2016 · b1a08e5 · b1a08e5
1 parent e7d0e44
commit b1a08e5
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 12 deletions.
diff --git a/src/plugins/dependencies.js b/src/plugins/dependencies.js
@@ -141,7 +141,7 @@
                         });
 
                         api.tasks.on.add(scope, function(task) {
-                            manager.addDependenciesFromTask(task);
+                            manager.addDependenciesFromTask(task, true);
                         });
 
                         api.tasks.on.remove(scope, function(task) {
@@ -190,7 +190,21 @@
 
                         api.dependencies.on.remove(scope, function(dependency) {
                             if (scope.conflictChecker && scope.enabled) {
-                                checker.refresh([dependency.getFromTask(), dependency.getToTask()]);
+                                var fromTask = dependency.getFromTask();
+                                var toTask = dependency.getToTask();
+
+                                if (fromTask && toTask) {
+                                    checker.refresh([fromTask, toTask]);
+                                } else {
+
+                                    if (fromTask) {
+                                        checker.removeConflictClass(fromTask);
+                                    } else {
+                                        checker.removeConflictClass(toTask);
+                                    }
+
+                                }
+
                             }
                         });
 

diff --git a/src/plugins/dependencies/dependenciesChecker.factory.js b/src/plugins/dependencies/dependenciesChecker.factory.js
@@ -60,6 +60,10 @@
                 handleTaskNonConflict(conflictsList, allTasks);
             };
 
+            this.removeConflictClass = function(task) {
+                task.$element.removeClass('gantt-task-conflict');
+            };
+
             /**
              * Remove the conflict status of given tasks.
              *

diff --git a/src/plugins/dependencies/dependenciesManager.factory.js b/src/plugins/dependencies/dependenciesManager.factory.js
@@ -50,7 +50,7 @@
              *
              * @param task
              */
-            this.addDependenciesFromTask = function(task) {
+            this.addDependenciesFromTask = function(task, isDraw) {
                 if (this.pluginScope.enabled) {
                     var taskDependencies = task.model.dependencies;
 
@@ -61,8 +61,11 @@
                         }
 
                         for (var i = 0, l = taskDependencies.length; i < l; i++) {
-                            var dependency = self.addDependency(task, taskDependencies[i]);
-                            dependency.connect();
+                            var dependency = self.addDependency(task, taskDependencies[i], isDraw);
+
+                            if (dependency !== null) {
+                                dependency.connect();
+                            }
                         }
                     }
                 }
@@ -93,11 +96,12 @@
              * @param task Task defining the dependency.
              * @param model Model object for the dependency.
              */
-            this.addDependency = function(task, model) {
+            this.addDependency = function(task, model, isDraw) {
                 var dependency = new Dependency(this, task, model);
-
                 var fromTaskId = dependency.getFromTaskId();
                 var toTaskId = dependency.getToTaskId();
+                var toTask = dependency.getToTask();
+                var manager = dependency.manager;
 
                 if (!(fromTaskId in this.dependenciesFrom)) {
                     this.dependenciesFrom[fromTaskId] = [];
@@ -106,14 +110,25 @@
                     this.dependenciesTo[toTaskId] = [];
                 }
 
-                if (fromTaskId) {
-                    this.dependenciesFrom[fromTaskId].push(dependency);
-                }
+                 if (!isDraw && !toTask) {
+                    // Remove dependency
+                    this.removeDependency(dependency, true);
+                    dependency.canConnect = false;
 
-                if (toTaskId) {
-                    this.dependenciesTo[toTaskId].push(dependency);
+                    manager.api.dependencies.raise.remove(dependency);
+
+                    return null;
+                } else {
+                    if (fromTaskId) {
+                        this.dependenciesFrom[fromTaskId].push(dependency);
+                    }
+
+                    if (toTaskId) {
+                        this.dependenciesTo[toTaskId].push(dependency);
+                    }
                 }
 
+
                 return dependency;
             };