package sqlancer.spark;

import java.sql.SQLException;
import java.util.Arrays;
import java.util.List;

import com.beust.jcommander.Parameter;
import com.beust.jcommander.Parameters;

import sqlancer.DBMSSpecificOptions;
import sqlancer.OracleFactory;
import sqlancer.common.oracle.TLPWhereOracle;
import sqlancer.common.oracle.TestOracle;
import sqlancer.common.query.ExpectedErrors;
import sqlancer.spark.gen.SparkExpressionGenerator;

@Parameters(separators = "=", commandDescription = "Spark SQL (default port: " + SparkOptions.DEFAULT_PORT
        + ", default host: " + SparkOptions.DEFAULT_HOST + ")")
public class SparkOptions implements DBMSSpecificOptions<SparkOptions.SparkOracleFactory> {
    public static final String DEFAULT_HOST = "localhost";
    public static final int DEFAULT_PORT = 10000;

    @Parameter(names = "--oracle")
    public List<SparkOracleFactory> oracle = Arrays.asList(SparkOracleFactory.TLPWhere);

    public enum SparkOracleFactory implements OracleFactory<SparkGlobalState> {
        TLPWhere {
            @Override
            public TestOracle<SparkGlobalState> create(SparkGlobalState globalState) throws SQLException {
                SparkExpressionGenerator gen = new SparkExpressionGenerator(globalState);
                ExpectedErrors expectedErrors = ExpectedErrors.newErrors().with(SparkErrors.getExpressionErrors())
                        .build();

                return new TLPWhereOracle<>(globalState, gen, expectedErrors);
            }
        };
    }

    @Override
    public List<SparkOracleFactory> getTestOracleFactory() {
        return oracle;
    }
}