grammar org.xtext.example.forest.Forest with org.eclipse.xtext.common.Terminals

generate forest "http://www.xtext.org/example/forest/Forest"

Country:
	"country" name=ID ("extends" supercountry=[Country|QualifiedName])? "{"
	forests=Forest
	"}"	;
Forest:	
	"forest" "{"
	("park" parks+=Park ("," parks+=Park)*)*
	("workingarea"  workingarea=WorkingArea)*
	"}";
Park:
     name=ID ("{"
	"enter" enter+=Enter
	("park" parks+=Park ("," parks+=Park)*)*
	("workingarea"  workingarea=WorkingArea)*
	"}")* ;
WorkingArea:
	name=ID "from" from=[ParkEnter|QualifiedName]  ;
Enter:
	name=ID;
ParkEnter:
	Park | Enter;
QualifiedName:
 ID ('.' ID)*;

park P4 {
	enter EN3 
	park P5
	workingarea W4 from P4.EN3 // Here it should only  be EN3, not P4.EN3 because EN3 and W4 are both contained by P4
				}

country C1 {
	forest {
		park P1 {
			enter EN1			
			park P2 {
				enter EN2			
				park P3 
				park P4 {
					enter EN3 
					park P5
					workingarea W4 from P4.EN3 // Only EN3, not P4.EN3 because EN3 and W4 are both contained by P4
				}
			 workingarea W2 from P4.EN3 // Here it is fine as it is because P4 and W2 are both contained inside P2 and EN3 is contained inside P4
			}
         workingarea W1 from P1.EN1 // Only EN1, not P1.EN1 because EN1 and W1 are both contained by P1
		}
		park P6 {
			enter EN4 
			}
		workingarea WF from P5.P5 //here it should be P1.P2.P4.P5
  } 
} 

public class ForestScopeProvider extends AbstractForestScopeProvider {
	@Override
	public IScope getScope(EObject context, EReference reference) {
	    
	   if (reference == ForestPackage.Literals.WORKING_AREA__FROM) {
	        Country rootCountry = (Country) EcoreUtil2.getRootContainer(context);

	       List<EObject> workingarea = new ArrayList<>();
	        while (rootCountry != null) {
	        			workingarea.addAll(EcoreUtil2.getAllContentsOfType(rootCountry, ParkEnter.class));   		        
	        	rootCountry = rootCountry.getSupercountry();	        	
	        }
	       System.out.println(workingarea);
	        
	       Function<EObject, QualifiedName> nameComputation = (EObject e) -> {
	   
	    	   if (e instanceof Enter)
	    	   {		  
	    		 Enter en = (Enter) e;
	             Park p = EcoreUtil2.getContainerOfType(en, Park.class);
	            
	    	      return QualifiedName.create(p.getName(), en.getName());
	    	   }
	    	   
	    	   else if (e instanceof Park)
	    	   {		  
	    		 Park p1 = (Park) e;
	             Park p2 = EcoreUtil2.getContainerOfType(p1, Park.class);
	            
	    	      return QualifiedName.create(p2.getName(), p1.getName());
	    	   }
	    	   return QualifiedName.create(((ParkEnter)e).getName());    
	       };
	       return Scopes.scopeFor(workingarea, nameComputation, IScope.NULLSCOPE);
	    }	  
	    return super.getScope(context, reference); 
	}
}

public class ForestScopeProvider extends AbstractForestScopeProvider {
	@Override
	public IScope getScope(EObject context, EReference reference) {
	    
	   if (reference == ForestPackage.Literals.WORKING_AREA__FROM) {
	        Country rootCountry = (Country) EcoreUtil2.getRootContainer(context);

	       List<EObject> workingarea = new ArrayList<>();
	        while (rootCountry != null) {
	        			workingarea.addAll(EcoreUtil2.getAllContentsOfType(rootCountry, ParkEnter.class));   		        
	        	rootCountry = rootCountry.getSupercountry();	        	
	        }
	       System.out.println(workingarea);
	        
	       Function<EObject, QualifiedName> nameComputation = (EObject e) -> {
	    	   if (e instanceof Enter)
	    	   {		  
	    		 Enter en = (Enter) e;
	             Park parkTheEnterIsIn = (Park)context.eContainer();
	             
	             List <String> allParks=new ArrayList<>();
	             EObject p = en.eContainer();
	             while (p instanceof Park && !p.equals(parkTheEnterIsIn)) {
	             allParks.add(((Park) p).getName());
	             p = p.eContainer();
	             }
	             QualifiedName nameWithAllParks = QualifiedName.create(ListExtensions.reverse(allParks)).append(en.getName());
		    	   return nameWithAllParks;      
	    	   }
	    	   else if (e instanceof Park)
	    	   {		  
	    		 Park p = (Park) e;
	             Park parkTheParkIsIn = (Park)context.eContainer();
	             
	             List <String> allParks=new ArrayList<>();
	             EObject pl = p.eContainer();
	             while (pl instanceof Park && !p.equals(parkTheParkIsIn)) {
	             allParks.add(((Park) pl).getName());
	             pl = pl.eContainer();
	             }
	             while (pl instanceof Park&& !pl.equals(parkTheParkIsIn)) {
		             allParks.add(((Park) pl).getName());
		             pl = pl.eContainer();
		             }
	             QualifiedName nameWithAllParks = QualifiedName.create(ListExtensions.reverse(allParks)).append(p.getName());
		    	   return nameWithAllParks;      
	    	   }
	    	   return QualifiedName.create(((ParkEnter)e).getName()); 
	       };
	
	       return Scopes.scopeFor(workingarea, nameComputation, IScope.NULLSCOPE);
	    }	  
	    return super.getScope(context, reference); 
	}
}

