com.github.leeonky.dal.ast.node.table.TransposedTableHead Maven / Gradle / Ivy

Go to download
package com.github.leeonky.dal.ast.node.table;

import com.github.leeonky.dal.ast.node.DALNode;
import com.github.leeonky.dal.ast.node.TableNode;
import com.github.leeonky.interpreter.SyntaxException;

import java.util.List;
import java.util.stream.Collectors;

import static com.github.leeonky.interpreter.InterpreterException.Position.Type.ROW;
import static java.util.Optional.empty;

public class TransposedTableHead extends DALNode {
    private final List prefixes;

    public TransposedTableHead(List prefixes) {
        this.prefixes = prefixes.stream().map(TableRowPrefixNode.class::cast).collect(Collectors.toList());
        if (!prefixes.isEmpty())
            setPositionBegin(prefixes.get(prefixes.size() - 1).getPositionBegin());
    }

    @Override
    public String inspect() {
        return "| >> " + TableNode.printLine(prefixes) + "\n";
    }

    public TableRowPrefixNode getPrefix(int i) {
        if (i >= prefixes.size())
            return new TableRowPrefixNode(empty(), empty(), empty());
        return prefixes.get(i);
    }

    public void checkSize(TransposedRowNode row) {
        if (row.cellCount() != prefixes.size())
            throw new SyntaxException("Different cell size", getPositionBegin(), ROW);
    }
}