package sqlancer.postgres;

import sqlancer.postgres.ast.PostgresAggregate;

import sqlancer.postgres.ast.PostgresBetweenOperation;

import sqlancer.postgres.ast.PostgresCastOperation;

import sqlancer.postgres.ast.PostgresCollate;

import sqlancer.postgres.ast.PostgresColumnValue;

import sqlancer.postgres.ast.PostgresConstant;

import sqlancer.postgres.ast.PostgresExpression;

import sqlancer.postgres.ast.PostgresFunction;

import sqlancer.postgres.ast.PostgresInOperation;

import sqlancer.postgres.ast.PostgresOrderByTerm;

import sqlancer.postgres.ast.PostgresPOSIXRegularExpression;

import sqlancer.postgres.ast.PostgresPostfixOperation;

import sqlancer.postgres.ast.PostgresPostfixText;

import sqlancer.postgres.ast.PostgresPrefixOperation;

import sqlancer.postgres.ast.PostgresSelect;

import sqlancer.postgres.ast.PostgresSelect.PostgresFromTable;

import sqlancer.postgres.ast.PostgresSimilarTo;

public final class PostgresExpectedValueVisitor implements PostgresVisitor {

private final StringBuilder sb = new StringBuilder();

private int nrTabs = 0;

private void print(PostgresExpression expr) {

PostgresToStringVisitor v = new PostgresToStringVisitor();

v.visit(expr);

for (int i = 0; i < nrTabs; i++) {

sb.append("\t");

}

sb.append(v.get());

sb.append(" -- " + expr.getExpectedValue());

sb.append("\n");

}

// @Override

// public void visit(PostgresExpression expr) {

// nrTabs++;

// try {

// super.visit(expr);

// } catch (IgnoreMeException e) {

//

// }

// nrTabs--;

// }

@Override

public void visit(PostgresConstant constant) {

print(constant);

}

@Override

public void visit(PostgresPostfixOperation op) {

print(op);

visit(op.getExpression());

}

public String get() {

return sb.toString();

}

@Override

public void visit(PostgresColumnValue c) {

print(c);

}

@Override

public void visit(PostgresPrefixOperation op) {

print(op);

visit(op.getExpression());

}

@Override

public void visit(PostgresSelect op) {

visit(op.getWhereClause());

}

@Override

public void visit(PostgresOrderByTerm op) {

}

@Override

public void visit(PostgresFunction f) {

print(f);

for (int i = 0; i < f.getArguments().length; i++) {

visit(f.getArguments()[i]);

}

}

@Override

public void visit(PostgresCastOperation cast) {

print(cast);

visit(cast.getExpression());

}

@Override

public void visit(PostgresBetweenOperation op) {

print(op);

visit(op.getExpr());

visit(op.getLeft());

visit(op.getRight());

}

@Override

public void visit(PostgresInOperation op) {

print(op);

visit(op.getExpr());

for (PostgresExpression right : op.getListElements()) {

visit(right);

}

}

@Override

public void visit(PostgresPostfixText op) {

print(op);

visit(op.getExpr());

}

@Override

public void visit(PostgresAggregate op) {

print(op);

for (PostgresExpression expr : op.getArgs()) {

visit(expr);

}

}

@Override

public void visit(PostgresSimilarTo op) {

print(op);

visit(op.getString());

visit(op.getSimilarTo());

if (op.getEscapeCharacter() != null) {

visit(op.getEscapeCharacter());

}

}

@Override

public void visit(PostgresPOSIXRegularExpression op) {

print(op);

visit(op.getString());

visit(op.getRegex());

}

@Override

public void visit(PostgresCollate op) {

print(op);

visit(op.getExpr());

}

@Override

public void visit(PostgresFromTable from) {

print(from);

}

}