package gudusoft.gsqlparser.stmt;

import gudusoft.gsqlparser.*;
import gudusoft.gsqlparser.nodes.*;
import gudusoft.gsqlparser.sqlenv.TSQLEnv;
import gudusoft.gsqlparser.sqlenv.TSQLTable;

/**
 * Define a view, which is a logical table based on one or more tables or views.
 */

public class TCreateViewSqlStatement extends TCustomSqlStatement {
    private boolean isSecure  = false;
    private boolean isRecursive = false;
    private ETableKind tableKind = ETableKind.etkBase;

    public boolean isSecure() {
        return isSecure;
    }

    public boolean isRecursive() {
        return isRecursive;
    }

    public ETableKind getTableKind() {
        return tableKind;
    }

    public TObjectName getComment(){
        if (commentClause == null) return null;
        if (commentClause.getComment() == null) return null;
        return commentClause.getComment();
    }
    private TCreateTableOption commentClause;

//    public TCreateTableOption getCommentClause() {
//        return commentClause;
//    }

    private TSelectSqlStatement subquery = null;
    private TViewAliasClause viewAliasClause = null;
    private TSourceToken stForce = null;
    private TSourceToken stReplace = null;
    private TObjectNameList viewAttributeList;
    private boolean isForce = false;
    private boolean isNoForce = false;

    public void setForce(boolean isForce) {
        this.isForce = isForce;
    }

    public void setNoForce(boolean isNoForce) {
        this.isNoForce = isNoForce;
    }

    public boolean isForce() {

        return isForce;
    }

    public boolean isNoForce() {
        return isNoForce;
    }

    public void setViewAttributeList(TObjectNameList viewAttributeList) {
        this.viewAttributeList = viewAttributeList;
    }

    public TObjectNameList getViewAttributeList() {

        return viewAttributeList;
    }

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

    public void setViewAliasClause(TViewAliasClause viewAliasClause) {
        this.viewAliasClause = viewAliasClause;
    }

    public void setStForce(TSourceToken stForce) {
        this.stForce = stForce;
    }

    public void setStReplace(TSourceToken stReplace) {
        this.stReplace = stReplace;
    }

    public void setRowTypeName(TObjectName rowTypeName) {
        this.rowTypeName = rowTypeName;
    }

    public void setRestrictionClause(TRestrictionClause restrictionClause) {
        this.restrictionClause = restrictionClause;
    }

    public void setViewName(TObjectName viewName) {
        this.viewName = viewName;
    }

    private TObjectName rowTypeName;

    public TObjectName getRowTypeName() {
        return rowTypeName;
    }

    public TSourceToken getStForce() {
        return stForce;
    }

    public TSourceToken getStReplace() {
        return stReplace;
    }

    /**
     * Oracle, DB2 restriction clause
     * @return DB2 restriction clause
     */
    public TRestrictionClause getRestrictionClause() {
        return restrictionClause;
    }

    private TRestrictionClause restrictionClause = null;


    public TViewAliasClause getViewAliasClause() {
        return viewAliasClause;
    }

    public TCreateViewSqlStatement(EDbVendor dbvendor) {
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sstcreateview;
    }

    /**
     * 
     * @return   a subquery that identifies columns and rows of the table(s) that the view is based on.
     */
    public TSelectSqlStatement getSubquery() {
        return subquery;
    }

    void buildsql() {
    }

    void clear() {
    }

    String getasprettytext() {
        return "";
    }

    void iterate(TVisitorAbs pvisitor) {
    }

    public int doParseStatement(TCustomSqlStatement psql) {
        if (rootNode == null) return -1;
        TCreateViewSqlNode createViewSqlNode = (TCreateViewSqlNode)rootNode;
        super.doParseStatement(psql);
        this.viewName = createViewSqlNode.getViewName();
        this.viewName.setDbObjectType(EDbObjectType.view);
        this.isSecure = createViewSqlNode.isSecure();
        this.isRecursive = createViewSqlNode.isRecursive();
        this.tableKind = createViewSqlNode.getTableKind();
        this.viewName.setTableKind(this.tableKind);

        TSQLTable sqlView = null;
        if (getSqlEnv() != null)  {
            sqlView = getSqlEnv().addView(viewName.toString(),true);

//            if (getSqlEnv().getDefaultCatalogName() != null){
//                if (this.viewName.getDatabaseToken() == null){
//                    this.viewName.setDatabaseToken(new TSourceToken(getSqlEnv().getDefaultCatalogName()));
//                }
//            }

            this.viewName.setSqlEnv(getSqlEnv());
        }
        rowTypeName = createViewSqlNode.getRowTypeName();

        if (createViewSqlNode.getSelectSqlNode() != null) {
            // gaussdb create view using jdbc option does not support subquery
            // https://support.huaweicloud.com/sqlref-spark-dli/dli_08_0197.html
            subquery = new TSelectSqlStatement(this.dbvendor);
            subquery.rootNode = createViewSqlNode.getSelectSqlNode();
            subquery.doParseStatement(this);
        }

        if (createViewSqlNode.getViewAliasClause() != null){
            createViewSqlNode.getViewAliasClause().doParse(this, ESqlClause.viewAlias);
            this.viewAliasClause = createViewSqlNode.getViewAliasClause();
            if (sqlView != null){
                for(int i=0;i<viewAliasClause.getViewAliasItemList().size();i++){
                    TViewAliasItem item = viewAliasClause.getViewAliasItemList().getViewAliasItem(i);
                    if (item.getAlias() != null){
                        sqlView.addColumn(item.getAlias().toString());
                    }
                }
            }
        }else{
            if ((sqlView != null)&&(subquery != null)) {
                TSelectSqlStatement viewQuery = subquery;
                if (subquery.isCombinedQuery()){
                    viewQuery = subquery.getRightStmt();
                }
                viewQuery.addColumnInSelectListToSQLEnv(sqlView);
//                for(int i=0;i<viewQuery.getResultColumnList().size();i++){
//                    TResultColumn resultColumn = viewQuery.getResultColumnList().getResultColumn(i);
//                    String aliasName = null;
//                    if (resultColumn.getAliasClause() != null){
//                        aliasName = resultColumn.getAliasClause().getAliasName().toString();
//                    }else{
//                        if (resultColumn.getExpr().getObjectOperand() != null){
//                            aliasName = resultColumn.getExpr().getObjectOperand().toString();
//                        }
//                    }
//                    if (aliasName != null){
//                        sqlView.addColumn(TSQLEnv.getObjectName(aliasName));
//                    }
//                }
            }
        }

        this.restrictionClause = createViewSqlNode.getRestrictionClause();
        this.stForce = createViewSqlNode.getStForce();
        isForce = createViewSqlNode.isForce();
        isNoForce = createViewSqlNode.isNoForce();
        this.stReplace = createViewSqlNode.getStReplace();
        viewAttributeList = createViewSqlNode.getViewAttributeList();
        this.commentClause = createViewSqlNode.getCommentClause();

        return 0;
    }

    private TObjectName viewName = null;

    /**
     *
     * @return the name of the view or the object view.
     */
    public TObjectName getViewName() {
        return viewName;
    }

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

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

        if (restrictionClause != null) restrictionClause.acceptChildren(v);
        v.postVisit(this);
    }
}