package gudusoft.gsqlparser.stmt.teradata;

import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.TCustomSqlStatement;
import gudusoft.gsqlparser.nodes.TParseTreeVisitor;
import gudusoft.gsqlparser.ESqlStatementType;

public class TTeradataMultiLoadExport extends TCustomSqlStatement {
    private CommandType commandType;
    private String filePath;
    private String delimiter = "|";  // default delimiter
    private String quote = "\"";     // default quote
    private String encoding = "UTF8"; // default encoding
    private boolean indicData = false;
    private Integer width;
    private String command;

    public enum CommandType {
        OUTFILE,
        RESET,
        UNKNOWN
    }

    public TTeradataMultiLoadExport(EDbVendor dbvendor, String exportCommand) {
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sstteradataMultiLoadExport;
        this.command = exportCommand;
        parse(exportCommand);
    }

    private void parse(String exportCommand) {
        if (exportCommand == null || exportCommand.trim().isEmpty()) {
            throw new IllegalArgumentException("Export command cannot be empty");
        }

        String command = exportCommand.trim();

        if (!command.startsWith(".EXPORT")) {
            throw new IllegalArgumentException("Invalid export command format. Must start with .EXPORT");
        }

        if (command.equals(".EXPORT RESET")) {
            this.commandType = CommandType.RESET;
            return;
        }

        if (command.startsWith(".EXPORT OUTFILE")) {
            this.commandType = CommandType.OUTFILE;
            parseOutfileCommand(command);
        } else {
            this.commandType = CommandType.UNKNOWN;
            throw new IllegalArgumentException("Unknown export command type");
        }
    }

    private void parseOutfileCommand(String command) {
        // Remove .EXPORT OUTFILE from the start
        String options = command.substring(".EXPORT OUTFILE".length()).trim();
        
        // Parse each option
        String[] parts = options.split("\\s+(?=(?:[^'\"]*['\"][^'\"]*['\"])*[^'\"]*$)");
        
        for (int i = 0; i < parts.length; i++) {
            String part = parts[i].trim();
            
            if (i == 0) {
                // First part should be the file path
                this.filePath = extractQuotedString(part);
                continue;
            }

            switch (part.toUpperCase()) {
                case "DELIMITER":
                    if (i + 1 < parts.length) {
                        this.delimiter = extractQuotedString(parts[++i]);
                    }
                    break;
                case "QUOTE":
                    if (i + 1 < parts.length) {
                        this.quote = extractQuotedString(parts[++i]);
                    }
                    break;
                case "ENCODING":
                    if (i + 1 < parts.length) {
                        this.encoding = extractQuotedString(parts[++i]);
                    }
                    break;
                case "INDICDATA":
                    this.indicData = true;
                    break;
                case "WIDTH":
                    if (i + 1 < parts.length) {
                        try {
                            this.width = Integer.parseInt(parts[++i].replace(";", ""));
                        } catch (NumberFormatException e) {
                            throw new IllegalArgumentException("Invalid WIDTH value");
                        }
                    }
                    break;
            }
        }

        if (this.filePath == null || this.filePath.isEmpty()) {
            throw new IllegalArgumentException("File path is required for OUTFILE command");
        }
    }

    private String extractQuotedString(String input) {
        input = input.replace(";", ""); // Remove trailing semicolon if present
        if (input.startsWith("'") && input.endsWith("'")) {
            return input.substring(1, input.length() - 1);
        }
        if (input.startsWith("\"") && input.endsWith("\"")) {
            return input.substring(1, input.length() - 1);
        }
        return input;
    }

    public String toCommandString() {
        if (commandType == CommandType.RESET) {
            return ".EXPORT RESET;";
        }

        if (commandType == CommandType.OUTFILE) {
            StringBuilder command = new StringBuilder();
            command.append(".EXPORT OUTFILE '").append(filePath).append("'");
            
            if (!"|".equals(delimiter)) {
                command.append(" DELIMITER '").append(delimiter).append("'");
            }
            if (!"\"".equals(quote)) {
                command.append(" QUOTE '").append(quote).append("'");
            }
            if (!"UTF8".equals(encoding)) {
                command.append(" ENCODING '").append(encoding).append("'");
            }
            if (indicData) {
                command.append(" INDICDATA");
            }
            if (width != null) {
                command.append(" WIDTH ").append(width);
            }
            command.append(";");
            
            return command.toString();
        }

        return "";
    }

    // Getters
    public CommandType getCommandType() { return commandType; }
    public String getFilePath() { return filePath; }
    public String getDelimiter() { return delimiter; }
    public String getQuote() { return quote; }
    public String getEncoding() { return encoding; }
    public boolean isIndicData() { return indicData; }
    public Integer getWidth() { return width; }

    // Setters
    public void setFilePath(String filePath) { this.filePath = filePath; }
    public void setDelimiter(String delimiter) { this.delimiter = delimiter; }
    public void setQuote(String quote) { this.quote = quote; }
    public void setEncoding(String encoding) { this.encoding = encoding; }
    public void setIndicData(boolean indicData) { this.indicData = indicData; }
    public void setWidth(Integer width) { this.width = width; }

    public void accept(TParseTreeVisitor v){
        v.preVisit(this);
        v.postVisit(this);
    }

    public void acceptChildren(TParseTreeVisitor v){
        v.preVisit(this);
        v.postVisit(this);
    }
}