tw.yukina.notion.sdk.builder.RichTextBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of notion-sdk-java Show documentation

Notion in Java

There is a newer version: 1.0.1

package tw.yukina.notion.sdk.builder;

import org.jetbrains.annotations.NotNull;
import tw.yukina.notion.sdk.model.TextColor;
import tw.yukina.notion.sdk.model.common.rich.Annotation;
import tw.yukina.notion.sdk.model.common.rich.RichText;

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

public abstract class RichTextBuilder {

    private final List> childRichText = new ArrayList<>();

    private boolean isBold = false;

    private boolean isItalic = false;

    private boolean isStrikethrough = false;

    private boolean isUnderline = false;

    private boolean isCode = false;

    private TextColor color = TextColor.DEFAULT;

    public RichTextBuilder setBold(boolean isBold){
        this.isBold = isBold;
        return this;
    }

    public RichTextBuilder setItalic(boolean isItalic){
        this.isItalic = isItalic;
        return this;
    }

    public RichTextBuilder setStrolethrough(boolean isStrikethrough){
        this.isStrikethrough = isStrikethrough;
        return this;
    }

    public RichTextBuilder setUnderline(boolean isUnderline){
        this.isUnderline = isUnderline;
        return this;
    }

    public RichTextBuilder setCodeStyle(boolean isCode){
        this.isCode = isCode;
        return this;
    }

    public RichTextBuilder setColor(@NotNull TextColor color){
        this.color = color;
        return this;
    }

    public RichTextBuilder append(RichTextBuilder richTextBuilder){
        this.childRichText.add(richTextBuilder);
        return this;
    }

    public List build(){
        List richTexts = new ArrayList<>();
        richTexts.add(buildSelf());

        for(RichTextBuilder richTextBuilder: childRichText){
            richTexts.addAll(richTextBuilder.build());
        }
        return richTexts;
    }

    public abstract RichText buildSelf();

    protected void setRichTextStyle(@NotNull T richText){
        Annotation annotation = new Annotation();
        annotation.setBold(this.isBold);
        annotation.setItalic(this.isItalic);
        annotation.setStrikethrough(this.isStrikethrough);
        annotation.setUnderline(this.isUnderline);
        annotation.setCode(this.isCode);
        annotation.setColor(this.color);
        richText.setAnnotations(annotation);
    }
}