001/**
002 *
003 * Copyright (c) 2014, the Railo Company Ltd. All rights reserved.
004 *
005 * This library is free software; you can redistribute it and/or
006 * modify it under the terms of the GNU Lesser General Public
007 * License as published by the Free Software Foundation; either 
008 * version 2.1 of the License, or (at your option) any later version.
009 * 
010 * This library is distributed in the hope that it will be useful,
011 * but WITHOUT ANY WARRANTY; without even the implied warranty of
012 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
013 * Lesser General Public License for more details.
014 * 
015 * You should have received a copy of the GNU Lesser General Public 
016 * License along with this library.  If not, see <http://www.gnu.org/licenses/>.
017 * 
018 **/
019/**
020 * Implements the CFML Function queryaddcolumn
021 */
022package lucee.runtime.functions.query;
023
024import lucee.runtime.PageContext;
025import lucee.runtime.exp.PageException;
026import lucee.runtime.functions.BIF;
027import lucee.runtime.op.Caster;
028import lucee.runtime.type.Array;
029import lucee.runtime.type.ArrayImpl;
030import lucee.runtime.type.Collection;
031import lucee.runtime.type.KeyImpl;
032import lucee.runtime.type.Query;
033import lucee.runtime.type.QueryColumn;
034
035public final class QuerySetColumn extends BIF {
036
037        private static final long serialVersionUID = -268309857190767441L;
038
039        public static String call(PageContext pc , Query query, String columnName,String newColumnName) throws PageException {
040                columnName=columnName.trim();
041                newColumnName=newColumnName.trim();
042                Collection.Key src=KeyImpl.getInstance(columnName);
043                Collection.Key trg=KeyImpl.getInstance(newColumnName);
044                
045                Query qp = Caster.toQuery(query,null);
046                if(qp!=null) qp.rename(src, trg);
047                else {
048                        QueryColumn qc = query.removeColumn(src);
049                        Array content=new ArrayImpl();
050                        int len=qc.size();
051                        for(int i=1;i<=len;i++){
052                                content.setE(i, qc.get(i,null));
053                        }
054                        query.addColumn(trg, content, qc.getType());
055                }
056                return null;
057        }
058        
059        @Override
060        public Object invoke(PageContext pc, Object[] args) throws PageException {
061                return call(pc,Caster.toQuery(args[0]),Caster.toString(args[1]),Caster.toString(args[2]));
062        }
063}