org.snapscript.core.type.extend.ModuleExtender Maven / Gradle / Ivy

package org.snapscript.core.type.extend;

import static org.snapscript.core.function.Origin.SYSTEM;

import java.util.ArrayList;
import java.util.List;

import org.snapscript.core.Context;
import org.snapscript.core.error.InternalStateException;
import org.snapscript.core.function.Function;
import org.snapscript.core.module.Module;
import org.snapscript.core.scope.Scope;
import org.snapscript.core.type.TypeLoader;

public class ModuleExtender {
   
   private final List functions;
   private final Context context;
   
   public ModuleExtender(Context context) {
      this.functions = new ArrayList();
      this.context = context;
   }
   
   public synchronized void extend(Module module){
      List available = module.getFunctions();
      TypeLoader loader = context.getLoader();
      
      if(functions.isEmpty()) {
         FunctionExtractor extractor = new FunctionExtractor(loader, SYSTEM);
         ScopeExtension extension = new ScopeExtension(context);
         
         try {
            List list = extractor.extract(module, Scope.class, extension);
            
            for(Function function : list) {
               functions.add(function);
            }
         } catch(Exception e) {
            throw new InternalStateException("Could not export runtime", e);
         }
      }
      available.addAll(functions);
   }

}