country C1 {
	forest {
		park P1 {
			enter EN1			
			park P2 {
				enter EN2			
				park P3 
				park P4 {
					enter EN3 
					park P5
					workingarea W4 from P1.P2.P4.P5 // Enter is fine but the park should be P5 not P1.P2.P5.P5
				}
			 workingarea W2 from  P4.EN3   //  Enter is fine but in park I get P1.P2.P3 instead of P3, P1.P2.P4 instead of P4, P1.P2.P4.P5 instead of P4.P5
			}
         workingarea W1 from  P2.EN2 // Enter is fine but in park I get P1.P2 instead of P2, P1.P2.P3 instead of P2.P3, P1.P2.P4 instead of P2.P4, P1.P2.P4.P5 instead of P2.P4.P5
		}
		park P6 {
			enter EN4 
			}
		workingarea WF from  //here I don't get any suggestions nor for the enter and not for parks. Maybe because we only consider when the container is a park while here it is a forest.
  } 
} 

public class ForestScopeProvider extends AbstractForestScopeProvider {
	@Override
	public IScope getScope(EObject context, EReference reference) {
	    
	   if (reference == ForestPackage.Literals.WORKING_AREA__FROM) {
	        Country rootCountry = (Country) EcoreUtil2.getRootContainer(context);

	       List<EObject> workingarea = new ArrayList<>();
	        while (rootCountry != null) {
	        			workingarea.addAll(EcoreUtil2.getAllContentsOfType(rootCountry, ParkEnter.class));   		        
	        	rootCountry = rootCountry.getSupercountry();	        	
	        }
	       System.out.println(workingarea);
	        
	       Function<EObject, QualifiedName> nameComputation = (EObject e) -> {
	    	   if (e instanceof Enter)
	    	   {		  
	    		 Enter en = (Enter) e;
	    		 if (context.eContainer() instanceof Park) {
	             Park parkTheEnterIsIn = (Park)context.eContainer();
	         
	             List <String> allParks=new ArrayList<>();
	             EObject p = en.eContainer();
	             while (p instanceof Park && !p.equals(parkTheEnterIsIn)) {
	             allParks.add(((Park) p).getName());
	             p = p.eContainer();
	             }
	             QualifiedName nameWithAllParks = QualifiedName.create(ListExtensions.reverse(allParks)).append(en.getName());
		    	   return nameWithAllParks;      
	    	   }
	    	 }
	    	   else if (e instanceof Park)
	    	   {		  
	    		 Park p = (Park) e;
	    		 
	    		 if (context.eContainer() instanceof Park)
	    		 {
	             Park parkTheParkIsIn = (Park)context.eContainer();
	             
	             List <String> allParks=new ArrayList<>();
	             EObject pl = p.eContainer();
	             while (pl instanceof Park && !pl.equals(parkTheParkIsIn)) {
	             allParks.add(((Park) pl).getName());
	             pl = pl.eContainer();
	             }
	            
	             QualifiedName nameWithAllParks = QualifiedName.create(ListExtensions.reverse(allParks)).append(p.getName());
		    	   return nameWithAllParks;  
	    		 }
	    		   return QualifiedName.create(((ParkEnter)e).getName()); 
	    	   }
	    	   return QualifiedName.create(((ParkEnter)e).getName()); 
	       };
	       
   
	       return Scopes.scopeFor(workingarea, nameComputation, IScope.NULLSCOPE);
	    }	  
	    return super.getScope(context, reference); 
	}
}