/*
 * Copyright 2013 Guidewire Software, Inc.
 */

package gw.plugin.ij.lang.psi.stubs.elements;

import com.intellij.lang.ASTNode;
import com.intellij.psi.stubs.IndexSink;
import com.intellij.psi.stubs.StubElement;
import com.intellij.psi.stubs.StubInputStream;
import com.intellij.psi.stubs.StubOutputStream;
import com.intellij.util.Function;
import com.intellij.util.containers.ContainerUtil;
import com.intellij.util.io.StringRef;
import gw.lang.parser.statements.ISuperTypeClause;
import gw.plugin.ij.lang.parser.GosuCompositeElement;
import gw.plugin.ij.lang.parser.GosuElementTypes;
import gw.plugin.ij.lang.psi.GosuStubElementType;
import gw.plugin.ij.lang.psi.api.statements.typedef.IGosuExtendsClause;
import gw.plugin.ij.lang.psi.api.types.IGosuCodeReferenceElement;
import gw.plugin.ij.lang.psi.impl.statements.typedef.GosuExtendsClauseImpl;
import gw.plugin.ij.lang.psi.stubs.GosuReferenceListStub;
import gw.plugin.ij.lang.psi.stubs.impl.GosuReferenceListStubImpl;
import gw.plugin.ij.lang.psi.stubs.index.GosuDirectInheritorsIndex;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.io.IOException;

public class GosuExtendsClauseElementType extends GosuStubElementType<GosuReferenceListStub, IGosuExtendsClause> {
  public GosuExtendsClauseElementType() {
    super("SuperTypeClause", ISuperTypeClause.class);
  }

  @NotNull
  public IGosuExtendsClause createElement(ASTNode node) {
    return new GosuExtendsClauseImpl((GosuCompositeElement) node);
  }

  @NotNull
  public IGosuExtendsClause createPsi(@NotNull GosuReferenceListStub stub) {
    return new GosuExtendsClauseImpl(stub);
  }

  @NotNull
  public GosuReferenceListStub createStub(@NotNull IGosuExtendsClause psi, StubElement parentStub) {
    final IGosuCodeReferenceElement[] elements = psi.getReferenceElements();
    String[] refNames = ContainerUtil.map(elements,
        new Function<IGosuCodeReferenceElement, String>() {
          @Nullable
          public String fun(@NotNull final IGosuCodeReferenceElement element) {
            return element.getReferenceName();
          }
        }, new String[elements.length]);

    return new GosuReferenceListStubImpl(parentStub, GosuElementTypes.EXTENDS_CLAUSE, refNames);
  }

  public void serialize(@NotNull GosuReferenceListStub stub, @NotNull StubOutputStream dataStream) throws IOException {
    final String[] names = stub.getBaseClasses();
    dataStream.writeByte(names.length);
    for (String s : names) {
      dataStream.writeName(s);
    }
  }

  @NotNull
  public GosuReferenceListStub deserialize(@NotNull StubInputStream dataStream, StubElement parentStub) throws IOException {
    final byte b = dataStream.readByte();
    final String[] names = new String[b];
    for (int i = 0; i < b; i++) {
      StringRef name = dataStream.readName();
      names[i] = name == null ? null : name.toString();
    }
    return new GosuReferenceListStubImpl(parentStub, GosuElementTypes.EXTENDS_CLAUSE, names);
  }

  public void indexStub(@NotNull GosuReferenceListStub stub, @NotNull IndexSink sink) {
    for (String name : stub.getBaseClasses()) {
      if (name != null) {
        sink.occurrence(GosuDirectInheritorsIndex.KEY, name);
      }
    }
  }
}