com.vladsch.flexmark.ext.zzzzzz.internal.ZzzzzzRepository Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of flexmark-ext-zzzzzz Show documentation

flexmark-java extension for zzzzzz

There is a newer version: 0.64.8

package com.vladsch.flexmark.ext.zzzzzz.internal;

import com.vladsch.flexmark.ext.zzzzzz.Zzzzzz;
import com.vladsch.flexmark.ext.zzzzzz.ZzzzzzBlock;
import com.vladsch.flexmark.ext.zzzzzz.ZzzzzzExtension;
import com.vladsch.flexmark.util.ast.KeepType;
import com.vladsch.flexmark.util.ast.Node;
import com.vladsch.flexmark.util.ast.NodeRepository;
import com.vladsch.flexmark.util.data.DataHolder;
import com.vladsch.flexmark.util.data.DataKey;
import org.jetbrains.annotations.NotNull;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

@SuppressWarnings("WeakerAccess")
public class ZzzzzzRepository extends NodeRepository {
    private ArrayList referencedZzzzzzBlocks = new ArrayList<>();

    public ZzzzzzRepository(DataHolder options) {
        super(ZzzzzzExtension.ZZZZZZS_KEEP.get(options));
    }

    public void addZzzzzzReference(ZzzzzzBlock zzzzzzBlock, Zzzzzz zzzzzz) {
        if (!zzzzzzBlock.isReferenced()) {
            referencedZzzzzzBlocks.add(zzzzzzBlock);
        }

        zzzzzzBlock.setFirstReferenceOffset(zzzzzz.getStartOffset());
    }

    public void resolveZzzzzzOrdinals() {
        // need to sort by first referenced offset then set each to its ordinal position in the array+1
        referencedZzzzzzBlocks.sort((f1, f2) -> f1.getFirstReferenceOffset() - f2.getFirstReferenceOffset());
        int ordinal = 0;
        for (ZzzzzzBlock zzzzzzBlock : referencedZzzzzzBlocks) {
            zzzzzzBlock.setZzzzzzOrdinal(++ordinal);
        }
    }

    public List getReferencedZzzzzzBlocks() {
        return referencedZzzzzzBlocks;
    }

    @NotNull
    @Override
    public DataKey getDataKey() {
        return ZzzzzzExtension.ZZZZZZS;
    }

    @NotNull
    @Override
    public DataKey getKeepDataKey() {
        return ZzzzzzExtension.ZZZZZZS_KEEP;
    }

    @NotNull
    @Override
    public Set getReferencedElements(Node parent) {
        HashSet references = new HashSet<>();
        //visitNodes(parent, new ValueRunnable() {
        //    @Override
        //    public void run(Node value) {
        //        if (value instanceof Zzzzzz) {
        //            //Reference reference = ((RefNode) value).getReferenceNode(ZzzzzzRepository.this);
        //            //if (reference != null) {
        //            //    references.add(reference);
        //            //}
        //        }
        //    }
        //}, Zzzzzz.class);
        return references;
    }
}