package de.dhbwstuttgart.syntaxtree;

import de.dhbwstuttgart.core.IItemWithOffset;
import de.dhbwstuttgart.syntaxtree.type.RefType;
import de.dhbwstuttgart.syntaxtree.visual.ASTPrinter;
import de.dhbwstuttgart.typecheck.JavaClassName;
import de.dhbwstuttgart.syntaxtree.type.RefTypeOrTPHOrWildcardOrGeneric;
import de.dhbwstuttgart.typeinference.constraints.Constraint;
import de.dhbwstuttgart.typeinference.constraints.ConstraintSet;
import de.dhbwstuttgart.typeinference.assumptions.TypeInferenceInformation;
import org.antlr.v4.runtime.Token;

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

/**
 * Stellt jede Art von Klasse dar. Auch abstrakte Klassen und Interfaces
 */
public class ClassOrInterface extends SyntaxTreeNode {
  protected int modifiers;
  protected JavaClassName name;
  private List<Field> fields = new ArrayList<>();
  private List<Method> methods = new ArrayList<>();
  private GenericDeclarationList genericClassParameters;
  private RefTypeOrTPHOrWildcardOrGeneric superClass;
  protected boolean isInterface;
  private List<? extends RefTypeOrTPHOrWildcardOrGeneric> implementedInterfaces;
  private List<Constructor> constructors;

  public ClassOrInterface(int modifiers, JavaClassName name, List<Field> fielddecl, List<Method> methods, List<Constructor> constructors, GenericDeclarationList genericClassParameters,
                          RefTypeOrTPHOrWildcardOrGeneric superClass, Boolean isInterface, List<? extends RefTypeOrTPHOrWildcardOrGeneric> implementedInterfaces, Token offset){
    super(offset);
      this.modifiers = modifiers;
      this.name = name;
      this.fields = fielddecl;
    this.genericClassParameters = genericClassParameters;
      this.superClass = superClass;
    this.isInterface = isInterface;
      this.implementedInterfaces = implementedInterfaces;
    this.methods = methods;
    this.constructors = constructors;
  }
  
  // Gets class name
  public JavaClassName getClassName(){
    return this.name;
  }
  
  // Get modifiers
  public int getModifiers(){
    return this.modifiers;
  }

  public List<Field> getFieldDecl(){
    return this.fields;
  }
  public List<Method> getMethods(){
    return this.methods;
  }

  public ConstraintSet getConstraints(TypeInferenceInformation info) {
    ConstraintSet ret = new ConstraintSet();
    for(Method m : this.getMethods()){
      ret.addAll(m.getConstraints(info, this));
    }
    return ret;
  }

  public RefType getType() {
    return generateTypeOfClass(this.getClassName(), this.getGenerics(), this.getOffset());
  }

  public static RefType generateTypeOfClass(JavaClassName name, GenericDeclarationList genericsOfClass ,Token offset){
    return new RefType(name, offset);
  }

  public RefTypeOrTPHOrWildcardOrGeneric getSuperClass() {
    return superClass;
  }

  public GenericDeclarationList getGenerics() {
    return this.genericClassParameters;
  }

  public List<? extends Method> getConstructors() {
    return constructors;
  }

  @Override
  public void accept(ASTVisitor visitor) {
    visitor.visit(this);
  }
}