com.hadii.stiff.diagram.DiagramComponent Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of stiff-lib Show documentation
stiff-lib is a java library for generating stiff diagrams.
The newest version!
package com.hadii.stiff.diagram;

import com.hadii.clarpse.reference.ComponentReference;
import com.hadii.clarpse.sourcemodel.Component;
import com.hadii.clarpse.sourcemodel.OOPSourceCodeModel;
import com.hadii.clarpse.sourcemodel.OOPSourceModelConstants;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;

/**
 * Represents the building blocks of a Striff diagram.
 */
public class DiagramComponent {

    private final Component cmp;
    private final List children = new ArrayList<>();

    public DiagramComponent(Component cmp, OOPSourceCodeModel srcModel) {
        this.cmp = cmp;
        if (srcModel != null) {
            for (String child : cmp.children()) {
                Optional childCmp = srcModel.getComponent(child);
                if (childCmp.isPresent()) {
                    if (childCmp.get().componentType() == OOPSourceModelConstants.ComponentType.FIELD
                            || childCmp.get().componentType() == OOPSourceModelConstants.ComponentType.INTERFACE_CONSTANT) {
                        children.add(new DiagramComponent(childCmp.get(), srcModel).uniqueName());
                    } else {
                        children.add(child);
                    }
                }
            }
        }
    }

    public DiagramComponent(String componentName) {
        this(new Component(), new OOPSourceCodeModel());
        this.cmp.setComponentName(componentName);
    }

    public List children() {
        return this.children;
    }

    public String uniqueName() {
        if (this.cmp.componentType() == OOPSourceModelConstants.ComponentType.FIELD
                || this.cmp.componentType() == OOPSourceModelConstants.ComponentType.INTERFACE_CONSTANT) {
            return cmp.uniqueName() + "." + cmp.codeFragment();
        } else {
            return cmp.uniqueName();
        }
    }

    public List componentInvocations(OOPSourceModelConstants.TypeReferences implementation) {
        return cmp.references(implementation);
    }

    public Set modifiers() {
        return this.cmp.modifiers();
    }

    public OOPSourceModelConstants.ComponentType componentType() {
        return this.cmp.componentType();
    }

    public String parentUniqueName() {
        return this.cmp.parentUniqueName();
    }

    public Set componentInvocations() {
        return this.cmp.references();
    }

    public String name() {
        return this.cmp.name();
    }

    public String codeFragment() {
        return this.cmp.codeFragment();
    }

    public String comment() {
        return this.cmp.comment();
    }

    public String sourceFile() {
        return this.cmp.sourceFile();
    }

    public void setName(String name) {
        this.cmp.setName(name);
    }

    public String packageName() {
        return this.cmp.packageName();
    }

    /**
     * Fetches the current component's parent base component given the set of components in the code base.
     */
    public DiagramComponent parentBaseComponent(Map codeBase) {
        String currParentClassName = this.cmp.parentUniqueName();
        DiagramComponent parent;
        for (parent = codeBase.get(currParentClassName); parent != null && !parent.componentType().isBaseComponent();
             parent = codeBase.get(currParentClassName)) {
            currParentClassName = parent.parentUniqueName();
        }
        return parent;
    }

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

    @Override
    public boolean equals(Object obj) {
        if (!(obj instanceof DiagramComponent)) {
            return false;
        }
        DiagramComponent other = (DiagramComponent) obj;
        return other.uniqueName().equals(this.uniqueName());
    }
}