TypePointerModelImpl.java

/*
 *  Copyright (c) 2012 Sam Harwell, Tunnel Vision Laboratories LLC
 *  All rights reserved.
 *
 *  The source code of this document is proprietary work, and is not licensed for
 *  distribution. For information about licensing, contact Sam Harwell at:
 *      sam@tunnelvisionlabs.com
 */
package org.tvl.goworks.editor.go.codemodel.impl;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import org.tvl.goworks.editor.go.codemodel.ParameterModel;
import org.tvl.goworks.editor.go.codemodel.TypeKind;
import org.tvl.goworks.editor.go.codemodel.TypeModel;
import org.tvl.goworks.editor.go.codemodel.TypePointerModel;

/**
 *
 * @author Sam Harwell
 */
public class TypePointerModelImpl extends TypeWrapperModelImpl implements TypePointerModel {

    public TypePointerModelImpl(TypeModelImpl elementType) {
        super("*" + elementType.getName(), elementType);
    }

    @Override
    public Collection<? extends TypeModelImpl> resolve() {
        if (isResolved()) {
            return Collections.singletonList(this);
        }

        List<TypeModelImpl> resolved = new ArrayList<>(getElementType().resolve());
        for (int i = 0; i < resolved.size(); i++) {
            resolved.set(i, new TypePointerModelImpl(resolved.get(i)));
        }

        return resolved;
    }

    @Override
    public TypeKind getKind() {
        return TypeKind.POINTER;
    }

    @Override
    public String getSimpleName() {
        return "*" + getElementType().getSimpleName();
    }

    @Override
    public PackageModelImpl getPackage() {
        return getElementType().getPackage();
    }

    @Override
    public Collection<FieldModelImpl> getFields() {
        if (getElementType().getKind() == TypeKind.INTRINSIC) {
            return Collections.emptyList();
        }

        return getElementType().getFields();
    }

    @Override
    public Collection<FieldModelImpl> getFields(String name) {
        if (getElementType().getKind() == TypeKind.INTRINSIC) {
            return Collections.emptyList();
        }

        return getElementType().getFields(name);
    }

    @Override
    public Collection<FunctionModelImpl> getMethods() {
        if (getElementType().getKind() == TypeKind.INTRINSIC) {
            return Collections.emptyList();
        }

        List<FunctionModelImpl> functions = new ArrayList<>();
        if (!isResolved()) {
            for (TypeModelImpl resolved : this.resolve()) {
                functions.addAll(resolved.getMethods());
            }
        } else {
            Collection<? extends FunctionModelImpl> packageFunctions = getElementType().getPackage().getFunctions();
            for (FunctionModelImpl function : packageFunctions) {
                ParameterModel receiver = function.getReceiverParameter();
                if (receiver == null) {
                    continue;
                }

                TypeModel receiverType = receiver.getVarType();
                if (this.equals(receiverType) || this.getElementType().equals(receiverType)) {
                    functions.add(function);
                }
            }
        }

        return Collections.unmodifiableList(functions);
    }

    @Override
    public Collection<FunctionModelImpl> getMethods(String name) {
        if (getElementType().getKind() == TypeKind.INTRINSIC) {
            return Collections.emptyList();
        }

        List<FunctionModelImpl> functions = new ArrayList<>();
        if (!isResolved()) {
            for (TypeModelImpl resolved : this.resolve()) {
                functions.addAll(resolved.getMethods(name));
            }
        } else {
            Collection<? extends FunctionModelImpl> packageFunctions = getElementType().getPackage().getFunctions(name);
            for (FunctionModelImpl function : packageFunctions) {
                ParameterModel receiver = function.getReceiverParameter();
                if (receiver == null) {
                    continue;
                }

                TypeModel receiverType = receiver.getVarType();
                if (this.equals(receiverType) || this.getElementType().equals(receiverType)) {
                    functions.add(function);
                }
            }
        }

        return Collections.unmodifiableList(functions);
    }

    @Override
    public boolean equals(Object obj) {
        if (!(obj instanceof TypePointerModel)) {
            return false;
        }

        return super.equals(obj);
    }

    @Override
    public int hashCode() {
        return super.hashCode();
    }

}