package gudusoft.gsqlparser.stmt;

import gudusoft.gsqlparser.*;
import gudusoft.gsqlparser.nodes.*;

/**
 * Represents cursor related statement, including:
 * <p>Cursor declaration
 * <p>Cursor specification
 * <p>Cursor body
 * <p>Ref cursor type definition
 */

public class TCursorDeclStmt extends TCustomSqlStatement {

    public enum CursorKind { declaration, specification, body, typeDefinition,resultsetName};
    private CursorKind cursorKind = CursorKind.declaration;

    public CursorKind getCursorKind() {
        return cursorKind;
    }

    public final static int     kind_cursor_declaration = 1;
    public final static int     kind_cursor_specification = 2;
    public final static int     kind_cursor_body = 3;
    public final static int     kind_ref_cursor_type_definition = 4;

    private int kind = kind_cursor_declaration;

    public void setKind(int kind) {
        this.kind = kind;
        switch (kind){
            case kind_cursor_declaration:
                this.cursorKind = CursorKind.declaration;
                break;
            case kind_cursor_specification:
                this.cursorKind = CursorKind.specification;
                break;
            case kind_cursor_body:
                this.cursorKind = CursorKind.body;
                break;
            case kind_ref_cursor_type_definition:
                this.cursorKind =  CursorKind.typeDefinition;
                break;
        }

    }

    /**
     * what's kind of SQL statement this class represents for.
     *
     * @deprecated since 2.5.0.9, please use {@link #getCursorKind()} instead.
     */
    public int getKind() {

        return kind;
    }

    public TCursorDeclStmt(){
        this(EDbVendor.dbvoracle);
    }
     public TCursorDeclStmt(EDbVendor dbvendor){
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sst_cursordecl;
        }

    void buildsql() {
    }

    void clear() {
    }

    /**
     * A SQL SELECT statement. If the cursor declaration declares parameters, each parameter must appear in select_statement.
     * @return
     */
    public TSelectSqlStatement getSubquery() {
        return subquery;
    }

    String getasprettytext() {
        return "";
    }

    void iterate(TVisitorAbs pvisitor) {
    }

    public int doParseStatement(TCustomSqlStatement psql) {
        super.doParseStatement(psql);
        
        if (this.cursorName != null){
            this.cursorName.setDbObjectType(EDbObjectType.cursor);
        }

         // push parameterDeclarations into symbolTable
         if (this.getCursorParameterDeclarations() != null){
             for(int i=0;i< this.getCursorParameterDeclarations().size();i++){
                this.getTopStatement().getSymbolTable().push( new TSymbolTableItem(TObjectName.ttobjParameter,this, this.getCursorParameterDeclarations().getParameterDeclarationItem(i)));
             }
         }

        if (subquery != null){
            subquery.parsestatement(this,false);
        }else if (selectSqlNode != null){
            //postgresql
            subquery = new TSelectSqlStatement(this.dbvendor);
            subquery.rootNode = selectSqlNode;
            subquery.doParseStatement(this);
        }

         // pop parameterDeclarations from symbolTable
         if (this.getCursorParameterDeclarations() != null){
             for(int i=0;i< this.getCursorParameterDeclarations().size();i++){
                this.getTopStatement().getSymbolTable().pop();
             }
         }

        return 0;
    }

    private TSelectSqlStatement subquery = null;
    private TSelectSqlNode selectSqlNode = null;

    private TObjectName resultsetName = null;//snowflake

    public TObjectName getResultsetName() {
        return resultsetName;
    }

    public void init(Object arg1)
    {
        if (arg1 instanceof TSelectSqlStatement){
            subquery = (TSelectSqlStatement)arg1;
        }else if (arg1 instanceof TSelectSqlNode){
            selectSqlNode = (TSelectSqlNode)arg1;
        }else if (arg1 instanceof TObjectName){
            resultsetName = (TObjectName)arg1;
        }
    }

    public void init(Object arg1,Object arg2){
        this.cursorKind = (CursorKind)arg1;
        switch (cursorKind){
            case declaration:
                if (arg1 instanceof TSelectSqlStatement){
                    subquery = (TSelectSqlStatement)arg1;
                }else if (arg1 instanceof TSelectSqlNode){
                    selectSqlNode = (TSelectSqlNode)arg1;
                }
                break;
            case resultsetName:
                resultsetName = (TObjectName)arg2;
                break;
            default:
                if (arg1 instanceof TSelectSqlStatement){
                    subquery = (TSelectSqlStatement)arg1;
                }else if (arg1 instanceof TSelectSqlNode){
                    selectSqlNode = (TSelectSqlNode)arg1;
                }
                break;
        }
    }

    private TObjectName cursorName;

    public void setCursorName(TObjectName cursorName) {
        this.cursorName = cursorName;
    }

    /**
     * Name of an explicit cursor.
     * @return
     */
    public TObjectName getCursorName() {

        return cursorName;
    }

    private TParameterDeclarationList cursorParameterDeclarations = null;

    /**
     * List of cursor parameter declaration, which is represented by {@link gudusoft.gsqlparser.nodes.TParameterDeclaration}.
     * @return
     */
    public TParameterDeclarationList getCursorParameterDeclarations() {
        return cursorParameterDeclarations;
    }

    public void setCursorParameterDeclarations(TParameterDeclarationList cursorParameterDeclarations) {
        this.cursorParameterDeclarations = cursorParameterDeclarations;
    }

    private TTypeName rowtype;

    public void setRowtype(TTypeName rowtype) {
        this.rowtype = rowtype;
    }

    /**
     * A record type that represents a row in a database table or a row fetched from a previously declared cursor or cursor variable.
     * @return
     */
    public TTypeName getRowtype() {

        return rowtype;
    }

    private TObjectName cursorTypeName;

    public void setCursorTypeName(TObjectName cursorTypeName) {
        this.cursorTypeName = cursorTypeName;
    }

    /**
     * Name of a REF CURSOR type. Valid only when kind = kind_ref_cursor_type_definition
     * To create a cursor variable, define a REF CURSOR type, and then declare the cursor
     * variable to be of that type. Declaring a cursor variable creates a pointer, not an item.
     * @return
     */
    public TObjectName getCursorTypeName() {

        return cursorTypeName;
    }

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

    public void acceptChildren(TParseTreeVisitor v){
        v.preVisit(this);
        if (getCursorParameterDeclarations() != null) getCursorParameterDeclarations().acceptChildren(v);
        if (subquery != null) subquery.acceptChildren(v);
        v.postVisit(this);
    }

    public void setSubquery(TSelectSqlStatement subquery) {
        this.subquery = subquery;
    }
}