fix tracking projects

domain/project.go

@@ -104,6 +104,9 @@ func (p *Project) NewStage(stageName, nodeName, path string) (*Stage, error) {
 	}
 	p.Stages = append(p.Stages, stage)
 	stage.node.AddStage(stage.Name, p.Name, path)
+	if p.NumberOfStages() == 1 {
+		stage.node.workspace.AddProject(p.Name, path)
+	}
 	return stage, nil
 }
 
@@ -112,6 +115,9 @@ func (p *Project) RemoveStage(stageName string) {
 		if s.Name == stageName {
 			p.Stages = append(p.Stages[:idx], p.Stages[idx+1:]...)
 			s.node.RemoveStage(s.Name)
+			if p.NumberOfStages() == 0 {
+				s.node.workspace.RemoveProject(p.Name)
+			}
 		}
 	}
 	p.setDefaultStage()

domain/workspace.go

@@ -283,6 +283,7 @@ func (w *Workspace) RemoveProject(name string) {
 	for idx, p := range w.Projects {
 		if p.Name == name {
 			w.Projects = append(w.Projects[:idx], w.Projects[idx+1:]...)
+			return
 		}
 	}
 }
@@ -292,18 +293,12 @@ func (n *Node) AddStage(name, projectName, path string) {
 		Name:        name,
 		ProjectName: projectName,
 	})
-	if len(n.Stages) == 1 {
-		n.workspace.AddProject(projectName, path)
-	}
 }
 
 func (n *Node) RemoveStage(name string) {
 	for idx, s := range n.Stages {
 		if s.Name == name {
 			n.Stages = append(n.Stages[:idx], n.Stages[idx+1:]...)
-			if len(n.Stages) == 0 {
-				n.workspace.RemoveProject(s.ProjectName)
-			}
 			return
 		}
 	}