package gudusoft.gsqlparser.dlineage.dataflow.model;

import java.util.*;

/**
 * Captures the output row structure and per-column upstream lineage of a pipelined table function.
 * This is the single source of truth for all stitching operations.
 */
public class PipelinedFunctionSignature {
    private String functionKey;
    private String nakedKey;
    private String procedureKey;
    private String rowTypeKey;
    private String collectionTypeKey;
    private int argCount;
    private boolean pipelined;
    private boolean resolved;
    private ResolutionStatus status;
    private List<String> outputColumns;
    private Map<String, List<PipelinedSourceRef>> lineageByOutputColumn;

    public enum ResolutionStatus {
        OK, AMBIGUOUS, UNRESOLVED, SKIPPED
    }

    public PipelinedFunctionSignature() {
        this.outputColumns = new ArrayList<>();
        this.lineageByOutputColumn = new LinkedHashMap<>();
        this.status = ResolutionStatus.UNRESOLVED;
        this.resolved = false;
        this.pipelined = true;
    }

    private Object functionIdentity;
    private Object nakedIdentity;

    /** The identity of the function's name as written, package included; see {@link #getFunctionKey()}. */
    public Object getFunctionIdentity() {
        return functionIdentity;
    }

    public void setFunctionIdentity(Object functionIdentity) {
        this.functionIdentity = functionIdentity;
    }

    /** The identity of the function's bare name as written. */
    public Object getNakedIdentity() {
        return nakedIdentity;
    }

    public void setNakedIdentity(Object nakedIdentity) {
        this.nakedIdentity = nakedIdentity;
    }

    /**
     * The function's name as written, package included: a label. The signature is registered under
     * {@link #getFunctionIdentity()} when it has one.
     */
    public String getFunctionKey() {
        return functionKey;
    }

    public void setFunctionKey(String functionKey) {
        this.functionKey = functionKey;
    }

    public String getNakedKey() {
        return nakedKey;
    }

    public void setNakedKey(String nakedKey) {
        this.nakedKey = nakedKey;
    }

    public String getProcedureKey() {
        return procedureKey;
    }

    public void setProcedureKey(String procedureKey) {
        this.procedureKey = procedureKey;
    }

    public String getRowTypeKey() {
        return rowTypeKey;
    }

    public void setRowTypeKey(String rowTypeKey) {
        this.rowTypeKey = rowTypeKey;
    }

    public String getCollectionTypeKey() {
        return collectionTypeKey;
    }

    public void setCollectionTypeKey(String collectionTypeKey) {
        this.collectionTypeKey = collectionTypeKey;
    }

    public int getArgCount() {
        return argCount;
    }

    public void setArgCount(int argCount) {
        this.argCount = argCount;
    }

    public boolean isPipelined() {
        return pipelined;
    }

    public void setPipelined(boolean pipelined) {
        this.pipelined = pipelined;
    }

    public boolean isResolved() {
        return resolved;
    }

    public void setResolved(boolean resolved) {
        this.resolved = resolved;
    }

    public ResolutionStatus getStatus() {
        return status;
    }

    public void setStatus(ResolutionStatus status) {
        this.status = status;
    }

    public List<String> getOutputColumns() {
        return outputColumns;
    }

    public void setOutputColumns(List<String> outputColumns) {
        this.outputColumns = outputColumns;
    }

    public Map<String, List<PipelinedSourceRef>> getLineageByOutputColumn() {
        return lineageByOutputColumn;
    }

    public void addSourceRef(String outputColumnNormalized, PipelinedSourceRef ref) {
        lineageByOutputColumn.computeIfAbsent(outputColumnNormalized, k -> new ArrayList<>()).add(ref);
    }

    @Override
    public String toString() {
        return "PipelinedFunctionSignature{" + functionKey + ", cols=" + outputColumns.size()
                + ", resolved=" + resolved + ", status=" + status + "}";
    }
}